rust/landgen/src/outline.rs
changeset 14139 09f62bb046ef
parent 14138 a65b60f36b96
child 14140 7f5a591e1c43
--- a/rust/landgen/src/outline.rs	Mon Nov 05 20:22:09 2018 +0300
+++ b/rust/landgen/src/outline.rs	Mon Nov 05 21:21:53 2018 +0300
@@ -54,11 +54,11 @@
             .chain(self.fill_points.iter())
     }
 
-    pub fn for_each<F>(&mut self, f: F)
-        where F: (Fn(&mut Point))
-    {
-        self.islands.iter_mut().for_each(|p| p.for_each(|x| f(x)));
-        self.fill_points.iter_mut().for_each(|p| f(p));
+    pub fn iter_mut(&mut self) -> impl Iterator<Item = &mut Point> {
+        self.islands
+            .iter_mut()
+            .flat_map(|i| i.iter_mut())
+            .chain(self.fill_points.iter_mut())
     }
 
     fn divide_edge<I: Iterator<Item = u32>>(
@@ -289,12 +289,14 @@
 
     pub fn mirror(&mut self) {
         let r = self.size.width as i32 - 1;
-        self.for_each(|p| p.x = r - p.x);
+
+        self.iter_mut().for_each(|p| p.x = r - p.x);
     }
 
     pub fn flip(&mut self) {
         let t = self.size.height as i32 - 1;
-        self.for_each(|p| p.y = t - p.y);
+
+        self.iter_mut().for_each(|p| p.y = t - p.y);
     }
 }
 
@@ -321,7 +323,7 @@
         Some(&Line::new(Point::new(20, 15), Point::new(10, 15)))
     );
 
-    points.for_each(|p| p.x = 2);
+    points.iter_mut().for_each(|p| p.x = 2);
 
     assert_eq!(points.fill_points[0].x, 2);
     assert_eq!(points.islands[0].get_edge(0).start.x, 2);