tensor_view.hpp

Go to the documentation of this file.

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

#ifndef LIN_VIEWS_TENSOR_VIEW_HPP_
#define LIN_VIEWS_TENSOR_VIEW_HPP_

#include "../core.hpp"

namespace lin {
namespace internal {

template <class D>
class TensorView : public Base<D> {
  static_assert(has_valid_traits<D>::value,
      "Derived types to Tensor<...> must have valid traits");

 public:
  typedef traits<D> Traits;

 private:
  typename Traits::elem_t *const elems;

 protected:
  using Base<D>::derived;

 public:
  using Base<D>::rows;
  using Base<D>::cols;
  using Base<D>::resize;
  using Base<D>::size;
  using Base<D>::operator=;
  using Base<D>::operator();
  using Base<D>::data;
  using Base<D>::eval;

  constexpr TensorView() = delete;
  constexpr TensorView(TensorView<D> const &) = default;
  constexpr TensorView(TensorView<D> &&) = default;
  constexpr TensorView<D> &operator=(TensorView<D> const &) = default;
  constexpr TensorView<D> &operator=(TensorView<D> &&) = default;

  constexpr TensorView(typename Traits::elem_t *elems)
  : elems(elems) {
    resize(Traits::max_rows, Traits::max_cols);
  }

  constexpr TensorView(typename Traits::elem_t *elems, size_t r, size_t c)
  : elems(elems) {
    resize(r, c);
  }

  constexpr typename Traits::elem_t *data() {
    return elems;
  }
};
}  // namespace internal
}  // namespace lin

#endif