00001
00004
00005
00006
00007
00008
00009
00010
00011
00012
00013
00014
00015
00016
00017
00018
00019
00020
00021
00022
00023
00024
00025
00026
00027
00028
00029
00030
00031
00032
00033
00034
00035
00036
00037
00038
00039
00040
00041
00042
00043
00044
00045 #ifndef CLIPPER_FFTMAP_SPARSE
00046 #define CLIPPER_FFTMAP_SPARSE
00047
00048
00049 #include "fftmap.h"
00050
00051
00052 namespace clipper
00053 {
00054
00056 class FFTmap_sparse_p1_base : public FFTmap_base {
00057 public:
00059 void init( const Grid_sampling& grid_sam, const FFTtype type = Default );
00061 ~FFTmap_sparse_p1_base();
00063 const Grid_sampling& grid_real() const { return grid_real_; }
00065 const Grid& grid_reci() const { return grid_reci_; }
00066
00068 static FFTtype& default_type() { return default_type_; }
00069 protected:
00071 ffttype* map_uv( const int& u, const int& v );
00073 std::complex<ffttype>* map_kl( const int& k, const int& l );
00074
00075 Grid_sampling grid_real_;
00076 Grid grid_reci_;
00077 FFTtype type_;
00078
00079 Array2d<std::complex<ffttype>*> row_kl;
00080 Array2d<ffttype*> row_uv;
00081
00082 static FFTtype default_type_;
00083 };
00084
00086
00091 class FFTmap_sparse_p1_hx : public FFTmap_sparse_p1_base
00092 {
00093 public:
00095 FFTmap_sparse_p1_hx();
00097 FFTmap_sparse_p1_hx( const Grid_sampling& grid_sam, const FFTtype type = Default );
00098
00099
00100
00101
00103 void set_hkl( const HKL& hkl, const std::complex<ffttype>& f );
00105 std::complex<ffttype>& cplx_data( const Coord_grid& uvw )
00106 { return map_kl( uvw.v(), uvw.w() )[ uvw.u() ]; }
00108 void require_real_data( const Coord_grid& uvw )
00109 { map_uv( uvw.u(), uvw.v() ); }
00111 const ffttype& real_data( const Coord_grid& uvw ) const
00112 { return row_uv( uvw.u(), uvw.v() )[ uvw.w() ]; }
00113
00115 void fft_h_to_x( const ftype& scale );
00116 };
00117
00119
00124 class FFTmap_sparse_p1_xh : public FFTmap_sparse_p1_base
00125 {
00126 public:
00128 FFTmap_sparse_p1_xh();
00130 FFTmap_sparse_p1_xh( const Grid_sampling& grid_sam, const FFTtype type = Default );
00131
00132
00133
00134
00136 ffttype& real_data( const Coord_grid& uvw )
00137 { return map_uv( uvw.u(), uvw.v() )[ uvw.w() ]; }
00139 void require_hkl( const HKL& hkl );
00141 const std::complex<ffttype> get_hkl( const HKL& hkl ) const;
00143 void require_cplx_data( const Coord_grid& hkl )
00144 { map_kl( hkl.v(), hkl.w() ); }
00146 const std::complex<ffttype>& cplx_data( const Coord_grid& hkl ) const
00147 { return row_kl( hkl.v(), hkl.w() )[ hkl.u() ]; }
00148
00150 void fft_x_to_h( const ftype& scale );
00151 };
00152
00153
00154 }
00155
00156 #endif