rust/integral-geometry/src/lib.rs
changeset 14134 09f62bb046ef
parent 14133 a65b60f36b96
child 14135 7f5a591e1c43
--- a/rust/integral-geometry/src/lib.rs	Mon Nov 05 20:22:09 2018 +0300
+++ b/rust/integral-geometry/src/lib.rs	Mon Nov 05 21:21:53 2018 +0300
@@ -1,8 +1,11 @@
 extern crate fpnum;
 
 use fpnum::distance;
-use std::cmp::max;
-use std::ops::{Add, AddAssign, Div, DivAssign, Mul, MulAssign, Range, RangeInclusive, Sub, SubAssign};
+use std::{
+    cmp::max,
+    ops::{Add, AddAssign, Div, DivAssign, Mul, MulAssign, Range, RangeInclusive, Sub, SubAssign}
+};
+
 
 #[derive(PartialEq, Eq, Clone, Copy, Debug)]
 pub struct Point {
@@ -464,18 +467,16 @@
         (&self.vertices[..self.edges_count()]).iter()
     }
 
-    fn iter_mut<'a>(&'a mut self) -> impl Iterator<Item = &mut Point> + 'a {
+    pub fn iter_mut<'a>(&'a mut self) -> impl Iterator<Item = &mut Point> + 'a {
         let edges_count = self.edges_count();
-        (&mut self.vertices[..edges_count]).iter_mut()
+        let start = self.vertices.as_mut_ptr();
+        let end = unsafe { start.add(self.vertices.len()) };
+        PolygonPointsIteratorMut { source: self, start, end }
     }
 
-    pub fn for_each<F>(&mut self, f: F)
-        where F: (Fn(&mut Point))
-    {
+    fn force_close(&mut self) {
         if !self.vertices.is_empty() {
-            self.iter_mut().for_each(f);
-            let edges_count = self.edges_count();
-            self.vertices[edges_count] = self.vertices[0]
+            self.vertices[0] = self.vertices[self.vertices.len() - 1];
         }
     }
 
@@ -487,6 +488,34 @@
     }
 }
 
+struct PolygonPointsIteratorMut<'a> {
+    source: &'a mut Polygon,
+    start: *mut Point,
+    end: *mut Point
+}
+
+impl <'a> Iterator for PolygonPointsIteratorMut<'a> {
+    type Item = &'a mut Point;
+
+    fn next(&mut self) -> Option<<Self as Iterator>::Item> {
+        if self.start == self.end {
+            None
+        } else {
+            unsafe {
+                let result = &mut *self.start;
+                self.start = self.start.add(1);
+                Some(result)
+            }
+        }
+    }
+}
+
+impl <'a> Drop for PolygonPointsIteratorMut<'a> {
+    fn drop(&mut self) {
+        self.source.force_close();
+    }
+}
+
 impl From<Vec<Point>> for Polygon {
     fn from(mut v: Vec<Point>) -> Self {
         if !v.is_empty() && v[0] != v[v.len() - 1] {