shenjack
f9eeafe322
readme update 看起来更像 Dear ImGui 一些(looks more like Dear ImGui and some intersting feature to the button remove debug 确认一下action 404 修改 writing theme looks good better? a ? alpha=255 not 256 looks better try new pyglet first 看起来好一些 sync pyglet 水一手 这波必须得水一手了，要不然commit太少了（确信 丢点正经东西上去 顺手继承一下Options 补充docs 坏了，忘记水commit了（ 至少我能早睡了（ 这里还能水一点来着（ 试试再说 reee 保证能跑（ 同步lib not dr 的修改 忘记带上 None了 还是加上一个额外的判断参数吧 刷点commit也不错 先更新一下依赖版本 水commit啦 理论可行，实践开始！ 构建参数喜加一 reeeee 更新一下 pyproject 的依赖 fix typing looks better 水一个（ 测试？ sync pyglet to master A | Try use rust-cache looks good? what? C | sync pyglet A | 添加了一个 Button Draw Theme A | Magic Number （确信） A | 尽量不继承Options sync pyglet A | Add theme A | Add more Theme information Enhance | Theme sync pyglet Add | add unifont Enhance | use os.walk in font loading Enhance | Use i18n in font loading Enhance | doc sync pyglet Add | add 3.12 build option to build_rs Fix | Button position have a z sync pyglet A | Logger.py 启动！ sync pyglet Changed | Bump pyo3 to 0.20.0 add logger.py update logger！ Add | more logger! Add | lib-not-dr some lib-not-dr
494 lines
21 KiB
Python
494 lines
21 KiB
Python
# -------------------------------
|
||
# Difficult Rocket
|
||
# Copyright © 2020-2023 by shenjackyuanjie 3695888@qq.com
|
||
# All rights reserved
|
||
# -------------------------------
|
||
|
||
import time
|
||
import random
|
||
import logging
|
||
import traceback
|
||
|
||
from pathlib import Path
|
||
from typing import List, Dict, Optional, Generator, Tuple
|
||
|
||
from pyglet.gl import gl
|
||
from pyglet.math import Mat4
|
||
from pyglet.text import Label
|
||
from pyglet.sprite import Sprite
|
||
from pyglet.graphics import Batch, Group
|
||
from pyglet.shapes import Line, Rectangle
|
||
|
||
from . import DR_mod_runtime
|
||
from .types import SR1Textures, SR1Rotation
|
||
|
||
# Difficult Rocket
|
||
from Difficult_Rocket import DR_status
|
||
from Difficult_Rocket.utils.translate import Tr
|
||
from Difficult_Rocket.client import ClientWindow
|
||
from Difficult_Rocket.api.types import Fonts, Options
|
||
from Difficult_Rocket.command.line import CommandText
|
||
from Difficult_Rocket.client.screen import BaseScreen
|
||
from Difficult_Rocket.api.camera import CenterGroupCamera
|
||
from Difficult_Rocket.api.gui.widget import PressTextButton
|
||
|
||
if DR_mod_runtime.use_DR_rust:
|
||
from .Difficult_Rocket_rs import (SR1PartList_rs,
|
||
SR1Ship_rs,
|
||
SR1PartData_rs,
|
||
SR1PartType_rs)
|
||
|
||
logger = logging.getLogger('client.dr_game_sr1_ship')
|
||
logger.level = logging.DEBUG
|
||
sr_tr = Tr(lang_path=Path(__file__).parent / 'lang')
|
||
|
||
|
||
class SR1ShipRenderStatus(Options): # NOQA
|
||
name = "SR1ShipRenderStatus"
|
||
# main status
|
||
draw_done: bool = False
|
||
draw_call: bool = False
|
||
update_call: bool = False
|
||
focus: bool = True
|
||
moving: bool = False
|
||
|
||
# button status
|
||
show_moving: bool = False
|
||
show_focus: bool = False
|
||
show_scale: bool = False
|
||
|
||
# debug status
|
||
draw_d_pos: bool = False
|
||
draw_mouse_pos: bool = False
|
||
draw_mouse_d_pos: bool = False
|
||
|
||
|
||
class SR1ShipRender(BaseScreen):
|
||
"""用于渲染 sr1 船的类"""
|
||
|
||
name = 'DR_game_sr1_ship_render'
|
||
|
||
def __init__(self,
|
||
main_window: ClientWindow):
|
||
super().__init__(main_window)
|
||
self.logger = logger
|
||
logger.info(sr_tr().mod.info.setup.start())
|
||
load_start_time = time.time_ns()
|
||
# status
|
||
self.status = SR1ShipRenderStatus()
|
||
|
||
self.dx = 0
|
||
self.dy = 0
|
||
self.width = main_window.width
|
||
self.height = main_window.height
|
||
|
||
self.main_batch = Batch()
|
||
self.group_camera = CenterGroupCamera(window=main_window,
|
||
order=10,
|
||
parent=main_window.main_group,
|
||
min_zoom=(1 / 2) ** 10,
|
||
max_zoom=10)
|
||
self.part_group = Group(0, parent=self.group_camera)
|
||
|
||
self.debug_label = Label(x=20, y=main_window.height - 100, font_size=DR_status.std_font_size,
|
||
text='SR1 render!', font_name=Fonts.微软等宽无线,
|
||
width=main_window.width - 20, height=20,
|
||
anchor_x='left', anchor_y='top',
|
||
batch=self.main_batch, group=Group(5, parent=self.part_group))
|
||
self.render_d_line = Line(0, 0, 0, 0, width=5, color=(200, 200, 10, 255),
|
||
batch=self.main_batch, group=Group(5, parent=self.part_group))
|
||
self.render_d_line.visible = self.status.draw_d_pos
|
||
self.render_d_label = Label('debug label NODATA', font_name=Fonts.微软等宽无线,
|
||
x=main_window.width / 2, y=main_window.height / 2)
|
||
self.render_d_label.visible = self.status.draw_d_pos
|
||
|
||
self.test_button = PressTextButton(x=100, y=100, width=150, height=30, text='test button',
|
||
batch=self.main_batch, group=Group(5, parent=main_window.main_group))
|
||
# self.test_button.push_handlers(main_window)
|
||
main_window.push_handlers(self.test_button)
|
||
|
||
# Optional data
|
||
self.textures: SR1Textures = SR1Textures()
|
||
self.gen_draw: Optional[Generator] = None
|
||
self.rust_ship: Optional[SR1Ship_rs] = None
|
||
self.ship_name: Optional[str] = None
|
||
|
||
# List/Dict data
|
||
self.parts_sprite: Dict[int, List[Sprite]] = {}
|
||
self.part_box_dict: Dict[int, Rectangle] = {}
|
||
self.part_line_box: Dict[int, List[Line]] = {}
|
||
self.part_line_list: List[Line] = []
|
||
|
||
if DR_mod_runtime.use_DR_rust:
|
||
self.rust_parts = None
|
||
self.part_list_rs = SR1PartList_rs('assets/builtin/PartList.xml', 'builtin_part_list')
|
||
|
||
self.load_xml('assets/builtin/dock1.xml')
|
||
|
||
load_end_time = time.time_ns()
|
||
logger.info(sr_tr().mod.info.setup.use_time().format((load_end_time - load_start_time) / 1000000000))
|
||
|
||
@property
|
||
def size(self) -> Tuple[int, int]:
|
||
""" 渲染器的渲染大小 """
|
||
return self.width, self.height
|
||
|
||
@size.setter
|
||
def size(self, value: Tuple[int, int]):
|
||
if not self.width == value[0] or not self.height == value[1]:
|
||
self.width, self.height = value
|
||
|
||
def load_xml(self, file_path: str) -> bool:
|
||
"""
|
||
加载 xml 文件
|
||
:param file_path:
|
||
:return:
|
||
"""
|
||
try:
|
||
start_time = time.time_ns()
|
||
logger.info(sr_tr().sr1.ship.xml.loading().format(file_path))
|
||
self.ship_name = file_path.split('/')[-1].split('.')[0]
|
||
if DR_mod_runtime.use_DR_rust:
|
||
self.rust_ship = SR1Ship_rs(file_path, self.part_list_rs, 'a_new_ship')
|
||
logger.info(sr_tr().sr1.ship.xml.load_done())
|
||
logger.info(sr_tr().sr1.ship.xml.load_time().format(
|
||
(time.time_ns() - start_time) / 1000000000))
|
||
return True
|
||
except Exception:
|
||
traceback.print_exc()
|
||
self.logger.error(traceback.format_exc())
|
||
return False
|
||
|
||
def gen_sprite(self, each_count: int = 100) -> Generator:
|
||
"""
|
||
生成 sprite
|
||
通过生成器减少一次性渲染的压力
|
||
:param each_count: 每次生成的数量 (默认 100) (过大会导致卡顿)
|
||
:return: 生成器
|
||
"""
|
||
count = 0
|
||
self.status.draw_done = False
|
||
# rust 渲染
|
||
if DR_mod_runtime.use_DR_rust:
|
||
cache = self.rust_ship.as_dict()
|
||
part_group = Group(2, parent=self.part_group)
|
||
line_box_group = Group(6, parent=self.part_group)
|
||
for p_id, parts in cache.items():
|
||
p_id: int
|
||
parts: List[Tuple[SR1PartType_rs, SR1PartData_rs]]
|
||
batch = []
|
||
for p_type, p_data in parts:
|
||
sprite_name = self.part_list_rs.get_part_type(p_data.part_type_id).sprite
|
||
part_sprite = Sprite(img=self.textures.get_texture(sprite_name),
|
||
x=p_data.x * 60, y=p_data.y * 60, z=random.random(),
|
||
batch=self.main_batch, group=part_group)
|
||
part_sprite.rotation = p_data.angle_r
|
||
part_sprite.scale_x = -1 if p_data.flip_x else 1
|
||
part_sprite.scale_y = -1 if p_data.flip_y else 1
|
||
|
||
batch.append(part_sprite)
|
||
part_box = self.rust_ship.get_part_box(p_id)
|
||
if part_box:
|
||
# 线框
|
||
part_line_box = []
|
||
width = 4
|
||
color = (random.randrange(0, 255), random.randrange(0, 255), random.randrange(0, 255),
|
||
random.randrange(100, 200))
|
||
(x, y), (x2, y2) = part_box
|
||
part_line_box.append(Line(x=x * 30, y=y * 30, x2=x * 30, y2=y2 * 30,
|
||
batch=self.main_batch, width=width, color=color, group=line_box_group))
|
||
part_line_box.append(Line(x=x * 30, y=y2 * 30, x2=x2 * 30, y2=y2 * 30,
|
||
batch=self.main_batch, width=width, color=color, group=line_box_group))
|
||
part_line_box.append(Line(x=x2 * 30, y=y2 * 30, x2=x2 * 30, y2=y * 30,
|
||
batch=self.main_batch, width=width, color=color, group=line_box_group))
|
||
part_line_box.append(Line(x=x2 * 30, y=y * 30, x2=x * 30, y2=y * 30,
|
||
batch=self.main_batch, width=width, color=color, group=line_box_group))
|
||
# 直接用循环得了
|
||
self.part_line_box[p_id] = part_line_box
|
||
self.parts_sprite[p_id] = batch
|
||
count += 1
|
||
if count >= each_count:
|
||
count = 0
|
||
yield
|
||
connect_line_group = Group(7, parent=self.part_group)
|
||
for connect in self.rust_ship.connection:
|
||
# 连接线
|
||
parent_part_data = cache[connect[2]][0][1]
|
||
child_part_data = cache[connect[3]][0][1]
|
||
color = (random.randrange(100, 255), random.randrange(0, 255), random.randrange(0, 255), 255)
|
||
self.part_line_list.append(Line(x=parent_part_data.x * 60, y=parent_part_data.y * 60,
|
||
x2=child_part_data.x * 60, y2=child_part_data.y * 60,
|
||
batch=self.main_batch, group=connect_line_group,
|
||
width=1, color=color))
|
||
count += 1
|
||
if count >= each_count * 3:
|
||
count = 0
|
||
yield count
|
||
|
||
# python 渲染
|
||
# for part_id, part in part_datas.items():
|
||
# # 下面就是调用 pyglet 去渲染的部分
|
||
# # render_scale = DR_status.gui_scale # 这个是 DR 的缩放比例 可以调节的
|
||
# # 在不缩放的情况下，XML的1个单位长度对应60个像素
|
||
# # render_x = part.x * 60
|
||
# # render_y = part.y * 60
|
||
# # cache_sprite = Sprite(img=self.textures.get_texture(part.textures),
|
||
# # x=render_x, y=render_y, z=random.random(),
|
||
# # batch=self.main_batch, group=self.part_group)
|
||
# # # 你得帮我换算一下 XML 里的 x y 和这里的屏幕像素的关系
|
||
# # # 旋转啥的不是大问题, 我找你要那个渲染代码就是要 x y 的换算逻辑
|
||
# # cache_sprite.rotation = SR1Rotation.get_rotation(part.angle)
|
||
# # if part.flip_x:
|
||
# # cache_sprite.scale_x = -1
|
||
# # if part.flip_y:
|
||
# # cache_sprite.scale_y = -1
|
||
# # self.parts_sprite[part.id] = cache_sprite
|
||
#
|
||
# if DR_mod_runtime.use_DR_rust:
|
||
# count += 1
|
||
# if count >= each_count:
|
||
# count = 0
|
||
# yield count
|
||
self.status.draw_done = True
|
||
raise GeneratorExit
|
||
|
||
def render_ship(self):
|
||
"""
|
||
渲染船
|
||
"""
|
||
self.status.draw_done = False
|
||
logger.info(sr_tr().sr1.ship.ship.load().format(self.ship_name))
|
||
start_time = time.perf_counter_ns()
|
||
self.parts_sprite: Dict[int, Sprite] = {}
|
||
self.part_line_box = {}
|
||
self.part_line_list = []
|
||
self.group_camera.reset()
|
||
# 调用生成器 减少卡顿
|
||
try:
|
||
self.gen_draw = self.gen_sprite()
|
||
if not self.status.draw_done:
|
||
next(self.gen_draw)
|
||
except (GeneratorExit, StopIteration):
|
||
self.status.draw_done = True
|
||
self.status.draw_call = False
|
||
full_mass = 0
|
||
if DR_mod_runtime.use_DR_rust:
|
||
full_mass = self.rust_ship.mass
|
||
logger.info(sr_tr().sr1.ship.ship.load_time().format(
|
||
(time.perf_counter_ns() - start_time) / 1000000000))
|
||
logger.info(sr_tr().sr1.ship.ship.info().format(
|
||
len(self.rust_ship.as_list()),
|
||
f'{full_mass}kg' if DR_mod_runtime.use_DR_rust else sr_tr().game.require_DR_rs()))
|
||
|
||
def draw_batch(self, window: ClientWindow):
|
||
if self.status.draw_done:
|
||
self.render_d_label.text = f'x: {self.group_camera.view_x} y: {self.group_camera.view_y}'
|
||
self.render_d_label.position = self.group_camera.view_x + (
|
||
self.window_pointer.width / 2), self.group_camera.view_y + (
|
||
self.window_pointer.height / 2) + 10, 0 # 0 for z
|
||
self.render_d_line.x2 = self.group_camera.view_x
|
||
self.render_d_line.y2 = self.group_camera.view_y
|
||
|
||
gl.glEnable(gl.GL_SCISSOR_TEST)
|
||
gl.glScissor(int(self.dx), int(self.dy), int(self.width), int(self.height))
|
||
gl.glViewport(int(self.dx), int(self.dy), self.window_pointer.width, self.window_pointer.height)
|
||
self.main_batch.draw() # use group camera, no need to with
|
||
gl.glViewport(0, 0, self.window_pointer.width, self.window_pointer.height)
|
||
gl.glScissor(0, 0, self.window_pointer.width, self.window_pointer.height)
|
||
gl.glDisable(gl.GL_SCISSOR_TEST)
|
||
|
||
# def on_draw(self, dt: float, window): # TODO: wait for pyglet 2.1
|
||
def on_draw(self, window: ClientWindow):
|
||
if self.status.draw_call:
|
||
self.render_ship()
|
||
|
||
if not self.status.draw_done:
|
||
try:
|
||
next(self.gen_draw)
|
||
except (GeneratorExit, StopIteration):
|
||
self.status.draw_done = True
|
||
self.logger.info(sr_tr().sr1.ship.ship.render.done())
|
||
except TypeError:
|
||
pass
|
||
|
||
self.debug_label.draw()
|
||
|
||
def on_resize(self, width: int, height: int, window: ClientWindow):
|
||
self.debug_label.y = height - 100
|
||
if not self.status.draw_done:
|
||
return
|
||
self.render_d_line.x2 = width // 2
|
||
self.render_d_line.y2 = height // 2
|
||
self.test_button._update_position()
|
||
|
||
def on_mouse_scroll(self, x: int, y: int, scroll_x: int, scroll_y: int, window: ClientWindow):
|
||
if not self.status.draw_done:
|
||
return
|
||
if self.status.focus:
|
||
mouse_dx = x - (self.width / 2) + self.dx
|
||
mouse_dy = y - (self.height / 2) + self.dy
|
||
# 鼠标缩放位置相对于屏幕中心的位置
|
||
mouse_dx_d = mouse_dx - self.group_camera.view_x
|
||
mouse_dy_d = mouse_dy - self.group_camera.view_y
|
||
# 鼠标相对偏移量的偏移量
|
||
if scroll_y == 0:
|
||
zoom_d = 1
|
||
else:
|
||
zoom_d = ((2 ** scroll_y) - 1) * 0.5 + 1
|
||
# 缩放的变换量
|
||
if not (self.group_camera.zoom == 10 and scroll_y > 0):
|
||
if self.group_camera.zoom * zoom_d >= 10:
|
||
zoom_d = 10 / self.group_camera.zoom
|
||
self.group_camera.zoom = 10
|
||
else:
|
||
self.group_camera.zoom *= zoom_d
|
||
mouse_dx_d *= (1 - zoom_d)
|
||
mouse_dy_d *= (1 - zoom_d)
|
||
self.group_camera.view_x += mouse_dx_d
|
||
self.group_camera.view_y += mouse_dy_d
|
||
elif self.status.moving:
|
||
# 如果是在移动整体渲染位置
|
||
size_x, size_y = self.size
|
||
size_x += round(scroll_y) * 10
|
||
size_y += round(scroll_y) * 10
|
||
if size_x < 10:
|
||
size_x = 10
|
||
if size_y < 10:
|
||
size_y = 10
|
||
self.size = size_x, size_y
|
||
|
||
def on_command(self, command: CommandText, window: ClientWindow):
|
||
""" 解析命令 """
|
||
self.logger.info(f'command: {command}')
|
||
if command.find('render'):
|
||
if command.find('reset'):
|
||
self.group_camera.reset()
|
||
else:
|
||
self.status.draw_call = True
|
||
print('应该渲染飞船的')
|
||
|
||
elif command.find('debug'):
|
||
if command.find('delta'):
|
||
self.render_d_line.visible = not self.render_d_line.visible
|
||
self.status.draw_mouse_d_pos = self.render_d_line.visible
|
||
self.logger.info(f'sr1 mouse {self.status.draw_mouse_d_pos}')
|
||
elif command.find('ship'):
|
||
if self.status.draw_done:
|
||
for index, sprite in self.parts_sprite.items():
|
||
sprite.visible = not sprite.visible
|
||
|
||
elif command.find('get_buf'):
|
||
|
||
def screenshot(window):
|
||
"""
|
||
从窗口截图
|
||
:param window:
|
||
:return:
|
||
"""
|
||
from pyglet.gl import GLubyte, GL_RGBA, GL_UNSIGNED_BYTE, glReadPixels
|
||
import pyglet
|
||
format_str = "RGBA"
|
||
buf = (GLubyte * (len(format_str) * window.width * window.height))()
|
||
glReadPixels(0, 0, window.width, window.height, GL_RGBA, GL_UNSIGNED_BYTE, buf)
|
||
return pyglet.image.ImageData(window.width, window.height, format_str, buf)
|
||
|
||
image_data = screenshot(self.window_pointer)
|
||
image_data.save('test.png')
|
||
elif command.find('gen_img'):
|
||
if not self.status.draw_done:
|
||
return
|
||
if not DR_mod_runtime.use_DR_rust:
|
||
# 这个功能依赖于 DR rs (简称,我懒得在Python端实现)
|
||
return
|
||
img_box = self.rust_ship.img_pos
|
||
img_size = (img_box[2] - img_box[0], img_box[3] - img_box[1])
|
||
# 中心点是左上角坐标
|
||
img_center = (abs(img_box[0]), abs(img_box[3]))
|
||
try:
|
||
from PIL import Image
|
||
except ImportError:
|
||
traceback.print_exc()
|
||
print('PIL not found')
|
||
return
|
||
img = Image.new('RGBA', img_size)
|
||
part_data = self.rust_ship.as_dict()
|
||
for part, sprites in self.parts_sprite.items():
|
||
for index, sprite in enumerate(sprites):
|
||
sprite_img = sprite.image
|
||
print(
|
||
f"sprite_img: {sprite_img} {part_data[part][index][1].x * 60} {part_data[part][index][1].y * 60}")
|
||
img_data = sprite_img.get_image_data()
|
||
fmt = img_data.format
|
||
if fmt != 'RGB':
|
||
fmt = 'RGBA'
|
||
pitch = -(img_data.width * len(fmt))
|
||
pil_image = Image.frombytes(fmt, (img_data.width, img_data.height), img_data.get_data(fmt, pitch))
|
||
|
||
pil_image = pil_image.rotate(-SR1Rotation.get_rotation(part_data[part][index][1].angle),
|
||
expand=True)
|
||
|
||
if part_data[part][index][1].flip_y:
|
||
pil_image.transpose(Image.FLIP_TOP_BOTTOM)
|
||
if part_data[part][index][1].flip_x:
|
||
pil_image.transpose(Image.FLIP_LEFT_RIGHT)
|
||
|
||
img.paste(pil_image, (
|
||
int(part_data[part][index][1].x * 60 + img_center[0]),
|
||
int(-part_data[part][index][1].y * 60 + img_center[1])))
|
||
|
||
img.save(f'test{time.time()}.png', 'PNG')
|
||
|
||
elif command.find('test'):
|
||
if command.find('save'):
|
||
if not self.status.draw_done:
|
||
return
|
||
if not DR_mod_runtime.use_DR_rust:
|
||
return
|
||
logger.info(sr_tr().sr1.ship.save.start().format(self.rust_ship))
|
||
self.rust_ship.save('./test-save.xml')
|
||
|
||
def on_mouse_drag(self, x: int, y: int, dx: int, dy: int, buttons: int, modifiers: int, window: ClientWindow):
|
||
if self.status.focus:
|
||
self.group_camera.view_x += dx
|
||
self.group_camera.view_y += dy
|
||
self.status.update_call = True
|
||
elif self.status.moving:
|
||
# 如果是在移动整体渲染位置
|
||
self.dx += dx
|
||
self.dy += dy
|
||
|
||
def on_file_drop(self, x: int, y: int, paths: List[str], window: ClientWindow):
|
||
if len(paths) > 1:
|
||
for path in paths:
|
||
try:
|
||
self.load_xml(path)
|
||
except Exception:
|
||
traceback.print_exc()
|
||
else:
|
||
if Path(paths[0]).is_dir():
|
||
for path in Path(paths[0]).glob('*.xml'):
|
||
try:
|
||
self.load_xml(str(path))
|
||
except ValueError:
|
||
traceback.print_exc()
|
||
if self.load_xml(paths[0]):
|
||
self.render_ship()
|
||
# for path in paths:
|
||
# if self.load_xml(path): # 加载成功一个就停下
|
||
# break
|
||
# self.render_ship()
|
||
|
||
@property
|
||
def view(self):
|
||
return self.window_pointer.view
|
||
|
||
@view.setter
|
||
def view(self, value: Mat4):
|
||
self.window_pointer.view = value
|
||
|
||
|
||
if __name__ == '__main__':
|
||
from objprint import op
|
||
|
||
op(SR1ShipRenderStatus())
|