//# ChebyshevParam.h: Parameter handling for Chebyshev polynomial
//# Copyright (C) 2000,2001,2002,2003,2005
//# Associated Universities, Inc. Washington DC, USA.
//#
//# This library is free software; you can redistribute it and/or modify it
//# under the terms of the GNU Library General Public License as published by
//# the Free Software Foundation; either version 2 of the License, or (at your
//# option) any later version.
//#
//# This library is distributed in the hope that it will be useful, but WITHOUT
//# ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
//# FITNESS FOR A PARTICULAR PURPOSE.  See the GNU Library General Public
//# License for more details.
//#
//# You should have received a copy of the GNU Library General Public License
//# along with this library; if not, write to the Free Software Foundation,
//# Inc., 675 Massachusetts Ave, Cambridge, MA 02139, USA.
//#
//# Correspondence concerning AIPS++ should be addressed as follows:
//#        Internet email: casa-feedback@nrao.edu.
//#        Postal address: AIPS++ Project Office
//#                        National Radio Astronomy Observatory
//#                        520 Edgemont Road
//#                        Charlottesville, VA 22903-2475 USA
//#
//#! ========================================================================

#ifndef SCIMATH_CHEBYSHEVPARAM_H
#define SCIMATH_CHEBYSHEVPARAM_H

#include <casacore/casa/aips.h>
#include <casacore/casa/Arrays/ArrayFwd.h>
#include <casacore/casa/BasicSL/String.h>
#include <casacore/scimath/Functionals/Function1D.h>

namespace casacore { //# NAMESPACE CASACORE - BEGIN

//# Forward Declarations
class RecordInterface;

// <summary>
// Define enums for Chebyshev classes
// </summary>
class ChebyshevEnums
{
public:
    // Modes that identify how this function behaves outside its Chebyshev
    // interval (see setInterval()).
    enum OutOfIntervalMode {

	// return a constant, default value.  The value returned is 
	// set with setDefault().
	CONSTANT,

	// return a constant value equal to the zero-th order coefficient
	ZEROTH,

	// evaluate the polynomial based on its coefficients just as it
	// would be inside the interval.  Thus, the function's range is not
	// guaranteed to remain within the characteristic bounds of the
	// Chebyshev interval.  
	EXTRAPOLATE,

	// evaluate the function as if the range is cyclic, repeating the
	// range values from its canonical domain.  The period of the cycle
	// will be equal to getIntervalMax()-getIntervalMin().  When the 
	// function is evaluated outside this interval, the input value will 
	// shifted an integer number of periods until it falls within the 
	// Chebyshev interval; the value returned is the polynomial evaluated 
	// at the shifted (x-axis) value.  Obviously, this mode is most 
	// expensive computationally when evaluating outside the range.
	CYCLIC,

	// evaluate the function at nearest interval edge
	EDGE,

	// number of enumerators
	NOutOfIntervalModes };
};


// <summary> Parameter handling for Chebyshev polynomial parameters
// </summary>

// <use visibility=local>

// <reviewed reviewer="wbrouw" date="2001/11/12" tests="tChebyshev" demos="">
// </reviewed>

// <prerequisite>
//   <li> <linkto class="FunctionParam">FunctionParam</linkto> class
//   <li> <linkto class="Function1D">Function1D</linkto>
//   <li> <linkto class="Chebyshev">Chebyshev</linkto>
// </prerequisite>
//
// <etymology>
// This class is named after Chebyshev Type I polynomials; it handles the 
// "fixed" parameters for the function.
// </etymology>
//
// <synopsis>
// This class assists in forming and evaluating a function as a
// Chebyshev series, a linear combination of so-called Chebyshev
// polynomials.  Users do not instantiate this abstract class directly; 
// instead they instantiate the child class 
// <linkto class="Chebyshev">Chebyshev</linkto>.  This class holds the part 
// of the implementation used by the 
// <linkto class="Chebyshev">Chebyshev</linkto> class that manages the "fixed" 
// parameters of the function (e.g. the polynomial coefficients, interval of
// interest, etc.)
// 
// For a full description, see the 
// <linkto class="Chebyshev">Chebyshev</linkto> class.
// 
// </synopsis>
//
// <example>
// In this example, a 2nd order Chebyshev polynomial series is
// created.
// <srcblock>
//   // set coeffs to desired values
//   Vector<Double> coeffs(3, 1);   
//
//   // configure the function   
//   Chebyshev<Double> cheb;
//   cheb.setInterval(-0.8, 7.2);
//   cheb.setDefault(1.0);
//   cheb.setCoefficients(coeffs);
//
//   // evaluate the function as necessary
//   Double z = cheb(-0.5);    // -0.5 is within range, z = 0.78625
//   z = cheb(4.2);            // 4.2 is within range, z = 0.375
//   z = cheb(-3);             // -3 is out of the interval, z = 1
// </srcblock>
// </example>
//
// <motivation>
// This class was created to support systematic errors in the simulator tool.  
// It can be used by Jones matrix classes to vary gains in a predictable way,
// mimicing natural processes of the atmosphere or instrumental effects.  
//
// The Chebyshev implementation is split between this class, 
// <src>ChebyshevParam</src> and its child 
// <linkto class="Chebyshev">Chebyshev</linkto> to better support the 
// <linkto class="AutoDiff">AutoDiff framework</linkto> for evaluating 
// derivatives.
// </motivation>
//
// <templating arg=T>
//  <li> T should have standard numerical operators. Current
//	implementation only tested for real types (and their AutoDiffs).
// </templating>
//
// <thrown>
//    <li> Assertion if indices out-of-range
// </thrown>
//
// <todo asof="2001/08/22">
//    <li> It would be helpful to be able to convert to and from the 
//         Polynomial<T> type; this would be supported via a function,
//         Polynomial<T> polynomial(), and constructor, 
//         Chebyshev(Polynomial<T>)
// </todo>

template<class T>
class  ChebyshevParam : public Function1D<T>
{
public: 

