-
Notifications
You must be signed in to change notification settings - Fork 0
/
bsp_to_dict.py
125 lines (98 loc) · 3.64 KB
/
bsp_to_dict.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
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
import json
from PIL import Image
from elasticsearch import Elasticsearch
import annotate_map
from calc import DATA_PATH
import calc
# These are obfuscated
KEY_MAPPINGS = {
'212': 'type',
'1072': 'desc1',
'845': 'desc2',
'709': 'translation',
'80': 'rotation',
'2553': 'scale',
'652': 'model',
'43840': 'modes',
'43887': 'extra_data',
'44105': 'team', # (allies, axis) for spawn selection camera
# These were left in in one of the S03 builds
'63913': 'lgbudgetingprobesize',
'63914': 'lgsplittransients',
'64017': 'locked_proxylod',
'64197': 'mapnamefilter',
'66774': 'scriptable_reserved_count',
'68080': 'superterrainlightbakelodoverride',
'68450': 'tr_vis_facing_dist_add_override',
'68451': 'tr_vis_radius_override_lod1',
'68452': 'tr_vis_radius_override_lod2',
'68489': 'transient_prefab_group',
'68490': 'transient_world_autolod_enabled'
}
def to_dict(f_in):
result = []
for line in f_in.readlines():
line = line[:-1] # Remove newline
if line == '{':
current_dict = {}
elif line == '}':
# noinspection PyUnboundLocalVariable
result.append(current_dict)
else:
key, value = line.split(' ', 1)
value = value[1:-1] # Remove extra quotes at start and end
# Remap keys to something readable
if key in KEY_MAPPINGS.keys():
key = KEY_MAPPINGS[key]
# Reformat rotation and translation
if key == 'rotation' or key == 'translation':
value = value.split(' ') # Split x, y, z values
value = list(map(float, value)) # Convert strings to floats
elif key == '20185': # Not sure what it means though
value = bool(int(value)) # 0, 1 -> False, True
current_dict[key] = value
return result
def get_all_in_box(res, xy_sw, xy_ne):
# Convert coords
xy_sw = calc.calc_coord(*xy_sw, True)
xy_ne = calc.calc_coord(*xy_ne, True)
if xy_sw[0] >= xy_ne[0] or xy_sw[1] >= xy_ne[1]:
raise TypeError("invalid bounding box")
result = []
for entity in res:
if 'translation' in entity.keys():
posx, posy = entity['translation'][:2]
if xy_sw[0] <= posx <= xy_ne[0] and xy_sw[1] <= posy <= xy_ne[1]:
result.append(entity)
return result
def main():
res = to_dict((DATA_PATH / 'mp_don3.d3dbsp').open())
# Restrict to coords in bounding box
if input('bounding box?') in ['yes', 'y']:
xy = []
for i in range(4):
if i % 2 == 0:
xy.append(float(input('x?')))
else:
xy.append(float(input('y?')))
res = get_all_in_box(res, xy[:-2], xy[-2:])
# Dump
dump_type = input('dump type? (es/json)')
if dump_type == 'es':
es = Elasticsearch()
index = input('index?')
for i, entity in enumerate(res):
res = es.index(index=index, id=i, body=entity)
elif dump_type == 'json':
json.dump(res, (DATA_PATH / 'mp_don3.json').open('w'), indent=4, sort_keys=True)
elif dump_type == 'map':
# Why?
with Image.open(DATA_PATH / 'stitched_map.png') as im:
for ent in res:
if 'translation' in ent.keys() and ent.get('model') != 'fullbody_hero_price_urban'\
and ent.get('type') == 'script_struct':
print('kek')
im = annotate_map.annotate(im, ent['translation'][:2], scale=.3)
im.convert('RGB').save(DATA_PATH / 'annotated_map.jpg')
if __name__ == '__main__':
main()