109 lines
3.1 KiB
Python
109 lines
3.1 KiB
Python
from PySide6.QtOpenGLWidgets import QOpenGLWidget
|
|
from PySide6.QtCore import QSize, Qt, QPoint
|
|
from OpenGL.GL import *
|
|
from OpenGL.GLU import *
|
|
from stl import mesh
|
|
|
|
|
|
class OpenGLWidget(QOpenGLWidget):
|
|
def __init__(self, parent=None):
|
|
super().__init__(parent)
|
|
self.stl_file = "out.stl" # Replace with your STL file path
|
|
self.lastPos = QPoint()
|
|
self.xRot = 0
|
|
self.yRot = 0
|
|
self.zoom = -10.0
|
|
|
|
def load_stl(self, filename):
|
|
try:
|
|
stl_mesh = mesh.Mesh.from_file(filename)
|
|
return stl_mesh.vectors
|
|
except FileNotFoundError:
|
|
print(f"Error: File {filename} not found.")
|
|
except Exception as e:
|
|
print(f"Error loading {filename}: {e}")
|
|
return []
|
|
|
|
def initializeGL(self):
|
|
glClearColor(0, 0, 0, 1)
|
|
glEnable(GL_DEPTH_TEST)
|
|
|
|
def resizeGL(self, w, h):
|
|
glViewport(0, 0, w, h)
|
|
glMatrixMode(GL_PROJECTION)
|
|
glLoadIdentity()
|
|
gluPerspective(45, w/h, 0.1, 100.0)
|
|
glMatrixMode(GL_MODELVIEW)
|
|
|
|
def paintGL(self):
|
|
glClear(GL_COLOR_BUFFER_BIT | GL_DEPTH_BUFFER_BIT)
|
|
glLoadIdentity()
|
|
glTranslatef(0.0, 0.0, self.zoom)
|
|
glRotatef(self.xRot, 1.0, 0.0, 0.0)
|
|
glRotatef(self.yRot, 0.0, 1.0, 0.0)
|
|
glColor3f(1.0, 1.0, 1.0)
|
|
|
|
mesh_data = self.load_stl(self.stl_file)
|
|
if mesh_data.any():
|
|
self.draw_stl(mesh_data)
|
|
|
|
def draw_stl(self, vertices):
|
|
glEnable(GL_LIGHTING)
|
|
glEnable(GL_LIGHT0)
|
|
glEnable(GL_DEPTH_TEST)
|
|
glEnable(GL_COLOR_MATERIAL)
|
|
glColorMaterial(GL_FRONT_AND_BACK, GL_AMBIENT_AND_DIFFUSE)
|
|
|
|
glLightfv(GL_LIGHT0, GL_POSITION, (0, 1, 1, 0))
|
|
glLightfv(GL_LIGHT0, GL_DIFFUSE, (0.6, 0.6, 0.6, 0.6))
|
|
|
|
glBegin(GL_TRIANGLES)
|
|
for triangle in vertices:
|
|
for vertex in triangle:
|
|
glVertex3fv(vertex)
|
|
glEnd()
|
|
|
|
# Draw outer vertices as points
|
|
glDisable(GL_LIGHTING)
|
|
glColor3f(1.0, 0.0, 0.0) # Set color to red
|
|
glPointSize(5.0) # Set point size
|
|
glBegin(GL_POINTS)
|
|
for triangle in vertices:
|
|
for vertex in triangle:
|
|
glVertex3fv(vertex)
|
|
glEnd()
|
|
|
|
# Draw edges
|
|
glColor3f(0.0, 0.0, 1.0) # Set color to blue
|
|
glLineWidth(2.0) # Set line width
|
|
glBegin(GL_LINES)
|
|
for triangle in vertices:
|
|
glVertex3fv(triangle[0])
|
|
glVertex3fv(triangle[1])
|
|
|
|
glVertex3fv(triangle[1])
|
|
glVertex3fv(triangle[2])
|
|
|
|
glVertex3fv(triangle[2])
|
|
glVertex3fv(triangle[0])
|
|
glEnd()
|
|
|
|
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.xRot += 0.5 * dy
|
|
self.yRot += 0.5 * dx
|
|
self.update()
|
|
|
|
self.lastPos = event.pos()
|
|
|
|
def wheelEvent(self, event):
|
|
delta = event.angleDelta().y()
|
|
self.zoom += delta / 120
|
|
self.update()
|