X-Git-Url: http://git.kpe.io/?a=blobdiff_plain;f=include%2Ffilter.h;h=9ee7390a4acd83c96bd9a72b7f9990fbfe7bede9;hb=dfa390de2efc04d85b03718a6480f735516df0e8;hp=19266a0edbce9e909d39bb127436f7de8a86fa68;hpb=00bead9fe7ab835c1584eac4ab586cc56742eaf3;p=ctsim.git diff --git a/include/filter.h b/include/filter.h index 19266a0..9ee7390 100644 --- a/include/filter.h +++ b/include/filter.h @@ -9,7 +9,7 @@ ** This is part of the CTSim program ** Copyright (C) 1983-2000 Kevin Rosenberg ** -** $Id: filter.h,v 1.5 2000/06/30 02:03:27 kevin Exp $ +** $Id: filter.h,v 1.12 2000/07/07 15:30:59 kevin Exp $ ** ** This program is free software; you can redistribute it and/or modify ** it under the terms of the GNU General Public License (version 2) as @@ -28,6 +28,7 @@ #ifndef FILTER_H #define FILTER_H + class SignalFilter { public: @@ -49,10 +50,12 @@ class SignalFilter { FILTER_METHOD_INVALID, FILTER_METHOD_CONVOLUTION, FILTER_METHOD_FOURIER, + FILTER_METHOD_FOURIER_TABLE, FILTER_METHOD_FFT, - FILTER_METHOD_FFT_ZEROPAD_2, - FILTER_METHOD_FFT_ZEROPAD_4, - FILTER_METHOD_FFT_ZEROPAD_6 +#if HAVE_FFTW + FILTER_METHOD_FFTW, + FILTER_METHOD_RFFTW, +#endif } FilterMethodID; typedef enum { @@ -74,18 +77,20 @@ class SignalFilter { static const char FILTER_METHOD_CONVOLUTION_STR[]= "convolution"; static const char FILTER_METHOD_FOURIER_STR[]= "fourier"; + static const char FILTER_METHOD_FOURIER_TABLE_STR[]="fourier_table"; static const char FILTER_METHOD_FFT_STR[]= "fft"; - static const char FILTER_METHOD_FFT_ZEROPAD_2_STR[]="fft_zeropad2"; - static const char FILTER_METHOD_FFT_ZEROPAD_4_STR[]="fft_zeropad4"; - static const char FILTER_METHOD_FFT_ZEROPAD_6_STR[]="fft_zeropad6"; +#if HAVE_FFTW + static const char FILTER_METHOD_FFTW_STR[]= "fftw"; + static const char FILTER_METHOD_RFFTW_STR[]= "rfftw"; +#endif - static const char DOMAIN_FREQUENCY_STR[]= "frequency"; - static const char DOMAIN_SPATIAL_STR[]= "spatial"; + static const char DOMAIN_FREQUENCY_STR[]="frequency"; + static const char DOMAIN_SPATIAL_STR[]="spatial"; - SignalFilter (const char* filterName, double bw, double xmin, double xmax, int n, double param, const char* domainName, const int numIntegral = 0); + SignalFilter (const char* filterName, const char* filterMethodName,double bw, double signalIncrement, int n, double param, const char* domainName, const int zeropad = 0, const int numIntegral = 0); - SignalFilter (const FilterID filt_type, double bw, double xmin, double xmax, int n, double param, const DomainID domain, const int numIntegral = 0); + SignalFilter (const FilterID filt_type, FilterMethodID filterMethodID, double bw, double signalIncrement, int n, double param, const DomainID domain, const int zeropad = 0, const int numIntegral = 0); SignalFilter (const char* filterName, const char* domainName, double bw, double param, int numIntegral = 0); @@ -94,10 +99,26 @@ class SignalFilter { double* getFilter (void) const { return m_vecFilter; } + int getNFilterPoints (void) const + { return m_nFilterPoints; } + double convolve (const double f[], const double dx, const int n, const int np) const; double convolve (const float f[], const double dx, const int n, const int np) const; + void filterSignal (const double input[], double output[]) const; + void filterSignal (const float input[], double output[]) const; + + static void finiteFourierTransform (const double input[], complex output[], const int n, const int direction); + static void finiteFourierTransform (const complex input[], complex output[], const int n, const int direction); + static void finiteFourierTransform (const complex input[], double output[], const int n, const int direction); + + void finiteFourierTransform (const double input[], complex output[], const int direction) const; + void finiteFourierTransform (const complex input[], complex output[], const int direction) const; + void finiteFourierTransform (const complex input[], double output[], const int direction) const; + + void setTraceLevel (int traceLevel) {m_traceLevel = traceLevel; } + bool fail(void) const {return m_fail;} const string& failMessage(void) const {return m_failMessage;} @@ -105,6 +126,9 @@ class SignalFilter { const string& nameDomain(void) const { return m_nameDomain;} const FilterID idFilter(void) const { return m_idFilter;} const DomainID idDomain(void) const { return m_idDomain;} + const double getFilterMin(void) const {return m_filterMin;} + const double getFilterMax(void) const {return m_filterMax;} + const double getFilterIncrement(void) const {return m_filterInc;} double response (double x); @@ -118,10 +142,23 @@ class SignalFilter { private: double m_bw; - int m_nPoints; - double m_xmin; - double m_xmax; + int m_nFilterPoints; + int m_nSignalPoints; + double m_signalInc; + double m_filterMin; + double m_filterMax; + double m_filterInc; double* m_vecFilter; + double* m_vecFourierCosTable; + double* m_vecFourierSinTable; + complex* m_complexVecFilter; +#ifdef HAVE_FFTW + fftw_real* m_vecRealFftInput; + rfftw_plan m_realPlanForward, m_realPlanBackward; + fftw_complex* m_vecComplexFftInput; + fftw_plan m_complexPlanForward, m_complexPlanBackward; +#endif + bool m_fail; string m_failMessage; string m_nameFilter; @@ -132,13 +169,17 @@ class SignalFilter { DomainID m_idDomain; double m_filterParam; int m_numIntegral; + int m_traceLevel; + int m_zeropad; - static FilterID convertFilterNameToID (const char* filterName); + static const FilterID convertFilterNameToID (const char* filterName); static const char* convertFilterIDToName (const FilterID filterID); + static const FilterMethodID convertFilterMethodNameToID (const char* filterMethodName); + static const char* convertFilterMethodIDToName (const FilterMethodID filterMethodID); static const DomainID convertDomainNameToID (const char* domainName); static const char* convertDomainIDToName (const DomainID domainID); - void init (const FilterID filt_type, double bw, double xmin, double xmax, int n, double param, const DomainID domain, const int numInt); + void init (const FilterID filt_type, const FilterMethodID filterMethod, double bw, double signalIncrement, int n, double param, const DomainID domain, const int zeropad, const int numInt); double spatialResponseCalc (double x, double param, int n) const;