mirror of
https://github.com/dcs-retribution/dcs-retribution.git
synced 2025-11-10 15:41:24 +00:00
25 lines
1.0 KiB
Python
25 lines
1.0 KiB
Python
from theater.base import *
|
|
from theater.conflicttheater import *
|
|
|
|
UNIT_VARIETY = 3
|
|
UNIT_AMOUNT_FACTOR = 16
|
|
|
|
|
|
def generate_initial(theater: ConflictTheater, enemy: str):
|
|
for cp in theater.enemy_points():
|
|
if cp.captured:
|
|
continue
|
|
|
|
for task in [CAP, FighterSweep, CAS, AirDefence]:
|
|
assert cp.importance <= IMPORTANCE_HIGH, "invalid importance {}".format(cp.importance)
|
|
assert cp.importance >= IMPORTANCE_LOW, "invalid importance {}".format(cp.importance)
|
|
|
|
importance_factor = (cp.importance - IMPORTANCE_LOW) / (IMPORTANCE_HIGH - IMPORTANCE_LOW)
|
|
variety = int(UNIT_VARIETY + UNIT_VARIETY * importance_factor / 2)
|
|
unittypes = db.choose_units(task, importance_factor, variety, enemy)
|
|
|
|
count = max(int(importance_factor * UNIT_AMOUNT_FACTOR), 1)
|
|
count_per_type = max(int(float(count) / len(unittypes)), 1)
|
|
for unit_type in unittypes:
|
|
cp.base.commision_units({unit_type: count_per_type})
|