    //# Constructors
    // create a zero-th order Chebyshev polynomial with the first coefficient
    // equal to zero.  The bounded domain is [T(-1), T(1)].  The 
    // OutOfDomainMode is CONSTANT, and the default value is T(0).
    ChebyshevParam();

    // create an n-th order Chebyshev polynomial with the coefficients
    // equal to zero.  The bounded domain is [T(-1), T(1)].  The 
    // OutOfDomainMode is CONSTANT, and the default value is T(0).
    explicit ChebyshevParam(const uInt n);

    // create a zero-th order Chebyshev polynomical with the first coefficient
    // equal to one.  
    //   min is the minimum value of its Chebyshev interval, and 
    //   max is the maximum value.  
    //   mode sets the behavior of the function outside the Chebyshev interval
    //      (see setOutOfIntervalMode() and OutOfIntervalMode enumeration 
    //      definition for details).  
    //   defval is the value returned when the function is evaluated outside
    //      the Chebyshev interval and mode=CONSTANT.
    ChebyshevParam(const T &min, const T &max,
		   ChebyshevEnums::OutOfIntervalMode
		   mode=ChebyshevEnums::CONSTANT, const T &defval=T(0));
  
    // create a fully specified Chebyshev polynomial.  
    //   coeffs holds the coefficients of the Chebyshev polynomial (see 
    //      setCoefficients() for details).
    //   min is the minimum value of its canonical range, and 
    //   max is the maximum value.  
    //   mode sets the behavior of the function outside the Chebyshev interval
    //      (see setOutOfIntervalMode() and OutOfIntervalMode enumeration 
    //      definition for details).  
    //   defval is the value returned when the function is evaluated outside
    //      the canonical range and mode=CONSTANT.
    ChebyshevParam(const Vector<T> &coeffs, const T &min, const T &max, 
		   ChebyshevEnums::OutOfIntervalMode
		   mode=ChebyshevEnums::CONSTANT, const T &defval=T(0));
  
    // create a fully specified Chebyshev polynomial.
    //   config  is a record that contains the non-coefficient data 
    //             that configures this class.
    // The fields recognized by this class are those documented for the 
    // setMode() function below.
    // <group>
    ChebyshevParam(uInt order, const RecordInterface& mode);
    ChebyshevParam(const Vector<T> &coeffs, const RecordInterface& mode);
    // </group>
  
    // create a deep copy of another Chebyshev polynomial
    // <group>
    ChebyshevParam(const  ChebyshevParam &other);
    template <class W>
      ChebyshevParam(const ChebyshevParam<W> &other) :
      Function1D<T>(other), def_p(other.getDefault()), 
      minx_p(other.getIntervalMin()), maxx_p(other.getIntervalMax()),
      mode_p(other.getOutOfIntervalMode()) {}
    // </group>

    // make a (deep) copy of another Chebyshev polynomial
    ChebyshevParam<T> &operator=(const ChebyshevParam<T> &other);
  
    // Destructor
    virtual ~ChebyshevParam();
  
    // set the Chebyshev coefficients.  
    //   coeffs holds the coefficients in order, beginning with the zero-th 
    //      order term.  The order of the polynomial, then, would be the size 
    //      of the Vector minus one.  
    void setCoefficients(const Vector<T> &coeffs);

