Attempt to weave in visualizer2 example to show volume waveform

Author: zachfeldman

inputmodule-control/Cargo.toml

@@ -11,3 +11,5 @@ serialport = "4.2.0"
 image = "0.24.5"
 rand = "0.8.5"
 chrono = "0.4.23"
+
+vis-core = { git = "https://github.com/Rahix/visualizer2.git" }

inputmodule-control/src/inputmodule.rs

@@ -242,6 +242,10 @@ pub fn serial_commands(args: &crate::ClapCli) {
                 random_eq_cmd(&serialdevs);
             }
 
+            if ledmatrix_args.input_eq {
+                input_eq_cmd(&serialdevs);
+            }
+
             if ledmatrix_args.clock {
                 clock_cmd(&serialdevs);
             }
@@ -636,6 +640,81 @@ fn random_eq_cmd(serialdevs: &Vec<String>) {
     }
 }
 
+
+// The data-type for storing analyzer results
+#[derive(Debug, Clone)]
+pub struct AnalyzerResult {
+    spectrum: vis_core::analyzer::Spectrum<Vec<f32>>,
+    volume: f32,
+    beat: f32,
+}
+
+// Equalizer-like animation that expands as volume goes up and retracts as it goes down
+fn input_eq_cmd(serialdevs: &Vec<String>) {
+    // Example from https://github.com/Rahix/visualizer2/blob/canon/README.md
+
+    // Initialize the logger.  Take a look at the sources if you want to customize
+    // the logger.
+    vis_core::default_log();
+
+    // Load the default config source.  More about config later on.  You can also
+    // do this manually if you have special requirements.
+    vis_core::default_config();
+
+    // Initialize some analyzer-tools.  These will be moved into the analyzer closure
+    // later on.
+    let mut analyzer = vis_core::analyzer::FourierBuilder::new()
+        .length(512)
+        .window(vis_core::analyzer::window::nuttall)
+        .plan();
+
+    let spectrum = vis_core::analyzer::Spectrum::new(vec![0.0; analyzer.buckets()], 0.0, 1.0);
+
+    let mut frames = vis_core::Visualizer::new(
+        AnalyzerResult {
+            spectrum,
+            volume: 0.0,
+            beat: 0.0,
+        },
+        // This closure is the "analyzer".  It will be executed in a loop to always
+        // have the latest data available.
+        move |info, samples| {
+            analyzer.analyze(samples);
+
+            info.spectrum.fill_from(&analyzer.average());
+            info.volume = samples.volume(0.3) * 400.0;
+            info.beat = info.spectrum.slice(50.0, 100.0).max() * 0.01;
+            info
+        },
+    )
+    // Build the frame iterator which is the base of your loop later on
+    .frames();
+
+    for frame in frames.iter() {
+        // This is just a primitive example, your vis core belongs here
+
+        frame.info(|info| {
+            for serialdev in serialdevs {
+                eq_cmd(
+                    serialdev,
+                    &[
+                        info.volume as u8,
+                        info.volume as u8,
+                        info.volume as u8,
+                        info.volume as u8,
+                        info.volume as u8,
+                        info.volume as u8,
+                        info.volume as u8,
+                        info.volume as u8,
+                        info.volume as u8,
+                    ],
+                )
+            }
+        });
+        thread::sleep(Duration::from_millis(30));
+    }
+}
+
 /// Display 9 values in equalizer diagram starting from the middle, going up and down
 /// TODO: Implement a commandline parameter for this
 fn eq_cmd(serialdev: &str, vals: &[u8]) {

inputmodule-control/src/ledmatrix.rs

@@ -86,6 +86,10 @@ pub struct LedMatrixSubcommand {
     #[arg(long)]
     pub random_eq: bool,
 
+    /// Input EQ
+    #[arg(long)]
+    pub input_eq: bool,
+
     /// EQ with custom values
     #[arg(long, num_args(9))]
     pub eq: Option<Vec<u8>>,