index_type.hpp

Go to the documentation of this file.

#ifndef STAN_MATH_PRIM_META_INDEX_TYPE_HPP
#define STAN_MATH_PRIM_META_INDEX_TYPE_HPP
 
#include <stan/math/prim/fun/Eigen.hpp>
#include <stan/math/prim/meta/is_eigen.hpp>
#include <stan/math/prim/meta/is_vector.hpp>
#include <stan/math/prim/meta/require_generics.hpp>
#include <type_traits>
#include <vector>
 
namespace stan {
namespace math {
 
template <typename T, typename = void>
struct index_type {};
 
template <typename T>
struct index_type<T, std::enable_if_t<std::is_pointer<T>::value>> {
  using type = int;
};
 
template <typename T>
using index_type_t = typename index_type<T>::type;
 
template <typename T>
struct index_type<T, require_std_vector_t<T>> {
  using type = typename std::decay_t<T>::size_type;
};
 
template <typename T>
struct index_type<T, require_eigen_t<T>> {
  using type = typename std::decay_t<T>::Index;
};
 
}  // namespace math
 
}  // namespace stan
 
#endif