    // set a particular Chebyshev coefficient.
    //   which is the coefficient order (i.e. 0 refers to the constant offset).
    //   value is the coefficient value.
    // If which is larger than current order of the function, the order will
    // be increased to the value of which, and that coefficient is set to 
    // value; missing coefficients less than this value will be set to zero.
    // Thus, the order can be increased with this function; however, it cannot
    // be decreased (even if the highest order coefficient is set to zero).
    // To lower the order, use setCoefficients() with a Vector having the 
    // desired number of coefficients.
    void setCoefficient(const uInt which, const T &value);
  
    // return the current set of coefficients into a given Vector.  
    const Vector<T> &getCoefficients() const;
  
    // return a particular coefficient.  
    //   which is the coefficient order (i.e. 0 refers to the constant offset).
    //     If which is out of range, zero is returned.
    T getCoefficient(const uInt which) const {
	return ((which < nparameters()) ? param_p[which] : T(0)); }
  
    // return the number of coeefficients currently loaded.  This does not
    // guarantee that the coefficients are non-zero
    uInt nCoefficients() const { return nparameters(); }
    
    // set the Chebyshev interval for this function.  The function will 
    // be scaled and shifted to such that the central bounded range of the 
    // Chebyshev polynomials ([-1, 1] in untransformed space) spans the 
    // given range.  
    //   min is the minimum value for the interval, and 
    //   max is the maximum value.  See setOutOfIntervalMode() for the behavior 
    //      of this function outside the set range.
    void setInterval(T xmin, T xmax) {
	if (xmin < xmax) { minx_p = xmin; maxx_p = xmax;
	} else { minx_p = xmax; maxx_p = xmin; } }   

    // return the minimum value for the currently Chebyshev interval.
    // See setInterval() for additional details.
    T getIntervalMin() const { return minx_p; }

    // return the maximum value for the currently Chebyshev interval.
    // See setInterval() for additional details.
    T getIntervalMax() const { return maxx_p; }

    // set the behavior of this function when it is evaluated outside its
    // Chebyshev interval
    void setOutOfIntervalMode(ChebyshevEnums::OutOfIntervalMode mode)
      { mode_p = mode; }

    // return the behavior of this function when it is evaluated outside of 
    // its Chebyshev interval.
    ChebyshevEnums::OutOfIntervalMode getOutOfIntervalMode() const
      { return mode_p; }

    // set the default value of this function.  This value is used when 
    // the getOutOfIntervalMode() returns Chebyshev::CONSTANT; it is returned
    // when the a value outside of the Chebyshev interval is passed to 
    // the () operator.
    void setDefault(const T &val) { def_p = val; }

    // return the currently set default value.  See setDefault() for details 
    // on the use of this value.
    const T &getDefault() const { return def_p; }

    // return the order of this polynomial.  This returns the value of 
    // nCoefficients()-1;
    uInt order() const { return param_p.nelements() - 1; }

    // transform a set of Chebyshev polynomial coefficients into a set 
    // representing the series' derivative.  coeffs should be assuming
    // an interval of [-1, 1].  xmin and xmax can be provided to transform
    // the series to another interval.
    static void derivativeCoeffs(Vector<T> &coeffs, const T &xmin=T(-1),
				 const T &xmax=T(1));
  
    // convert a set of Chebyshev polynomial coefficients to power series
    // coefficients.  The values passed in coeffs are taken to 
    // be chebyshev coefficients; these values will be replaced with the 
    // power series coefficients.  They should be ordered beginning
    // with the zero-th order coefficient.  
    static void chebyshevToPower(Vector<T> &coeffs);

    // convert a set of power series coefficients to Chebyshev
    // polynomial coefficients.  The values passed in coeffs are taken to 
    // be power series coefficients; these values will be replaced with the 
    // Chebyshev polynomial coefficients.  They should be ordered beginning
    // with the zero-th order coefficient.  
    static void powerToChebyshev(Vector<T> &coeffs);

    // Give name of function
    virtual const String &name() const { static String x("chebyshev");
    return x; }

protected:

    // Default value if outside interval
    T def_p;
    // Lowest interval bound
    T minx_p;
    // Highest inetrval bound
    T maxx_p;
    // Out-of-interval handling type
    ChebyshevEnums::OutOfIntervalMode mode_p;

    static Vector<String> modes_s;

