mirror of
https://github.com/dcs-liberation/dcs_liberation.git
synced 2025-11-10 14:22:26 +00:00
127 lines
4.5 KiB
Python
127 lines
4.5 KiB
Python
import typing
|
|
import random
|
|
from datetime import datetime, timedelta
|
|
|
|
from dcs.mission import Mission
|
|
from dcs.triggers import *
|
|
from dcs.condition import *
|
|
from dcs.action import *
|
|
|
|
from theater.weatherforecast import WeatherForecast
|
|
from theater.conflicttheater import Conflict
|
|
|
|
ACTIVATION_TRIGGER_SIZE = 80000
|
|
ACTIVATION_TRIGGER_MIN_DISTANCE = 5000
|
|
|
|
RANDOM_TIME = {
|
|
"night": 0,
|
|
"dusk": 5,
|
|
"dawn": 35,
|
|
"noon": 75,
|
|
"day": 100,
|
|
}
|
|
|
|
RANDOM_WEATHER = {
|
|
0: 0, # thunderstorm
|
|
1: 5, # heavy rain
|
|
2: 20, # rain
|
|
3: 40, # random dynamic
|
|
}
|
|
|
|
|
|
class EnvironmentSettingsGenerator:
|
|
def __init__(self, mission: Mission, conflict: Conflict, game):
|
|
self.mission = mission
|
|
self.conflict = conflict
|
|
self.game = game
|
|
|
|
def _gen_random_time(self):
|
|
time_roll = random.randint(0, 100)
|
|
time_period = None
|
|
for k, v in RANDOM_TIME.items():
|
|
if v >= time_roll:
|
|
time_period = k
|
|
break
|
|
|
|
self.mission.random_daytime(time_period)
|
|
|
|
def _gen_random_time_2(self):
|
|
start_time = datetime.now(self.mission.start_time.tzinfo).date()
|
|
daytime_map = {
|
|
"day": timedelta(hours=random.randrange(9, 18)),
|
|
"night": timedelta(hours=random.randrange(-3, 6)),
|
|
"dusk": timedelta(hours=random.randrange(18, 21)),
|
|
"dawn": timedelta(hours=random.randrange(6, 9)),
|
|
}
|
|
|
|
time_roll = random.randint(0, 100)
|
|
time_period = None
|
|
for k, v in RANDOM_TIME.items():
|
|
if v >= time_roll:
|
|
time_period = k
|
|
break
|
|
|
|
start_time += daytime_map[time_period]
|
|
self.mission.start_time = start_time
|
|
|
|
def _gen_random_weather(self):
|
|
weather_roll = random.randint(0, 100)
|
|
weather_type = None
|
|
for k, v in RANDOM_TIME.items():
|
|
if v >= weather_roll:
|
|
weather_type = k
|
|
break
|
|
|
|
if weather_type == 0:
|
|
self.mission.weather.random_thunderstorm()
|
|
elif weather_type == 1:
|
|
self.mission.weather.heavy_rain()
|
|
elif weather_type == 2:
|
|
self.mission.weather.heavy_rain()
|
|
self.mission.weather.enable_fog = False
|
|
elif weather_type == 3:
|
|
self.mission.weather.random(self.mission.start_time, self.mission.terrain)
|
|
|
|
def _gen_activation_trigger(self, player_coalition: str, enemy_coalition: str):
|
|
activate_by_trigger = []
|
|
for coalition_name, coalition in self.mission.coalition.items():
|
|
for country in coalition.countries.values():
|
|
if coalition_name == enemy_coalition:
|
|
for plane_group in country.plane_group:
|
|
plane_group.late_activation = True
|
|
activate_by_trigger.append(plane_group)
|
|
|
|
for vehicle_group in country.vehicle_group:
|
|
vehicle_group.late_activation = True
|
|
activate_by_trigger.append(vehicle_group)
|
|
|
|
zone_distance_to_aircraft = self.conflict.air_attackers_location.distance_to_point(self.conflict.position)
|
|
zone_size = min(zone_distance_to_aircraft - ACTIVATION_TRIGGER_MIN_DISTANCE, ACTIVATION_TRIGGER_SIZE)
|
|
|
|
activation_trigger_zone = self.mission.triggers.add_triggerzone(self.conflict.position, zone_size)
|
|
activation_trigger = TriggerOnce(Event.NoEvent, "Activation trigger")
|
|
activation_trigger.add_condition(PartOfCoalitionInZone(player_coalition, activation_trigger_zone.id))
|
|
for group in activate_by_trigger:
|
|
activation_trigger.add_action(ActivateGroup(group.id))
|
|
|
|
self.mission.triggerrules.triggers.append(activation_trigger)
|
|
|
|
def _set_allegiances(self, player_coalition: str, enemy_coalition: str):
|
|
for cp in self.game.theater.controlpoints:
|
|
if cp.is_global:
|
|
continue
|
|
self.mission.terrain.airport_by_id(cp.at.id).set_coalition(cp.captured and player_coalition or enemy_coalition)
|
|
|
|
def generate(self, is_quick: bool):
|
|
player_coalition = self.game.player == "USA" and "blue" or "red"
|
|
enemy_coalition = player_coalition == "blue" and "red" or "blue"
|
|
|
|
self.mission.coalition[player_coalition].bullseye = self.conflict.position
|
|
|
|
self._gen_random_time_2()
|
|
self._gen_random_weather()
|
|
self._set_allegiances(player_coalition, enemy_coalition)
|
|
|
|
if not is_quick:
|
|
self._gen_activation_trigger(player_coalition, enemy_coalition)
|