fix: Line now always includes points at start and end

shark-visualizer-interface/src/lib.rs

@@ -1,8 +1,8 @@
 use palette::{IntoColor, LinSrgb};
 use shark::point::Point;
-use shark::shader::{Fragment, Shader};
 #[cfg(target_arch = "wasm32")]
 use shark::shader::FragThree;
+use shark::shader::{Fragment, Shader};
 
 #[repr(C)]
 #[derive(Clone)]

shark-visualizer/src/shader_compiler.rs

@@ -11,7 +11,9 @@ use shark::shader::{FragThree, Fragment};
 use shark_visualizer_interface::VisualizationExports;
 
 use crate::{
-    ui::ErrorMessageEvent, user_config::{UserConfigState, RespawnLedsEvent}, visualization::VisualizationState,
+    ui::ErrorMessageEvent,
+    user_config::{RespawnLedsEvent, UserConfigState},
+    visualization::VisualizationState,
 };
 
 #[derive(Event)]

shark-visualizer/src/visualization.rs

@@ -3,9 +3,7 @@ pub use bevy::{prelude::*, window::CursorGrabMode};
 use shark::shader::{FragThree, Shader};
 
 use crate::{
-    shader_compiler::VisualizationExportsWrapper,
-    user_config::RespawnLedsEvent,
-    PlayBackState,
+    shader_compiler::VisualizationExportsWrapper, user_config::RespawnLedsEvent, PlayBackState,
 };
 
 pub struct VisualizationPlugin;
@@ -84,8 +82,7 @@ fn respawn_leds(
         for entity in leds.iter() {
             commands.entity(entity).despawn();
         }
-
-
+
         info!("Spawning LEDs");
         let led_root = root.single();
         for led in state.exports.as_ref().unwrap().points() {

shark/src/point/mod.rs

@@ -83,4 +83,4 @@ pub mod tests {
 
         let points: Vec<Point> = points.collect();
     }
-}
+}

shark/src/point/primitives.rs

@@ -21,8 +21,12 @@ impl Iterator for Line {
     type Item = Point;
 
     fn next(&mut self) -> Option<Self::Item> {
+        if self.num_points == 0 {
+            return None;
+        }
+
         if self.current_point < self.num_points {
-            let t = self.current_point as f64 / self.num_points as f64;
+            let t = self.current_point as f64 / (self.num_points - 1) as f64;
             self.current_point += 1;
             Some((1.0 - t) * self.a + t * self.b)
         } else {
@@ -33,4 +37,4 @@ impl Iterator for Line {
 
 pub fn line(a: Point, b: Point, num_points: usize) -> Line {
     Line::new(a, b, num_points)
-}
+}

shark/src/shader/mod.rs

@@ -2,11 +2,11 @@ pub mod primitives;
 
 use core::slice;
 
+use palette::{IntoColor, LinSrgb};
 use primitives::{
     checkerboard, extrude, mix, mod_position, mod_time, rotate_hue, Checkerboard, Extrude,
     Interpolate, ModPosition, ModTime, RotateHue,
 };
-use palette::{IntoColor, LinSrgb};
 
 pub trait Shader<F: Fragment> {
     type Output: IntoColor<LinSrgb<f64>>;