[SpatialPartioning] Expose operator() in KdTreeRangeQueryBase

poncateam · Oct 4, 2023 · a25da02 · a25da02
1 parent 1073d26
commit a25da02
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/Ponca/src/SpatialPartitioning/KdTree/Query/kdTreeRangeQueries.h b/Ponca/src/SpatialPartitioning/KdTree/Query/kdTreeRangeQueries.h
@@ -25,6 +25,7 @@ class KdTreeRangeQueryBase : public KdTreeQuery<Traits>, public QueryType
     using VectorType     = typename DataPoint::VectorType;
     using QueryAccelType = KdTreeQuery<Traits>;
     using Iterator       = IteratorType<IndexType, DataPoint, KdTreeRangeQueryBase>;
+    using Self           = KdTreeRangeQueryBase<Traits, IteratorType, QueryType>;
 
 protected:
     friend Iterator;
@@ -34,6 +35,11 @@ class KdTreeRangeQueryBase : public KdTreeQuery<Traits>, public QueryType
             KdTreeQuery<Traits>(kdtree), QueryType(radius, input){}
 
 public:
+    inline Self& operator()(typename QueryType::InputType input, Scalar radius)
+    { QueryAccelType::reset(); return QueryType::template operator()<Self>(input, radius); }
+    inline Self& operator()(typename QueryType::InputType input)
+    { QueryAccelType::reset(); return QueryType::template operator()<Self>(input); }
+
     inline Iterator begin(){
         QueryAccelType::reset();
         QueryType::reset();