feat(NnSearch): add temporary possibility to find mapping in nn_search

Geode-solutions · Nov 28, 2024 · fd41b99 · fd41b99
1 parent 784f19d
commit fd41b99
Showing 1 changed file with 12 additions and 0 deletions.
diff --git a/include/geode/geometry/nn_search.hpp b/include/geode/geometry/nn_search.hpp
@@ -23,6 +23,7 @@
 
 #pragma once
 
+#include <geode/basic/mapping.hpp>
 #include <geode/basic/pimpl.hpp>
 
 #include <geode/geometry/common.hpp>
@@ -53,6 +54,17 @@ namespace geode
                 return colocated_mapping.size() - nb_unique_points();
             }
 
+            [[nodiscard]] geode::GenericMapping< geode::index_t >
+                points_mapping() const
+            {
+                geode::GenericMapping< geode::index_t > mapping;
+                for( const auto& p : geode::Indices{ colocated_mapping } )
+                {
+                    mapping.map( p, colocated_mapping[p] );
+                }
+                return mapping;
+            }
+
             std::vector< Point< dimension > > unique_points;
             /*!
              * This list has the size of the number of points in the tree.