  //# Make members of parent classes known.
protected:
  using Function1D<T>::param_p;
public:
  using Function1D<T>::nparameters;
  using Function1D<T>::setMode;
};


// <summary> A ChebyshevParam with the get/setMode implementation </summary>
//
// <synopsis>
// The get/setMode() implementation is separated from ChebyshevParam
// to enable simple specialization for AutoDiff.  See
// <linkto class="ChebyshevParam">ChebyshevParam</linkto> for documentation
// </synopsis>
template <class T>
class ChebyshevParamModeImpl : public ChebyshevParam<T>
{
public:
    ChebyshevParamModeImpl() : ChebyshevParam<T>() { }

    explicit ChebyshevParamModeImpl(const uInt n) : ChebyshevParam<T>(n) {}

    ChebyshevParamModeImpl(const T &min, const T &max,
			   typename ChebyshevEnums::OutOfIntervalMode mode=ChebyshevEnums::CONSTANT,
			   const T &defval=T(0))
	: ChebyshevParam<T>(min, max, mode, defval) {}
  
    ChebyshevParamModeImpl(const Vector<T> &coeffs, 
			   const T &min, const T &max, 
			   typename ChebyshevEnums::OutOfIntervalMode mode=ChebyshevEnums::CONSTANT, 
			   const T &defval=T(0))
	: ChebyshevParam<T>(coeffs, min, max, mode, defval) {}
  
    ChebyshevParamModeImpl(uInt order, const RecordInterface& mode)
	: ChebyshevParam<T>(order, mode) { setMode(mode); }
    ChebyshevParamModeImpl(const Vector<T> &coeffs, 
			   const RecordInterface& mode)
	: ChebyshevParam<T>(coeffs, mode) { setMode(mode); }
  
    ChebyshevParamModeImpl(const ChebyshevParamModeImpl &other) 
	: ChebyshevParam<T>(other) {}
  
    // get/set the function mode.  This is an alternate way to get/set the 
    // non-coefficient data for this function.  The supported record fields 
    // are as follows:
    // <pre>
    // Field Name     Type            Role
    // -------------------------------------------------------------------
    // min            template type   the minimum value of the Chebyshev 
    //              			interval of interest
    // max            template type   the maximum value of the Chebyshev 
    //              			interval of interest
    // intervalMode   TpString        the out-of-interval mode; recognized
    //                                  values are "constant", "zeroth",
    //                                  "extrapolate", "cyclic", and "edge".
    //                                  setMode() recognizes a 
    //                                  case-insensitive, minimum match.
    // default        template type   the out-of-range value that is returned
    //                                  when the out-of-interval mode is 
    //                                  "constant".
    // </pre>
    // An exception is thrown if interval mode is unrecognized.
    // <group>
    virtual void setMode(const RecordInterface& mode);
    virtual void getMode(RecordInterface& mode) const;
    // </group>

    // return True if the implementing function supports a mode.  This
    // implementation always returns True.
    virtual Bool hasMode() const;

