tensor_mapping_reference.hpp

Go to the documentation of this file.

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

#ifndef LIN_REFERENCES_TENSOR_MAPPING_REFERENCE_HPP_
#define LIN_REFERENCES_TENSOR_MAPPING_REFERENCE_HPP_

#include "../core.hpp"

namespace lin {
namespace internal {

template <class D, class E>
class TensorMappingReference : public Mapping<D>, public Dimensions<D> {
 public:
  typedef traits<D> Traits;

 private:
  Mapping<E> &_mapping;
  size_t const _i;
  size_t const _j;

 protected:
  using Mapping<D>::derived;

  using Dimensions<D>::resize;

 public:
  using Mapping<D>::size;
  using Mapping<D>::operator=;
  using Mapping<D>::operator();
  using Mapping<D>::eval;

  using Dimensions<D>::rows;
  using Dimensions<D>::cols;

  constexpr TensorMappingReference() = delete;
  constexpr TensorMappingReference(TensorMappingReference<D, E> const &) = default;
  constexpr TensorMappingReference(TensorMappingReference<D, E> &&) = default;
  constexpr TensorMappingReference<D, E> &operator=(TensorMappingReference<D, E> const &) = default;
  constexpr TensorMappingReference<D, E> &operator=(TensorMappingReference<D, E> &&) = default;

  constexpr TensorMappingReference(Mapping<E> &mapping, size_t i, size_t j)
  : _mapping(mapping), _i(i), _j(j) {
    LIN_ASSERT((i >= 0) && (i + Traits::max_rows <= mapping.rows()));
    LIN_ASSERT((j >= 0) && (j + Traits::max_cols <= mapping.cols()));

    resize(Traits::max_rows, Traits::max_cols);
  }

  constexpr TensorMappingReference(Mapping<E> &mapping, size_t i, size_t j, size_t r, size_t c)
  : _mapping(mapping), _i(i), _j(j) {
    LIN_ASSERT((i >= 0) && (i + r <= mapping.rows()));
    LIN_ASSERT((j >= 0) && (j + c <= mapping.cols()));

    resize(r, c);
  }

  constexpr typename Traits::elem_t &operator()(size_t i, size_t j) {
    LIN_ASSERT((i >= 0) && (i < rows()));
    LIN_ASSERT((j >= 0) && (j < cols()));

    return _mapping(_i + i, _j + j);
  }

  constexpr typename Traits::elem_t &operator()(size_t i) {
    LIN_ASSERT((i >= 0) && (i < size()));

    return operator()(i / cols(), i % cols());
  }
};
}  // namespace internal
}  // namespace lin

#endif