dcs-retribution/gen/naming.py

import random
import time
from typing import List

from dcs.country import Country
from dcs.unittype import UnitType as DcsUnitType

from game.dcs.aircrafttype import AircraftType
from game.dcs.unittype import UnitType
from gen.flights.flight import Flight

ALPHA_MILITARY = [
    "Alpha",
    "Bravo",
    "Charlie",
    "Delta",
    "Echo",
    "Foxtrot",
    "Golf",
    "Hotel",
    "India",
    "Juliet",
    "Kilo",
    "Lima",
    "Mike",
    "November",
    "Oscar",
    "Papa",
    "Quebec",
    "Romeo",
    "Sierra",
    "Tango",
    "Uniform",
    "Victor",
    "Whisky",
    "XRay",
    "Yankee",
    "Zulu",
    "Zero",
]

ANIMALS: tuple[str, ...] = (
    "SHARK",
    "TORTOISE",
    "BAT",
    "PANGOLIN",
    "AARDWOLF",
    "MONKEY",
    "BUFFALO",
    "DOG",
    "BOBCAT",
    "LYNX",
    "PANTHER",
    "TIGER",
    "LION",
    "OWL",
    "BUTTERFLY",
    "BISON",
    "DUCK",
    "COBRA",
    "MAMBA",
    "DOLPHIN",
    "PHEASANT",
    "ARMADILLO",
    "RACOON",
    "ZEBRA",
    "COW",
    "COYOTE",
    "FOX",
    "LIGHTFOOT",
    "COTTONMOUTH",
    "TAURUS",
    "VIPER",
    "CASTOR",
    "GIRAFFE",
    "SNAKE",
    "MONSTER",
    "ALBATROSS",
    "HAWK",
    "DOVE",
    "MOCKINGBIRD",
    "GECKO",
    "ORYX",
    "GORILLA",
    "HARAMBE",
    "GOOSE",
    "MAVERICK",
    "HARE",
    "JACKAL",
    "LEOPARD",
    "CAT",
    "MUSK",
    "ORCA",
    "OCELOT",
    "BEAR",
    "PANDA",
    "GULL",
    "PENGUIN",
    "PYTHON",
    "RAVEN",
    "DEER",
    "MOOSE",
    "REINDEER",
    "SHEEP",
    "GAZELLE",
    "INSECT",
    "VULTURE",
    "WALLABY",
    "KANGAROO",
    "KOALA",
    "KIWI",
    "WHALE",
    "FISH",
    "RHINO",
    "HIPPO",
    "RAT",
    "WOODPECKER",
    "WORM",
    "BABOON",
    "YAK",
    "SCORPIO",
    "HORSE",
    "POODLE",
    "CENTIPEDE",
    "CHICKEN",
    "CHEETAH",
    "CHAMELEON",
    "CATFISH",
    "CATERPILLAR",
    "CARACAL",
    "CAMEL",
    "CAIMAN",
    "BARRACUDA",
    "BANDICOOT",
    "ALLIGATOR",
    "BONGO",
    "CORAL",
    "ELEPHANT",
    "ANTELOPE",
    "CRAB",
    "DACHSHUND",
    "DODO",
    "FLAMINGO",
    "FERRET",
    "FALCON",
    "BULLDOG",
    "DONKEY",
    "IGUANA",
    "TAMARIN",
    "HARRIER",
    "GRIZZLY",
    "GREYHOUND",
    "GRASSHOPPER",
    "JAGUAR",
    "LADYBUG",
    "KOMODO",
    "DRAGON",
    "LIZARD",
    "LLAMA",
    "LOBSTER",
    "OCTOPUS",
    "MANATEE",
    "MAGPIE",
    "MACAW",
    "OSTRICH",
    "OYSTER",
    "MOLE",
    "MULE",
    "MOTH",
    "MONGOOSE",
    "MOLLY",
    "MEERKAT",
    "MOUSE",
    "PEACOCK",
    "PIKE",
    "ROBIN",
    "RAGDOLL",
    "PLATYPUS",
    "PELICAN",
    "PARROT",
    "PORCUPINE",
    "PIRANHA",
    "PUMA",
    "PUG",
    "TAPIR",
    "TERMITE",
    "URCHIN",
    "SHRIMP",
    "TURKEY",
    "TOUCAN",
    "TETRA",
    "HUSKY",
    "STARFISH",
    "SWAN",
    "FROG",
    "SQUIRREL",
    "WALRUS",
    "WARTHOG",
    "CORGI",
    "WEASEL",
    "WOMBAT",
    "WOLVERINE",
    "MAMMOTH",
    "TOAD",
    "WOLF",
    "ZEBU",
    "SEAL",
    "SKATE",
    "JELLYFISH",
    "MOSQUITO",
    "LOCUST",
    "SLUG",
    "SNAIL",
    "HEDGEHOG",
    "PIGLET",
    "FENNEC",
    "BADGER",
    "ALPACA",
    "DINGO",
    "COLT",
    "SKUNK",
    "BUNNY",
    "IMPALA",
    "GUANACO",
    "CAPYBARA",
    "ELK",
    "MINK",
    "PRONGHORN",
    "CROW",
    "BUMBLEBEE",
    "FAWN",
    "OTTER",
    "WATERBUCK",
    "JERBOA",
    "KITTEN",
    "ARGALI",
    "OX",
    "MARE",
    "FINCH",
    "BASILISK",
    "GOPHER",
    "HAMSTER",
    "CANARY",
    "WOODCHUCK",
    "ANACONDA",
)


