update readme and front
This commit is contained in:
bridge
209
src/front/rendering.py
Normal file
209
src/front/rendering.py
Normal file
@@ -0,0 +1,209 @@
|
||||
import math
|
||||
from typing import List, Optional, Tuple
|
||||
from src.classes.avatar import Avatar, Gender
|
||||
from src.classes.tile import TileType
|
||||
from src.utils.text_wrap import wrap_text
|
||||
|
||||
|
||||
def draw_grid(pygame_mod, screen, colors, map_obj, ts: int, m: int):
|
||||
grid_color = colors["grid"]
|
||||
for gx in range(map_obj.width + 1):
|
||||
start_pos = (m + gx * ts, m)
|
||||
end_pos = (m + gx * ts, m + 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 + gy * ts)
|
||||
end_pos = (m + map_obj.width * ts, m + 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):
|
||||
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 + y * ts)
|
||||
screen.blit(tile_image, pos)
|
||||
else:
|
||||
color = (80, 80, 80)
|
||||
rect = pygame_mod.Rect(m + x * ts, m + y * ts, ts, ts)
|
||||
pygame_mod.draw.rect(screen, color, rect)
|
||||
draw_grid(pygame_mod, screen, colors, map_obj, ts, m)
|
||||
|
||||
|
||||
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))))
|
||||
return max(16, min(40, base + growth))
|
||||
|
||||
|
||||
def draw_region_labels(pygame_mod, screen, colors, world, get_region_font, tile_size: int, margin: int):
|
||||
ts = tile_size
|
||||
m = margin
|
||||
mouse_x, mouse_y = pygame_mod.mouse.get_pos()
|
||||
from src.classes.region import regions_by_id
|
||||
hovered_region = None
|
||||
for region in regions_by_id.values():
|
||||
name = getattr(region, "name", None)
|
||||
if not name:
|
||||
continue
|
||||
center_x, center_y = region.center_loc
|
||||
screen_x = m + center_x * ts + ts // 2
|
||||
screen_y = m + center_y * ts + ts // 2
|
||||
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"])
|
||||
shadow_surface = region_font.render(str(name), True, (0, 0, 0))
|
||||
text_w = text_surface.get_width()
|
||||
text_h = text_surface.get_height()
|
||||
x = int(screen_x - text_w / 2)
|
||||
y = int(screen_y - text_h / 2)
|
||||
if (x <= mouse_x <= x + text_w and y <= mouse_y <= y + text_h):
|
||||
hovered_region = region
|
||||
screen.blit(shadow_surface, (x + 1, y + 1))
|
||||
screen.blit(text_surface, (x, y))
|
||||
return hovered_region
|
||||
|
||||
|
||||
def avatar_center_pixel(avatar: Avatar, tile_size: int, margin: int) -> Tuple[int, int]:
|
||||
px = margin + avatar.pos_x * tile_size + tile_size // 2
|
||||
py = margin + 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) -> Optional[Avatar]:
|
||||
mouse_x, mouse_y = pygame_mod.mouse.get_pos()
|
||||
hovered = None
|
||||
min_dist = float("inf")
|
||||
for avatar_id, avatar in simulator.avatars.items():
|
||||
cx, cy = avatar_center_pixel(avatar, tile_size, margin)
|
||||
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 image_rect.collidepoint(mouse_x - image_x, mouse_y - image_y):
|
||||
hovered = avatar
|
||||
min_dist = 0
|
||||
else:
|
||||
radius = max(8, tile_size // 3)
|
||||
pygame_mod.draw.circle(screen, colors["avatar"], (cx, cy), radius)
|
||||
dist = math.hypot(mouse_x - cx, mouse_y - cy)
|
||||
if dist <= radius and dist < min_dist:
|
||||
hovered = avatar
|
||||
min_dist = dist
|
||||
return hovered
|
||||
|
||||
|
||||
def draw_tooltip(pygame_mod, screen, colors, lines: List[str], mouse_x: int, mouse_y: int, font):
|
||||
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
|
||||
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
|
||||
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):
|
||||
lines = [
|
||||
f"{avatar.name}",
|
||||
f"性别: {avatar.gender}",
|
||||
f"年龄: {avatar.age}",
|
||||
f"境界: {str(avatar.cultivation_progress)}",
|
||||
f"灵根: {avatar.root.value}",
|
||||
f"个性: {avatar.persona.name}",
|
||||
f"位置: ({avatar.pos_x}, {avatar.pos_y})",
|
||||
]
|
||||
lines.append(f"灵石: {str(avatar.magic_stone)}")
|
||||
if avatar.items:
|
||||
lines.append("物品:")
|
||||
for item, quantity in avatar.items.items():
|
||||
lines.append(f" {item.name} x{quantity}")
|
||||
else:
|
||||
lines.append("")
|
||||
lines.append("物品: 无")
|
||||
if avatar.thinking:
|
||||
lines.append("")
|
||||
lines.append("思考:")
|
||||
thinking_lines = wrap_text(avatar.thinking, 20)
|
||||
lines.extend(thinking_lines)
|
||||
draw_tooltip(pygame_mod, screen, colors, lines, *pygame_mod.mouse.get_pos(), font)
|
||||
|
||||
|
||||
def draw_tooltip_for_region(pygame_mod, screen, colors, font, region, mouse_x: int, mouse_y: int):
|
||||
if region is None:
|
||||
return
|
||||
lines = [
|
||||
f"区域: {region.name}",
|
||||
f"描述: {region.desc}",
|
||||
]
|
||||
from src.classes.region import CultivateRegion, NormalRegion
|
||||
if isinstance(region, CultivateRegion):
|
||||
stars = "★" * region.essence_density + "☆" * (10 - region.essence_density)
|
||||
lines.append(f"主要灵气: {region.essence_type} {stars}")
|
||||
elif isinstance(region, NormalRegion):
|
||||
species_info = region.get_species_info()
|
||||
if species_info and species_info != "暂无特色物种":
|
||||
lines.append("物种分布:")
|
||||
for species in species_info.split("; "):
|
||||
lines.append(f" {species}")
|
||||
else:
|
||||
lines.append("物种分布: 暂无特色物种")
|
||||
draw_tooltip(pygame_mod, screen, colors, lines, mouse_x, mouse_y, font)
|
||||
|
||||
|
||||
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 = 32
|
||||
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",
|
||||
]
|
||||
|
||||
|
||||
Reference in New Issue
Block a user