FilterbankEngineCPU.h

//-*-C++-*-
 
/***************************************************************************
 *
 *   Copyright (C) 2009 by Willem van Straten
 *   Licensed under the Academic Free License version 2.1
 *
 ***************************************************************************/
 
// dspsr/Signal/General/dsp/FilterbankEngineCPU.h
 
#ifndef __FilterbankEngineCPU_h
#define __FilterbankEngineCPU_h
 
#include "dsp/FilterbankEngine.h"
 
namespace dsp
{
  class FilterbankEngineCPU : public Filterbank::Engine
  {
  public:
 
    FilterbankEngineCPU ();
 
    ~FilterbankEngineCPU ();
 
    void setup (dsp::Filterbank*);
 
    void set_scratch (float *);
 
    void perform (const dsp::TimeSeries* in, dsp::TimeSeries* out,
                  uint64_t npart, uint64_t in_step, uint64_t out_step);
 
    void perform (const dsp::TimeSeries* in, dsp::TimeSeries* out, dsp::TimeSeries* zero_DM_out,
                  uint64_t npart, uint64_t in_step, uint64_t out_step);
 
 
    void finish ();
 
    FTransform::Plan* get_forward ();
 
    FTransform::Plan* get_backward ();
 
  protected:
 
    FTransform::Plan* forward;
 
    FTransform::Plan* backward;
 
    bool real_to_complex;
 
    float* scratch;
 
    float* freq_domain_scratch[3];
 
    float* time_domain_scratch;
 
    float* windowed_time_domain_scratch;
 
    const dsp::Response* response;
 
    const dsp::Response* zero_DM_response;
 
    const dsp::Apodization* temporal_apodization;
    const dsp::Apodization* spectral_apodization;
 
    dsp::Response* passband;
 
    bool matrix_convolution;
 
    unsigned nchan_subband;
 
    unsigned freq_res;
 
    unsigned nfilt_pos;
 
    unsigned nfilt_neg;
 
    uint64_t nsamp_fft;
 
    unsigned nkeep;
 
    bool verbose;
 
  private:
 
    // size of the forward FFT size
    unsigned bigfftsize;
 
  };
 
}
 
#endif