Introduced geopoint.

examples/geo_json.rs

@@ -40,7 +40,16 @@ fn main() -> tantivy::Result<()> {
     let field = schema.get_field("geometry").unwrap();
     let query = SpatialQuery::new(
         field,
-        [GeoPoint { lon:-99.49, lat: 45.56}, GeoPoint {lon:-99.45, lat: 45.59}],
+        [
+            GeoPoint {
+                lon: -99.49,
+                lat: 45.56,
+            },
+            GeoPoint {
+                lon: -99.45,
+                lat: 45.59,
+            },
+        ],
         tantivy::query::SpatialQueryType::Intersects,
     );
     let hits = searcher.search(&query, &TopDocs::with_limit(10).order_by_score())?;

src/fastfield/mod.rs

@@ -699,18 +699,24 @@ mod tests {
         let index = Index::create_in_ram(schema);
         let mut index_writer = index.writer_for_tests().unwrap();
         index_writer.set_merge_policy(Box::new(NoMergePolicy));
-        index_writer.add_document(doc!(
-            date_field => DateTime::from_u64(1i64.to_u64()),
-            multi_date_field => DateTime::from_u64(2i64.to_u64()),
-            multi_date_field => DateTime::from_u64(3i64.to_u64())
-        )).unwrap();
-        index_writer.add_document(doc!(
-            date_field => DateTime::from_u64(4i64.to_u64())
-        )).unwrap();
-        index_writer.add_document(doc!(
-            multi_date_field => DateTime::from_u64(5i64.to_u64()),
-            multi_date_field => DateTime::from_u64(6i64.to_u64())
-        )).unwrap();
+        index_writer
+            .add_document(doc!(
+                date_field => DateTime::from_u64(1i64.to_u64()),
+                multi_date_field => DateTime::from_u64(2i64.to_u64()),
+                multi_date_field => DateTime::from_u64(3i64.to_u64())
+            ))
+            .unwrap();
+        index_writer
+            .add_document(doc!(
+                date_field => DateTime::from_u64(4i64.to_u64())
+            ))
+            .unwrap();
+        index_writer
+            .add_document(doc!(
+                multi_date_field => DateTime::from_u64(5i64.to_u64()),
+                multi_date_field => DateTime::from_u64(6i64.to_u64())
+            ))
+            .unwrap();
         index_writer.commit().unwrap();
         let reader = index.reader().unwrap();
         let searcher = reader.searcher();

src/index/segment_reader.rs

@@ -181,12 +181,11 @@ impl SegmentReader {
         let fieldnorm_data = segment.open_read(SegmentComponent::FieldNorms)?;
         let fieldnorm_readers = FieldNormReaders::open(fieldnorm_data)?;
         let spatial_readers = if schema.contains_spatial_field() {
-
             let spatial_data = segment.open_read(SegmentComponent::Spatial)?;
-                SpatialReaders::open(spatial_data)?
-            } else {
-                SpatialReaders::empty()
-            };
+            SpatialReaders::open(spatial_data)?
+        } else {
+            SpatialReaders::empty()
+        };
 
         let original_bitset = if segment.meta().has_deletes() {
             let alive_doc_file_slice = segment.open_read(SegmentComponent::Delete)?;

src/indexer/merger.rs

@@ -546,7 +546,7 @@ impl IndexMerger {
 
         let Some(mut spatial_serializer) = serializer.extract_spatial_serializer() else {
             // The schema does not contain any spatial field.
-            return Ok(())
+            return Ok(());
         };
 
         let mut segment_mappings: Vec<Vec<Option<DocId>>> = Vec::new();
@@ -592,19 +592,19 @@ impl IndexMerger {
                 // No need to fsync here. This file is not here for persistency.
             }
         }
-            for (field, temp_file) in temp_files {
-                // Memory map the triangle file.
-                use memmap2::MmapOptions;
-                let mmap = unsafe { MmapOptions::new().map_mut(temp_file.as_file())? };
-                // Cast to &[Triangle] slice
-                let triangle_count = mmap.len() / std::mem::size_of::<Triangle>();
-                let triangles = unsafe {
-                    std::slice::from_raw_parts_mut(mmap.as_ptr() as *mut Triangle, triangle_count)
-                };
-                // Get spatial writer and rebuild block kd-tree.
-                spatial_serializer.serialize_field(field, triangles)?;
-            }
-            spatial_serializer.close()?;
+        for (field, temp_file) in temp_files {
+            // Memory map the triangle file.
+            use memmap2::MmapOptions;
+            let mmap = unsafe { MmapOptions::new().map_mut(temp_file.as_file())? };
+            // Cast to &[Triangle] slice
+            let triangle_count = mmap.len() / std::mem::size_of::<Triangle>();
+            let triangles = unsafe {
+                std::slice::from_raw_parts_mut(mmap.as_ptr() as *mut Triangle, triangle_count)
+            };
+            // Get spatial writer and rebuild block kd-tree.
+            spatial_serializer.serialize_field(field, triangles)?;
+        }
+        spatial_serializer.close()?;
 
         Ok(())
     }

src/spatial/geometry.rs

@@ -189,7 +189,9 @@ impl Geometry {
                             polygon
                                 .iter()
                                 .map(|ring| {
-                                    Value::Array(ring.iter().map(|p| json!([p.lon, p.lat])).collect())
+                                    Value::Array(
+                                        ring.iter().map(|p| json!([p.lon, p.lat])).collect(),
+                                    )
                                 })
                                 .collect(),
                         )
@@ -400,10 +402,7 @@ impl BinarySerializable for Geometry {
     }
 }
 
-fn serialize_line_string<W: Write + ?Sized>(
-    line: &[GeoPoint],
-    writer: &mut W,
-) -> io::Result<()> {
+fn serialize_line_string<W: Write + ?Sized>(line: &[GeoPoint], writer: &mut W) -> io::Result<()> {
     BinarySerializable::serialize(&VInt(line.len() as u64), writer)?;
     let mut lon = Vec::new();
     let mut lat = Vec::new();
@@ -453,7 +452,10 @@ fn deserialize_line_string<R: Read>(reader: &mut R) -> io::Result<Vec<GeoPoint>>
     let lat: Vec<f64> = decompress_f64(&lat_bytes, point_count);
     let mut line_string: Vec<GeoPoint> = Vec::new();
     for offset in 0..point_count {
-        line_string.push(GeoPoint { lon: lon[offset], lat: lat[offset] });
+        line_string.push(GeoPoint {
+            lon: lon[offset],
+            lat: lat[offset],
+        });
     }
     Ok(line_string)
 }
@@ -476,7 +478,10 @@ fn deserialize_polygon<R: Read>(reader: &mut R) -> io::Result<Vec<Vec<GeoPoint>>
     for point_count in rings {
         let mut ring = Vec::new();
         for _ in 0..point_count {
-            ring.push(GeoPoint { lon: lon[offset], lat: lat[offset] });
+            ring.push(GeoPoint {
+                lon: lon[offset],
+                lat: lat[offset],
+            });
             offset += 1;
         }
         polygon.push(ring);

src/spatial/point.rs

@@ -1,4 +1,3 @@
-
 /// A point in the geographical coordinate system.
 #[derive(Debug, Clone, Copy, PartialEq)]
 pub struct GeoPoint {

src/spatial/writer.rs

@@ -95,11 +95,7 @@ fn append_point(triangles: &mut Vec<Triangle>, doc_id: DocId, point: GeoPoint) {
     triangles.push(Triangle::from_point(doc_id, point.0, point.1));
 }
 
-fn append_line_string(
-    triangles: &mut Vec<Triangle>,
-    doc_id: DocId,
-    line_string: Vec<GeoPoint>,
-) {
+fn append_line_string(triangles: &mut Vec<Triangle>, doc_id: DocId, line_string: Vec<GeoPoint>) {
     let mut previous = as_point_i32(line_string[0]);
     for point in line_string.into_iter().skip(1) {
         let point = as_point_i32(point);