class NameGenerator:
    number = 0
    infantry_number = 0
    aircraft_number = 0
    convoy_number = 0
    cargo_ship_number = 0

    animals: list[str] = list(ANIMALS)
    existing_alphas: List[str] = []

    @classmethod
    def reset(cls):
        cls.number = 0
        cls.infantry_number = 0
        cls.convoy_number = 0
        cls.cargo_ship_number = 0
        cls.animals = list(ANIMALS)
        cls.existing_alphas = []

    @classmethod
    def reset_numbers(cls):
        cls.number = 0
        cls.infantry_number = 0
        cls.aircraft_number = 0
        cls.convoy_number = 0
        cls.cargo_ship_number = 0

    @classmethod
    def next_aircraft_name(cls, country: Country, parent_base_id: int, flight: Flight):
        cls.aircraft_number += 1
        try:
            if flight.custom_name:
                name_str = flight.custom_name
            else:
                name_str = "{} {}".format(
                    flight.package.target.name, flight.flight_type
                )
        except AttributeError:  # Here to maintain save compatibility with 2.3
            name_str = "{} {}".format(flight.package.target.name, flight.flight_type)
        return "{}|{}|{}|{}|{}|".format(
            name_str,
            country.id,
            cls.aircraft_number,
            parent_base_id,
            flight.unit_type.name,
        )

    @classmethod
    def next_unit_name(
        cls, country: Country, parent_base_id: int, unit_type: UnitType[DcsUnitType]
    ) -> str:
        cls.number += 1
        return "unit|{}|{}|{}|{}|".format(
            country.id, cls.number, parent_base_id, unit_type.name
        )

    @classmethod
    def next_infantry_name(
        cls, country: Country, parent_base_id: int, unit_type: UnitType[DcsUnitType]
    ) -> str:
        cls.infantry_number += 1
        return "infantry|{}|{}|{}|{}|".format(
            country.id,
            cls.infantry_number,
            parent_base_id,
            unit_type.name,
        )

    @classmethod
    def next_awacs_name(cls, country: Country):
        cls.number += 1
        return "awacs|{}|{}|0|".format(country.id, cls.number)

    @classmethod
    def next_tanker_name(cls, country: Country, unit_type: AircraftType):
        cls.number += 1
        return "tanker|{}|{}|0|{}".format(country.id, cls.number, unit_type.name)

    @classmethod
    def next_carrier_name(cls, country: Country):
        cls.number += 1
        return "carrier|{}|{}|0|".format(country.id, cls.number)

    @classmethod
    def next_convoy_name(cls) -> str:
        cls.convoy_number += 1
        return f"Convoy {cls.convoy_number:03}"

    @classmethod
    def next_cargo_ship_name(cls) -> str:
        cls.cargo_ship_number += 1
        return f"Cargo Ship {cls.cargo_ship_number:03}"

    @classmethod
    def random_objective_name(cls):
        if cls.animals:
            animal = random.choice(cls.animals)
            cls.animals.remove(animal)
            return animal

        for _ in range(10):
            alpha = random.choice(ALPHA_MILITARY).upper()
            number = random.randint(0, 100)
            alpha_mil_name = f"{alpha} #{number:02}"
            if alpha_mil_name not in cls.existing_alphas:
                cls.existing_alphas.append(alpha_mil_name)
                return alpha_mil_name

        # At this point, give up trying - something has gone wrong and we haven't been
        # able to make a new name in 10 tries. We'll just make a longer name using the
        # current unix epoch in nanoseconds. That should be unique... right?
        last_chance_name = alpha_mil_name + str(time.time_ns())
        cls.existing_alphas.append(last_chance_name)
        return last_chance_name


namegen = NameGenerator