在Eigen中，如何将多个映射矩阵组合为矩阵阵列

Question

我有一个具有多个映射矩阵作为成员的类。的代码如下所示：

class MyClass { 
public: 
    MyClass(int nrow, int ncol, int nmat) : 
    arr(new int [nmat*nrow*ncol]), 
    mat1(arr, nrow, ncol, Stride<Dynamic, Dynamic>(nrow * nmat, nmat)), 
    mat2(arr + 1, nrow, ncol, Stride<Dynamic, Dynamic>(nrow * nmat, nmat)) {} 

    int *arr; 
    Map<MatrixXi, 0, Stride<Dynamic, Dynamic> > mat1; 
    Map<MatrixXi, 0, Stride<Dynamic, Dynamic> > mat2; 
}; 

int main() 
{ 
    int nmat = 2;  // number of matrices 
    int nrow = 3;  // number of rows for each matrix 
    int ncol = 4;  // number of columns for each matrix 
    int arr_size = nmat*nrow*ncol; // size of the array. 

    MyClass A (nrow, ncol, nmat); 
    for (int i=0; i<arr_size; i++) { 
    A.arr[i] = i+1; 
    } 

    cout <<"mat1 = \n"<<A.mat1<<endl; 
    cout <<"mat2 = \n"<<A.mat2<<endl; 

    return 0; 
} 

现在我想使mat1和mat2是两个MATRICS的阵列，如mat[0]和mat[1]。我试图做这样的事情：

class MyClass { 
public: 
    MyClass(int nrow, int ncol, int nmat) : 
    arr(new int [nmat*nrow*ncol]), 
    mat[0](arr, nrow, ncol, Stride<Dynamic, Dynamic>(nrow * nmat, nmat)), 
    mat[1](arr + 1, nrow, ncol, Stride<Dynamic, Dynamic>(nrow * nmat, nmat)) {} 

    int *arr; 
    Map<MatrixXi, 0, Stride<Dynamic, Dynamic> > mat[2]; 
}; 

它显然不工作，因为我不能在初始化程序列表中执行此操作。但要做这种跨步的映射，我必须使用初始化列表。任何人都有如何实现这个想法？

Answer 1

用C++ 11，你可以做到这一点

class MyClass { 
public: 
    MyClass(int* arr, int nrow, int ncol, int nmat) : 
     mat { { arr, nrow, ncol, Stride<Dynamic, Dynamic>(nrow * nmat, nmat) }, 
      { arr + 1, nrow, ncol, Stride<Dynamic, Dynamic>(nrow * nmat, nmat) } } { 
    std::cout << mat[1].data() - arr << std::endl; 
    std::cout << mat[1].rows() << std::endl; 
    std::cout << mat[1].cols() << std::endl; 
    std::cout << mat[1].innerStride() << std::endl; 
    std::cout << mat[1].outerStride() << std::endl; 
    } 

    Map<MatrixXi, 0, Stride<Dynamic, Dynamic> > mat[2]; 
}; 

应该在以下链接列表初始化的情况下，5。

http://en.cppreference.com/w/cpp/language/list_initialization

在另一方面，Eigen::Map仅仅是一个重量轻的包装器原始指针和尺寸。您可能需要将arr，nrow，ncol和nmat作为类成员，并在运行中构造Eigen::Map。