Changeset 3089 in Sophya for trunk/SophyaProg/Tests/tfft.cc

Timestamp:

Oct 11, 2006, 12:58:10 PM (19 years ago)

Author:

ansari

Message:

prog tests: ajout test perfo multi-fft (tfft.cc) et amelioration test SwFits (tswfits.cc), Reza 11/10/2006

File:

• trunk/SophyaProg/Tests/tfft.cc (modified) (8 diffs)

trunk/SophyaProg/Tests/tfft.cc

@@ -11 +11 @@
 #include "fftpserver.h"
 #include "fftmserver.h"
+#include "fftmayer.h"
 #include "fftwserver.h"
 #include "ntoolsinit.h"
@@ -48 +49 @@
 static  int prtlev = 0;
 
+//-------------------------------------------------
 template <class T>
 void TestFFTPack(T seuil, sa_size_t num)
@@ -106 +108 @@
 }
 
+//-------------------------------------------------
 template <class T>
 int TestFFTS(T seuil, FFTServerInterface & ffts, sa_size_t num)
@@ -197 +200 @@
 }
 
-
-
-
-
+inline void MayerFFTForw(r_4* d, int sz) {
+  fht_r4(d, sz);
+}
+inline void MayerFFTForw(r_8* d, int sz) {
+  fht_r8(d, sz);
+}
+
+//-------------------------------------------------
+template <class T>
+int MultiFFTTest(T t, sa_size_t sz, int nloop, bool fgp)
+{
+  cout <<" ===> MultiFFTTest<T=r_" << sizeof(T) << "> NLoop= " << nloop << " ArrSz= " << sz << endl;  
+  TVector< T > in(sz), incopie(sz);
+
+  incopie = in = RandomSequence();
+  FFTPackServer fftp(false);
+  TVector< complex<T> > outc;
+    
+  PrtTim("MultiFFT-LoopStart");
+
+  if (fgp) {
+    cout << " --- Test effectue avec FFTPack " << endl;
+    for(int kk=0; kk<nloop; kk++) {
+      in = incopie;
+      //      fftp.FFTForward(in, outc);
+      fftp.fftf(sz, in.Data());
+    }
+  }
+  else { 
+    cout << " --- Test effectue avec FFTMayer " << endl;
+    for(int kk=0; kk<nloop; kk++) {
+      in = incopie;
+      //      in += 0.01;
+      MayerFFTForw(in.Data(), sz);
+    }
+  }
+  PrtTim("MultiFFT-LoopEnd");
+  cout << " ----- End ---- MultiFFTTest<T> ----" << endl;
+  return 0;
+}
+
+
+
+//-------------------------------------------------
+//-------------------------------------------------
+//-------------------------------------------------
 int main(int narg, char* arg[])
 {
@@ -209 +254 @@
   if (narg < 3) {
     cout << "tfft/ args error - \n  Usage tfft size p/P/M/W [f/d/F/D  PrtLev=0 MaxNPrt=50 diffthr] \n" 
+         << "    OR  tfft size P/M FL/DL/FZ/DZ  [NFFT=10]  \n" 
          << " 1D real/complex-FFT test (FFTServer) \n"
          << " size: input vector length \n " 
          << " p=FFTPackTest  P=FFTPack, M=FFTMayer, W= FFTWServer \n " 
          << " F/f:float, D/d:double F/D:random in_vector (default=D) \n" 
-         << " diffthr : Threshold for diff checks (=10^-6/10^-4 double/float)" << endl;
+         << "  FL/DL : perform NFFT on a serie of float[size] or double[size] " 
+         << "  FZ/DZ : perform NFFT on a series of complex<float>[size] or complex<double>[size] " 
+         << " diffthr : Threshold for diff checks (=10^-6/10^-4 double/float)" 
+         << endl;
     return(1);
   }
@@ -239 +288 @@
   dtyp = toupper(dtyp);
 
+
   FFTServerInterface * ffts;
   if (*arg[2] == 'M')  ffts = fftm.Clone();
   else if (*arg[2] == 'W')  ffts = fftw.Clone();
   else ffts = fftp.Clone();
+
+  int nloop = 0;
+  bool fgpack = true;
+  if (arg[3][1] == 'L') {  // MultiFFT test 
+    nloop = 10;
+    if (narg > 4) nloop = atoi(arg[4]);
+    if (*arg[2] == 'M')  fgpack = false;
+  }
+
 
   cout << "\n ============================================= \n"
@@ -255 +314 @@
       cout << "   ------ Testing FFTServer for double (r_8)----- " << endl;
       if (*arg[2] == 'p') TestFFTPack(ds, sz);
+      else if (nloop > 0) rc = MultiFFTTest(ds, sz, nloop, fgpack);
       else rc = TestFFTS(ds, *ffts, sz);
     }
@@ -260 +320 @@
       cout << "   ------ Testing FFTServer for float (r_4)----- " << endl;
       if (*arg[2] == 'p') TestFFTPack(fs, sz);
+      else if (nloop > 0) rc = MultiFFTTest(fs, sz, nloop, fgpack);
       else rc = TestFFTS(fs, *ffts, sz);
     }