在 Python 中使用 PySide6 和 OpenGL 实现绘制多个图形并对这些图形进行拾取操作，需要结合帧缓冲区（FBO）、颜色编码和像素读取技术。以下是一个完整的实现示例，涵盖图形绘制、对象拾取以及相关 OpenGL 状态管理。 ### ### 图形绘制与拾取的基本流程 在 OpenGL 中实现拾取功能的核心是使用帧缓冲区（FBO）进行离屏渲染，并为每个可交互对象分配唯一的颜色标识。当用户点击屏幕时，程序会读取点击位置的像素值，并将其转换为对应的对象 ID。 以下是一个使用 PySide6 和 OpenGL 实现绘制 4 个点并拾取的完整示例： ```python import sys import numpy as np from PySide6.QtCore import Qt from PySide6.QtGui import QSurfaceFormat, QOpenGLShader, QOpenGLShaderProgram from PySide6.QtWidgets import QApplication, QMainWindow from PySide6.QtOpenGLWidgets import QOpenGLWidget from OpenGL.GL import * class GLPickWidget(QOpenGLWidget): def __init__(self): super().__init__() self.setMouseTracking(True) # 每个点的数据 (x, y, z, r, g, b) self.vertex_datas = [ np.array([0.0, 0.0, 0.0, 1.0, 0.0, 0.0], dtype=np.float32), np.array([0.5, 0.5, 0.0, 0.0, 1.0, 0.0], dtype=np.float32), np.array([-0.5, 0.5, 0.0, 0.0, 0.0, 1.0], dtype=np.float32), np.array([0.0, -0.5, 0.0, 1.0, 1.0, 0.0], dtype=np.float32), ] self.picking_shader = None self.vbo = None self.fbo = None self.picking_texture = None def initializeGL(self): glEnable(GL_DEPTH_TEST) # 创建帧缓冲区 self.fbo = glGenFramebuffers(1) glBindFramebuffer(GL_FRAMEBUFFER, self.fbo) # 创建纹理附件 self.picking_texture = glGenTextures(1) glBindTexture(GL_TEXTURE_2D, self.picking_texture) glTexImage2D(GL_TEXTURE_2D, 0, GL_RGB32UI, self.width(), self.height(), 0, GL_RGB_INTEGER, GL_UNSIGNED_INT, None) glTexParameteri(GL_TEXTURE_2D, GL_TEXTURE_MIN_FILTER, GL_NEAREST) glTexParameteri(GL_TEXTURE_2D, GL_TEXTURE_MAG_FILTER, GL_NEAREST) glFramebufferTexture2D(GL_FRAMEBUFFER, GL_COLOR_ATTACHMENT0, GL_TEXTURE_2D, self.picking_texture, 0) glBindFramebuffer(GL_FRAMEBUFFER, 0) # 创建着色器程序 self.picking_shader = QOpenGLShaderProgram() vertex_shader = QOpenGLShader(QOpenGLShader.Vertex) vertex_shader.compileSourceCode(""" #version 330 core layout(location = 0) in vec3 position; layout(location = 1) in vec3 color; out vec3 fragColor; void main() { gl_Position = vec4(position, 1.0); fragColor = color; } """) fragment_shader = QOpenGLShader(QOpenGLShader.Fragment) fragment_shader.compileSourceCode(""" #version 330 core in vec3 fragColor; out vec4 outColor; void main() { outColor = vec4(fragColor, 1.0); } """) self.picking_shader.addShader(vertex_shader) self.picking_shader.addShader(fragment_shader) self.picking_shader.link() # 创建 VBO self.vbo = glGenBuffers(1) def paintGL(self): glClearColor(0.0, 0.0, 0.0, 1.0) glClear(GL_COLOR_BUFFER_BIT | GL_DEPTH_BUFFER_BIT) # 绘制主场景 self.picking_shader.bind() glBindBuffer(GL_ARRAY_BUFFER, self.vbo) glEnableVertexAttribArray(0) glVertexAttribPointer(0, 3, GL_FLOAT, GL_FALSE, 6 * 4, None) glEnableVertexAttribArray(1) glVertexAttribPointer(1, 3, GL_FLOAT, GL_FALSE, 6 * 4, ctypes.c_void_p(3 * 4)) for idx, vertex_data in enumerate(self.vertex_datas): glBufferSubData(GL_ARRAY_BUFFER, 0, vertex_data.nbytes, vertex_data) glDrawArrays(GL_POINTS, 0, len(vertex_data)) glDisableVertexAttribArray(0) glDisableVertexAttribArray(1) self.picking_shader.release() def mousePressEvent(self, event): if event.button() == Qt.LeftButton: x = event.position().x() y = event.position().y() # 绑定帧缓冲区进行离屏渲染 glBindFramebuffer(GL_FRAMEBUFFER, self.fbo) glViewport(0, 0, self.width(), self.height()) glClearColor(0.0, 0.0, 0.0, 1.0) glClear(GL_COLOR_BUFFER_BIT | GL_DEPTH_BUFFER_BIT) self.picking_shader.bind() glBindBuffer(GL_ARRAY_BUFFER, self.vbo) glEnableVertexAttribArray(0) glVertexAttribPointer(0, 3, GL_FLOAT, GL_FALSE, 6 * 4, None) glEnableVertexAttribArray(1) glVertexAttribPointer(1, 3, GL_FLOAT, GL_FALSE, 6 * 4, ctypes.c_void_p(3 * 4)) for idx, vertex_data in enumerate(self.vertex_datas): id = idx + 1 r = ((id >> 16) & 0xFF) / 255.0 g = ((id >> 8) & 0xFF) / 255.0 b = (id & 0xFF) / 255.0 vertex_data[3:] = [r, g, b] glBufferSubData(GL_ARRAY_BUFFER, 0, vertex_data.nbytes, vertex_data) glDrawArrays(GL_POINTS, 0, len(vertex_data)) glDisableVertexAttribArray(0) glDisableVertexAttribArray(1) self.picking_shader.release() # 读取像素 glReadBuffer(GL_COLOR_ATTACHMENT0) pixel = np.zeros((1, 1, 3), dtype=np.uint32) glReadPixels(x, self.height() - y, 1, 1, GL_RGB_INTEGER, GL_UNSIGNED_INT, pixel) glBindFramebuffer(GL_FRAMEBUFFER, 0) objectId = (pixel[0, 0, 0] << 16) | (pixel[0, 0, 1] << 8) | pixel[0, 0, 2] print(f"Picked object ID: {objectId}") super().mousePressEvent(event) if __name__ == "__main__": app = QApplication(sys.argv) window = QMainWindow() widget = GLPickWidget() window.setCentralWidget(widget) window.resize(600, 600) window.show() sys.exit(app.exec()) ``` ### ### 代码说明 - **帧缓冲区初始化**：通过 `glGenFramebuffers` 和 `glTexImage2D` 创建帧缓冲区及其颜色附件，使用 `GL_RGB32UI` 格式以支持整数颜色值[^1]。 - **着色器设置**：定义顶点和片段着色器，用于在离屏渲染阶段为每个图形分配唯一颜色标识。 - **离屏绘制**：在 `mousePressEvent` 中绑定帧缓冲区，为每个图形设置唯一颜色，并绘制图形。 - **像素读取与对象识别**：调用 `glReadPixels` 读取点击位置的像素值，并将其转换为对象 ID。 ###