from dataclasses import dataclass, field from typing import Union, TypeVar, Type, Optional, TYPE_CHECKING from enum import Enum from abc import ABC, abstractmethod from src.utils.df import game_configs, get_str, get_int, get_list_int from src.utils.config import CONFIG from src.utils.distance import chebyshev_distance from src.classes.essence import EssenceType, Essence from src.classes.animal import Animal, animals_by_id from src.classes.plant import Plant, plants_by_id from src.classes.sect import sects_by_name if TYPE_CHECKING: from src.classes.avatar import Avatar @dataclass class Region(ABC): """ 区域抽象基类 """ id: int name: str desc: str # 核心坐标数据，由 load_map.py 注入 cors: list[tuple[int, int]] = field(default_factory=list) # 计算字段 center_loc: tuple[int, int] = field(init=False) area: int = field(init=False) def __post_init__(self): """初始化计算字段""" # 基于坐标点计算面积 self.area = len(self.cors) # 计算中心位置 if self.cors: avg_x = sum(coord[0] for coord in self.cors) // len(self.cors) avg_y = sum(coord[1] for coord in self.cors) // len(self.cors) candidate = (avg_x, avg_y) if candidate in self.cors: self.center_loc = candidate else: def _dist2(p: tuple[int, int]) -> int: return (p[0] - avg_x) ** 2 + (p[1] - avg_y) ** 2 self.center_loc = min(self.cors, key=_dist2) else: # Fallback self.center_loc = (0, 0) def __hash__(self) -> int: return hash(self.id) def __eq__(self, other) -> bool: if not isinstance(other, Region): return False return self.id == other.id @abstractmethod def get_region_type(self) -> str: pass def get_hover_info(self) -> list[str]: return [ f"区域: {self.name}", f"描述: {self.desc}", ] @abstractmethod def _get_desc(self) -> str: """ 返回紧跟在名字后的描述，通常包含括号，例如 '（金行灵气：5）' 注意，不需要包含self.desc """ pass def _get_distance_desc(self, current_loc: tuple[int, int] = None, step_len: int = 1) -> str: if current_loc is None: return "" dist = chebyshev_distance(current_loc, self.center_loc) # 估算到达时间：距离 / 步长 (向上取整) months = (dist + step_len - 1) // step_len # 避免显示 0 个月 months = max(1, months) return f"（距离：{months}月）" def get_info(self, current_loc: tuple[int, int] = None, step_len: int = 1) -> str: return f"{self.name}{self._get_distance_desc(current_loc, step_len)}" def get_detailed_info(self, current_loc: tuple[int, int] = None, step_len: int = 1) -> str: return f"{self.name}{self._get_desc()} - {self.desc}{self._get_distance_desc(current_loc, step_len)}" def get_structured_info(self) -> dict: return { "id": str(self.id), "name": self.name, "desc": self.desc, "type": self.get_region_type(), "type_name": "区域" } @dataclass(eq=False) class NormalRegion(Region): """普通区域""" animal_ids: list[int] = field(default_factory=list) plant_ids: list[int] = field(default_factory=list) animals: list[Animal] = field(init=False, default_factory=list) plants: list[Plant] = field(init=False, default_factory=list) def __post_init__(self): super().__post_init__() for animal_id in self.animal_ids: if animal_id in animals_by_id: self.animals.append(animals_by_id[animal_id]) for plant_id in self.plant_ids: if plant_id in plants_by_id: self.plants.append(plants_by_id[plant_id]) def get_region_type(self) -> str: return "normal" def get_species_info(self) -> str: info_parts = [] if self.animals: info_parts.extend([a.get_info() for a in self.animals]) if self.plants: info_parts.extend([p.get_info() for p in self.plants]) return "; ".join(info_parts) if info_parts else "无特色物种" def _get_desc(self) -> str: species_info = self.get_species_info() return f"（物种分布：{species_info}）" def __str__(self) -> str: species_info = self.get_species_info() return f"普通区域：{self.name} - {self.desc} | 物种分布：{species_info}" def get_hover_info(self) -> list[str]: lines = super().get_hover_info() species_info = self.get_species_info() if species_info and species_info != "暂无特色物种": lines.append("物种分布:") for species in species_info.split("; "): lines.append(f" {species}") else: lines.append("物种分布: 暂无特色物种") return lines @property def is_huntable(self) -> bool: return len(self.animals) > 0 @property def is_harvestable(self) -> bool: return len(self.plants) > 0 def get_structured_info(self) -> dict: info = super().get_structured_info() info["type_name"] = "普通区域" # Fix: Return the actual structure instead of just calling get_structured_info on elements but never assigning # The previous implementation (if it existed) was inherited from base or incorrect # Assuming animals and plants are populated in __post_init__ info["animals"] = [a.get_structured_info() for a in self.animals] if self.animals else [] info["plants"] = [p.get_structured_info() for p in self.plants] if self.plants else [] return info @dataclass(eq=False) class CultivateRegion(Region): """修炼区域""" essence_type: EssenceType = EssenceType.GOLD # 默认值避免 dataclass 继承错误 essence_density: int = 0 essence: Essence = field(init=False) # 洞府主人：默认为空（无主） host_avatar: Optional["Avatar"] = field(default=None, init=False) def __post_init__(self): super().__post_init__() essence_density_dict = {essence_type: 0 for essence_type in EssenceType} essence_density_dict[self.essence_type] = self.essence_density self.essence = Essence(essence_density_dict) def get_region_type(self) -> str: return "cultivate" def _get_desc(self) -> str: return f"（{self.essence_type}行灵气：{self.essence_density}）" def __str__(self) -> str: return f"修炼区域：{self.name}（{self.essence_type}行灵气：{self.essence_density}）- {self.desc}" def get_hover_info(self) -> list[str]: lines = super().get_hover_info() stars = "★" * self.essence_density + "☆" * (10 - self.essence_density) lines.append(f"主要灵气: {self.essence_type} {stars}") if self.host_avatar: lines.append(f"主人: {self.host_avatar.name}") else: lines.append("主人: 无（可占据）") return lines def get_structured_info(self) -> dict: info = super().get_structured_info() info["type_name"] = "修炼区域" info["essence"] = { "type": str(self.essence_type), "density": self.essence_density } if self.host_avatar: info["host"] = { "id": self.host_avatar.id, "name": self.host_avatar.name } else: info["host"] = None return info @dataclass(eq=False) class CityRegion(Region): """城市区域""" def get_region_type(self) -> str: return "city" def _get_desc(self) -> str: return "" def __str__(self) -> str: return f"城市区域：{self.name} - {self.desc}" def get_structured_info(self) -> dict: info = super().get_structured_info() info["type_name"] = "城市区域" return info def _normalize_region_name(name: str) -> str: s = str(name).strip() brackets = [("(", ")"), ("（", "）"), ("[", "]"), ("【", "】"), ("「", "」"), ("『", "』"), ("<", ">"), ("《", "》")] for left, right in brackets: while True: start = s.find(left) end = s.rfind(right) if start != -1 and end != -1 and end > start: s = (s[:start] + s[end + 1:]).strip() else: break return s def resolve_region(world, region: Union[Region, str]) -> Region: """ 解析字符串或 Region 为当前 world.map 中的 Region 实例 """ from typing import Dict if isinstance(region, str): region_name = region by_name: Dict[str, Region] = getattr(world.map, "region_names", {}) # 1) 精确匹配 r = by_name.get(region_name) if r is not None: return r # 2) 归一化后再精确匹配 normalized = _normalize_region_name(region_name) if normalized and normalized != region_name: r2 = by_name.get(normalized) if r2 is not None: return r2 # 3) 唯一包含匹配 candidates = [name for name in by_name.keys() if name and (name in region_name or (normalized and name in normalized))] if len(candidates) == 1: return by_name[candidates[0]] # 4) 兜底：若传入为宗门名，则解析为其总部区域 sect = sects_by_name.get(region_name) or (sects_by_name.get(normalized) if normalized and normalized != region_name else None) if sect is not None: sect_regions = getattr(world.map, "sect_regions", {}) or {} matched = [r for r in sect_regions.values() if getattr(r, "sect_name", None) == sect.name] if len(matched) == 1: return matched[0] if candidates: sample = ", ".join(candidates[:5]) raise ValueError(f"区域名不唯一: {region_name}，候选: {sample}") raise ValueError(f"未知区域名: {region_name}") if isinstance(region, Region): by_id = getattr(world.map, "regions", None) if isinstance(by_id, dict) and region.id in by_id: return by_id[region.id] return region raise TypeError(f"不支持的region类型: {type(region).__name__}")