from typing import Set

from dcs import task
from dcs.liveries_scanner import Liveries
from dcs.planes import PlaneType
from dcs.weapons_data import Weapons
from dcs import unittype

from game.modsupport import vehiclemod
from game.modsupport import planemod
from pydcs_extensions.weapon_injector import inject_weapons


class SWWeapons:
    PROTONB = {
        "clsid": "{PROTONB}",
        "name": "PROTONB",
        "weight": 50,
    }
    PROTONBGG = {
        "clsid": "{PROTONBGG}",
        "name": "PROTONBGG",
        "weight": 1000,
    }
    PROTONF = {
        "clsid": "{PROTONF}",
        "name": "PROTONF",
        "weight": 50,
    }
    PROTONM = {
        "clsid": "{PROTONM}",
        "name": "PROTONM",
        "weight": 50,
    }
    PROTONM1 = {
        "clsid": "{PROTONM1}",
        "name": "PROTONM1",
        "weight": 50,
    }
    PROTONM2 = {
        "clsid": "{PROTONM2}",
        "name": "PROTONM2",
        "weight": 50,
    }
    PROTONM3 = {
        "clsid": "{PROTONM3}",
        "name": "PROTONM3",
        "weight": 0.1,
    }
    PROTONMissile = {
        "clsid": "{PROTONMissile}",
        "name": "PROTONMissile",
        "weight": 50,
    }
    ENERGY_CELL = {
        "clsid": "{AFUEL}",
        "name": "ENERGY CELL",
        "weight": 1005,
    }
    ENERGY_CELL_ = {
        "clsid": "{TIEFUEL}",
        "name": "ENERGY CELL",
        "weight": 1005,
    }
    ENERGY_CELL__ = {
        "clsid": "{HUNTFUEL}",
        "name": "ENERGY CELL",
        "weight": 1005,
    }
    ENERGY_CELL___ = {
        "clsid": "{TIRFUEL}",
        "name": "ENERGY CELL",
        "weight": 1005,
    }
    ENERGY_CELL____ = {
        "clsid": "{XFUELTANK}",
        "name": "ENERGY CELL",
        "weight": 2005,
    }


inject_weapons(SWWeapons)


@planemod
class XWING(PlaneType):
    id = "XWING"
    flyable = True
    height = 5.63
    width = 6
    length = 19.43
    fuel_max = 5000
    max_speed = 1152.396
    chaff = 5000
    flare = 5000
    charge_total = 10000
    chaff_charge_size = 1
    flare_charge_size = 1
    category = "Interceptor"  # {78EFB7A2-FD52-4b57-A6A6-3BF0E1D6555F}
    radio_frequency = 127.5

    livery_name = "XWING"  # from type
    Liveries = Liveries()[livery_name]

    class Pylon1:
        PROTONM = (1, Weapons.PROTONM)

    class Pylon2:
        PROTONM = (2, Weapons.PROTONM)
        ENERGY_CELL____ = (2, Weapons.ENERGY_CELL____)

    class Pylon3:
        PROTONM = (3, Weapons.PROTONM)

    class Pylon4:
        PROTONM = (4, Weapons.PROTONM)

    class Pylon5:
        PROTONM = (5, Weapons.PROTONM)

    class Pylon6:
        PROTONM = (6, Weapons.PROTONM)

    class Pylon10:
        ENERGY_CELL____ = (10, Weapons.ENERGY_CELL____)

    pylons: Set[int] = {1, 2, 3, 4, 5, 6, 10}

    tasks = [
        task.CAP,
        task.Escort,
        task.FighterSweep,
        task.Intercept,
        task.Reconnaissance,
    ]
    task_default = task.CAP


@planemod
class XWINGAI(PlaneType):
    id = "XWINGAI"
    height = 5.63
    width = 6
    length = 19.43
    fuel_max = 10000
    max_speed = 1152.396
    chaff = 5000
    flare = 5000
    charge_total = 10000
    chaff_charge_size = 1
    flare_charge_size = 1
    category = "Interceptor"  # {78EFB7A2-FD52-4b57-A6A6-3BF0E1D6555F}
    radio_frequency = 127.5

    livery_name = "XWINGAI"  # from type
    Liveries = Liveries()[livery_name]

    class Pylon1:
        PROTONM = (1, Weapons.PROTONM)

    class Pylon2:
        PROTONM = (2, Weapons.PROTONM)

    ENERGY_CELL____ = (2, Weapons.ENERGY_CELL____)

    class Pylon3:
        PROTONM = (3, Weapons.PROTONM)

    class Pylon4:
        PROTONM = (4, Weapons.PROTONM)

    class Pylon5:
        PROTONM = (5, Weapons.PROTONM)

    class Pylon6:
        PROTONM = (6, Weapons.PROTONM)

    class Pylon10:
        ENERGY_CELL____ = (10, Weapons.ENERGY_CELL____)

    pylons: Set[int] = {1, 2, 3, 4, 5, 6, 10}

    tasks = [
        task.CAP,
        task.Escort,
        task.FighterSweep,
        task.Intercept,
        task.Reconnaissance,
    ]
    task_default = task.CAP


