deduce_hp_types.hpp

Go to the documentation of this file.

#ifndef MLPACK_CORE_HPT_DEDUCE_HP_TYPES_HPP
#define MLPACK_CORE_HPT_DEDUCE_HP_TYPES_HPP

#include <mlpack/core/hpt/fixed.hpp>

namespace mlpack {
namespace hpt {

template<typename... Args>
struct DeduceHyperParameterTypes
{
  template<typename... HPTypes>
  struct ResultHolder
  {
    using TupleType = std::tuple;
  };
};

template<typename T, typename... Args>
struct DeduceHyperParameterTypes
{
  template<typename ArgumentType,
           bool IsArithmetic = std::is_arithmetic::value>
  struct ResultHPType;

  template<typename ArithmeticType>
  struct ResultHPType
  {
    using Type = ArithmeticType;
  };

  template<typename Type>
  struct IsCollectionType
  {
    using Yes = char[1];
    using No = char[2];

    template<typename TypeToCheck>
    static Yes& Check(typename TypeToCheck::value_type*);
    template<typename>
    static No& Check(...);

    static const bool value  =
      sizeof(decltype(Check(0))) == sizeof(Yes);
  };

  template<typename CollectionType>
  struct ResultHPType
  {
    static_assert(IsCollectionType::value,
        "One of the passed arguments is neither of an arithmetic type, nor of "
        "a collection type, nor fixed with the Fixed function.");

    using Type = typename CollectionType::value_type;
  };

  template<typename... HPTypes>
  struct ResultHolder
  {
    using TupleType = typename DeduceHyperParameterTypes::template
        ResultHoldertypename ResultHPType::Type>::TupleType;
  };

  using TupleType = typename ResultHolder<>::TupleType;
};

template<typename T, typename... Args>
struct DeduceHyperParameterTypes<PreFixedArg, Args...>
{
  template<typename... HPTypes>
  struct ResultHolder
  {
    using TupleType = typename DeduceHyperParameterTypes::template
        ResultHolder::TupleType;
  };

  using TupleType = typename ResultHolder<>::TupleType;
};

template<typename... Args>
using TupleOfHyperParameters =
    typename DeduceHyperParameterTypes::TupleType;

} // namespace hpt
} // namespace mlpack

#endif