/*
 * SaturationPotential.hpp
 *
 *  Created on: Oct 11, 2012
 *      Author: heber
 */

#ifndef SATURATIONPOTENTIAL_HPP_
#define SATURATIONPOTENTIAL_HPP_


// include config.h
#ifdef HAVE_CONFIG_H
#include <config.h>
#endif

#include <boost/function.hpp>
#include <limits>

#include "Potentials/EmpiricalPotential.hpp"
#include "Potentials/Specifics/PairPotential_Angle.hpp"
#include "Potentials/Specifics/PairPotential_Morse.hpp"
#include "FunctionApproximation/FunctionModel.hpp"

/** This is a combination of Morse and Angle potentials for saturated elements.
 *
 */
class SaturationPotential : virtual public EmpiricalPotential, virtual public FunctionModel
{
  //!> grant unit test access to internal parts
  friend class SaturationPotentialTest;
  // some repeated typedefs to avoid ambiguities
  typedef FunctionModel::arguments_t arguments_t;
  typedef FunctionModel::result_t result_t;
  typedef FunctionModel::results_t results_t;
  typedef EmpiricalPotential::derivative_components_t derivative_components_t;
  typedef FunctionModel::parameters_t parameters_t;
public:
  SaturationPotential(
      const double _saturation_cutoff,
      boost::function< std::vector<arguments_t>(const argument_t &, const double)> &_triplefunction
      );
  SaturationPotential(
      const double _morse_spring_constant,
      const double _morse_equilibrium_distance,
      const double _morse_dissociation_energy,
      const double _angle_spring_constant,
      const double _angle_equilibrium_distance,
      const double _energy_offset,
      const double _saturation_cutoff,
      boost::function< std::vector<arguments_t>(const argument_t &, const double)> &_triplefunction
      );
  virtual ~SaturationPotential() {}

  /** Setter for parameters as required by FunctionModel interface.
   *
   * \param _params given set of parameters
   */
  void setParameters(const parameters_t &_params);

  /** Getter for parameters as required by FunctionModel interface.
   *
   * \return set of parameters
   */
  parameters_t getParameters() const;

  /** Getter for the number of parameters of this model function.
   *
   * \return number of parameters
   */
  size_t getParameterDimension() const
  { return MAXPARAMS; }

  /** Evaluates the harmonic potential function for the given arguments.
   *
   * @param arguments single distance
   * @return value of the potential function
   */
  results_t operator()(const arguments_t &arguments) const;

  /** Evaluates the derivative of the potential function.
   *
   * @param arguments single distance
   * @return vector with derivative with respect to the input degrees of freedom
   */
  derivative_components_t derivative(const arguments_t &arguments) const;

  /** Evaluates the derivative of the function with the given \a arguments
   * with respect to a specific parameter indicated by \a index.
   *
   * \param arguments set of arguments as input variables to the function
   * \param index derivative of which parameter
   * \return result vector containing the derivative with respect to the given
   *         input
   */
  results_t parameter_derivative(const arguments_t &arguments, const size_t index) const;

  /** States whether lower and upper boundaries should be used to constraint
   * the parameter search for this function model.
   *
   * \return true - constraints should be used, false - else
   */
  bool isBoxConstraint() const {
    return true;
  }

  /** Returns a vector which are the lower boundaries for each parameter_t
   * of this FunctionModel.
   *
   * \return vector of parameter_t resembling lowest allowed values
   */
  parameters_t getLowerBoxConstraints() const {
    parameters_t lowerbounds(getParameterDimension(), -std::numeric_limits<double>::max());
    lowerbounds[morse_equilibrium_distance] = 0.;
    lowerbounds[angle_equilibrium_distance] = -1.;
    return lowerbounds;
  }

  /** Returns a vector which are the upper boundaries for each parameter_t
   * of this FunctionModel.
   *
   * \return vector of parameter_t resembling highest allowed values
   */
  parameters_t getUpperBoxConstraints() const {
    parameters_t upperbounds(getParameterDimension(), std::numeric_limits<double>::max());
    upperbounds[angle_equilibrium_distance] = 1.;
    return upperbounds;
  }

  enum parameter_enum_t {
    all_energy_offset,
    morse_spring_constant,
    morse_equilibrium_distance,
    morse_dissociation_energy,
    angle_spring_constant,
    angle_equilibrium_distance,
    MAXPARAMS
  };

private:
  PairPotential_Morse morse;
  PairPotential_Angle angle;
  double energy_offset;

  //!> bound function that obtains the triples for the internal coordinationb summation.
  const boost::function< std::vector< arguments_t >(const argument_t &, const double)> &triplefunction;
  const double saturation_cutoff;
};

#endif /* SATURATIONPOTENTIAL_HPP_ */