Merge pull request #1114 from Geode-solutions/feat/anisotropic_distance_to_tetrahedron

Feat/anisotropic_distance_to_tetrahedron

Author: MelchiorSchuh

include/geode/mesh/builder/hybrid_solid_builder.hpp

@@ -58,7 +58,7 @@ namespace geode
         index_t create_tetrahedron( const std::array< index_t, 4 >& vertices );
 
         /*!
-         * Create a new hexahedron from four vertices.
+         * Create a new hexahedron from eight vertices.
          * @param[in] vertices The eight vertices defining the hexahedron to
          * create
          * @return the index of the created hexahedron

include/geode/mesh/helpers/aabb_solid_helpers.hpp

@@ -31,6 +31,8 @@ namespace geode
     FORWARD_DECLARATION_DIMENSION_CLASS( AABBTree );
     FORWARD_DECLARATION_DIMENSION_CLASS( SolidMesh );
     FORWARD_DECLARATION_DIMENSION_CLASS( TetrahedralSolid );
+    FORWARD_DECLARATION_DIMENSION_CLASS( CoordinateSystem );
+    ALIAS_3D( CoordinateSystem );
     template < typename T >
     class GenericMeshAABB;
 } // namespace geode
@@ -59,6 +61,26 @@ namespace geode
     };
     ALIAS_3D( DistanceToTetrahedron );
 
+    template < index_t dimension >
+    class AnisotropicDistanceToTetrahedron
+    {
+    public:
+        explicit AnisotropicDistanceToTetrahedron(
+            const TetrahedralSolid< dimension >& mesh,
+            const CoordinateSystem< dimension >& coordinate_system )
+            : mesh_( mesh ), coordinate_system_( coordinate_system )
+        {
+        }
+
+        [[nodiscard]] double operator()(
+            const Point< dimension >& query, index_t cur_box ) const;
+
+    private:
+        const TetrahedralSolid< dimension >& mesh_;
+        const CoordinateSystem< dimension >& coordinate_system_;
+    };
+    ALIAS_3D( AnisotropicDistanceToTetrahedron );
+
     template < index_t dimension >
     class GenericMeshAABB< SolidMesh< dimension > >
     {

src/geode/mesh/helpers/aabb_solid_helpers.cpp

@@ -60,9 +60,31 @@ namespace geode
             point_tetrahedron_distance( query, mesh_.tetrahedron( cur_box ) ) );
     }
 
+    template < index_t dimension >
+    double AnisotropicDistanceToTetrahedron< dimension >::operator()(
+        const Point< dimension >& query, index_t cur_box ) const
+    {
+        const auto tetrahedron_vertices =
+            mesh_.tetrahedron( cur_box ).vertices();
+        const auto p0 =
+            coordinate_system_.coordinates( tetrahedron_vertices[0] );
+        const auto p1 =
+            coordinate_system_.coordinates( tetrahedron_vertices[1] );
+        const auto p2 =
+            coordinate_system_.coordinates( tetrahedron_vertices[2] );
+        const auto p3 =
+            coordinate_system_.coordinates( tetrahedron_vertices[3] );
+        const Tetrahedron tetrahedron_in_metric_space{ p0, p1, p2, p3 };
+        const auto query_in_metric_space =
+            coordinate_system_.coordinates( query );
+        return std::get< 0 >( point_tetrahedron_distance(
+            query_in_metric_space, tetrahedron_in_metric_space ) );
+    }
+
     template opengeode_mesh_api AABBTree3D create_aabb_tree< 3 >(
         const SolidMesh3D& );
 
     template class opengeode_mesh_api DistanceToTetrahedron< 3 >;
+    template class opengeode_mesh_api AnisotropicDistanceToTetrahedron< 3 >;
 
 } // namespace geode