Clean up the category property of TGOs.

This really needs to be a proper type, but this is a start: create new
categories for the types of TGOs that are missing. This removes some
icon special cases.
This commit is contained in:
Dan Albert
2021-05-18 22:53:38 -07:00
parent 2769d32c81
commit 4e37666037
5 changed files with 23 additions and 91 deletions

View File

@@ -5,9 +5,9 @@ import logging
from typing import Iterator, List, TYPE_CHECKING
from dcs.mapping import Point
from dcs.triggers import TriggerZone
from dcs.unit import Unit
from dcs.unitgroup import Group
from dcs.triggers import TriggerZone, Triggers
from .. import db
from ..data.radar_db import UNITS_WITH_RADAR
@@ -20,78 +20,25 @@ if TYPE_CHECKING:
from .missiontarget import MissionTarget
NAME_BY_CATEGORY = {
"power": "Power plant",
"ammo": "Ammo depot",
"fuel": "Fuel depot",
"ewr": "Early Warning Radar",
"aa": "AA Defense Site",
"ware": "Warehouse",
"farp": "FARP",
"fob": "FOB",
"factory": "Factory",
"comms": "Comms. tower",
"oil": "Oil platform",
"derrick": "Derrick",
"ww2bunker": "Bunker",
"village": "Village",
"allycamp": "Camp",
"EWR": "EWR",
}
ABBREV_NAME = {
"power": "PLANT",
"ammo": "AMMO",
"fuel": "FUEL",
"aa": "AA",
"ware": "WARE",
"ammo": "Ammo depot",
"armor": "Armor group",
"coastal": "Coastal defense",
"comms": "Communications tower",
"derrick": "Derrick",
"factory": "Factory",
"farp": "FARP",
"fob": "FOB",
"factory": "FACTORY",
"comms": "COMMST",
"oil": "OILP",
"derrick": "DERK",
"ww2bunker": "BUNK",
"village": "VLG",
"allycamp": "CMP",
}
CATEGORY_MAP = {
# Special cases
"CARRIER": ["CARRIER"],
"LHA": ["LHA"],
"aa": ["AA"],
# Buildings
"power": [
"Workshop A",
"Electric power box",
"Garage small A",
"Farm B",
"Repair workshop",
"Garage B",
],
"ware": ["Warehouse", "Hangar A"],
"fuel": ["Tank", "Tank 2", "Tank 3", "Fuel tank"],
"ammo": [".Ammunition depot", "Hangar B"],
"farp": [
"FARP Tent",
"FARP Ammo Dump Coating",
"FARP Fuel Depot",
"FARP Command Post",
"FARP CP Blindage",
],
"fob": ["Bunker 2", "Bunker 1", "Garage small B", ".Command Center", "Barracks 2"],
"factory": ["Tech combine", "Tech hangar A"],
"comms": ["TV tower", "Comms tower M"],
"oil": ["Oil platform"],
"derrick": ["Oil derrick", "Pump station", "Subsidiary structure 2"],
"ww2bunker": [
"Siegfried Line",
"Fire Control Bunker",
"SK_C_28_naval_gun",
"Concertina Wire",
"Czech hedgehogs 1",
],
"village": ["Small house 1B", "Small House 1A", "Small warehouse 1"],
"allycamp": [],
"fuel": "Fuel depot",
"missile": "Missile site",
"oil": "Oil platform",
"power": "Power plant",
"ship": "Ship",
"village": "Village",
"ware": "Warehouse",
"ww2bunker": "Bunker",
}
@@ -414,7 +361,7 @@ class MissileSiteGroundObject(TheaterGroundObject):
) -> None:
super().__init__(
name=name,
category="aa",
category="missile",
group_id=group_id,
position=position,
heading=0,
@@ -436,7 +383,7 @@ class CoastalSiteGroundObject(TheaterGroundObject):
) -> None:
super().__init__(
name=name,
category="aa",
category="coastal",
group_id=group_id,
position=position,
heading=heading,
@@ -510,7 +457,7 @@ class VehicleGroupGroundObject(BaseDefenseGroundObject):
) -> None:
super().__init__(
name=name,
category="aa",
category="armor",
group_id=group_id,
position=position,
heading=0,
@@ -532,7 +479,7 @@ class EwrGroundObject(BaseDefenseGroundObject):
) -> None:
super().__init__(
name=name,
category="EWR",
category="ewr",
group_id=group_id,
position=position,
heading=0,
@@ -565,7 +512,7 @@ class ShipGroundObject(NavalGroundObject):
) -> None:
super().__init__(
name=name,
category="aa",
category="ship",
group_id=group_id,
position=position,
heading=0,