python_graphics_demos/00_qt5_window_test.py at master · chatyan/python_graphics_demos · GitHub

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
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
import moderngl
import pyrr
from PyQt5 import QtOpenGL, QtWidgets, QtCore
import numpy

from util_moderngl_qt import DrawerMeshVtxColor


class MyQtGLWidget(QtOpenGL.QGLWidget):

    def __init__(self, drawer, parent=None):
        self.ctx = None
        self.parent = parent
        fmt = QtOpenGL.QGLFormat()
        fmt.setVersion(3, 3)
        fmt.setProfile(QtOpenGL.QGLFormat.CoreProfile)
        fmt.setSampleBuffers(True)
        super(MyQtGLWidget, self).__init__(fmt, None)
        #
        self.resize(640, 480)
        self.setWindowTitle('Mesh Viewer')
        self.drawer = drawer

    def initializeGL(self):
        self.ctx = moderngl.create_context()
        self.drawer.init_gl(self.ctx)

    def paintGL(self):
        self.ctx.clear(1.0, 0.8, 1.0)
        self.ctx.enable(moderngl.DEPTH_TEST)
        mvp = pyrr.Matrix44.identity()
        self.drawer.paint_gl(mvp)

    def resizeGL(self, width, height):
        width = max(2, width)
        height = max(2, height)
        self.ctx.viewport = (0, 0, width, height)


if __name__ == '__main__':
    V = numpy.array([
        [-0.5, -0.5, 0],
        [+0.5, -0.5, 0],
        [+0, +0.5, 0]], dtype=numpy.float32)
    C = numpy.array([
        [1, 0, 0],
        [0, 1, 0],
        [0, 0, 1]], dtype=numpy.float32)
    F = numpy.array([
        [0, 1, 2]], dtype=numpy.uint32)

    with QtWidgets.QApplication([]) as app:
        drawer = DrawerMeshVtxColor.Drawer(V=V.tobytes(), C=C.tobytes(), F=F.tobytes())
        win = MyQtGLWidget(drawer)
        win.show()
        app.exec()