123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345 |
- import os.path
- from UM.View.View import View
- from UM.Scene.Iterator.DepthFirstIterator import DepthFirstIterator
- from UM.Scene.Selection import Selection
- from UM.Resources import Resources
- from PyQt6.QtGui import QOpenGLContext, QDesktopServices, QImage
- from PyQt6.QtCore import QSize, QUrl
- import numpy as np
- import time
- from UM.Application import Application
- from UM.Logger import Logger
- from UM.Message import Message
- from UM.Math.Color import Color
- from UM.PluginRegistry import PluginRegistry
- from UM.Platform import Platform
- from UM.Event import Event
- from UM.View.RenderBatch import RenderBatch
- from UM.View.GL.OpenGL import OpenGL
- from UM.i18n import i18nCatalog
- from cura.Settings.ExtruderManager import ExtruderManager
- from cura import XRayPass
- import math
- catalog = i18nCatalog("cura")
- class SolidView(View):
- """Standard view for mesh models."""
- _show_xray_warning_preference = "view/show_xray_warning"
- def __init__(self):
- super().__init__()
- application = Application.getInstance()
- application.getPreferences().addPreference("view/show_overhang", True)
- application.globalContainerStackChanged.connect(self._onGlobalContainerChanged)
- self._enabled_shader = None
- self._disabled_shader = None
- self._non_printing_shader = None
- self._support_mesh_shader = None
- self._xray_shader = None
- self._xray_pass = None
- self._xray_composite_shader = None
- self._composite_pass = None
- self._extruders_model = None
- self._theme = None
- self._support_angle = self._retrieveSupportAngle()
- self._lowest_printable_height = self._retrieveLowestPrintHeight()
- self._global_stack = None
- self._old_composite_shader = None
- self._old_layer_bindings = None
- self._next_xray_checking_time = time.time()
- self._xray_checking_update_time = 30.0
- self._xray_warning_cooldown = 60 * 10
- self._xray_warning_message = Message(
- catalog.i18nc("@info:status", "The highlighted areas indicate either missing or extraneous surfaces. Fix your model and open it again into Cura."),
- lifetime = 60 * 5,
- title = catalog.i18nc("@info:title", "Model Errors"),
- option_text = catalog.i18nc("@info:option_text", "Do not show this message again"),
- option_state = False,
- message_type=Message.MessageType.WARNING
- )
- self._xray_warning_message.optionToggled.connect(self._onDontAskMeAgain)
- application.getPreferences().addPreference(self._show_xray_warning_preference, True)
- self._xray_warning_message.addAction("manifold", catalog.i18nc("@action:button", "Learn more"), "[no_icon]", "[no_description]",
- button_style = Message.ActionButtonStyle.LINK,
- button_align = Message.ActionButtonAlignment.ALIGN_LEFT)
- self._xray_warning_message.actionTriggered.connect(self._onNonManifoldLearnMoreClicked)
- application.engineCreatedSignal.connect(self._onGlobalContainerChanged)
- def _onDontAskMeAgain(self, checked: bool) -> None:
- Application.getInstance().getPreferences().setValue(self._show_xray_warning_preference, not checked)
- def _onNonManifoldLearnMoreClicked(self, action, message) -> None:
- QDesktopServices.openUrl(QUrl("https://support.ultimaker.com/hc/en-us/articles/360014055959"))
- def _onGlobalContainerChanged(self) -> None:
- if self._global_stack:
- try:
- self._global_stack.propertyChanged.disconnect(self._onPropertyChanged)
- except TypeError:
- pass
- for extruder_stack in ExtruderManager.getInstance().getActiveExtruderStacks():
- extruder_stack.propertyChanged.disconnect(self._onPropertyChanged)
- self._global_stack = Application.getInstance().getGlobalContainerStack()
- if self._global_stack:
- self._global_stack.propertyChanged.connect(self._onPropertyChanged)
- for extruder_stack in ExtruderManager.getInstance().getActiveExtruderStacks():
- extruder_stack.propertyChanged.connect(self._onPropertyChanged)
-
- self._support_angle = self._retrieveSupportAngle()
- self._lowest_printable_height = self._retrieveLowestPrintHeight()
- def _onPropertyChanged(self, key: str, property_name: str) -> None:
- if property_name != "value":
- return
-
- if key == "support_angle":
- self._support_angle = self._retrieveSupportAngle()
- elif key == "layer_height_0" or key == "slicing_tolerance":
- self._lowest_printable_height = self._retrieveLowestPrintHeight()
- def _retrieveSupportAngle(self) -> float:
- global_container_stack = Application.getInstance().getGlobalContainerStack()
- if global_container_stack:
- support_extruder_nr = int(global_container_stack.getExtruderPositionValueWithDefault("support_extruder_nr"))
- try:
- support_angle_stack = global_container_stack.extruderList[support_extruder_nr]
- except IndexError:
- pass
- else:
- angle = support_angle_stack.getProperty("support_angle", "value")
- if angle is not None:
- return angle
- return 90.0
- def _retrieveLowestPrintHeight(self) -> float:
- min_height = 0.0
- for extruder in Application.getInstance().getExtruderManager().getActiveExtruderStacks():
- init_layer_height = extruder.getProperty("layer_height_0", "value")
- tolerance_setting = extruder.getProperty("slicing_tolerance", "value")
- if tolerance_setting == "middle":
- init_layer_height /= 2.0
- min_height = max(min_height, init_layer_height)
- return min_height
- def _checkSetup(self):
- if not self._extruders_model:
- self._extruders_model = Application.getInstance().getExtrudersModel()
- if not self._theme:
- self._theme = Application.getInstance().getTheme()
- if not self._enabled_shader:
- self._enabled_shader = OpenGL.getInstance().createShaderProgram(Resources.getPath(Resources.Shaders, "overhang.shader"))
- self._enabled_shader.setUniformValue("u_overhangColor", Color(*self._theme.getColor("model_overhang").getRgb()))
- self._enabled_shader.setUniformValue("u_renderError", 0.0)
- if not self._disabled_shader:
- self._disabled_shader = OpenGL.getInstance().createShaderProgram(Resources.getPath(Resources.Shaders, "striped.shader"))
- self._disabled_shader.setUniformValue("u_diffuseColor1", Color(*self._theme.getColor("model_unslicable").getRgb()))
- self._disabled_shader.setUniformValue("u_diffuseColor2", Color(*self._theme.getColor("model_unslicable_alt").getRgb()))
- self._disabled_shader.setUniformValue("u_width", 50.0)
- if not self._non_printing_shader:
- self._non_printing_shader = OpenGL.getInstance().createShaderProgram(Resources.getPath(Resources.Shaders, "transparent_object.shader"))
- self._non_printing_shader.setUniformValue("u_diffuseColor", Color(*self._theme.getColor("model_non_printing").getRgb()))
- self._non_printing_shader.setUniformValue("u_opacity", 0.6)
- if not self._support_mesh_shader:
- self._support_mesh_shader = OpenGL.getInstance().createShaderProgram(Resources.getPath(Resources.Shaders, "striped.shader"))
- self._support_mesh_shader.setUniformValue("u_vertical_stripes", True)
- self._support_mesh_shader.setUniformValue("u_width", 5.0)
- if not Application.getInstance().getPreferences().getValue(self._show_xray_warning_preference):
- self._xray_shader = None
- self._xray_composite_shader = None
- if self._composite_pass and 'xray' in self._composite_pass.getLayerBindings():
- self._composite_pass.setLayerBindings(self._old_layer_bindings)
- self._composite_pass.setCompositeShader(self._old_composite_shader)
- self._old_layer_bindings = None
- self._old_composite_shader = None
- self._enabled_shader.setUniformValue("u_renderError", 0.0)
- self._xray_warning_message.hide()
- else:
- if not self._xray_shader:
- self._xray_shader = OpenGL.getInstance().createShaderProgram(Resources.getPath(Resources.Shaders, "xray.shader"))
- if not self._xray_composite_shader:
- self._xray_composite_shader = OpenGL.getInstance().createShaderProgram(Resources.getPath(Resources.Shaders, "xray_composite.shader"))
- theme = Application.getInstance().getTheme()
- self._xray_composite_shader.setUniformValue("u_background_color", Color(*theme.getColor("viewport_background").getRgb()))
- self._xray_composite_shader.setUniformValue("u_outline_color", Color(*theme.getColor("model_selection_outline").getRgb()))
- self._xray_composite_shader.setUniformValue("u_flat_error_color_mix", 0.)
- renderer = self.getRenderer()
- if not self._composite_pass or not 'xray' in self._composite_pass.getLayerBindings():
-
-
- self._xray_pass = XRayPass.XRayPass(1, 1)
- self._enabled_shader.setUniformValue("u_renderError", 1.0)
- renderer.addRenderPass(self._xray_pass)
- if not self._composite_pass:
- self._composite_pass = self.getRenderer().getRenderPass("composite")
- self._old_layer_bindings = self._composite_pass.getLayerBindings()
- self._composite_pass.setLayerBindings(["default", "selection", "xray"])
- self._old_composite_shader = self._composite_pass.getCompositeShader()
- self._composite_pass.setCompositeShader(self._xray_composite_shader)
- def beginRendering(self):
- scene = self.getController().getScene()
- renderer = self.getRenderer()
- self._checkSetup()
- global_container_stack = Application.getInstance().getGlobalContainerStack()
- if global_container_stack:
- if Application.getInstance().getPreferences().getValue("view/show_overhang"):
-
- if self._support_angle >= 0 and self._support_angle <= 90:
- self._enabled_shader.setUniformValue("u_overhangAngle", math.cos(math.radians(90 - self._support_angle)))
- else:
- self._enabled_shader.setUniformValue("u_overhangAngle", math.cos(math.radians(0)))
- else:
- self._enabled_shader.setUniformValue("u_overhangAngle", math.cos(math.radians(0)))
- self._enabled_shader.setUniformValue("u_lowestPrintableHeight", self._lowest_printable_height)
- disabled_batch = renderer.createRenderBatch(shader = self._disabled_shader)
- normal_object_batch = renderer.createRenderBatch(shader = self._enabled_shader)
- renderer.addRenderBatch(disabled_batch)
- renderer.addRenderBatch(normal_object_batch)
- for node in DepthFirstIterator(scene.getRoot()):
- if node.render(renderer):
- continue
- if node.getMeshData() and node.isVisible():
- uniforms = {}
- shade_factor = 1.0
- per_mesh_stack = node.callDecoration("getStack")
- extruder_index = node.callDecoration("getActiveExtruderPosition")
- if extruder_index is None:
- extruder_index = "0"
- extruder_index = int(extruder_index)
- try:
- material_color = self._extruders_model.getItem(extruder_index)["color"]
- except KeyError:
- material_color = self._extruders_model.defaultColors[0]
- if extruder_index != ExtruderManager.getInstance().activeExtruderIndex:
-
- shade_factor = 0.6
- try:
-
-
- uniforms["diffuse_color"] = [
- shade_factor * int(material_color[1:3], 16) / 255,
- shade_factor * int(material_color[3:5], 16) / 255,
- shade_factor * int(material_color[5:7], 16) / 255,
- 1.0
- ]
-
-
- uniforms["hover_face"] = -1
- except ValueError:
- pass
- if node.callDecoration("isNonPrintingMesh"):
- if per_mesh_stack and (node.callDecoration("isInfillMesh") or node.callDecoration("isCuttingMesh")):
- renderer.queueNode(node, shader = self._non_printing_shader, uniforms = uniforms, transparent = True)
- else:
- renderer.queueNode(node, shader = self._non_printing_shader, transparent = True)
- elif getattr(node, "_outside_buildarea", False):
- disabled_batch.addItem(node.getWorldTransformation(copy = False), node.getMeshData(), normal_transformation = node.getCachedNormalMatrix())
- elif per_mesh_stack and node.callDecoration("isSupportMesh"):
-
- shade_factor = 0.6
- uniforms["diffuse_color_2"] = [
- uniforms["diffuse_color"][0] * shade_factor,
- uniforms["diffuse_color"][1] * shade_factor,
- uniforms["diffuse_color"][2] * shade_factor,
- 1.0
- ]
- renderer.queueNode(node, shader = self._support_mesh_shader, uniforms = uniforms)
- else:
- normal_object_batch.addItem(node.getWorldTransformation(copy=False), node.getMeshData(), uniforms=uniforms, normal_transformation = node.getCachedNormalMatrix())
- if node.callDecoration("isGroup") and Selection.isSelected(node):
- renderer.queueNode(scene.getRoot(), mesh = node.getBoundingBoxMesh(), mode = RenderBatch.RenderMode.LineLoop)
- def endRendering(self):
-
- if time.time() > self._next_xray_checking_time\
- and Application.getInstance().getPreferences().getValue(self._show_xray_warning_preference):
- self._next_xray_checking_time = time.time() + self._xray_checking_update_time
- xray_img = self._xray_pass.getOutput()
- xray_img = xray_img.convertToFormat(QImage.Format.Format_RGB888)
-
-
-
-
-
- class QImageArrayView:
- """
- Class that ducktypes to be a Numpy ndarray.
- """
- def __init__(self, qimage):
- bits_pointer = qimage.bits()
- if bits_pointer is None:
- self.__array_interface__ = {
- "shape": (0, 0),
- "typestr": "|u4",
- "data": (0, False),
- "strides": (1, 3),
- "version": 3
- }
- else:
- self.__array_interface__ = {
- "shape": (qimage.height(), qimage.width()),
- "typestr": "|u4",
- "data": (int(bits_pointer), False),
- "strides": (qimage.bytesPerLine(), 3),
- "version": 3
- }
- array = np.asarray(QImageArrayView(xray_img)).view(np.dtype({
- "r": (np.uint8, 0, "red"),
- "g": (np.uint8, 1, "green"),
- "b": (np.uint8, 2, "blue"),
- "a": (np.uint8, 3, "alpha")
- }), np.recarray)
- if np.any(np.mod(array.r, 2)):
- self._next_xray_checking_time = time.time() + self._xray_warning_cooldown
- self._xray_warning_message.show()
- Logger.log("i", "X-Ray overlay found non-manifold pixels.")
- def event(self, event):
- if event.type == Event.ViewDeactivateEvent:
- if self._composite_pass and 'xray' in self._composite_pass.getLayerBindings():
- self.getRenderer().removeRenderPass(self._xray_pass)
- self._composite_pass.setLayerBindings(self._old_layer_bindings)
- self._composite_pass.setCompositeShader(self._old_composite_shader)
- self._xray_warning_message.hide()
|