add constant time exp avg

quartiq · Sep 28, 2023 · 03ef6dc · 03ef6dc
1 parent 9f3ad77
commit 03ef6dc
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 2 deletions.
diff --git a/src/bin/main.rs b/src/bin/main.rs
@@ -40,7 +40,7 @@ pub struct Opts {
     #[arg(short, long, default_value_t = 1.0f32)]
     fs: f32,
 
-    /// Exponential averaging
+    /// Exponential averaging alpha, negative for constant time, positive for constant noise
     #[arg(short, long)]
     avg: Option<f32>,
 }

diff --git a/src/psd.rs b/src/psd.rs
@@ -402,7 +402,14 @@ impl<const N: usize> PsdCascade<N> {
             stage.set_stage_depth(self.stage_length);
             stage.set_detrend(self.detrend);
             stage.set_overlap(self.overlap);
-            stage.set_avg(self.avg);
+
+            stage.set_avg(self.avg.map(|avg| {
+                if avg < 0.0 {
+                    (-avg * (1 << (self.stage_length * i)) as f32).min(1.0)
+                } else {
+                    avg
+                }
+            }));
             self.stages.push(stage);
         }
         &mut self.stages[i]