Continue cylinder2d refinement setup encapsulation

1 files changed, 12 insertions, 0 deletions

diff --git a/src/refinement/grid2D.hh b/src/refinement/grid2D.hh
index fdf6461..9ca8aaf 100644
--- a/src/refinement/grid2D.hh
+++ b/src/refinement/grid2D.hh
@@ -281,6 +281,18 @@ void Grid2D<T,DESCRIPTOR>::forEachGrid(
 }
 
 template <typename T, template<typename> class DESCRIPTOR>
+Grid2D<T,DESCRIPTOR>& Grid2D<T,DESCRIPTOR>::locate(Vector<T,2> pos)
+{
+  int iC;
+  for (auto& grid : _fineGrids) {
+    if (grid->getCuboidGeometry().getC(pos, iC)) {
+      return grid->locate(pos);
+    }
+  }
+  return *this;
+}
+
+template <typename T, template<typename> class DESCRIPTOR>
 std::size_t Grid2D<T,DESCRIPTOR>::getActiveVoxelN() const
 {
   std::size_t n = _geometry->getStatistics().getNvoxel();