is_var.hpp

Go to the documentation of this file.

#ifndef STAN_MATH_PRIM_META_IS_VAR_HPP
#define STAN_MATH_PRIM_META_IS_VAR_HPP
 
#include <stan/math/prim/meta/require_helpers.hpp>
 
#include <type_traits>
 
namespace stan {
template <typename T, typename = void>
struct is_var : std::false_type {};
 
template <typename T>
using has_var_scalar_type = is_var<scalar_type_t<T>>;
 
template <typename T>
inline constexpr bool is_var_v = is_var<T>::value;
 
namespace internal {
 
template <typename... Types>
struct is_any_var_scalar_impl {
  static constexpr bool value
      = (has_var_scalar_type<std::decay_t<Types>>::value || ...);
};
 
template <typename... Types>
struct is_any_var_scalar_impl<std::tuple<Types...>> {
  static constexpr bool value
      = (is_any_var_scalar_impl<std::decay_t<Types>>::value || ...);
};
 
template <typename T, typename... VecArgs>
struct is_any_var_scalar_impl<std::vector<T, VecArgs...>> {
  static constexpr bool value = is_any_var_scalar_impl<std::decay_t<T>>::value;
};
 
}  // namespace internal
 
template <typename... Types>
struct is_any_var_scalar
    : std::disjunction<
          internal::is_any_var_scalar_impl<std::decay_t<Types>>...> {};
 
template <typename... Types>
constexpr bool is_any_var_scalar_v
    = is_any_var_scalar<std::decay_t<Types>...>::value;
 
namespace internal {
template <typename... Types>
struct is_all_var_scalar_impl {
  static constexpr bool value
      = (has_var_scalar_type<std::decay_t<Types>>::value && ...);
};
template <typename... Types>
struct is_all_var_scalar_impl<std::tuple<Types...>> {
  static constexpr bool value
      = (is_all_var_scalar_impl<std::decay_t<Types>>::value && ...);
};
 
template <typename... Types, typename... VecArgs>
struct is_all_var_scalar_impl<std::vector<std::tuple<Types...>, VecArgs...>> {
  static constexpr bool value
      = (is_all_var_scalar_impl<std::decay_t<Types>>::value && ...);
};
 
}  // namespace internal
 
template <typename... Types>
struct is_all_var_scalar
    : std::disjunction<
          internal::is_all_var_scalar_impl<std::decay_t<Types>>...> {};
 
template <typename... Types>
constexpr bool is_all_var_scalar_v
    = is_all_var_scalar<std::decay_t<Types>...>::value;
 
template <typename T>
using require_var_t = require_t<is_var<std::decay_t<T>>>;
 
template <typename T>
using require_not_var_t = require_not_t<is_var<std::decay_t<T>>>;
 
template <typename... Types>
using require_all_var_t = require_all_t<is_var<std::decay_t<Types>>...>;
 
template <typename... Types>
using require_any_var_t = require_any_t<is_var<std::decay_t<Types>>...>;
 
template <typename... Types>
using require_all_not_var_t = require_all_not_t<is_var<std::decay_t<Types>>...>;
 
template <typename... Types>
using require_any_not_var_t = require_any_not_t<is_var<std::decay_t<Types>>...>;
template <template <class...> class TypeCheck, class... Check>
using require_var_vt = require_t<
    container_type_check_base<is_var, value_type_t, TypeCheck, Check...>>;
 
template <template <class...> class TypeCheck, class... Check>
using require_any_var_vt = require_any_t<
    container_type_check_base<is_var, value_type_t, TypeCheck, Check>...>;
 
template <template <class...> class TypeCheck, class... Check>
using require_all_var_vt = require_all_t<
    container_type_check_base<is_var, value_type_t, TypeCheck, Check>...>;
template <typename T>
using require_not_vt_var = require_not_t<is_var<value_type_t<std::decay_t<T>>>>;
 
template <typename... Types>
using require_any_vt_var
    = require_any_t<is_var<value_type_t<std::decay_t<Types>>>...>;
 
template <typename... Types>
using require_all_not_vt_var
    = require_all_not_t<is_var<value_type_t<std::decay_t<Types>>>...>;
 
template <typename T>
using require_st_var = require_t<is_var<scalar_type_t<std::decay_t<T>>>>;
 
template <typename T>
using require_not_st_var
    = require_not_t<is_var<scalar_type_t<std::decay_t<T>>>>;
 
template <typename... Types>
using require_all_st_var
    = require_all_t<is_var<scalar_type_t<std::decay_t<Types>>>...>;
 
template <typename... Types>
using require_any_st_var
    = require_any_t<is_var<scalar_type_t<std::decay_t<Types>>>...>;
 
template <typename... Types>
using require_all_not_st_var
    = require_all_not_t<is_var<scalar_type_t<std::decay_t<Types>>>...>;
template <typename T>
struct value_type<T, std::enable_if_t<is_var<T>::value>> {
  using type = typename std::decay_t<T>::value_type;
};
 
}  // namespace stan
#endif