vector.hpp

Go to the documentation of this file.

// vim: set tabstop=2:softtabstop=2:shiftwidth=2:expandtab

#ifndef LIN_CORE_TRAITS_VECTOR_HPP_
#define LIN_CORE_TRAITS_VECTOR_HPP_

#include "tensor.hpp"
#include "utilities.hpp"

#include <type_traits>

namespace lin {
namespace internal {

template <class C>
struct is_row_vector : std::integral_constant<bool, (
    has_fixed_rows<C>::value &&
    (_dims<C>::rows == 1) &&
    (_dims<C>::max_cols > 1)
  )> { };

template <class C>
struct is_col_vector : std::integral_constant<bool, (
    has_fixed_cols<C>::value &&
    (_dims<C>::cols == 1) &&
    (_dims<C>::max_rows > 1)
  )> { };

template <class C>
struct is_vector : disjunction<is_row_vector<C>, is_col_vector<C>> { };

template <class C, typename = void>
struct _vector_dims;

template <class C>
struct _vector_dims<C, std::enable_if_t<is_row_vector<C>::value>> {
  static constexpr size_t length = _dims<C>::cols;
  static constexpr size_t max_length = _dims<C>::max_cols;
};

template <class C>
struct _vector_dims<C, std::enable_if_t<is_col_vector<C>::value>> {
  static constexpr size_t length = _dims<C>::rows;
  static constexpr size_t max_length = _dims<C>::max_rows;
};

template <class C>
struct vector_traits {
  static_assert(is_vector<C>::value,
      "lin::internal::vector_traits can only be populated for vector types");
  typedef _elem_t<C> elem_t;
  typedef _eval_t<C> eval_t;
  static constexpr size_t length = _vector_dims<C>::length;
  static constexpr size_t max_length = _vector_dims<C>::max_length;
};

template <class C, class D>
struct _have_same_vector_dimensions : std::integral_constant<bool, (
    (_vector_dims<C>::length == _vector_dims<D>::length) &&
    (_vector_dims<C>::max_length == _vector_dims<D>::max_length)
  )> { };

template <class... Cs>
struct have_same_vector_dimensions;

template <>
struct have_same_vector_dimensions<> : std::true_type { };

template <class C>
struct have_same_vector_dimensions<C> : std::true_type {
  static_assert(is_vector<C>::value,
      "lin::internal::have_same_vector_dimensions cannot be passed a non-vector type");
};

template <class C, class... Cs>
struct have_same_vector_dimensions<C, Cs...>
    : conjunction<_have_same_vector_dimensions<C, Cs>...> { };

template <class... Cs>
struct have_same_vector_traits : conjunction<
    have_same_elements<Cs...>, have_same_vector_dimensions<Cs...>
  > { };

}  // namespace internal
}  // namespace lin

#endif