DynamicResponse.h

//-*-C++-*-
/***************************************************************************
 *
 *   Copyright (C) 2024 Willem van Straten
 *   Licensed under the Academic Free License version 2.1
 *
 ***************************************************************************/
 
// psrchive/Base/Extensions/Pulsar/DynamicResponse.h
 
#ifndef __DynamicResponse_h
#define __DynamicResponse_h
 
#include "Pulsar/ArchiveExtension.h"
#include "Pulsar/Profile.h"
 
#include <vector> 
#include <complex>
 
namespace Pulsar {
 
  class DynamicResponse : public Pulsar::Archive::Extension
  {
 
  public:
 
    DynamicResponse ();

    DynamicResponse (const DynamicResponse& extension);

    ~DynamicResponse ();

    DynamicResponse* clone () const { return new DynamicResponse(*this); }

    void set_nchan (unsigned);
    
    unsigned get_nchan () const;

    void set_ntime (unsigned);

    unsigned get_ntime () const;

    void set_npol (unsigned);
    
    unsigned get_npol () const;

    void set_minimum_epoch (const MJD& epoch) { minimum_epoch = epoch; }
    MJD get_minimum_epoch () const { return minimum_epoch; }

    void set_maximum_epoch (const MJD& epoch) { maximum_epoch = epoch; }
    MJD get_maximum_epoch () const { return maximum_epoch; }

    void set_centre_frequency (double freq) { centre_frequency = freq; }
    double get_centre_frequency () const { return centre_frequency; }

    void set_bandwidth (double freq) { bandwidth = freq; }
    double get_bandwidth () const { return bandwidth; }

    void set_impulse_pos (unsigned nsamp) { impulse_pos = nsamp; }

    unsigned get_impulse_pos () const { return impulse_pos; }

    void set_impulse_neg (unsigned nsamp) { impulse_neg = nsamp; }

    unsigned get_impulse_neg () const { return impulse_neg; }

    void resize_data ();
    
    std::vector<std::complex<double>>& get_data () { return response; }
    
    const std::vector<std::complex<double>>& get_data () const { return response; }

    /* @pre vector must have size=ntime*nchan*npol */
    void set_data (const std::vector<std::complex<double>>&);
 
  protected:

    unsigned nchan = 0;
    unsigned ntime = 0;
    unsigned npol = 0;

    unsigned impulse_pos = 0;

    unsigned impulse_neg = 0;

    MJD minimum_epoch;
    MJD maximum_epoch;

    double centre_frequency = 0;
    double bandwidth = 0;

    std::vector<std::complex<double>> response;            
  };
 
}
 
#endif