comparison dsp/transforms/FFT.cpp @ 131:a586888bc06c kissfft

Add forwardMagnitude

             ro[m_n - i - 1] =  ro[i + 1];
             io[m_n - i - 1] = -io[i + 1];
         }
     }
 
+    void forwardMagnitude(const double *ri, double *mo) {
+
+        double *io = new double[m_n];
+
+        forward(ri, mo, io);
+
+        for (int i = 0; i < m_n; ++i) {
+            mo[i] = sqrt(mo[i] * mo[i] + io[i] * io[i]);
+        }
+
+        delete[] io;
+    }
+
     void inverse(const double *ri, const double *ii, double *ro) {
 
         for (int i = 0; i < m_n; ++i) {
             m_c[i].r = ri[i];
             m_c[i].i = ii[i];

 {
     m_d->forward(ri, ro, io);
 }
 
 void
+FFTReal::forwardMagnitude(const double *ri, double *mo)
+{
+    m_d->forwardMagnitude(ri, mo);
+}
+
+void
 FFTReal::inverse(const double *ri, const double *ii, double *ro)
 {
     m_d->inverse(ri, ii, ro);
 }