VariableTransformation.h

//-*-C++-*-
/***************************************************************************
 *
 *   Copyright (C) 2022 by Willem van Straten
 *   Licensed under the Academic Free License version 2.1
 *
 ***************************************************************************/
 
// psrchive/More/Polarimetry/Pulsar/VariableTransformation.h
 
#ifndef __CalibrationVariableTransformation_H
#define __CalibrationVariableTransformation_H
 
#include "MEAL/Complex2.h"
#include "MEAL/Nvariate.h"
#include "MEAL/ProductRule.h"
#include "MEAL/ChainRule.h"
#include "MEAL/Value.h"
 
#include "Estimate.h"
#include "Stokes.h"
 
#include <map>
 
namespace Calibration {


  class VariableTransformation : public MEAL::ProductRule<MEAL::Complex2>
  {
    MEAL::ChainRule<MEAL::Complex2> chain;

    MEAL::Value<MEAL::Complex2> pre_correction;

    Reference::To<MEAL::Complex2> model;

    MEAL::Value<MEAL::Complex2> post_correction;

    std::map< unsigned, Reference::To< MEAL::Nvariate<MEAL::Scalar> > > function;
 
  public:

    VariableTransformation ();

    VariableTransformation (const VariableTransformation& s);

    const VariableTransformation& operator = (const VariableTransformation& s);

    ~VariableTransformation ();

    VariableTransformation* clone () const;

    void set_model (MEAL::Complex2*);

    const MEAL::Complex2* get_model () const;
    MEAL::Complex2* get_model ();

    void set_constraint (unsigned index, MEAL::Nvariate<MEAL::Scalar>*);

    unsigned get_nconstraint () const { return function.size(); }

    unsigned get_index (unsigned j);
    
    bool has_constraint (unsigned index) { return function.find(index) != function.end(); }

    MEAL::Nvariate<MEAL::Scalar>* get_constraint (unsigned index) { return function.at(index); }
 
    class Argument
    {
    public:

      Jones<double> pre_correction;

      Jones<double> post_correction;

      std::map< unsigned, std::vector<double> > arguments;
    };
 
    void set_argument (const Argument&);
 
  private:

    void init ();
 
  };

  std::string axis_value_to_string(const VariableTransformation::Argument&);
 
}
 
#endif