bambam 0.3.1

The Behavior and Advanced Mobility Big Access Model
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

use geo::{Centroid, ConvexHull, Geometry, MultiPoint, Point};
use rstar::{primitives::PointWithData, PointDistance, RTreeObject, AABB};

/// An object stored in an opportunity spatial index which can be found
/// by a spatial query against its geometry and stores the lookup index
/// for the corresponding opportunity vector.
pub struct OpportunitySpatialRow {
    pub geometry: Geometry,
    pub index: usize,
}

impl OpportunitySpatialRow {
    /// creates a dummy query [`OpportunitySpatialRow`] object wrapping
    /// the query geometry for use with the [`rstar`] library.
    pub fn query(geometry: Geometry) -> OpportunitySpatialRow {
        OpportunitySpatialRow {
            geometry,
            index: 99999999999,
        }
    }
}

impl RTreeObject for OpportunitySpatialRow {
    type Envelope = AABB<Point>;

    fn envelope(&self) -> Self::Envelope {
        match &self.geometry {
            Geometry::Point(g) => g.envelope(),
            Geometry::Polygon(g) => g.envelope(),
            Geometry::MultiPolygon(g) => {
                let points = g.iter().flat_map(|i| i.centroid()).collect();
                let mp = MultiPoint(points);
                mp.convex_hull().envelope()
            },
            _ => panic!("opportunities can only be paired with POINT, POLYGON, or MULTIPOLYGON geometry types")
        }
    }
}

impl PointDistance for OpportunitySpatialRow {
    fn distance_2(
        &self,
        point: &<Self::Envelope as rstar::Envelope>::Point,
    ) -> <<Self::Envelope as rstar::Envelope>::Point as rstar::Point>::Scalar {
        match self.geometry.centroid() {
            Some(c) => c.distance_2(point),
            None => 999999999.0,
        }
    }
}