Program Listing for File grid2D_view_entity.hpp

↰ Return to documentation for file (cosm/repr/grid2D_view_entity.hpp)

#pragma once

/*******************************************************************************
 * Includes
 ******************************************************************************/
#include "cosm/cosm.hpp"
#include "cosm/repr/base_grid_view_entity.hpp"
#include "cosm/repr/entity2D.hpp"

/*******************************************************************************
 * Namespaces
 ******************************************************************************/
namespace cosm::repr {

/*******************************************************************************
 * Class Definitions
 ******************************************************************************/
template <typename TGridType, typename TGridViewType>
class grid2D_view_entity
    : public crepr::entity2D,
      public crepr::base_grid_view_entity<TGridType, TGridViewType>,
      public rer::client<grid2D_view_entity<TGridType, TGridViewType>> {
 public:
  using base_grid_view_entity_type =
      base_grid_view_entity<TGridType, TGridViewType>;

  using typename base_grid_view_entity_type::cell_type;
  using typename base_grid_view_entity_type::coord_type;
  using typename base_grid_view_entity_type::grid_type;
  using typename base_grid_view_entity_type::grid_view_type;

  using base_grid_view_entity_type::access;
  using base_grid_view_entity_type::resolution;

  grid2D_view_entity(const rtypes::type_uuid& id,
                     const grid_view_type& the_view,
                     const rtypes::discretize_ratio& res)
      : entity2D(id,
                 rmath::vector3z({ the_view.shape()[0], the_view.shape()[1] }, 0),
                 rmath::vector3z(the_view.origin()->loc(), 0),
                 rspatial::euclidean_dist(res.v())),
        base_grid_view_entity_type(the_view, res),
        ER_CLIENT_INIT("cosm.repr.grid2D_view_entity") {}

  ~grid2D_view_entity(void) override = default;

  const cell_type& access(size_t i, size_t j) const {
    ER_ASSERT(i < xdsize(), "Out of bounds X access: %zu >= %zu", i, xdsize());
    ER_ASSERT(j < ydsize(), "Out of bounds Y access: %zu >= %zu", j, ydsize());
    return view()[i][j];
  }

  const cell_type& access(const coord_type& c) const override {
    return access(c.x(), c.y());
  }

  bool contains_abs(const coord_type& cell) const override {
    return xdspan().contains(cell.x()) && ydspan().contains(cell.y());
  }

  bool contains_rel(const coord_type& cell) const override {
    return (cell.x() < xdsize()) && (cell.y() < ydsize());
  }

 protected:
  using base_grid_view_entity_type::view;
};

} /* namespace cosm::repr */