RcppArmadillo ifft2对R的原生mvfft

时间:2016-08-14 13:50:59

标签: c++ r fft rcpp

我注意到RcppArmadillo支持FFT&二维FFT。不幸的是,ifft2RcppArmadillo)和R&#39 mvfft(..., inverse = TRUE)与我的数据之间存在显着差异。这在第0个bin中特别大(这在我的应用程序中非常重要)。差异不是标量倍数。我找不到任何关于这些偏差的文档或说明,特别是在第0个区域。

我已经专门针对ifft(arma::cx_mat input)函数调用调试了该问题。除非可能存在无法预料的内存管理问题,否则这就是罪魁祸首。

示例:ifft2结果(1列前5个条目):

[1] 0.513297156-0.423498014i -0.129250939+0.300225299i  
0.039722228-0.093052563i -0.007956237+0.018643534i 0.001181177-0.002768473i

mvfft反向结果(1列前5个条目):

[1] 0.278131988-0.633838170i -0.195699114+0.445980950i  
0.060070320-0.136894940i -0.011924932+0.027175865i 0.001754788-0.003999007i

问题

  • RcppArmadillo FFT仍在开发中吗?
  • 这是跨越FFT变体的常见问题(FP或DP噪声之外的数值偏差)?
  • 是否有低级别的'来自Rcpp或RcppArmadillo的函数调用来调用R的原生FFT?

再现性 - 下面我尽可能地解决了这个问题并重现了这个问题。 针对最小代码进行了更新 Rcpp代码:

#include <RcppArmadillo.h>
// [[Rcpp::depends(RcppArmadillo)]]

using namespace Rcpp;
 // [[Rcpp::export]]
 //profile is the dependent variable of a given variable x,
 //q is a vector containing complex valued information for a single column after a tcrossprod
 //Size is a scalar value which the FFT depends upon.
 arma::cx_mat DebugLmnCPP( arma::cx_vec Profile, arma::cx_vec q) {
   std::complex<double> oneeye (0,1);//Cmplx number (0 + 1i)
   arma::cx_mat qFFT = ifft2( exp( oneeye * (Profile * q.st() )  ) );
   return(qFFT );
 }
 // [[Rcpp::export]]
 //For pedagogical purposes
 arma::cx_mat DebugIFFTRCPP( arma::cx_mat input) {
   arma::cx_mat qFFT = ifft2( input );
   return( qFFT );
 }

RCode(对不起,这很草率)

library(Rcpp)
library(RcppArmadillo)
sourceCpp("/home/FILE.cpp")

#Use C++ function
qt <- c(6.0+0i, 5.95+0i, 0.10+0i)
prof <-  0.25* sin( (1:512)*(2*3.1415)/512 )  + 0.25#Offset Sine wave
Debug1 <- DebugLmnCPP( Profile = prof, q = qt )

#Use R function
FFTSize <- 2^9
DebugLmnR <- function(Profile, q) {
  g <- (0+1i)*(as.matrix(Profile ) %*% t(q))
  qFFT <- mvfft( exp(g) , inverse = TRUE) / FFTSize 
  return( qFFT )
}
#Call function
Debug2 <- DebugLmnR( Profile = prof, q = qt )

#Use R and C++
DebugLmnRC <- function(Profile, q) {
  g <- (0+1i)*(as.matrix(Profile ) %*% t(q))
  qFFT <-  DebugIFFTRCPP(exp(g))
  return( qFFT )
}
#Call function
Debug3 <- DebugLmnRC( Profile = prof, q = qt )
#Compare Results
Debug1[1:5,1] #CPP
Debug2[1:5,1] #R
Debug3[1:5,1] #R and CPP

产量:

> Debug1[1:5,1]
[1]  0.359632774+0.35083419i -0.037254305-0.36995074i  0.015576046+0.15288379i -0.004552119-0.03992962i
[5]  0.000967252+0.00765564i
> Debug2[1:5,1]
[1]  0.03620451+0.51053116i -0.04624384-0.55604273i  0.02204910+0.23101589i -0.00653108-0.06061692i
[5]  0.00140213+0.01167389i
> Debug3[1:5,1]
[1]  0.359632774+0.35083419i -0.037254305-0.36995074i  0.015576046+0.15288379i -0.004552119-0.03992962i
[5]  0.000967252+0.00765564i

1 个答案:

答案 0 :(得分:4)

我不是特别喜欢你的例子:

  • 因为它仍然太复杂
  • 您正在转换您正在比较的函数中的数据 - 通常是一个坏主意
  • 所以我建议您修改输入

这是一个更简单的例子。使用此示例的R引导中的help(fft)

fftR> x <- 1:4

fftR> fft(x)
[1] 10+0i -2+2i -2+0i -2-2i

fftR> fft(fft(x), inverse = TRUE)/length(x)
[1] 1+0i 2+0i 3+0i 4+0i

我们可以使用RcppArmadillo轻松复制:

R> cppFunction("arma::cx_mat armafft(arma::vec x) { return fft(x); }", 
+              depends="RcppArmadillo")
R> armafft(1:4)
      [,1]
[1,] 10+0i
[2,] -2+2i
[3,] -2+0i
[4,] -2-2i
R> 

并添加反向

R> cppFunction("arma::cx_mat armaifft(arma::cx_mat x) { return ifft(x); }", 
+              depends="RcppArmadillo")
R> armaifft(armafft(1:4))
     [,1]
[1,] 1+0i
[2,] 2+0i
[3,] 3+0i
[4,] 4+0i
R> 

恢复输入,如R示例所示。

据我所知,没有任何错误,我没有理由相信这对于2d案件有任何不同......

编辑/跟进:错误在于OP,而不在于Armadillo。这里的主要问题是

  • 未仔细阅读文档
  • 没有使用最少的例子

这里的主要问题是 Armadillo的fft()可以处理向量或矩阵,因此(在矩阵的情况下)对应于R的mvfft()。犰狳的fft2()只是其他的东西而且与此无关。

让我们继续/扩展我们之前的例子。我们重新定义了我们的访问器以使用复杂的矩阵值:

R> cppFunction("arma::cx_mat armafft(arma::cx_mat x) { return fft(x); }",
+              depends="RcppArmadillo")
R>

然后定义一个尺寸为5 x 2的复杂数组,我们将其提供给它:

R> z <- array(1:10 + 1i, dim=c(5,2))
R> z
     [,1]  [,2]
[1,] 1+1i  6+1i
[2,] 2+1i  7+1i
[3,] 3+1i  8+1i
[4,] 4+1i  9+1i
[5,] 5+1i 10+1i
R> 
R> armafft(z)
              [,1]          [,2]
[1,] 15.0+5.00000i 40.0+5.00000i
[2,] -2.5+3.44095i -2.5+3.44095i
[3,] -2.5+0.81230i -2.5+0.81230i
[4,] -2.5-0.81230i -2.5-0.81230i
[5,] -2.5-3.44095i -2.5-3.44095i
R> 

这与我们在每列上单独运行函数所获得的输出相同。这也是R对mvfft()所做的事(cf help(fft)

R> mvfft(z)
              [,1]          [,2]
[1,] 15.0+5.00000i 40.0+5.00000i
[2,] -2.5+3.44095i -2.5+3.44095i
[3,] -2.5+0.81230i -2.5+0.81230i
[4,] -2.5-0.81230i -2.5-0.81230i
[5,] -2.5-3.44095i -2.5-3.44095i
R> 

同样的结果,不同的库/包,就我所见,没有错误。