Merge pull request #936 from Geode-solutions/feat/generic_aabb_search

feat(GenericMeshAABB): Created a class to access a mesh AABB generica…

Author: panquez

include/geode/mesh/helpers/aabb_edged_curve_helpers.h

@@ -36,6 +36,8 @@ namespace geode
     FORWARD_DECLARATION_DIMENSION_CLASS( Point );
     FORWARD_DECLARATION_DIMENSION_CLASS( AABBTree );
     FORWARD_DECLARATION_DIMENSION_CLASS( EdgedCurve );
+    template < typename T >
+    class GenericMeshAABB;
 } // namespace geode
 
 namespace geode
@@ -60,4 +62,29 @@ namespace geode
         const EdgedCurve< dimension >& mesh_;
     };
     ALIAS_2D_AND_3D( DistanceToEdge );
+
+    template < index_t dimension >
+    class GenericMeshAABB< EdgedCurve< dimension > >
+    {
+    public:
+        GenericMeshAABB( const EdgedCurve< dimension >& mesh )
+            : elements_tree_{ create_aabb_tree( mesh ) },
+              distance_action_{ mesh }
+        {
+        }
+
+        std::tuple< index_t, Point< dimension >, double > closest_element(
+            const Point< dimension >& query ) const
+        {
+            return this->elements_aabb().closest_element_box(
+                query, distance_action_ );
+        }
+
+    private:
+        AABBTree< dimension > elements_tree_;
+        const DistanceToEdge< dimension > distance_action_;
+    };
+    template < index_t dimension >
+    using EdgedCurveAABB = GenericMeshAABB< EdgedCurve< dimension > >;
+    ALIAS_2D_AND_3D( EdgedCurveAABB );
 } // namespace geode

include/geode/mesh/helpers/aabb_solid_helpers.h

@@ -31,6 +31,8 @@ namespace geode
     FORWARD_DECLARATION_DIMENSION_CLASS( AABBTree );
     FORWARD_DECLARATION_DIMENSION_CLASS( SolidMesh );
     FORWARD_DECLARATION_DIMENSION_CLASS( TetrahedralSolid );
+    template < typename T >
+    class GenericMeshAABB;
 } // namespace geode
 
 namespace geode
@@ -55,4 +57,53 @@ namespace geode
         const TetrahedralSolid< dimension >& mesh_;
     };
     ALIAS_3D( DistanceToTetrahedron );
+
+    template < index_t dimension >
+    class GenericMeshAABB< SolidMesh< dimension > >
+    {
+    public:
+        GenericMeshAABB( const SolidMesh< dimension >& mesh )
+            : elements_tree_{ create_aabb_tree( mesh ) }
+        {
+        }
+
+    protected:
+        const AABBTree< dimension >& elements_aabb() const
+        {
+            return elements_tree_;
+        }
+
+    private:
+        AABBTree< dimension > elements_tree_;
+    };
+    template < index_t dimension >
+    using SolidMeshAABB = GenericMeshAABB< SolidMesh< dimension > >;
+    ALIAS_3D( SolidMeshAABB );
+
+    template < index_t dimension >
+    class GenericMeshAABB< TetrahedralSolid< dimension > >
+        : public GenericMeshAABB< SolidMesh< dimension > >
+    {
+    public:
+        using Base = GenericMeshAABB< SolidMesh< dimension > >;
+
+        GenericMeshAABB( const TetrahedralSolid< dimension >& mesh )
+            : Base{ mesh }, distance_action_{ mesh }
+        {
+        }
+
+        std::tuple< index_t, Point< dimension >, double > closest_element(
+            const Point< dimension >& query ) const
+        {
+            return this->elements_aabb().closest_element_box(
+                query, distance_action_ );
+        }
+
+    private:
+        const DistanceToTetrahedron< dimension > distance_action_;
+    };
+    template < index_t dimension >
+    using TetrahedralSolidAABB =
+        GenericMeshAABB< TetrahedralSolid< dimension > >;
+    ALIAS_3D( TetrahedralSolidAABB );
 } // namespace geode

include/geode/mesh/helpers/aabb_surface_helpers.h

@@ -37,6 +37,8 @@ namespace geode
     FORWARD_DECLARATION_DIMENSION_CLASS( AABBTree );
     FORWARD_DECLARATION_DIMENSION_CLASS( SurfaceMesh );
     FORWARD_DECLARATION_DIMENSION_CLASS( TriangulatedSurface );
+    template < typename T >
+    class GenericMeshAABB;
 } // namespace geode
 
 namespace geode
@@ -61,4 +63,53 @@ namespace geode
         const TriangulatedSurface< dimension >& mesh_;
     };
     ALIAS_2D_AND_3D( DistanceToTriangle );
+
+    template < index_t dimension >
+    class GenericMeshAABB< SurfaceMesh< dimension > >
+    {
+    public:
+        GenericMeshAABB( const SurfaceMesh< dimension >& mesh )
+            : elements_tree_{ create_aabb_tree( mesh ) }
+        {
+        }
+
+    protected:
+        const AABBTree< dimension >& elements_aabb() const
+        {
+            return elements_tree_;
+        }
+
+    private:
+        AABBTree< dimension > elements_tree_;
+    };
+    template < index_t dimension >
+    using SurfaceMeshAABB = GenericMeshAABB< SurfaceMesh< dimension > >;
+    ALIAS_2D_AND_3D( SurfaceMeshAABB );
+
+    template < index_t dimension >
+    class GenericMeshAABB< TriangulatedSurface< dimension > >
+        : public GenericMeshAABB< SurfaceMesh< dimension > >
+    {
+    public:
+        using Base = GenericMeshAABB< SurfaceMesh< dimension > >;
+
+        GenericMeshAABB( const TriangulatedSurface< dimension >& mesh )
+            : Base{ mesh }, distance_action_{ mesh }
+        {
+        }
+
+        std::tuple< index_t, Point< dimension >, double > closest_element(
+            const Point< dimension >& query ) const
+        {
+            return this->elements_aabb().closest_element_box(
+                query, distance_action_ );
+        }
+
+    private:
+        const DistanceToTriangle< dimension > distance_action_;
+    };
+    template < index_t dimension >
+    using TriangulatedSurfaceAABB =
+        GenericMeshAABB< TriangulatedSurface< dimension > >;
+    ALIAS_2D_AND_3D( TriangulatedSurfaceAABB );
 } // namespace geode