VariableBackendEstimate.h

//-*-C++-*-
 
/***************************************************************************
 *
 *   Copyright (C) 2021 by Willem van Straten
 *   Licensed under the Academic Free License version 2.1
 *
 ***************************************************************************/
 
// psrchive/More/Polarimetry/Pulsar/VariableBackendEstimate.h
 
#ifndef __Pulsar_VariableBackendEstimate_H
#define __Pulsar_VariableBackendEstimate_H
 
#include "Pulsar/BackendEstimate.h"
#include "Pulsar/VariableBackend.h"
#include "Pulsar/VariableGain.h"
#include "Pulsar/ConvertMJD.h"
 
#include "MEAL/Scalar.h"
#include "MEAL/Univariate.h"
 
#include <map>
 
namespace Calibration
{
  class IndexedProduct : public MEAL::ProductRule<MEAL::Complex2>
  {
    int index;
    
  public:
    IndexedProduct () { index = -1; }
 
    bool has_index () const { return index >= 0; }
    void set_index (unsigned _index) { index = _index; }
    unsigned get_index () const { return index; }
  };
  
  class VariableBackendEstimate : public BackendEstimate
  {
    Reference::To< IndexedProduct > psr_response;
    
    Reference::To< IndexedProduct > cal_response;

    Reference::To<VariableBackend> variable_backend;

    bool cal_backend_only = false;
    

    Reference::To<VariableGain> cal_gain;
    
    Reference::To< MEAL::Scalar > gain_variation;
    Reference::To< MEAL::Scalar > diff_gain_variation;
    Reference::To< MEAL::Scalar > diff_phase_variation;

    std::vector< unsigned > gain_imap;
    std::vector< unsigned > diff_gain_imap;
    std::vector< unsigned > diff_phase_imap;
    std::vector< unsigned > backend_imap;
    
  public:

    VariableBackendEstimate (MEAL::Complex2* response = 0);

    void set_response (MEAL::Complex2* xform);

    void add_model (MEAL::Complex2* xform);

    IndexedProduct* get_psr_response () { return psr_response; }

    IndexedProduct* get_cal_response () { return cal_response; }

    SingleAxis* get_backend () { return variable_backend->get_backend(); }
    
    void set_psr_constant_gain (bool = true);

    void set_cal_backend_only (bool = true);
    
    void set_gain_variation (MEAL::Univariate<MEAL::Scalar>*);
    void set_diff_gain_variation (MEAL::Univariate<MEAL::Scalar>*);
    void set_diff_phase_variation (MEAL::Univariate<MEAL::Scalar>*);

    Calibration::ConvertMJD convert;
    
    void update ();

    void update (MEAL::Scalar* function, double value);


    bool reduce_nfree ();

    void update_reference_epoch ();

    void engage_time_variations ();
    void disengage_time_variations ();
 
    void unmap_variations (std::vector<unsigned>& imap,
                           MEAL::Complex2* composite);
    
    void compute_covariance (std::vector<unsigned>& imap,
                             std::vector< std::vector<double> >& Ctotal);
    
 
  };
}
 
#endif