MeanArrivalTime.h

//-*-C++-*-
/***************************************************************************
 *
 *   Copyright (C) 2024 by Willem van Straten and Matthew Bailes
 *   Licensed under the Academic Free License version 2.1
 *
 ***************************************************************************/
 
// psrchive/More/Timing/Pulsar/MeanArrivalTime.h
 
#ifndef __Pulsar_MeanArrivalTime_h
#define __Pulsar_MeanArrivalTime_h
 
#include "LinearRegression.h"
#include "ReferenceAble.h"
#include "Estimate.h"
 
#include <vector>
 
namespace Pulsar {

  class MeanArrivalTime : public Reference::Able
  {
    std::vector< Estimate<double> > delay_sec;
    std::vector< double > lambda_sq;

    Estimate<double> fit_delay;
    Estimate<double> fit_delta_DM;

    double reference_frequency = 0;

    unsigned num_subbands = 1;
 
  public:

    void integrate (double freq_MHz, Estimate<double> delay_seconds);

    void fit ();

    void reset ();

    Estimate<double> get_delay () const { return fit_delay; }

    Estimate<double> get_delta_DM () const { return fit_delta_DM; }


    double get_reference_frequency () const { return reference_frequency; }

    void set_num_subbands(unsigned n) { num_subbands = n; }


    std::vector<std::pair<Estimate<double>,double>> get_subband_delays();
 
  private:

    LinearRegression linear_fit;
 
  };
 
}
 
#endif