X-Git-Url: http://git.kpe.io/?a=blobdiff_plain;f=include%2Ffilter.h;h=6ad776792158c2f4d6d7604268a39f2baa4a67fd;hb=ab4a3ba2c1ba9672136b5022897f81fc55d03493;hp=b544a1f2132ac16671cbab6bde3e6c320cebcbbf;hpb=2d39e823ba389fc68e5317c422b55be006094252;p=ctsim.git diff --git a/include/filter.h b/include/filter.h index b544a1f..6ad7767 100644 --- a/include/filter.h +++ b/include/filter.h @@ -1,7 +1,45 @@ +/***************************************************************************** +** FILE IDENTIFICATION +** +** Name: filter.h +** Purpose: Signal filter header file +** Programmer: Kevin Rosenberg +** Date Started: June 2000 +** +** This is part of the CTSim program +** Copyright (C) 1983-2000 Kevin Rosenberg +** +** $Id: filter.h,v 1.14 2000/07/19 04:33:27 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 +** published by the Free Software Foundation. +** +** This program is distributed in the hope that it will be useful, +** but WITHOUT ANY WARRANTY; without even the implied warranty of +** MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the +** GNU General Public License for more details. +** +** You should have received a copy of the GNU General Public License +** along with this program; if not, write to the Free Software +** Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA +******************************************************************************/ + #ifndef FILTER_H #define FILTER_H +#ifdef HAVE_CONFIG_H +#include "config.h" +#endif +#ifdef HAVE_FFTW +#include +#include +#endif + +#include + + class SignalFilter { public: @@ -18,75 +56,149 @@ class SignalFilter { FILTER_ABS_COSINE, FILTER_SHEPP } FilterID; - + + typedef enum { + FILTER_METHOD_INVALID, + FILTER_METHOD_CONVOLUTION, + FILTER_METHOD_FOURIER, + FILTER_METHOD_FOURIER_TABLE, + FILTER_METHOD_FFT, +#if HAVE_FFTW + FILTER_METHOD_FFTW, + FILTER_METHOD_RFFTW, +#endif + } FilterMethodID; + typedef enum { DOMAIN_INVALID, - DOMAIN_FREQ, + DOMAIN_FREQUENCY, DOMAIN_SPATIAL } DomainID; + static const char FILTER_ABS_BANDLIMIT_STR[]= "abs_bandlimit"; + static const char FILTER_ABS_SINC_STR[]= "abs_sinc"; + static const char FILTER_ABS_COS_STR[]= "abs_cos"; + static const char FILTER_ABS_HAMMING_STR[]= "abs_hamming"; + static const char FILTER_SHEPP_STR[]= "shepp"; + static const char FILTER_BANDLIMIT_STR[]= "bandlimit"; + static const char FILTER_SINC_STR[]= "sinc"; + static const char FILTER_COS_STR[]= "cos"; + static const char FILTER_HAMMING_STR[]= "hamming"; + static const char FILTER_TRIANGLE_STR[]= "triangle"; + + 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"; +#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"; + - SignalFilter (const char* filterName, double bw, double xmin, double xmax, int n, double param, const char* domainName, const int numInt); + SignalFilter (const char* filterName, const char* filterMethodName,double bw, double signalIncrement, int n, double param, const char* domainName, const int zeropad = 0, const int preinterpolationFactor = 1); - SignalFilter (const FilterID filt_type, double bw, double xmin, double xmax, int n, double param, const DomainID domain, const int numInt); + SignalFilter (const FilterID filt_type, FilterMethodID filterMethodID, double bw, double signalIncrement, int n, double param, const DomainID domain, const int zeropad = 0, const int preinterpolationFactor = 1); + + SignalFilter (const char* filterName, const char* domainName, double bw, double param); ~SignalFilter (void); 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;} + const string& nameFilter(void) const { return m_nameFilter;} 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;} - static double response (const char* const FilterName, const char* const DomainName, double bw, double x, double param); + double response (double x); - static double spatialResponseCalc (FilterID fType, double bw, double x, double param, int n); + static double spatialResponse (FilterID fType, double bw, double x, double param); + + static double frequencyResponse (FilterID fType, double bw, double u, double param); static double spatialResponseAnalytic (FilterID fType, double bw, double x, double param); - static double frequencyResponse (FilterID fType, double bw, double u, double param); + static double spatialResponseCalc (FilterID fType, double bw, double x, double param, int nIntegral); + + static void setNumIntegral(int nIntegral) {N_INTEGRAL = nIntegral;} 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, *m_vecRealFftSignal; + rfftw_plan m_realPlanForward, m_realPlanBackward; + fftw_complex* m_vecComplexFftInput, *m_vecComplexFftSignal; + fftw_plan m_complexPlanForward, m_complexPlanBackward; +#endif + bool m_fail; + string m_failMessage; string m_nameFilter; + string m_nameFilterMethod; string m_nameDomain; FilterID m_idFilter; + FilterMethodID m_idFilterMethod; DomainID m_idDomain; + double m_filterParam; + int m_traceLevel; + int m_zeropad; + int m_nOutputPoints; + int m_preinterpolationFactor; - static const char FILTER_ABS_BANDLIMIT_STR[]= "abs_bandlimit"; - static const char FILTER_ABS_SINC_STR[]= "abs_sinc"; - static const char FILTER_ABS_COS_STR[]= "abs_cos"; - static const char FILTER_ABS_HAMMING_STR[]= "abs_hamming"; - static const char FILTER_SHEPP_STR[]= "shepp"; - static const char FILTER_BANDLIMIT_STR[]= "bandlimit"; - static const char FILTER_SINC_STR[]= "sinc"; - static const char FILTER_COS_STR[]= "cos"; - static const char FILTER_HAMMING_STR[]= "hamming"; - static const char FILTER_TRIANGLE_STR[]= "triangle"; - - static const char DOMAIN_FREQ_STR[]= "freq"; - static const char DOMAIN_SPATIAL_STR[]= "spatial"; + static int N_INTEGRAL; - static FilterID convertFilterNameToID (const char* filterName); + static const bool haveAnalyticSpatial (const FilterID filterID); + 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 preInterpScale); -double spatialResponseCalc (double x, double param, int n) const; + double spatialResponseCalc (double x, double param) const; double spatialResponseAnalytic (double x, double param) const; @@ -99,5 +211,4 @@ double spatialResponseCalc (double x, double param, int n) const; }; - #endif