Program Listing for File arena_grid.hpp

↰ Return to documentation for file (cosm/arena/ds/arena_grid.hpp)

#pragma once

/*******************************************************************************
 * Includes
 ******************************************************************************/
#include <shared_mutex>
#include <tuple>

#include "rcppsw/ds/stacked_grid2D.hpp"
#include "rcppsw/types/discretize_ratio.hpp"

#include "cosm/ds/cell2D.hpp"

/*******************************************************************************
 * Namespaces
 ******************************************************************************/
namespace cosm::arena::ds {

using arena_layer_stack = std::tuple<cds::cell2D>;

/*******************************************************************************
 * Class Definitions
 ******************************************************************************/
class arena_grid : public rds::stacked_grid2D<arena_layer_stack> {
 public:
  using value_type = cds::cell2D;
  using coord_type = rmath::vector2z;
  using view = rds::base_grid2D<cds::cell2D>::grid_view;
  using const_view = rds::base_grid2D<cds::cell2D>::const_grid_view;

  static constexpr const size_t kCell = 0;

  static bool has_overlap(const view& v1, const view& v2) {
    auto v1_xspan = rmath::range(v1.origin()->loc().x(),
                                 v1.origin()->loc().x() + v1.shape()[0]);
    auto v1_yspan = rmath::range(v1.origin()->loc().y(),
                                 v1.origin()->loc().y() + v1.shape()[1]);
    auto v2_xspan = rmath::range(v2.origin()->loc().x(),
                                 v2.origin()->loc().x() + v2.shape()[0]);
    auto v2_yspan = rmath::range(v2.origin()->loc().y(),
                                 v2.origin()->loc().y() + v2.shape()[1]);
    return v1_xspan.overlaps_with(v2_xspan) && v1_yspan.overlaps_with(v2_yspan);
  }
  arena_grid(const rmath::vector2d& dims,
             const rtypes::discretize_ratio& resolution)
      : stacked_grid2D(rmath::vector2d(0.0, 0.0), dims, resolution, resolution) {
    for (size_t i = 0; i < xdsize(); ++i) {
      for (size_t j = 0; j < ydsize(); ++j) {
        access<kCell>(i, j).loc(rmath::vector2z(i, j));
      } /* for(j..) */
    } /* for(i..) */
  }

  void reset(void) {
    for (size_t i = 0; i < xdsize(); ++i) {
      for (size_t j = 0; j < ydsize(); ++j) {
        access<kCell>(i, j).reset();
      } /* for(j..) */
    } /* for(i..) */
  } /* reset */

  std::shared_mutex* mtx(void) { return &m_mtx; }

 private:
  /* clang-format off */
  std::shared_mutex m_mtx{};
  /* clang-format on */
};

} /* namespace cosm::arena::ds */