reorg

2024-12-07 20:16:47 -08:00
parent 21d9863a0f
commit 9728669d4e
45 changed files with 4279 additions and 1840 deletions
@@ -0,0 +1,61 @@
+#!/usr/bin/env python3
+# -*- coding: utf-8 -*-
+"""
+Author: C0ffymachyne
+License: GPLv3
+Version: 1.0.0
+
+Description:
+    Varios normalizatin techqniues node
+"""
+
+from ..core.io import audio_to_comfy_3d, audio_from_comfy_2d
+from ..core.loudness import rms_normalization, lufs_normalization, peak_normalization, automatic_gain_control
+
+class SignalProcessingNormalizer:
+    @classmethod
+    def INPUT_TYPES(cls):
+        return {
+            "required": {
+                "audio_input": ("AUDIO",),
+                "mode" : (["lufs","rms", "peak","auto"],),
+                "target_rms": ("FLOAT", {"default": 0.1, "min": 0, "max": 10.0, "step": 0.1}),
+                "target_lufs_db": ("FLOAT", {"default": -14.0, "min": -100, "max": 100.0, "step": 0.1}),
+                "target_peak": ("FLOAT", {"default": 0.9, "min": 0.0, "max": 1.0, "step": 0.1}),
+                "target_auto": ("FLOAT", {"default": 0.7, "min": 0.0, "max": 1.0, "step": 0.1}),
+                "target_auto_alpha": ("FLOAT", {"default": 0.1, "min": 0.0, "max": 10.0, "step": 0.1})
+            },
+        }
+
+    RETURN_TYPES = ("AUDIO",)
+    RETURN_NAMES = ("processed_audio",)
+    CATEGORY = "Audio Processing"
+    FUNCTION = "process"
+
+    def process(
+        self,
+        audio_input,
+        mode : str,
+        target_rms : float,
+        target_lufs_db : float,
+        target_peak : float,
+        target_auto : float,
+        target_auto_alpha : float
+        
+    ):
+
+        waveform, sample_rate = audio_from_comfy_2d(audio_input,try_gpu=True)
+
+        if mode == 'rms':
+            processed_waveform = rms_normalization(waveform,target_rms)
+        elif mode == 'lufs':
+            processed_waveform = lufs_normalization(waveform,sample_rate,target_lufs_db)
+        elif mode == 'peak':
+            processed_waveform = peak_normalization(waveform,target_peak)
+        elif mode == 'auto':
+            processed_waveform = automatic_gain_control(waveform,target_auto,target_auto_alpha)
+        else:
+            processed_waveform = waveform
+
+        return audio_to_comfy_3d(processed_waveform,sample_rate)
+