mirror of
https://github.com/cainan-c/TaikoPythonTools.git
synced 2024-11-30 18:24:33 +01:00
71 lines
2.8 KiB
Python
71 lines
2.8 KiB
Python
|
import argparse
|
||
|
|
import subprocess
|
||
|
|
import os
|
||
|
|
import sys
|
||
|
|
|
||
|
|
def convert_audio_to_nus3bank(input_audio, audio_type, game, preview_point, song_id):
|
||
|
|
# Determine the output filename for the nus3bank
|
||
|
|
output_filename = f"song_{song_id}.nus3bank"
|
||
|
|
converted_audio_file = f"{input_audio}.{audio_type}"
|
||
|
|
|
||
|
|
# Determine the path to the run.py script within the 'script' folder
|
||
|
|
templates_folder = os.path.join(os.path.dirname(__file__), 'script')
|
||
|
|
run_py_path = os.path.join(templates_folder, 'run.py')
|
||
|
|
|
||
|
|
# Prepare the command based on the audio type
|
||
|
|
if audio_type in ["bnsf", "at9", "idsp", "lopus", "wav"]:
|
||
|
|
# Construct the command to convert input audio to the specified type
|
||
|
|
conversion_command = ["python", run_py_path, audio_type, input_audio, f"{input_audio}.{audio_type}"]
|
||
|
|
|
||
|
|
# Construct the command to create the nus3bank
|
||
|
|
nus3_command = ["python", run_py_path, "nus3", game, f"{input_audio}.{audio_type}", str(preview_point), output_filename]
|
||
|
|
|
||
|
|
try:
|
||
|
|
# Execute the conversion command
|
||
|
|
subprocess.run(conversion_command, check=True)
|
||
|
|
|
||
|
|
# Execute the nus3 command
|
||
|
|
subprocess.run(nus3_command, check=True)
|
||
|
|
|
||
|
|
print(f"Conversion successful! Created {output_filename}")
|
||
|
|
|
||
|
|
# Delete the non-nus3bank file after successful conversion
|
||
|
|
if os.path.exists(converted_audio_file):
|
||
|
|
os.remove(converted_audio_file)
|
||
|
|
print(f"Deleted {converted_audio_file}")
|
||
|
|
except subprocess.CalledProcessError as e:
|
||
|
|
print(f"Error: {e}")
|
||
|
|
else:
|
||
|
|
print(f"Unsupported audio type: {audio_type}")
|
||
|
|
|
||
|
|
def main():
|
||
|
|
# Create an argument parser
|
||
|
|
parser = argparse.ArgumentParser(description="Convert audio to nus3bank")
|
||
|
|
|
||
|
|
# Define command-line arguments
|
||
|
|
parser.add_argument("input_audio", type=str, nargs="?", help="Input audio file path.")
|
||
|
|
parser.add_argument("audio_type", type=str, nargs="?", help="Type of input audio (e.g., wav, bnsf, at9, idsp, lopus).")
|
||
|
|
parser.add_argument("game", type=str, nargs="?", help="Game type (e.g., nijiiro, ns1, ps4, wiiu3).")
|
||
|
|
parser.add_argument("preview_point", type=int, nargs="?", help="Audio preview point in ms.")
|
||
|
|
parser.add_argument("song_id", type=str, nargs="?", help="Song ID for the nus3bank file.")
|
||
|
|
|
||
|
|
# Parse the command-line arguments
|
||
|
|
args = parser.parse_args()
|
||
|
|
|
||
|
|
# If no arguments are provided, display usage information
|
||
|
|
if len(sys.argv) == 1:
|
||
|
|
parser.print_help()
|
||
|
|
sys.exit(0)
|
||
|
|
|
||
|
|
# Validate input audio file path
|
||
|
|
if not args.input_audio:
|
||
|
|
print("Error: Input audio file path is required.")
|
||
|
|
parser.print_help()
|
||
|
|
sys.exit(1)
|
||
|
|
|
||
|
|
# Call function to convert audio to nus3bank
|
||
|
|
convert_audio_to_nus3bank(args.input_audio, args.audio_type, args.game, args.preview_point, args.song_id)
|
||
|
|
|
||
|
|
if __name__ == "__main__":
|
||
|
|
main()
|