1 files changed, 23 insertions, 26 deletions

diff --git a/klangfarbrs/src/line.rs b/klangfarbrs/src/line.rs
index f57b103..5f862a8 100644
--- a/klangfarbrs/src/line.rs
+++ b/klangfarbrs/src/line.rs
@@ -1,19 +1,25 @@
 use super::{Millisecond, Amplitude, SamplesPerSecond};
+use super::utils::*;
 
 pub struct Line {
-    pub start: Amplitude,
-    pub end: Amplitude,
-    pub duration: Millisecond,
-    pub index: u32,
+    index: u32,
+    samples: u32,
     slope: f32,
-    samples: u32
+    y_intercept: f32,
 }
 
 impl Line {
     pub fn new(
         start: Amplitude, end: Amplitude, duration: Millisecond, sample_rate: SamplesPerSecond
     ) -> Self {
-        Self { start, end, duration, index: 0, slope: slope(start, end, ms_to_samples(duration, sample_rate)), samples: ms_to_samples(duration, sample_rate) }
+        let number_of_samples = ms_to_samples(duration, sample_rate);
+
+        Self {
+            index: 0,
+            slope: slope(start, end, number_of_samples),
+            samples: number_of_samples,
+            y_intercept: start,
+        }
     }
 }
 
@@ -22,25 +28,16 @@ impl Iterator for Line {
 
     fn next(&mut self) -> Option<Self::Item> {
         let idx = self.index;
-        let val = self.slope * idx as f32 + self.start;
+        let val = self.slope * idx as f32 + self.y_intercept;
         self.index += 1;
- 
+
         if idx <= self.samples {
             Some(val)
         } else {
             None
         }
-          
-    }
-}
-
-fn slope(start: Amplitude, end: Amplitude, duration: Millisecond) -> f32 {
-  return (end - start) / duration as f32 ;
-}
 
-fn ms_to_samples(ms: Millisecond, sample_rate: SamplesPerSecond) -> u32 {
-    let multiplier = sample_rate as u32 / 1000;
-    multiplier * ms
+    }
 }
 
 #[cfg(test)]
@@ -51,18 +48,18 @@ mod tests {
     fn it_calculates_a_slope() {
       let expected = 0.5;
       let slope = slope(0.0, 0.5, 1);
-      assert_eq! (expected, slope)
+      assert_eq! (slope, expected)
     }
 
     #[test]
     fn it_calculates_the_next_values() {
       let mut line = Line::new(0.0, 0.5, 1, 5000.0);
-      assert_eq!(0.0, line.next().unwrap());
-      assert_eq!(0.1, line.next().unwrap());
-      assert_eq!(0.2, line.next().unwrap());
-      assert_eq!(0.3, line.next().unwrap());
-      assert_eq!(0.4, line.next().unwrap());
-      assert_eq!(0.5, line.next().unwrap());
-      assert_eq!(None, line.next());
+      assert_eq!(line.next(), Some(0.0));
+      assert_eq!(line.next(), Some(0.1));
+      assert_eq!(line.next(), Some(0.2));
+      assert_eq!(line.next(), Some(0.3));
+      assert_eq!(line.next(), Some(0.4));
+      assert_eq!(line.next(), Some(0.5));
+      assert_eq!(line.next(), None);
     }
 }