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

use std::cmp::Ordering;

#[derive(Copy, Clone, Debug, Eq)]
pub struct Range {
    pub begin: usize,
    pub end: usize,
}

impl Range {
    pub fn new(begin: usize, end: usize) -> Self {
        assert!(begin <= end);
        Range { begin, end }
    }

    pub fn offset(&mut self, offset: usize) {
        self.begin += offset;
        self.end += offset;
    }
}

impl PartialEq for Range {
    fn eq(&self, other: &Self) -> bool {
        if self.begin <= other.begin {
            other.begin < self.end
        } else {
            self.begin < other.end
        }
    }
}

impl Ord for Range {
    fn cmp(&self, other: &Self) -> Ordering {
        if self.eq(other) {
            Ordering::Equal
        } else {
            self.begin.cmp(&other.begin)
        }
    }
}

impl PartialOrd for Range {
    fn partial_cmp(&self, other: &Self) -> Option<Ordering> {
        Some(self.cmp(other))
    }
}

#[cfg(test)]
mod tests {
    use super::*;
    use std::collections::BTreeMap;

    #[test]
    fn test_btreemap() {
        let mut map = BTreeMap::new();
        map.insert(Range::new(0, 10), String::from("0-10"));
        map.insert(Range::new(10, 15), String::from("10-15"));
        assert_eq!(map.get(&Range::new(0, 1)), Some(&String::from("0-10")));
        assert_eq!(map.get(&Range::new(3, 4)), Some(&String::from("0-10")));
        assert_eq!(map.get(&Range::new(10, 11)), Some(&String::from("10-15")));
        assert_eq!(map.get(&Range::new(15, 16)), None);
    }
}