mirror of
https://github.com/coop-deluxe/sm64coopdx.git
synced 2024-11-25 13:35:12 +00:00
65 lines
1.9 KiB
Python
65 lines
1.9 KiB
Python
|
#!/usr/bin/env python3
|
||
|
import sys
|
||
|
import os
|
||
|
import time
|
||
|
|
||
|
# find sound/samples/*custom* -name '*.aiff' | while read filename; do python3 tools/append_aiffc_table.py "$filename"; done
|
||
|
|
||
|
# Get the filename from the first command line argument
|
||
|
if len(sys.argv) < 2:
|
||
|
print("Usage: {} <filename>".format(sys.argv[0]))
|
||
|
sys.exit(1)
|
||
|
filename = sys.argv[1]
|
||
|
|
||
|
# check if the binary file contains the string 'VADPCMCODES', and if so print that it was already found and exit
|
||
|
with open(filename, 'rb') as f:
|
||
|
content = f.read()
|
||
|
if b'VADPCMCODES' in content:
|
||
|
print('Already contains a table for ' + filename + '. Exiting.')
|
||
|
sys.exit(0)
|
||
|
|
||
|
# run the program ./tools/tabledesign on the specified filename and capture the output
|
||
|
import subprocess
|
||
|
command = ['./tools/tabledesign', '-s', '1', filename]
|
||
|
result = subprocess.run(command, stdout=subprocess.PIPE, stderr=subprocess.PIPE)
|
||
|
output = result.stdout.decode('utf-8').strip()
|
||
|
|
||
|
# split the output by whitespace
|
||
|
words = output.split()
|
||
|
|
||
|
# calculate header info
|
||
|
size = len(words) * 2 + 18
|
||
|
unknown = 0x0b
|
||
|
version = 1
|
||
|
|
||
|
# append the chunk header
|
||
|
chunk = b''
|
||
|
chunk += b'APPL'
|
||
|
chunk += size.to_bytes(4, 'big', signed=True)
|
||
|
chunk += b'stoc'
|
||
|
chunk += unknown.to_bytes(1, 'big', signed=False)
|
||
|
chunk += b'VADPCMCODES'
|
||
|
chunk += version.to_bytes(2, 'big', signed=False)
|
||
|
|
||
|
# convert each word to a signed 16 bit integer and append them to content
|
||
|
for word in words:
|
||
|
value = int(word)
|
||
|
chunk += value.to_bytes(2, 'big', signed=True)
|
||
|
|
||
|
# find the index of the first instance of 'SSND'
|
||
|
index = content.find(b'SSND')
|
||
|
if index == -1:
|
||
|
print("Couldn't find 'SSND' for ' + filename + '. Aborting.")
|
||
|
sys.exit(1)
|
||
|
|
||
|
if size != len(chunk) - 8:
|
||
|
print('Failed to generate the correct size for ' + filename)
|
||
|
sys,exit(1)
|
||
|
|
||
|
# insert the chunk before 'SSND'
|
||
|
content = content[:index] + chunk + content[index:]
|
||
|
|
||
|
# write the modified content to the binary file
|
||
|
with open(filename, 'wb') as f:
|
||
|
f.write(content)
|