nodes/data.py

Part of Tanks of Freedom.

"""Game constants and data tables for Tanks of Freedom.

Unit stats are lifted from upstream `scripts/objects/units/{soldier,tank,
helicopter}.gd`; the map layout and starting line-up are this port's own,
sized for a single 12x12 skirmish.
"""

from __future__ import annotations

# ============================================================================
# Window / world layout
# ============================================================================

WINDOW_WIDTH = 1280
WINDOW_HEIGHT = 720

# Map size: 12x12 fits a small skirmish in screen space comfortably
MAP_W = 12
MAP_H = 12

# Iso tile dimensions. Upstream uses 64x32 for the diamond; using same here.
TILE_W = 64
TILE_H = 32

# World origin offset so the diamond sits centred horizontally below the HUD bar
WORLD_OFFSET_X = WINDOW_WIDTH * 0.5
WORLD_OFFSET_Y = 120

HUD_HEIGHT = 60  # top bar
INFO_PANEL_W = 280  # right-hand info panel

# ============================================================================
# Player ids
# ============================================================================

PLAYER_BLUE = 0
PLAYER_RED = 1
PLAYER_NEUTRAL = -1

PLAYER_COLOUR = {
    PLAYER_BLUE: (0.30, 0.55, 0.95, 1.0),
    PLAYER_RED: (0.95, 0.35, 0.30, 1.0),
    PLAYER_NEUTRAL: (0.75, 0.75, 0.78, 1.0),
}

PLAYER_NAME = {
    PLAYER_BLUE: "BLUE",
    PLAYER_RED: "RED",
    PLAYER_NEUTRAL: "NEUTRAL",
}

# ============================================================================
# Unit types  (from upstream unit.gd subclasses)
# ============================================================================

UNIT_SOLDIER = 0
UNIT_TANK = 1
UNIT_HELICOPTER = 2

UNIT_STATS = {
    UNIT_SOLDIER: {
        "name": "INFANTRY",
        "life": 10,
        "max_life": 10,
        "attack": 5,
        "max_ap": 4,
        "attack_ap": 1,
        "max_attacks": 1,
        "can_capture": True,
    },
    UNIT_TANK: {
        "name": "TANK",
        "life": 15,
        "max_life": 15,
        "attack": 10,
        "max_ap": 6,
        "attack_ap": 1,
        "max_attacks": 1,
        "can_capture": False,
    },
    UNIT_HELICOPTER: {
        "name": "HELICOPTER",
        "life": 10,
        "max_life": 10,
        "attack": 8,
        "max_ap": 8,
        "attack_ap": 1,
        "max_attacks": 1,
        "can_capture": False,
    },
}

# How far a unit can reach a target, in grid steps. All three units are
# melee-range in this port, matching upstream's adjacency-only combat.
ATTACK_RANGE = {
    UNIT_SOLDIER: 1,
    UNIT_TANK: 1,
    UNIT_HELICOPTER: 1,
}

# ============================================================================
# Building types  (from building.gd)
# ============================================================================

# Upstream builds units out of barracks / factory / airport; this port has no
# production, so the four kinds differ only in look and in the HQ win condition.
BLDG_HQ = 0
BLDG_BARRACKS = 1
BLDG_FACTORY = 2
BLDG_AIRPORT = 3

BUILDING_STATS = {
    BLDG_HQ: {"name": "HQ"},
    BLDG_BARRACKS: {"name": "BARRACKS"},
    BLDG_FACTORY: {"name": "FACTORY"},
    BLDG_AIRPORT: {"name": "AIRPORT"},
}

# ============================================================================
# Terrain  (impassable bits: placed on tile map by world)
# ============================================================================

TERR_GRASS = 0
TERR_WATER = 1  # impassable for ground units
TERR_TREE = 2  # impassable for everyone except heli
TERR_MOUNTAIN = 3  # impassable

TERRAIN_FROM_CHAR = {
    "g": TERR_GRASS,
    "w": TERR_WATER,
    "f": TERR_TREE,
    "t": TERR_TREE,
    "m": TERR_MOUNTAIN,
    "r": TERR_GRASS,  # 'road' visually identical to grass for our procedural look
}

# 12x12 map. 'g' grass, 'f' forest/tree, 'w' water, 'm' mountain.
# Symmetric: blue south-west, red north-east, with a neutral mid corridor.
MAP_LAYOUT_DEFAULT = [
    "ggggwwggffgg",
    "gggwwggggfgg",
    "ggffmgggrrgg",
    "gffggggggrgg",
    "ggggggwwgggg",
    "gggggggwgggg",
    "ggggwggggggg",
    "ggggwwgggggg",
    "ggrggggggffg",
    "ggrrgggmffgg",
    "ggfggggmmggg",
    "ggffggggwwgg",
]

# ============================================================================
# Starting state: buildings & units. Tied to MAP_LAYOUT_DEFAULT.
# ============================================================================

# (x, y, building_type, owner)
STARTING_BUILDINGS = [
    (1, 10, BLDG_HQ, PLAYER_BLUE),
    (3, 10, BLDG_BARRACKS, PLAYER_BLUE),
    (1, 8, BLDG_FACTORY, PLAYER_BLUE),
    (3, 8, BLDG_AIRPORT, PLAYER_BLUE),
    (10, 1, BLDG_HQ, PLAYER_RED),
    (8, 1, BLDG_BARRACKS, PLAYER_RED),
    (10, 3, BLDG_FACTORY, PLAYER_RED),
    (8, 3, BLDG_AIRPORT, PLAYER_RED),
    # Neutral mid-map captures
    (5, 5, BLDG_BARRACKS, PLAYER_NEUTRAL),
    (6, 6, BLDG_FACTORY, PLAYER_NEUTRAL),
]

# (x, y, unit_type, owner)
STARTING_UNITS = [
    (2, 10, UNIT_SOLDIER, PLAYER_BLUE),
    (3, 9, UNIT_SOLDIER, PLAYER_BLUE),
    (2, 8, UNIT_TANK, PLAYER_BLUE),
    (4, 9, UNIT_HELICOPTER, PLAYER_BLUE),
    (9, 1, UNIT_SOLDIER, PLAYER_RED),
    (8, 2, UNIT_SOLDIER, PLAYER_RED),
    (9, 3, UNIT_TANK, PLAYER_RED),
    (7, 2, UNIT_HELICOPTER, PLAYER_RED),
]

# ============================================================================
# Audio (CC-BY-SA assets bundled from upstream)
# ============================================================================

SFX_SELECT = "select.wav"
SFX_MOVE = "move.wav"
SFX_END_TURN = "end_turn.wav"
SFX_EXPLOSION = "explosion.wav"
SFX_HURT = "hurt.wav"
SFX_BUTTON = "button.wav"
SFX_NO_MOVES = "no_moves.wav"


# ============================================================================
# Combat tables
# ============================================================================

# Attacker -> Defender ban list (upstream: tank can't shoot helicopters).
_NO_ATTACK = {(UNIT_TANK, UNIT_HELICOPTER)}


def can_attack_unit_type(att_type: int, def_type: int) -> bool:
    return (att_type, def_type) not in _NO_ATTACK