#!/usr/bin/env python3 # -*- coding: utf-8 -*- """ 独立的坐标轴点击测试demo 专门用于测试和调试坐标轴的点击检测和拖拽约束功能 """ import sys import math from direct.showbase.ShowBase import ShowBase from panda3d.core import ( Vec3, Vec4, Point3, Point2, LineSegs, AmbientLight, DirectionalLight, TextNode, CollisionTraverser, CollisionHandlerQueue, CollisionNode, CollisionRay, GeomNode, CardMaker, Material ) from direct.task import Task from direct.gui.DirectGui import DirectFrame, DirectLabel class GizmoTest(ShowBase): def __init__(self): ShowBase.__init__(self) # 关闭默认的鼠标控制 self.disableMouse() print("坐标轴测试程序启动") # 设置背景色 self.setBackgroundColor(0.2, 0.2, 0.3) # 设置相机位置 self.cam.setPos(10, -20, 10) self.cam.lookAt(0, 0, 0) # 添加光照 self.setupLighting() # 创建参考立方体 self.createReferenceCube() # 创建坐标轴 self.createGizmo() # 设置点击检测 self.setupClickDetection() # 创建信息显示 self.createInfoDisplay() # 拖拽状态变量 self.isDragging = False self.dragAxis = None self.dragStartMousePos = None self.dragStartCubePos = None # 高亮状态 self.highlightAxis = None # 设置事件处理 self.accept("mouse1", self.onMouseClick) self.accept("mouse1-up", self.onMouseRelease) self.taskMgr.add(self.mouseTask, "mouseTask") # 添加颜色测试任务 self.taskMgr.add(self.colorTestTask, "colorTest") self.test_time = 0 print("=== Gizmo Click Test Demo Started ===") print("Instructions:") print("- ONLY click on red/green/blue gizmo axes to drag") print("- Each axis constrains movement to that direction only") print("- Click elsewhere should do nothing") print("- Real-time display of mouse and projection info") print("- Color test will cycle through colors every 3 seconds") def setupLighting(self): """设置光照""" # 环境光 alight = AmbientLight('alight') alight.setColor((0.3, 0.3, 0.3, 1)) alnp = self.render.attachNewNode(alight) self.render.setLight(alnp) # 定向光 dlight = DirectionalLight('dlight') dlight.setColor((0.8, 0.8, 0.8, 1)) dlnp = self.render.attachNewNode(dlight) dlnp.setHpr(45, -45, 0) self.render.setLight(dlnp) def createReferenceCube(self): """创建参考立方体""" # 创建立方体 cm = CardMaker('cube') cm.setFrame(-1, 1, -1, 1) # 创建6个面 faces = [ ('front', (0, 1, 0), (0, 0, 0), (1, 0, 0, 1)), # 红色前面 ('back', (0, -1, 0), (0, 0, 180), (0, 1, 0, 1)), # 绿色后面 ('left', (-1, 0, 0), (0, 0, 90), (0, 0, 1, 1)), # 蓝色左面 ('right', (1, 0, 0), (0, 0, -90), (1, 1, 0, 1)), # 黄色右面 ('top', (0, 0, 1), (-90, 0, 0), (1, 0, 1, 1)), # 紫色顶面 ('bottom', (0, 0, -1), (90, 0, 0), (0, 1, 1, 1)) # 青色底面 ] self.cube = self.render.attachNewNode("cube") for name, pos, hpr, color in faces: face = self.cube.attachNewNode(cm.generate()) face.setPos(*pos) face.setHpr(*hpr) face.setColor(*color) face.setName(name) print(f"✓ 参考立方体创建完成，位置: {self.cube.getPos()}") def createGizmo(self): """创建坐标轴""" self.gizmo = self.render.attachNewNode("gizmo") self.gizmo.setPos(0, 0, 3) # 放在立方体上方 self.axis_length = 3.0 self.click_threshold = 25 # 点击检测阈值（像素） # 创建X轴（不在LineSegs级别设置颜色） x_lines = LineSegs("x_axis") x_lines.setThickness(6.0) # 不在LineSegs级别设置颜色！ x_lines.moveTo(0, 0, 0) x_lines.drawTo(self.axis_length, 0, 0) # 箭头 x_lines.moveTo(self.axis_length - 0.3, -0.1, 0) x_lines.drawTo(self.axis_length, 0, 0) x_lines.drawTo(self.axis_length - 0.3, 0.1, 0) x_geom = x_lines.create() self.gizmoXAxis = self.gizmo.attachNewNode(x_geom) self.gizmoXAxis.setLightOff() # 使用RenderState强制设置颜色 from panda3d.core import RenderState, ColorAttrib red_state = RenderState.make(ColorAttrib.makeFlat((1, 0, 0, 1))) self.gizmoXAxis.setState(red_state) self.gizmoXAxis.setColor(1, 0, 0, 1) # 创建Y轴（不在LineSegs级别设置颜色） y_lines = LineSegs("y_axis") y_lines.setThickness(6.0) # 不在LineSegs级别设置颜色！ y_lines.moveTo(0, 0, 0) y_lines.drawTo(0, self.axis_length, 0) # 箭头 y_lines.moveTo(-0.1, self.axis_length - 0.3, 0) y_lines.drawTo(0, self.axis_length, 0) y_lines.drawTo(0.1, self.axis_length - 0.3, 0) y_geom = y_lines.create() self.gizmoYAxis = self.gizmo.attachNewNode(y_geom) self.gizmoYAxis.setLightOff() # 使用RenderState强制设置颜色 green_state = RenderState.make(ColorAttrib.makeFlat((0, 1, 0, 1))) self.gizmoYAxis.setState(green_state) self.gizmoYAxis.setColor(0, 1, 0, 1) # 创建Z轴（不在LineSegs级别设置颜色） z_lines = LineSegs("z_axis") z_lines.setThickness(6.0) # 不在LineSegs级别设置颜色！ z_lines.moveTo(0, 0, 0) z_lines.drawTo(0, 0, self.axis_length) # 箭头 z_lines.moveTo(-0.1, 0, self.axis_length - 0.3) z_lines.drawTo(0, 0, self.axis_length) z_lines.drawTo(0.1, 0, self.axis_length - 0.3) z_geom = z_lines.create() self.gizmoZAxis = self.gizmo.attachNewNode(z_geom) self.gizmoZAxis.setLightOff() # 使用RenderState强制设置颜色 blue_state = RenderState.make(ColorAttrib.makeFlat((0, 0, 1, 1))) self.gizmoZAxis.setState(blue_state) self.gizmoZAxis.setColor(0, 0, 1, 1) # 轴颜色字典 self.axis_colors = { "x": (1, 0, 0, 1), # 红色 "y": (0, 1, 0, 1), # 绿色 "z": (0, 0, 1, 1) # 蓝色 } print(f"✓ 坐标轴创建完成，长度={self.axis_length}") print(f"X轴节点颜色: {self.gizmoXAxis.getColor()}") print(f"Y轴节点颜色: {self.gizmoYAxis.getColor()}") print(f"Z轴节点颜色: {self.gizmoZAxis.getColor()}") def setupClickDetection(self): """设置点击检测""" self.picker = CollisionTraverser() self.pq = CollisionHandlerQueue() self.pickerNode = CollisionNode('mouseRay') self.pickerNP = self.camera.attachNewNode(self.pickerNode) self.pickerNode.setFromCollideMask(GeomNode.getDefaultCollideMask()) self.picker.addCollider(self.pickerNP, self.pq) def createInfoDisplay(self): """创建信息显示面板""" self.infoPanel = DirectFrame( frameColor=(0, 0, 0, 0.7), frameSize=(-0.4, 0.4, -0.3, 0.3), pos=(-0.6, 0, 0.8) ) self.infoText = DirectLabel( parent=self.infoPanel, text="Mouse Info\nWaiting...", text_scale=0.05, text_fg=(1, 1, 1, 1), frameColor=(0, 0, 0, 0), pos=(0, 0, 0) ) def worldToScreen(self, world_pos): """将世界坐标转换为屏幕坐标""" try: # 转换为相机坐标系 cam_pos = self.cam.getRelativePoint(self.render, world_pos) # 检查是否在相机前方 if cam_pos.getY() <= 0: return None # 使用相机lens进行投影 screen_pos = Point2() if self.cam.node().getLens().project(cam_pos, screen_pos): # 转换为像素坐标 win_x = (screen_pos.getX() + 1.0) * 0.5 * self.win.getXSize() win_y = (1.0 - screen_pos.getY()) * 0.5 * self.win.getYSize() return (win_x, win_y) return None except: return None def distanceToLine(self, point, line_start, line_end): """计算点到线段的距离""" try: px, py = point x1, y1 = line_start x2, y2 = line_end # 计算线段长度 line_length_sq = (x2 - x1) ** 2 + (y2 - y1) ** 2 if line_length_sq < 0.001: return math.sqrt((px - x1) ** 2 + (py - y1) ** 2) # 计算投影参数 t = max(0, min(1, ((px - x1) * (x2 - x1) + (py - y1) * (y2 - y1)) / line_length_sq)) # 计算投影点 projection_x = x1 + t * (x2 - x1) projection_y = y1 + t * (y2 - y1) # 返回距离 distance = math.sqrt((px - projection_x) ** 2 + (py - projection_y) ** 2) return distance except: return float('inf') def checkGizmoClick(self, mouse_x, mouse_y): """检查是否点击了坐标轴""" if not self.gizmo: return None # 获取坐标轴中心的世界坐标 gizmo_world_pos = self.gizmo.getPos(self.render) # 计算各轴端点的世界坐标 x_end = gizmo_world_pos + Vec3(self.axis_length, 0, 0) y_end = gizmo_world_pos + Vec3(0, self.axis_length, 0) z_end = gizmo_world_pos + Vec3(0, 0, self.axis_length) # 投影到屏幕坐标 center_screen = self.worldToScreen(gizmo_world_pos) x_screen = self.worldToScreen(x_end) y_screen = self.worldToScreen(y_end) z_screen = self.worldToScreen(z_end) if not center_screen: return None # 检测各个轴 axes_data = [ ("x", x_screen, "X轴"), ("y", y_screen, "Y轴"), ("z", z_screen, "Z轴") ] closest_axis = None closest_distance = float('inf') for axis_name, axis_screen, axis_label in axes_data: if axis_screen: distance = self.distanceToLine( (mouse_x, mouse_y), center_screen, axis_screen ) if distance < self.click_threshold and distance < closest_distance: closest_axis = axis_name closest_distance = distance return closest_axis def setAxisColor(self, axis, color): """设置坐标轴颜色 - 使用RenderState强制覆盖""" from panda3d.core import RenderState, ColorAttrib # 创建强制颜色状态 color_state = RenderState.make(ColorAttrib.makeFlat(color)) if axis == "x" and self.gizmoXAxis: self.gizmoXAxis.setState(color_state) self.gizmoXAxis.setColor(*color) self.gizmoXAxis.setColorScale(*color) print(f"✓ 强制设置X轴颜色: {color}") elif axis == "y" and self.gizmoYAxis: self.gizmoYAxis.setState(color_state) self.gizmoYAxis.setColor(*color) self.gizmoYAxis.setColorScale(*color) print(f"✓ 强制设置Y轴颜色: {color}") elif axis == "z" and self.gizmoZAxis: self.gizmoZAxis.setState(color_state) self.gizmoZAxis.setColor(*color) self.gizmoZAxis.setColorScale(*color) print(f"✓ 强制设置Z轴颜色: {color}") def mouseTask(self, task): """鼠标任务 - 处理高亮和拖拽""" if not self.mouseWatcherNode.hasMouse(): return task.cont # 获取鼠标屏幕坐标 mouse_x = (self.mouseWatcherNode.getMouseX() + 1.0) * 0.5 * self.win.getXSize() mouse_y = (1.0 - self.mouseWatcherNode.getMouseY()) * 0.5 * self.win.getYSize() # 检查是否在颜色测试周期内 cycle_time = int(self.test_time) % 15 is_color_testing = (cycle_time >= 3 and cycle_time < 12) # 在黄、白、橙色测试期间 # 更新信息显示 info_text = f"Mouse Info:\n" info_text += f"Normalized: ({self.mouseWatcherNode.getMouseX():.3f}, {self.mouseWatcherNode.getMouseY():.3f})\n" info_text += f"Pixels: ({mouse_x:.1f}, {mouse_y:.1f})\n" info_text += f"Window Size: {self.win.getXSize()} x {self.win.getYSize()}\n" if is_color_testing: info_text += f"Color Test Active (cycle: {cycle_time})\n" # 只在不拖拽且不在颜色测试时检测高亮 if not self.isDragging and not is_color_testing: # 简化的高亮检测 hovered_axis = None if self.gizmo: # 获取坐标轴中心的世界坐标 gizmo_world_pos = self.gizmo.getPos(self.render) # 计算各轴端点的世界坐标 x_end = gizmo_world_pos + Vec3(self.axis_length, 0, 0) y_end = gizmo_world_pos + Vec3(0, self.axis_length, 0) z_end = gizmo_world_pos + Vec3(0, 0, self.axis_length) # 投影到屏幕坐标 center_screen = self.worldToScreen(gizmo_world_pos) x_screen = self.worldToScreen(x_end) y_screen = self.worldToScreen(y_end) z_screen = self.worldToScreen(z_end) if center_screen: # 检测距离最近的轴 min_distance = float('inf') for axis_name, axis_screen in [("x", x_screen), ("y", y_screen), ("z", z_screen)]: if axis_screen: distance = self.distanceToLine((mouse_x, mouse_y), center_screen, axis_screen) if distance < self.click_threshold and distance < min_distance: min_distance = distance hovered_axis = axis_name # 处理高亮状态变化 if hovered_axis != self.highlightAxis: # 恢复之前高亮的轴 if self.highlightAxis: self.setAxisColor(self.highlightAxis, self.axis_colors[self.highlightAxis]) print(f"恢复 {self.highlightAxis} 轴原色") # 高亮新的轴 if hovered_axis: self.setAxisColor(hovered_axis, (1, 1, 0, 1)) # 纯黄色高亮 print(f"高亮 {hovered_axis} 轴为黄色") self.highlightAxis = hovered_axis # 更新信息显示 if hovered_axis: info_text += f"Hover: {hovered_axis.upper()} axis (YELLOW)" else: info_text += "Hover: None" elif is_color_testing: info_text += "Hover: Disabled (Color Test)" else: info_text += f"Dragging: {self.dragAxis.upper()} axis" # 处理拖拽 if self.isDragging: self.handleDrag(mouse_x, mouse_y) self.infoText.setText(info_text) return task.cont def onMouseClick(self): """鼠标点击事件 - 只有点击坐标轴才开始拖拽""" if not self.mouseWatcherNode.hasMouse(): return # 检查是否在颜色测试期间 cycle_time = int(self.test_time) % 15 is_color_testing = (cycle_time >= 3 and cycle_time < 12) if is_color_testing: print("点击被禁用 - 颜色测试进行中") return # 获取鼠标坐标 mouse_x = (self.mouseWatcherNode.getMouseX() + 1.0) * 0.5 * self.win.getXSize() mouse_y = (1.0 - self.mouseWatcherNode.getMouseY()) * 0.5 * self.win.getYSize() print(f"\n=== Mouse Click ===") print(f"Pixel Coords: ({mouse_x:.1f}, {mouse_y:.1f})") # 检测坐标轴点击 clicked_axis = self.checkGizmoClick(mouse_x, mouse_y) if clicked_axis: print(f"✓ Clicked {clicked_axis.upper()} Axis - Starting drag") # 只有真正点击到轴才开始拖拽 self.isDragging = True self.dragAxis = clicked_axis self.dragStartMousePos = (mouse_x, mouse_y) self.dragStartCubePos = self.cube.getPos() # 显示拖拽颜色 - 使用更明显的橙色 self.setAxisColor(clicked_axis, (1, 0.5, 0, 1)) # 更亮的橙色表示拖拽 print(f" 开始位置: {self.dragStartCubePos}") else: print("✗ No axis clicked - No action") # 确保没有点击到轴时不开始拖拽 self.isDragging = False self.dragAxis = None def onMouseRelease(self): """鼠标释放事件""" if self.isDragging and self.dragAxis: print(f"Stop dragging {self.dragAxis.upper()} axis") # 恢复轴颜色 self.setAxisColor(self.dragAxis, self.axis_colors[self.dragAxis]) # 清理所有拖拽状态 self.isDragging = False self.dragAxis = None self.dragStartMousePos = None self.dragStartCubePos = None def handleDrag(self, mouse_x, mouse_y): """处理拖拽 - 真正的轴约束拖拽""" if not self.dragAxis or not self.dragStartMousePos or not self.dragStartCubePos: return # 计算从拖拽开始到现在的总鼠标移动距离 total_dx = mouse_x - self.dragStartMousePos[0] total_dy = mouse_y - self.dragStartMousePos[1] # 根据拖拽的轴，确定世界空间中的轴方向向量 if self.dragAxis == "x": world_axis_dir = Vec3(1, 0, 0) # X轴方向 elif self.dragAxis == "y": world_axis_dir = Vec3(0, 1, 0) # Y轴方向 elif self.dragAxis == "z": world_axis_dir = Vec3(0, 0, 1) # Z轴方向 else: return # 将世界轴方向投影到屏幕空间 axis_start_world = self.dragStartCubePos axis_end_world = self.dragStartCubePos + world_axis_dir axis_start_screen = self.worldToScreen(axis_start_world) axis_end_screen = self.worldToScreen(axis_end_world) if not axis_start_screen or not axis_end_screen: return # 计算轴在屏幕空间的方向向量 screen_axis_dir = ( axis_end_screen[0] - axis_start_screen[0], axis_end_screen[1] - axis_start_screen[1] ) # 归一化屏幕轴方向 screen_axis_length = math.sqrt(screen_axis_dir[0]**2 + screen_axis_dir[1]**2) if screen_axis_length > 0.001: screen_axis_dir = ( screen_axis_dir[0] / screen_axis_length, screen_axis_dir[1] / screen_axis_length ) else: return # 将总鼠标移动向量投影到屏幕轴方向上 projected_distance = ( total_dx * screen_axis_dir[0] + total_dy * screen_axis_dir[1] ) # 将投影距离转换为世界空间移动量 move_scale = 0.02 # 移动灵敏度 world_movement = world_axis_dir * projected_distance * move_scale # 从拖拽开始位置计算新位置（确保只沿选定轴移动） new_cube_pos = self.dragStartCubePos + world_movement self.cube.setPos(new_cube_pos) # 坐标轴跟随立方体移动 self.gizmo.setPos(new_cube_pos + Vec3(0, 0, 3)) def colorTestTask(self, task): """颜色测试任务 - 每3秒循环测试坐标轴颜色""" from panda3d.core import RenderState, ColorAttrib self.test_time += globalClock.getDt() # 每3秒切换一次测试 cycle_time = int(self.test_time) % 15 # 15秒一个完整周期 if cycle_time < 3: # 正常颜色 if hasattr(self, 'gizmoXAxis'): self.setAxisColor("x", (1, 0, 0, 1)) # 红色 self.setAxisColor("y", (0, 1, 0, 1)) # 绿色 self.setAxisColor("z", (0, 0, 1, 1)) # 蓝色 elif cycle_time < 6: # 全部黄色 if hasattr(self, 'gizmoXAxis'): self.setAxisColor("x", (1, 1, 0, 1)) self.setAxisColor("y", (1, 1, 0, 1)) self.setAxisColor("z", (1, 1, 0, 1)) elif cycle_time < 9: # 全部白色 if hasattr(self, 'gizmoXAxis'): self.setAxisColor("x", (1, 1, 1, 1)) self.setAxisColor("y", (1, 1, 1, 1)) self.setAxisColor("z", (1, 1, 1, 1)) elif cycle_time < 12: # 全部橙色 if hasattr(self, 'gizmoXAxis'): self.setAxisColor("x", (1, 0.5, 0, 1)) self.setAxisColor("y", (1, 0.5, 0, 1)) self.setAxisColor("z", (1, 0.5, 0, 1)) else: # 恢复正常颜色 if hasattr(self, 'gizmoXAxis'): self.setAxisColor("x", (1, 0, 0, 1)) # 红色 self.setAxisColor("y", (0, 1, 0, 1)) # 绿色 self.setAxisColor("z", (0, 0, 1, 1)) # 蓝色 return task.cont if __name__ == "__main__": app = GizmoTest() app.run()