@planemod
class YWINGA(PlaneType):
    id = "YWINGA"
    height = 5.63
    width = 20
    length = 19.43
    fuel_max = 10000
    max_speed = 1260.396
    chaff = 5000
    flare = 5000
    charge_total = 10000
    chaff_charge_size = 1
    flare_charge_size = 1
    category = "Interceptor"  # {78EFB7A2-FD52-4b57-A6A6-3BF0E1D6555F}
    radio_frequency = 127.5

    livery_name = "YWINGA"  # from type
    Liveries = Liveries()[livery_name]

    class Pylon1:
        LAU_88_with_3_x_AGM_65D___Maverick_D__IIR_ASM_ = (
            1,
            Weapons.LAU_88_with_3_x_AGM_65D___Maverick_D__IIR_ASM_,
        )
        PROTONMissile = (
            1,
            Weapons.PROTONMissile,
        )
        Mk_84___2000lb_GP_Bomb_LD = (
            1,
            Weapons.Mk_84___2000lb_GP_Bomb_LD,
        )
        AGM_88C_HARM___High_Speed_Anti_Radiation_Missile = (
            1,
            Weapons.AGM_88C_HARM___High_Speed_Anti_Radiation_Missile,
        )

    class Pylon2:
        LAU_88_with_3_x_AGM_65D___Maverick_D__IIR_ASM_ = (
            2,
            Weapons.LAU_88_with_3_x_AGM_65D___Maverick_D__IIR_ASM_,
        )
        PROTONMissile = (
            2,
            Weapons.PROTONMissile,
        )
        Mk_84___2000lb_GP_Bomb_LD = (
            2,
            Weapons.Mk_84___2000lb_GP_Bomb_LD,
        )
        AGM_88C_HARM___High_Speed_Anti_Radiation_Missile = (
            2,
            Weapons.AGM_88C_HARM___High_Speed_Anti_Radiation_Missile,
        )

    class Pylon3:
        PROTONBGG = (
            3,
            Weapons.PROTONBGG,
        )
        Mk_84___2000lb_GP_Bomb_LD = (
            3,
            Weapons.Mk_84___2000lb_GP_Bomb_LD,
        )
        AGM_88C_HARM___High_Speed_Anti_Radiation_Missile = (
            3,
            Weapons.AGM_88C_HARM___High_Speed_Anti_Radiation_Missile,
        )

    class Pylon4:
        LAU_88_with_3_x_AGM_65D___Maverick_D__IIR_ASM_ = (
            4,
            Weapons.LAU_88_with_3_x_AGM_65D___Maverick_D__IIR_ASM_,
        )
        PROTONMissile = (
            4,
            Weapons.PROTONMissile,
        )
        Mk_84___2000lb_GP_Bomb_LD = (
            4,
            Weapons.Mk_84___2000lb_GP_Bomb_LD,
        )
        AGM_88C_HARM___High_Speed_Anti_Radiation_Missile = (
            4,
            Weapons.AGM_88C_HARM___High_Speed_Anti_Radiation_Missile,
        )

    class Pylon5:
        LAU_88_with_3_x_AGM_65D___Maverick_D__IIR_ASM_ = (
            5,
            Weapons.LAU_88_with_3_x_AGM_65D___Maverick_D__IIR_ASM_,
        )
        PROTONMissile = (
            5,
            Weapons.PROTONMissile,
        )
        Mk_84___2000lb_GP_Bomb_LD = (
            5,
            Weapons.Mk_84___2000lb_GP_Bomb_LD,
        )
        AGM_88C_HARM___High_Speed_Anti_Radiation_Missile = (
            5,
            Weapons.AGM_88C_HARM___High_Speed_Anti_Radiation_Missile,
        )

    class Pylon6:
        PROTONMissile = (6, Weapons.PROTONMissile)
        LAU_88_with_3_x_AGM_65D___Maverick_D__IIR_ASM_ = (
            6,
            Weapons.LAU_88_with_3_x_AGM_65D___Maverick_D__IIR_ASM_,
        )
        Mk_84___2000lb_GP_Bomb_LD = (
            6,
            Weapons.Mk_84___2000lb_GP_Bomb_LD,
        )
        AGM_88C_HARM___High_Speed_Anti_Radiation_Missile = (
            6,
            Weapons.AGM_88C_HARM___High_Speed_Anti_Radiation_Missile,
        )

    class Pylon7:
        ENERGY_CELL___ = (7, Weapons.ENERGY_CELL___)

    class Pylon8:
        ENERGY_CELL___ = (8, Weapons.ENERGY_CELL___)

    pylons: Set[int] = {1, 2, 3, 4, 5, 6, 7, 8}

    tasks = [
        task.CAS,
        task.Escort,
        task.FighterSweep,
        task.Intercept,
        task.Reconnaissance,
    ]
    task_default = task.CAS


