我刚开始在我的代码中使用模板元编程。我有一个类,它的成员是一个多维笛卡尔点的向量。以下是该类的基本设置:
template<size_t N>
class TGrid{
public:
void round_points_3(){
for(std::size_t i = 0; i < Xp.size();i++){
Xp[i][0] = min[0] + (std::floor((Xp[i][0] - min[0]) * nbins[0] / (max[0] - min[0])) * bin_w[0]) + bin_w[0]/2.0;
Xp[i][1] = min[1] + (std::floor((Xp[i][1] - min[1]) * nbins[1] / (max[1] - min[1])) * bin_w[1]) + bin_w[1]/2.0;
Xp[i][2] = min[2] + (std::floor((Xp[i][2] - min[2]) * nbins[2] / (max[2] - min[2])) * bin_w[2]) + bin_w[2]/2.0;
}
}
void round_points_2(){
for(std::size_t i = 0; i < Xp.size();i++){
Xp[i][0] = min[0] + (std::floor((Xp[i][0] - min[0]) * nbins[0] / (max[0] - min[0])) * bin_w[0]) + bin_w[0]/2.0;
Xp[i][1] = min[1] + (std::floor((Xp[i][1] - min[1]) * nbins[1] / (max[1] - min[1])) * bin_w[1]) + bin_w[1]/2.0;
}
}
void round_points_1(){
for(std::size_t i = 0; i < Xp.size();i++){
Xp[i][0] = min[0] + (std::floor((Xp[i][0] - min[0]) * nbins[0] / (max[0] - min[0])) * bin_w[0]) + bin_w[0]/2.0;
}
}
public:
std::vector<std::array<double,N> > Xp;
std::vector<double> min, max, nbins, bin_w;
};
这个类表示一个多维网格,维度由模板值N指定。我将进行许多操作,通过具有针对特定维度的模板特定成员函数(如循环展开)可以使这些操作更加高效。
在TGrid类中,我有3个特定于D = 1,D = 2和D = 3的函数。这是由函数的下标_1,_2和_3表示的。
我正在寻找一种面向模板元编程的方法来更紧凑地编写这三个函数。
我已经看过循环展开的示例,但所有这些示例都没有考虑模板类的成员函数。
round_points
和第二个循环。array::size()
是 constexpr 的,编译器会自行决定是否展开循环是一个好主意。 - pmr