SKFilterbank.h

//-*-C++-*-
/***************************************************************************
 *
 *   Copyright (C) 2010 by Willem van Straten and Andrew Jameson
 *   Licensed under the Academic Free License version 2.1
 *
 ***************************************************************************/
 
// dspsr/Signal/General/dsp/SKFilterbank.h
 
#ifndef __SKFilterbank_h
#define __SKFilterbank_h
 
#include "dsp/Filterbank.h"
#include "ThreadContext.h"
 
namespace dsp {
  
 
  class SKFilterbank: public Filterbank {
 
  public:
 
    SKFilterbank ( unsigned _n_threads=1 );
    ~SKFilterbank ();
 
    class Engine;
 
    void set_engine (Engine*);
 
    void set_M (unsigned _M) { tscrunch = _M; }
 
    uint64_t get_skfb_inc (uint64_t blocksize);
 
    void set_output_tscr (TimeSeries * _output_tscr);
 
  protected:
 
    virtual void filterbank ();
    virtual void custom_prepare ();
 
    Reference::To<Engine> engine;
 
  private:
 
    unsigned tscrunch;
 
    unsigned debugd;
 
    ThreadContext* context;
 
    unsigned n_threads;
 
    unsigned thread_count;
 
    std::vector <pthread_t> ids;
 
    void start_threads ();
 
    void wait_threads ();
 
    void stop_threads ();
 
    void join_threads ();
 
    static void* sk_thread (void*);
 
    void thread ();
 
    enum State { Idle, Active, Quit };
 
    State state;
 
    std::vector <State> states;
 
    Reference::To<TimeSeries> output_tscr;
 
    std::vector <float> S1_tscr;
    std::vector <float> S2_tscr;
 
  };
 
  class SKFilterbank::Engine : public Reference::Able
  {
  public:
 
      virtual void setup () = 0;
 
      virtual void prepare (const dsp::TimeSeries* input, unsigned _nfft) = 0;
 
      virtual void perform (const dsp::TimeSeries* input, dsp::TimeSeries* output,
                            dsp::TimeSeries *output_tscr) = 0;
 
  }; 
}
 
#endif