from dcs.vehicles import MissilesSS, Unarmed, AirDefence

from gen.sam.group_generator import GroupGenerator


class SilkwormGenerator(GroupGenerator):
    def __init__(self, game, ground_object, faction):
        super(SilkwormGenerator, self).__init__(game, ground_object)
        self.faction = faction

    def generate(self):

        positions = self.get_circular_position(5, launcher_distance=120, coverage=180)

        self.add_unit(
            MissilesSS.Silkworm_Radar,
            "SR#0",
            self.position.x,
            self.position.y,
            self.heading,
        )

        # Launchers
        for i, p in enumerate(positions):
            self.add_unit(
                MissilesSS.SS_N_2_Silkworm,
                "Missile#" + str(i),
                p[0],
                p[1],
                self.heading,
            )

        # Commander
        self.add_unit(
            Unarmed.Truck_KAMAZ_43101,
            "KAMAZ#0",
            self.position.x - 35,
            self.position.y - 20,
            self.heading,
        )

        # Shorad
        self.add_unit(
            AirDefence.SPAAA_ZSU_23_4_Shilka_Gun_Dish,
            "SHILKA#0",
            self.position.x - 55,
            self.position.y - 38,
            self.heading,
        )

        # Shorad 2
        self.add_unit(
            AirDefence.SAM_SA_9_Strela_1_Gaskin_TEL,
            "STRELA#0",
            self.position.x + 200,
            self.position.y + 15,
            90,
        )