mirror of
https://github.com/dcs-retribution/dcs-retribution.git
synced 2025-11-10 15:41:24 +00:00
86 lines
3.3 KiB
Python
86 lines
3.3 KiB
Python
import math
|
|
import random
|
|
|
|
from dcs import unitgroup
|
|
from dcs.point import PointAction
|
|
from dcs.unit import Vehicle
|
|
from dcs.unittype import UnitType
|
|
from dcs.vehicles import AirDefence
|
|
|
|
from game import db
|
|
from gen.sam.sam_avenger import AvengerGenerator
|
|
from gen.sam.sam_chaparral import ChaparralGenerator
|
|
from gen.sam.sam_gepard import GepardGenerator
|
|
from gen.sam.sam_hawk import HawkGenerator
|
|
from gen.sam.sam_hq7 import HQ7Generator
|
|
from gen.sam.sam_linebacker import LinebackerGenerator
|
|
from gen.sam.sam_patriot import PatriotGenerator
|
|
from gen.sam.sam_rapier import RapierGenerator
|
|
from gen.sam.sam_roland import RolandGenerator
|
|
from gen.sam.sam_sa10 import SA10Generator
|
|
from gen.sam.sam_sa11 import SA11Generator
|
|
from gen.sam.sam_sa13 import SA13Generator
|
|
from gen.sam.sam_sa15 import SA15Generator
|
|
from gen.sam.sam_sa19 import SA19Generator
|
|
from gen.sam.sam_sa2 import SA2Generator
|
|
from gen.sam.sam_sa3 import SA3Generator
|
|
from gen.sam.sam_sa6 import SA6Generator
|
|
from gen.sam.sam_sa8 import SA8Generator
|
|
from gen.sam.sam_sa9 import SA9Generator
|
|
from gen.sam.sam_vulcan import VulcanGenerator
|
|
from gen.sam.sam_zsu23 import ZSU23Generator
|
|
from gen.sam.sam_zu23 import ZU23Generator
|
|
from gen.sam.sam_zu23_ural import ZU23UralGenerator
|
|
|
|
|
|
def generate_anti_air_group(game, parent_cp, ground_object, faction:str):
|
|
"""
|
|
This generate a SAM group
|
|
:param parentCp: The parent control point
|
|
:param ground_object: The ground object which will own the sam group
|
|
:param country: Owner country
|
|
:return: Nothing, but put the group reference inside the ground object
|
|
"""
|
|
|
|
SAM_MAP = {
|
|
|
|
AirDefence.SAM_Hawk_PCP:HawkGenerator,
|
|
AirDefence.AAA_ZU_23_Closed:ZU23Generator,
|
|
AirDefence.AAA_ZU_23_Insurgent_on_Ural_375:ZU23UralGenerator,
|
|
AirDefence.SPAAA_ZSU_23_4_Shilka:ZSU23Generator,
|
|
AirDefence.AAA_Vulcan_M163: VulcanGenerator,
|
|
AirDefence.SAM_Linebacker_M6: LinebackerGenerator,
|
|
AirDefence.Rapier_FSA_Launcher: RapierGenerator,
|
|
AirDefence.SAM_Avenger_M1097: AvengerGenerator,
|
|
AirDefence.SPAAA_Gepard: GepardGenerator,
|
|
AirDefence.SAM_Roland_ADS: RolandGenerator,
|
|
AirDefence.SAM_Patriot_LN_M901: PatriotGenerator,
|
|
AirDefence.SAM_Patriot_EPP_III: PatriotGenerator,
|
|
AirDefence.SAM_Chaparral_M48: ChaparralGenerator,
|
|
|
|
AirDefence.SAM_SA_2_LN_SM_90: SA2Generator,
|
|
AirDefence.SAM_SA_3_S_125_LN_5P73: SA3Generator,
|
|
AirDefence.SAM_SA_6_Kub_LN_2P25: SA6Generator,
|
|
AirDefence.SAM_SA_8_Osa_9A33: SA8Generator,
|
|
AirDefence.SAM_SA_9_Strela_1_9P31: SA9Generator,
|
|
AirDefence.SAM_SA_10_S_300PS_LN_5P85C: SA10Generator,
|
|
AirDefence.SAM_SA_10_S_300PS_CP_54K6: SA10Generator,
|
|
AirDefence.SAM_SA_11_Buk_LN_9A310M1: SA11Generator,
|
|
AirDefence.SAM_SA_13_Strela_10M3_9A35M3: SA13Generator,
|
|
AirDefence.SAM_SA_15_Tor_9A331: SA15Generator,
|
|
AirDefence.SAM_SA_19_Tunguska_2S6: SA19Generator,
|
|
AirDefence.HQ_7_Self_Propelled_LN: HQ7Generator
|
|
}
|
|
|
|
possible_sams = [u for u in db.FACTIONS[faction]["units"] if u in AirDefence.__dict__.values()]
|
|
if len(possible_sams) > 0:
|
|
sam = random.choice(possible_sams)
|
|
generator = SAM_MAP[sam](game, ground_object, 0)
|
|
generator.generate()
|
|
ground_object.groups = [generator.get_generated_group()]
|
|
|
|
|
|
|
|
|
|
|