type_traits.h

#ifndef LINEARALGEBRA_SRC_MOLPRO_LINALG_ARRAY_UTIL_TYPE_TRAITS_H
#define LINEARALGEBRA_SRC_MOLPRO_LINALG_ARRAY_UTIL_TYPE_TRAITS_H
#include <cmath>
#include <cstdlib>
#include <type_traits>
 
namespace molpro::linalg::array {
template <class... Ts>
using void_t = void;
 
template <class A, class = void_t<>>
struct has_mapped_type : std::false_type {};
 
template <class A>
struct has_mapped_type<A, void_t<typename A::mapped_type>> : std::true_type {};
 
template <class A>
constexpr bool has_mapped_type_v = has_mapped_type<A>{};
 
template <class T>
constexpr bool is_sparse_v = has_mapped_type_v<T>;
 
template <class A, bool = has_mapped_type_v<A>>
struct mapped_or_value_type {
  using type = typename A::value_type;
};
 
template <class A>
struct mapped_or_value_type<A, true> {
  using type = typename A::mapped_type;
};
 
template <class A>
using mapped_or_value_type_t = typename mapped_or_value_type<A>::type;
 
template <class T, typename = void>
struct is_iterable : std::false_type {};
 
template <class T>
struct is_iterable<T, void_t<decltype(std::begin(std::declval<T>())), decltype(std::end(std::declval<T>())),
                             std::enable_if_t<!is_sparse_v<T>>>> : std::true_type {};
 
template <class T>
constexpr bool is_iterable_v = is_iterable<T>{};
 
template <class T, typename = void>
struct is_distributed : std::false_type {};
 
template <class T>
struct is_distributed<T, void_t<typename T::distributed_array>> : std::true_type {};
 
template <class T>
constexpr bool is_distributed_v = is_distributed<T>::value;
 
template <class T, typename = void>
struct is_disk : std::false_type {};
 
template <class T>
struct is_disk<T, void_t<typename T::disk_array>> : std::true_type {};
 
template <class T>
constexpr bool is_disk_v = is_disk<T>::value;
 
enum class ArrayFamily { None, Iterable, Sparse, Distributed, DistributedDisk };
 
template <class T, bool = is_iterable_v<T>, bool = is_sparse_v<T>, bool = is_distributed_v<T>, bool = is_disk_v<T>>
struct array_family {
  constexpr auto value() { return ArrayFamily::None; }
};
 
template <class T>
struct array_family<T, true, false, false, false> {
  constexpr auto value() { return ArrayFamily::Iterable; }
};
 
template <class T>
struct array_family<T, false, true, false, false> {
  constexpr auto value() { return ArrayFamily::Sparse; }
};
 
template <class T>
struct array_family<T, false, false, true, false> {
  constexpr auto value() { return ArrayFamily::Distributed; }
};
 
template <class T>
struct array_family<T, false, false, true, true> {
  constexpr auto value() { return ArrayFamily::DistributedDisk; }
};
 
template <class T>
constexpr auto array_family_v = array_family<T>{}.value();
 
template <typename T>
constexpr auto check_abs() {
  using std::abs;
  return abs(T{});
}
} // namespace molpro::linalg::array
#endif // LINEARALGEBRA_SRC_MOLPRO_LINALG_ARRAY_UTIL_TYPE_TRAITS_H