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