From ca2fbaa6716b3cb5bb789e1669dfd92e498ff304 Mon Sep 17 00:00:00 2001
From: Weird Constructor <weirdconstructor@gmail.com>
Date: Wed, 16 Jun 2021 19:54:20 +0200
Subject: [PATCH] wrote a test for frequency cleanliness

---
 tests/common/mod.rs | 10 +++++++
 tests/node_ad.rs    | 66 +++++++++++++++++++++++++++++++++++++++++++++
 2 files changed, 76 insertions(+)

diff --git a/tests/common/mod.rs b/tests/common/mod.rs
index f145238..819733a 100644
--- a/tests/common/mod.rs
+++ b/tests/common/mod.rs
@@ -334,6 +334,16 @@ pub fn run_and_get_fft4096(
     fft_thres_at_ms(&mut out_l[..], FFT::F4096, thres, offs_ms)
 }
 
+pub fn run_and_get_fft4096_now(
+    node_exec: &mut hexodsp::nodes::NodeExecutor,
+    thres: u32) -> Vec<(u16, u32)>
+{
+    let min_samples_for_fft = 4096.0 * 1.5; // 1.5 for some extra margin
+    let run_len_s = min_samples_for_fft / SAMPLE_RATE;
+    let (mut out_l, _out_r) = run_no_input(node_exec, run_len_s);
+    fft_thres_at_ms(&mut out_l[..], FFT::F4096, thres, 0.0)
+}
+
 #[allow(unused)]
 pub enum FFT {
     F16,
diff --git a/tests/node_ad.rs b/tests/node_ad.rs
index 99211ce..873133e 100644
--- a/tests/node_ad.rs
+++ b/tests/node_ad.rs
@@ -130,3 +130,69 @@ fn check_node_ad_retrig() {
         -0.002267599, -0.002267599, -0.0022675395, -0.002267599, -0.0022675395
     ]);
 }
+
+#[test]
+fn check_node_ad_inp_sin() {
+    let (node_conf, mut node_exec) = new_node_engine();
+    let mut matrix = Matrix::new(node_conf, 3, 3);
+
+    let sin = NodeId::Sin(0);
+    let ad  = NodeId::Ad(0);
+    let out = NodeId::Out(0);
+    matrix.place(0, 0, Cell::empty(sin)
+                       .out(None, None, sin.out("sig")));
+    matrix.place(0, 1, Cell::empty(ad)
+                       .input(ad.inp("inp"), None, None)
+                       .out(None, None, ad.out("sig")));
+    matrix.place(0, 2, Cell::empty(out)
+                       .input(out.inp("ch1"), None, None));
+    matrix.sync().unwrap();
+
+    let trig_p = ad.inp_param("trig").unwrap();
+    let atk_p  = ad.inp_param("atk").unwrap();
+    let dcy_p  = ad.inp_param("dcy").unwrap();
+
+    // check if we have any frequencies resembling 440Hz
+    matrix.set_param(trig_p, SAtom::param(1.0));
+    let res = run_for_ms(&mut node_exec, 7.0);
+
+    let fft = run_and_get_fft4096_now(&mut node_exec, 6);
+    assert_eq!(fft[0], (420, 6));
+    assert_eq!(fft[1], (431, 6));
+    assert_eq!(fft[2], (441, 6));
+    assert_eq!(fft[3], (452, 6));
+    assert_eq!(fft[4], (463, 6));
+
+    // Next we test if lengthening the attack has
+    // effect on the captured frequencies.
+    matrix.set_param(trig_p, SAtom::param(0.0));
+    run_for_ms(&mut node_exec, 8.0);
+
+    matrix.set_param(atk_p, SAtom::param(atk_p.norm(40.0)));
+    matrix.set_param(trig_p, SAtom::param(1.0));
+    let res = run_for_ms(&mut node_exec, 7.0);
+
+    let fft = run_and_get_fft4096_now(&mut node_exec, 300);
+    assert_eq!(fft[0], (431, 322));
+    assert_eq!(fft[1], (441, 360));
+
+    matrix.set_param(trig_p, SAtom::param(0.0));
+    run_for_ms(&mut node_exec, 8.0);
+
+    // Next we test if lengthening the decay too has
+    // effect on the captured frequencies.
+    matrix.set_param(trig_p, SAtom::param(0.0));
+    run_for_ms(&mut node_exec, 8.0);
+
+    matrix.set_param(dcy_p, SAtom::param(dcy_p.norm(40.0)));
+    matrix.set_param(trig_p, SAtom::param(1.0));
+    let res = run_for_ms(&mut node_exec, 7.0);
+
+    let fft = run_and_get_fft4096_now(&mut node_exec, 300);
+    assert_eq!(fft[0], (431, 489));
+    assert_eq!(fft[1], (441, 647));
+    assert_eq!(fft[2], (452, 398));
+
+    matrix.set_param(trig_p, SAtom::param(0.0));
+    run_for_ms(&mut node_exec, 8.0);
+}