2021Graphics/Source/03_Primitives/step3.py at main · dknife/2021Graphics · 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
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
# opengl 라이브러리 임포트
from OpenGL.GL import *
from OpenGL.GLU import *

# pyQt 임포트
from PyQt5.QtWidgets import *
from PyQt5.QtCore import *
from PyQt5.QtGui import QPainter, QPen

import sys
import numpy as np

PRIMITIVES = ['GL_POINTS', 'GL_LINES', 'GL_LINE_STRIP',
              'GL_LINE_LOOP', 'GL_TRIANGLES', 'GL_TRIANGLE_STRIP',
              'GL_TRIANGLE_FAN', 'GL_QUADS', 'GL_QUAD_STRIP', 'GL_POLYGON']

PRIMITIVE_VALUES = [GL_POINTS, GL_LINES, GL_LINE_STRIP,
              GL_LINE_LOOP, GL_TRIANGLES, GL_TRIANGLE_STRIP,
              GL_TRIANGLE_FAN, GL_QUADS, GL_QUAD_STRIP, GL_POLYGON]

selected = 0

points = [ [0, 0], [10, 10], [100, 50]]

class MyWindow(QMainWindow):
    def __init__(self, title=''):
        super().__init__()

        central_widget = QWidget()
        layout = QHBoxLayout()
        self.setCentralWidget(central_widget)
        central_widget.setLayout(layout)

        # 1. OpenGL Widget
        self.opengl = MyGLWidget()
        # 2. Control Widget
        self.control = QGroupBox('Control')
        self.make_control()
        # 3. Painter Widget
        self.input = Drawer(parent = self)

        layout.addWidget(self.opengl)
        layout.addWidget(self.control)
        layout.addWidget(self.input)

    def make_control(self):
        control_layout = QVBoxLayout()
        selectCombo = QComboBox()
        resetButton = QPushButton('reset points', self)
        control_layout.addWidget(selectCombo)
        control_layout.addWidget(resetButton)
        self.control.setLayout(control_layout)

        for i in range( len(PRIMITIVES)) :
            selectCombo.addItem(PRIMITIVES[i])

        selectCombo.currentIndexChanged.connect(self.select)

    def select(self, value):
        global selected

        selected = value
        self.opengl.update()

class MyGLWidget(QOpenGLWidget):
    def __init__(self):
        super().__init__()

    def initializeGL(self):
        # OpenGL 그리기를 수행하기 전에 각종 상태값을 초기화
        glClearColor(0.8, 0.8, 0.6, 1.0)

    def resizeGL(self, w: int, h: int):
        # 카메라의 투영 특성을 여기서 설정
        glMatrixMode(GL_PROJECTION)
        glLoadIdentity()
        # 새로운 렌즈를 사용한다.
        glOrtho(0, 250, 360, 0, -1, 1)

    def paintGL(self):

        glClear(GL_COLOR_BUFFER_BIT)
        glMatrixMode(GL_MODELVIEW)
        glLoadIdentity()
        print(selected)
        glBegin(PRIMITIVE_VALUES[selected])
        for i in range(len(points)):
            glVertex2fv( points[i])
        glEnd()

class Drawer(QWidget):
    def __init__(self, parent=None):
        QWidget.__init__(self, parent)
        self.parent = parent
        self.painter = QPainter()

    def paintEvent(self, event):
        global points

        self.painter.begin(self)

        self.painter.setPen( QPen(Qt.red, 6))
        for i in range( len(points) ):
            self.painter.drawPoint( points[i][0], points[i][1] )

        self.painter.end()

    def mousePressEvent(self, event):
        points.append( [event.x(), event.y()] )
        self.update()
        self.parent.opengl.update()

def main():
    app = QApplication(sys.argv)
    win = MyWindow('My Primitive Test App')
    win.setFixedSize(800, 400)
    win.show()
    sys.exit(app.exec_())

if __name__ == '__main__':
    main()