Changeset 2989 in Sophya for trunk/SophyaExt/IFFTW/fftwserver.cc

Timestamp:

Jun 23, 2006, 12:31:39 PM (19 years ago)

Author:

ansari

Message:

Ajout flag use_out_size d'utilisation de la taille du tableau reel de sortie pour FFTBack(complex->real) dans ReShapetoReal() pour FFTWServer - fichiers fftwserver.h .cc) - Reza 23 Juin 2006

File:

• trunk/SophyaExt/IFFTW/fftwserver.cc (modified) (3 diffs)

trunk/SophyaExt/IFFTW/fftwserver.cc

@@ -334 +334 @@
     else _p1drb = new FFTWServerPlan(out.Size(), FFTW_COMPLEX_TO_REAL, true);
 
-    ReShapetoReal(in, intemp);
+    ReShapetoReal(in, intemp, usoutsz);
     //    cerr << " DEBUG-FFTWServer::FFTBackward() in = \n" << in << endl;
     //    cerr << " DEBUG-FFTWServer::FFTBackward() intemp = \n" << intemp << endl;
@@ -361 +361 @@
 
 /* --Methode-- */
-void FFTWServer::ReShapetoReal(TArray< complex<r_8> > const & ina, TArray< r_8 > & outa)
+void FFTWServer::ReShapetoReal(TArray< complex<r_8> > const & ina, TArray< r_8 > & outa, 
+                               bool usz)
 {
   TVector< complex<r_8> > in(ina);
@@ -367 +368 @@
   int n = in.NElts();
   r_8 thr = FFTArrayChecker<r_8>::ZeroThreshold();
-  sa_size_t ncs = ( (in(n-1).imag() < -thr) || (in(n-1).imag() > thr) ) ?
+  sa_size_t ncs;
+  if (usz) {
+    if ( (out.NElts() != 2*n-1) && (out.NElts() != 2*n-2) )
+      throw SzMismatchError("FFTWServer::ReShapetoReal(..., true) - Wrong output array size ");
+    ncs = out.NElts();
+  }
+  else {
+    sa_size_t ncs = ( (in(n-1).imag() < -thr) || (in(n-1).imag() > thr) ) ?
                     ncs = 2*n-1 : ncs = 2*n-2;
 
-  if (out.NElts() != ncs) {
-    cerr << " DEBUG-FFTWServer::ReShapetoReal() ncs= " << ncs 
-         << " out.NElts()= " << out.NElts() << endl;
-    throw SzMismatchError("FFTWServer::ReShapetoReal() - Wrong output array size !");
+    if (out.NElts() != ncs) {
+      cerr << " DEBUG-FFTWServer::ReShapetoReal() ncs= " << ncs 
+           << " out.NElts()= " << out.NElts() << endl;
+      throw SzMismatchError("FFTWServer::ReShapetoReal() - Wrong output array size !");
+    }
   }
   //  if (ncs == 2*n-2)  {