configure client slots on scramble + fixes

theater/caucasus.py

@@ -3,6 +3,7 @@ from dcs.terrain import caucasus
 from .conflicttheater import *
 from .base import *
 
+
 class CaucasusTheater(ConflictTheater):
     kutaisi = ControlPoint(caucasus.Kutaisi, ALL_RADIALS, SIZE_SMALL, IMPORTANCE_LOW)
     senaki = ControlPoint(caucasus.Senaki, ALL_RADIALS, SIZE_REGULAR, IMPORTANCE_LOW)

theater/start_generator.py

@@ -3,9 +3,33 @@ import random
 
 import dcs
 
+from shop import db
 from theater.controlpoint import *
 from theater.base import *
 from theater.conflicttheater import *
 
-def generate_initial(theater: ConflictTheater):
-    pass
+UNIT_VARIETY = 2
+UNIT_AMOUNT_FACTOR = 1
+
+
+def generate_initial(theater: ConflictTheater, enemy: str):
+    for cp in theater.enemy_bases():
+        if cp.captured:
+            continue
+
+        for task in [CAP, FighterSweep, CAS, AirDefence]:
+            suitable_unittypes = db.find_unittype(task, enemy)
+            suitable_unittypes.sort(key=lambda x: db.PRICES[x])
+
+            importance = cp.importance * 10 - 10
+            units_idx_start = int(importance * UNIT_VARIETY)
+            units_idx_end = units_idx_start + UNIT_VARIETY
+
+            range_start = min(len(suitable_unittypes)-1, units_idx_start)
+            range_end = min(len(suitable_unittypes), units_idx_end)
+            unittypes = suitable_unittypes[range_start:range_end]
+
+            typecount = max(math.floor(importance * UNIT_AMOUNT_FACTOR), 1)
+            units = {unittype: typecount for unittype in unittypes}
+            print("{} - {}".format(cp.name, units))
+            cp.base.commision_units(units)