From 77869487e437b4050f98c28a24db82e1e83a872c Mon Sep 17 00:00:00 2001
From: Adrian Kummerlaender
Date: Wed, 6 Feb 2019 21:02:21 +0100
Subject: Continue cylinder2d refinement setup encapsulation

---
 src/refinement/grid2D.hh | 12 ++++++++++++
 1 file changed, 12 insertions(+)

(limited to 'src/refinement/grid2D.hh')

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
@@ -280,6 +280,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
 {
-- 
cgit v1.2.3