mirror of
https://github.com/dcs-retribution/dcs-retribution.git
synced 2025-11-10 15:41:24 +00:00
88 lines
3.2 KiB
Python
88 lines
3.2 KiB
Python
import pickle
|
|
import typing
|
|
|
|
from dcs.mission import Mission
|
|
from dcs.mapping import Point
|
|
from dcs.unitgroup import VehicleGroup, StaticGroup
|
|
from dcs.unit import *
|
|
from dcs.statics import warehouse_map, fortification_map
|
|
|
|
from game import db
|
|
from gen.groundobjectsgen import TheaterGroundObject
|
|
|
|
m = Mission()
|
|
m.load_file("./cau_groundobjects.miz")
|
|
|
|
|
|
def parse_name(name: str) -> int:
|
|
first_part = name.split()[0].split("|")
|
|
return int(first_part[0]) if len(first_part) == 1 else int(first_part[1])
|
|
|
|
|
|
if __name__ == "__main__":
|
|
theater_objects = []
|
|
|
|
for group in m.country("Russia").static_group + m.country("Russia").vehicle_group:
|
|
for unit in group.units:
|
|
theater_object = TheaterGroundObject()
|
|
theater_object.object_id = len(theater_objects) + 1
|
|
|
|
try:
|
|
theater_object.cp_id = parse_name(str(unit.name))
|
|
except Exception as e:
|
|
theater_object.cp_id = parse_name(str(group.name))
|
|
|
|
theater_object.position = unit.position
|
|
theater_object.heading = unit.heading
|
|
|
|
if isinstance(unit, Vehicle):
|
|
theater_object.dcs_identifier = "AA"
|
|
else:
|
|
theater_object.dcs_identifier = unit.type
|
|
|
|
airport_distance = m.terrain.airport_by_id(theater_object.cp_id).position.distance_to_point(theater_object.position)
|
|
if airport_distance > 150000:
|
|
print("Object {} {} is placed {}m from airport {}!".format(theater_object.dcs_identifier,
|
|
group.name,
|
|
airport_distance,
|
|
m.terrain.airport_by_id(theater_object.cp_id)))
|
|
|
|
assert theater_object.dcs_identifier
|
|
assert theater_object.cp_id
|
|
assert theater_object.object_id
|
|
|
|
theater_objects.append(theater_object)
|
|
|
|
group_ids = 1
|
|
for object_a in theater_objects:
|
|
for object_b in theater_objects:
|
|
if object_a.position.distance_to_point(object_b.position) < 2000:
|
|
if object_a.group_id and object_b.group_id:
|
|
continue
|
|
elif object_a.group_id:
|
|
object_b.group_id = object_a.group_id
|
|
elif object_b.group_id:
|
|
object_a.group_id = object_b.group_id
|
|
else:
|
|
object_a.group_id = group_ids
|
|
object_b.group_id = group_ids
|
|
group_ids += 1
|
|
|
|
assert object_a.cp_id == object_b.cp_id, "Object {} and {} are placed in group with different airports!".format(object_a.string_identifier, object_b.string_identifier)
|
|
|
|
for a in theater_objects:
|
|
if not a.group_id:
|
|
a.group_id = group_ids
|
|
group_ids += 1
|
|
|
|
print("Total {} objects".format(len(theater_objects)))
|
|
with open("../cau_groundobjects.p", "wb") as f:
|
|
result = {}
|
|
for theater_object in theater_objects:
|
|
if theater_object.cp_id not in result:
|
|
result[theater_object.cp_id] = []
|
|
result[theater_object.cp_id].append(theater_object)
|
|
|
|
pickle.dump(result, f)
|
|
|