mirror of
https://github.com/dcs-retribution/dcs-retribution.git
synced 2025-11-10 15:41:24 +00:00
45 lines
1.2 KiB
Python
45 lines
1.2 KiB
Python
import pickle
|
|
import typing
|
|
|
|
from game import db
|
|
from gen.groundobjectsgen import TheaterGroundObject
|
|
from dcs.mission import Mission
|
|
from dcs.terrain import PersianGulf
|
|
|
|
m = Mission()
|
|
m.load_file("./cau_groundobjects.miz")
|
|
|
|
result = {}
|
|
|
|
|
|
def append_group(cp_id, category, group_id, object_id, position, heading):
|
|
global result
|
|
|
|
if cp_id not in result:
|
|
result[cp_id] = []
|
|
|
|
result[cp_id].append(TheaterGroundObject(category, cp_id, group_id, object_id, position, heading))
|
|
|
|
|
|
def parse_name(name: str) -> typing.Tuple:
|
|
args = str(name).split("|")
|
|
if len(args) == 3:
|
|
args.append("1")
|
|
|
|
return args[0], int(args[1]), int(args[2]), int(args[3])
|
|
|
|
|
|
for group in m.country("Russia").static_group + m.country("Russia").vehicle_group:
|
|
try:
|
|
category, cp_id, group_id, object_id = parse_name(str(group.name))
|
|
except:
|
|
print("Failed to parse {}".format(group.name))
|
|
continue
|
|
|
|
append_group(cp_id, category, group_id, object_id, [group.position.x, group.position.y], group.units[0].heading)
|
|
|
|
print("Total {} objects".format(sum([len(x) for x in result.values()])))
|
|
with open("../cau_groundobjects.p", "wb") as f:
|
|
pickle.dump(result, f)
|
|
|