mirror of
https://github.com/dcs-retribution/dcs-retribution.git
synced 2025-11-10 15:41:24 +00:00
54 lines
1.8 KiB
Python
54 lines
1.8 KiB
Python
from game import db
|
|
from .conflictgen import *
|
|
from .naming import *
|
|
|
|
from dcs.mission import *
|
|
from dcs.unittype import *
|
|
from dcs.point import *
|
|
from dcs.task import *
|
|
from dcs.country import *
|
|
|
|
SPREAD_DISTANCE_FACTOR = 0.1, 0.3
|
|
SPREAD_DISTANCE_SIZE_FACTOR = 0.1
|
|
|
|
|
|
class ArmorConflictGenerator:
|
|
def __init__(self, mission: Mission, conflict: Conflict):
|
|
self.m = mission
|
|
self.conflict = conflict
|
|
|
|
def _group_point(self, point) -> Point:
|
|
distance = randint(
|
|
int(self.conflict.size * SPREAD_DISTANCE_FACTOR[0]),
|
|
int(self.conflict.size * SPREAD_DISTANCE_FACTOR[1]),
|
|
)
|
|
|
|
return point.random_point_within(distance, self.conflict.size * SPREAD_DISTANCE_SIZE_FACTOR)
|
|
|
|
def _generate_group(self, side: Country, unit: VehicleType, count: int, at: Point):
|
|
for c in range(count):
|
|
group = self.m.vehicle_group(
|
|
side,
|
|
namegen.next_armor_group_name(),
|
|
unit,
|
|
position=self._group_point(at),
|
|
group_size=1,
|
|
move_formation=PointAction.OffRoad)
|
|
wayp = group.add_waypoint(self.conflict.position.point_from_heading(0, 500))
|
|
wayp.tasks = []
|
|
|
|
def generate(self, attackers: db.ArmorDict, defenders: db.ArmorDict):
|
|
for type, count in attackers.items():
|
|
self._generate_group(
|
|
side=self.conflict.attackers_side,
|
|
unit=type,
|
|
count=count,
|
|
at=self.conflict.ground_attackers_location)
|
|
|
|
for type, count in defenders.items():
|
|
self._generate_group(
|
|
side=self.conflict.defenders_side,
|
|
unit=type,
|
|
count=count,
|
|
at=self.conflict.ground_defenders_location)
|