summaryrefslogtreecommitdiff
path: root/tools/libjaudio/baa.py
blob: b1bfe6efe30c4a6b412ed292efd3576227e6028d (plain)
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
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
import struct

class JAUAudioArcInterpreter:
    def __init__(self,file):
        self.offset = 0
        self.file = file
        self.nameTable = {"Sections": []}
    
    def readU32_string(self):
        val = struct.unpack(">4s",self.file[self.offset:self.offset+4])[0]
        self.offset += 4
        return val

    def readU32_(self):
        val = struct.unpack(">I",self.file[self.offset:self.offset+4])[0]
        self.offset += 4
        return val
    
    def readU8_(self):
        val = struct.unpack(">B",self.file[self.offset:self.offset+1])[0]
        self.offset += 1
        return val
    
    def getString(self,bytes,offset):
        string = bytes[offset:]
        string = string[:string.index(0)]
        return string

    # More JAudio Decomp is required for this
    def readWS(self,bank_no,offset,waveArc):
        # print(bank_no)
        # print(offset)
        # print(waveArc)

        size = struct.unpack(">I",self.file[offset+4:offset+8])[0]
        # print(size)
        wsys = self.file[offset:offset+size]
        someNumber,archiveBankOffset,ctrlGroupOffset = struct.unpack(">III",wsys[12:24])
        # print(ctrlGroupOffset)

    def readBNK(self,param_1,param_2):
        pass

    def beginBNKList(self,param_1,param_2):
        pass
    
    def endBNKList(self):
        pass
    
    def readBSC(self,param_1,param_2):
        pass

    def readBST(self,offset,size):
        # print(offset)
        bst = self.file[offset:offset+size]
        numSections = struct.unpack(">I",bst[32:36])[0]

        sectionOffsets = []        
        for i in range(numSections):
             sectionOffsets.append(struct.unpack(">I",bst[36+(i*4):40+(i*4)])[0])
        for sectionOffset in sectionOffsets:
            numGroups = struct.unpack(">I",bst[sectionOffset:sectionOffset+4])[0]
            groupOffsets = []
            for i in range(numGroups):
                groupOffsets.append(struct.unpack(">I",bst[sectionOffset+4+(i*4):sectionOffset+8+(i*4)])[0])
            # print(groupOffsets)
            for groupOffset in groupOffsets:
                numEntries,unk = struct.unpack(">II",bst[groupOffset:groupOffset+8])
                entries = []
                for i in range(numEntries):
                    val = struct.unpack(">I",bst[groupOffset+8+(i*4):groupOffset+12+(i*4)])[0]
                    entries.append({"Type":val>>24,"Offset":val&0xFFFFFF})
                for entry in entries:
                    
                    # print(f"{entry['Type']} {entry['Offset']}")
                    match entry["Type"] & 0xf0:
                        case 0x50: #SFX
                            priority,unk,pad,swBit,volume = struct.unpack(">BBHIf",bst[entry["Offset"]:entry["Offset"]+12])
                            # print(f"{priority:01x} {unk:01x} {pad:02x} {swBit:04x} {volume}")
                        case 0x60: #Arc BGM
                            priority,unk,arcIndex = struct.unpack(">BBH",bst[entry["Offset"]:entry["Offset"]+4])
                            # print(f"{priority:01x} {unk:01x} Arc Index: {arcIndex}")
                        case 0x70: #Streams
                            priority,unk,resourceID,stringOffset = struct.unpack(">BBHI",bst[entry["Offset"]:entry["Offset"]+8])
                            # print(f"{priority:01x} {unk:01x} {resourceID:02x} {stringOffset:04x} ")
                            streamPath = self.getString(bst,stringOffset)
                            # print(f"{streamPath}")
                    

    def readBSTN(self,offset,size):
        # print(param_1)
        bstn = self.file[offset:offset+size]
        numSections = struct.unpack(">I",bstn[32:36])[0]
        # print(numSections)
        sectionOffsets = []
        for i in range(numSections):
            sectionOffsets.append(struct.unpack(">I",bstn[36+(i*4):40+(i*4)])[0])
        # print(sectionOffsets)
        sections = []
        for offset in sectionOffsets:
            numGroups,sectionNameOffset = struct.unpack(">II",bstn[offset:offset+8])
            sectionName = self.getString(bstn,sectionNameOffset)
            # print(f"Section {sectionName} has {numGroups} groups")
            section = {"Section Name":sectionName.decode('ascii'),
                       "Groups":[]}
            groupOffsets = []
            for i in range(numGroups):
                groupOffsets.append(struct.unpack(">I",bstn[offset+8+(i*4):offset+12+(i*4)])[0])
            groups = []
            for groupOffset in groupOffsets:
                groupSize,groupNameOffset = struct.unpack(">II",bstn[groupOffset:groupOffset+8])
                groupName = self.getString(bstn,groupNameOffset)
                group = {"Group Name":groupName.decode('ascii'),
                         "Names": []}
                # print(f"Group {groupName} has {groupSize} entries")
                nameOffsets = []
                for i in range(groupSize):
                    nameOffsets.append(struct.unpack(">I",bstn[groupOffset+8+(i*4):groupOffset+12+(i*4)])[0])
                names = []
                for nameOffset in nameOffsets:
                    names.append(self.getString(bstn,nameOffset).decode("ascii"))
                # print(names)
                group["Names"] = names
                groups.append(group)
            section["Groups"] = groups
            sections.append(section)
        self.nameTable["Sections"] = sections


    def readBMS(self,param_1,param_2,param_3):
        pass

    def readBMS_fromArchive(self):
        pass

    def newVoiceBank(self,param_1,param_2):
        pass

    def newDynamicSeqBlock(self,param_1):
        pass

    def readBSFT(self,param_1):
        pass

    def readMaxSeCategory(self,param_1,param_2,param_3):
        pass

    def parse(self):
        if self.readU32_string() != b'AA_<':
            return False
                
        while self.readCommand_() == True:
            pass

        return True

    def readCommand_(self):
        command = self.readU32_string()
        print(command)
        match command:
            case b'>_AA':
                return False
            case b'ws  ':
                self.readWS(self.readU32_(),self.readU32_(),self.readU32_())
            case b'bnk ':
                self.readBNK(self.readU32_(),self.readU32_())
            case b'bl_<':
                self.beginBNKList(self.readU32_(),self.readU32_())
            case b'>_bl':
                self.endBNKList()
            case b'bsc ':
                start = self.readU32_()
                end = self.readU32_()
                self.readBSC(start,end-start)
            case b'bst ':
                start = self.readU32_()
                end = self.readU32_()
                self.readBST(start,end-start)
            case b'bstn':
                start = self.readU32_()
                end = self.readU32_()
                self.readBSTN(start,end-start)
            case b'bms ':
                param_1 = self.readU32_()
                start = self.readU32_()
                end = self.readU32_()
                self.readBMS(param_1,start,end-start)
            case b'bmsa':
                self.readBMS_fromArchive(self.readU32_())
            case b'vbnk':
                self.newVoiceBank(self.readU32_(),self.readU32_())
            case b'dsqb':
                self.newDynamicSeqBlock(self.readU32_())
            case b'bsft':
                self.readBSFT(self.readU32_())
            case b'sect':
                self.readU8_()
                self.readMaxSeCategory(self.readU8_(),self.readU8_(),self.readU8_())
        return True



def main():
    baa = JAUAudioArcInterpreter(open("Z2Sound.baa","rb").read())
    baa.parse()

if __name__ == "__main__":
    main()