00_qt5_window_test.py

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()