InverseFilterbankEngineCPU.h

//-*-C++-*-
 
/***************************************************************************
 *
 *   Copyright (C) 2009 by Willem van Straten
 *   Licensed under the Academic Free License version 2.1
 *
 ***************************************************************************/
 
// dspsr/Signal/General/dsp/InverseFilterbankEngineCPU.h
 
#ifndef __InverseFilterbankEngineCPU_h
#define __InverseFilterbankEngineCPU_h
 
#include "dsp/InverseFilterbankEngine.h"
 
#include "FTransform.h"
 
namespace dsp
{
 
  class InverseFilterbankEngineCPU : public dsp::InverseFilterbank::Engine
  {
 
  public:
 
    InverseFilterbankEngineCPU ();
 
    ~InverseFilterbankEngineCPU ();
 
    void setup (InverseFilterbank*);
 
    void set_scratch (float *);
 
    void perform (const TimeSeries* in, TimeSeries* out,
                  uint64_t npart, uint64_t in_step=0, uint64_t out_step=0);
 
    void perform (const TimeSeries * in,
                  TimeSeries * out,
                  TimeSeries* zero_DM_out,
                  uint64_t npart,
                  const uint64_t in_step=0,
                  const uint64_t out_step=0);
 
    void finish ();
 
  protected:
 
    bool real_to_complex;
 
    bool dual_sideband;
 
    bool verbose;
 
    Response* response;
 
    Response* zero_DM_response;
 
    Apodization* temporal_apodization;
    Apodization* spectral_apodization;
 
    unsigned input_npol;
 
    unsigned input_nchan;
    unsigned output_nchan;
 
    unsigned input_discard_neg;
    unsigned input_discard_pos;
    unsigned input_discard_total;
 
    unsigned output_discard_neg;
    unsigned output_discard_pos;
    unsigned output_discard_total;
 
    unsigned input_fft_length;
    unsigned output_fft_length;
 
    unsigned input_sample_step;
 
    unsigned output_sample_step;
 
    unsigned input_os_keep;
    unsigned input_os_discard;
 
 
    bool fft_plans_setup;
 
    bool pfb_dc_chan;
 
    bool pfb_all_chan;
 
    unsigned pfb_nchan;
 
    const TimeSeries* input;
    TimeSeries* output;
    TimeSeries* zero_DM_out;
 
    unsigned coarse_channels;
 
 
    unsigned fine_channels_per_coarse_channel;
 
 
    unsigned fine_channels_per_output_channel;
 
 
    unsigned coarse_channels_per_output_channel;
 
 
    unsigned output_channels_per_coarse_channel;
 
    unsigned spectral_steps;
 
    unsigned output_channels_per_spectral_step;
 
    void stitch (float* dest, unsigned ipol, unsigned fine_channel_offset, uint64_t input_offset_nfloat, bool shift_by_half_chan);
    void stitch_one (float* dest, unsigned ipol, unsigned chan_offset, uint64_t input_offset_nfloat, bool shift_by_half_chan);
 
    void filter (float* spectrum, unsigned ipol, unsigned chan_offset);
 
    void invert (float* spectrum, unsigned ipol, unsigned chan_offset, uint64_t output_offset_nfloat);
 
    void build_fine_channel_plan ();
    std::vector<unsigned> fine_channel_plan;
    unsigned matlab_fine_channel_plan (unsigned jchan);
 
  private:
 
 
    float* scratch;
 
    FTransform::Plan* forward;
 
    FTransform::Plan* backward;
 
    float* input_fft_scratch;
 
    unsigned input_fft_scratch_nfloat;
 
    float* input_time_scratch;
 
    unsigned input_time_scratch_nfloat;
 
    float* output_fft_scratch;
 
    unsigned output_fft_scratch_nfloat;
 
    float* stitch_scratch;
 
    unsigned stitch_scratch_nfloat;
 
    float* stitch_scratch_zero_DM;
  };
 
}
 
#endif