GNU Radio 3.6.5 C++ API
|
00001 /* -*- c++ -*- */ 00002 /* 00003 * Copyright 2010 Free Software Foundation, Inc. 00004 * 00005 * This file is part of GNU Radio 00006 * 00007 * GNU Radio is free software; you can redistribute it and/or modify 00008 * it under the terms of the GNU General Public License as published by 00009 * the Free Software Foundation; either version 3, or (at your option) 00010 * any later version. 00011 * 00012 * GNU Radio is distributed in the hope that it will be useful, 00013 * but WITHOUT ANY WARRANTY; without even the implied warranty of 00014 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the 00015 * GNU General Public License for more details. 00016 * 00017 * You should have received a copy of the GNU General Public License 00018 * along with GNU Radio; see the file COPYING. If not, write to 00019 * the Free Software Foundation, Inc., 51 Franklin Street, 00020 * Boston, MA 02110-1301, USA. 00021 */ 00022 00023 #ifndef INCLUDED_GRI_FFT_FILTER_FFF_GENERIC_H 00024 #define INCLUDED_GRI_FFT_FILTER_FFF_GENERIC_H 00025 00026 #include <gr_core_api.h> 00027 #include <gr_complex.h> 00028 #include <vector> 00029 00030 class gri_fft_real_fwd; 00031 class gri_fft_real_rev; 00032 00033 class GR_CORE_API gri_fft_filter_fff_generic 00034 { 00035 private: 00036 int d_ntaps; 00037 int d_nsamples; 00038 int d_fftsize; // fftsize = ntaps + nsamples - 1 00039 int d_decimation; 00040 gri_fft_real_fwd *d_fwdfft; // forward "plan" 00041 gri_fft_real_rev *d_invfft; // inverse "plan" 00042 int d_nthreads; // number of FFTW threads to use 00043 std::vector<float> d_tail; // state carried between blocks for overlap-add 00044 std::vector<float> d_new_taps; 00045 gr_complex *d_xformed_taps; // Fourier xformed taps 00046 00047 00048 void compute_sizes(int ntaps); 00049 int tailsize() const { return d_ntaps - 1; } 00050 00051 public: 00052 /*! 00053 * \brief Construct a FFT filter for float vectors with the given taps and decimation rate. 00054 * 00055 * This is the basic implementation for performing FFT filter for fast convolution 00056 * in other blocks for floating point vectors (such as gr_fft_filter_fff). 00057 * \param decimation The decimation rate of the filter (int) 00058 * \param taps The filter taps (float) 00059 * \param nthreads The number of threads for the FFT to use (int) 00060 */ 00061 gri_fft_filter_fff_generic (int decimation, const std::vector<float> &taps, 00062 int nthreads=1); 00063 ~gri_fft_filter_fff_generic (); 00064 00065 /*! 00066 * \brief Set new taps for the filter. 00067 * 00068 * Sets new taps and resets the class properties to handle different sizes 00069 * \param taps The filter taps (float) 00070 */ 00071 int set_taps (const std::vector<float> &taps); 00072 00073 /*! 00074 * \brief Set number of threads to use. 00075 */ 00076 void set_nthreads(int n); 00077 00078 /*! 00079 * \brief Get number of threads being used. 00080 */ 00081 int nthreads() const; 00082 00083 /*! 00084 * \brief Perform the filter operation 00085 * 00086 * \param nitems The number of items to produce 00087 * \param input The input vector to be filtered 00088 * \param output The result of the filter operation 00089 */ 00090 int filter (int nitems, const float *input, float *output); 00091 00092 }; 00093 00094 #endif /* INCLUDED_GRI_FFT_FILTER_FFF_GENERIC_H */