From 2c48e979c31f39e52365d0f7e30bf9ffaa582f01 Mon Sep 17 00:00:00 2001
From: Weird Constructor <weirdconstructor@gmail.com>
Date: Tue, 3 Aug 2021 06:21:04 +0200
Subject: [PATCH] Added function and selector to prevent selection of specific
 random NodeIds

---
 src/dsp/mod.rs | 48 ++++++++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 48 insertions(+)

diff --git a/src/dsp/mod.rs b/src/dsp/mod.rs
index c117d56..4a1c870 100644
--- a/src/dsp/mod.rs
+++ b/src/dsp/mod.rs
@@ -652,6 +652,54 @@ impl UICategory {
     }
 }
 
+#[derive(Debug, Clone, Copy)]
+pub enum RandNodeSelector {
+    Any,
+    OnlyUseful,
+}
+
+fn rand_node_satisfies_spec(nid: NodeId, sel: RandNodeSelector) -> bool {
+    if let NodeId::Nop = nid {
+        return false;
+    }
+
+    match sel {
+        RandNodeSelector::Any => true,
+        RandNodeSelector::OnlyUseful => {
+            match nid {
+                NodeId::Nop     => false,
+                NodeId::Out(_)  => false,
+                NodeId::FbRd(_) => false,
+                NodeId::Test(_) => false,
+                _ => true,
+            }
+        },
+    }
+}
+
+pub fn get_rand_node_id(count: usize, sel: RandNodeSelector) -> Vec<NodeId> {
+    let mut sm  = crate::dsp::helpers::SplitMix64::new_time_seed();
+    let mut out = vec![];
+
+    let mut cnt = 0;
+    while cnt < 100 && out.len() < count {
+        let cur = ALL_NODE_IDS[sm.next_u64() as usize % ALL_NODE_IDS.len()];
+
+        if rand_node_satisfies_spec(cur, sel) {
+            out.push(cur);
+        }
+
+        cnt += 1;
+    }
+
+    while out.len() < count {
+        out.push(NodeId::Nop);
+    }
+
+    out
+}
+
+
 macro_rules! make_node_info_enum {
     ($s1: ident => $v1: ident,
         $($str: ident => $variant: ident