visioncortex 0.9.2

Semantic Computer Vision
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
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153

use std::fmt::Debug;

use crate::PointF64;

use super::Orientation;

/// Common properties/methods for any polygons.
///
/// To be composed into any structs that represents a polygon
#[derive(Clone, Default)]
pub struct PolygonProps {
    pub points: Vec<PointF64>,
    pub is_hole: bool,
}

impl Debug for PolygonProps {
    fn fmt(&self, f: &mut std::fmt::Formatter<'_>) -> std::fmt::Result {
        let mut formatted = vec![(if self.is_hole {"HOLE"} else {"SOLID"}).to_string()];

        for point in self.points.iter() {
            formatted.push(format!("({}, {})", point.x, point.y));
        }

        f.write_str(&("<".to_owned() + &formatted.join(", ") + ">\n"))
    }
}

pub trait PolygonInterface {
    fn props(&self) -> &PolygonProps;
    fn props_mut(&mut self) -> &mut PolygonProps;

    fn num_points(&self) -> usize {
        self.props().num_points()
    }

    fn get_point(&self, i: usize) -> PointF64 {
        self.props().points[i]
    }

    fn set_point(&mut self, i: usize, p: PointF64) {
        self.props_mut().points[i] = p;
    }

    fn is_hole(&self) -> bool {
        self.props().is_hole
    }
}

#[derive(Clone, Debug, Default)]
// The most basic and generic polygon class
pub struct Polygon {
    props: PolygonProps,
}

impl PolygonInterface for Polygon {
    fn props(&self) -> &PolygonProps {
        &self.props
    }

    fn props_mut(&mut self) -> &mut PolygonProps {
        &mut self.props
    }
}

impl Polygon {
    pub fn from_points_and_is_hole(points: Vec<PointF64>, is_hole: bool) -> Self {
        Self {
            props: PolygonProps::from_points_and_is_hole(points, is_hole)
        }
    }

    /// Create a triangle with the default props
    pub fn triangle(p1: PointF64, p2: PointF64, p3: PointF64) -> Self {
        let mut triangle = Self::default();
        triangle.props.points = vec![p1, p2, p3];
        triangle
    }

    pub fn is_valid(&self) -> bool {
        self.props.is_valid()
    }
}

impl PolygonProps {
    pub fn from_points_and_is_hole(points: Vec<PointF64>, is_hole: bool) -> Self {
        Self {
            points,
            is_hole
        }
    }

    pub fn num_points(&self) -> usize {
        self.points.len()
    }

    pub fn get_point_safe(&self, i: usize) -> Option<PointF64> {
        if i < self.num_points() {
            Some(self.points[i])
        } else {
            None
        }
    }

    pub fn clear(&mut self) {
        self.points.clear()
    }

    /// Invert the order of vertices
    pub fn invert(&mut self) {
        self.points.reverse()
    }

    pub fn get_orientation(&self) -> Orientation {
        let mut area = 0.0;
        let len = self.num_points();
        for curr in 0..len {
            let next = (curr+1) % len;
            let (curr_pt, next_pt) = (self.points[curr], self.points[next]);
            area += curr_pt.x * next_pt.y - curr_pt.y * next_pt.x;
        }

        if area.is_sign_positive() {
            Orientation::CounterClockwise
        } else if area.is_sign_negative() {
            Orientation::Clockwise
        } else {
            Orientation::None
        }
    }

    pub fn set_orientation(&mut self, orientation: Orientation) {
        let poly_orientation = self.get_orientation();
        if poly_orientation != Orientation::None && poly_orientation != orientation {
            self.invert();
        }
    }

    /// A valid PolygonProps should have at least 3 points
    pub fn is_valid(&self) -> bool {
        self.num_points() >= 3
    }

    pub fn dump(&self, decimal: bool) -> String {
        let mut dump = vec![self.num_points().to_string(), (self.is_hole as i32).to_string()];
        for p in self.points.iter() {
            dump.push(format!("{} {}",
                if decimal {p.x.to_string()} else {(p.x as i32).to_string()},
                if decimal {p.y.to_string()} else {(p.y as i32).to_string()}
            ));
        }
        dump.join("\n")
    }
}