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

// Copyright 2016 spatial-rs Developers
//
// Licensed under the Apache License, Version 2.0, <LICENSE-APACHE or
// http://apache.org/licenses/LICENSE-2.0> or the MIT license <LICENSE-MIT or
// http://opensource.org/licenses/MIT>, at your option. This file may not be
// copied, modified, or distributed except according to those terms.

use num::{Signed, Float, Bounded, ToPrimitive, FromPrimitive};
use std::ops::{MulAssign, AddAssign};
use geometry::Rect;
use tree::mbr::{MbrLeafGeometry, MbrLeaf, MbrNode};
use std::fmt::Debug;
use generic_array::ArrayLength;

/// Query trait for navigating the tree
pub trait MbrQuery<P, DIM, LG, T>
    where DIM: ArrayLength<P> + ArrayLength<(P, P)>
{
    /// Returns true if the leaf matches the query
    fn accept_leaf(&self, leaf: &MbrLeaf<P, DIM, LG, T>) -> bool;
    /// Returns true if the level matches the query
    fn accept_level(&self, level: &MbrNode<P, DIM, LG, T>) -> bool;
}

/// Rect based query
#[derive(Debug, Clone)]
pub enum MbrRectQuery<P, DIM>
    where DIM: ArrayLength<P> + ArrayLength<(P, P)>
{
    /// Matching leaves are ones that are completely contained by this rect
    ContainedBy(Rect<P, DIM>),
    /// Matching leaves are ones that overlap this rect
    Overlaps(Rect<P, DIM>),
}

impl<P, DIM, LG, T> MbrQuery<P, DIM, LG, T> for MbrRectQuery<P, DIM>
    where P: Float + Signed + Bounded + MulAssign + AddAssign + ToPrimitive + FromPrimitive + Copy + Debug + Default,
          DIM: ArrayLength<P> + ArrayLength<(P,P)>,
          LG: MbrLeafGeometry<P, DIM>
{

// Does this query accept the given leaf?
    fn accept_leaf(&self, leaf: &MbrLeaf<P, DIM, LG, T>) -> bool {
        match *self {
            MbrRectQuery::ContainedBy(ref query) => leaf.geometry.contained_by_mbr(query),
            MbrRectQuery::Overlaps(ref query) => leaf.geometry.overlapped_by_mbr(query),
        }
    }

// Does this query accept the given level?
    fn accept_level(&self, level: &MbrNode<P, DIM, LG, T>) -> bool {
        match *self {
            MbrRectQuery::ContainedBy(ref query) => level.overlapped_by_mbr(query),
            MbrRectQuery::Overlaps(ref query) => level.overlapped_by_mbr(query),
        }
    }
}