stream.hpp

Go to the documentation of this file.

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

#ifndef LIN_CORE_TYPES_STREAM_HPP_
#define LIN_CORE_TYPES_STREAM_HPP_

#ifdef LIN_DESKTOP
  #include <iostream>
#endif

#include "../config.hpp"
#include "../traits.hpp"

namespace lin {
namespace internal {

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

 public:
  typedef traits<D> Traits;

 protected:
  inline constexpr D &derived() {
    return static_cast<D &>(*this);
  }

  inline constexpr D const &derived() const {
    return static_cast<D const &>(*this);
  }

 public:
  constexpr Stream() = default;
  constexpr Stream(Stream<D> const &) = default;
  constexpr Stream(Stream<D> &&) = default;
  constexpr Stream<D> &operator=(Stream<D> const &) = default;
  constexpr Stream<D> &operator=(Stream<D> &&) = default;

  inline constexpr size_t rows() const {
    return derived().rows();
  }

  inline constexpr size_t cols() const {
    return derived().cols();
  }

  inline constexpr size_t size() const {
    return rows() * cols();
  }

  inline constexpr typename Traits::elem_t operator()(size_t i, size_t j) const {
    return derived()(i, j);
  }

  inline constexpr typename Traits::elem_t operator()(size_t i) const {
    return derived()(i);
  }

  inline constexpr typename Traits::elem_t operator[](size_t i) const {
    return derived()(i);
  }

  inline constexpr typename Traits::eval_t eval() const {
    return typename Traits::eval_t(derived());
  }
};

#ifdef LIN_DESKTOP
template <class C>
std::ostream &operator<<(std::ostream &os, Stream<C> const &s) {
  for (size_t i = 0; i < s.rows() - 1; i++) {
    for (size_t j = 0; j < s.cols() - 1; j++)
      os << s(i, j) << " ";
    os << s(i, s.cols() - 1) << "\n";
  }
  for (size_t j = 0; j < s.cols() - 1; j++)
    os << s(s.rows() - 1, j) << " ";
  os << s(s.rows() - 1, s.cols() - 1) << "\n";
  return os;
}
#endif

}  // namespace internal
}  // namespace lin

#endif