@planemod
class YWING(PlaneType):
    id = "YWING"
    flyable = True
    height = 5.63
    width = 10
    length = 19.43
    fuel_max = 10000
    max_speed = 2649.996
    chaff = 5000
    flare = 5000
    charge_total = 10000
    chaff_charge_size = 1
    flare_charge_size = 1
    category = "Interceptor"  # {78EFB7A2-FD52-4b57-A6A6-3BF0E1D6555F}
    radio_frequency = 127.5

    livery_name = "YWING"  # from type
    Liveries = Liveries()[livery_name]

    class Pylon1:
        PROTONB = (1, Weapons.PROTONB)

    class Pylon2:
        PROTONB = (2, Weapons.PROTONB)

    class Pylon3:
        PROTONB = (3, Weapons.PROTONB)

    class Pylon4:
        PROTONB = (4, Weapons.PROTONB)

    class Pylon5:
        PROTONB = (5, Weapons.PROTONB)

    class Pylon6:
        PROTONB = (6, Weapons.PROTONB)

    class Pylon7:
        ENERGY_CELL___ = (7, Weapons.ENERGY_CELL___)

    class Pylon8:
        ENERGY_CELL___ = (8, Weapons.ENERGY_CELL___)

    pylons: Set[int] = {1, 2, 3, 4, 5, 6, 7, 8}

    tasks = [
        task.CAS,
        task.Escort,
        task.FighterSweep,
        task.Intercept,
        task.Reconnaissance,
    ]
    task_default = task.CAS


@planemod
class AWING(PlaneType):
    id = "AWING"
    flyable = True
    height = 5.63
    width = 10
    length = 19.43
    fuel_max = 5000
    max_speed = 1440.396
    chaff = 5000
    flare = 5000
    charge_total = 10000
    chaff_charge_size = 1
    flare_charge_size = 1
    category = "Interceptor"  # {78EFB7A2-FD52-4b57-A6A6-3BF0E1D6555F}
    radio_frequency = 127.5

    livery_name = "AWING"  # from type
    Liveries = Liveries()[livery_name]

    class Pylon1:
        PROTONM1 = (1, Weapons.PROTONM1)

    class Pylon2:
        ENERGY_CELL = (2, Weapons.ENERGY_CELL)

    class Pylon3:
        PROTONM1 = (3, Weapons.PROTONM1)

    class Pylon10:
        ENERGY_CELL = (10, Weapons.ENERGY_CELL)

    pylons: Set[int] = {1, 2, 3, 10}

    tasks = [
        task.CAP,
        task.Escort,
        task.FighterSweep,
        task.Intercept,
        task.Reconnaissance,
    ]
    task_default = task.CAP


@planemod
class AWINGA(PlaneType):
    id = "AWINGA"
    height = 5.63
    width = 10
    length = 19.43
    fuel_max = 10000
    max_speed = 1440.396
    chaff = 5000
    flare = 5000
    charge_total = 10000
    chaff_charge_size = 1
    flare_charge_size = 1
    category = "Interceptor"  # {78EFB7A2-FD52-4b57-A6A6-3BF0E1D6555F}
    radio_frequency = 127.5

    livery_name = "AWINGA"  # from type
    Liveries = Liveries()[livery_name]

    class Pylon1:
        PROTONM1 = (1, Weapons.PROTONM1)

    class Pylon2:
        ENERGY_CELL = (2, Weapons.ENERGY_CELL)

    class Pylon3:
        PROTONM1 = (3, Weapons.PROTONM1)

    class Pylon10:
        ENERGY_CELL = (10, Weapons.ENERGY_CELL)

    pylons: Set[int] = {1, 2, 3, 10}

    tasks = [
        task.CAP,
        task.Escort,
        task.FighterSweep,
        task.Intercept,
        task.Reconnaissance,
    ]
    task_default = task.CAP


@planemod
class CORVETTE(PlaneType):
    id = "CORVETTE"
    flyable = True
    height = 5.63
    width = 30
    length = 19.43
    fuel_max = 11000
    max_speed = 792.396
    chaff = 5000
    flare = 5000
    charge_total = 10000
    chaff_charge_size = 1
    flare_charge_size = 1
    category = "Interceptor"  # {78EFB7A2-FD52-4b57-A6A6-3BF0E1D6555F}
    radio_frequency = 127.5

    livery_name = "CORVETTE"  # from type
    Liveries = Liveries()[livery_name]

    class Pylon3:
        PROTONM3 = (3, Weapons.PROTONM3)

    class Pylon4:
        PROTONM3 = (4, Weapons.PROTONM3)

    class Pylon5:
        PROTONM3 = (5, Weapons.PROTONM3)

    class Pylon7:
        PROTONM3 = (7, Weapons.PROTONM3)

    class Pylon8:
        PROTONM3 = (8, Weapons.PROTONM3)

    class Pylon9:
        PROTONM3 = (9, Weapons.PROTONM3)

    pylons: Set[int] = {3, 4, 5, 7, 8, 9}

    tasks = [
        task.CAP,
        task.Escort,
        task.FighterSweep,
        task.Intercept,
        task.Reconnaissance,
    ]
    task_default = task.CAP


