from dcs.ships import (
    Battlecruiser_1144_2_Pyotr_Velikiy,
    Cruiser_1164_Moskva,
    CVN_70_Carl_Vinson,
    CVN_71_Theodore_Roosevelt,
    CVN_72_Abraham_Lincoln,
    CVN_73_George_Washington,
    CVN_74_John_C__Stennis,
    CV_1143_5_Admiral_Kuznetsov,
    CV_1143_5_Admiral_Kuznetsov_2017,
    Frigate_11540_Neustrashimy,
    Corvette_1124_4_Grisha,
    Frigate_1135M_Rezky,
    Corvette_1241_1_Molniya,
    LHA_1_Tarawa,
    FFG_Oliver_Hazzard_Perry,
    CG_Ticonderoga,
    Type_052B_Destroyer,
    Type_052C_Destroyer,
    Type_054A_Frigate,
    DDG_Arleigh_Burke_IIa,
)
from dcs.vehicles import AirDefence

TELARS = {
    AirDefence.SAM_SA_19_Tunguska_Grison,
    AirDefence.SAM_SA_11_Buk_Gadfly_Fire_Dome_TEL,
    AirDefence.SAM_SA_8_Osa_Gecko_TEL,
    AirDefence.SAM_SA_15_Tor_Gauntlet,
    AirDefence.SAM_Roland_ADS,
}

TRACK_RADARS = {
    AirDefence.SAM_SA_6_Kub_Straight_Flush_STR,
    AirDefence.SAM_SA_3_S_125_Low_Blow_TR,
    AirDefence.SAM_SA_10_S_300_Grumble_Flap_Lid_TR,
    AirDefence.SAM_Hawk_TR__AN_MPQ_46,
    AirDefence.SAM_Patriot_STR,
    AirDefence.SAM_SA_2_S_75_Fan_Song_TR,
    AirDefence.SAM_Rapier_Blindfire_TR,
    AirDefence.HQ_7_Self_Propelled_STR,
}

LAUNCHER_TRACKER_PAIRS = {
    AirDefence.SAM_SA_6_Kub_Gainful_TEL: AirDefence.SAM_SA_6_Kub_Straight_Flush_STR,
    AirDefence.SAM_SA_3_S_125_Goa_LN: AirDefence.SAM_SA_3_S_125_Low_Blow_TR,
    AirDefence.SAM_SA_10_S_300_Grumble_TEL_D: AirDefence.SAM_SA_10_S_300_Grumble_Flap_Lid_TR,
    AirDefence.SAM_SA_10_S_300_Grumble_TEL_C: AirDefence.SAM_SA_10_S_300_Grumble_Flap_Lid_TR,
    AirDefence.SAM_Hawk_LN_M192: AirDefence.SAM_Hawk_TR__AN_MPQ_46,
    AirDefence.SAM_Patriot_LN: AirDefence.SAM_Patriot_STR,
    AirDefence.SAM_SA_2_S_75_Guideline_LN: AirDefence.SAM_SA_2_S_75_Fan_Song_TR,
    AirDefence.SAM_Rapier_LN: AirDefence.SAM_Rapier_Blindfire_TR,
    AirDefence.HQ_7_Self_Propelled_LN: AirDefence.HQ_7_Self_Propelled_STR,
}

UNITS_WITH_RADAR = {
    # Radars
    AirDefence.SAM_SA_19_Tunguska_Grison,
    AirDefence.SAM_SA_11_Buk_Gadfly_Fire_Dome_TEL,
    AirDefence.SAM_SA_8_Osa_Gecko_TEL,
    AirDefence.SAM_SA_15_Tor_Gauntlet,
    AirDefence.SPAAA_Gepard,
    AirDefence.SPAAA_Vulcan_M163,
    AirDefence.SAM_Roland_ADS,
    AirDefence.SPAAA_ZSU_23_4_Shilka_Gun_Dish,
    AirDefence.EWR_1L13,
    AirDefence.SAM_SA_6_Kub_Straight_Flush_STR,
    AirDefence.SAM_SA_10_S_300_Grumble_Flap_Lid_TR,
    AirDefence.SAM_SA_10_S_300_Grumble_Clam_Shell_SR,
    AirDefence.EWR_55G6,
    AirDefence.SAM_SA_10_S_300_Grumble_Big_Bird_SR,
    AirDefence.SAM_SA_11_Buk_Gadfly_Snow_Drift_SR,
    AirDefence.MCC_SR_Sborka_Dog_Ear_SR,
    AirDefence.SAM_Hawk_TR__AN_MPQ_46,
    AirDefence.SAM_Hawk_SR__AN_MPQ_50,
    AirDefence.SAM_Patriot_STR,
    AirDefence.SAM_Hawk_CWAR_AN_MPQ_55,
    AirDefence.SAM_P19_Flat_Face_SR__SA_2_3,
    AirDefence.SAM_Roland_EWR,
    AirDefence.SAM_SA_3_S_125_Low_Blow_TR,
    AirDefence.SAM_SA_2_S_75_Fan_Song_TR,
    AirDefence.SAM_Rapier_Blindfire_TR,
    AirDefence.HQ_7_Self_Propelled_LN,
    AirDefence.HQ_7_Self_Propelled_STR,
    AirDefence.EWR_FuMG_401_Freya_LZ,
    AirDefence.EWR_FuSe_65_Würzburg_Riese,
    # Ships
    CVN_70_Carl_Vinson,
    FFG_Oliver_Hazzard_Perry,
    CG_Ticonderoga,
    Corvette_1124_4_Grisha,
    CV_1143_5_Admiral_Kuznetsov,
    Corvette_1241_1_Molniya,
    Cruiser_1164_Moskva,
    Frigate_11540_Neustrashimy,
    Battlecruiser_1144_2_Pyotr_Velikiy,
    Frigate_1135M_Rezky,
    CV_1143_5_Admiral_Kuznetsov_2017,
    CVN_74_John_C__Stennis,
    CVN_71_Theodore_Roosevelt,
    CVN_72_Abraham_Lincoln,
    CVN_73_George_Washington,
    DDG_Arleigh_Burke_IIa,
    LHA_1_Tarawa,
    Type_052B_Destroyer,
    Type_054A_Frigate,
    Type_052C_Destroyer,
}