54 lines
854 B
C++
54 lines
854 B
C++
#include <cstdio>
|
|
#include <iostream>
|
|
using namespace std;
|
|
#include "transpose_blocks.h"
|
|
using namespace igl;
|
|
|
|
int main(int argc,char * argv[])
|
|
{
|
|
// hieght of block
|
|
int m = 2;
|
|
// width of block
|
|
int n = 4;
|
|
// number of blocks
|
|
int k = 3;
|
|
// dimension blocks run along
|
|
int dim = 1;
|
|
|
|
// Input
|
|
Eigen::MatrixXd A;
|
|
|
|
if(dim == 1)
|
|
{
|
|
A.resize(m*k,n);
|
|
}else{
|
|
A.resize(m,n*k);
|
|
}
|
|
|
|
// loop over blocks
|
|
for(int b = 0;b<k;b++)
|
|
{
|
|
for(int i = 0;i<m;i++)
|
|
{
|
|
for(int j = 0;j<n;j++)
|
|
{
|
|
if(dim == 1)
|
|
{
|
|
A(b*m+i,j) = 100*b + i*n + j;
|
|
}else// dim == 2
|
|
{
|
|
A(i,b*n+j) = 100*b + i*n + j;
|
|
}
|
|
}
|
|
}
|
|
}
|
|
cout<<"A=["<<endl<<A<<endl<<"];"<<endl;
|
|
|
|
Eigen::MatrixXd B;
|
|
transpose_blocks(A,k,dim,B);
|
|
|
|
|
|
cout<<"B=["<<endl<<B<<endl<<"];"<<endl;
|
|
return 0;
|
|
}
|