@planemod
class CORVETTEA(PlaneType):
    id = "CORVETTEA"
    height = 5.63
    width = 30
    length = 19.43
    fuel_max = 11000
    max_speed = 648.396
    chaff = 5000
    flare = 5000
    charge_total = 10000
    chaff_charge_size = 1
    flare_charge_size = 1
    category = "Interceptor"  # {78EFB7A2-FD52-4b57-A6A6-3BF0E1D6555F}
    radio_frequency = 127.5

    livery_name = "CORVETTEA"  # from type
    Liveries = Liveries()[livery_name]

    class Pylon3:
        PROTONM3 = (3, Weapons.PROTONM3)

    class Pylon4:
        PROTONM3 = (4, Weapons.PROTONM3)

    class Pylon5:
        PROTONM3 = (5, Weapons.PROTONM3)

    class Pylon7:
        PROTONM3 = (7, Weapons.PROTONM3)

    class Pylon8:
        PROTONM3 = (8, Weapons.PROTONM3)

    class Pylon9:
        PROTONM3 = (9, Weapons.PROTONM3)

    pylons: Set[int] = {3, 4, 5, 7, 8, 9}

    tasks = [
        task.CAP,
        task.Escort,
        task.FighterSweep,
        task.Intercept,
        task.Reconnaissance,
    ]
    task_default = task.CAP


@planemod
class FAUCON(PlaneType):
    id = "FAUCON"
    flyable = True
    height = 5.63
    width = 10
    length = 19.43
    fuel_max = 10000
    max_speed = 2649.996
    chaff = 5000
    flare = 5000
    charge_total = 10000
    chaff_charge_size = 1
    flare_charge_size = 1
    category = "Interceptor"  # {78EFB7A2-FD52-4b57-A6A6-3BF0E1D6555F}
    radio_frequency = 127.5

    livery_name = "FAUCON"  # from type
    Liveries = Liveries()[livery_name]

    class Pylon1:
        PROTONF = (1, Weapons.PROTONF)

    pylons: Set[int] = {1}

    tasks = [
        task.CAP,
        task.Escort,
        task.FighterSweep,
        task.Intercept,
        task.Reconnaissance,
    ]
    task_default = task.FighterSweep


@planemod
class FAUCON_AI(PlaneType):
    id = "FAUCON_AI"
    height = 5.63
    width = 20
    length = 19.43
    fuel_max = 10000
    max_speed = 2649.996
    chaff = 5000
    flare = 5000
    charge_total = 10000
    chaff_charge_size = 1
    flare_charge_size = 1
    category = "Interceptor"  # {78EFB7A2-FD52-4b57-A6A6-3BF0E1D6555F}
    radio_frequency = 127.5

    livery_name = "FAUCON_AI"  # from type
    Liveries = Liveries()[livery_name]

    class Pylon1:
        PROTONF = (1, Weapons.PROTONF)

    pylons: Set[int] = {1}

    tasks = [
        task.CAP,
        task.Escort,
        task.FighterSweep,
        task.Intercept,
        task.Reconnaissance,
    ]
    task_default = task.FighterSweep


@planemod
class TIE(PlaneType):
    id = "TIE"
    flyable = True
    height = 5.63
    width = 10
    length = 19.43
    fuel_max = 8000
    max_speed = 2649.996
    chaff = 5000
    flare = 5000
    charge_total = 10000
    chaff_charge_size = 1
    flare_charge_size = 1
    category = "Interceptor"  # {78EFB7A2-FD52-4b57-A6A6-3BF0E1D6555F}
    radio_frequency = 127.5

    livery_name = "TIE"  # from type
    Liveries = Liveries()[livery_name]

    class Pylon2:
        ENERGY_CELL_ = (2, Weapons.ENERGY_CELL_)

    class Pylon10:
        ENERGY_CELL_ = (10, Weapons.ENERGY_CELL_)

    pylons: Set[int] = {2, 10}

    tasks = [
        task.CAP,
        task.Escort,
        task.FighterSweep,
        task.Intercept,
        task.Reconnaissance,
    ]
    task_default = task.FighterSweep


@planemod
class TIE_AI(PlaneType):
    id = "TIE_AI"
    height = 5.63
    width = 5
    length = 19.43
    fuel_max = 8000
    max_speed = 1440
    chaff = 5000
    flare = 5000
    charge_total = 10000
    chaff_charge_size = 1
    flare_charge_size = 1
    category = "Interceptor"  # {78EFB7A2-FD52-4b57-A6A6-3BF0E1D6555F}
    radio_frequency = 127.5

    livery_name = "TIE_AI"  # from type
    Liveries = Liveries()[livery_name]

    class Pylon2:
        ENERGY_CELL_ = (2, Weapons.ENERGY_CELL_)

    class Pylon10:
        ENERGY_CELL_ = (10, Weapons.ENERGY_CELL_)

    pylons: Set[int] = {2, 10}

    tasks = [
        task.CAP,
        task.Escort,
        task.FighterSweep,
        task.Intercept,
        task.Reconnaissance,
    ]
    task_default = task.FighterSweep


