Kontinuation commented on code in PR #523:
URL: https://github.com/apache/sedona-db/pull/523#discussion_r2699012938
##########
rust/sedona-spatial-join/src/index/spatial_index.rs:
##########
@@ -409,6 +416,177 @@ impl SpatialIndex {
})
}
+ /// Query the spatial index with a batch of probe geometries to find
matching build-side geometries.
+ ///
+ /// This method iterates over the probe geometries in the given range of
the evaluated batch.
+ /// For each probe geometry, it performs the two-phase spatial join query:
+ /// 1. **Filter phase**: Uses the R-tree index with the probe geometry's
bounding rectangle
+ /// to quickly identify candidate geometries.
+ /// 2. **Refinement phase**: Evaluates the exact spatial predicate on
candidates to determine
+ /// actual matches.
+ ///
+ /// # Arguments
+ /// * `evaluated_batch` - The batch containing probe geometries and their
bounding rectangles
+ /// * `range` - The range of rows in the evaluated batch to process
+ /// * `max_result_size` - The maximum number of results to collect before
stopping. If the
+ /// number of results exceeds this limit, the method returns early.
+ /// * `build_batch_positions` - Output vector that will be populated with
(batch_idx, row_idx)
+ /// pairs for each matching build-side geometry
+ /// * `probe_indices` - Output vector that will be populated with the
indices of probe geometries
+ /// that have matches each probe geometry processed
+ ///
+ /// # Returns
+ /// * A tuple containing:
+ /// - `QueryResultMetrics`: Aggregated metrics (total matches and
candidates) for the processed rows
+ /// - `usize`: The index of the next row to process (exclusive end of
the processed range)
+ pub(crate) async fn query_batch(
+ self: &Arc<Self>,
+ evaluated_batch: &Arc<EvaluatedBatch>,
+ range: Range<usize>,
+ max_result_size: usize,
+ build_batch_positions: &mut Vec<(i32, i32)>,
+ probe_indices: &mut Vec<u32>,
+ ) -> Result<(QueryResultMetrics, usize)> {
Review Comment:
Improved documentation for `probe_indices`.
--
This is an automated message from the Apache Git Service.
To respond to the message, please log on to GitHub and use the
URL above to go to the specific comment.
To unsubscribe, e-mail: [email protected]
For queries about this service, please contact Infrastructure at:
[email protected]
