| 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 | 
|---|