fix bug
This commit is contained in:
34
src/classes/avatar/__init__.py
Normal file
34
src/classes/avatar/__init__.py
Normal file
@@ -0,0 +1,34 @@
|
||||
"""
|
||||
Avatar 模块
|
||||
|
||||
将原 avatar.py 拆分为多个子模块,通过此 __init__.py 导出以保持向后兼容。
|
||||
"""
|
||||
from src.classes.avatar.core import (
|
||||
Avatar,
|
||||
Gender,
|
||||
gender_strs,
|
||||
MAX_HISTORY_EVENTS,
|
||||
)
|
||||
|
||||
from src.classes.avatar.info_presenter import (
|
||||
get_avatar_info,
|
||||
get_avatar_structured_info,
|
||||
get_avatar_hover_info,
|
||||
get_avatar_expanded_info,
|
||||
get_other_avatar_info,
|
||||
)
|
||||
|
||||
__all__ = [
|
||||
# 核心类
|
||||
"Avatar",
|
||||
"Gender",
|
||||
"gender_strs",
|
||||
"MAX_HISTORY_EVENTS",
|
||||
# 信息展示函数
|
||||
"get_avatar_info",
|
||||
"get_avatar_structured_info",
|
||||
"get_avatar_hover_info",
|
||||
"get_avatar_expanded_info",
|
||||
"get_other_avatar_info",
|
||||
]
|
||||
|
||||
158
src/classes/avatar/action_mixin.py
Normal file
158
src/classes/avatar/action_mixin.py
Normal file
@@ -0,0 +1,158 @@
|
||||
"""
|
||||
Avatar 动作管理 Mixin
|
||||
"""
|
||||
from __future__ import annotations
|
||||
|
||||
from typing import TYPE_CHECKING, Optional, List
|
||||
|
||||
if TYPE_CHECKING:
|
||||
from src.classes.avatar.core import Avatar
|
||||
|
||||
from src.classes.action import Action
|
||||
from src.classes.action_runtime import ActionStatus, ActionResult, ActionPlan, ActionInstance
|
||||
from src.classes.action.registry import ActionRegistry
|
||||
from src.classes.event import Event
|
||||
from src.classes.typings import ACTION_NAME, ACTION_NAME_PARAMS_PAIRS
|
||||
from src.utils.params import filter_kwargs_for_callable
|
||||
from src.run.log import get_logger
|
||||
|
||||
|
||||
class ActionMixin:
|
||||
"""动作管理相关方法"""
|
||||
|
||||
def create_action(self: "Avatar", action_name: ACTION_NAME) -> Action:
|
||||
"""
|
||||
根据动作名称创建新的action实例
|
||||
|
||||
Args:
|
||||
action_name: 动作类的名称(如 'Cultivate', 'Breakthrough' 等)
|
||||
|
||||
Returns:
|
||||
新创建的Action实例
|
||||
|
||||
Raises:
|
||||
ValueError: 如果找不到对应的动作类
|
||||
"""
|
||||
action_cls = ActionRegistry.get(action_name)
|
||||
return action_cls(self, self.world)
|
||||
|
||||
def load_decide_result_chain(
|
||||
self: "Avatar",
|
||||
action_name_params_pairs: ACTION_NAME_PARAMS_PAIRS,
|
||||
avatar_thinking: str,
|
||||
short_term_objective: str,
|
||||
prepend: bool = False
|
||||
):
|
||||
"""
|
||||
加载AI的决策结果(动作链),立即设置第一个为当前动作,其余进入队列。
|
||||
|
||||
Args:
|
||||
action_name_params_pairs: 动作名和参数对列表
|
||||
avatar_thinking: 思考内容
|
||||
short_term_objective: 短期目标
|
||||
prepend: 是否插队到最前面(默认False,即追加到末尾)
|
||||
"""
|
||||
if not action_name_params_pairs:
|
||||
return
|
||||
self.thinking = avatar_thinking
|
||||
self.short_term_objective = short_term_objective
|
||||
# 转为计划并入队(不立即提交,交由提交阶段统一触发开始事件)
|
||||
plans: List[ActionPlan] = [ActionPlan(name, params) for name, params in action_name_params_pairs]
|
||||
if prepend:
|
||||
self.planned_actions[0:0] = plans
|
||||
else:
|
||||
self.planned_actions.extend(plans)
|
||||
|
||||
def clear_plans(self: "Avatar") -> None:
|
||||
self.planned_actions.clear()
|
||||
|
||||
def has_plans(self: "Avatar") -> bool:
|
||||
return len(self.planned_actions) > 0
|
||||
|
||||
def commit_next_plan(self: "Avatar") -> Optional[Event]:
|
||||
"""
|
||||
提交下一个可启动的计划为当前动作;返回开始事件(若有)。
|
||||
"""
|
||||
if self.current_action is not None:
|
||||
return None
|
||||
while self.planned_actions:
|
||||
plan = self.planned_actions.pop(0)
|
||||
try:
|
||||
action = self.create_action(plan.action_name)
|
||||
except Exception as e:
|
||||
logger = get_logger().logger
|
||||
logger.warning(
|
||||
"非法动作: Avatar(name=%s,id=%s) 的动作 %s 参数=%s 无法启动,原因=%s",
|
||||
self.name, self.id, plan.action_name, plan.params, e
|
||||
)
|
||||
continue
|
||||
# 再验证
|
||||
params_for_can_start = filter_kwargs_for_callable(action.can_start, plan.params)
|
||||
can_start, reason = action.can_start(**params_for_can_start)
|
||||
if not can_start:
|
||||
# 记录不合法动作
|
||||
logger = get_logger().logger
|
||||
logger.warning(
|
||||
"非法动作: Avatar(name=%s,id=%s) 的动作 %s 参数=%s 无法启动,原因=%s",
|
||||
self.name, self.id, plan.action_name, plan.params, reason
|
||||
)
|
||||
continue
|
||||
# 启动
|
||||
params_for_start = filter_kwargs_for_callable(action.start, plan.params)
|
||||
start_event = action.start(**params_for_start)
|
||||
self.current_action = ActionInstance(action=action, params=plan.params, status="running")
|
||||
# 标记为"本轮新设动作",用于本月补充执行
|
||||
self._new_action_set_this_step = True
|
||||
return start_event
|
||||
return None
|
||||
|
||||
def peek_next_plan(self: "Avatar") -> Optional[ActionPlan]:
|
||||
if not self.planned_actions:
|
||||
return None
|
||||
return self.planned_actions[0]
|
||||
|
||||
async def tick_action(self: "Avatar") -> List[Event]:
|
||||
"""
|
||||
推进当前动作一步;返回过程中由动作内部产生的事件(通过 add_event 收集)。
|
||||
"""
|
||||
if self.current_action is None:
|
||||
return []
|
||||
# 记录当前动作实例引用,用于检测执行过程中是否发生了"抢占/切换"
|
||||
action_instance_before = self.current_action
|
||||
action = action_instance_before.action
|
||||
params = action_instance_before.params
|
||||
params_for_step = filter_kwargs_for_callable(action.step, params)
|
||||
result: ActionResult = action.step(**params_for_step)
|
||||
if result.status == ActionStatus.COMPLETED:
|
||||
params_for_finish = filter_kwargs_for_callable(action.finish, params)
|
||||
finish_events = await action.finish(**params_for_finish)
|
||||
# 仅当当前动作仍然是刚才执行的那个实例时才清空
|
||||
# 若在 step() 内部通过"抢占"机制切换了动作(如 Escape 失败立即切到 Attack),不要清空新动作
|
||||
if self.current_action is action_instance_before:
|
||||
self.current_action = None
|
||||
if finish_events:
|
||||
# 允许 finish 直接返回事件(极少用),统一并入 pending
|
||||
for e in finish_events:
|
||||
self._pending_events.append(e)
|
||||
# 合并动作返回的事件(通常为空)
|
||||
if result.events:
|
||||
for e in result.events:
|
||||
self._pending_events.append(e)
|
||||
events, self._pending_events = self._pending_events, []
|
||||
# 本轮已执行过,清除"新设动作"标记(但如果刚刚提交了新动作,commit_next_plan会重新设置为True)
|
||||
if self.current_action is None:
|
||||
# 当前无动作时才清除标记,避免清除新提交动作的标记
|
||||
self._new_action_set_this_step = False
|
||||
|
||||
return events
|
||||
|
||||
def add_event(self: "Avatar", event: Event, *, to_sidebar: bool = True) -> None:
|
||||
"""
|
||||
添加事件:
|
||||
- to_sidebar: 是否进入全局侧边栏(通过 Avatar._pending_events 暂存)
|
||||
|
||||
注意:事件会先存入_pending_events,统一由Simulator写入event_manager,避免重复
|
||||
"""
|
||||
if to_sidebar:
|
||||
self._pending_events.append(event)
|
||||
|
||||
311
src/classes/avatar/core.py
Normal file
311
src/classes/avatar/core.py
Normal file
@@ -0,0 +1,311 @@
|
||||
"""
|
||||
Avatar 核心类
|
||||
|
||||
精简后的 Avatar 类,通过 Mixin 组合完整功能。
|
||||
"""
|
||||
import random
|
||||
from dataclasses import dataclass, field
|
||||
from enum import Enum
|
||||
from typing import Optional, List, TYPE_CHECKING
|
||||
|
||||
if TYPE_CHECKING:
|
||||
from src.classes.sect_ranks import SectRank
|
||||
|
||||
from src.classes.calendar import MonthStamp
|
||||
from src.classes.world import World
|
||||
from src.sim.save.avatar_save_mixin import AvatarSaveMixin
|
||||
from src.sim.load.avatar_load_mixin import AvatarLoadMixin
|
||||
from src.classes.tile import Tile
|
||||
from src.classes.region import Region
|
||||
from src.classes.cultivation import CultivationProgress
|
||||
from src.classes.root import Root
|
||||
from src.classes.technique import Technique, get_technique_by_sect
|
||||
from src.classes.age import Age
|
||||
from src.classes.event import Event
|
||||
from src.classes.action_runtime import ActionPlan, ActionInstance
|
||||
from src.classes.alignment import Alignment
|
||||
from src.classes.persona import Persona, get_random_compatible_personas
|
||||
from src.classes.item import Item
|
||||
from src.classes.weapon import Weapon
|
||||
from src.classes.auxiliary import Auxiliary
|
||||
from src.classes.magic_stone import MagicStone
|
||||
from src.classes.hp_and_mp import HP, HP_MAX_BY_REALM
|
||||
from src.classes.relation import Relation
|
||||
from src.classes.sect import Sect
|
||||
from src.classes.appearance import Appearance, get_random_appearance
|
||||
from src.classes.spirit_animal import SpiritAnimal
|
||||
from src.classes.long_term_objective import LongTermObjective
|
||||
from src.classes.nickname_data import Nickname
|
||||
from src.utils.config import CONFIG
|
||||
|
||||
# Mixin 导入
|
||||
from src.classes.avatar.effects_mixin import EffectsMixin
|
||||
from src.classes.avatar.inventory_mixin import InventoryMixin
|
||||
from src.classes.avatar.action_mixin import ActionMixin
|
||||
|
||||
persona_num = CONFIG.avatar.persona_num
|
||||
|
||||
|
||||
class Gender(Enum):
|
||||
MALE = "male"
|
||||
FEMALE = "female"
|
||||
|
||||
def __str__(self) -> str:
|
||||
return gender_strs.get(self, self.value)
|
||||
|
||||
|
||||
gender_strs = {
|
||||
Gender.MALE: "男",
|
||||
Gender.FEMALE: "女",
|
||||
}
|
||||
|
||||
# 历史事件的最大数量
|
||||
MAX_HISTORY_EVENTS = 10
|
||||
|
||||
|
||||
@dataclass
|
||||
class Avatar(
|
||||
AvatarSaveMixin,
|
||||
AvatarLoadMixin,
|
||||
EffectsMixin,
|
||||
InventoryMixin,
|
||||
ActionMixin,
|
||||
):
|
||||
"""
|
||||
NPC的类。
|
||||
包含了这个角色的一切信息。
|
||||
"""
|
||||
world: World
|
||||
name: str
|
||||
id: str
|
||||
birth_month_stamp: MonthStamp
|
||||
age: Age
|
||||
gender: Gender
|
||||
cultivation_progress: CultivationProgress = field(default_factory=lambda: CultivationProgress(0))
|
||||
pos_x: int = 0
|
||||
pos_y: int = 0
|
||||
tile: Optional[Tile] = None
|
||||
|
||||
root: Root = field(default_factory=lambda: random.choice(list(Root)))
|
||||
personas: List[Persona] = field(default_factory=list)
|
||||
technique: Technique | None = None
|
||||
history_events: List[Event] = field(default_factory=list)
|
||||
_pending_events: List[Event] = field(default_factory=list)
|
||||
current_action: Optional[ActionInstance] = None
|
||||
planned_actions: List[ActionPlan] = field(default_factory=list)
|
||||
thinking: str = ""
|
||||
short_term_objective: str = ""
|
||||
long_term_objective: Optional[LongTermObjective] = None
|
||||
magic_stone: MagicStone = field(default_factory=lambda: MagicStone(0))
|
||||
items: dict[Item, int] = field(default_factory=dict)
|
||||
hp: HP = field(default_factory=lambda: HP(0, 0))
|
||||
relations: dict["Avatar", Relation] = field(default_factory=dict)
|
||||
alignment: Alignment | None = None
|
||||
sect: Sect | None = None
|
||||
sect_rank: "SectRank | None" = None
|
||||
appearance: Appearance = field(default_factory=get_random_appearance)
|
||||
weapon: Optional[Weapon] = None
|
||||
weapon_proficiency: float = 0.0
|
||||
auxiliary: Optional[Auxiliary] = None
|
||||
spirit_animal: Optional[SpiritAnimal] = None
|
||||
nickname: Optional[Nickname] = None
|
||||
custom_pic_id: Optional[int] = None
|
||||
|
||||
is_dead: bool = False
|
||||
death_info: Optional[dict] = None
|
||||
|
||||
_new_action_set_this_step: bool = False
|
||||
_action_cd_last_months: dict[str, int] = field(default_factory=dict)
|
||||
|
||||
known_regions: set[int] = field(default_factory=set)
|
||||
|
||||
# ========== 宗门相关 ==========
|
||||
|
||||
def join_sect(self, sect: Sect, rank: "SectRank") -> None:
|
||||
"""加入宗门"""
|
||||
if self.is_dead:
|
||||
return
|
||||
if self.sect:
|
||||
self.leave_sect()
|
||||
self.sect = sect
|
||||
self.sect_rank = rank
|
||||
sect.add_member(self)
|
||||
|
||||
def leave_sect(self) -> None:
|
||||
"""退出宗门"""
|
||||
if self.sect:
|
||||
self.sect.remove_member(self)
|
||||
self.sect = None
|
||||
self.sect_rank = None
|
||||
|
||||
def get_sect_str(self) -> str:
|
||||
"""获取宗门显示名:有宗门则返回"宗门名+职位",否则返回"散修"。"""
|
||||
if self.sect is None:
|
||||
return "散修"
|
||||
if self.sect_rank is None:
|
||||
return self.sect.name
|
||||
from src.classes.sect_ranks import get_rank_display_name
|
||||
rank_name = get_rank_display_name(self.sect_rank, self.sect)
|
||||
return f"{self.sect.name}{rank_name}"
|
||||
|
||||
def get_sect_rank_name(self) -> str:
|
||||
"""获取宗门职位的显示名称"""
|
||||
if self.sect is None or self.sect_rank is None:
|
||||
return "散修"
|
||||
from src.classes.sect_ranks import get_rank_display_name
|
||||
return get_rank_display_name(self.sect_rank, self.sect)
|
||||
|
||||
# ========== 死亡相关 ==========
|
||||
|
||||
def set_dead(self, reason: str, time: MonthStamp) -> None:
|
||||
"""设置角色死亡状态。"""
|
||||
if self.is_dead:
|
||||
return
|
||||
|
||||
self.is_dead = True
|
||||
self.death_info = {
|
||||
"time": int(time),
|
||||
"reason": reason,
|
||||
"location": (self.pos_x, self.pos_y)
|
||||
}
|
||||
|
||||
self.planned_actions.clear()
|
||||
self.current_action = None
|
||||
self._pending_events.clear()
|
||||
self.thinking = ""
|
||||
self.short_term_objective = ""
|
||||
|
||||
if self.sect:
|
||||
self.sect.remove_member(self)
|
||||
|
||||
def death_by_old_age(self) -> bool:
|
||||
"""检查是否老死"""
|
||||
return self.age.death_by_old_age(self.cultivation_progress.realm)
|
||||
|
||||
# ========== 年龄与修为 ==========
|
||||
|
||||
def update_age(self, current_month_stamp: MonthStamp):
|
||||
"""更新年龄"""
|
||||
self.age.update_age(current_month_stamp, self.birth_month_stamp)
|
||||
|
||||
def update_cultivation(self, new_level: int):
|
||||
"""更新修仙进度,并在境界提升时更新寿命和宗门职位"""
|
||||
old_realm = self.cultivation_progress.realm
|
||||
self.cultivation_progress.level = new_level
|
||||
self.cultivation_progress.realm = self.cultivation_progress.get_realm(new_level)
|
||||
|
||||
if self.cultivation_progress.realm != old_realm:
|
||||
self.age.update_realm(self.cultivation_progress.realm)
|
||||
self.recalc_effects()
|
||||
from src.classes.sect_ranks import check_and_promote_sect_rank
|
||||
check_and_promote_sect_rank(self, old_realm, self.cultivation_progress.realm)
|
||||
|
||||
# ========== 区域与位置 ==========
|
||||
|
||||
def is_in_region(self, region: Region | None) -> bool:
|
||||
current_region = self.tile.region
|
||||
if current_region is None:
|
||||
tile = self.world.map.get_tile(self.pos_x, self.pos_y)
|
||||
current_region = tile.region
|
||||
return current_region == region
|
||||
|
||||
def get_co_region_avatars(self, avatars: List["Avatar"]) -> List["Avatar"]:
|
||||
"""返回与自己处于同一区域的角色列表(不含自己)。"""
|
||||
if self.tile is None:
|
||||
return []
|
||||
same_region: list[Avatar] = []
|
||||
for other in avatars:
|
||||
if other is self or other.tile is None:
|
||||
continue
|
||||
if other.tile.region == self.tile.region:
|
||||
same_region.append(other)
|
||||
return same_region
|
||||
|
||||
def _init_known_regions(self):
|
||||
"""初始化已知区域:当前位置 + 宗门驻地"""
|
||||
if self.tile and self.tile.region:
|
||||
self.known_regions.add(self.tile.region.id)
|
||||
|
||||
if self.sect:
|
||||
for r in self.world.map.sect_regions.values():
|
||||
if r.sect_id == self.sect.id:
|
||||
self.known_regions.add(r.id)
|
||||
break
|
||||
|
||||
# ========== 关系相关 ==========
|
||||
|
||||
def set_relation(self, other: "Avatar", relation: Relation) -> None:
|
||||
"""设置与另一个角色的关系。"""
|
||||
from src.classes.relations import set_relation
|
||||
set_relation(self, other, relation)
|
||||
|
||||
def get_relation(self, other: "Avatar") -> Optional[Relation]:
|
||||
"""获取与另一个角色的关系。"""
|
||||
from src.classes.relations import get_relation
|
||||
return get_relation(self, other)
|
||||
|
||||
def clear_relation(self, other: "Avatar") -> None:
|
||||
"""清除与另一个角色的关系。"""
|
||||
from src.classes.relations import clear_relation
|
||||
clear_relation(self, other)
|
||||
|
||||
# ========== 信息展示(委托) ==========
|
||||
|
||||
def get_info(self, detailed: bool = False) -> dict:
|
||||
from src.classes.avatar.info_presenter import get_avatar_info
|
||||
return get_avatar_info(self, detailed)
|
||||
|
||||
def get_structured_info(self) -> dict:
|
||||
from src.classes.avatar.info_presenter import get_avatar_structured_info
|
||||
return get_avatar_structured_info(self)
|
||||
|
||||
def get_hover_info(self) -> list[str]:
|
||||
from src.classes.avatar.info_presenter import get_avatar_hover_info
|
||||
return get_avatar_hover_info(self)
|
||||
|
||||
def get_expanded_info(
|
||||
self,
|
||||
co_region_avatars: Optional[List["Avatar"]] = None,
|
||||
other_avatar: Optional["Avatar"] = None,
|
||||
detailed: bool = False
|
||||
) -> dict:
|
||||
from src.classes.avatar.info_presenter import get_avatar_expanded_info
|
||||
return get_avatar_expanded_info(self, co_region_avatars, other_avatar, detailed)
|
||||
|
||||
def get_other_avatar_info(self, other_avatar: "Avatar") -> str:
|
||||
from src.classes.avatar.info_presenter import get_other_avatar_info
|
||||
return get_other_avatar_info(self, other_avatar)
|
||||
|
||||
# ========== 魔法方法 ==========
|
||||
|
||||
def __post_init__(self):
|
||||
"""在Avatar创建后自动初始化tile和HP"""
|
||||
self.tile = self.world.map.get_tile(self.pos_x, self.pos_y)
|
||||
|
||||
max_hp = HP_MAX_BY_REALM.get(self.cultivation_progress.realm, 100)
|
||||
self.hp = HP(max_hp, max_hp)
|
||||
|
||||
if not self.personas:
|
||||
self.personas = get_random_compatible_personas(persona_num, avatar=self)
|
||||
|
||||
if self.technique is None:
|
||||
self.technique = get_technique_by_sect(self.sect)
|
||||
|
||||
if self.sect:
|
||||
self.sect.add_member(self)
|
||||
|
||||
if self.alignment is None:
|
||||
if self.sect is not None:
|
||||
self.alignment = self.sect.alignment
|
||||
else:
|
||||
self.alignment = random.choice(list(Alignment))
|
||||
|
||||
self.recalc_effects()
|
||||
self._init_known_regions()
|
||||
|
||||
def __hash__(self) -> int:
|
||||
return hash(self.id)
|
||||
|
||||
def __str__(self) -> str:
|
||||
return str(self.get_info(detailed=False))
|
||||
|
||||
119
src/classes/avatar/effects_mixin.py
Normal file
119
src/classes/avatar/effects_mixin.py
Normal file
@@ -0,0 +1,119 @@
|
||||
"""
|
||||
Avatar 效果计算 Mixin
|
||||
|
||||
负责角色效果的计算和应用。
|
||||
"""
|
||||
from __future__ import annotations
|
||||
|
||||
from collections import defaultdict
|
||||
from typing import TYPE_CHECKING
|
||||
|
||||
if TYPE_CHECKING:
|
||||
from src.classes.avatar.core import Avatar
|
||||
|
||||
from src.classes.effect import _merge_effects, _evaluate_conditional_effect
|
||||
from src.classes.hp_and_mp import HP_MAX_BY_REALM
|
||||
|
||||
|
||||
class EffectsMixin:
|
||||
"""效果计算相关方法"""
|
||||
|
||||
@property
|
||||
def effects(self: "Avatar") -> dict[str, object]:
|
||||
"""
|
||||
合并所有来源的效果:宗门、功法、灵根、特质、兵器、辅助装备、灵兽、天地灵机
|
||||
"""
|
||||
merged: dict[str, object] = defaultdict(str)
|
||||
# 来自宗门
|
||||
if self.sect is not None:
|
||||
evaluated = _evaluate_conditional_effect(self.sect.effects, self)
|
||||
merged = _merge_effects(merged, evaluated)
|
||||
# 来自功法
|
||||
evaluated = _evaluate_conditional_effect(self.technique.effects, self)
|
||||
merged = _merge_effects(merged, evaluated)
|
||||
# 来自灵根
|
||||
evaluated = _evaluate_conditional_effect(self.root.effects, self)
|
||||
merged = _merge_effects(merged, evaluated)
|
||||
# 来自特质(persona)
|
||||
for persona in self.personas:
|
||||
evaluated = _evaluate_conditional_effect(persona.effects, self)
|
||||
merged = _merge_effects(merged, evaluated)
|
||||
# 来自兵器
|
||||
if self.weapon is not None:
|
||||
evaluated = _evaluate_conditional_effect(self.weapon.effects, self)
|
||||
merged = _merge_effects(merged, evaluated)
|
||||
# 来自辅助装备
|
||||
if self.auxiliary is not None:
|
||||
evaluated = _evaluate_conditional_effect(self.auxiliary.effects, self)
|
||||
merged = _merge_effects(merged, evaluated)
|
||||
# 来自灵兽
|
||||
if self.spirit_animal is not None:
|
||||
evaluated = _evaluate_conditional_effect(self.spirit_animal.effects, self)
|
||||
merged = _merge_effects(merged, evaluated)
|
||||
# 来自天地灵机(世界级buff/debuff)
|
||||
if self.world.current_phenomenon is not None:
|
||||
evaluated = _evaluate_conditional_effect(self.world.current_phenomenon.effects, self)
|
||||
merged = _merge_effects(merged, evaluated)
|
||||
# 评估动态效果表达式:值以 "eval(...)" 形式给出
|
||||
final: dict[str, object] = {}
|
||||
for k, v in merged.items():
|
||||
if isinstance(v, str):
|
||||
s = v.strip()
|
||||
if s.startswith("eval(") and s.endswith(")"):
|
||||
expr = s[5:-1]
|
||||
final[k] = eval(expr, {"__builtins__": {}}, {"avatar": self})
|
||||
continue
|
||||
final[k] = v
|
||||
return final
|
||||
|
||||
def recalc_effects(self: "Avatar") -> None:
|
||||
"""
|
||||
重新计算所有长期效果
|
||||
在装备更换、突破境界等情况下调用
|
||||
|
||||
当前包括:
|
||||
- HP 最大值
|
||||
- 寿命最大值
|
||||
"""
|
||||
# 计算基础最大值(基于境界)
|
||||
base_max_hp = HP_MAX_BY_REALM.get(self.cultivation_progress.realm, 100)
|
||||
|
||||
# 访问 self.effects 会触发 @property,重新 merge 所有 effects
|
||||
effects = self.effects
|
||||
extra_max_hp = int(effects.get("extra_max_hp", 0))
|
||||
extra_max_lifespan = int(effects.get("extra_max_lifespan", 0))
|
||||
|
||||
# 计算新的最大值
|
||||
new_max_hp = base_max_hp + extra_max_hp
|
||||
|
||||
# 更新最大值
|
||||
self.hp.max = new_max_hp
|
||||
|
||||
# 更新寿命
|
||||
if self.age:
|
||||
self.age.max_lifespan = self.age.base_max_lifespan + extra_max_lifespan
|
||||
|
||||
# 调整当前值(不超过新的最大值)
|
||||
if self.hp.cur > new_max_hp:
|
||||
self.hp.cur = new_max_hp
|
||||
|
||||
def update_time_effect(self: "Avatar") -> None:
|
||||
"""
|
||||
随时间更新的被动效果。
|
||||
当前实现:当 HP 未满时,回复最大生命值的 1%(受HP恢复速率加成影响)。
|
||||
"""
|
||||
if self.hp.cur < self.hp.max:
|
||||
base_recover = self.hp.max * 0.01
|
||||
|
||||
# 应用HP恢复速率加成
|
||||
recovery_rate_raw = self.effects.get("extra_hp_recovery_rate", 0.0)
|
||||
recovery_rate_multiplier = 1.0 + float(recovery_rate_raw or 0.0)
|
||||
|
||||
recover_amount = int(base_recover * recovery_rate_multiplier)
|
||||
self.hp.recover(recover_amount)
|
||||
|
||||
@property
|
||||
def move_step_length(self: "Avatar") -> int:
|
||||
"""获取角色的移动步长"""
|
||||
return self.cultivation_progress.get_move_step()
|
||||
|
||||
349
src/classes/avatar/info_presenter.py
Normal file
349
src/classes/avatar/info_presenter.py
Normal file
@@ -0,0 +1,349 @@
|
||||
"""
|
||||
Avatar 信息展示模块
|
||||
|
||||
将信息格式化逻辑从 Avatar 类中分离,作为独立函数提供。
|
||||
"""
|
||||
from __future__ import annotations
|
||||
|
||||
from typing import TYPE_CHECKING, Optional, List
|
||||
|
||||
if TYPE_CHECKING:
|
||||
from src.classes.avatar.core import Avatar
|
||||
|
||||
from src.classes.battle import get_base_strength
|
||||
from src.classes.relation import get_relation_label
|
||||
from src.utils.config import CONFIG
|
||||
|
||||
|
||||
def get_avatar_info(avatar: "Avatar", detailed: bool = False) -> dict:
|
||||
"""
|
||||
获取 avatar 的信息,返回 dict;根据 detailed 控制信息粒度。
|
||||
"""
|
||||
region = avatar.tile.region if avatar.tile is not None else None
|
||||
from src.classes.relation import get_relations_strs
|
||||
relation_lines = get_relations_strs(avatar, max_lines=8)
|
||||
relations_info = ";".join(relation_lines) if relation_lines else "无"
|
||||
magic_stone_info = str(avatar.magic_stone)
|
||||
|
||||
from src.classes.sect import get_sect_info_with_rank
|
||||
|
||||
if detailed:
|
||||
weapon_info = f"{avatar.weapon.get_detailed_info()},熟练度:{avatar.weapon_proficiency:.1f}%"
|
||||
auxiliary_info = avatar.auxiliary.get_detailed_info() if avatar.auxiliary is not None else "无"
|
||||
sect_info = get_sect_info_with_rank(avatar, detailed=True)
|
||||
alignment_info = avatar.alignment.get_detailed_info() if avatar.alignment is not None else "未知"
|
||||
region_info = region.get_detailed_info() if region is not None else "无"
|
||||
root_info = avatar.root.get_detailed_info()
|
||||
technique_info = avatar.technique.get_detailed_info() if avatar.technique is not None else "无"
|
||||
cultivation_info = avatar.cultivation_progress.get_detailed_info()
|
||||
personas_info = ", ".join([p.get_detailed_info() for p in avatar.personas]) if avatar.personas else "无"
|
||||
items_info = ",".join([f"{item.get_detailed_info()}x{quantity}" for item, quantity in avatar.items.items()]) if avatar.items else "无"
|
||||
appearance_info = avatar.appearance.get_detailed_info(avatar.gender)
|
||||
spirit_animal_info = avatar.spirit_animal.get_info() if avatar.spirit_animal is not None else "无"
|
||||
else:
|
||||
weapon_info = avatar.weapon.get_info() if avatar.weapon is not None else "无"
|
||||
auxiliary_info = avatar.auxiliary.get_info() if avatar.auxiliary is not None else "无"
|
||||
sect_info = get_sect_info_with_rank(avatar, detailed=False)
|
||||
region_info = region.get_info() if region is not None else "无"
|
||||
alignment_info = avatar.alignment.get_info() if avatar.alignment is not None else "未知"
|
||||
root_info = avatar.root.get_info()
|
||||
technique_info = avatar.technique.get_info() if avatar.technique is not None else "无"
|
||||
cultivation_info = avatar.cultivation_progress.get_info()
|
||||
personas_info = ", ".join([p.get_detailed_info() for p in avatar.personas]) if avatar.personas else "无"
|
||||
items_info = ",".join([f"{item.get_info()}x{quantity}" for item, quantity in avatar.items.items()]) if avatar.items else "无"
|
||||
appearance_info = avatar.appearance.get_info()
|
||||
spirit_animal_info = avatar.spirit_animal.get_info() if avatar.spirit_animal is not None else "无"
|
||||
|
||||
info_dict = {
|
||||
"名字": avatar.name,
|
||||
"性别": str(avatar.gender),
|
||||
"年龄": str(avatar.age),
|
||||
"hp": str(avatar.hp),
|
||||
"灵石": magic_stone_info,
|
||||
"关系": relations_info,
|
||||
"宗门": sect_info,
|
||||
"阵营": alignment_info,
|
||||
"地区": region_info,
|
||||
"灵根": root_info,
|
||||
"功法": technique_info,
|
||||
"境界": cultivation_info,
|
||||
"特质": personas_info,
|
||||
"物品": items_info,
|
||||
"外貌": appearance_info,
|
||||
"兵器": weapon_info,
|
||||
"辅助装备": auxiliary_info,
|
||||
}
|
||||
# 绰号:仅在存在时显示
|
||||
if avatar.nickname is not None:
|
||||
info_dict["绰号"] = avatar.nickname.value
|
||||
# 灵兽:仅在存在时显示
|
||||
if avatar.spirit_animal is not None:
|
||||
info_dict["灵兽"] = spirit_animal_info
|
||||
# 长期目标:仅在存在时显示
|
||||
if avatar.long_term_objective is not None:
|
||||
info_dict["长期目标"] = avatar.long_term_objective.content
|
||||
# 短期目标:仅在存在时显示
|
||||
if avatar.short_term_objective:
|
||||
info_dict["短期目标"] = avatar.short_term_objective
|
||||
return info_dict
|
||||
|
||||
|
||||
def get_avatar_structured_info(avatar: "Avatar") -> dict:
|
||||
"""
|
||||
获取结构化的角色信息,用于前端展示和交互。
|
||||
"""
|
||||
# 基础信息
|
||||
info = {
|
||||
"id": avatar.id,
|
||||
"name": avatar.name,
|
||||
"gender": str(avatar.gender),
|
||||
"age": avatar.age.age,
|
||||
"lifespan": avatar.age.max_lifespan,
|
||||
"realm": avatar.cultivation_progress.realm.value,
|
||||
"level": avatar.cultivation_progress.level,
|
||||
"hp": {"cur": avatar.hp.cur, "max": avatar.hp.max},
|
||||
"alignment": str(avatar.alignment) if avatar.alignment else "未知",
|
||||
"magic_stone": avatar.magic_stone.value,
|
||||
"thinking": avatar.thinking,
|
||||
"short_term_objective": avatar.short_term_objective,
|
||||
"long_term_objective": avatar.long_term_objective.content if avatar.long_term_objective else "",
|
||||
"nickname": avatar.nickname.value if avatar.nickname else None,
|
||||
"nickname_reason": avatar.nickname.reason if avatar.nickname else None,
|
||||
"is_dead": avatar.is_dead,
|
||||
"death_info": avatar.death_info,
|
||||
}
|
||||
|
||||
# 1. 特质 (Personas)
|
||||
info["personas"] = [p.get_structured_info() for p in avatar.personas]
|
||||
|
||||
# 2. 功法 (Technique)
|
||||
if avatar.technique:
|
||||
info["technique"] = avatar.technique.get_structured_info()
|
||||
else:
|
||||
info["technique"] = None
|
||||
|
||||
# 3. 宗门 (Sect)
|
||||
if avatar.sect:
|
||||
sect_info = avatar.sect.get_structured_info()
|
||||
if avatar.sect_rank:
|
||||
from src.classes.sect_ranks import get_rank_display_name
|
||||
sect_info["rank"] = get_rank_display_name(avatar.sect_rank, avatar.sect)
|
||||
else:
|
||||
sect_info["rank"] = "弟子"
|
||||
info["sect"] = sect_info
|
||||
else:
|
||||
info["sect"] = None
|
||||
|
||||
# 补充:阵营详情
|
||||
from src.classes.alignment import alignment_infos, alignment_strs
|
||||
info["alignment"] = str(avatar.alignment) if avatar.alignment else "未知"
|
||||
if avatar.alignment:
|
||||
cn_name = alignment_strs.get(avatar.alignment, avatar.alignment.value)
|
||||
desc = alignment_infos.get(avatar.alignment, "")
|
||||
info["alignment_detail"] = {
|
||||
"name": cn_name,
|
||||
"desc": desc,
|
||||
}
|
||||
|
||||
# 4. 装备 (Weapon & Auxiliary)
|
||||
if avatar.weapon:
|
||||
w_info = avatar.weapon.get_structured_info()
|
||||
w_info["proficiency"] = f"{avatar.weapon_proficiency:.1f}%"
|
||||
info["weapon"] = w_info
|
||||
else:
|
||||
info["weapon"] = None
|
||||
|
||||
if avatar.auxiliary:
|
||||
info["auxiliary"] = avatar.auxiliary.get_structured_info()
|
||||
else:
|
||||
info["auxiliary"] = None
|
||||
|
||||
# 5. 物品 (Items)
|
||||
items_list = []
|
||||
for item, count in avatar.items.items():
|
||||
i_info = item.get_structured_info()
|
||||
i_info["count"] = count
|
||||
items_list.append(i_info)
|
||||
info["items"] = items_list
|
||||
|
||||
# 6. 关系 (Relations)
|
||||
relations_list = []
|
||||
for other, relation in avatar.relations.items():
|
||||
relations_list.append({
|
||||
"target_id": other.id,
|
||||
"name": other.name,
|
||||
"relation": get_relation_label(relation, avatar, other),
|
||||
"realm": other.cultivation_progress.realm.value,
|
||||
"sect": other.sect.name if other.sect else "散修"
|
||||
})
|
||||
info["relations"] = relations_list
|
||||
|
||||
# 7. 外貌
|
||||
info["appearance"] = avatar.appearance.get_info()
|
||||
|
||||
# 8. 灵根
|
||||
from src.classes.root import format_root_cn
|
||||
root_str = format_root_cn(avatar.root)
|
||||
info["root"] = root_str
|
||||
info["root_detail"] = {
|
||||
"name": root_str,
|
||||
"desc": f"包含元素:{'、'.join(str(e) for e in avatar.root.elements)}",
|
||||
"effect_desc": avatar.root.effect_desc
|
||||
}
|
||||
|
||||
# 9. 灵兽
|
||||
if avatar.spirit_animal:
|
||||
info["spirit_animal"] = avatar.spirit_animal.get_structured_info()
|
||||
|
||||
return info
|
||||
|
||||
|
||||
def get_avatar_hover_info(avatar: "Avatar") -> list[str]:
|
||||
"""
|
||||
返回用于前端悬浮提示的多行信息。
|
||||
"""
|
||||
def add_kv(lines: list[str], key: str, value: object) -> None:
|
||||
lines.append(f"{key}: {value}")
|
||||
|
||||
def add_section(lines: list[str], title: str, body: list[str]) -> None:
|
||||
lines.append("")
|
||||
lines.append(f"{title}:")
|
||||
lines.extend(body)
|
||||
|
||||
lines: list[str] = []
|
||||
# 基础信息
|
||||
if avatar.nickname:
|
||||
add_kv(lines, "绰号", f"「{avatar.nickname.value}」")
|
||||
|
||||
add_kv(lines, "性别", avatar.gender)
|
||||
add_kv(lines, "年龄", avatar.age)
|
||||
add_kv(lines, "外貌", avatar.appearance.get_info())
|
||||
add_kv(lines, "阵营", avatar.alignment)
|
||||
add_kv(lines, "境界", str(avatar.cultivation_progress))
|
||||
add_kv(lines, "HP", avatar.hp)
|
||||
add_kv(lines, "战斗力", int(get_base_strength(avatar)))
|
||||
add_kv(lines, "宗门", avatar.get_sect_str())
|
||||
|
||||
from src.classes.root import format_root_cn
|
||||
add_kv(lines, "灵根", format_root_cn(avatar.root))
|
||||
|
||||
tech_str = avatar.technique.get_colored_info() if avatar.technique is not None else "无"
|
||||
add_kv(lines, "功法", tech_str)
|
||||
|
||||
if avatar.personas:
|
||||
persona_parts = [p.get_colored_info() for p in avatar.personas]
|
||||
add_kv(lines, "特质", ", ".join(persona_parts))
|
||||
|
||||
add_kv(lines, "灵石", str(avatar.magic_stone))
|
||||
|
||||
# 物品
|
||||
if avatar.items:
|
||||
items_lines = [f" {item.name} x{quantity}" for item, quantity in avatar.items.items()]
|
||||
add_section(lines, "物品", items_lines)
|
||||
else:
|
||||
add_kv(lines, "物品", "无")
|
||||
|
||||
# 思考与目标
|
||||
if avatar.thinking:
|
||||
add_section(lines, "思考", [avatar.thinking])
|
||||
if avatar.long_term_objective:
|
||||
add_section(lines, "长期目标", [avatar.long_term_objective.content])
|
||||
if avatar.short_term_objective:
|
||||
add_section(lines, "短期目标", [avatar.short_term_objective])
|
||||
|
||||
# 兵器(必有,使用颜色标记等级)
|
||||
if avatar.weapon is not None:
|
||||
weapon_text = avatar.weapon.get_colored_info()
|
||||
if avatar.weapon.desc:
|
||||
weapon_text += f"({avatar.weapon.desc})"
|
||||
add_kv(lines, "兵器", weapon_text)
|
||||
|
||||
# 辅助装备(可选,使用颜色标记等级)
|
||||
if avatar.auxiliary is not None:
|
||||
auxiliary_text = avatar.auxiliary.get_colored_info()
|
||||
if avatar.auxiliary.desc:
|
||||
auxiliary_text += f"({avatar.auxiliary.desc})"
|
||||
add_kv(lines, "辅助装备", auxiliary_text)
|
||||
else:
|
||||
add_kv(lines, "辅助装备", "无")
|
||||
|
||||
# 灵兽:仅在存在时显示
|
||||
if avatar.spirit_animal is not None:
|
||||
add_kv(lines, "灵兽", avatar.spirit_animal.get_info())
|
||||
|
||||
# 关系(从自身视角分组展示)
|
||||
from src.classes.relation import get_relations_strs
|
||||
relation_lines = get_relations_strs(avatar, max_lines=15)
|
||||
if relation_lines:
|
||||
add_section(lines, "关系", [f" {s}" for s in relation_lines])
|
||||
else:
|
||||
add_kv(lines, "关系", "无")
|
||||
|
||||
return lines
|
||||
|
||||
|
||||
def get_avatar_expanded_info(
|
||||
avatar: "Avatar",
|
||||
co_region_avatars: Optional[List["Avatar"]] = None,
|
||||
other_avatar: Optional["Avatar"] = None,
|
||||
detailed: bool = False
|
||||
) -> dict:
|
||||
"""
|
||||
获取角色的扩展信息,包含基础信息、观察到的角色和事件历史。
|
||||
|
||||
Args:
|
||||
avatar: 目标角色
|
||||
co_region_avatars: 同区域的其他角色列表,用于"观察到的角色"字段
|
||||
other_avatar: 另一个角色,如果提供则返回两人共同经历的事件,否则返回单人事件
|
||||
detailed: 是否返回详细信息
|
||||
"""
|
||||
info = get_avatar_info(avatar, detailed=detailed)
|
||||
|
||||
observed: list[str] = []
|
||||
if co_region_avatars:
|
||||
for other in co_region_avatars[:8]:
|
||||
observed.append(f"{other.name},境界:{other.cultivation_progress.get_info()}")
|
||||
|
||||
# 历史事件改为从全局事件管理器分类查询
|
||||
em = avatar.world.event_manager
|
||||
major_limit = CONFIG.social.major_event_context_num
|
||||
minor_limit = CONFIG.social.minor_event_context_num
|
||||
|
||||
# 根据是否提供 other_avatar 决定获取单人事件还是双人共同事件
|
||||
if other_avatar is not None:
|
||||
major_events = em.get_major_events_between(avatar.id, other_avatar.id, limit=major_limit)
|
||||
minor_events = em.get_minor_events_between(avatar.id, other_avatar.id, limit=minor_limit)
|
||||
else:
|
||||
major_events = em.get_major_events_by_avatar(avatar.id, limit=major_limit)
|
||||
minor_events = em.get_minor_events_by_avatar(avatar.id, limit=minor_limit)
|
||||
|
||||
major_list = [str(e) for e in major_events]
|
||||
minor_list = [str(e) for e in minor_events]
|
||||
|
||||
info["周围角色"] = observed
|
||||
info["重大事件"] = major_list
|
||||
info["短期事件"] = minor_list
|
||||
info["长期目标"] = avatar.long_term_objective.content if avatar.long_term_objective else "无"
|
||||
return info
|
||||
|
||||
|
||||
def get_other_avatar_info(from_avatar: "Avatar", to_avatar: "Avatar") -> str:
|
||||
"""
|
||||
仅显示几个字段:名字、绰号、境界、关系、宗门、阵营、外貌、功法、武器、辅助装备、HP
|
||||
"""
|
||||
nickname = to_avatar.nickname.value if to_avatar.nickname else "无"
|
||||
sect = to_avatar.sect.name if to_avatar.sect else "散修"
|
||||
tech = to_avatar.technique.get_info() if to_avatar.technique else "无"
|
||||
weapon = to_avatar.weapon.get_info() if to_avatar.weapon else "无"
|
||||
aux = to_avatar.auxiliary.get_info() if to_avatar.auxiliary else "无"
|
||||
alignment = to_avatar.alignment
|
||||
|
||||
# 关系可能为空
|
||||
relation = from_avatar.get_relation(to_avatar) or "无"
|
||||
|
||||
return (
|
||||
f"{to_avatar.name},绰号:{nickname},境界:{to_avatar.cultivation_progress.get_info()},"
|
||||
f"关系:{relation},宗门:{sect},阵营:{alignment},"
|
||||
f"外貌:{to_avatar.appearance.get_info()},功法:{tech},兵器:{weapon},辅助:{aux},HP:{to_avatar.hp}"
|
||||
)
|
||||
|
||||
119
src/classes/avatar/inventory_mixin.py
Normal file
119
src/classes/avatar/inventory_mixin.py
Normal file
@@ -0,0 +1,119 @@
|
||||
"""
|
||||
Avatar 物品与装备管理 Mixin
|
||||
"""
|
||||
from __future__ import annotations
|
||||
|
||||
from typing import TYPE_CHECKING, Optional
|
||||
|
||||
if TYPE_CHECKING:
|
||||
from src.classes.avatar.core import Avatar
|
||||
from src.classes.item import Item
|
||||
from src.classes.weapon import Weapon
|
||||
from src.classes.auxiliary import Auxiliary
|
||||
|
||||
|
||||
class InventoryMixin:
|
||||
"""物品与装备管理相关方法"""
|
||||
|
||||
def add_item(self: "Avatar", item: "Item", quantity: int = 1) -> None:
|
||||
"""
|
||||
添加物品到背包
|
||||
|
||||
Args:
|
||||
item: 要添加的物品
|
||||
quantity: 添加数量,默认为1
|
||||
"""
|
||||
if quantity <= 0:
|
||||
return
|
||||
|
||||
if item in self.items:
|
||||
self.items[item] += quantity
|
||||
else:
|
||||
self.items[item] = quantity
|
||||
|
||||
def remove_item(self: "Avatar", item: "Item", quantity: int = 1) -> bool:
|
||||
"""
|
||||
从背包移除物品
|
||||
|
||||
Args:
|
||||
item: 要移除的物品
|
||||
quantity: 移除数量,默认为1
|
||||
|
||||
Returns:
|
||||
bool: 是否成功移除(如果物品不足则返回False)
|
||||
"""
|
||||
if quantity <= 0:
|
||||
return True
|
||||
|
||||
if item not in self.items:
|
||||
return False
|
||||
|
||||
if self.items[item] < quantity:
|
||||
return False
|
||||
|
||||
self.items[item] -= quantity
|
||||
|
||||
# 如果数量为0,从字典中移除该物品
|
||||
if self.items[item] == 0:
|
||||
del self.items[item]
|
||||
|
||||
return True
|
||||
|
||||
def has_item(self: "Avatar", item: "Item", quantity: int = 1) -> bool:
|
||||
"""
|
||||
检查是否拥有足够数量的物品
|
||||
|
||||
Args:
|
||||
item: 要检查的物品
|
||||
quantity: 需要的数量,默认为1
|
||||
|
||||
Returns:
|
||||
bool: 是否拥有足够数量的物品
|
||||
"""
|
||||
return item in self.items and self.items[item] >= quantity
|
||||
|
||||
def get_item_quantity(self: "Avatar", item: "Item") -> int:
|
||||
"""
|
||||
获取指定物品的数量
|
||||
|
||||
Args:
|
||||
item: 要查询的物品
|
||||
|
||||
Returns:
|
||||
int: 物品数量,如果没有该物品则返回0
|
||||
"""
|
||||
return self.items.get(item, 0)
|
||||
|
||||
def change_weapon(self: "Avatar", new_weapon: "Weapon") -> None:
|
||||
"""
|
||||
更换兵器,熟练度归零,并重新计算长期效果
|
||||
|
||||
Args:
|
||||
new_weapon: 新的兵器
|
||||
"""
|
||||
self.weapon = new_weapon
|
||||
self.weapon_proficiency = 0.0
|
||||
self.recalc_effects()
|
||||
|
||||
def change_auxiliary(self: "Avatar", new_auxiliary: Optional["Auxiliary"]) -> None:
|
||||
"""
|
||||
更换辅助装备,并重新计算长期效果
|
||||
|
||||
Args:
|
||||
new_auxiliary: 新的辅助装备(可为 None 表示卸下)
|
||||
"""
|
||||
self.auxiliary = new_auxiliary
|
||||
self.recalc_effects()
|
||||
|
||||
def increase_weapon_proficiency(self: "Avatar", amount: float) -> None:
|
||||
"""
|
||||
增加兵器熟练度,上限100
|
||||
|
||||
Args:
|
||||
amount: 增加的熟练度值
|
||||
"""
|
||||
# 应用extra_weapon_proficiency_gain效果(倍率加成)
|
||||
gain_multiplier = 1.0 + self.effects.get("extra_weapon_proficiency_gain", 0.0)
|
||||
actual_amount = amount * gain_multiplier
|
||||
self.weapon_proficiency = min(100.0, self.weapon_proficiency + actual_amount)
|
||||
|
||||
Reference in New Issue
Block a user