ui替换
This commit is contained in:
parent
1ad2086525
commit
3f6ae9185e
@ -1,166 +0,0 @@
|
||||
"""
|
||||
pyimgui管理器 - 在Panda3D引擎上集成ImGui界面
|
||||
"""
|
||||
import imgui
|
||||
import OpenGL.GL as gl
|
||||
from panda3d.core import RenderState, ShaderAttrib, Texture, CardMaker, NodePath
|
||||
from panda3d.core import TransparencyAttrib, ColorBlendAttrib
|
||||
from direct.task.Task import Task
|
||||
import numpy as np
|
||||
|
||||
class ImGuiManager:
|
||||
"""ImGui管理器，负责在Panda3D引擎上渲染ImGui界面"""
|
||||
|
||||
def __init__(self, world):
|
||||
self.world = world
|
||||
self.io = None
|
||||
self.font_texture = None
|
||||
self._setup_imgui()
|
||||
self._create_imgui_task()
|
||||
|
||||
# UI状态
|
||||
self.show_demo_window = False
|
||||
self.show_test_button = True
|
||||
self.button_clicked = False
|
||||
|
||||
def _setup_imgui(self):
|
||||
"""初始化ImGui设置"""
|
||||
# 初始化ImGui上下文
|
||||
imgui.create_context()
|
||||
self.io = imgui.get_io()
|
||||
|
||||
# 设置显示大小
|
||||
win_props = self.world.win.getProperties()
|
||||
self.io.display_size = (win_props.get_x_size(), win_props.get_y_size())
|
||||
|
||||
# 设置渲染回调
|
||||
self.io.render_callback = self._render_imgui
|
||||
|
||||
# 创建字体纹理
|
||||
self._create_font_texture()
|
||||
|
||||
print("✓ ImGui初始化完成")
|
||||
|
||||
def _create_font_texture(self):
|
||||
"""创建ImGui字体纹理"""
|
||||
# 获取字体数据
|
||||
font_pixels = self.io.fonts.get_tex_data_as_rgba32()
|
||||
width, height = self.io.fonts.get_tex_data_as_rgba32()[1]
|
||||
|
||||
# 创建Panda3D纹理
|
||||
self.font_texture = Texture()
|
||||
self.font_texture.setup_2d_texture(width, height, Texture.T_unsigned_byte, Texture.F_rgba8)
|
||||
self.font_texture.set_ram_image(font_pixels)
|
||||
|
||||
# 设置纹理ID
|
||||
self.io.fonts.tex_id = self.font_texture
|
||||
|
||||
# 清除字体数据
|
||||
self.io.fonts.clear_tex_data()
|
||||
|
||||
def _render_imgui(self, draw_data):
|
||||
"""渲染ImGui绘制数据"""
|
||||
if not draw_data:
|
||||
return
|
||||
|
||||
# 设置OpenGL状态
|
||||
gl.glEnable(gl.GL_BLEND)
|
||||
gl.glBlendEquation(gl.GL_FUNC_ADD)
|
||||
gl.glBlendFunc(gl.GL_SRC_ALPHA, gl.GL_ONE_MINUS_SRC_ALPHA)
|
||||
gl.glDisable(gl.GL_CULL_FACE)
|
||||
gl.glDisable(gl.GL_DEPTH_TEST)
|
||||
gl.glEnable(gl.GL_SCISSOR_TEST)
|
||||
|
||||
# 设置视口和投影矩阵
|
||||
display_w, display_h = self.io.display_size
|
||||
gl.glViewport(0, 0, int(display_w), int(display_h))
|
||||
|
||||
# 获取投影矩阵
|
||||
projection_matrix = self._get_projection_matrix()
|
||||
gl.glMatrixMode(gl.GL_PROJECTION)
|
||||
gl.glLoadMatrixf(projection_matrix)
|
||||
|
||||
# 渲染每个命令列表
|
||||
for commands in draw_data.commands_lists:
|
||||
# 渲染命令
|
||||
self._render_commands(commands)
|
||||
|
||||
# 恢复OpenGL状态
|
||||
gl.glDisable(gl.GL_SCISSOR_TEST)
|
||||
gl.glEnable(gl.GL_DEPTH_TEST)
|
||||
|
||||
def _get_projection_matrix(self):
|
||||
"""获取正交投影矩阵"""
|
||||
L, R, T, B = 0, self.io.display_size[0], self.io.display_size[1], 0
|
||||
projection_matrix = np.array([
|
||||
[2.0/(R-L), 0.0, 0.0, 0.0],
|
||||
[0.0, 2.0/(T-B), 0.0, 0.0],
|
||||
[0.0, 0.0, -1.0, 0.0],
|
||||
[(R+L)/(L-R), (T+B)/(B-T), 0.0, 1.0]
|
||||
], dtype=np.float32)
|
||||
return projection_matrix
|
||||
|
||||
def _render_commands(self, commands):
|
||||
"""渲染ImGui命令列表"""
|
||||
# 这里需要实现具体的渲染逻辑
|
||||
# 由于Panda3D的渲染管线与ImGui不完全兼容，我们需要创建一个自定义的渲染方法
|
||||
pass
|
||||
|
||||
def _create_imgui_task(self):
|
||||
"""创建ImGui更新任务"""
|
||||
taskMgr.add(self._update_imgui, "update_imgui")
|
||||
|
||||
def _update_imgui(self, task):
|
||||
"""更新ImGui界面"""
|
||||
# 开始新帧
|
||||
imgui.new_frame()
|
||||
|
||||
# 绘制UI
|
||||
self._draw_ui()
|
||||
|
||||
# 渲染ImGui
|
||||
imgui.render()
|
||||
|
||||
return Task.cont
|
||||
|
||||
def _draw_ui(self):
|
||||
"""绘制UI元素"""
|
||||
# 绘制测试按钮
|
||||
if self.show_test_button:
|
||||
imgui.begin("测试窗口", True)
|
||||
if imgui.button("点击我!"):
|
||||
self.button_clicked = True
|
||||
print("按钮被点击了!")
|
||||
|
||||
if self.button_clicked:
|
||||
imgui.text("按钮已被点击!")
|
||||
|
||||
imgui.end()
|
||||
|
||||
# 绘制演示窗口（可选）
|
||||
if self.show_demo_window:
|
||||
self.show_demo_window = imgui.show_demo_window(self.show_demo_window)
|
||||
|
||||
def toggle_demo_window(self):
|
||||
"""切换演示窗口显示"""
|
||||
self.show_demo_window = not self.show_demo_window
|
||||
|
||||
def toggle_test_button(self):
|
||||
"""切换测试按钮显示"""
|
||||
self.show_test_button = not self.show_test_button
|
||||
|
||||
def process_mouse_input(self, button_event):
|
||||
"""处理鼠标输入"""
|
||||
# 这里需要将Panda3D的鼠标事件转换为ImGui可以理解的事件
|
||||
pass
|
||||
|
||||
def process_keyboard_input(self, key_event):
|
||||
"""处理键盘输入"""
|
||||
# 这里需要将Panda3D的键盘事件转换为ImGui可以理解的事件
|
||||
pass
|
||||
|
||||
def cleanup(self):
|
||||
"""清理资源"""
|
||||
if self.font_texture:
|
||||
self.font_texture = None
|
||||
imgui.destroy_context()
|
||||
@ -1,204 +0,0 @@
|
||||
"""
|
||||
pyimgui管理器 - 在Panda3D引擎上集成ImGui界面
|
||||
使用更简单的方法，通过创建覆盖层来实现
|
||||
"""
|
||||
import imgui
|
||||
import numpy as np
|
||||
from panda3d.core import RenderState, ShaderAttrib, Texture, CardMaker, NodePath
|
||||
from panda3d.core import TransparencyAttrib, ColorBlendAttrib, PNMImage
|
||||
from panda3d.core import GraphicsOutput, RenderBuffer, TextureStage, TexGenAttrib
|
||||
from direct.task.Task import Task
|
||||
from direct.gui.OnscreenImage import OnscreenImage
|
||||
from direct.gui.OnscreenText import OnscreenText
|
||||
import PIL.Image
|
||||
import PIL.ImageDraw
|
||||
import io
|
||||
|
||||
class SimpleImGuiManager:
|
||||
"""简化的ImGui管理器，使用2D覆盖层在Panda3D引擎上显示ImGui界面"""
|
||||
|
||||
def __init__(self, world):
|
||||
self.world = world
|
||||
self.io = None
|
||||
self._setup_imgui()
|
||||
self._create_imgui_task()
|
||||
|
||||
# UI状态
|
||||
self.show_demo_window = False
|
||||
self.show_test_button = True
|
||||
self.button_clicked = False
|
||||
self.counter = 0
|
||||
|
||||
# 渲染相关
|
||||
self.imgui_texture = None
|
||||
self.imgui_image = None
|
||||
self.texture_buffer = None
|
||||
|
||||
# 创建渲染纹理
|
||||
self._create_render_texture()
|
||||
|
||||
def _setup_imgui(self):
|
||||
"""初始化ImGui设置"""
|
||||
# 初始化ImGui上下文
|
||||
imgui.create_context()
|
||||
self.io = imgui.get_io()
|
||||
|
||||
# 设置显示大小
|
||||
win_props = self.world.win.getProperties()
|
||||
self.io.display_size = (win_props.get_x_size(), win_props.get_y_size())
|
||||
|
||||
# 设置字体大小
|
||||
font_size = 16
|
||||
self.io.font_global_scale = font_size / 13.0
|
||||
|
||||
print("✓ ImGui初始化完成")
|
||||
|
||||
def _create_render_texture(self):
|
||||
"""创建用于渲染ImGui的纹理"""
|
||||
win_props = self.world.win.getProperties()
|
||||
width = win_props.get_x_size()
|
||||
height = win_props.get_y_size()
|
||||
|
||||
# 创建纹理
|
||||
self.imgui_texture = Texture()
|
||||
self.imgui_texture.setup_2d_texture(width, height, Texture.T_unsigned_byte, Texture.F_rgba8)
|
||||
|
||||
# 创建屏幕图像
|
||||
self.imgui_image = OnscreenImage(
|
||||
image=self.imgui_texture,
|
||||
pos=(0, 0, 0),
|
||||
scale=(1, 1, 1)
|
||||
)
|
||||
self.imgui_image.setTransparency(TransparencyAttrib.MAlpha)
|
||||
|
||||
def _create_imgui_task(self):
|
||||
"""创建ImGui更新任务"""
|
||||
taskMgr.add(self._update_imgui, "update_imgui")
|
||||
|
||||
def _update_imgui(self, task):
|
||||
"""更新ImGui界面"""
|
||||
# 开始新帧
|
||||
imgui.new_frame()
|
||||
|
||||
# 绘制UI
|
||||
self._draw_ui()
|
||||
|
||||
# 渲染ImGui到纹理
|
||||
self._render_to_texture()
|
||||
|
||||
return Task.cont
|
||||
|
||||
def _draw_ui(self):
|
||||
"""绘制UI元素"""
|
||||
# 绘制测试窗口
|
||||
if self.show_test_button:
|
||||
imgui.begin("测试窗口", True)
|
||||
|
||||
if imgui.button("点击我!"):
|
||||
self.button_clicked = True
|
||||
self.counter += 1
|
||||
print(f"按钮被点击了! 计数: {self.counter}")
|
||||
|
||||
if self.button_clicked:
|
||||
imgui.text(f"按钮已被点击 {self.counter} 次!")
|
||||
|
||||
imgui.text("这是一个在Panda3D引擎上显示的ImGui按钮")
|
||||
imgui.text(f"帧率: {globalClock.getAverageFrameRate():.1f} FPS")
|
||||
|
||||
imgui.end()
|
||||
|
||||
# 绘制演示窗口（可选）
|
||||
if self.show_demo_window:
|
||||
self.show_demo_window = imgui.show_demo_window(self.show_demo_window)
|
||||
|
||||
def _render_to_texture(self):
|
||||
"""将ImGui渲染到纹理"""
|
||||
# 渲染ImGui
|
||||
imgui.render()
|
||||
|
||||
# 获取绘制数据
|
||||
draw_data = imgui.get_draw_data()
|
||||
|
||||
if draw_data:
|
||||
# 创建PIL图像
|
||||
width, height = int(self.io.display_size[0]), int(self.io.display_size[1])
|
||||
image = PIL.Image.new('RGBA', (width, height), (0, 0, 0, 0))
|
||||
draw = PIL.ImageDraw.Draw(image)
|
||||
|
||||
# 简单的文本渲染（这里只是示例，实际需要完整的ImGui渲染）
|
||||
if self.show_test_button:
|
||||
# 绘制简单的按钮模拟
|
||||
button_x, button_y = 50, 50
|
||||
button_width, button_height = 100, 30
|
||||
draw.rectangle([button_x, button_y, button_x + button_width, button_y + button_height],
|
||||
fill=(100, 100, 200, 255), outline=(255, 255, 255, 255))
|
||||
draw.text((button_x + 10, button_y + 5), "点击我!", fill=(255, 255, 255, 255))
|
||||
|
||||
# 显示点击计数
|
||||
if self.counter > 0:
|
||||
draw.text((button_x, button_y + button_height + 10),
|
||||
f"点击次数: {self.counter}", fill=(255, 255, 255, 255))
|
||||
|
||||
# 将PIL图像转换为Panda3D纹理
|
||||
img_buffer = io.BytesIO()
|
||||
image.save(img_buffer, format='RGBA')
|
||||
img_buffer.seek(0)
|
||||
|
||||
pnm_image = PNMImage()
|
||||
pnm_image.read(StringStream(img_buffer.read()))
|
||||
self.imgui_texture.load(pnm_image)
|
||||
|
||||
def toggle_demo_window(self):
|
||||
"""切换演示窗口显示"""
|
||||
self.show_demo_window = not self.show_demo_window
|
||||
|
||||
def toggle_test_button(self):
|
||||
"""切换测试按钮显示"""
|
||||
self.show_test_button = not self.show_test_button
|
||||
|
||||
def process_mouse_click(self, x, y):
|
||||
"""处理鼠标点击"""
|
||||
# 检查是否点击了按钮区域
|
||||
if self.show_test_button:
|
||||
button_x, button_y = 50, 50
|
||||
button_width, button_height = 100, 30
|
||||
|
||||
# 转换坐标系（Panda3D坐标系原点在屏幕中心）
|
||||
screen_x = (x + self.io.display_size[0] / 2)
|
||||
screen_y = (self.io.display_size[1] / 2 - y)
|
||||
|
||||
if (button_x <= screen_x <= button_x + button_width and
|
||||
button_y <= screen_y <= button_y + button_height):
|
||||
self.button_clicked = True
|
||||
self.counter += 1
|
||||
print(f"按钮被点击了! 计数: {self.counter}")
|
||||
|
||||
def cleanup(self):
|
||||
"""清理资源"""
|
||||
if self.imgui_image:
|
||||
self.imgui_image.removeNode()
|
||||
if self.imgui_texture:
|
||||
self.imgui_texture = None
|
||||
imgui.destroy_context()
|
||||
|
||||
# 辅助类用于内存中的字符串流
|
||||
class StringStream:
|
||||
"""简单的内存字符串流，用于PNMImage读取"""
|
||||
def __init__(self, data):
|
||||
self.data = data
|
||||
self.position = 0
|
||||
|
||||
def read(self, size=-1):
|
||||
if size == -1:
|
||||
result = self.data[self.position:]
|
||||
self.position = len(self.data)
|
||||
else:
|
||||
result = self.data[self.position:self.position + size]
|
||||
self.position += size
|
||||
return result
|
||||
|
||||
def seek(self, position):
|
||||
self.position = position
|
||||
|
||||
def tell(self):
|
||||
return self.position
|
||||
Loading…
Reference in New Issue
Block a user