不同非类型模板参数的不同成员函数

Question

我有一个Vector 课程和一个Matrix 课程。我想使用 = 运算符将 Vector 类隐式转换为 Matrix 类。类的定义如下所示。

template <unsigned int dim>
class Vector {
public:
  std::array<double, dim> x;
};

template <unsigned int nrow, unsigned int ncol>
class Matrix {
public:
  std::array<Vector<ncol>, nrow> x;

  auto& operator=(Vector<ncol> v)
  {
    x.fill(v);
    return *this;
  }
};

int main()
{
  Vector<2> vec1;
  vec1.x[0] = 1;
  vec1.x[1] = 2;

  Matrix<1, 2> mat1;
  mat1 = vec1;

  std::cout << mat1.x[0].x[0] << std::endl;
  std::cout << mat1.x[0].x[1] << std::endl;
}

这可以正常工作。但是，我想介绍另一个operator=，它应该如下所示。

auto& operator=(Vector<nrow> v)
{
  for(unsigned int row=0; row < nrow; ++row){
    x[row].x[0] = v[row];
  }
  return *this;
}

但是，由于重载实例化为相同的签名（对于nrow=ncol 的情况），我不确定如何设计这样的方法。具体来说，我想知道是否只有nrow!=ncol才可以定义第二种方法。

我已经尝试通过定义两种类型来使用std::enable_if

using nrow_t = std::integral_constant<unsigned int,nrow>;
using ncol_t = std::integral_constant<unsigned int,ncol>;

然后使用std::is_same，但我不确定如何继续。

Answer 1

这可能是使用命名函数会容易得多的情况。它将准确地传达正在发生的事情，不会让任何人感到惊讶。也就是说你可以使用这样的东西

template <unsigned int nrow, unsigned int ncol>
class Matrix {
public:
    std::array<Vector<ncol>, nrow> x;

    template<unsigned int cols,
             std::enable_if_t<cols == ncol || (cols == ncol && ncol == nrow)>* = nullptr>
    auto& operator=(Vector<cols> v)
    {
        x.fill(v);
        return *this;
    }
    template<unsigned int rows,
             std::enable_if_t<rows == nrow && nrow != ncol>* = nullptr>
    auto& operator=(Vector<rows> v)
    {
        for(unsigned int row=0; row < nrow; ++row){
            x[row].x[0] = v[row];
        }
        return *this;
    }
};

当cols == ncol || (cols == ncol && ncol == nrow) 为真时，我们enable_if 第一次重载。当它具有正确的数量时会发生这种情况，因为列或行和列相同并且它具有正确的列数量。第二个重载使用rows == nrow && nrow != ncol，并且仅在行和列不相同且行数匹配时才会激活。