-
Notifications
You must be signed in to change notification settings - Fork 0
/
rtsoundbank.py
74 lines (55 loc) · 2.57 KB
/
rtsoundbank.py
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
import struct, json
from pathlib import Path
from functools import partial
struct_fmt = "<IIIIIcccc" # little-endian, 5 integers, 4 bytes (a gba pointer)
# friendly reminder: "integer" tipically means a 16-bit integer
# and a 16-bit integer is the same as a DWORD
def parse2list(_bin, unpack, length):
''' https://stackoverflow.com/a/14216741 '''
return [unpack(chunk) for chunk in iter(partial(_bin.read, length), b'')]
def banklist2dict(_list):
dictIndex = []
for item in _list:
itemDict = {}
itemDict['length'] = item[0]
itemDict['samplerate'] = item[1]
itemDict['pitch'] = item[2]
itemDict['looppoints'] = item[3]
itemDict['looppointl'] = item[4]
itemDict['start'] = [item[5].hex(), item[6].hex(), item[7].hex(), item[8].hex()]
dictIndex.append(itemDict)
return dictIndex
def pointer_conv(p):
ptrBytes = bytearray()
for i in range(3):
ptrBytes += bytes.fromhex(p[i])
return int.from_bytes(bytes(ptrBytes), byteorder="little")
#return int.from_bytes(b''.join(struct.unpack(">ccc", (struct.pack('>i', p)[1:]))), byteorder="big")
def decode_table(_bin):
struct_len = struct.calcsize(struct_fmt)
struct_unpack = struct.Struct(struct_fmt).unpack_from
parsedIndex = []
#with open(_bin, 'rb') as f:
parsedIndex = parse2list(_bin, struct_unpack, struct_len)
return(banklist2dict(parsedIndex))
def decode_table_json(_bin):
return json.dumps(decode_table(_bin), indent=4)
def encode_table(f):
binIndex = bytearray()
dictIndex = json.loads(f.read()) # json -> dict
for item in dictIndex:
binItem = struct.pack(struct_fmt, item["length"], item["samplerate"], item["pitch"], item["looppoints"], item["looppointl"],
bytes.fromhex(item["start"][0]), bytes.fromhex(item["start"][1]), bytes.fromhex(item["start"][2]), bytes.fromhex(item["start"][3]))
binIndex += binItem
return binIndex
def export_samples(ROMIn, sampleTable, outputFolder): # samples is a list containing the id for the samples that will be exported
sample_count = 000
tengokuROM = ROMIn.read()
dictIndex = sampleTable
Path(outputFolder).mkdir(parents=True, exist_ok=True) # https://stackoverflow.com/a/273227
for item in dictIndex:
startOffset = pointer_conv(item['start'])
lengthOffset = startOffset + item['length']
with open(Path(outputFolder, str(sample_count)+".raw"), "wb") as outFile:
outFile.write(tengokuROM[startOffset:lengthOffset])
sample_count += 1