TimeFrequencySmooth.h

//-*-C++-*-
/***************************************************************************
 *
 *   Copyright (C) 2018 by Paul Demorest
 *   Licensed under the Academic Free License version 2.1
 *
 ***************************************************************************/
 
#ifndef __Pulsar_TimeFrequencySmooth_h
#define __Pulsar_TimeFrequencySmooth_h
 
#include "Pulsar/Algorithm.h"
#include "Pulsar/Archive.h"
 
namespace Pulsar {


  class TimeFrequencySmooth : public Algorithm
  {
 
    public:

      static TimeFrequencySmooth* factory (const std::string& name);

      static const std::vector<TimeFrequencySmooth*>& children ();
 
      TimeFrequencySmooth();


      virtual void smooth (std::vector<float> &smoothed, 
          std::vector<float> &raw, std::vector<float> &weight,
          std::vector<float> &freqs, std::vector<float> &times) = 0;

      virtual TextInterface::Parser* get_interface () { return 0; }

      virtual TimeFrequencySmooth* clone () const = 0;
 
    protected:
 
      void check_dimensions (std::vector<float> &smoothed, 
          std::vector<float> &raw, std::vector<float> &weight,
          std::vector<float> &freqs, std::vector<float> &times);

      unsigned idx(unsigned isubint, unsigned ichan) const { 
        return nchan*isubint + ichan;
      }

      unsigned idx(unsigned isubint, unsigned ichan, unsigned ipol) const { 
        return nchan*npol*isubint + npol*ichan + ipol;
      }
 
      // Current data dimensions
      unsigned nsub;
      unsigned nchan;
      unsigned npol;
 
    private:
      static void build ();
  };
 
  std::ostream& operator<< (std::ostream&, TimeFrequencySmooth*);
 
  std::istream& operator>> (std::istream&, TimeFrequencySmooth*&);
 
}
 
#endif