MathToolbox.hpp

Go to the documentation of this file.

// -*- mode: C++; tab-width: 4; indent-tabs-mode: nil; c-basic-offset: 4 -*-
// vi: set et ts=4 sw=4 sts=4:
/*
  This file is part of the Open Porous Media project (OPM).

  OPM 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 2 of the License, or
  (at your option) any later version.

  OPM 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 OPM.  If not, see <http://www.gnu.org/licenses/>.

  Consult the COPYING file in the top-level source directory of this
  module for the precise wording of the license and the list of
  copyright holders.
*/
#ifndef OPM_MATERIAL_MATH_TOOLBOX_HPP
#define OPM_MATERIAL_MATH_TOOLBOX_HPP

#include <cmath>
#include <algorithm>
#include <type_traits>
#include <stdexcept>
#include <opm/common/utility/gpuDecorators.hpp>

namespace Opm {
/*
 * \brief A traits class which provides basic mathematical functions for arbitrary scalar
 *        floating point values.
 *
 * The reason why this is done in such a complicated way is to enable other approaches,
 * in particular automatic differentiation based ones.
 */
template <class ScalarT>
struct MathToolbox
{
    static_assert(std::is_floating_point<ScalarT>::value,
                  "This class expects floating point scalars! (specialization missing?)");
public:
    typedef ScalarT Scalar;

    typedef ScalarT ValueType;

    typedef MathToolbox<Scalar> InnerToolbox;

    OPM_HOST_DEVICE static Scalar value(Scalar value)
    { return value; }

    OPM_HOST_DEVICE static Scalar scalarValue(Scalar value)
    { return value; }

    OPM_HOST_DEVICE static Scalar createBlank(Scalar /*value*/)
    { return Scalar(); }

    OPM_HOST_DEVICE static Scalar createConstant(Scalar value)
    { return value; }

    OPM_HOST_DEVICE static Scalar createConstant(unsigned numDerivatives, Scalar value)
    {
        if (numDerivatives != 0)
            throw std::logic_error("Plain floating point objects cannot represent any derivatives");
        return value;
    }

    OPM_HOST_DEVICE static Scalar createConstant(Scalar /*x*/, Scalar value)
    { return value; }

    OPM_HOST_DEVICE static Scalar createVariable(Scalar /*value*/, unsigned /*varIdx*/)
    { throw std::logic_error("Plain floating point objects cannot represent variables"); }

    OPM_HOST_DEVICE static Scalar createVariable(Scalar /*x*/, Scalar /*value*/, unsigned /*varIdx*/)
    { throw std::logic_error("Plain floating point objects cannot represent variables"); }

    template <class LhsEval>
    OPM_HOST_DEVICE static LhsEval decay(Scalar value)
    {
        static_assert(std::is_floating_point<LhsEval>::value,
                      "The left-hand side must be a primitive floating point type!");

        return value;
    }

    OPM_HOST_DEVICE static bool isSame(Scalar a, Scalar b, Scalar tolerance)
    {
        Scalar valueDiff = a - b;
        Scalar denom = std::max<Scalar>(1.0, std::abs(a + b));

        return std::abs(valueDiff) < tolerance || std::abs(valueDiff)/denom < tolerance;
    }

    // arithmetic functions

    OPM_HOST_DEVICE static Scalar max(Scalar arg1, Scalar arg2)
    { return std::max(arg1, arg2); }

    OPM_HOST_DEVICE static Scalar min(Scalar arg1, Scalar arg2)
    { return std::min(arg1, arg2); }

    OPM_HOST_DEVICE static Scalar abs(Scalar arg)
    { return std::abs(arg); }

    OPM_HOST_DEVICE static Scalar tan(Scalar arg)
    { return std::tan(arg); }

    OPM_HOST_DEVICE static Scalar atan(Scalar arg)
    { return std::atan(arg); }

    OPM_HOST_DEVICE static Scalar atan2(Scalar arg1, Scalar arg2)
    { return std::atan2(arg1, arg2); }

    OPM_HOST_DEVICE static Scalar sin(Scalar arg)
    { return std::sin(arg); }

    OPM_HOST_DEVICE static Scalar asin(Scalar arg)
    { return std::asin(arg); }

