hprtree 0.2.3

A Hilbert-Packed-R-Tree implementation for rust
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91

extern crate hprtree;

use std::sync::Mutex;

use actix_web::{
    post,
    web::{self, Data},
    App, HttpResponse, HttpServer, Responder,
};
use hprtree::{BBox, HPRTree, HPRTreeBuilder, SpatiallyIndexable};
use serde::{Deserialize, Serialize};

#[derive(Debug, Clone, Serialize, Deserialize)]
struct Element {
    pub x: f32,
    pub y: f32,
    pub data: String,
}

impl SpatiallyIndexable for Element {
    fn x(&self) -> hprtree::CoordinateType {
        self.x
    }

    fn y(&self) -> hprtree::CoordinateType {
        self.y
    }
}

#[derive(Debug, Clone, Serialize, Deserialize)]
pub struct FindBBox {
    pub minx: f32,
    pub miny: f32,
    pub maxx: f32,
    pub maxy: f32,
}

impl Into<BBox> for FindBBox {
    fn into(self) -> BBox {
        BBox {
            minx: self.minx,
            miny: self.miny,
            maxx: self.maxx,
            maxy: self.maxy,
        }
    }
}

#[post("/add")]
async fn add(
    index: Data<Mutex<HPRTree<Element>>>,
    index_builder: Data<Mutex<HPRTreeBuilder<Element>>>,
    data: web::Json<Element>,
) -> impl Responder {
    let mut index = index.lock().unwrap();
    let mut index_builder = index_builder.lock().unwrap();
    let data = data.into_inner();
    index_builder.insert(data);
    index_builder.sort_items();
    *index = index_builder.clone().build_sorted();

    HttpResponse::Ok()
}

#[post("/find")]
async fn find(index: Data<Mutex<HPRTree<Element>>>, data: web::Json<FindBBox>) -> impl Responder {
    let index = index.lock().unwrap();
    let data: BBox = data.into_inner().into();

    let res = index.query(&data);

    HttpResponse::Ok().json(res)
}

#[actix_web::main]
async fn main() -> std::io::Result<()> {
    let builder = HPRTreeBuilder::<Element>::new(16);
    let index_builder = Data::new(Mutex::new(builder.clone()));
    let index = Data::new(Mutex::new(builder.build()));

    HttpServer::new(move || {
        App::new()
            .app_data(index_builder.clone())
            .app_data(index.clone())
            .service(add)
            .service(find)
    })
    .bind(("localhost", 3030))?
    .run()
    .await
}