Spaces:
Sleeping
Sleeping
File size: 2,848 Bytes
f5799e7 |
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 |
"""Vocal chunks module."""
import os
import re
import wave
from modules.console_colors import ULTRASINGER_HEAD
from modules.os_helper import create_folder
from modules.Ultrastar.ultrastar_converter import (
get_end_time_from_ultrastar,
get_start_time_from_ultrastar,
)
from modules.Ultrastar.ultrastar_txt import UltrastarTxtValue
class AudioManipulation:
"""Docstring"""
def export_chunks_from_transcribed_data(
audio_filename: str, transcribed_data: [], output_folder_name: str
) -> None:
"""Export transcribed_data as vocal chunks wav files"""
print(
f"{ULTRASINGER_HEAD} Export transcribed data as vocal chunks wav files"
)
wave_file = wave.open(audio_filename, "rb")
sample_rate, n_channels = wave_file.getparams()[2], wave_file.getparams()[0]
for i, data in enumerate(transcribed_data):
start_byte = int(data.start * sample_rate * n_channels)
end_byte = int(data.end * sample_rate * n_channels)
chunk = get_chunk(end_byte, start_byte, wave_file)
export_chunk_to_wav_file(
chunk, output_folder_name, i, data.word, wave_file
)
wave_file.close()
def export_chunks_from_ultrastar_data(
audio_filename: str, ultrastar_data: UltrastarTxtValue, folder_name: str
) -> None:
"""Export ultrastar data as vocal chunks wav files"""
print(f"{ULTRASINGER_HEAD} Export Ultrastar data as vocal chunks wav files")
create_folder(folder_name)
wave_file = wave.open(audio_filename, "rb")
sample_rate, n_channels = wave_file.getparams()[2], wave_file.getparams()[0]
for i, word in enumerate(ultrastar_data.words):
start_time = get_start_time_from_ultrastar(ultrastar_data, i)
end_time = get_end_time_from_ultrastar(ultrastar_data, i)
start_byte = int(start_time * sample_rate * n_channels)
end_byte = int(end_time * sample_rate * n_channels)
chunk = get_chunk(end_byte, start_byte, wave_file)
export_chunk_to_wav_file(
chunk, folder_name, i, word, wave_file
)
def export_chunk_to_wav_file(chunk, folder_name: str, i: int, word: str, wave_file) -> None:
"""Export vocal chunks to wav file"""
clean_word = re.sub("[^A-Za-z0-9]+", "", word)
# todo: Progress?
# print(f"{str(i)} {clean_word}")
with wave.open(
os.path.join(folder_name, f"chunk_{i}_{clean_word}.wav"), "wb"
) as chunk_file:
chunk_file.setparams(wave_file.getparams())
chunk_file.writeframes(chunk)
def get_chunk(end_byte: int, start_byte: int, wave_file):
"""
Gets the chunk from wave file.
Returns chunk as n frames of audio, as a bytes object.
"""
# todo: get out of position error message
wave_file.setpos(start_byte) # ({:.2f})
chunk = wave_file.readframes(end_byte - start_byte)
return chunk
|