119 lines
3.3 KiB
Python
119 lines
3.3 KiB
Python
from enum import Enum
|
||
from dataclasses import dataclass
|
||
from typing import TYPE_CHECKING
|
||
|
||
if TYPE_CHECKING:
|
||
from src.classes.region import Region
|
||
from src.classes.avatar import Avatar
|
||
|
||
class TileType(Enum):
|
||
PLAIN = "plain" # 平原
|
||
WATER = "water" # 水域
|
||
SEA = "sea" # 海洋
|
||
MOUNTAIN = "mountain" # 山脉
|
||
FOREST = "forest" # 森林
|
||
CITY = "city" # 城市
|
||
DESERT = "desert" # 沙漠
|
||
RAINFOREST = "rainforest" # 热带雨林
|
||
GLACIER = "glacier" # 冰川/冰原
|
||
SNOW_MOUNTAIN = "snow_mountain" # 雪山
|
||
VOLCANO = "volcano" # 火山
|
||
GRASSLAND = "grassland" # 草原
|
||
SWAMP = "swamp" # 沼泽
|
||
CAVE = "cave" # 洞穴
|
||
RUIN = "ruin" # 遗迹
|
||
FARM = "farm" # 农田
|
||
SECT = "sect" # 宗门
|
||
ISLAND = "island" # 岛屿
|
||
BAMBOO = "bamboo" # 竹林
|
||
GOBI = "gobi" # 戈壁
|
||
TUNDRA = "tundra" # 苔原
|
||
MARSH = "marsh" # 湿地
|
||
|
||
|
||
# ============================================================
|
||
# 2x2 大型Tile 配置
|
||
# ============================================================
|
||
|
||
# 固定的大型Tile类型 (基于 TileType)
|
||
LARGE_TILE_TYPES: set[TileType] = {
|
||
TileType.CAVE,
|
||
TileType.RUIN,
|
||
TileType.SECT,
|
||
TileType.CITY,
|
||
}
|
||
|
||
# 动态的大型Tile前缀 (用于宗门名、城市名等)
|
||
# 地图存储的tile名称如果以这些类别开头的子文件夹对应，则视为大型Tile
|
||
LARGE_TILE_CATEGORIES: set[str] = {"sect", "city"}
|
||
|
||
|
||
def is_large_tile(tile_name: str) -> bool:
|
||
"""
|
||
判断一个tile名称是否为2x2大型tile。
|
||
|
||
Args:
|
||
tile_name: tile的名称（如 "cave", "明心剑宗", "青云城" 等）
|
||
|
||
Returns:
|
||
是否为大型tile
|
||
"""
|
||
# 1. 检查是否为标准大型TileType
|
||
try:
|
||
t = TileType(tile_name.lower())
|
||
return t in LARGE_TILE_TYPES
|
||
except ValueError:
|
||
pass
|
||
|
||
# 2. 非标准类型：宗门名、城市名等
|
||
# 这些tile的名称直接是中文名，它们的切片存在于 sects/ 或 cities/ 目录
|
||
# 由于无法在此静态判断，我们假设所有非标准TileType的名称都是大型tile
|
||
# (因为普通tile都在TileType枚举中)
|
||
return True
|
||
|
||
|
||
def get_large_tile_slices(tile_name: str) -> list[str]:
|
||
"""
|
||
获取大型tile的4个切片名称。
|
||
|
||
Args:
|
||
tile_name: 大型tile的名称
|
||
|
||
Returns:
|
||
4个切片名称的列表 [TL, TR, BL, BR]
|
||
"""
|
||
return [f"{tile_name}_{i}" for i in range(4)]
|
||
|
||
|
||
@dataclass
|
||
class Tile():
|
||
# 实际的地块
|
||
type: TileType
|
||
x: int
|
||
y: int
|
||
region: 'Region' = None # 可以是一个region的一部分，也可以不属于任何region
|
||
|
||
@property
|
||
def location_name(self) -> str:
|
||
"""
|
||
优雅地获取当前地点的名称。
|
||
如果属于某个区域，返回区域名；否则返回'荒野'。
|
||
"""
|
||
return self.region.name if self.region else "荒野"
|
||
|
||
|
||
from src.utils.distance import manhattan_distance
|
||
|
||
def get_avatar_distance(avatar1: 'Avatar', avatar2: 'Avatar') -> int:
|
||
"""
|
||
计算两个 Avatar 之间的曼哈顿距离
|
||
|
||
Args:
|
||
avatar1: 第一个角色
|
||
avatar2: 第二个角色
|
||
|
||
Returns:
|
||
两个角色之间的距离
|
||
"""
|
||
return manhattan_distance((avatar1.pos_x, avatar1.pos_y), (avatar2.pos_x, avatar2.pos_y))
|