156 lines
7.4 KiB
Python
156 lines
7.4 KiB
Python
import numpy as np
|
|
import pygame
|
|
import math
|
|
from .utils import *
|
|
from .agent import TileNode, Agent, AgentType
|
|
|
|
terrain_colors = [
|
|
(245, 245, 220), # 米色 Plain
|
|
(210, 180, 140), # 褐色 Road
|
|
(224, 255, 255), # 青色 Water
|
|
(173, 216, 230) # 蓝色 Subwater
|
|
]
|
|
|
|
team_colors = [
|
|
(255, 0, 0), # 红色
|
|
(0, 0, 255), # 蓝色
|
|
(0, 255, 0), # 绿色
|
|
(255, 255, 0), # 黄色
|
|
(255, 165, 0), # 橙色
|
|
(128, 0, 128), # 紫色
|
|
(0, 255, 255), # 青色
|
|
(255, 192, 203) # 粉色
|
|
]
|
|
|
|
icons_path = "./tianqiong/envs/icons"
|
|
|
|
unit_icons: Dict[str, Dict[int, pygame.Surface]] = {
|
|
# AgentType.AntiAir: {0: pygame.image.load(f"{icons_path}/AntiAir_b.png"), 1: pygame.image.load(f"{icons_path}/AntiAir_r.png")},
|
|
# AgentType.Airport: {0: pygame.image.load(f"{icons_path}/Airport_b.png"), 1: pygame.image.load(f"{icons_path}/Airport_r.png")},
|
|
# AgentType.Artillery: {0: pygame.image.load(f"{icons_path}/Artillery_b.png"), 1: pygame.image.load(f"{icons_path}/Artillery_r.png")},
|
|
# AgentType.Bomber: {0: pygame.image.load(f"{icons_path}/Bomber_b.png"), 1: pygame.image.load(f"{icons_path}/Bomber_r.png")},
|
|
# AgentType.Carrier: {0: pygame.image.load(f"{icons_path}/Carrier_b.png"), 1: pygame.image.load(f"{icons_path}/Carrier_r.png")},
|
|
# AgentType.Fighter: {0: pygame.image.load(f"{icons_path}/Fighter_b.png"), 1: pygame.image.load(f"{icons_path}/Fighter_r.png")},
|
|
# AgentType.Helicopter: {0: pygame.image.load(f"{icons_path}/Helicopter_b.png"), 1: pygame.image.load(f"{icons_path}/Helicopter_r.png")},
|
|
# AgentType.Infantry: {0: pygame.image.load(f"{icons_path}/Infantry_b.png"), 1: pygame.image.load(f"{icons_path}/Infantry_r.png")},
|
|
# AgentType.Tank: {0: pygame.image.load(f"{icons_path}/Tank_b.png"), 1: pygame.image.load(f"{icons_path}/Tank_r.png")},
|
|
# AgentType.TransportHelicopter: {0: pygame.image.load(f"{icons_path}/TransportHelicopter_b.png"), 1: pygame.image.load(f"{icons_path}/TransportHelicopter_r.png")},
|
|
# AgentType.TransportShip: {0: pygame.image.load(f"{icons_path}/TransportShip_b.png"), 1: pygame.image.load(f"{icons_path}/TransportShip_r.png")},
|
|
# AgentType.CombatShip: {0: pygame.image.load(f"{icons_path}/CombatShip_b.png"), 1: pygame.image.load(f"{icons_path}/CombatShip_r.png")},
|
|
}
|
|
|
|
class Renderer:
|
|
def __init__(self, nodes: List[TileNode], hex_size=20):
|
|
self.nodes = nodes
|
|
self.hex_size = hex_size
|
|
self.window_size, (self.offset_x, self.offset_y) = get_window_size_and_offsets(nodes, hex_size)
|
|
|
|
def render(self, player_agents: List[Agent], spotted_enemy_agents: List[Agent], attack_agent: Agent=None, defend_agent: Agent=None) -> np.ndarray:
|
|
screen = pygame.Surface(self.window_size)
|
|
offset_x = self.offset_x
|
|
offset_y = self.offset_y
|
|
hex_size = self.hex_size
|
|
nodes = self.nodes
|
|
draw_hex_grid(screen, offset_x, offset_y, hex_size, nodes)
|
|
for agent in player_agents:
|
|
pos = agent.pos
|
|
alpha = agent.endurance / agent.max_endurance * 255
|
|
draw_unit(screen, offset_x, offset_y, hex_size, axial_to_pixel(pos, hex_size), alpha=alpha, agent_type=agent.agent_type, team_id=agent.team_id)
|
|
for agent in spotted_enemy_agents:
|
|
pos = agent.pos
|
|
alpha = agent.endurance / agent.max_endurance * 255
|
|
draw_unit(screen, offset_x, offset_y, hex_size, axial_to_pixel(pos, hex_size), alpha=alpha, agent_type=agent.agent_type, team_id=agent.team_id)
|
|
|
|
if attack_agent and defend_agent:
|
|
start_pos = axial_to_pixel(attack_agent.pos, hex_size)
|
|
end_pos = axial_to_pixel(defend_agent.pos, hex_size)
|
|
draw_hexagon(screen, hex_size, (start_pos[0] + offset_x, start_pos[1] + offset_y), team_colors[attack_agent.team_id], True, 5)
|
|
draw_hexagon(screen, hex_size, (end_pos[0] + offset_x, end_pos[1] + offset_y), team_colors[attack_agent.team_id], True, 5)
|
|
|
|
rgb_array = pygame.surfarray.array3d(screen)
|
|
|
|
return np.transpose(rgb_array, (1, 0, 2))
|
|
|
|
# 轴坐标系到像素坐标的转换函数（尖角朝上）
|
|
def axial_to_pixel(pos: Tuple[int, int], hex_size):
|
|
q, r = pos
|
|
x = hex_size * math.sqrt(3) * q
|
|
y = hex_size * 3/2 * r
|
|
x += hex_size * math.sqrt(3) / 2 * r
|
|
return x, y
|
|
|
|
# 绘制六边形的函数
|
|
def draw_hexagon(surface, hex_size, center, color=(255, 255, 255), only_frame=False, line_width=1):
|
|
line_color = (169, 169, 169) # 灰色
|
|
angle_offset = math.pi / 6 # 顶点指向上方
|
|
points = [
|
|
(
|
|
center[0] + hex_size * math.cos(angle_offset + math.pi / 3 * i),
|
|
center[1] + hex_size * math.sin(angle_offset + math.pi / 3 * i)
|
|
)
|
|
for i in range(6)
|
|
]
|
|
if not only_frame:
|
|
pygame.draw.polygon(surface, color, points)
|
|
pygame.draw.polygon(surface, line_color, points, width=1) # 绘制六边形边框
|
|
else:
|
|
pygame.draw.polygon(surface, color, points, width=line_width) # 绘制六边形边框
|
|
|
|
def get_window_size_and_offsets(nodes: List[TileNode], hex_size: int) -> Tuple[Tuple[int, int], Tuple[int, int]]:
|
|
min_x = min_y = float('inf')
|
|
max_x = max_y = float('-inf')
|
|
offset_x = offset_y = float('-inf')
|
|
|
|
for node in nodes:
|
|
q, r = node.pos
|
|
x, y = axial_to_pixel((q, r), hex_size)
|
|
min_x = min(min_x, x)
|
|
max_x = max(max_x, x)
|
|
min_y = min(min_y, y)
|
|
max_y = max(max_y, y)
|
|
offset_x = max(offset_x, -x)
|
|
offset_y = max(offset_y, -y)
|
|
|
|
# 加一些边距
|
|
margin = hex_size * 2
|
|
offset_x += margin
|
|
offset_y += margin
|
|
width = int(max_x - min_x + 2 * margin)
|
|
height = int(max_y - min_y + 2 * margin)
|
|
|
|
width = (width - 15) // 16 * 16 + 16
|
|
height = (height - 15) // 16 * 16 + 16
|
|
|
|
return (width, height), (int(offset_x), int(offset_y))
|
|
|
|
# 绘制单位的函数，带透明度
|
|
def draw_unit(surface, offset_x, offset_y, hex_size, center, alpha=128, agent_type=None, team_id=None):
|
|
# print(agent_type, team_id)
|
|
if agent_type in unit_icons and team_id in unit_icons[agent_type]:
|
|
# 如果有图标，用图标表示
|
|
icon = unit_icons[agent_type][team_id]
|
|
icon_size = hex_size * 1.2
|
|
icon = pygame.transform.scale(icon, (icon_size, icon_size)) # 缩放图标至合适尺寸
|
|
icon.set_alpha(alpha) # 设置透明度
|
|
surface.blit(icon, (center[0] + offset_x - icon_size // 2, center[1] + offset_y - icon_size // 2))
|
|
else:
|
|
# 如果没有图标，用圆圈表示单位
|
|
unit_surface = pygame.Surface((hex_size, hex_size), pygame.SRCALPHA)
|
|
unit_surface.set_alpha(alpha) # 设置透明度
|
|
color = team_colors[team_id]
|
|
pygame.draw.circle(unit_surface, color, (hex_size // 2, hex_size // 2), hex_size // 3)
|
|
surface.blit(unit_surface, (center[0] + offset_x - hex_size // 2, center[1] + offset_y - hex_size // 2))
|
|
|
|
# 主绘制功能
|
|
def draw_hex_grid(surface, offset_x, offset_y, hex_size, nodes: List[TileNode]):
|
|
background_color = (255, 255, 255) # 白色
|
|
surface.fill(background_color)
|
|
|
|
for node in nodes:
|
|
q, r = node.pos
|
|
color = terrain_colors[node.terrain_type.value]
|
|
pixel_x, pixel_y = axial_to_pixel((q, r), hex_size)
|
|
center = (pixel_x + offset_x, pixel_y + offset_y)
|
|
draw_hexagon(surface, hex_size, center, color)
|
|
|