Multivariate.h

//-*-C++-*-
/***************************************************************************
 *
 *   Copyright (C) 2022 by Willem van Straten
 *   Licensed under the Academic Free License version 2.1
 *
 ***************************************************************************/
 
// psrchive/More/MEAL/MEAL/Multivariate.h
 
#ifndef __MEAL_Multivariate_H
#define __MEAL_Multivariate_H
 
#include "MEAL/MultivariatePolicy.h"
#include "MEAL/Nvariate.h"
#include "Reference.h"
 
namespace MEAL {


  template<class T>
  class Multivariate : public Nvariate<T>
  {
    unsigned ndim;
 
  public:

    Multivariate (unsigned _ndim)
    { ndim = _ndim;
      set_multivariate_policy ( new MultivariatePolicy (ndim, this) ); }

    Multivariate (const Multivariate& copy) : T (copy) { ndim = copy.ndim; }

    Multivariate& operator = (const Multivariate& copy);

    Multivariate<T>* clone () const
    { throw Error (InvalidState, "Multivariate<T>::clone", "not implemented"); }


    unsigned get_ndim () const { return ndim; }

    void set_abscissa_value (unsigned idim, double value)
    { multivariate_policy->set_abscissa (idim, value); }

    double get_abscissa_value (unsigned idim) const
    { return multivariate_policy->get_abscissa (idim); }

 
  protected:
 
    template <class U>
    friend class FunctionPolicyTraits;
    
    void set_multivariate_policy (MultivariatePolicy* policy)
    { this->argument_policy = multivariate_policy = policy; }
 
    MultivariatePolicy* get_multivariate_policy () const
    { return multivariate_policy; }

    void copy_multivariate_policy (const Multivariate* other)
    { set_multivariate_policy (other->get_multivariate_policy()); }
 
  private:

    Reference::To<MultivariatePolicy> multivariate_policy;
 
  };
 
  template<class T>
  class FunctionPolicyTraits< Multivariate<T> >
  {
  public:
    static void composite_component (Multivariate<T>* composite,
                                     Multivariate<T>* component)
    {
      component->copy_multivariate_policy (composite);
    }
  };
}
 
template<class T>
MEAL::Multivariate<T>& MEAL::Multivariate<T>::operator = (const Multivariate& copy)
{ 
  if (&copy == this)
    return *this;
 
  for (unsigned idim=0; idim < ndim; idim++)
    set_abscissa_value( idim, copy.get_abscissa_value(idim) );
 
  T::operator=( copy );
 
  return *this;
}
 
#endif