Rust version of IFS

contents/IFS/IFS.md

@@ -136,6 +136,8 @@ Here, instead of tracking children of children, we track a single individual tha
 [import:18-29, lang:"c"](code/c/IFS.c)
 {% sample lang="coco" %}
 [import:4-16, lang:"coconut"](code/coconut/IFS.coco)
+{% sample lang="rust" %}
+[import:9-20, lang:"rust"](code/rust/IFS.rs)
 {% endmethod %}
 
 If we set the initial points to the on the equilateral triangle we saw before, we can see the Sierpinski triangle again after a few thousand iterations, as shown below:
@@ -207,6 +209,8 @@ In addition, we have written the chaos game code to take in a set of points so t
 [import, lang:"c"](code/c/IFS.c)
 {%sample lang="coco" %}
 [import, lang:"coconut"](code/coconut/IFS.coco)
+{%sample lang="rust" %}
+[import, lang:"rust"](code/rust/IFS.rs)
 {% endmethod %}
 
 ### Bibliography

contents/IFS/code/rust/IFS.rs

@@ -0,0 +1,36 @@
+use rand::prelude::*;
+
+#[derive(Clone, Copy)]
+struct Point {
+    x: f64,
+    y: f64,
+}
+
+fn chaos_game(iters: usize, shapes: Vec<Point>) -> Vec<Point> {
+    let mut rng = rand::thread_rng();
+    let mut p   = Point{x: rng.gen(), y: rng.gen()};
+
+    (0..iters).into_iter().map(|_| {
+        let old_point = p;
+        let tmp = shapes[rng.gen_range(0, shapes.len())];
+        p.x = 0.5 * (p.x + tmp.x);
+        p.y = 0.5 * (p.y + tmp.y);
+        old_point
+    }).collect()
+}
+
+fn main() {
+    let shapes = vec![
+        Point{x: 0., y: 0.},
+        Point{x: 0.5, y: (0.75 as f64).sqrt()},
+        Point{x: 1., y: 0.},
+    ];
+
+    let mut out = String::new();
+
+    for point in chaos_game(10_000, shapes) {
+        out += format!("{}\t{}\n", point.x, point.y).as_str();
+    }
+
+    std::fs::write("./out.dat", out).unwrap();
+}