mirror of
https://github.com/dcs-liberation/dcs_liberation.git
synced 2025-11-10 14:22:26 +00:00
79 lines
2.7 KiB
Python
79 lines
2.7 KiB
Python
import logging
|
|
|
|
from game import db
|
|
from .conflictgen import *
|
|
from .naming import *
|
|
|
|
from dcs.mission import *
|
|
from dcs.statics import *
|
|
|
|
FARP_FRONTLINE_DISTANCE = 10000
|
|
|
|
|
|
CATEGORY_MAPPING = {
|
|
"power": [Fortification.Workshop_A],
|
|
"warehouse": [Warehouse.Warehouse],
|
|
"fuel": [Warehouse.Tank],
|
|
"ammo": [Warehouse.Ammunition_depot],
|
|
"farp": [Fortification.FARP_Tent],
|
|
"comms": [Fortification.TV_tower],
|
|
"oil": [Fortification.Oil_platform],
|
|
}
|
|
|
|
|
|
class GroundObjectsGenerator:
|
|
FARP_CAPACITY = 4
|
|
|
|
def __init__(self, mission: Mission, conflict: Conflict, game):
|
|
self.m = mission
|
|
self.conflict = conflict
|
|
self.game = game
|
|
|
|
def generate_farps(self, number_of_units=1) -> typing.Collection[StaticGroup]:
|
|
assert self.conflict.is_vector, "FARP could be generated only on frontline conflicts!"
|
|
|
|
for i, _ in enumerate(range(0, number_of_units, self.FARP_CAPACITY)):
|
|
heading = self.conflict.heading - 90
|
|
position = self.conflict.find_ground_position(self.conflict.center.point_from_heading(heading, FARP_FRONTLINE_DISTANCE), heading)
|
|
position = position.point_from_heading(0, i * 275)
|
|
|
|
yield self.m.farp(
|
|
country=self.m.country(self.game.player),
|
|
name="FARP",
|
|
position=position,
|
|
)
|
|
|
|
def generate(self):
|
|
side = self.m.country(self.game.enemy)
|
|
|
|
cp = None # type: ControlPoint
|
|
if self.conflict.attackers_side.name == self.game.player:
|
|
cp = self.conflict.to_cp
|
|
else:
|
|
cp = self.conflict.from_cp
|
|
|
|
for ground_object in cp.ground_objects:
|
|
if ground_object.category == "defense":
|
|
unit_type = random.choice(self.game.commision_unit_types(cp, AirDefence))
|
|
assert unit_type is not None, "Cannot find unit type for GroundObject defense ({})!".format(cp)
|
|
|
|
group = self.m.vehicle_group(
|
|
country=side,
|
|
name=ground_object.string_identifier,
|
|
_type=unit_type,
|
|
position=ground_object.position,
|
|
heading=ground_object.heading
|
|
)
|
|
|
|
logging.info("generated defense object identifier {} with mission id {}".format(group.name, group.id))
|
|
else:
|
|
group = self.m.static_group(
|
|
country=side,
|
|
name=ground_object.string_identifier,
|
|
_type=random.choice(CATEGORY_MAPPING[ground_object.category]),
|
|
position=ground_object.position,
|
|
heading=ground_object.heading
|
|
)
|
|
|
|
logging.info("generated object identifier {} with mission id {}".format(group.name, group.id))
|