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_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_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,
    }

    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()]