def __init__(self, parent, main_window, scene):
super().__init__(parent)
self.parent = parent
self.main_window = main_window
self.ui = Ui_SceneViewWidget()
self.ui.setupUi(self)
self.gl_widget = GLWidget(self)
self.ui.gl_layout.addWidget(self.gl_widget)
self.scene = scene
self.initialize()
def __init__(self, parent, main_window, scene):
super().__init__(parent)
self.parent = parent
self.main_window = main_window
self.ui = Ui_SceneViewWidget()
self.ui.setupUi(self)
self.gl_widget = GLWidget(self)
self.ui.gl_layout.addWidget(self.gl_widget)
self.scene = scene
self.initialize()
class SceneViewWidget(QtWidgets.QWidget):
def __init__(self, parent, main_window, scene):
super().__init__(parent)
self.parent = parent
self.main_window = main_window
self.ui = Ui_SceneViewWidget()
self.ui.setupUi(self)
self.gl_widget = GLWidget(self)
self.ui.gl_layout.addWidget(self.gl_widget)
self.scene = scene
self.initialize()
def initialize(self):
self.ui.tx.valueChanged.connect(self.update_transform)
self.ui.ty.valueChanged.connect(self.update_transform)
self.ui.tz.valueChanged.connect(self.update_transform)
self.ui.rx.valueChanged.connect(self.update_transform)
self.ui.ry.valueChanged.connect(self.update_transform)
self.ui.rz.valueChanged.connect(self.update_transform)
self.ui.lposx.valueChanged.connect(self.update_light)
self.ui.lposy.valueChanged.connect(self.update_light)
self.ui.lposz.valueChanged.connect(self.update_light)
self.ui.intensity.valueChanged.connect(self.update_light)
self.ui.meshes_list.itemClicked.connect(self.update_transform_fields)
self.ui.render_btn.clicked.connect(self.render)
self.ui.delete_btn.clicked.connect(self.delete_mesh)
self.update_light()
def update_transform_fields(self):
mesh = self.get_selected_mesh()
if not mesh:
return
self.ui.tx.setValue(mesh.transform.tx)
self.ui.ty.setValue(mesh.transform.ty)
self.ui.tz.setValue(mesh.transform.tz)
self.ui.rx.setValue(mesh.transform.rx)
self.ui.ry.setValue(mesh.transform.ry)
self.ui.rz.setValue(mesh.transform.rz)
def update_transform(self):
mesh = self.get_selected_mesh()
if not mesh:
return
mesh.transform.tx = self.ui.tx.value()
mesh.transform.ty = self.ui.ty.value()
mesh.transform.tz = self.ui.tz.value()
mesh.transform.rx = self.ui.rx.value()
mesh.transform.ry = self.ui.ry.value()
mesh.transform.rz = self.ui.rz.value()
self.gl_widget.update()
def update_list(self):
self.ui.meshes_list.clear()
for mesh_name in self.scene.meshes.keys():
self.ui.meshes_list.addItem(mesh_name)
def get_selected_mesh(self):
selection = self.ui.meshes_list.selectedItems()
if not selection:
return None
return self.scene.meshes[selection[0].text()]
def delete_mesh(self):
selection = self.ui.meshes_list.selectedItems()
if not selection:
return
del self.scene.meshes[selection[0].text()]
self.refresh()
def update_light(self):
self.gl_widget.light_pos = (self.ui.lposx.value(),
self.ui.lposy.value(),
self.ui.lposz.value())
self.gl_widget.light_intensity = self.ui.intensity.value()
self.gl_widget.update()
def refresh(self):
self.gl_widget.schedule_buffer_update(self.scene.meshes.values())
self.gl_widget.update_buffers()
self.update_light()
self.update_list()
def render(self):
camera_pos = self.gl_widget.camera_pos
intensity = self.gl_widget.light_intensity
filename = self.ui.filename.text() or 'output'
render(filename, (self.gl_widget.width, self.gl_widget.height),
self.scene.meshes, camera_pos, intensity)
self.main_window.load_png(filename + '.png')
class SceneViewWidget(QtWidgets.QWidget):
def __init__(self, parent, main_window, scene):
super().__init__(parent)
self.parent = parent
self.main_window = main_window
self.ui = Ui_SceneViewWidget()
self.ui.setupUi(self)
self.gl_widget = GLWidget(self)
self.ui.gl_layout.addWidget(self.gl_widget)
self.scene = scene
self.initialize()
def initialize(self):
self.ui.tx.valueChanged.connect(self.update_transform)
self.ui.ty.valueChanged.connect(self.update_transform)
self.ui.tz.valueChanged.connect(self.update_transform)
self.ui.rx.valueChanged.connect(self.update_transform)
self.ui.ry.valueChanged.connect(self.update_transform)
self.ui.rz.valueChanged.connect(self.update_transform)
self.ui.lposx.valueChanged.connect(self.update_light)
self.ui.lposy.valueChanged.connect(self.update_light)
self.ui.lposz.valueChanged.connect(self.update_light)
self.ui.intensity.valueChanged.connect(self.update_light)
self.ui.meshes_list.itemClicked.connect(self.update_transform_fields)
self.ui.render_btn.clicked.connect(self.render)
self.ui.delete_btn.clicked.connect(self.delete_mesh)
self.update_light()
def update_transform_fields(self):
mesh = self.get_selected_mesh()
if not mesh:
return
self.ui.tx.setValue(mesh.transform.tx)
self.ui.ty.setValue(mesh.transform.ty)
self.ui.tz.setValue(mesh.transform.tz)
self.ui.rx.setValue(mesh.transform.rx)
self.ui.ry.setValue(mesh.transform.ry)
self.ui.rz.setValue(mesh.transform.rz)
def update_transform(self):
mesh = self.get_selected_mesh()
if not mesh:
return
mesh.transform.tx = self.ui.tx.value()
mesh.transform.ty = self.ui.ty.value()
mesh.transform.tz = self.ui.tz.value()
mesh.transform.rx = self.ui.rx.value()
mesh.transform.ry = self.ui.ry.value()
mesh.transform.rz = self.ui.rz.value()
self.gl_widget.update()
def update_list(self):
self.ui.meshes_list.clear()
for mesh_name in self.scene.meshes.keys():
self.ui.meshes_list.addItem(mesh_name)
def get_selected_mesh(self):
selection = self.ui.meshes_list.selectedItems()
if not selection:
return None
return self.scene.meshes[selection[0].text()]
def delete_mesh(self):
selection = self.ui.meshes_list.selectedItems()
if not selection:
return
del self.scene.meshes[selection[0].text()]
self.refresh()
def update_light(self):
self.gl_widget.light_pos = (self.ui.lposx.value(), self.ui.lposy.value(), self.ui.lposz.value())
self.gl_widget.light_intensity = self.ui.intensity.value()
self.gl_widget.update()
def refresh(self):
self.gl_widget.schedule_buffer_update(self.scene.meshes.values())
self.gl_widget.update_buffers()
self.update_light()
self.update_list()
def render(self):
camera_pos = self.gl_widget.camera_pos
intensity = self.gl_widget.light_intensity
filename = self.ui.filename.text() or "output"
render(filename, (self.gl_widget.width, self.gl_widget.height), self.scene.meshes, camera_pos, intensity)
self.main_window.load_png(filename + ".png")
