add_decomposable_evaluate_const.hpp

Go to the documentation of this file.

#ifndef MLPACK_CORE_OPTIMIZERS_FUNCTION_ADD_DECOMPOSABLE_EVALUATE_HPP
#define MLPACK_CORE_OPTIMIZERS_FUNCTION_ADD_DECOMPOSABLE_EVALUATE_HPP

#include "traits.hpp"

namespace mlpack {
namespace optimization {

template<typename FunctionType,
         bool HasDecomposableEvaluateWithGradient =
             traits::HasEvaluateWithGradient<FunctionType,
                 traits::DecomposableEvaluateWithGradientForm>::value,
         bool HasDecomposableEvaluate =
             traits::HasEvaluate<FunctionType,
                 traits::DecomposableEvaluateForm>::value>
class AddDecomposableEvaluate
{
 public:
  // Provide a dummy overload so the name 'Evaluate' exists for this object.
  double Evaluate(traits::UnconstructableType&, const size_t);
};

template<typename FunctionType, bool HasDecomposableEvaluateWithGradient>
class AddDecomposableEvaluate<FunctionType, HasDecomposableEvaluateWithGradient,
    true>
{
 public:
  // Reflect the existing Evaluate().
  double Evaluate(const arma::mat& coordinates, const size_t index)
  {
    return static_cast<FunctionType*>(
        static_cast<Function<FunctionType>*>(this))->Evaluate(coordinates,
        index);
  }
};

template<typename FunctionType>
class AddDecomposableEvaluate<FunctionType, true, false>
{
 public:
  double Evaluate(const arma::mat& coordinates,
                  const size_t begin,
                  const size_t batchSize)
  {
    arma::mat gradient; // This will be ignored.
    return static_cast<Function<FunctionType>*>(this)->EvaluateWithGradient(
        coordinates, begin, gradient, batchSize);
  }
};

template<typename FunctionType,
         bool HasDecomposableEvaluateWithGradient =
             traits::HasEvaluateWithGradient<FunctionType,
                 traits::DecomposableEvaluateWithGradientConstForm>::value,
         bool HasDecomposableEvaluate =
             traits::HasEvaluate<FunctionType,
                 traits::DecomposableEvaluateConstForm>::value>
class AddDecomposableEvaluateConst { };

template<typename FunctionType>
class AddDecomposableEvaluateConst<FunctionType, true, false>
{
 public:
  double Evaluate(const arma::mat& coordinates,
                  const size_t begin,
                  const size_t batchSize) const
  {
    arma::mat gradient; // This will be ignored.
    return
        static_cast<const Function<FunctionType>*>(this)->EvaluateWithGradient(
        coordinates, begin, gradient, batchSize);
  }
};

template<typename FunctionType,
         bool HasDecomposableEvaluateWithGradient =
             traits::HasEvaluateWithGradient<FunctionType,
                 traits::DecomposableEvaluateWithGradientStaticForm>::value,
         bool HasDecomposableEvaluate =
             traits::HasEvaluate<FunctionType,
                 traits::DecomposableEvaluateStaticForm>::value>
class AddDecomposableEvaluateStatic { };

template<typename FunctionType>
class AddDecomposableEvaluateStatic<FunctionType, true, false>
{
 public:
  static double Evaluate(const arma::mat& coordinates,
                         const size_t begin,
                         const size_t batchSize)
  {
    arma::mat gradient; // This will be ignored.
    return FunctionType::EvaluateWithGradient(coordinates, begin, gradient,
        batchSize);
  }
};

} // namespace optimization
} // namespace mlpack

#endif