devtools/eigen
1
0
Fork 0
mirror of https://gitlab.com/libeigen/eigen.git synced 2026-04-10 11:34:33 +08:00
Code Issues Packages Projects Releases Wiki Activity

Fix FFT when destination does not have unit stride.

Browse Source
This commit is contained in:
Antonio Sánchez
2024-05-07 17:18:29 +00:00
committed by Rasmus Munk Larsen
parent 99c18bce6e
commit e16d70bd4e
2 changed files with 67 additions and 9 deletions
Download Patch File Download Diff File Expand all files Collapse all files

44
unsupported/Eigen/FFT
View File

@@ -231,11 +231,12 @@ class FFT {
THIS_METHOD_IS_ONLY_FOR_EXPRESSIONS_WITH_DIRECT_MEMORY_ACCESS_SUCH_AS_MAP_OR_PLAIN_MATRICES)
if (nfft < 1) nfft = src.size();
if (NumTraits<src_type>::IsComplex == 0 && HasFlag(HalfSpectrum))
dst.derived().resize((nfft >> 1) + 1);
else
dst.derived().resize(nfft);
Index dst_size = nfft;
if (NumTraits<src_type>::IsComplex == 0 && HasFlag(HalfSpectrum)) {
dst_size = (nfft >> 1) + 1;
}
dst.derived().resize(dst_size);
if (src.innerStride() != 1 || src.size() < nfft) {
Matrix<src_type, 1, Dynamic> tmp;
@@ -245,9 +246,21 @@ class FFT {
} else {
tmp = src;
}
fwd(&dst[0], &tmp[0], nfft);
if (dst.innerStride() != 1) {
Matrix<dst_type, 1, Dynamic> out(1, dst_size);
fwd(&out[0], &tmp[0], nfft);
dst.derived() = out;
} else {
fwd(&dst[0], &tmp[0], nfft);
}
} else {
fwd(&dst[0], &src[0], nfft);
if (dst.innerStride() != 1) {
Matrix<dst_type, 1, Dynamic> out(1, dst_size);
fwd(&out[0], &src[0], nfft);
dst.derived() = out;
} else {
fwd(&dst[0], &src[0], nfft);
}
}
}
@@ -326,9 +339,22 @@ class FFT {
} else {
tmp = src;
}
inv(&dst[0], &tmp[0], nfft);
if (dst.innerStride() != 1) {
Matrix<dst_type, 1, Dynamic> out(1, nfft);
inv(&out[0], &tmp[0], nfft);
dst.derived() = out;
} else {
inv(&dst[0], &tmp[0], nfft);
}
} else {
inv(&dst[0], &src[0], nfft);
if (dst.innerStride() != 1) {
Matrix<dst_type, 1, Dynamic> out(1, nfft);
inv(&out[0], &src[0], nfft);
dst.derived() = out;
} else {
inv(&dst[0], &src[0], nfft);
}
}
}