fix(Mesh): allow graph identifier with a max number of vertices in clustering

Author: panquez

include/geode/mesh/helpers/detail/component_identifier.hpp

@@ -53,6 +53,8 @@ namespace geode
 
             index_t identify_vertices();
 
+            index_t identify_vertices( index_t nb_max_vertices );
+
             [[nodiscard]] index_t vertex_identifier( index_t vertex_id ) const;
 
             [[nodiscard]] absl::FixedArray< std::vector< index_t > >

src/geode/mesh/helpers/detail/component_identifier.cpp

@@ -27,6 +27,7 @@
 
 #include <geode/basic/attribute.hpp>
 #include <geode/basic/attribute_manager.hpp>
+#include <geode/basic/logger.hpp>
 #include <geode/basic/pimpl_impl.hpp>
 
 #include <geode/mesh/core/edged_curve.hpp>
@@ -78,12 +79,17 @@ namespace geode
             }
 
             index_t identify_elements()
+            {
+                return identify_elements( identification_.size() );
+            }
+
+            index_t identify_elements( index_t nb_max_elements )
             {
                 for( const auto p : Indices{ identification_ } )
                 {
                     if( identifier( p ) == NO_ID )
                     {
-                        propagate( p, nb_components_ );
+                        propagate( p, nb_components_, nb_max_elements );
                         nb_components_++;
                     }
                 }
@@ -114,11 +120,13 @@ namespace geode
             }
 
         private:
-            void propagate( index_t element_from, index_t tag_id )
+            void propagate(
+                index_t element_from, index_t tag_id, index_t nb_max_elements )
             {
+                index_t counter{ 0 };
                 std::queue< index_t > queue;
                 queue.emplace( element_from );
-                while( !queue.empty() )
+                while( !queue.empty() && counter < nb_max_elements )
                 {
                     const auto cur_el = queue.front();
                     queue.pop();
@@ -127,6 +135,7 @@ namespace geode
                         continue;
                     }
                     identification_[cur_el] = tag_id;
+                    counter++;
                     add_adjacents( cur_el, queue );
                 }
             }
@@ -176,6 +185,11 @@ namespace geode
             return impl_->identify_elements();
         }
 
+        index_t GraphIdentifier::identify_vertices( index_t nb_max_vertices )
+        {
+            return impl_->identify_elements( nb_max_vertices );
+        }
+
         index_t GraphIdentifier::vertex_identifier( index_t vertex_id ) const
         {
             return impl_->identifier( vertex_id );