File size: 11,505 Bytes
1e4a2ab |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127 128 129 130 131 132 133 134 135 136 137 138 139 140 141 142 143 144 145 146 147 148 149 150 151 152 153 154 155 156 157 158 159 160 161 162 163 164 165 |
import os
import sys
import json
sys.path.append(os.getcwd())
from main.app.variables import translations, configs
from main.app.core.ui import gr_info, gr_warning, change_preset_choices, change_effect_preset_choices
def load_presets(presets, cleaner, autotune, pitch, clean_strength, index_strength, resample_sr, filter_radius, rms_mix_rate, protect, split_audio, f0_autotune_strength, formant_shifting, formant_qfrency, formant_timbre):
if not presets: gr_warning(translations["provide_file_settings"])
file = {}
if presets:
with open(os.path.join(configs["presets_path"], presets)) as f:
file = json.load(f)
gr_info(translations["load_presets"].format(presets=presets))
return [file.get("cleaner", cleaner), file.get("autotune", autotune), file.get("pitch", pitch), file.get("clean_strength", clean_strength), file.get("index_strength", index_strength), file.get("resample_sr", resample_sr), file.get("filter_radius", filter_radius), file.get("rms_mix_rate", rms_mix_rate), file.get("protect", protect), file.get("split_audio", split_audio), file.get("f0_autotune_strength", f0_autotune_strength), file.get("formant_shifting", formant_shifting), file.get("formant_qfrency", formant_qfrency), file.get("formant_timbre", formant_timbre)]
def save_presets(name, cleaner, autotune, pitch, clean_strength, index_strength, resample_sr, filter_radius, rms_mix_rate, protect, split_audio, f0_autotune_strength, cleaner_chbox, autotune_chbox, pitch_chbox, index_strength_chbox, resample_sr_chbox, filter_radius_chbox, rms_mix_rate_chbox, protect_chbox, split_audio_chbox, formant_shifting_chbox, formant_shifting, formant_qfrency, formant_timbre):
if not name: return gr_warning(translations["provide_filename_settings"])
if not any([cleaner_chbox, autotune_chbox, pitch_chbox, index_strength_chbox, resample_sr_chbox, filter_radius_chbox, rms_mix_rate_chbox, protect_chbox, split_audio_chbox, formant_shifting_chbox]): return gr_warning(translations["choose1"])
settings = {}
for checkbox, data in [(cleaner_chbox, {"cleaner": cleaner, "clean_strength": clean_strength}), (autotune_chbox, {"autotune": autotune, "f0_autotune_strength": f0_autotune_strength}), (pitch_chbox, {"pitch": pitch}), (index_strength_chbox, {"index_strength": index_strength}), (resample_sr_chbox, {"resample_sr": resample_sr}), (filter_radius_chbox, {"filter_radius": filter_radius}), (rms_mix_rate_chbox, {"rms_mix_rate": rms_mix_rate}), (protect_chbox, {"protect": protect}), (split_audio_chbox, {"split_audio": split_audio}), (formant_shifting_chbox, {"formant_shifting": formant_shifting, "formant_qfrency": formant_qfrency, "formant_timbre": formant_timbre})]:
if checkbox: settings.update(data)
with open(os.path.join(configs["presets_path"], name + ".conversion.json"), "w") as f:
json.dump(settings, f, indent=4)
gr_info(translations["export_settings"].format(name=name))
return change_preset_choices()
def audio_effect_load_presets(presets, resample_checkbox, audio_effect_resample_sr, chorus_depth, chorus_rate_hz, chorus_mix, chorus_centre_delay_ms, chorus_feedback, distortion_drive_db, reverb_room_size, reverb_damping, reverb_wet_level, reverb_dry_level, reverb_width, reverb_freeze_mode, pitch_shift_semitones, delay_second, delay_feedback, delay_mix, compressor_threshold_db, compressor_ratio, compressor_attack_ms, compressor_release_ms, limiter_threshold_db, limiter_release_ms, gain_db, bitcrush_bit_depth, clipping_threshold_db, phaser_rate_hz, phaser_depth, phaser_centre_frequency_hz, phaser_feedback, phaser_mix, bass_boost, bass_frequency, treble_boost, treble_frequency, fade_in, fade_out, chorus_check_box, distortion_checkbox, reverb_check_box, delay_check_box, compressor_check_box, limiter, gain_checkbox, bitcrush_checkbox, clipping_checkbox, phaser_check_box, bass_or_treble, fade):
if not presets: gr_warning(translations["provide_file_settings"])
file = {}
if presets:
with open(os.path.join(configs["presets_path"], presets)) as f:
file = json.load(f)
gr_info(translations["load_presets"].format(presets=presets))
return [
file.get("resample_checkbox", resample_checkbox), file.get("audio_effect_resample_sr", audio_effect_resample_sr),
file.get("chorus_depth", chorus_depth), file.get("chorus_rate_hz", chorus_rate_hz),
file.get("chorus_mix", chorus_mix), file.get("chorus_centre_delay_ms", chorus_centre_delay_ms),
file.get("chorus_feedback", chorus_feedback), file.get("distortion_drive_db", distortion_drive_db),
file.get("reverb_room_size", reverb_room_size), file.get("reverb_damping", reverb_damping),
file.get("reverb_wet_level", reverb_wet_level), file.get("reverb_dry_level", reverb_dry_level),
file.get("reverb_width", reverb_width), file.get("reverb_freeze_mode", reverb_freeze_mode),
file.get("pitch_shift_semitones", pitch_shift_semitones), file.get("delay_second", delay_second),
file.get("delay_feedback", delay_feedback), file.get("delay_mix", delay_mix),
file.get("compressor_threshold_db", compressor_threshold_db), file.get("compressor_ratio", compressor_ratio),
file.get("compressor_attack_ms", compressor_attack_ms), file.get("compressor_release_ms", compressor_release_ms),
file.get("limiter_threshold_db", limiter_threshold_db), file.get("limiter_release_ms", limiter_release_ms),
file.get("gain_db", gain_db), file.get("bitcrush_bit_depth", bitcrush_bit_depth),
file.get("clipping_threshold_db", clipping_threshold_db), file.get("phaser_rate_hz", phaser_rate_hz),
file.get("phaser_depth", phaser_depth), file.get("phaser_centre_frequency_hz", phaser_centre_frequency_hz),
file.get("phaser_feedback", phaser_feedback), file.get("phaser_mix", phaser_mix),
file.get("bass_boost", bass_boost), file.get("bass_frequency", bass_frequency),
file.get("treble_boost", treble_boost), file.get("treble_frequency", treble_frequency),
file.get("fade_in", fade_in), file.get("fade_out", fade_out),
file.get("chorus_check_box", chorus_check_box), file.get("distortion_checkbox", distortion_checkbox),
file.get("reverb_check_box", reverb_check_box), file.get("delay_check_box", delay_check_box),
file.get("compressor_check_box", compressor_check_box), file.get("limiter", limiter),
file.get("gain_checkbox", gain_checkbox), file.get("bitcrush_checkbox", bitcrush_checkbox),
file.get("clipping_checkbox", clipping_checkbox), file.get("phaser_check_box", phaser_check_box),
file.get("bass_or_treble", bass_or_treble), file.get("fade", fade)
]
def audio_effect_save_presets(name, resample_checkbox, audio_effect_resample_sr, chorus_depth, chorus_rate_hz, chorus_mix, chorus_centre_delay_ms, chorus_feedback, distortion_drive_db, reverb_room_size, reverb_damping, reverb_wet_level, reverb_dry_level, reverb_width, reverb_freeze_mode, pitch_shift_semitones, delay_second, delay_feedback, delay_mix, compressor_threshold_db, compressor_ratio, compressor_attack_ms, compressor_release_ms, limiter_threshold_db, limiter_release_ms, gain_db, bitcrush_bit_depth, clipping_threshold_db, phaser_rate_hz, phaser_depth, phaser_centre_frequency_hz, phaser_feedback, phaser_mix, bass_boost, bass_frequency, treble_boost, treble_frequency, fade_in, fade_out, chorus_check_box, distortion_checkbox, reverb_check_box, delay_check_box, compressor_check_box, limiter, gain_checkbox, bitcrush_checkbox, clipping_checkbox, phaser_check_box, bass_or_treble, fade):
if not name: return gr_warning(translations["provide_filename_settings"])
if not any([resample_checkbox, chorus_check_box, distortion_checkbox, reverb_check_box, delay_check_box, compressor_check_box, limiter, gain_checkbox, bitcrush_checkbox, clipping_checkbox, phaser_check_box, bass_or_treble, fade, pitch_shift_semitones != 0]): return gr_warning(translations["choose1"])
settings = {}
for checkbox, data in [
(resample_checkbox, {
"resample_checkbox": resample_checkbox,
"audio_effect_resample_sr": audio_effect_resample_sr
}),
(chorus_check_box, {
"chorus_check_box": chorus_check_box,
"chorus_depth": chorus_depth,
"chorus_rate_hz": chorus_rate_hz,
"chorus_mix": chorus_mix,
"chorus_centre_delay_ms": chorus_centre_delay_ms,
"chorus_feedback": chorus_feedback
}),
(distortion_checkbox, {
"distortion_checkbox": distortion_checkbox,
"distortion_drive_db": distortion_drive_db
}),
(reverb_check_box, {
"reverb_check_box": reverb_check_box,
"reverb_room_size": reverb_room_size,
"reverb_damping": reverb_damping,
"reverb_wet_level": reverb_wet_level,
"reverb_dry_level": reverb_dry_level,
"reverb_width": reverb_width,
"reverb_freeze_mode": reverb_freeze_mode
}),
(pitch_shift_semitones != 0, {
"pitch_shift_semitones": pitch_shift_semitones
}),
(delay_check_box, {
"delay_check_box": delay_check_box,
"delay_second": delay_second,
"delay_feedback": delay_feedback,
"delay_mix": delay_mix
}),
(compressor_check_box, {
"compressor_check_box": compressor_check_box,
"compressor_threshold_db": compressor_threshold_db,
"compressor_ratio": compressor_ratio,
"compressor_attack_ms": compressor_attack_ms,
"compressor_release_ms": compressor_release_ms
}),
(limiter, {
"limiter": limiter,
"limiter_threshold_db": limiter_threshold_db,
"limiter_release_ms": limiter_release_ms
}),
(gain_checkbox, {
"gain_checkbox": gain_checkbox,
"gain_db": gain_db
}),
(bitcrush_checkbox, {
"bitcrush_checkbox": bitcrush_checkbox,
"bitcrush_bit_depth": bitcrush_bit_depth
}),
(clipping_checkbox, {
"clipping_checkbox": clipping_checkbox,
"clipping_threshold_db": clipping_threshold_db
}),
(phaser_check_box, {
"phaser_check_box": phaser_check_box,
"phaser_rate_hz": phaser_rate_hz,
"phaser_depth": phaser_depth,
"phaser_centre_frequency_hz": phaser_centre_frequency_hz,
"phaser_feedback": phaser_feedback,
"phaser_mix": phaser_mix
}),
(bass_or_treble, {
"bass_or_treble": bass_or_treble,
"bass_boost": bass_boost,
"bass_frequency": bass_frequency,
"treble_boost": treble_boost,
"treble_frequency": treble_frequency
}),
(fade, {
"fade": fade,
"fade_in": fade_in,
"fade_out": fade_out
})
]:
if checkbox: settings.update(data)
with open(os.path.join(configs["presets_path"], name + ".effect.json"), "w") as f:
json.dump(settings, f, indent=4)
gr_info(translations["export_settings"].format(name=name))
return change_effect_preset_choices() |