1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
|
# Copyright (c) 2022 Ultimaker B.V.
# Uranium is released under the terms of the LGPLv3 or higher.
from PyQt6.QtGui import QImage
from PyQt6.QtOpenGL import QOpenGLFramebufferObject, QOpenGLFramebufferObjectFormat
class FrameBufferObject:
"""An interface for OpenGL FrameBuffer Objects.
This class describes a minimal interface that is expected of FrameBuffer Object
classes.
"""
def __init__(self, width: int, height: int) -> None:
super().__init__()
buffer_format = QOpenGLFramebufferObjectFormat()
buffer_format.setAttachment(QOpenGLFramebufferObject.Attachment.Depth)
self._fbo = QOpenGLFramebufferObject(width, height, buffer_format)
self._contents = None
def getTextureId(self) -> int:
"""Get the texture ID of the texture target of this FBO."""
return self._fbo.texture()
def bind(self) -> None:
"""Bind the FBO so it can be rendered to."""
self._contents = None
self._fbo.bind()
def release(self) -> None:
"""Release the FBO so it will no longer be rendered to."""
self._fbo.release()
def getContents(self) -> QImage:
"""Get the contents of the FBO as an image data object."""
if not self._contents:
self._contents = self._fbo.toImage()
return self._contents
|