@planemod
class HUNTER(PlaneType):
    id = "HUNTER"
    flyable = True
    height = 5.63
    width = 6
    length = 19.43
    fuel_max = 10000
    max_speed = 2649.996
    chaff = 5000
    flare = 5000
    charge_total = 10000
    chaff_charge_size = 1
    flare_charge_size = 1
    category = "Interceptor"  # {78EFB7A2-FD52-4b57-A6A6-3BF0E1D6555F}
    radio_frequency = 127.5

    livery_name = "HUNTER"  # from type
    Liveries = Liveries()[livery_name]

    class Pylon1:
        PROTONM2 = (1, Weapons.PROTONM2)

    class Pylon2:
        ENERGY_CELL__ = (2, Weapons.ENERGY_CELL__)

    class Pylon3:
        PROTONM2 = (3, Weapons.PROTONM2)

    class Pylon10:
        ENERGY_CELL__ = (10, Weapons.ENERGY_CELL__)

    pylons: Set[int] = {1, 2, 3, 10}

    tasks = [
        task.CAP,
        task.Escort,
        task.FighterSweep,
        task.Intercept,
        task.Reconnaissance,
    ]
    task_default = task.CAP


@planemod
class HUNTERA(PlaneType):
    id = "HUNTERA"
    height = 5.63
    width = 5
    length = 19.43
    fuel_max = 10000
    max_speed = 1512.396
    chaff = 5000
    flare = 5000
    charge_total = 10000
    chaff_charge_size = 1
    flare_charge_size = 1
    category = "Interceptor"  # {78EFB7A2-FD52-4b57-A6A6-3BF0E1D6555F}
    radio_frequency = 127.5

    livery_name = "HUNTERA"  # from type
    Liveries = Liveries()[livery_name]

    class Pylon1:
        PROTONM2 = (1, Weapons.PROTONM2)

    class Pylon2:
        ENERGY_CELL__ = (2, Weapons.ENERGY_CELL__)

    class Pylon3:
        PROTONM2 = (3, Weapons.PROTONM2)

    class Pylon10:
        ENERGY_CELL__ = (10, Weapons.ENERGY_CELL__)

    pylons: Set[int] = {1, 2, 3, 10}

    tasks = [
        task.CAP,
        task.Escort,
        task.FighterSweep,
        task.Intercept,
        task.Reconnaissance,
    ]
    task_default = task.CAP


@planemod
class TIE_INTER(PlaneType):
    id = "TIE_INTER"
    flyable = True
    height = 5.63
    width = 6
    length = 19.43
    fuel_max = 5000
    max_speed = 1440.396
    chaff = 5000
    flare = 5000
    charge_total = 10000
    chaff_charge_size = 1
    flare_charge_size = 1
    category = "Interceptor"  # {78EFB7A2-FD52-4b57-A6A6-3BF0E1D6555F}
    radio_frequency = 127.5

    livery_name = "TIE_INTER"  # from type
    Liveries = Liveries()[livery_name]

    class Pylon1:
        PROTONM1 = (1, Weapons.PROTONM1)

    class Pylon2:
        ENERGY_CELL___ = (2, Weapons.ENERGY_CELL___)

    class Pylon3:
        PROTONM1 = (3, Weapons.PROTONM1)

    class Pylon10:
        ENERGY_CELL___ = (10, Weapons.ENERGY_CELL___)

    pylons: Set[int] = {1, 2, 3, 10}

    tasks = [
        task.CAP,
        task.Escort,
        task.FighterSweep,
        task.Intercept,
        task.Reconnaissance,
    ]
    task_default = task.CAP


@planemod
class TIE_INTERA(PlaneType):
    id = "TIE_INTERA"
    height = 5.63
    width = 10
    length = 19.43
    fuel_max = 5000
    max_speed = 1548.396
    chaff = 5000
    flare = 5000
    charge_total = 10000
    chaff_charge_size = 1
    flare_charge_size = 1
    category = "Interceptor"  # {78EFB7A2-FD52-4b57-A6A6-3BF0E1D6555F}
    radio_frequency = 127.5

    livery_name = "TIE_INTERA"  # from type
    Liveries = Liveries()[livery_name]

    class Pylon1:
        PROTONM1 = (1, Weapons.PROTONM1)

    class Pylon2:
        ENERGY_CELL___ = (2, Weapons.ENERGY_CELL___)

    class Pylon3:
        PROTONM1 = (3, Weapons.PROTONM1)

    class Pylon10:
        ENERGY_CELL___ = (10, Weapons.ENERGY_CELL___)

    pylons: Set[int] = {1, 2, 3, 10}

    tasks = [
        task.CAP,
        task.Escort,
        task.FighterSweep,
        task.Intercept,
        task.Reconnaissance,
    ]
    task_default = task.CAP


@planemod
class naboo_starfighter(PlaneType):
    id = "naboo_starfighter"
    flyable = True
    height = 5.63
    width = 5
    length = 10.43
    fuel_max = 9249
    max_speed = 1152.396
    chaff = 0
    flare = 500
    charge_total = 500
    chaff_charge_size = 1
    flare_charge_size = 1
    category = "Interceptor"  # {78EFB7A2-FD52-4b57-A6A6-3BF0E1D6555F}
    radio_frequency = 127.5

    livery_name = "NABOO_STARFIGHTER"  # from type
    Liveries = Liveries()[livery_name]

    class Pylon1:
        PROTONM = (1, Weapons.PROTONM)

    class Pylon2:
        PROTONM = (2, Weapons.PROTONM)
        ENERGY_CELL____ = (2, Weapons.ENERGY_CELL____)

    class Pylon3:
        PROTONM = (3, Weapons.PROTONM)

    class Pylon4:
        PROTONM = (4, Weapons.PROTONM)

    class Pylon5:
        PROTONM = (5, Weapons.PROTONM)

    class Pylon6:
        PROTONM = (6, Weapons.PROTONM)

    class Pylon10:
        ENERGY_CELL____ = (10, Weapons.ENERGY_CELL____)

    pylons: Set[int] = {1, 2, 3, 4, 5, 6, 10}

    tasks = [
        task.CAP,
        task.Escort,
        task.FighterSweep,
        task.Intercept,
        task.Reconnaissance,
        task.GroundAttack,
        task.CAS,
        task.AFAC,
        task.RunwayAttack,
    ]
    task_default = task.FighterSweep


