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
|
#!/usr/bin/env python3
import argparse
import os
import struct
import sys
from pathlib import Path
script_dir = os.path.dirname(os.path.realpath(__file__))
root_dir = script_dir + "/../"
src_dir = root_dir + "src/"
def get_c_dir(dirname):
for root, dirs, files in os.walk(src_dir):
for directory in dirs:
if directory == dirname:
return os.path.join(root, directory)
def get_c_file(directory):
for root, dirs, files in os.walk(directory):
for file in files:
if file.endswith(".c") and "data" not in file:
return file
def parse_data(file):
entries = dict()
with open(file, 'r') as f:
glabels = f.read().split('glabel ')
for glabel in glabels[2:]:
split = glabel.split('\n')
label = split[0]
rest_of_label = split[1:]
values = list()
for f in rest_of_label:
f = f.split('\t')
values.append(f[-1])
entries[split[0]] = values[:-1]
return entries
def construct_arrays(data):
output = ""
for label, values in data.items():
output += "s32 " + label + "[] = {"
for value in values:
output += "\n\t" + value + ","
#labels floats as floats
try:
binary_data = bytes.fromhex(value[2:])
number = struct.unpack('!f', binary_data)[0]
if abs(number) > 0.0001 and abs(number) < 100000.0:
output += " // f32: " + str(number)
except Exception as e:
pass
output += "\n};\n\n"
return output
def construct_extern(data):
output = ""
for label, values in data.items():
output += "extern UNK_TYPE " + label + ";\n"
return output
def main():
parser = argparse.ArgumentParser(usage="./data_parser.py --file path/to/file.c or ./data_parser.py (from an actor or gamestate's asm dir)",
description="Creates a data.c file. "
"Output will be saved as data.c")
parser.add_argument('--file', help="path of c file to be processed", required=False)
parser.add_argument('--method', help="Method of output - options are 'extern' and 'arrays'.", required=False)
args = parser.parse_args()
if args.method not in ['extern', 'arrays']:
print("Acceptable methods are {extern} and {arrays}.")
print("Example usage: ./data_parser.py --method arrays --file ../../asm/overlays/ovl_En_Wallmas_data.asm")
return
if args.file:
c_file_path = args.file
print("Using file: {}".format(c_file_path))
else:
this_dir = Path.cwd()
c_dir_path = get_c_dir(this_dir.name)
if c_dir_path is None:
sys.exit(
"Cannot find appropriate c file dir. In argumentless mode, run this script from the c file's corresponding asm dir.")
print("Example usage: ./data_parser.py --method arrays --file ../../asm/overlays/ovl_En_Wallmas_data.asm")
c_file = get_c_file(c_dir_path)
c_file_path = os.path.join(c_dir_path, c_file)
print("Using file: {}".format(c_file_path))
data = parse_data(c_file_path)
output = None
if args.method == 'extern':
output = construct_extern(data)
elif args.method == 'arrays':
output = construct_arrays(data)
with open(os.path.join(root_dir, "data.c"), "w", encoding="UTF-8") as f:
f.write(output)
if __name__ == "__main__":
main()
|