    OPM_HOST_DEVICE static Scalar sinh(Scalar arg)
    { return std::sinh(arg); }

    OPM_HOST_DEVICE static Scalar asinh(Scalar arg)
    { return std::asinh(arg); }

    OPM_HOST_DEVICE static Scalar cos(Scalar arg)
    { return std::cos(arg); }

    OPM_HOST_DEVICE static Scalar acos(Scalar arg)
    { return std::acos(arg); }

    OPM_HOST_DEVICE static Scalar cosh(Scalar arg)
    { return std::cosh(arg); }

    OPM_HOST_DEVICE static Scalar acosh(Scalar arg)
    { return std::acosh(arg); }

    OPM_HOST_DEVICE static Scalar sqrt(Scalar arg)
    { return std::sqrt(arg); }

    OPM_HOST_DEVICE static Scalar exp(Scalar arg)
    { return std::exp(arg); }

    OPM_HOST_DEVICE static Scalar log10(Scalar arg)
    { return std::log10(arg); }

    OPM_HOST_DEVICE static Scalar log(Scalar arg)
    { return std::log(arg); }

    OPM_HOST_DEVICE static Scalar pow(Scalar base, Scalar exp)
    { return std::pow(base, exp); }

    OPM_HOST_DEVICE static bool isfinite(Scalar arg)
    { return std::isfinite(arg); }

    OPM_HOST_DEVICE static bool isnan(Scalar arg)
    { return std::isnan(arg); }
};

template <class Eval1, class Eval2>
struct ReturnEval_
{
    typedef typename std::remove_const< typename std::remove_reference<Eval1>::type >::type T;
    typedef typename std::remove_const< typename std::remove_reference<Eval2>::type >::type U;

    //static_assert(std::is_constructible<T, U>::value || std::is_constructible<U, T>::value,
    //              "One of the argument types must be constructible to the other");