@planemod
class naboo_starfighter_AI(PlaneType):
    id = "naboo_starfighter_AI"
    height = 5.63
    width = 6
    length = 19.43
    fuel_max = 10000
    max_speed = 1152.396
    chaff = 0
    flare = 500
    charge_total = 500
    chaff_charge_size = 1
    flare_charge_size = 1
    category = "Interceptor"  # {78EFB7A2-FD52-4b57-A6A6-3BF0E1D6555F}
    radio_frequency = 127.5

    livery_name = "NABOO_STARFIGHTER_AI"  # from type
    Liveries = Liveries()[livery_name]

    class Pylon1:
        PROTONM = (1, Weapons.PROTONM)

    class Pylon2:
        PROTONM = (2, Weapons.PROTONM)
        ENERGY_CELL____ = (2, Weapons.ENERGY_CELL____)

    class Pylon3:
        PROTONM = (3, Weapons.PROTONM)

    class Pylon4:
        PROTONM = (4, Weapons.PROTONM)

    class Pylon5:
        PROTONM = (5, Weapons.PROTONM)

    class Pylon6:
        PROTONM = (6, Weapons.PROTONM)

    class Pylon10:
        ENERGY_CELL____ = (10, Weapons.ENERGY_CELL____)

    pylons: Set[int] = {1, 2, 3, 4, 5, 6, 10}

    tasks = [
        task.CAP,
        task.Escort,
        task.FighterSweep,
        task.Intercept,
        task.Reconnaissance,
        task.GroundAttack,
        task.CAS,
        task.AFAC,
        task.RunwayAttack,
    ]
    task_default = task.FighterSweep


@planemod
class tie_bomber_2(PlaneType):
    id = "tie_bomber_2"
    flyable = True
    height = 5.63
    width = 10
    length = 19.43
    fuel_max = 10000
    max_speed = 2649.996
    chaff = 5000
    flare = 5000
    charge_total = 10000
    chaff_charge_size = 1
    flare_charge_size = 1
    category = "Interceptor"  # {78EFB7A2-FD52-4b57-A6A6-3BF0E1D6555F}
    radio_frequency = 127.5

    livery_name = "TIE_BOMBER_2"  # from type
    Liveries = Liveries()[livery_name]

    class Pylon1:
        PROTONM1 = (1, Weapons.PROTONM1)
        TER_9A_with_3_x_Mk_82___500lb_GP_Bomb_LD = (
            1,
            Weapons.TER_9A_with_3_x_Mk_82___500lb_GP_Bomb_LD,
        )
        Mk_84___2000lb_GP_Bomb_LD = (1, Weapons.Mk_84___2000lb_GP_Bomb_LD)

    class Pylon2:
        ENERGY_CELL___ = (2, Weapons.ENERGY_CELL___)

    class Pylon3:
        PROTONM1 = (3, Weapons.PROTONM1)
        Mk_84___2000lb_GP_Bomb_LD = (3, Weapons.Mk_84___2000lb_GP_Bomb_LD)
        TER_9A_with_3_x_Mk_82___500lb_GP_Bomb_LD = (
            3,
            Weapons.TER_9A_with_3_x_Mk_82___500lb_GP_Bomb_LD,
        )

    class Pylon4:
        TER_9A_with_3_x_Mk_82___500lb_GP_Bomb_LD = (
            4,
            Weapons.TER_9A_with_3_x_Mk_82___500lb_GP_Bomb_LD,
        )
        PROTONM2 = (4, Weapons.PROTONM2)
        Mk_84___2000lb_GP_Bomb_LD = (4, Weapons.Mk_84___2000lb_GP_Bomb_LD)

    class Pylon5:
        TER_9A_with_3_x_Mk_82___500lb_GP_Bomb_LD = (
            5,
            Weapons.TER_9A_with_3_x_Mk_82___500lb_GP_Bomb_LD,
        )
        PROTONM2 = (5, Weapons.PROTONM2)
        Mk_84___2000lb_GP_Bomb_LD = (5, Weapons.Mk_84___2000lb_GP_Bomb_LD)

    class Pylon6:
        TER_9A_with_3_x_Mk_82___500lb_GP_Bomb_LD = (
            6,
            Weapons.TER_9A_with_3_x_Mk_82___500lb_GP_Bomb_LD,
        )
        PROTONM2 = (6, Weapons.PROTONM2)
        Mk_84___2000lb_GP_Bomb_LD = (6, Weapons.Mk_84___2000lb_GP_Bomb_LD)

    class Pylon7:
        TER_9A_with_3_x_Mk_82___500lb_GP_Bomb_LD = (
            7,
            Weapons.TER_9A_with_3_x_Mk_82___500lb_GP_Bomb_LD,
        )
        Mk_84___2000lb_GP_Bomb_LD = (7, Weapons.Mk_84___2000lb_GP_Bomb_LD)

    class Pylon8:
        TER_9A_with_3_x_Mk_82___500lb_GP_Bomb_LD = (
            8,
            Weapons.TER_9A_with_3_x_Mk_82___500lb_GP_Bomb_LD,
        )
        PROTONM2 = (8, Weapons.PROTONM2)
        Mk_84___2000lb_GP_Bomb_LD = (8, Weapons.Mk_84___2000lb_GP_Bomb_LD)

    class Pylon9:
        TER_9A_with_3_x_Mk_82___500lb_GP_Bomb_LD = (
            9,
            Weapons.TER_9A_with_3_x_Mk_82___500lb_GP_Bomb_LD,
        )
        Mk_84___2000lb_GP_Bomb_LD = (9, Weapons.Mk_84___2000lb_GP_Bomb_LD)

    class Pylon10:
        ENERGY_CELL___ = (10, Weapons.ENERGY_CELL___)

    pylons: Set[int] = {1, 2, 3, 4, 5, 6, 7, 8, 9, 10}

    tasks = [
        task.CAP,
        task.Escort,
        task.FighterSweep,
        task.Intercept,
        task.Reconnaissance,
        task.GroundAttack,
        task.CAS,
        task.RunwayAttack,
    ]
    task_default = task.GroundAttack


