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
|
#!/usr/bin/env python
"""
This script generates subtitles for Toonstrack cutscenes
in a single SUBTITLES.PAK file from given directory of .SBV subtitles
Usage:
```
pakdir.py SUBTITLES_DIR
````
"""
import struct
from itertools import chain
from submaker import sbv2tss
def write_uint32_le(number):
return struct.pack('<I', number)
def calculate_index_length(pak_index):
return sum(len(write_index_entry(fname, 0)) for fname in pak_index)
def write_index_entry(fname, offset):
return write_uint32_le(offset) + fname.encode() + b'\00'
def generate_index(data_files):
end = ('\00\00\00\00', b'')
pak_index, rdata = zip(*chain(data_files, (end,)))
off = calculate_index_length(pak_index)
for fname, fdata in zip(pak_index, rdata):
yield write_index_entry(fname, off), fdata
off += len(fdata)
def create_entry(filename):
name, _ = os.path.splitext(os.path.basename(filename))
return name + '.tss', ''.join(sbv2tss(filename)).encode()
if __name__ == "__main__":
import os
import glob
import sys
if len(sys.argv) < 2:
print("Usage: pakdir.py SUBTITLES_DIR")
exit(1)
paths = sys.argv[1:]
paths = (os.path.join(path, '*.sbv') for path in paths)
files = sorted(set(chain.from_iterable(glob.iglob(r) for r in paths)))
print(files)
index, data = zip(*generate_index(create_entry(filename) for filename in files))
with open('SUBTITLES.PAK', 'wb') as output:
output.write(b''.join(index))
output.write(b''.join(data))
|
