Fixed potential index out of bounds error

1 files changed, 2 insertions, 2 deletions

diff --git a/src/transforms.rs b/src/transforms.rs
index 8c60aad..ceb0bd3 100644
--- a/src/transforms.rs
+++ b/src/transforms.rs
@@ -87,10 +87,10 @@ fn score_guess(correlation: &[f32], period: f32, data_points: usize) -> f32 {
 }
 
 fn interpolate(correlation: &[f32], x: f32) -> f32 {
-    if x < 0.0 {
+    if x.floor() < 0.0 {
         correlation[0]
     }
-    else if x >= correlation.len() as f32 {
+    else if x.ceil() >= correlation.len() as f32 {
         correlation[correlation.len()-1]
     }
     else {