mirror/eigen
2
0
Fork 0
mirror of https://gitlab.com/libeigen/eigen.git synced 2024-12-21 07:19:46 +08:00
Code Issues Packages Projects Releases Wiki Activity

if the src.stride() != 1, then the layout is not continuous -- need to copy to temporary

Browse Source
This commit is contained in:
Mark Borgerding 2010-01-22 01:17:36 -05:00
parent cd7912313d
commit 141c746fc7
2 changed files with 18 additions and 23 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

15
unsupported/Eigen/FFT
View File

@ -185,7 +185,13 @@ class FFT
dst->derived().resize( (src.size()>>1)+1);
else
dst->derived().resize(src.size());
fwd( &(*dst)[0],&src[0],src.size() );
if (src.stride() != 1) {
Matrix<typename InputDerived::Scalar,1,Dynamic> tmp = src;
fwd( &(*dst)[0],&tmp[0],src.size() );
}else{
fwd( &(*dst)[0],&src[0],src.size() );
}
}
inline
@ -219,7 +225,12 @@ class FFT
int nfft = src.size();
int nout = HasFlag(HalfSpectrum) ? ((nfft>>1)+1) : nfft;
dst->derived().resize( nout );
inv( &(*dst)[0],&src[0], nfft);
if (src.stride() != 1) {
Matrix<typename ComplexDerived::Scalar,1,Dynamic> tmp = src;
inv( &(*dst)[0],&tmp[0], nfft);
}else{
inv( &(*dst)[0],&src[0], nfft);
}
}
template <typename _Output>

26
unsupported/test/FFTW.cpp
View File

@ -123,43 +123,27 @@ void test_complex(int nfft)
template <typename T,int nrows,int ncols>
void test_complex2d()
{
typedef typename Eigen::FFT<T>::Complex Complex;
FFT<T> fft;
Eigen::Matrix<Complex,nrows,ncols> src,src2,dst,dst2;
Eigen::Matrix<Complex,nrows,ncols> src;
Eigen::Matrix<Complex,nrows,ncols> dst;
Eigen::Matrix<Complex,nrows,ncols> src2;
Eigen::Matrix<Complex,nrows,ncols> dst2;
//src = Eigen::Matrix<Complex,nrows,ncols>::Random();
src = Eigen::Matrix<Complex,nrows,ncols>::Identity();
src = Eigen::Matrix<Complex,nrows,ncols>::Random();
//src = Eigen::Matrix<Complex,nrows,ncols>::Identity();
for (int k=0;k<ncols;k++) {
Eigen::Matrix<Complex,nrows,1> tmpIn = src.col(k);
Eigen::Matrix<Complex,nrows,1> tmpOut;
fft.fwd( &tmpOut,tmpIn );
fft.fwd( &tmpOut,src.col(k) );
dst2.col(k) = tmpOut;
}
//cout << "dst2: " << dst2 << "\n\n";
for (int k=0;k<nrows;k++) {
Eigen::Matrix<Complex,1,ncols> tmpIn = dst2.row(k);
Eigen::Matrix<Complex,1,ncols> tmpOut;
fft.fwd( &tmpOut, tmpIn);
fft.fwd( &tmpOut, dst2.row(k) );
dst2.row(k) = tmpOut;
}
/*
*/
fft.fwd2(dst.data(),src.data(),nrows,ncols);
fft.inv2(src2.data(),dst.data(),nrows,ncols);
/*
cout << "src: " << src << "\n\n";
cout << "dst: " << dst << "\n\n";
cout << "src2: " << src2 << "\n\n";
cout << "dst2: " << dst2 << "\n\n";
*/
VERIFY( (src-src2).norm() < test_precision<T>() );
VERIFY( (dst-dst2).norm() < test_precision<T>() );
}