/*****************************************************************************
-** FILE IDENTIFICATION
+** File IDENTIFICATION
**
** Name: filter.cpp
** Purpose: Routines for signal-procesing filters
** This is part of the CTSim program
** Copyright (C) 1983-2000 Kevin Rosenberg
**
-** $Id: filter.cpp,v 1.2 2000/06/20 17:54:51 kevin Exp $
+** $Id: filter.cpp,v 1.11 2000/07/05 17:24:33 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
/* NAME
- * filter_generate Generate a filter
+ * SignalFilter::SignalFilter Construct a signal
*
* SYNOPSIS
- * f = filter_generate (filt_type, bw, xmin, xmax, n, param, domain, analytic)
- * double f Generated filter vector
- * int filt_type Type of filter wanted
- * double bw Bandwidth of filter
- * double xmin, xmax Filter limits
- * int n Number of points in filter
- * double param General input parameter to filters
- * int domain FREQ or SPATIAL domain wanted
- * int numint Number if intervals for calculating
- * discrete inverse fourier xform
- * for spatial domain filters. For
- * ANALYTIC solutions, use numint = 0
+ * f = SignalFilter (filt_type, bw, filterMin, filterMax, n, param, domain, analytic)
+ * double f Generated filter vector
+ * int filt_type Type of filter wanted
+ * double bw Bandwidth of filter
+ * double filterMin, filterMax Filter limits
+ * int nSignalPoints Number of points in signal
+ * double param General input parameter to filters
+ * int domain FREQUENCY or SPATIAL domain wanted
+ * int numint Number if intervals for calculating discrete inverse fourier xform
+ * for spatial domain filters. For ANALYTIC solutions, use numint = 0
*/
-SignalFilter::SignalFilter (const FilterType filt_type, double bw, double xmin, double xmax, int n, double param, const DomainType domain, int numint)
+SignalFilter::SignalFilter (const char* filterName, const char* filterMethodName, double bw, double signalIncrement, int nSignalPoints, double param, const char* domainName, int numIntegral = 0)
+{
+ m_vecFilter = NULL;
+ m_vecFourierCosTable = NULL;
+ m_vecFourierSinTable = NULL;
+ m_idFilter = convertFilterNameToID (filterName);
+ if (m_idFilter == FILTER_INVALID) {
+ m_fail = true;
+ m_failMessage = "Invalid Filter name ";
+ m_failMessage += filterName;
+ return;
+ }
+ m_idFilterMethod = convertFilterMethodNameToID (filterMethodName);
+ if (m_idFilterMethod == FILTER_METHOD_INVALID) {
+ m_fail = true;
+ m_failMessage = "Invalid filter method name ";
+ m_failMessage += filterMethodName;
+ return;
+ }
+ m_idDomain = convertDomainNameToID (domainName);
+ if (m_idDomain == DOMAIN_INVALID) {
+ m_fail = true;
+ m_failMessage = "Invalid domain name ";
+ m_failMessage += domainName;
+ return;
+ }
+ init (m_idFilter, m_idFilterMethod, bw, signalIncrement, nSignalPoints, param, m_idDomain, numIntegral);
+}
+
+SignalFilter::SignalFilter (const FilterID filterID, const FilterMethodID filterMethodID, double bw, double signalIncrement, int nSignalPoints, double param, const DomainID domainID, int numIntegral = 0)
+{
+ init (filterID, filterMethodID, bw, signalIncrement, nSignalPoints, param, domainID, numIntegral);
+}
+
+SignalFilter::SignalFilter (const char* filterName, const char* domainName, double bw, double param, int numIntegral = 0)
{
- m_vecFilter = new double [n];
- m_filterType = filt_type;
m_bw = bw;
+ m_nSignalPoints = 0;
+ m_nFilterPoints = 0;
+ m_vecFilter = NULL;
+ m_vecFourierCosTable = NULL;
+ m_vecFourierSinTable = NULL;
+ m_filterParam = param;
+ m_numIntegral = numIntegral;
+ m_idFilter = convertFilterNameToID (filterName);
+ if (m_idFilter == FILTER_INVALID) {
+ m_fail = true;
+ m_failMessage = "Invalid Filter name ";
+ m_failMessage += filterName;
+ return;
+ }
+ m_idDomain = convertDomainNameToID (domainName);
+ if (m_idDomain == DOMAIN_INVALID) {
+ m_fail = true;
+ m_failMessage = "Invalid domain name ";
+ m_failMessage += domainName;
+ return;
+ }
+}
- double xinc = (xmax - xmin) / (n - 1);
+void
+SignalFilter::init (const FilterID filterID, const FilterMethodID filterMethodID, double bw, double signalIncrement, int nSignalPoints, double param, const DomainID domainID, int numint)
+{
+ m_bw = bw;
+ m_idFilter = filterID;
+ m_idDomain = domainID;
+ m_idFilterMethod = filterMethodID;
+ if (m_idFilter == FILTER_INVALID || m_idDomain == DOMAIN_INVALID || m_idFilterMethod == FILTER_METHOD_INVALID) {
+ m_fail = true;
+ return;
+ }
+ m_traceLevel = TRACE_NONE;
+ m_nameFilter = convertFilterIDToName (m_idFilter);
+ m_nameDomain = convertDomainIDToName (m_idDomain);
+ m_nameFilterMethod = convertFilterMethodIDToName (m_idFilterMethod);
+ m_fail = false;
+ m_nSignalPoints = nSignalPoints;
+ m_signalInc = signalIncrement;
+ m_filterParam = param;
+
+ if (m_idFilterMethod == FILTER_METHOD_FOURIER) {
+ int nFourier = m_nSignalPoints * m_nSignalPoints + 1;
+ double angleIncrement = (2. * PI) / m_nSignalPoints;
+ m_vecFourierCosTable = new double[ nFourier ];
+ m_vecFourierSinTable = new double[ nFourier ];
+ for (int i = 0; i < nFourier; i++) {
+ m_vecFourierCosTable[i] = cos (angleIncrement * i);
+ m_vecFourierSinTable[i] = sin (angleIncrement * i);
+ }
+ m_nFilterPoints = m_nSignalPoints;
+ m_filterMin = -1. / (2 * m_signalInc);
+ m_filterMax = 1. / (2 * m_signalInc);
+ m_filterInc = (m_filterMax - m_filterMin) / m_nFilterPoints;
+ m_vecFilter = new double [m_nFilterPoints];
+ int halfFilter = m_nFilterPoints / 2;
+ for (int i = 0; i < halfFilter; i++)
+ m_vecFilter[i] = static_cast<double>(i) / (halfFilter - 1) / (2 * m_signalInc);
+ for (int i = 0; i < halfFilter; i++)
+ m_vecFilter[m_nFilterPoints - i - 1] = static_cast<double>(i+1) / (halfFilter - 1) / (2 * m_signalInc);
+ if (halfFilter % 2) // odd
+ m_vecFilter[halfFilter] = 1 / (2 * m_signalInc);
+ } else if (m_idFilterMethod == FILTER_METHOD_FFT || m_idFilterMethod == FILTER_METHOD_FFT_ZEROPAD_2 || m_idFilterMethod == FILTER_METHOD_FFT_ZEROPAD_4) {
+ m_nFilterPoints = m_nSignalPoints;
+ if (m_idFilterMethod == FILTER_METHOD_FFT_ZEROPAD_2 || m_idFilterMethod == FILTER_METHOD_FFT_ZEROPAD_4) {
+ double logBase2 = log(m_nSignalPoints) / log(2);
+ int nextPowerOf2 = static_cast<int>(floor(logBase2)) + 1;
+ if (m_idFilterMethod == FILTER_METHOD_FFT_ZEROPAD_4)
+ nextPowerOf2++;
+ if (logBase2 != floor(logBase2))
+ nextPowerOf2++;
+ m_nFilterPoints = 1 << nextPowerOf2;
+ cout << "nFilterPoints = " << m_nFilterPoints << endl;
+ }
+ m_filterMin = -1. / (2 * m_signalInc);
+ m_filterMax = 1. / (2 * m_signalInc);
+ m_filterInc = (m_filterMax - m_filterMin) / m_nFilterPoints;
+ m_vecFilter = new double [m_nFilterPoints];
+ int halfFilter = m_nFilterPoints / 2;
+ for (int i = 0; i < halfFilter; i++)
+ m_vecFilter[i] = static_cast<double>(i) / (halfFilter - 1) / (2 * m_signalInc) / m_nSignalPoints;
+ for (int i = 0; i < halfFilter; i++)
+ m_vecFilter[m_nFilterPoints - i - 1] = static_cast<double>(i+1) / (halfFilter - 1) / (2 * m_signalInc) / m_nSignalPoints;
+ if (halfFilter % 2) // odd
+ m_vecFilter[halfFilter] = 1 / (2 * m_signalInc) / m_nSignalPoints;
+
+#if HAVE_FFTW
+ m_planForward = fftw_create_plan (m_nFilterPoints, FFTW_FORWARD, FFTW_ESTIMATE);
+ m_planBackward = fftw_create_plan (m_nFilterPoints, FFTW_BACKWARD, FFTW_ESTIMATE);
+#endif
+ }
- if (m_filterType == FILTER_SHEPP) {
- double a = 2 * m_bw;
- double c = - 4. / (a * a);
- int center = (n - 1) / 2;
- int sidelen = center;
- m_vecFilter[center] = 4. / (a * a);
+ if (m_idFilterMethod == FILTER_METHOD_CONVOLUTION) {
+ m_nFilterPoints = 2 * m_nSignalPoints - 1;
+ m_filterMin = -m_signalInc * (m_nSignalPoints - 1);
+ m_filterMax = m_signalInc * (m_nSignalPoints - 1);
+ m_filterInc = (m_filterMax - m_filterMin) / (m_nFilterPoints - 1);
+ m_numIntegral = numint;
+ m_vecFilter = new double[ m_nFilterPoints ];
- for (int i = 1; i <= sidelen; i++ )
- m_vecFilter [center + i] = m_vecFilter [center - i] = c / (4 * (i * i) - 1);
- } else if (domain == D_FREQ) {
- double x;
- int i;
- for (x = xmin, i = 0; i < n; x += xinc, i++)
- m_vecFilter[i] = frequencyResponse (x, param);
- } else if (domain == D_SPATIAL) {
- double x;
- int i;
- for (x = xmin, i = 0; i < n; x += xinc, i++)
- if (numint == 0)
- m_vecFilter[i] = spatialResponseAnalytic (x, param);
- else
- m_vecFilter[i] = spatialResponseCalc (x, param, numint);
- } else {
- sys_error (ERR_WARNING, "Illegal domain %d [filt_generate]", domain);
+ if (m_idFilter == FILTER_SHEPP) {
+ double a = 2 * m_bw;
+ double c = - 4. / (a * a);
+ int center = (m_nFilterPoints - 1) / 2;
+ int sidelen = center;
+ m_vecFilter[center] = 4. / (a * a);
+
+ for (int i = 1; i <= sidelen; i++ )
+ m_vecFilter [center + i] = m_vecFilter [center - i] = c / (4 * (i * i) - 1);
+ } else if (m_idDomain == DOMAIN_FREQUENCY) {
+ double x;
+ int i;
+ for (x = m_filterMin, i = 0; i < m_nFilterPoints; x += m_filterInc, i++)
+ m_vecFilter[i] = frequencyResponse (x, param);
+ } else if (m_idDomain == DOMAIN_SPATIAL) {
+ double x;
+ int i;
+ for (x = m_filterMin, i = 0; i < m_nFilterPoints; x += m_filterInc, i++)
+ if (numint == 0)
+ m_vecFilter[i] = spatialResponseAnalytic (x, param);
+ else
+ m_vecFilter[i] = spatialResponseCalc (x, param, numint);
+ } else {
+ m_failMessage = "Illegal domain name ";
+ m_failMessage += m_idDomain;
+ m_fail = true;
+ }
}
}
SignalFilter::~SignalFilter (void)
{
- delete m_vecFilter;
+ delete m_vecFilter;
+ delete m_vecFourierSinTable;
+ delete m_vecFourierCosTable;
+#if HAVE_FFTW
+ if (m_idFilterMethod == FILTER_METHOD_FFT) {
+ fftw_destroy_plan(m_planForward);
+ fftw_destroy_plan(m_planBackward);
+ }
+#endif
+}
+
+
+const SignalFilter::FilterID
+SignalFilter::convertFilterNameToID (const char *filterName)
+{
+ FilterID filterID = FILTER_INVALID;
+
+ if (strcasecmp (filterName, FILTER_BANDLIMIT_STR) == 0)
+ filterID = FILTER_BANDLIMIT;
+ else if (strcasecmp (filterName, FILTER_HAMMING_STR) == 0)
+ filterID = FILTER_G_HAMMING;
+ else if (strcasecmp (filterName, FILTER_SINC_STR) == 0)
+ filterID = FILTER_SINC;
+ else if (strcasecmp (filterName, FILTER_COS_STR) == 0)
+ filterID = FILTER_COSINE;
+ else if (strcasecmp (filterName, FILTER_TRIANGLE_STR) == 0)
+ filterID = FILTER_TRIANGLE;
+ else if (strcasecmp (filterName, FILTER_ABS_BANDLIMIT_STR) == 0)
+ filterID = FILTER_ABS_BANDLIMIT;
+ else if (strcasecmp (filterName, FILTER_ABS_HAMMING_STR) == 0)
+ filterID = FILTER_ABS_G_HAMMING;
+ else if (strcasecmp (filterName, FILTER_ABS_SINC_STR) == 0)
+ filterID = FILTER_ABS_SINC;
+ else if (strcasecmp (filterName, FILTER_ABS_COS_STR) == 0)
+ filterID = FILTER_ABS_COSINE;
+ else if (strcasecmp (filterName, FILTER_SHEPP_STR) == 0)
+ filterID = FILTER_SHEPP;
+
+ return (filterID);
+}
+
+const char *
+SignalFilter::convertFilterIDToName (const FilterID filterID)
+{
+ const char *name = "";
+
+ if (filterID == FILTER_SHEPP)
+ name = FILTER_SHEPP_STR;
+ else if (filterID == FILTER_ABS_COSINE)
+ name = FILTER_ABS_COS_STR;
+ else if (filterID == FILTER_ABS_SINC)
+ name = FILTER_ABS_SINC_STR;
+ else if (filterID == FILTER_ABS_G_HAMMING)
+ name = FILTER_ABS_HAMMING_STR;
+ else if (filterID == FILTER_ABS_BANDLIMIT)
+ name = FILTER_ABS_BANDLIMIT_STR;
+ else if (filterID == FILTER_COSINE)
+ name = FILTER_COS_STR;
+ else if (filterID == FILTER_SINC)
+ name = FILTER_SINC_STR;
+ else if (filterID == FILTER_G_HAMMING)
+ name = FILTER_HAMMING_STR;
+ else if (filterID == FILTER_BANDLIMIT)
+ name = FILTER_BANDLIMIT_STR;
+ else if (filterID == FILTER_TRIANGLE)
+ name = FILTER_TRIANGLE_STR;
+
+ return (name);
+}
+
+const SignalFilter::FilterMethodID
+SignalFilter::convertFilterMethodNameToID (const char* const filterMethodName)
+{
+ FilterMethodID fmID = FILTER_METHOD_INVALID;
+
+ if (strcasecmp (filterMethodName, FILTER_METHOD_CONVOLUTION_STR) == 0)
+ fmID = FILTER_METHOD_CONVOLUTION;
+ else if (strcasecmp (filterMethodName, FILTER_METHOD_FOURIER_STR) == 0)
+ fmID = FILTER_METHOD_FOURIER;
+ else if (strcasecmp (filterMethodName, FILTER_METHOD_FFT_STR) == 0)
+ fmID = FILTER_METHOD_FFT;
+ else if (strcasecmp (filterMethodName, FILTER_METHOD_FFT_ZEROPAD_2_STR) == 0)
+ fmID = FILTER_METHOD_FFT_ZEROPAD_2;
+ else if (strcasecmp (filterMethodName, FILTER_METHOD_FFT_ZEROPAD_4_STR) == 0)
+ fmID = FILTER_METHOD_FFT_ZEROPAD_4;
+ else if (strcasecmp (filterMethodName, FILTER_METHOD_FFT_ZEROPAD_6_STR) == 0)
+ fmID = FILTER_METHOD_FFT_ZEROPAD_6;
+
+ return (fmID);
+}
+
+const char *
+SignalFilter::convertFilterMethodIDToName (const FilterMethodID fmID)
+{
+ const char *name = "";
+
+ if (fmID == FILTER_METHOD_CONVOLUTION)
+ return (FILTER_METHOD_CONVOLUTION_STR);
+ else if (fmID == FILTER_METHOD_FOURIER)
+ return (FILTER_METHOD_FOURIER_STR);
+ else if (fmID == FILTER_METHOD_FFT)
+ return (FILTER_METHOD_FFT_STR);
+ else if (fmID == FILTER_METHOD_FFT_ZEROPAD_2)
+ return (FILTER_METHOD_FFT_ZEROPAD_2_STR);
+ else if (fmID == FILTER_METHOD_FFT_ZEROPAD_4)
+ return (FILTER_METHOD_FFT_ZEROPAD_4_STR);
+ else if (fmID == FILTER_METHOD_FFT_ZEROPAD_6)
+ return (FILTER_METHOD_FFT_ZEROPAD_6_STR);
+
+ return (name);
+}
+
+const SignalFilter::DomainID
+SignalFilter::convertDomainNameToID (const char* const domainName)
+{
+ DomainID dID = DOMAIN_INVALID;
+
+ if (strcasecmp (domainName, DOMAIN_SPATIAL_STR) == 0)
+ dID = DOMAIN_SPATIAL;
+ else if (strcasecmp (domainName, DOMAIN_FREQUENCY_STR) == 0)
+ dID = DOMAIN_FREQUENCY;
+
+ return (dID);
+}
+
+const char *
+SignalFilter::convertDomainIDToName (const DomainID domain)
+{
+ const char *name = "";
+
+ if (domain == DOMAIN_SPATIAL)
+ return (DOMAIN_SPATIAL_STR);
+ else if (domain == DOMAIN_FREQUENCY)
+ return (DOMAIN_FREQUENCY_STR);
+
+ return (name);
+}
+
+
+void
+SignalFilter::filterSignal (const float input[], double output[]) const
+{
+ if (m_idFilterMethod == FILTER_METHOD_CONVOLUTION) {
+ for (int i = 0; i < m_nSignalPoints; i++)
+ output[i] = convolve (input, m_signalInc, i, m_nSignalPoints);
+ } else if (m_idFilterMethod == FILTER_METHOD_FOURIER) {
+ complex<double> fftSignal[m_nSignalPoints];
+ complex<double> complexOutput[m_nSignalPoints];
+ complex<double> filteredSignal[m_nSignalPoints];
+ finiteFourierTransform (input, fftSignal, m_nSignalPoints, -1);
+ dotProduct (m_vecFilter, fftSignal, filteredSignal, m_nSignalPoints);
+ finiteFourierTransform (filteredSignal, complexOutput, m_nSignalPoints, 1);
+ for (int i = 0; i < m_nSignalPoints; i++)
+ output[i] = complexOutput[i].real();
+ } else if (m_idFilterMethod == FILTER_METHOD_FFT || FILTER_METHOD_FFT_ZEROPAD_2 || FILTER_METHOD_FFT_ZEROPAD_4) {
+ fftw_complex in[m_nFilterPoints], out[m_nFilterPoints];
+ for (int i = 0; i < m_nSignalPoints; i++) {
+ in[i].re = input[i];
+ in[i].im = 0;
+ }
+ for (int i = m_nSignalPoints; i < m_nFilterPoints; i++) {
+ in[i].re = in[i].im = 0; // ZeroPad
+ }
+ fftw_one(m_planForward, in, out);
+ for (int i = 0; i < m_nFilterPoints; i++) {
+ out[i].re = m_vecFilter[i] * out[i].re;
+ out[i].im = m_vecFilter[i] * out[i].im;
+ }
+ fftw_one(m_planBackward, out, in);
+ for (int i = 0; i < m_nSignalPoints; i++)
+ output[i] = in[i].re;
+ }
}
+double
+SignalFilter::response (double x)
+{
+ double response = 0;
+
+ if (m_idDomain == DOMAIN_SPATIAL)
+ response = spatialResponse (m_idFilter, m_bw, x, m_filterParam, m_numIntegral);
+ else if (m_idDomain == DOMAIN_FREQUENCY)
+ response = frequencyResponse (m_idFilter, m_bw, x, m_filterParam);
+
+ return (response);
+}
+
+
+double
+SignalFilter::spatialResponse (FilterID filterID, double bw, double x, double param, int nIntegral = 0)
+{
+ if (nIntegral == 0)
+ return spatialResponseAnalytic (filterID, bw, x, param);
+ else
+ return spatialResponseCalc (filterID, bw, x, param, nIntegral);
+}
/* NAME
* filter_spatial_response_calc Calculate filter by discrete inverse fourier
*/
double
-SignalFilter::spatialResponseCalc (double x, double param, int n) const
+SignalFilter::spatialResponseCalc (double x, double param, int nIntegral) const
{
- return (spatialResponseCalc (m_filterType, m_bw, x, param, n));
+ return (spatialResponseCalc (m_idFilter, m_bw, x, param, nIntegral));
}
double
-SignalFilter::spatialResponseCalc (FilterType fType, double bw, double x, double param, int n)
+SignalFilter::spatialResponseCalc (FilterID filterID, double bw, double x, double param, int n)
{
double zmin, zmax;
- if (fType == FILTER_TRIANGLE) {
+ if (filterID == FILTER_TRIANGLE) {
zmin = 0;
zmax = bw;
} else {
double z = zmin;
double q [n];
for (int i = 0; i < n; i++, z += zinc)
- q[i] = frequencyResponse (fType, bw, z, param) * cos (TWOPI * z * x);
+ q[i] = frequencyResponse (filterID, bw, z, param) * cos (TWOPI * z * x);
double y = 2 * integrateSimpson (zmin, zmax, q, n);
double
SignalFilter::frequencyResponse (double u, double param) const
{
- return frequencyResponse (m_filterType, m_bw, u, param);
+ return frequencyResponse (m_idFilter, m_bw, u, param);
}
double
-SignalFilter::frequencyResponse (FilterType fType, double bw, double u, double param)
+SignalFilter::frequencyResponse (FilterID filterID, double bw, double u, double param)
{
double q;
double au = fabs (u);
- switch (fType) {
+ switch (filterID) {
case FILTER_BANDLIMIT:
if (au >= bw / 2)
q = 0.;
break;
default:
q = 0;
- sys_error (ERR_WARNING, "Frequency response for filter %d not implemented [filter_frequency_response]", fType);
+ sys_error (ERR_WARNING, "Frequency response for filter %d not implemented [filter_frequency_response]", filterID);
break;
}
return (q);
double
SignalFilter::spatialResponseAnalytic (double x, double param) const
{
- return spatialResponseAnalytic (m_filterType, m_bw, x, param);
+ return spatialResponseAnalytic (m_idFilter, m_bw, x, param);
}
double
-SignalFilter::spatialResponseAnalytic (FilterType fType, double bw, double x, double param)
+SignalFilter::spatialResponseAnalytic (FilterID filterID, double bw, double x, double param)
{
double q, temp;
double u = TWOPI * x;
double b = PI / bw;
double b2 = TWOPI / bw;
- switch (fType) {
+ switch (filterID) {
case FILTER_BANDLIMIT:
q = bw * sinc(u * w, 1.0);
break;
break;
case FILTER_ABS_SINC:
default:
- sys_error (ERR_WARNING, "Analytic filter type %d not implemented [filter_spatial_response_analytic]", fType);
+ sys_error (ERR_WARNING, "Analytic filter type %d not implemented [filter_spatial_response_analytic]", filterID);
q = 0;
break;
}
double
SignalFilter::integral_abscos (double u, double w)
{
- if (fabs (u) > F_EPSILON)
- return (cos(u * w) - 1) / (u * u) + w / u * sin (u * w);
- else
- return (w * w / 2);
+ return (fabs (u) > F_EPSILON
+ ? (cos(u * w) - 1) / (u * u) + w / u * sin (u * w)
+ : (w * w / 2));
}
*/
double
-SignalFilter::convolve (const double func[], const double dx, const int n, const int np, const FunctionSymmetry func_type) const
+SignalFilter::convolve (const double func[], const double dx, const int n, const int np) const
{
double sum = 0.0;
- if (func_type == FUNC_BOTH) {
#if UNOPTIMIZED_CONVOLUTION
- for (int i = 0; i < np; i++)
- sum += func[i] * m_vecFilter[n - i + (np - 1)];
+ for (int i = 0; i < np; i++)
+ sum += func[i] * m_vecFilter[n - i + (np - 1)];
#else
- double* f2 = m_vecFilter + n + (np - 1);
- for (int i = 0; i < np; i++)
- sum += *func++ * *f2--;
+ double* f2 = m_vecFilter + n + (np - 1);
+ for (int i = 0; i < np; i++)
+ sum += *func++ * *f2--;
#endif
- } else if (func_type == FUNC_EVEN) {
- for (int i = 0; i < np; i++) {
- int k = abs (n - i);
- sum += func[i] * m_vecFilter[k];
- }
- } else if (func_type == FUNC_ODD) {
- for (int i = 0; i < np; i++) {
- int k = n - i;
- if (k < 0)
- sum -= func[i] * m_vecFilter[k];
- else
- sum += func[i] * m_vecFilter[k];
- }
- } else
- sys_error (ERR_WARNING, "Illegal function type %d [convolve]", func_type);
return (sum * dx);
}
double
-SignalFilter::convolve (const float func[], const double dx, const int n, const int np, const FunctionSymmetry func_type) const
+SignalFilter::convolve (const float func[], const double dx, const int n, const int np) const
{
double sum = 0.0;
- if (func_type == FUNC_BOTH) {
#if UNOPTIMIZED_CONVOLUTION
- for (int i = 0; i < np; i++)
- sum += func[i] * m_vecFilter[n - i + (np - 1)];
+for (int i = 0; i < np; i++)
+ sum += func[i] * m_vecFilter[n - i + (np - 1)];
#else
- double* f2 = m_vecFilter + n + (np - 1);
- for (int i = 0; i < np; i++)
- sum += *func++ * *f2--;
+double* f2 = m_vecFilter + n + (np - 1);
+for (int i = 0; i < np; i++)
+ sum += *func++ * *f2--;
#endif
- } else if (func_type == FUNC_EVEN) {
- for (int i = 0; i < np; i++) {
- int k = abs (n - i);
- sum += func[i] * m_vecFilter[k];
+
+ return (sum * dx);
+}
+
+
+void
+SignalFilter::finiteFourierTransform (const float input[], complex<double> output[], const int n, int direction)
+{
+ if (direction < 0)
+ direction = -1;
+ else
+ direction = 1;
+
+ double angleIncrement = 2 * PI / n;
+ for (int i = 0; i < n; i++) {
+ double sumReal = 0;
+ double sumImag = 0;
+ for (int j = 0; j < n; j++) {
+ double angle = i * j * angleIncrement * direction;
+ sumReal += input[j] * cos(angle);
+ sumImag += input[j] * sin(angle);
+ }
+ if (direction < 0) {
+ sumReal /= n;
+ sumImag /= n;
}
- } else if (func_type == FUNC_ODD) {
- for (int i = 0; i < np; i++) {
- int k = n - i;
- if (k < 0)
- sum -= func[i] * m_vecFilter[k];
- else
- sum += func[i] * m_vecFilter[k];
+ output[i] = complex<double> (sumReal, sumImag);
+ }
+}
+
+
+void
+SignalFilter::finiteFourierTransform (const complex<double> input[], complex<double> output[], const int n, int direction)
+{
+ if (direction < 0)
+ direction = -1;
+ else
+ direction = 1;
+
+ double angleIncrement = 2 * PI / n;
+ for (int i = 0; i < n; i++) {
+ complex<double> sum (0,0);
+ for (int j = 0; j < n; j++) {
+ double angle = i * j * angleIncrement * direction;
+ complex<double> exponentTerm (cos(angle), sin(angle));
+ sum += input[j] * exponentTerm;
+ }
+ if (direction < 0) {
+ sum /= n;
}
- } else
- sys_error (ERR_WARNING, "Illegal function type %d [convolve]", func_type);
+ output[i] = sum;
+ }
+}
- return (sum * dx);
+void
+SignalFilter::finiteFourierTransform (const float input[], complex<double> output[], int direction) const
+{
+ if (direction < 0)
+ direction = -1;
+ else
+ direction = 1;
+
+ for (int i = 0; i < m_nSignalPoints; i++) {
+ double sumReal = 0, sumImag = 0;
+ for (int j = 0; j < m_nSignalPoints; j++) {
+ int tableIndex = i * j;
+ if (direction > 0) {
+ sumReal += input[i] * m_vecFourierCosTable[tableIndex];
+ sumImag += input[i] * m_vecFourierSinTable[tableIndex];
+ } else {
+ sumReal += input[i] * m_vecFourierCosTable[tableIndex];
+ sumImag -= input[i] * m_vecFourierSinTable[tableIndex];
+ }
+ }
+ if (direction < 0) {
+ sumReal /= m_nSignalPoints;
+ sumImag /= m_nSignalPoints;
+ }
+ output[i] = complex<double> (sumReal, sumImag);
+ }
+}
+
+// (a+bi) * (c + di) = (ac - db) + (bc + da)i
+#if 0
+void
+SignalFilter::finiteFourierTransform (const complex<double> input[], complex<double> output[], int direction) const
+{
+ if (direction < 0)
+ direction = -1;
+ else
+ direction = 1;
+
+ for (int i = 0; i < m_nSignalPoints; i++) {
+ double sumReal = 0, sumImag = 0;
+ for (int j = 0; j < m_nSignalPoints; j++) {
+ int tableIndex = i * j;
+ if (direction > 0) {
+ sumReal += input[i] * m_vecFourierCosTable[tableIndex];
+ sumImag += input[i] * m_vecFourierSinTable[tableIndex];
+ } else {
+ sumReal += input[i] * m_vecFourierCosTable[tableIndex];
+ sumImag -= input[i] * m_vecFourierSinTable[tableIndex];
+ }
+ }
+ if (direction > 0) {
+ sumReal /= m_nSignalPoints;
+ sumImag /= m_nSignalPoints;
+ }
+ output[i] = complex<double> (sumReal, sumImag);
+ }
}
+#endif
+void
+SignalFilter::dotProduct (const double v1[], const complex<double> v2[], complex<double> output[], const int n)
+{
+ for (int i = 0; i < n; i++)
+ output[i] = v1[i] * v2[i];
+}