@planemod
class TIE_BA(PlaneType):
    id = "TIE_BA"
    height = 5.63
    width = 10
    length = 19.43
    fuel_max = 5000
    max_speed = 1548.396
    chaff = 5000
    flare = 5000
    charge_total = 10000
    chaff_charge_size = 1
    flare_charge_size = 1
    category = "Interceptor"  # {78EFB7A2-FD52-4b57-A6A6-3BF0E1D6555F}
    radio_frequency = 127.5

    livery_name = "TIE_BA"  # from type
    Liveries = Liveries()[livery_name]

    class Pylon1:
        PROTONM1 = (1, Weapons.PROTONM1)
        LAU_88_with_3_x_AGM_65D___Maverick_D__IIR_ASM_ = (
            1,
            Weapons.LAU_88_with_3_x_AGM_65D___Maverick_D__IIR_ASM_,
        )
        Mk_84___2000lb_GP_Bomb_LD = (1, Weapons.Mk_84___2000lb_GP_Bomb_LD)
        AGM_88C_HARM___High_Speed_Anti_Radiation_Missile = (
            1,
            Weapons.AGM_88C_HARM___High_Speed_Anti_Radiation_Missile,
        )

    class Pylon2:
        ENERGY_CELL___ = (2, Weapons.ENERGY_CELL___)

    class Pylon3:
        PROTONM1 = (3, Weapons.PROTONM1)
        LAU_88_with_3_x_AGM_65D___Maverick_D__IIR_ASM_ = (
            3,
            Weapons.LAU_88_with_3_x_AGM_65D___Maverick_D__IIR_ASM_,
        )
        TER_9A_with_3_x_Mk_82___500lb_GP_Bomb_LD = (
            3,
            Weapons.TER_9A_with_3_x_Mk_82___500lb_GP_Bomb_LD,
        )
        Mk_84___2000lb_GP_Bomb_LD = (3, Weapons.Mk_84___2000lb_GP_Bomb_LD)
        AGM_88C_HARM___High_Speed_Anti_Radiation_Missile = (
            3,
            Weapons.AGM_88C_HARM___High_Speed_Anti_Radiation_Missile,
        )

    class Pylon4:
        LAU_88_with_3_x_AGM_65D___Maverick_D__IIR_ASM_ = (
            4,
            Weapons.LAU_88_with_3_x_AGM_65D___Maverick_D__IIR_ASM_,
        )
        TER_9A_with_3_x_Mk_82___500lb_GP_Bomb_LD = (
            4,
            Weapons.TER_9A_with_3_x_Mk_82___500lb_GP_Bomb_LD,
        )
        PROTONM2 = (4, Weapons.PROTONM2)
        Mk_84___2000lb_GP_Bomb_LD = (4, Weapons.Mk_84___2000lb_GP_Bomb_LD)
        AGM_88C_HARM___High_Speed_Anti_Radiation_Missile = (
            4,
            Weapons.AGM_88C_HARM___High_Speed_Anti_Radiation_Missile,
        )

    class Pylon5:
        LAU_88_with_3_x_AGM_65D___Maverick_D__IIR_ASM_ = (
            5,
            Weapons.LAU_88_with_3_x_AGM_65D___Maverick_D__IIR_ASM_,
        )
        TER_9A_with_3_x_Mk_82___500lb_GP_Bomb_LD = (
            5,
            Weapons.TER_9A_with_3_x_Mk_82___500lb_GP_Bomb_LD,
        )
        PROTONM2 = (5, Weapons.PROTONM2)
        Mk_84___2000lb_GP_Bomb_LD = (5, Weapons.Mk_84___2000lb_GP_Bomb_LD)
        AGM_88C_HARM___High_Speed_Anti_Radiation_Missile = (
            5,
            Weapons.AGM_88C_HARM___High_Speed_Anti_Radiation_Missile,
        )

    class Pylon6:
        LAU_88_with_3_x_AGM_65D___Maverick_D__IIR_ASM_ = (
            6,
            Weapons.LAU_88_with_3_x_AGM_65D___Maverick_D__IIR_ASM_,
        )
        TER_9A_with_3_x_Mk_82___500lb_GP_Bomb_LD = (
            6,
            Weapons.TER_9A_with_3_x_Mk_82___500lb_GP_Bomb_LD,
        )
        PROTONM2 = (6, Weapons.PROTONM2)
        AGM_88C_HARM___High_Speed_Anti_Radiation_Missile = (
            6,
            Weapons.AGM_88C_HARM___High_Speed_Anti_Radiation_Missile,
        )
        Mk_84___2000lb_GP_Bomb_LD = (6, Weapons.Mk_84___2000lb_GP_Bomb_LD)

    class Pylon7:
        LAU_88_with_3_x_AGM_65D___Maverick_D__IIR_ASM_ = (
            7,
            Weapons.LAU_88_with_3_x_AGM_65D___Maverick_D__IIR_ASM_,
        )
        TER_9A_with_3_x_Mk_82___500lb_GP_Bomb_LD = (
            7,
            Weapons.TER_9A_with_3_x_Mk_82___500lb_GP_Bomb_LD,
        )
        AGM_88C_HARM___High_Speed_Anti_Radiation_Missile = (
            7,
            Weapons.AGM_88C_HARM___High_Speed_Anti_Radiation_Missile,
        )
        Mk_84___2000lb_GP_Bomb_LD = (7, Weapons.Mk_84___2000lb_GP_Bomb_LD)

    class Pylon8:
        TER_9A_with_3_x_Mk_82___500lb_GP_Bomb_LD = (
            8,
            Weapons.TER_9A_with_3_x_Mk_82___500lb_GP_Bomb_LD,
        )
        PROTONM2 = (8, Weapons.PROTONM2)
        AGM_88C_HARM___High_Speed_Anti_Radiation_Missile = (
            8,
            Weapons.AGM_88C_HARM___High_Speed_Anti_Radiation_Missile,
        )

    class Pylon9:
        TER_9A_with_3_x_Mk_82___500lb_GP_Bomb_LD = (
            9,
            Weapons.TER_9A_with_3_x_Mk_82___500lb_GP_Bomb_LD,
        )

    class Pylon10:
        ENERGY_CELL___ = (10, Weapons.ENERGY_CELL___)

    pylons: Set[int] = {1, 2, 3, 4, 5, 6, 7, 8, 9, 10}

    tasks = [
        task.CAP,
        task.Escort,
        task.FighterSweep,
        task.Intercept,
        task.Reconnaissance,
        task.GroundAttack,
        task.CAS,
        task.RunwayAttack,
    ]
    task_default = task.CAS