    typedef typename std::conditional<std::is_constructible<T, U>::value,
                                      T,
                                      U>::type type;
};

// these are convenience functions for not having to type MathToolbox<Scalar>::foo()
template <class Evaluation>
OPM_HOST_DEVICE Evaluation blank(const Evaluation& x)
{ return MathToolbox<Evaluation>::createBlank(x); }

template <class Evaluation, class Scalar>
OPM_HOST_DEVICE Evaluation constant(const Scalar& value)
{ return MathToolbox<Evaluation>::createConstant(value); }

template <class Evaluation, class Scalar>
OPM_HOST_DEVICE Evaluation constant(unsigned numDeriv, const Scalar& value)
{ return MathToolbox<Evaluation>::createConstant(numDeriv, value); }

template <class Evaluation, class Scalar>
OPM_HOST_DEVICE Evaluation constant(const Evaluation& x, const Scalar& value)
{ return MathToolbox<Evaluation>::createConstant(x, value); }

template <class Evaluation, class Scalar>
OPM_HOST_DEVICE Evaluation variable(unsigned numDeriv, const Scalar& value, unsigned idx)
{ return MathToolbox<Evaluation>::createVariable(numDeriv, value, idx); }

template <class Evaluation, class Scalar>
OPM_HOST_DEVICE Evaluation variable(const Evaluation& x, const Scalar& value, unsigned idx)
{ return MathToolbox<Evaluation>::createVariable(x, value, idx); }

template <class Evaluation, class Scalar>
OPM_HOST_DEVICE Evaluation variable(const Scalar& value, unsigned idx)
{ return MathToolbox<Evaluation>::createVariable(value, idx); }

template <class ResultEval, class Evaluation>
OPM_HOST_DEVICE auto decay(const Evaluation& value)
    -> decltype(MathToolbox<Evaluation>::template decay<ResultEval>(value))
{ return MathToolbox<Evaluation>::template decay<ResultEval>(value); }

template <class Evaluation>
OPM_HOST_DEVICE auto getValue(const Evaluation& val)
    -> decltype(MathToolbox<Evaluation>::value(val))
{ return MathToolbox<Evaluation>::value(val); }

template <class Evaluation>
OPM_HOST_DEVICE auto scalarValue(const Evaluation& val)
    -> decltype(MathToolbox<Evaluation>::scalarValue(val))
{ return MathToolbox<Evaluation>::scalarValue(val); }

template <class Evaluation1, class Evaluation2>
typename ReturnEval_<Evaluation1, Evaluation2>::type
OPM_HOST_DEVICE max(const Evaluation1& arg1, const Evaluation2& arg2)
{ return MathToolbox<typename ReturnEval_<Evaluation1, Evaluation2>::type>::max(arg1, arg2); }

template <class Evaluation1, class Evaluation2>
typename ReturnEval_<Evaluation1, Evaluation2>::type
OPM_HOST_DEVICE min(const Evaluation1& arg1, const Evaluation2& arg2)
{ return MathToolbox<typename ReturnEval_<Evaluation1, Evaluation2>::type>::min(arg1, arg2); }

template <class Evaluation>
OPM_HOST_DEVICE Evaluation abs(const Evaluation& value)
{ return MathToolbox<Evaluation>::abs(value); }

template <class Evaluation>
OPM_HOST_DEVICE Evaluation tan(const Evaluation& value)
{ return MathToolbox<Evaluation>::tan(value); }

template <class Evaluation>
OPM_HOST_DEVICE Evaluation atan(const Evaluation& value)
{ return MathToolbox<Evaluation>::atan(value); }

template <class Evaluation1, class Evaluation2>
typename ReturnEval_<Evaluation1, Evaluation2>::type
OPM_HOST_DEVICE atan2(const Evaluation1& value1, const Evaluation2& value2)
{ return MathToolbox<typename ReturnEval_<Evaluation1, Evaluation2>::type>::atan2(value1, value2); }

template <class Evaluation>
OPM_HOST_DEVICE Evaluation sin(const Evaluation& value)
{ return MathToolbox<Evaluation>::sin(value); }

template <class Evaluation>
OPM_HOST_DEVICE Evaluation asin(const Evaluation& value)
{ return MathToolbox<Evaluation>::asin(value); }

template <class Evaluation>
OPM_HOST_DEVICE Evaluation sinh(const Evaluation& value)
{ return MathToolbox<Evaluation>::sinh(value); }

template <class Evaluation>
OPM_HOST_DEVICE Evaluation asinh(const Evaluation& value)
{ return MathToolbox<Evaluation>::asinh(value); }

template <class Evaluation>
OPM_HOST_DEVICE Evaluation cos(const Evaluation& value)
{ return MathToolbox<Evaluation>::cos(value); }

template <class Evaluation>
OPM_HOST_DEVICE Evaluation acos(const Evaluation& value)
{ return MathToolbox<Evaluation>::acos(value); }

template <class Evaluation>
OPM_HOST_DEVICE Evaluation cosh(const Evaluation& value)
{ return MathToolbox<Evaluation>::cosh(value); }

template <class Evaluation>
OPM_HOST_DEVICE Evaluation acosh(const Evaluation& value)
{ return MathToolbox<Evaluation>::acosh(value); }

template <class Evaluation>
OPM_HOST_DEVICE Evaluation sqrt(const Evaluation& value)
{ return MathToolbox<Evaluation>::sqrt(value); }

template <class Evaluation>
OPM_HOST_DEVICE Evaluation exp(const Evaluation& value)
{ return MathToolbox<Evaluation>::exp(value); }

template <class Evaluation>
OPM_HOST_DEVICE Evaluation log(const Evaluation& value)
{ return MathToolbox<Evaluation>::log(value); }

template <class Evaluation>
OPM_HOST_DEVICE Evaluation log10(const Evaluation& value)
{ return MathToolbox<Evaluation>::log10(value); }

template <class Evaluation1, class Evaluation2>
typename ReturnEval_<Evaluation1, Evaluation2>::type
OPM_HOST_DEVICE pow(const Evaluation1& base, const Evaluation2& exp)
{ return MathToolbox<typename ReturnEval_<Evaluation1, Evaluation2>::type>::pow(base, exp); }

template <class Evaluation>
OPM_HOST_DEVICE bool isfinite(const Evaluation& value)
{ return MathToolbox<Evaluation>::isfinite(value); }

template <class Evaluation>
OPM_HOST_DEVICE bool isnan(const Evaluation& value)
{ return MathToolbox<Evaluation>::isnan(value); }

} // namespace Opm

#endif