array3d.hpp

Go to the documentation of this file.

#pragma once
#include <vector>
 
#include "NNL/common/contract.hpp"
#include "NNL/common/exception.hpp"
namespace nnl {
 
namespace utl {
template <typename T>
class Array3D {
 public:
  Array3D() noexcept {}
  Array3D(std::size_t z, std::size_t y, std::size_t x) : sz_(z), sy_(y), sx_(x) { data_.resize(sz_ * sy_ * sx_, T()); }
  void Resize(std::size_t z, std::size_t y, std::size_t x) {
    sz_ = z;
    sy_ = y;
    sx_ = x;
    data_.resize(sz_ * sy_ * sx_, T{});
  }

  std::size_t SizeZ() const noexcept { return sz_; }
  std::size_t SizeY() const noexcept { return sy_; }
  std::size_t SizeX() const noexcept { return sx_; }
  T& At(std::size_t z, std::size_t y, std::size_t x) {
    std::size_t index = GetIndex(z, y, x);
    if (index >= data_.size()) NNL_THROW(RangeError(NNL_SRCTAG("index out of range: " + std::to_string(index))));
    return data_.at(index);
  }

  const T& At(std::size_t z, std::size_t y, std::size_t x) const {
    std::size_t index = GetIndex(z, y, x);
    if (index >= data_.size()) NNL_THROW(RangeError(NNL_SRCTAG("index out of range: " + std::to_string(index))));
    return data_.at(index);
  }

  T& operator()(std::size_t z, std::size_t y, std::size_t x) noexcept {
    NNL_EXPECTS_DBG(GetIndex(z, y, x) < data_.size());
    return data_[GetIndex(z, y, x)];
  }

  const T& operator()(std::size_t z, std::size_t y, std::size_t x) const noexcept {
    NNL_EXPECTS_DBG(GetIndex(z, y, x) < data_.size());
    return data_[GetIndex(z, y, x)];
  }
 
 private:
  inline std::size_t GetIndex(std::size_t z, std::size_t y, std::size_t x) const noexcept {
    return x + y * sx_ + sx_ * sy_ * z;
  }
 
  std::vector<T> data_;
  std::size_t sz_;
  std::size_t sy_;
  std::size_t sx_;
};

}  // namespace utl
}  // namespace nnl