@vehiclemod
class MBT9_EMPIRE(unittype.VehicleType):
    id = "MBT9_EMPIRE"
    name = "SW - MBT-9 IMPERIAL"
    detection_range = 0
    threat_range = 5000
    air_weapon_dist = 5000


@vehiclemod
class MBT9_REBEL(unittype.VehicleType):
    id = "MBT9_REBEL"
    name = "SW - MBT-9 REBEL"
    detection_range = 0
    threat_range = 5000
    air_weapon_dist = 5000


@vehiclemod
class MBT9_AAA_EMPIRE(unittype.VehicleType):
    id = "MBT9_AAA EMPIRE"
    name = "SW - MBT-9AV IMPERIAL"
    detection_range = 0
    threat_range = 12000
    air_weapon_dist = 12000


@vehiclemod
class MBT9_AAA_REBEL(unittype.VehicleType):
    id = "MBT9_AAA REBEL"
    name = "SW - MBT-9AV REBEL"
    detection_range = 0
    threat_range = 12000
    air_weapon_dist = 12000


@vehiclemod
class Jugger(unittype.VehicleType):
    id = "Jugger"
    name = "SW - Juggernaut"
    detection_range = 25000
    threat_range = 30
    air_weapon_dist = 30
    eplrs = True


@vehiclemod
class TB_TT(unittype.VehicleType):
    id = "TB_TT"
    name = "SW - AT-AT"
    detection_range = 5000
    threat_range = 20000
    air_weapon_dist = 20000


@vehiclemod
class SW___TR_TT(unittype.VehicleType):
    id = "SW - TR_TT"
    name = "SW - AT-ST"
    detection_range = 5000
    threat_range = 5000
    air_weapon_dist = 5000


@vehiclemod
class Gozanti(unittype.VehicleType):
    id = "Gozanti"
    name = "SW - Gozanti"
    detection_range = 0
    threat_range = 0
    air_weapon_dist = 0
    eplrs = True


@vehiclemod
class Destroyer_carrier(unittype.ShipType):
    id = "Destroyer_carrier"
    name = "SW - ISD carrier"
    plane_num = 72
    helicopter_num = 6
    parking = 11
    detection_range = 25000
    threat_range = 0
    air_weapon_dist = 15000