mirror of
https://github.com/dcs-retribution/dcs-retribution.git
synced 2025-11-10 15:41:24 +00:00
119 lines
2.7 KiB
Python
119 lines
2.7 KiB
Python
import typing
|
|
|
|
from dcs.vehicles import *
|
|
from dcs.unitgroup import *
|
|
from dcs.ships import *
|
|
from dcs.planes import *
|
|
from dcs.task import *
|
|
from dcs.unittype import *
|
|
|
|
PRICES = {
|
|
# planes
|
|
|
|
Su_25T: 11,
|
|
Su_25: 11,
|
|
A_10A: 18,
|
|
A_10C: 20,
|
|
|
|
FA_18C_hornet: 18,
|
|
AV8BNA: 15,
|
|
|
|
Su_27: 30,
|
|
Su_33: 33,
|
|
F_15C: 30,
|
|
M_2000C: 11,
|
|
|
|
MiG_21Bis: 13,
|
|
MiG_29A: 23,
|
|
|
|
IL_76MD: 13,
|
|
S_3B_Tanker: 13,
|
|
|
|
A_50: 8,
|
|
E_3A: 8,
|
|
|
|
# armor
|
|
|
|
Armor.MBT_T_55: 4,
|
|
Armor.MBT_T_80U: 8,
|
|
Armor.MBT_T_90: 10,
|
|
|
|
Armor.MBT_M60A3_Patton: 6,
|
|
Armor.MBT_M1A2_Abrams: 9,
|
|
|
|
Armor.ATGM_M1134_Stryker: 6,
|
|
Armor.APC_BTR_80: 6,
|
|
|
|
AirDefence.AAA_ZU_23_on_Ural_375: 4,
|
|
AirDefence.SAM_Avenger_M1097: 10,
|
|
}
|
|
|
|
UNIT_BY_TASK = {
|
|
FighterSweep: [Su_27, Su_33, FA_18C_hornet, F_15C, MiG_21Bis, MiG_29A, F_A_18C, AV8BNA],
|
|
CAS: [Su_25T, A_10A, A_10C, ],
|
|
CAP: [Armor.MBT_T_90, Armor.MBT_T_80U, Armor.MBT_T_55, Armor.MBT_M1A2_Abrams, Armor.MBT_M60A3_Patton, Armor.ATGM_M1134_Stryker, Armor.APC_BTR_80, ],
|
|
AirDefence: [AirDefence.AAA_ZU_23_on_Ural_375, AirDefence.SAM_Avenger_M1097 ],
|
|
Transport: [IL_76MD, S_3B_Tanker, ],
|
|
Carriage: [CVN_74_John_C__Stennis, CV_1143_5_Admiral_Kuznetsov, ],
|
|
AWACS: [E_3A, A_50, ],
|
|
}
|
|
|
|
UNIT_BY_COUNTRY = {
|
|
"Russia": [
|
|
Su_25T,
|
|
Su_27,
|
|
Su_33,
|
|
Su_25,
|
|
MiG_21Bis,
|
|
MiG_29A,
|
|
A_50,
|
|
AirDefence.AAA_ZU_23_on_Ural_375,
|
|
Armor.APC_BTR_80,
|
|
Armor.MBT_T_90,
|
|
Armor.MBT_T_80U,
|
|
Armor.MBT_T_55,
|
|
IL_76MD,
|
|
CV_1143_5_Admiral_Kuznetsov],
|
|
|
|
"USA": [F_15C,
|
|
A_10C,
|
|
FA_18C_hornet,
|
|
AV8BNA,
|
|
E_3A,
|
|
Armor.MBT_M1A2_Abrams,
|
|
Armor.MBT_M60A3_Patton,
|
|
Armor.ATGM_M1134_Stryker,
|
|
S_3B_Tanker,
|
|
AirDefence.SAM_Avenger_M1097,
|
|
CVN_74_John_C__Stennis],
|
|
}
|
|
|
|
UnitsDict = typing.Dict[UnitType, int]
|
|
PlaneDict = typing.Dict[FlyingType, int]
|
|
ArmorDict = typing.Dict[VehicleType, int]
|
|
AirDefenseDict = typing.Dict[AirDefence, int]
|
|
StartingPosition = typing.Optional[typing.Union[ShipGroup, Airport, Point]]
|
|
|
|
|
|
def unit_task(unit: UnitType) -> Task:
|
|
for task, units in UNIT_BY_TASK.items():
|
|
if unit in units:
|
|
return task
|
|
|
|
assert False
|
|
|
|
|
|
def find_unittype(for_task: Task, country_name: str) -> typing.List[UnitType]:
|
|
return [x for x in UNIT_BY_TASK[for_task] if x in UNIT_BY_COUNTRY[country_name]]
|
|
|
|
|
|
def unit_type_name(unit_type) -> str:
|
|
return unit_type.id and unit_type.id or unit_type.name
|
|
|
|
|
|
def task_name(task) -> str:
|
|
if task == AirDefence:
|
|
return "AirDefence"
|
|
else:
|
|
return task.name
|