vector_stream_reference.hpp

Go to the documentation of this file.

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

#ifndef LIN_REFERENCES_VECTOR_STREAM_REFERENCE_HPP_
#define LIN_REFERENCES_VECTOR_STREAM_REFERENCE_HPP_

#include "../core.hpp"
#include "tensor_stream_reference.hpp"

namespace lin {
namespace internal {

template <class E, size_t N, size_t MN>
class VectorStreamReference : public TensorStreamReference<VectorStreamReference<E, N, MN>, E> {
  static_assert(is_col_vector<VectorStreamReference<E, N, MN>>::value,
      "Derived types to VectorStreamReference<...> must have column vector traits");

 public:
  typedef traits<VectorStreamReference<E, N, MN>> Traits;

  typedef vector_traits<VectorStreamReference<E, N, MN>> VectorTraits;

 protected:
  using TensorStreamReference<VectorStreamReference<E, N, MN>, E>::derived;
  using TensorStreamReference<VectorStreamReference<E, N, MN>, E>::resize;

 public:
  using TensorStreamReference<VectorStreamReference<E, N, MN>, E>::TensorStreamReference;
  using TensorStreamReference<VectorStreamReference<E, N, MN>, E>::rows;
  using TensorStreamReference<VectorStreamReference<E, N, MN>, E>::cols;
  using TensorStreamReference<VectorStreamReference<E, N, MN>, E>::size;
  using TensorStreamReference<VectorStreamReference<E, N, MN>, E>::operator=;
  using TensorStreamReference<VectorStreamReference<E, N, MN>, E>::operator();
  using TensorStreamReference<VectorStreamReference<E, N, MN>, E>::eval;

  constexpr VectorStreamReference() = delete;
  constexpr VectorStreamReference(VectorStreamReference<E, N, MN> const &) = default;
  constexpr VectorStreamReference(VectorStreamReference<E, N, MN> &&) = default;
  constexpr VectorStreamReference<E, N, MN> &operator=(VectorStreamReference<E, N, MN> const &) = default;
  constexpr VectorStreamReference<E, N, MN> &operator=(VectorStreamReference<E, N, MN> &&) = default;

  constexpr VectorStreamReference(Stream<E> const &stream, size_t i, size_t j, size_t n)
  : TensorStreamReference<VectorStreamReference<E, N, MN>, E>(stream, i, j, n, 1) { }
};

template <class E, size_t N, size_t MN>
struct _elem<VectorStreamReference<E, N, MN>> {
  typedef _elem_t<E> type;
};

template <class E, size_t N, size_t MN>
struct _dims<VectorStreamReference<E, N, MN>> {
  static constexpr size_t rows = N;
  static constexpr size_t cols = 1;
  static constexpr size_t max_rows = MN;
  static constexpr size_t max_cols = 1;
};

template <class E, size_t N, size_t MN>
class RowVectorStreamReference : public TensorStreamReference<RowVectorStreamReference<E, N, MN>, E> {
  static_assert(is_row_vector<RowVectorStreamReference<E, N, MN>>::value,
      "Derived types to RowVectorStreamReference<...> must have row vector traits");

 public:
  typedef traits<RowVectorStreamReference<E, N, MN>> Traits;

  typedef vector_traits<RowVectorStreamReference<E, N, MN>> VectorTraits;

 protected:
  using TensorStreamReference<RowVectorStreamReference<E, N, MN>, E>::derived;
  using TensorStreamReference<RowVectorStreamReference<E, N, MN>, E>::resize;

 public:
  using TensorStreamReference<RowVectorStreamReference<E, N, MN>, E>::TensorStreamReference;
  using TensorStreamReference<RowVectorStreamReference<E, N, MN>, E>::rows;
  using TensorStreamReference<RowVectorStreamReference<E, N, MN>, E>::cols;
  using TensorStreamReference<RowVectorStreamReference<E, N, MN>, E>::size;
  using TensorStreamReference<RowVectorStreamReference<E, N, MN>, E>::operator=;
  using TensorStreamReference<RowVectorStreamReference<E, N, MN>, E>::operator();
  using TensorStreamReference<RowVectorStreamReference<E, N, MN>, E>::eval;

  constexpr RowVectorStreamReference() = delete;
  constexpr RowVectorStreamReference(RowVectorStreamReference<E, N, MN> const &) = default;
  constexpr RowVectorStreamReference(RowVectorStreamReference<E, N, MN> &&) = default;
  constexpr RowVectorStreamReference<E, N, MN> &operator=(RowVectorStreamReference<E, N, MN> const &) = default;
  constexpr RowVectorStreamReference<E, N, MN> &operator=(RowVectorStreamReference<E, N, MN> &&) = default;

  constexpr RowVectorStreamReference(Stream<E> const &stream, size_t i, size_t j, size_t n)
  : TensorStreamReference<RowVectorStreamReference<E, N, MN>, E>(stream, i, j, 1, n) { }
};

template <class E, size_t N, size_t MN>
struct _elem<RowVectorStreamReference<E, N, MN>> {
  typedef _elem_t<E> type;
};

template <class E, size_t N, size_t MN>
struct _dims<RowVectorStreamReference<E, N, MN>> {
  static constexpr size_t rows = 1;
  static constexpr size_t cols = N;
  static constexpr size_t max_rows = 1;
  static constexpr size_t max_cols = MN;
};
}  // namespace internal
}  // namespace lin

#endif