CalibrationInterpolator.h

//-*-C++-*-
/***************************************************************************
 *
 *   Copyright (C) 2022 by Willem van Straten
 *   Licensed under the Academic Free License version 2.1
 *
 ***************************************************************************/
 
// psrchive/More/Polarimetry/Pulsar/CalibrationInterpolator.h
 
#ifndef __CalibrationInterpolator_H
#define __CalibrationInterpolator_H
 
#include "Pulsar/Calibrator.h"
#include <map>
 
namespace Pulsar {
 
  class CalibrationInterpolatorExtension;
  class FluxCalibratorExtension;
  class PolnCalibratorExtension;
  class CalibratorStokes;
  class Integration;
  class SplineSmooth2D;
  
  class CalibrationInterpolator : public Calibrator::Variation
  {
    void construct (Archive*);
 
  public:

    CalibrationInterpolator (Calibrator*);

    CalibrationInterpolator (Archive*);

    ~CalibrationInterpolator ();

    const CalibrationInterpolatorExtension* get_extension();

    const Calibrator::Type* get_type () const;

    bool update (const Integration* subint)
    { return update<Integration>(subint); }


    template<class Container>
    bool update (const Container*);
    
  protected:

    Reference::To<const CalibrationInterpolatorExtension> interpolator;
    
    Reference::To<CalibratorStokes> calpoln;

    Reference::To<PolnCalibratorExtension> feedpar;

    Reference::To<FluxCalibratorExtension> fluxcal;
 
    std::map< unsigned, Reference::To<SplineSmooth2D> > feedpar_splines;
    std::map< unsigned, Reference::To<SplineSmooth2D> > calpoln_splines;
    std::map< unsigned, Reference::To<SplineSmooth2D> > fluxcal_splines;
 
    MJD last_computed;

    bool update (const MJD& epoch, const std::vector<double>& freqs);
  };
 
  template<class Container>
  bool CalibrationInterpolator::update (const Container* container)
  {
    unsigned nchan = container->get_nchan();
    std::vector<double> frequency (nchan);
  
    for (unsigned ichan=0; ichan<nchan; ++ichan)
      frequency[ichan] = container->get_centre_frequency (ichan);
  
    return update (container->get_epoch (), frequency);
  }
 
}
 
#endif