mirror of
https://github.com/Dvorinka/swingmusic-extended.git
synced 2026-06-03 20:13:02 +00:00
cwilvx
73 lines
2.0 KiB
Python
73 lines
2.0 KiB
Python
"""
|
|
This library contains all the functions related to tracks.
|
|
"""
|
|
|
|
import os
|
|
|
|
from app.lib.pydub.pydub import AudioSegment
|
|
from app.lib.pydub.pydub.silence import detect_leading_silence, detect_silence
|
|
from app.utils.threading import ThreadWithReturnValue
|
|
|
|
|
|
def get_leading_silence_end(filepath: str):
|
|
"""
|
|
Returns the leading silence of a track.
|
|
"""
|
|
format = filepath.split(".")[-1]
|
|
audio = AudioSegment.from_file(filepath, format=format)
|
|
silence = detect_leading_silence(audio, silence_threshold=-40.0, chunk_size=10)
|
|
|
|
return silence if silence > 1000 else 0
|
|
|
|
|
|
def get_trailing_silence_start(filepath: str):
|
|
"""
|
|
Returns the trailing silence of a track.
|
|
"""
|
|
format = filepath.split(".")[-1]
|
|
audio = AudioSegment.from_file(filepath, format=format)
|
|
duration = len(audio)
|
|
|
|
audio = audio[-30000:] if len(audio) > 30000 else audio
|
|
silence_groups = detect_silence(audio, silence_thresh=-40.0, seek_step=10)
|
|
|
|
if len(silence_groups) == 0:
|
|
return duration
|
|
|
|
silence_group = silence_groups[-1]
|
|
is_ok = silence_group[1] == len(audio)
|
|
|
|
if is_ok:
|
|
return duration - (silence_group[1] - silence_group[0])
|
|
|
|
return duration
|
|
|
|
|
|
def get_silence_paddings(ending_file: str, starting_file: str):
|
|
"""
|
|
Returns the ending silence of a track and the starting silence of the next.
|
|
"""
|
|
silence = {"starting_file": 0, "ending_file": 0}
|
|
ending_thread = None
|
|
starting_thread = None
|
|
|
|
if os.path.exists(ending_file):
|
|
ending_thread = ThreadWithReturnValue(
|
|
target=get_trailing_silence_start, args=(ending_file,)
|
|
)
|
|
ending_thread.start()
|
|
|
|
if os.path.exists(starting_file):
|
|
starting_thread = ThreadWithReturnValue(
|
|
target=get_leading_silence_end, args=(starting_file,)
|
|
)
|
|
starting_thread.start()
|
|
|
|
if ending_thread:
|
|
silence["ending_file"] = ending_thread.join()
|
|
|
|
if starting_thread:
|
|
silence["starting_file"] = starting_thread.join()
|
|
|
|
return silence
|