exp.hh

/* * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * */
/*                                                                             */
/*   This file is part of the C++-library RFFGen.                              */
/*   Copyright 2015 Lars Lubkoll                                               */
/*                                                                             */
/*   RFFGen is free software: you can redistribute it and/or modify            */
/*   it under the terms of the GNU General Public License as published by      */
/*   the Free Software Foundation, either version 3 of the License, or         */
/*   (at your option) any later version.                                       */
/*                                                                             */
/*   RFFGen 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 General Public License for more details.                              */
/*                                                                             */
/*   You should have received a copy of the GNU General Public License         */
/*   along with RFFGen.  If not, see <http://www.gnu.org/licenses/>.           */
/*                                                                             */
/* * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * */

#ifndef RFFGEN_CMATH_EXP_HH
#define RFFGEN_CMATH_EXP_HH

#include <cmath>
#include "../Util/base.hh"
#include "../Util/chainer.hh"

namespace RFFGen
{
  namespace CMath
  {
    struct Exp : Base , Chainer<Exp>
    {
      using Chainer<Exp>::operator ();
      explicit Exp(double x=0.) { update(x); }

      void update(double x)
      {
        e_x = ::exp(x);
      }

      double d0() const noexcept
      {
        return e_x;
      }

      template < int = -1 >
      double d1(double dx = 1.) const
      {
        return e_x * dx;
      }

      template < int = -1 , int = -1 >
      double d2(double dx = 1., double dy = 1.) const
      {
        return e_x * dx * dy;
      }

      template < int = -1 , int = -1 , int = -1 >
      double d3(double dx = 1., double dy = 1., double dz = 1.) const
      {
        return e_x * dx * dy * dz;
      }

    private:
      double e_x = 1.;
    };

    struct Exp2 : Base , Chainer<Exp2>
    {
      using Chainer<Exp2>::operator ();
      explicit Exp2(double x=0.) { update(x); }

      void update(double x)
      {
        value = exp2(x);
      }

      double d0() const noexcept
      {
        return value;
      }

      template < int = -1 >
      double d1(double dx = 1.) const
      {
        return value * ln2 * dx;
      }

      template < int = -1 , int = -1 >
      double d2(double dx = 1., double dy = 1.) const
      {
        return value * ln2 * ln2 * dx * dy;
      }

      template < int = -1 , int = -1 , int = -1 >
      double d3(double dx = 1., double dy = 1., double dz = 1.) const
      {
        return value * ln2 * ln2 * ln2 * dx * dy * dz;
      }

    private:
      double value = 1., ln2 = log(2.);
    };

    template <class Function, class = std::enable_if_t<std::is_base_of<Base,Function>::value> >
    auto exp(const Function& f)
    {
      return Exp()(f);
    }

    template <class Function, class = std::enable_if_t<std::is_base_of<Base,Function>::value> >
    auto ex2p(const Function& f)
    {
      return Exp2()(f);
    }
  }
}

#endif // RFFGEN_CMATH_EXP_HH