From c9073bec568efa90c85e381c12dbcd20fe86a8f0 Mon Sep 17 00:00:00 2001 From: Justin Worthe Date: Mon, 5 Dec 2016 21:22:42 +0200 Subject: Refactored number traits --- src/sinusoid.rs | 16 ++++++++-------- 1 file changed, 8 insertions(+), 8 deletions(-) (limited to 'src/sinusoid.rs') diff --git a/src/sinusoid.rs b/src/sinusoid.rs index 1f60b03..cc9c20c 100644 --- a/src/sinusoid.rs +++ b/src/sinusoid.rs @@ -1,6 +1,5 @@ use std::cmp::{PartialOrd}; -use std::ops::{Add, Sub, Mul, Div, Rem}; -use ::num_traits::{Trig, Float}; +use ::num_traits::{Trig, ArithmeticOps, FractionOps}; // generic number type, but realistically it's only useful for // floats. Maybe also complex @@ -20,21 +19,22 @@ impl Sinusoid { } } } -impl Sinusoid where T: Float + Copy { +impl Sinusoid where T: FractionOps + Copy { pub fn period(&self) -> T { self.frequency.recip() } } -impl Sinusoid where T: Float + Add + Sub + Mul + Div + From + Trig + Copy + PartialOrd + Rem { +impl Sinusoid where T: FractionOps + ArithmeticOps + Trig + Copy { pub fn radial_frequency(&self) -> T { - (T::from(2))*T::pi()*self.frequency + T::two_pi()*self.frequency } pub fn sample(&self, t: T) -> T { (self.radial_frequency()*(t%self.period()) + self.phase).cos() * self.amplitude } +} +impl Sinusoid where T: FractionOps + ArithmeticOps + From + Trig + Copy + PartialOrd { //inclusive of start, exclusive of end pub fn sample_range(&self, start: T, end: T, sample_rate: T) -> Vec { - let sample_resolution = T::from(1) / sample_rate; let mut result = Vec::new(); let mut i: u16 = 0; loop { @@ -68,7 +68,7 @@ mod tests { #[test] fn sample() { - let sinusoid = Sinusoid::new(1.0 as f32, 1.0, -f32::consts::PI/2.0); //AKA sin + let sinusoid = Sinusoid::new(1.0 as f32, 1.0, -f32::consts::FRAC_PI_2); //AKA sin assert!((sinusoid.sample(0.0)-0.0) < f32::EPSILON); assert!((sinusoid.sample(0.25)-1.0) < f32::EPSILON); @@ -79,7 +79,7 @@ mod tests { #[test] fn sample_range() { - let sinusoid = Sinusoid::new(1.0 as f32, 1.0, -f32::consts::PI/2.0); //AKA sin + let sinusoid = Sinusoid::new(1.0 as f32, 1.0, -f32::consts::FRAC_PI_2); //AKA sin let samples = sinusoid.sample_range(0.0, 100.0, 4.0); println!("Epsilon is {}", f32::EPSILON); assert_eq!(samples.len(), 400); -- cgit v1.2.3