ComplexRVM.h

//-*-C++-*-
/***************************************************************************
 *
 *   Copyright (C) 2009 - 2021 by Willem van Straten
 *   Licensed under the Academic Free License version 2.1
 *
 ***************************************************************************/
 
// psrchive/More/MEAL/MEAL/ComplexRVM.h
 
#ifndef __ComplexRVM_H
#define __ComplexRVM_H
 
#include "MEAL/Complex.h"
#include "MEAL/ProductRule.h"
 
namespace MEAL {
 
  template<typename> class VectorRule;
  
  class RVM;
  class Scalar;

  class ComplexRVM : public ProductRule<Complex>
  {
 
  public:

    ComplexRVM ();

    ComplexRVM (const ComplexRVM& copy);

    ComplexRVM& operator = (const ComplexRVM& copy);

    ~ComplexRVM ();

    RVM* get_rvm ();
    void set_rvm (RVM*);
    
    void add_state (double phase, std::complex< Estimate<double> >& L);
    void set_state (unsigned i);

    unsigned get_nstate () const;

    void set_phase (unsigned i, double phase);
    double get_phase (unsigned i) const;

    void set_linear (unsigned i, const Estimate<double>& L);
    Estimate<double> get_linear (unsigned i) const;
 
#if 0
    void renormalize (double renorm);
#endif

    void set_gains_infit (bool flag=true);

    void set_gains_maximum_likelihood (bool flag=true);
    bool get_gains_maximum_likelihood () const { return gains_maximum_likelihood; }
 
    // ///////////////////////////////////////////////////////////////////
    //
    // Function implementation
    //
    // ///////////////////////////////////////////////////////////////////

    std::string get_name () const;
 
  private:
 
    Reference::To<RVM> rvm;
 
    Reference::To<Scalar> modelQ;
    Reference::To<Scalar> modelU;

    Reference::To< VectorRule<Complex> > gain;

    bool gains_maximum_likelihood;
 
    class State;
    std::vector<State> state;
 
    class MaximumLikelihoodGain;
 
    void check (unsigned i, const char* method) const;
    void init ();
  };
 
}
 
#endif