| 1 | #ifndef  FFTWServer_H_SEEN
 | 
|---|
| 2 | #define  FFTWServer_H_SEEN
 | 
|---|
| 3 | 
 | 
|---|
| 4 | #include "machdefs.h"
 | 
|---|
| 5 | #include "sspvflags.h"
 | 
|---|
| 6 | #include "fftservintf.h"
 | 
|---|
| 7 | 
 | 
|---|
| 8 | 
 | 
|---|
| 9 | // Classe implementant l'interface FFTServerInterface en 
 | 
|---|
| 10 | // utilisant FFTW 
 | 
|---|
| 11 | 
 | 
|---|
| 12 | namespace SOPHYA {
 | 
|---|
| 13 | 
 | 
|---|
| 14 | #ifdef FFTW_V2_EXTSOP
 | 
|---|
| 15 | class FFTWServerPlan;
 | 
|---|
| 16 | #endif
 | 
|---|
| 17 | 
 | 
|---|
| 18 | class FFTWServer : public FFTServerInterface {
 | 
|---|
| 19 |  public:
 | 
|---|
| 20 |   FFTWServer(bool preserve_input=true);
 | 
|---|
| 21 |   virtual ~FFTWServer();
 | 
|---|
| 22 | 
 | 
|---|
| 23 |   
 | 
|---|
| 24 |   virtual FFTServerInterface * Clone();
 | 
|---|
| 25 | 
 | 
|---|
| 26 |   // Transforme unidimensionnelle , N-dimensionnel  (double precision - r_8)
 | 
|---|
| 27 |   virtual void FFTForward(TArray< complex<r_8> > & in, TArray< complex<r_8> > & out);
 | 
|---|
| 28 |   virtual void FFTBackward(TArray< complex<r_8> > & in, TArray< complex<r_8> > & out);
 | 
|---|
| 29 |   virtual void FFTForward(TArray< r_8 > & in, TArray< complex<r_8> > & out);
 | 
|---|
| 30 |   virtual void FFTBackward(TArray< complex<r_8> > & in, TArray< r_8 > & out,
 | 
|---|
| 31 |                            bool usoutsz=false);
 | 
|---|
| 32 | 
 | 
|---|
| 33 | // Methodes statiques pour reordonner les donnees en 1-D (double precision - r_8)
 | 
|---|
| 34 |   static void ReShapetoReal(TArray< complex<r_8> > const & in, TArray< r_8 > & out, bool usz);
 | 
|---|
| 35 |   static void ReShapetoCompl(TArray< r_8 > const & in, TArray< complex<r_8> > & out);
 | 
|---|
| 36 | 
 | 
|---|
| 37 | #ifdef ALSO_FFTW_FLOAT_EXTSOP 
 | 
|---|
| 38 |  /* ---------------------------------------------------------------------------
 | 
|---|
| 39 |    Declare and compile single precision (float) version of FFTWServr methods 
 | 
|---|
| 40 |    Needs the libfftw3f.a , in addition to libfftw3.a   
 | 
|---|
| 41 |    --------------------------------------------------------------------------- */
 | 
|---|
| 42 | 
 | 
|---|
| 43 |   // Transforme unidimensionnelle , N-dimensionnel  (single precision - r_4)
 | 
|---|
| 44 |   virtual void FFTForward(TArray< complex<r_4> > & in, TArray< complex<r_4> > & out);
 | 
|---|
| 45 |   virtual void FFTBackward(TArray< complex<r_4> > & in, TArray< complex<r_4> > & out);
 | 
|---|
| 46 |   virtual void FFTForward(TArray< r_4 > & in, TArray< complex<r_4> > & out);
 | 
|---|
| 47 |   virtual void FFTBackward(TArray< complex<r_4> > & in, TArray< r_4 > & out,
 | 
|---|
| 48 |                            bool usoutsz=false);
 | 
|---|
| 49 | 
 | 
|---|
| 50 | // Methodes statiques pour reordonner les donnees en 1-D (single precision - r_4)
 | 
|---|
| 51 |   static void ReShapetoReal(TArray< complex<r_4> > const & in, TArray< r_4 > & out, bool usz);
 | 
|---|
| 52 |   static void ReShapetoCompl(TArray< r_4 > const & in, TArray< complex<r_4> > & out);
 | 
|---|
| 53 |   
 | 
|---|
| 54 | #endif
 | 
|---|
| 55 | 
 | 
|---|
| 56 |  protected:
 | 
|---|
| 57 | 
 | 
|---|
| 58 | #ifndef FFTW_V2_EXTSOP    
 | 
|---|
| 59 | /*  For FFTW >= V3  */
 | 
|---|
| 60 |  FFTArrayChecker<r_8> ckR8;
 | 
|---|
| 61 |  FFTArrayChecker<r_4> ckR4;
 | 
|---|
| 62 |  bool _preserve_input; // if true, input arrays not overwritten
 | 
|---|
| 63 | 
 | 
|---|
| 64 | #else
 | 
|---|
| 65 | /*  For FFTW  V2  */
 | 
|---|
| 66 |  FFTWServerPlan * _p1df;
 | 
|---|
| 67 |  FFTWServerPlan * _p1db;
 | 
|---|
| 68 |  FFTWServerPlan * _pndf;
 | 
|---|
| 69 |  FFTWServerPlan * _pndb;
 | 
|---|
| 70 | 
 | 
|---|
| 71 |  FFTWServerPlan * _p1drf;
 | 
|---|
| 72 |  FFTWServerPlan * _p1drb;
 | 
|---|
| 73 |  FFTWServerPlan * _pndrf;
 | 
|---|
| 74 |  FFTWServerPlan * _pndrb;
 | 
|---|
| 75 | 
 | 
|---|
| 76 |  FFTArrayChecker<r_8> ckR8;
 | 
|---|
| 77 |  bool _preserve_input; // if true, input arrays not overwritten
 | 
|---|
| 78 | 
 | 
|---|
| 79 | #endif
 | 
|---|
| 80 | };
 | 
|---|
| 81 | 
 | 
|---|
| 82 | } // Fin du namespace
 | 
|---|
| 83 | 
 | 
|---|
| 84 | #endif
 | 
|---|