mirror of
https://github.com/cainan-c/TaikoPythonTools.git
synced 2024-12-02 19:17:23 +01:00
51 lines
1.8 KiB
Python
51 lines
1.8 KiB
Python
import os
|
|
import argparse
|
|
import subprocess
|
|
import shutil
|
|
import tempfile
|
|
from pydub import AudioSegment
|
|
|
|
def parse_arguments():
|
|
parser = argparse.ArgumentParser(description='CLI tool to create .acb files')
|
|
parser.add_argument('input_audio', type=str, help='Path to the input audio file')
|
|
parser.add_argument('song_id', type=str, help='Song ID')
|
|
return parser.parse_args()
|
|
|
|
def main():
|
|
args = parse_arguments()
|
|
|
|
# Generate a unique random temporary folder name
|
|
with tempfile.TemporaryDirectory(prefix='song_') as temp_folder:
|
|
try:
|
|
# Convert input audio to 44100Hz WAV
|
|
input_audio = args.input_audio
|
|
temp_wav_file = os.path.join(temp_folder, f'input_{args.song_id}.wav')
|
|
|
|
audio = AudioSegment.from_file(input_audio)
|
|
audio = audio.set_frame_rate(44100)
|
|
audio.export(temp_wav_file, format='wav')
|
|
|
|
# Generate .hca file using VGAudioCli.exe
|
|
hca_folder = os.path.join(temp_folder, f'song_{args.song_id}')
|
|
os.makedirs(hca_folder, exist_ok=True)
|
|
hca_file = os.path.join(hca_folder, '00000.hca')
|
|
subprocess.run(['bin/VGAudioCli.exe', temp_wav_file, hca_file], check=True)
|
|
|
|
# Copy sample .acb template to temporary location
|
|
acb_template = 'templates/song_sample.acb'
|
|
temp_acb_file = os.path.join(temp_folder, f'song_{args.song_id}.acb')
|
|
shutil.copy(acb_template, temp_acb_file)
|
|
|
|
# Edit .acb using ACBEditor
|
|
subprocess.run(['bin/ACBEditor.exe', hca_folder], check=True)
|
|
|
|
# Move .acb file to the current directory
|
|
final_acb_file = f'song_{args.song_id}.acb'
|
|
os.replace(temp_acb_file, final_acb_file)
|
|
|
|
except Exception as e:
|
|
print(f"Error: {e}")
|
|
|
|
if __name__ == '__main__':
|
|
main()
|