Changeset 1394 in Sophya for trunk/SophyaLib/NTools/fftpserver.cc

Timestamp:

Feb 12, 2001, 6:09:27 PM (25 years ago)

Author:

ansari

Message:

Changement interface FFTServer - Reza 12/2/2001

File:

• trunk/SophyaLib/NTools/fftpserver.cc (modified) (6 diffs)

trunk/SophyaLib/NTools/fftpserver.cc

@@ -28 +28 @@
 FFTPackServer::FFTPackServer()
    : FFTServerInterface("FFTPackServer using extended FFTPack (C-version) package")
-  , ckR4(true, true) , ckR8(true, true)
+  , ckR4("FFTPackServer: ", true, true) , ckR8("FFTPackServer: ", true, true)
 {
 //the working array and its size for the different
@@ -72 +72 @@
 
 
-
 void FFTPackServer::FFTForward(TArray< complex<r_4> > const & in, TArray< complex<r_4> > & out)
 {
   ckR4.CheckResize(in, out);  
   out = in;
+  cout << out << endl;
   fftf(out.Size(), out.Data());
   if (getNormalize()) out *= (1./(r_4)(in.Size()));
@@ -93 +93 @@
   TArray< r_4 > inout(in, false);
   fftf(inout.Size(), inout.Data());
-  ckR4.ReShapetoCompl(inout, out);
+  ReShapetoCompl(inout, out);
   if (getNormalize()) out *= complex<r_4>((1./(r_4)(in.Size())), 0.);
 }
@@ -100 +100 @@
 {
   ckR4.CheckResize(in, out);    
-  ckR4.ReShapetoReal(in, out);
+  ReShapetoReal(in, out);
   fftb(out.Size(), out.Data());
 }
@@ -110 +110 @@
   TArray< r_8 > inout(in, false);
   fftf(inout.Size(), inout.Data());
-  ckR8.ReShapetoCompl(inout, out);
+  ReShapetoCompl(inout, out);
   if (getNormalize()) out *= complex<r_8>((1./(r_8)(in.Size())), 0.);
 }
@@ -117 +117 @@
 {
   ckR8.CheckResize(in, out);  
-  ckR8.ReShapetoReal(in, out);
-  fftb(out.Size(), out.Data());
-}
-
- 
+  ReShapetoReal(in, out);
+  fftb(out.Size(), out.Data());
+}
+
+
+template <class T>
+void FFTPack_ReShapetoReal(TArray< complex<T> > const & ina, TArray< T >  & outa)
+{
+  TVector< complex<T> > in(ina);
+  TVector< T > out(outa);
+  sa_size_t n = in.NElts();
+  T thr = FFTArrayChecker<T>::ZeroThreshold();
+  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) 
+    throw SzMismatchError("FFTPack_ReShapetoReal() - Wrong output array size !");
+    //    cerr << "DEBUG-FFTPack_ReShapetoReal() ncs = " << ncs 
+    //   << " out.NElts()= " << out.NElts() << endl;
+
+  sa_size_t k;
+
+  out(0) = in(0).real();
+  for(k=1;k<n-1;k++) {
+    out(2*k-1) = in(k).real();
+    out(2*k) = in(k).imag();
+  }
+  if (ncs == n*2-2)  out(ncs-1) = in(n-1).real();
+  else { out(ncs-2) = in(n-1).real();  out(ncs-1) = in(n-1).imag(); }
+
+  return;
+}
+
+template <class T>
+void FFTPack_ReShapetoCompl(TArray< T > const & ina, TArray< complex<T> > & outa)
+{
+  TVector< T > in(ina);
+  TVector< complex<T> > out(outa);
+  sa_size_t n = in.NElts();
+  sa_size_t ncs = n/2+1;
+  sa_size_t nc = (n%2 != 0) ? n/2+1 : n/2;
+  if (out.NElts() != ncs)
+    throw SzMismatchError("FFTPack_ReShapetoCompl() - Wrong output array size !");
+
+  out(0) = complex<T> (in(0),0.);
+  for(int k=1;k<nc;k++)
+    out(k) =  complex<r_4> (in(2*k-1), in(2*k));
+  if (n%2 == 0) out(ncs-1) = complex<T>(in(n-1), 0.);
+
+  return;
+}
+
+void FFTPackServer::ReShapetoReal(TArray< complex<r_8> > const & in, TArray< r_8 >  & out)
+{
+  FFTPack_ReShapetoReal<r_8>(in, out);
+}
+
+void FFTPackServer::ReShapetoCompl(TArray< r_8 > const & in, TArray< complex<r_8> > & out)
+{
+  FFTPack_ReShapetoCompl<r_8>(in, out);
+}
+
+void FFTPackServer::ReShapetoReal(TArray< complex<r_4> > const & in, TArray< r_4 >  & out)
+{
+  FFTPack_ReShapetoReal<r_4>(in, out);
+}
+
+void FFTPackServer::ReShapetoCompl(TArray< r_4 > const & in, TArray< complex<r_4> > & out)
+{
+  FFTPack_ReShapetoCompl<r_4>(in, out);
+}
 
 void FFTPackServer::checkint_rfft(int_4 l)