  //# Make members of parent classes known.
protected:
  using ChebyshevParam<T>::modes_s;
public:
  using ChebyshevParam<T>::setOutOfIntervalMode;
  using ChebyshevParam<T>::getOutOfIntervalMode;
  using ChebyshevParam<T>::getIntervalMin;
  using ChebyshevParam<T>::getIntervalMax;
  using ChebyshevParam<T>::getDefault;
};

#define ChebyshevParamModeImpl_PS ChebyshevParamModeImpl

// <summary> Partial specialization of ChebyshevParamModeImpl for 
// <src>AutoDiff</src>
// </summary>
// <synopsis>
// <note role=warning> The name <src>ChebyshevParamModeImpl_PS</src> is only 
// for cxx2html limitations.  
// </note>
// </synopsis>
template <class T>
class ChebyshevParamModeImpl_PS<AutoDiff<T> > 
    : public ChebyshevParam<AutoDiff<T> > 
{
public:
    ChebyshevParamModeImpl_PS() : ChebyshevParam<AutoDiff<T> >() { }

    explicit ChebyshevParamModeImpl_PS(const uInt n) 
	: ChebyshevParam<AutoDiff<T> >(n) {}

    ChebyshevParamModeImpl_PS(const AutoDiff<T> &min, const AutoDiff<T> &max,
			      typename ChebyshevEnums::OutOfIntervalMode mode=ChebyshevEnums::CONSTANT,
			      const AutoDiff<T> &defval=AutoDiff<T>(0))
	: ChebyshevParam<AutoDiff<T> >(min, max, mode, defval) {}
  
    ChebyshevParamModeImpl_PS(const Vector<AutoDiff<T> > &coeffs, 
			      const AutoDiff<T> &min, const AutoDiff<T> &max, 
			      typename ChebyshevEnums::OutOfIntervalMode mode=ChebyshevEnums::CONSTANT, 
			      const AutoDiff<T> &defval=AutoDiff<T>(0))
	: ChebyshevParam<AutoDiff<T> >(coeffs, min, max, mode, defval) {}
  
    ChebyshevParamModeImpl_PS(uInt order, const RecordInterface& mode)
	: ChebyshevParam<AutoDiff<T> >(order, mode) {}
    ChebyshevParamModeImpl_PS(const Vector<AutoDiff<T> > &coeffs, 
			      const RecordInterface& mode)
	: ChebyshevParam<AutoDiff<T> >(coeffs, mode) {}
  
    ChebyshevParamModeImpl_PS(const ChebyshevParamModeImpl_PS &other) 
	: ChebyshevParam<AutoDiff<T> >(other) {}
  
    virtual void setMode(const RecordInterface& mode);
    virtual void getMode(RecordInterface& mode) const;

  //# Make members of parent classes known.
protected:
  using ChebyshevParam<AutoDiff<T> >::modes_s;
public:
  using ChebyshevParam<AutoDiff<T> >::setOutOfIntervalMode;
  using ChebyshevParam<AutoDiff<T> >::getOutOfIntervalMode;
  using ChebyshevParam<AutoDiff<T> >::getIntervalMin;
  using ChebyshevParam<AutoDiff<T> >::getIntervalMax;
  using ChebyshevParam<AutoDiff<T> >::getDefault;
};


#define ChebyshevParamModeImpl_PSA ChebyshevParamModeImpl

// <summary> Partial specialization of ChebyshevParamModeImpl for 
// <src>AutoDiff</src>
// </summary>
// <synopsis>
// <note role=warning> The name <src>ChebyshevParamModeImpl_PS</src> is only 
// for cxx2html limitations.  
// </note>
// </synopsis>
template <class T>
class ChebyshevParamModeImpl_PSA<AutoDiffA<T> > 
    : public ChebyshevParam<AutoDiffA<T> > 
{
public:
    ChebyshevParamModeImpl_PSA() : ChebyshevParam<AutoDiffA<T> >() { }

    explicit ChebyshevParamModeImpl_PSA(const uInt n) 
	: ChebyshevParam<AutoDiffA<T> >(n) {}

    ChebyshevParamModeImpl_PSA(const AutoDiffA<T> &min, 
			       const AutoDiffA<T> &max,
			       typename ChebyshevEnums::OutOfIntervalMode mode=ChebyshevEnums::CONSTANT,
			       const AutoDiffA<T> &defval=AutoDiffA<T>(0))
	: ChebyshevParam<AutoDiffA<T> >(min, max, mode, defval) {}
  
    ChebyshevParamModeImpl_PSA(const Vector<AutoDiffA<T> > &coeffs, 
			       const AutoDiffA<T> &min, 
			       const AutoDiffA<T> &max, 
			       typename ChebyshevEnums::OutOfIntervalMode mode=ChebyshevEnums::CONSTANT, 
			       const AutoDiffA<T> &defval=AutoDiffA<T>(0))
	: ChebyshevParam<AutoDiffA<T> >(coeffs, min, max, mode, defval) {}
  
    ChebyshevParamModeImpl_PSA(uInt order, const RecordInterface& mode)
	: ChebyshevParam<AutoDiffA<T> >(order, mode) {}
    ChebyshevParamModeImpl_PSA(const Vector<AutoDiffA<T> > &coeffs, 
			      const RecordInterface& mode)
	: ChebyshevParam<AutoDiffA<T> >(coeffs, mode) {}
  
    ChebyshevParamModeImpl_PSA(const ChebyshevParamModeImpl_PSA &other) 
	: ChebyshevParam<AutoDiffA<T> >(other) {}
  
    virtual void setMode(const RecordInterface& mode);
    virtual void getMode(RecordInterface& mode) const;

  //# Make members of parent classes known.
protected:
  using ChebyshevParam<AutoDiffA<T> >::modes_s;
public:
  using ChebyshevParam<AutoDiffA<T> >::setOutOfIntervalMode;
  using ChebyshevParam<AutoDiffA<T> >::getOutOfIntervalMode;
  using ChebyshevParam<AutoDiffA<T> >::getIntervalMin;
  using ChebyshevParam<AutoDiffA<T> >::getIntervalMax;
  using ChebyshevParam<AutoDiffA<T> >::getDefault;
};




} //# NAMESPACE CASACORE - END

#ifndef CASACORE_NO_AUTO_TEMPLATES
#include <casacore/scimath/Functionals/ChebyshevParam.tcc>
#endif //# CASACORE_NO_AUTO_TEMPLATES
#endif