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;
    
 
    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