From d63396b78e55ec6f49683bf9132d97348cc5ac02 Mon Sep 17 00:00:00 2001
From: Leo Le Bouter <lle-bout@zaclys.net>
Date: Wed, 7 Oct 2020 12:02:20 +0200
Subject: [PATCH] Make InterpolationConfig configurable from command line

---
 src/main.rs | 65 ++++++++++++++++++++++++++++++++++++++++-------------
 1 file changed, 50 insertions(+), 15 deletions(-)

diff --git a/src/main.rs b/src/main.rs
index 2d102e0..80fa01d 100644
--- a/src/main.rs
+++ b/src/main.rs
@@ -99,7 +99,11 @@ impl Lerp for Point {
     }
 }
 
-fn optimize_line(default_weight: u32, line: &str) -> Result<Vec<InputArrayEntry>> {
+fn optimize_line(
+    default_weight: u32,
+    line: &str,
+    config: &InterpolationConfig,
+) -> Result<Vec<InputArrayEntry>> {
     let frame: Vec<InputArrayEntry> = serde_json::from_str(line)?;
 
     let input_points: Vec<Point> = frame
@@ -116,13 +120,8 @@ fn optimize_line(default_weight: u32, line: &str) -> Result<Vec<InputArrayEntry>
     let pg = segments_to_point_graph(&input_points, segs);
     let eg = point_graph_to_euler_graph(&pg);
     let ec = euler_graph_to_euler_circuit(&input_points, &eg);
-    let output_points: Vec<Point> = interpolate_euler_circuit(
-        &input_points,
-        &ec,
-        &eg,
-        input_points.len() as u32,
-        &InterpolationConfig::default(),
-    );
+    let output_points: Vec<Point> =
+        interpolate_euler_circuit(&input_points, &ec, &eg, input_points.len() as u32, config);
 
     let output_frame: Vec<_> = output_points.iter().map(|p| (p.x, p.y, p.color)).collect();
 
@@ -131,20 +130,56 @@ fn optimize_line(default_weight: u32, line: &str) -> Result<Vec<InputArrayEntry>
 
 fn main() -> Result<()> {
     let matches = app_from_crate!()
-        .args(&[Arg::with_name("default-weight")
-            .short("w")
-            .takes_value(true)
-            .multiple(false)
-            .default_value("0")])
+        .args(&[
+            Arg::with_name("default-weight")
+                .short("w")
+                .takes_value(true)
+                .multiple(false)
+                .default_value("0"),
+            Arg::with_name("distance-per-point")
+                .short("d")
+                .takes_value(true)
+                .multiple(false),
+            Arg::with_name("blank-delay-points")
+                .short("b")
+                .takes_value(true)
+                .multiple(false),
+            Arg::with_name("radians-per-point")
+                .short("r")
+                .takes_value(true)
+                .multiple(false),
+        ])
         .get_matches();
 
     let default_weight = matches.value_of("default-weight").unwrap().parse().unwrap();
+    let mut config = InterpolationConfig::default();
+
+    if let Some(distance_per_point) = matches.value_of("distance-per-point") {
+        config.distance_per_point = distance_per_point
+            .parse()
+            .expect("distance-per-point must be a float")
+    }
+
+    if let Some(blank_delay_points) = matches.value_of("blank-delay-points") {
+        config.blank_delay_points = blank_delay_points
+            .parse()
+            .expect("blank-delay-points must be an unsigned integer")
+    }
+
+    if let Some(radians_per_point) = matches.value_of("radians-per-point") {
+        config.radians_per_point = radians_per_point
+            .parse()
+            .expect("radians-per-point must be a float")
+    }
 
     loop {
         let mut line = String::new();
         io::stdin().read_line(&mut line)?;
 
-        serde_json::to_writer(io::stdout(), &optimize_line(default_weight, &line)?)?;
+        serde_json::to_writer(
+            io::stdout(),
+            &optimize_line(default_weight, &line, &config)?,
+        )?;
         io::stdout().write(b"\n")?;
         io::stdout().flush()?;
     }
@@ -152,5 +187,5 @@ fn main() -> Result<()> {
 
 #[test]
 fn optimize_line_test() {
-    optimize_line(0, "[[100.0, 100.0, 65280], [100.0, 500.0, 65280], [500.0, 500.0, 65280], [500.0, 100.0, 65280], [100.0, 100.0, 65280]]").unwrap();
+    optimize_line(0, "[[100.0, 100.0, 65280], [100.0, 500.0, 65280], [500.0, 500.0, 65280], [500.0, 100.0, 65280], [100.0, 100.0, 65280]]", &InterpolationConfig::default()).unwrap();
 }