238 lines
6.7 KiB
Python
Executable File
238 lines
6.7 KiB
Python
Executable File
import sys
|
|
import math
|
|
|
|
from OpenGL import GL, GLU
|
|
|
|
from PyQt5.QtCore import pyqtSignal, QPoint, QSize, Qt
|
|
from PyQt5.QtGui import QColor
|
|
from PyQt5.QtWidgets import (QApplication, QHBoxLayout, QOpenGLWidget, QSlider,
|
|
QWidget)
|
|
|
|
|
|
class Window(QWidget):
|
|
def __init__(self):
|
|
super(Window, self).__init__()
|
|
|
|
self.glWidget = GLWidget()
|
|
|
|
self.xSlider = self.createSlider()
|
|
self.ySlider = self.createSlider()
|
|
self.zSlider = self.createSlider()
|
|
|
|
self.xSlider.valueChanged.connect(self.glWidget.setXRotation)
|
|
self.glWidget.xRotationChanged.connect(self.xSlider.setValue)
|
|
self.ySlider.valueChanged.connect(self.glWidget.setYRotation)
|
|
self.glWidget.yRotationChanged.connect(self.ySlider.setValue)
|
|
self.zSlider.valueChanged.connect(self.glWidget.setZRotation)
|
|
self.glWidget.zRotationChanged.connect(self.zSlider.setValue)
|
|
|
|
mainLayout = QHBoxLayout()
|
|
mainLayout.addWidget(self.glWidget)
|
|
mainLayout.addWidget(self.xSlider)
|
|
mainLayout.addWidget(self.ySlider)
|
|
mainLayout.addWidget(self.zSlider)
|
|
self.setLayout(mainLayout)
|
|
|
|
self.xSlider.setValue(15 * 16)
|
|
self.ySlider.setValue(345 * 16)
|
|
self.zSlider.setValue(0 * 16)
|
|
|
|
self.setWindowTitle("Hello GL")
|
|
|
|
def createSlider(self):
|
|
slider = QSlider(Qt.Vertical)
|
|
|
|
slider.setRange(0, 360 * 16)
|
|
slider.setSingleStep(16)
|
|
slider.setPageStep(15 * 16)
|
|
slider.setTickInterval(15 * 16)
|
|
slider.setTickPosition(QSlider.TicksRight)
|
|
|
|
return slider
|
|
|
|
|
|
class GLWidget(QOpenGLWidget):
|
|
xRotationChanged = pyqtSignal(int)
|
|
yRotationChanged = pyqtSignal(int)
|
|
zRotationChanged = pyqtSignal(int)
|
|
|
|
def __init__(self, parent=None):
|
|
super(GLWidget, self).__init__(parent)
|
|
|
|
self.object = 0
|
|
self.xRot = 0
|
|
self.yRot = 0
|
|
self.zRot = 0
|
|
|
|
self.lastPos = QPoint()
|
|
|
|
self.trolltechGreen = QColor.fromCmykF(0.40, 0.0, 1.0, 0.0)
|
|
self.trolltechPurple = QColor.fromCmykF(0.39, 0.39, 0.0, 0.0)
|
|
|
|
def minimumSizeHint(self):
|
|
return QSize(50, 50)
|
|
|
|
def sizeHint(self):
|
|
return QSize(400, 400)
|
|
|
|
def setXRotation(self, angle):
|
|
angle = self.normalizeAngle(angle)
|
|
if angle != self.xRot:
|
|
self.xRot = angle
|
|
self.xRotationChanged.emit(angle)
|
|
self.update()
|
|
|
|
def setYRotation(self, angle):
|
|
angle = self.normalizeAngle(angle)
|
|
if angle != self.yRot:
|
|
self.yRot = angle
|
|
self.yRotationChanged.emit(angle)
|
|
self.update()
|
|
|
|
def setZRotation(self, angle):
|
|
angle = self.normalizeAngle(angle)
|
|
if angle != self.zRot:
|
|
self.zRot = angle
|
|
self.zRotationChanged.emit(angle)
|
|
self.update()
|
|
|
|
def initializeGL(self):
|
|
|
|
self.setClearColor(self.trolltechPurple.darker())
|
|
self.object = self.makeObject()
|
|
GL.glShadeModel(GL.GL_FLAT)
|
|
GL.glEnable(GL.GL_DEPTH_TEST)
|
|
GL.glEnable(GL.GL_CULL_FACE)
|
|
|
|
def paintGL(self):
|
|
GL.glClear(
|
|
GL.GL_COLOR_BUFFER_BIT | GL.GL_DEPTH_BUFFER_BIT)
|
|
GL.glLoadIdentity()
|
|
GL.glTranslated(0.0, 0.0, -10.0)
|
|
GL.glRotated(self.xRot / 16.0, 1.0, 0.0, 0.0)
|
|
GL.glRotated(self.yRot / 16.0, 0.0, 1.0, 0.0)
|
|
GL.glRotated(self.zRot / 16.0, 0.0, 0.0, 1.0)
|
|
GL.glCallList(self.object)
|
|
|
|
def resizeGL(self, width, height):
|
|
side = min(width, height)
|
|
if side < 0:
|
|
return
|
|
|
|
GL.glViewport((width - side) // 2, (height - side) // 2, side,
|
|
side)
|
|
|
|
GL.glMatrixMode(GL.GL_PROJECTION)
|
|
GL.glLoadIdentity()
|
|
GL.glOrtho(-0.5, +0.5, +0.5, -0.5, 4.0, 15.0)
|
|
GL.glMatrixMode(GL.GL_MODELVIEW)
|
|
|
|
def mousePressEvent(self, event):
|
|
self.lastPos = event.pos()
|
|
|
|
def mouseMoveEvent(self, event):
|
|
dx = event.x() - self.lastPos.x()
|
|
dy = event.y() - self.lastPos.y()
|
|
|
|
if event.buttons() & Qt.LeftButton:
|
|
self.setXRotation(self.xRot + 8 * dy)
|
|
self.setYRotation(self.yRot + 8 * dx)
|
|
elif event.buttons() & Qt.RightButton:
|
|
self.setXRotation(self.xRot + 8 * dy)
|
|
self.setZRotation(self.zRot + 8 * dx)
|
|
|
|
self.lastPos = event.pos()
|
|
|
|
def makeObject(self):
|
|
genList = GL.glGenLists(1)
|
|
GL.glNewList(genList, GL.GL_COMPILE)
|
|
|
|
GL.glBegin(GL.GL_QUADS)
|
|
|
|
x1 = +0.06
|
|
y1 = -0.14
|
|
x2 = +0.14
|
|
y2 = -0.06
|
|
x3 = +0.08
|
|
y3 = +0.00
|
|
x4 = +0.30
|
|
y4 = +0.22
|
|
|
|
self.quad(x1, y1, x2, y2, y2, x2, y1, x1)
|
|
self.quad(x3, y3, x4, y4, y4, x4, y3, x3)
|
|
|
|
self.extrude(x1, y1, x2, y2)
|
|
self.extrude(x2, y2, y2, x2)
|
|
self.extrude(y2, x2, y1, x1)
|
|
self.extrude(y1, x1, x1, y1)
|
|
self.extrude(x3, y3, x4, y4)
|
|
self.extrude(x4, y4, y4, x4)
|
|
self.extrude(y4, x4, y3, x3)
|
|
|
|
NumSectors = 200
|
|
|
|
for i in range(NumSectors):
|
|
angle1 = (i * 2 * math.pi) / NumSectors
|
|
x5 = 0.30 * math.sin(angle1)
|
|
y5 = 0.30 * math.cos(angle1)
|
|
x6 = 0.20 * math.sin(angle1)
|
|
y6 = 0.20 * math.cos(angle1)
|
|
|
|
angle2 = ((i + 1) * 2 * math.pi) / NumSectors
|
|
x7 = 0.20 * math.sin(angle2)
|
|
y7 = 0.20 * math.cos(angle2)
|
|
x8 = 0.30 * math.sin(angle2)
|
|
y8 = 0.30 * math.cos(angle2)
|
|
|
|
self.quad(x5, y5, x6, y6, x7, y7, x8, y8)
|
|
|
|
self.extrude(x6, y6, x7, y7)
|
|
self.extrude(x8, y8, x5, y5)
|
|
|
|
GL.glEnd()
|
|
GL.glEndList()
|
|
|
|
return genList
|
|
|
|
def quad(self, x1, y1, x2, y2, x3, y3, x4, y4):
|
|
self.setColor(self.trolltechGreen)
|
|
|
|
GL.glVertex3d(x1, y1, -0.05)
|
|
GL.glVertex3d(x2, y2, -0.05)
|
|
GL.glVertex3d(x3, y3, -0.05)
|
|
GL.glVertex3d(x4, y4, -0.05)
|
|
|
|
GL.glVertex3d(x4, y4, +0.05)
|
|
GL.glVertex3d(x3, y3, +0.05)
|
|
GL.glVertex3d(x2, y2, +0.05)
|
|
GL.glVertex3d(x1, y1, +0.05)
|
|
|
|
def extrude(self, x1, y1, x2, y2):
|
|
self.setColor(self.trolltechGreen.darker(250 + int(100 * x1)))
|
|
|
|
GL.glVertex3d(x1, y1, +0.05)
|
|
GL.glVertex3d(x2, y2, +0.05)
|
|
GL.glVertex3d(x2, y2, -0.05)
|
|
GL.glVertex3d(x1, y1, -0.05)
|
|
|
|
def normalizeAngle(self, angle):
|
|
while angle < 0:
|
|
angle += 360 * 16
|
|
while angle > 360 * 16:
|
|
angle -= 360 * 16
|
|
return angle
|
|
|
|
def setClearColor(self, c):
|
|
GL.glClearColor(c.redF(), c.greenF(), c.blueF(), c.alphaF())
|
|
|
|
def setColor(self, c):
|
|
GL.glColor4f(c.redF(), c.greenF(), c.blueF(), c.alphaF())
|
|
|
|
|
|
if __name__ == '__main__':
|
|
|
|
app = QApplication(sys.argv)
|
|
window = Window()
|
|
window.show()
|
|
sys.exit(app.exec_())
|