Changeset 3912 in Sophya for trunk/AddOn

Timestamp:

Nov 22, 2010, 10:01:12 AM (15 years ago)

Author:

ansari

Message:

correction programme de test FFT multi-thread, Reza 22/11/2010

File:

• trunk/AddOn/TAcq/tmtfft.cc (modified) (7 diffs)

trunk/AddOn/TAcq/tmtfft.cc

@@ -4 +4 @@
 //   R. Ansari - C. Magneville  ,   Decembre 2007
 // Exemples de commandes pour test (dans l'ordre ci-dessous :
-// csh> tmtfft OCG 1000000 XXX
-// csh> tmtfft STG 1000000 1024
-// csh> tmtfft RDG 1000000 rg1.ppf
-// csh> tmtfft MTG 3000000 4 
+// csh> tmtfft VSZ NTH NbFFT [PaqSz=4096] 
+// csh> tmtfft 1310720 1 10000 16384
+// csh> tmtfft 1310720 1 10000 8192
+// csh> tmtfft 1310720 2 10000 8192
 //--------------------------------------------------------------------------------
 
@@ -37 +37 @@
 // Definition d'une classe heritant de ZThread, pour remplir un vecteur avec 
 // des aleatoires
+
 class MTDoFFT : public ZThread {
 public:
@@ -53 +54 @@
 
 static int mtrandId = 0;  // Pour donner un identificateur a chaque thread
+ZMutex* pmtx_=NULL;  // pour synchroniser creation de plan_fftw;
 
 MTDoFFT::MTDoFFT(TVector<r_4>& v, sa_size_t nfft, sa_size_t paqsz)
@@ -58 +60 @@
     nbfft_(nfft), paqsz_(paqsz), ffts_(true) // preserve input=true
 {
+  if (pmtx_==NULL)  pmtx_ = new ZMutex;
   char buff[32];
   sprintf(buff, "MTDoFFT-Id=%d", mtrandId);
@@ -63 +66 @@
   nom_ = buff;
   // Initialisation taille vecteurs 
-  TVector<r_4> vx(paqsz);
+  TVector<r_4> vx(paqsz_);
   vx = RegularSequence();
   ffts_.FFTForward(vx, cfour_);
   spectre_.ReSize(cfour_.Size());
-  cout << " Thread MTDoFFT(" << nom_ << " ) Created ... " << endl;
+  cout << " Thread MTDoFFT(" << nom_ << ", PaqSz=" << paqsz_ << " ) Created ... " << endl;
 }
 
@@ -81 +84 @@
   FMTRandGen rgen;
   TVector<r_4> vx(paqsz_);
+  
+  pmtx_->lock();
   fftwf_plan plan = fftwf_plan_dft_r2c_1d(paqsz_, vx.Data(), 
-                                       (fftwf_complex *)cfour_.Data(), FFTW_ESTIMATE);  
+                                          (fftwf_complex *)cfour_.Data(), FFTW_ESTIMATE);  
+  pmtx_->unlock();
 
   sa_size_t prm = nbfft_/10;
@@ -161 +167 @@
   sa_size_t nfft = atoi(arg[3]);
   sa_size_t paqsz = 4096;
-  if (narg > 4) paqsz = atoi(arg[3]);
+  if (narg > 4) paqsz = atoi(arg[4]);
   try {
     f3_tmtfft(VSZ, NTH, nfft, paqsz);