import math from typing import List, Optional, Tuple, Callable from src.classes.avatar import Avatar from src.utils.text_wrap import wrap_text # 顶部状态栏高度（像素） STATUS_BAR_HEIGHT = 32 TOOLTIP_MIN_WIDTH = 260 def wrap_lines_for_tooltip(lines: List[str], max_chars_per_line: int = 28) -> List[str]: """ 将一组 tooltip 行进行字符级换行： - 对形如 "前缀: 内容" 的行，仅对内容部分换行，并在续行添加两个空格缩进 - 其他行超过宽度则直接按宽度切分 """ wrapped: List[str] = [] for line in lines: # 仅处理简单前缀（到第一个": "为界） split_idx = line.find(": ") if split_idx != -1: prefix = line[: split_idx + 2] content = line[split_idx + 2 :] segs = wrap_text(content, max_chars_per_line) if segs: wrapped.append(prefix + segs[0]) for seg in segs[1:]: wrapped.append(" " + seg) else: wrapped.append(line) continue # 无前缀情形：必要时整行切分 if len(line) > max_chars_per_line: wrapped.extend(wrap_text(line, max_chars_per_line)) else: wrapped.append(line) return wrapped def draw_grid(pygame_mod, screen, colors, map_obj, ts: int, m: int, top_offset: int = 0): grid_color = colors["grid"] for gx in range(map_obj.width + 1): start_pos = (m + gx * ts, m + top_offset) end_pos = (m + gx * ts, m + top_offset + map_obj.height * ts) pygame_mod.draw.line(screen, grid_color, start_pos, end_pos, 1) for gy in range(map_obj.height + 1): start_pos = (m, m + top_offset + gy * ts) end_pos = (m + map_obj.width * ts, m + top_offset + gy * ts) pygame_mod.draw.line(screen, grid_color, start_pos, end_pos, 1) def draw_map(pygame_mod, screen, colors, world, tile_images, ts: int, m: int, top_offset: int = 0): map_obj = world.map for y in range(map_obj.height): for x in range(map_obj.width): tile = map_obj.get_tile(x, y) tile_image = tile_images.get(tile.type) if tile_image: pos = (m + x * ts, m + top_offset + y * ts) screen.blit(tile_image, pos) else: color = (80, 80, 80) rect = pygame_mod.Rect(m + x * ts, m + top_offset + y * ts, ts, ts) pygame_mod.draw.rect(screen, color, rect) draw_grid(pygame_mod, screen, colors, map_obj, ts, m, top_offset) def draw_sect_headquarters(pygame_mod, screen, world, sect_images: dict, ts: int, m: int, top_offset: int = 0): """ 在底图绘制完成后叠加绘制宗门总部（2x2 tile）。 以区域左上角（north_west_cor）为锚点绘制。 """ for region in world.map.regions.values(): if getattr(region, "get_region_type", lambda: "")() != "sect": continue img_path: str | None = getattr(region, "image_path", None) if not img_path: # 可回退到按名称找图：期望 assets/sects/{region.name}.png key = str(getattr(region, "name", "")) image = sect_images.get(key) else: key = str(pygame_mod.Path(img_path).stem) if hasattr(pygame_mod, "Path") else img_path.split("/")[-1].split("\\")[-1].split(".")[0] image = sect_images.get(key) if not image: # 未加载到图片则跳过 continue try: nw = tuple(map(int, str(getattr(region, "north_west_cor", "0,0")).split(","))) except Exception: continue x_px = m + nw[0] * ts y_px = m + top_offset + nw[1] * ts screen.blit(image, (x_px, y_px)) def _is_small_square_region(region) -> int: """ 若为 2x2 或 3x3 的矩形/正方形区域，返回边长（2或3）；否则返回0。 """ try: nw = tuple(map(int, str(getattr(region, "north_west_cor", "0,0")).split(","))) se = tuple(map(int, str(getattr(region, "south_east_cor", "0,0")).split(","))) except Exception: return 0 if getattr(region, "shape", None) is None: return 0 shape_name = getattr(region.shape, "name", "") if shape_name not in ("RECTANGLE", "SQUARE"): return 0 width = se[0] - nw[0] + 1 height = se[1] - nw[1] + 1 if width == height and width in (2, 3): return width return 0 def draw_small_regions(pygame_mod, screen, world, region_images: dict, tile_images: dict, ts: int, m: int, top_offset: int = 0, tile_originals: Optional[dict] = None): """ 使用整图绘制 2x2 / 3x3 的小区域： - 优先按名称从 region_images 中取 n×n 的整图（n 为 2 或 3） - 若没有整图，则将现有 tile 图裁切/合成为一张，避免重复边框 """ for region in world.map.regions.values(): n = _is_small_square_region(region) if n == 0: continue # 仅对 2x2 生效；3x3 不覆盖（保持每格一张图） if n != 2: continue try: nw = tuple(map(int, str(getattr(region, "north_west_cor", "0,0")).split(","))) except Exception: continue x_px = m + nw[0] * ts y_px = m + top_offset + nw[1] * ts name_key = str(getattr(region, "name", "")) variants = region_images.get(name_key) if variants and variants.get(n): screen.blit(variants[n], (x_px, y_px)) continue # 回退：从原始 tile 贴图一次性缩放到 n×n，避免“先缩1×1再放大”的二次缩放 try: tile = world.map.get_tile(nw[0], nw[1]) base_image = None if tile_originals is not None: base_image = tile_originals.get(tile.type) if base_image is None: base_image = tile_images.get(tile.type) except Exception: base_image = None if base_image is not None: scaled = pygame_mod.transform.scale(base_image, (ts * n, ts * n)) screen.blit(scaled, (x_px, y_px)) else: # 最后兜底：淡色块 tmp = pygame_mod.Surface((ts * n, ts * n), pygame_mod.SRCALPHA) tmp.fill((255, 255, 255, 24)) screen.blit(tmp, (x_px, y_px)) def calculate_font_size_by_area(tile_size: int, area: int) -> int: base = int(tile_size * 1.1) growth = int(max(0, min(24, (area ** 0.5)))) size = base + growth - 7 # 再降低2个字号 return max(16, min(40, size)) def draw_region_labels(pygame_mod, screen, colors, world, get_region_font, tile_size: int, margin: int, top_offset: int = 0, outline_px: int = 2): ts = tile_size m = margin mouse_x, mouse_y = pygame_mod.mouse.get_pos() hovered_region = None # 以区域面积降序放置，优先保证大区域标签可读性 regions = sorted(list(world.map.regions.values()), key=lambda r: getattr(r, "area", 0), reverse=True) placed_rects = [] # 已放置标签的矩形列表，用于碰撞检测 # 可放置范围（地图区域） map_px_w = world.map.width * ts map_px_h = world.map.height * ts min_x_allowed = m max_x_allowed = m + map_px_w min_y_allowed = m + top_offset max_y_allowed = m + top_offset + map_px_h def _clamp_rect(x0: int, y0: int, w: int, h: int) -> Tuple[int, int]: # 将标签限制在地图区域内 x = max(min_x_allowed, min(x0, max_x_allowed - w)) y = max(min_y_allowed, min(y0, max_y_allowed - h)) return x, y for region in regions: name = getattr(region, "name", None) if not name: continue # 小区域（面积<=9，例如2x2/3x3）标签放在底部；大区域放在中心 use_bottom = getattr(region, "area", 0) <= 9 if use_bottom and getattr(region, "cors", None): bottom_y = max(y for _, y in region.cors) xs_on_bottom = [x for x, y in region.cors if y == bottom_y] if xs_on_bottom: left_x = min(xs_on_bottom) right_x = max(xs_on_bottom) anchor_cx_tile = (left_x + right_x) / 2.0 else: anchor_cx_tile = float(region.center_loc[0]) screen_cx = int(m + anchor_cx_tile * ts + ts // 2) screen_cy = int(m + top_offset + (bottom_y + 1) * ts + 2) else: # 居中放置 screen_cx = int(m + float(region.center_loc[0]) * ts + ts // 2) screen_cy = int(m + top_offset + float(region.center_loc[1]) * ts) font_size = calculate_font_size_by_area(tile_size, region.area) region_font = get_region_font(font_size) text_surface = region_font.render(str(name), True, colors["text"]) border_surface = region_font.render(str(name), True, colors.get("text_border", (24, 24, 24))) text_w = text_surface.get_width() text_h = text_surface.get_height() # 候选偏移：优先“区域下方”，若越界或冲突，再尝试左右位移、其上方 pad = 6 dxw = max(8, int(0.6 * text_w)) + pad dyh = text_h + pad candidates = [ (0, 0), # 正下方（期望位置） (-dxw, 0), (dxw, 0), # 下方左右 (0, -dyh), # 底边上方一行 (-dxw, -dyh), (dxw, -dyh), (0, -2 * dyh), # 再上方，尽量避免覆盖区域 ] chosen_rect = None for (dx, dy) in candidates: # 以锚点为基准，文本顶部左上角坐标 x_try = int(screen_cx + dx - text_w / 2) y_try = int(screen_cy + dy) x_try, y_try = _clamp_rect(x_try, y_try, text_w, text_h) rect_try = pygame_mod.Rect(x_try, y_try, text_w, text_h) if not any(rect_try.colliderect(r) for r in placed_rects): chosen_rect = rect_try break if chosen_rect is None: # 如果所有候选均冲突，就退回锚点正下方 x0 = int(screen_cx - text_w / 2) y0 = int(screen_cy) x0, y0 = _clamp_rect(x0, y0, text_w, text_h) chosen_rect = pygame_mod.Rect(x0, y0, text_w, text_h) # 悬停检测使用最终位置 if chosen_rect.collidepoint(mouse_x, mouse_y): hovered_region = region # 多方向描边 if outline_px > 0: for dx in (-outline_px, 0, outline_px): for dy in (-outline_px, 0, outline_px): if dx == 0 and dy == 0: continue screen.blit(border_surface, (chosen_rect.x + dx, chosen_rect.y + dy)) screen.blit(text_surface, (chosen_rect.x, chosen_rect.y)) placed_rects.append(chosen_rect) return hovered_region def avatar_center_pixel(avatar: Avatar, tile_size: int, margin: int, top_offset: int = 0) -> Tuple[int, int]: px = margin + avatar.pos_x * tile_size + tile_size // 2 py = margin + top_offset + avatar.pos_y * tile_size + tile_size // 2 return px, py def draw_avatars_and_pick_hover( pygame_mod, screen, colors, simulator, avatar_images, tile_size: int, margin: int, get_display_center: Optional[Callable[[Avatar, int, int], Tuple[float, float]]] = None, top_offset: int = 0, name_font: Optional[object] = None, highlight_avatar_id: Optional[str] = None, ) -> Tuple[Optional[Avatar], List[Avatar]]: mouse_x, mouse_y = pygame_mod.mouse.get_pos() candidates_with_dist: List[Tuple[float, Avatar]] = [] for avatar_id, avatar in simulator.world.avatar_manager.avatars.items(): if get_display_center is not None: cx_f, cy_f = get_display_center(avatar, tile_size, margin) cx, cy = int(cx_f), int(cy_f) else: cx, cy = avatar_center_pixel(avatar, tile_size, margin) cy += top_offset avatar_image = avatar_images.get(avatar_id) if avatar_image: image_rect = avatar_image.get_rect() image_x = cx - image_rect.width // 2 image_y = cy - image_rect.height // 2 screen.blit(avatar_image, (image_x, image_y)) # 名字（置于头像下方居中） if name_font is not None: _draw_avatar_name_label( pygame_mod, screen, colors, name_font, str(getattr(avatar, "name", "")), is_highlight=bool(highlight_avatar_id and avatar.id == highlight_avatar_id), anchor_x=image_x + image_rect.width // 2, anchor_y=image_y + image_rect.height + 2, ) if image_rect.collidepoint(mouse_x - image_x, mouse_y - image_y): dist = math.hypot(mouse_x - cx, mouse_y - cy) candidates_with_dist.append((dist, avatar)) else: radius = max(8, tile_size // 3) pygame_mod.draw.circle(screen, colors["avatar"], (cx, cy), radius) # 名字（置于圆形下方居中） if name_font is not None: _draw_avatar_name_label( pygame_mod, screen, colors, name_font, str(getattr(avatar, "name", "")), is_highlight=bool(highlight_avatar_id and avatar.id == highlight_avatar_id), anchor_x=cx, anchor_y=int(cy + radius + 2), ) dist = math.hypot(mouse_x - cx, mouse_y - cy) if dist <= radius: candidates_with_dist.append((dist, avatar)) candidates_with_dist.sort(key=lambda t: t[0]) hovered = candidates_with_dist[0][1] if candidates_with_dist else None candidate_avatars: List[Avatar] = [a for _, a in candidates_with_dist] return hovered, candidate_avatars def draw_tooltip(pygame_mod, screen, colors, lines: List[str], mouse_x: int, mouse_y: int, font, min_width: Optional[int] = None, top_limit: int = 0): padding = 6 spacing = 2 surf_lines = [font.render(t, True, colors["text"]) for t in lines] width = max(s.get_width() for s in surf_lines) + padding * 2 if min_width is not None: width = max(width, min_width) height = sum(s.get_height() for s in surf_lines) + padding * 2 + spacing * (len(surf_lines) - 1) x = mouse_x + 12 y = mouse_y + 12 screen_w, screen_h = screen.get_size() if x + width > screen_w: x = mouse_x - width - 12 if y + height > screen_h: y = mouse_y - height - 12 # 进一步夹紧，避免位于窗口上边或左边之外 x = max(0, min(x, screen_w - width)) y = max(top_limit, min(y, screen_h - height)) bg_rect = pygame_mod.Rect(x, y, width, height) pygame_mod.draw.rect(screen, colors["tooltip_bg"], bg_rect, border_radius=6) pygame_mod.draw.rect(screen, colors["tooltip_bd"], bg_rect, 1, border_radius=6) cursor_y = y + padding for s in surf_lines: screen.blit(s, (x + padding, cursor_y)) cursor_y += s.get_height() + spacing def draw_tooltip_for_avatar(pygame_mod, screen, colors, font, avatar: Avatar): # 改为从 Avatar.get_hover_info 获取信息行，避免前端重复拼接 lines = avatar.get_hover_info() draw_tooltip(pygame_mod, screen, colors, lines, *pygame_mod.mouse.get_pos(), font, min_width=TOOLTIP_MIN_WIDTH, top_limit=STATUS_BAR_HEIGHT) def draw_tooltip_for_region(pygame_mod, screen, colors, font, region, mouse_x: int, mouse_y: int): if region is None: return # 改为调用 region.get_hover_info()，并统一用 wrap_lines_for_tooltip 进行换行 lines = region.get_hover_info() wrapped_lines = wrap_lines_for_tooltip(lines, 28) draw_tooltip(pygame_mod, screen, colors, wrapped_lines, mouse_x, mouse_y, font, min_width=TOOLTIP_MIN_WIDTH, top_limit=STATUS_BAR_HEIGHT) def draw_operation_guide(pygame_mod, screen, colors, font, margin: int, auto_step: bool): auto_status = "开" if auto_step else "关" guide_text = f"A:自动步进({auto_status}) SPACE:单步 ESC:退出" guide_surf = font.render(guide_text, True, colors["status_text"]) x_pos = margin + 8 screen.blit(guide_surf, (x_pos, 8)) return guide_surf.get_width() def draw_year_month_info(pygame_mod, screen, colors, font, margin: int, guide_width: int, world): year = int(world.month_stamp.get_year()) month_num = world.month_stamp.get_month().value ym_text = f"{year}年{month_num:02d}月" ym_surf = font.render(ym_text, True, colors["status_text"]) x_pos = margin + guide_width + 8 * 3 screen.blit(ym_surf, (x_pos, 8)) def draw_status_bar(pygame_mod, screen, colors, font, margin: int, world, auto_step: bool): status_y = 8 status_height = STATUS_BAR_HEIGHT status_rect = pygame_mod.Rect(0, 0, screen.get_width(), status_height) pygame_mod.draw.rect(screen, colors["status_bg"], status_rect) pygame_mod.draw.line(screen, colors["status_border"], (0, status_height), (screen.get_width(), status_height), 2) guide_w = draw_operation_guide(pygame_mod, screen, colors, font, margin, auto_step) draw_year_month_info(pygame_mod, screen, colors, font, margin, guide_w, world) __all__ = [ "draw_map", "draw_region_labels", "draw_avatars_and_pick_hover", "draw_tooltip_for_avatar", "draw_tooltip_for_region", "draw_status_bar", "STATUS_BAR_HEIGHT", "map_pixel_size", ] def draw_hover_badge(pygame_mod, screen, colors, font, center_x: int, center_y: int, index: int, total: int, top_offset: int = 0): """ 在给定中心附近绘制一个小徽标，显示 index/total（索引从1开始）。 徽标默认放在头像上方偏右位置。 """ label = f"{index}/{total}" surf = font.render(label, True, colors["text"]) pad_x = 6 pad_y = 2 w = surf.get_width() + pad_x * 2 h = surf.get_height() + pad_y * 2 # 徽标位置：头像中心的右上角 x = int(center_x + 10) y = int(center_y + top_offset - 24 - h) rect = pygame_mod.Rect(x, y, w, h) # 半透明背景与描边 bg = pygame_mod.Surface((w, h), pygame_mod.SRCALPHA) bg.fill((20, 20, 20, 180)) screen.blit(bg, (rect.x, rect.y)) pygame_mod.draw.rect(screen, colors.get("tooltip_bd", (90, 90, 90)), rect, 1, border_radius=6) # 文本 screen.blit(surf, (rect.x + pad_x, rect.y + pad_y)) def _draw_avatar_name_label(pygame_mod, screen, colors, font, name_text: str, *, is_highlight: bool, anchor_x: int, anchor_y: int) -> None: if not name_text: return text_color = (236, 236, 236) if is_highlight else colors["text"] text_surf = font.render(name_text, True, text_color) tx = int(anchor_x - text_surf.get_width() / 2) ty = int(anchor_y) if is_highlight: pad_x = 6 pad_y = 2 w = text_surf.get_width() + pad_x * 2 h = text_surf.get_height() + pad_y * 2 bg = pygame_mod.Surface((w, h), pygame_mod.SRCALPHA) bg.fill((0, 0, 0, 210)) screen.blit(bg, (tx - pad_x, ty - pad_y)) rect = pygame_mod.Rect(tx - pad_x, ty - pad_y, w, h) pygame_mod.draw.rect(screen, colors.get("tooltip_bd", (90, 90, 90)), rect, 1, border_radius=6) # 高亮时直接白字绘制（背景已提供对比） screen.blit(text_surf, (tx, ty)) return # 非高亮：加1px 阴影提升可读性（不加底板） shadow = font.render(name_text, True, colors.get("text_border", (24, 24, 24))) screen.blit(shadow, (tx + 1, ty + 1)) screen.blit(text_surf, (tx, ty)) def map_pixel_size(world_or_map, tile_size: int) -> Tuple[int, int]: """ 计算地图像素宽高（不含 margin 与顶部偏移）。 支持传入 world（含 .map）或 map 对象（含 .width/.height）。 """ map_obj = getattr(world_or_map, "map", world_or_map) return map_obj.width * tile_size, map_obj.height * tile_size