36 lines
1.1 KiB
Python
36 lines
1.1 KiB
Python
# Copyright (C) 2024 Reece H. Dunn. SPDX-License-Identifier: GPL-3
|
|
|
|
from .io_nodes import LoadAudio
|
|
from .plot_nodes import (PlotWaveform, PlotSpectrogram)
|
|
from .spect_nodes import (
|
|
Spectrogram, InverseSpectrogram,
|
|
GriffinLim,
|
|
MelSpectrogram
|
|
)
|
|
|
|
NODE_CLASS_MAPPINGS = {
|
|
# I/O
|
|
"ComfyAudio.LoadAudio": LoadAudio,
|
|
# Plotting
|
|
"ComfyAudio.PlotWaveform": PlotWaveform,
|
|
"ComfyAudio.PlotSpectrogram": PlotSpectrogram,
|
|
# Spectrogram
|
|
"ComfyAudio.Spectrogram": Spectrogram,
|
|
"ComfyAudio.InverseSpectrogram": InverseSpectrogram,
|
|
"ComfyAudio.GriffinLim": GriffinLim,
|
|
"ComfyAudio.MelSpectrogram": MelSpectrogram,
|
|
}
|
|
|
|
NODE_DISPLAY_NAME_MAPPINGS = {
|
|
# I/O
|
|
"ComfyAudio.LoadAudio": "Load Audio From Path",
|
|
# Plotting
|
|
"ComfyAudio.PlotWaveform": "Plot Waveform",
|
|
"ComfyAudio.PlotSpectrogram": "Plot Spectrogram",
|
|
# Spectrogram
|
|
"ComfyAudio.Spectrogram": "Spectrogram",
|
|
"ComfyAudio.InverseSpectrogram": "Inverse Spectrogram",
|
|
"ComfyAudio.GriffinLim": "Griffin Lim",
|
|
"ComfyAudio.MelSpectrogram": "Mel-scale Spectrogram",
|
|
}
|