add utility functions

plot_utils.py → utils.py

@@ -1,7 +1,8 @@
-import numpy as np
 import torch
+import numpy as np
 from scipy.signal import freqz
 from typing import Iterable
+from functools import reduce
 
 from modules import fx
 from modules.functional import (
@@ -39,3 +40,33 @@ def get_log_mags_from_eq(eq: Iterable, worN=1024, sr=44100):
 
     log_mags = list(map(f, eq))
     return log_mags[0][0], [x for _, x in log_mags]
+
+
+jsonparse2hydra = lambda d: (
+    (
+        {"_target_": d["class_path"]}
+        | (
+            {k: jsonparse2hydra(v) for k, v in d["init_args"].items()}
+            if "init_args" in d
+            else {}
+        )
+        if "class_path" in d
+        else {k: jsonparse2hydra(v) for k, v in d.items()}
+    )
+    if isinstance(d, dict)
+    else (list(map(jsonparse2hydra, d)) if isinstance(d, list) else d)
+)
+
+remove_window_fn = lambda d: (
+    {k: remove_window_fn(v) for k, v in d.items() if k != "window_fn"}
+    if isinstance(d, dict)
+    else (list(map(remove_window_fn, d)) if isinstance(d, list) else d)
+)
+
+
+def chain_functions(*functions):
+    return lambda *initial_args: reduce(
+        lambda xs, f: f(*xs) if isinstance(xs, tuple) else f(xs),
+        functions,
+        initial_args,
+    )