mirror of
https://git.mirrors.martin98.com/https://github.com/Ultimaker/Cura
synced 2025-04-21 21:29:41 +08:00
1f403f815d
**The diagnosis:** The issue arrises from the fact that while the original object is subscribed to be validated, that subscription is only created by setting it as a settings override object. A clone of that object, while still a settings-override object, never gets subscribed because it's not explicitly set with the tool. **The solution (?):** I moved all the validation stuff over to SettingOverrideDecorator.py, and use its onSettingChanged() function to trigger the validation. Unfortunately, I can't use the timer because of some limitation with QTTimer and threads. So it's _a bit laggy in some places, and I'd be open to tips about how to fix this. It does work reliably though.
216 lines
9.3 KiB
Python
216 lines
9.3 KiB
Python
# Copyright (c) 2016 Ultimaker B.V.
|
|
# Uranium is released under the terms of the LGPLv3 or higher.
|
|
|
|
from UM.Tool import Tool
|
|
from UM.Scene.Selection import Selection
|
|
from UM.Scene.Iterator.DepthFirstIterator import DepthFirstIterator
|
|
from UM.Application import Application
|
|
from UM.Preferences import Preferences
|
|
from cura.Settings.SettingOverrideDecorator import SettingOverrideDecorator
|
|
from cura.Settings.ExtruderManager import ExtruderManager
|
|
from UM.Settings.SettingInstance import SettingInstance
|
|
from UM.Event import Event
|
|
from UM.Settings.Validator import ValidatorState
|
|
from UM.Logger import Logger
|
|
|
|
from PyQt5.QtCore import QTimer
|
|
|
|
## This tool allows the user to add & change settings per node in the scene.
|
|
# The settings per object are kept in a ContainerStack, which is linked to a node by decorator.
|
|
class PerObjectSettingsTool(Tool):
|
|
def __init__(self):
|
|
super().__init__()
|
|
self._model = None
|
|
|
|
self.setExposedProperties("SelectedObjectId", "ContainerID", "SelectedActiveExtruder", "MeshType")
|
|
|
|
self._advanced_mode = False
|
|
self._multi_extrusion = False
|
|
self._single_model_selected = False
|
|
|
|
Selection.selectionChanged.connect(self.propertyChanged)
|
|
|
|
Preferences.getInstance().preferenceChanged.connect(self._onPreferenceChanged)
|
|
self._onPreferenceChanged("cura/active_mode")
|
|
|
|
Application.getInstance().globalContainerStackChanged.connect(self._onGlobalContainerChanged)
|
|
self._onGlobalContainerChanged()
|
|
Selection.selectionChanged.connect(self._updateEnabled)
|
|
|
|
self._scene = Application.getInstance().getController().getScene()
|
|
|
|
self._error_check_timer = QTimer()
|
|
self._error_check_timer.setInterval(250)
|
|
self._error_check_timer.setSingleShot(True)
|
|
self._error_check_timer.timeout.connect(self._updateStacksHaveErrors)
|
|
|
|
def event(self, event):
|
|
super().event(event)
|
|
if event.type == Event.MousePressEvent and self._controller.getToolsEnabled():
|
|
self.operationStopped.emit(self)
|
|
return False
|
|
|
|
def getSelectedObjectId(self):
|
|
selected_object = Selection.getSelectedObject(0)
|
|
selected_object_id = id(selected_object)
|
|
return selected_object_id
|
|
|
|
def getContainerID(self):
|
|
selected_object = Selection.getSelectedObject(0)
|
|
try:
|
|
return selected_object.callDecoration("getStack").getId()
|
|
except AttributeError:
|
|
return ""
|
|
|
|
## Gets the active extruder of the currently selected object.
|
|
#
|
|
# \return The active extruder of the currently selected object.
|
|
def getSelectedActiveExtruder(self):
|
|
selected_object = Selection.getSelectedObject(0)
|
|
return selected_object.callDecoration("getActiveExtruder")
|
|
|
|
## Changes the active extruder of the currently selected object.
|
|
#
|
|
# \param extruder_stack_id The ID of the extruder to print the currently
|
|
# selected object with.
|
|
def setSelectedActiveExtruder(self, extruder_stack_id):
|
|
selected_object = Selection.getSelectedObject(0)
|
|
stack = selected_object.callDecoration("getStack") #Don't try to get the active extruder since it may be None anyway.
|
|
if not stack:
|
|
selected_object.addDecorator(SettingOverrideDecorator())
|
|
selected_object.callDecoration("setActiveExtruder", extruder_stack_id)
|
|
|
|
def setMeshType(self, mesh_type):
|
|
selected_object = Selection.getSelectedObject(0)
|
|
stack = selected_object.callDecoration("getStack") #Don't try to get the active extruder since it may be None anyway.
|
|
if not stack:
|
|
selected_object.addDecorator(SettingOverrideDecorator())
|
|
stack = selected_object.callDecoration("getStack")
|
|
|
|
settings = stack.getTop()
|
|
for property_key in ["infill_mesh", "cutting_mesh", "support_mesh", "anti_overhang_mesh"]:
|
|
if property_key != mesh_type:
|
|
if settings.getInstance(property_key):
|
|
settings.removeInstance(property_key)
|
|
else:
|
|
if not (settings.getInstance(property_key) and settings.getProperty(property_key, "value")):
|
|
definition = stack.getSettingDefinition(property_key)
|
|
new_instance = SettingInstance(definition, settings)
|
|
new_instance.setProperty("value", True)
|
|
new_instance.resetState() # Ensure that the state is not seen as a user state.
|
|
settings.addInstance(new_instance)
|
|
|
|
def getMeshType(self):
|
|
selected_object = Selection.getSelectedObject(0)
|
|
stack = selected_object.callDecoration("getStack") #Don't try to get the active extruder since it may be None anyway.
|
|
if not stack:
|
|
return ""
|
|
|
|
settings = stack.getTop()
|
|
for property_key in ["infill_mesh", "cutting_mesh", "support_mesh", "anti_overhang_mesh"]:
|
|
if settings.getInstance(property_key) and settings.getProperty(property_key, "value"):
|
|
return property_key
|
|
|
|
return ""
|
|
|
|
def _onPreferenceChanged(self, preference):
|
|
if preference == "cura/active_mode":
|
|
self._advanced_mode = Preferences.getInstance().getValue(preference) == 1
|
|
self._updateEnabled()
|
|
|
|
def _onGlobalContainerChanged(self):
|
|
global_container_stack = Application.getInstance().getGlobalContainerStack()
|
|
if global_container_stack:
|
|
|
|
# used for enabling or disabling per extruder settings per object
|
|
self._multi_extrusion = global_container_stack.getProperty("machine_extruder_count", "value") > 1
|
|
|
|
extruder_stack = ExtruderManager.getInstance().getExtruderStack(0)
|
|
|
|
if extruder_stack:
|
|
root_node = Application.getInstance().getController().getScene().getRoot()
|
|
for node in DepthFirstIterator(root_node):
|
|
new_stack_id = extruder_stack.getId()
|
|
# Get position of old extruder stack for this node
|
|
old_extruder_pos = node.callDecoration("getActiveExtruderPosition")
|
|
if old_extruder_pos is not None:
|
|
# Fetch current (new) extruder stack at position
|
|
new_stack = ExtruderManager.getInstance().getExtruderStack(old_extruder_pos)
|
|
if new_stack:
|
|
new_stack_id = new_stack.getId()
|
|
node.callDecoration("setActiveExtruder", new_stack_id)
|
|
|
|
self._updateEnabled()
|
|
|
|
def _updateEnabled(self):
|
|
selected_objects = Selection.getAllSelectedObjects()
|
|
if len(selected_objects)> 1:
|
|
self._single_model_selected = False
|
|
elif len(selected_objects) == 1 and selected_objects[0].callDecoration("isGroup"):
|
|
self._single_model_selected = False # Group is selected, so tool needs to be disabled
|
|
else:
|
|
self._single_model_selected = True
|
|
Application.getInstance().getController().toolEnabledChanged.emit(self._plugin_id, self._advanced_mode and self._single_model_selected)
|
|
|
|
|
|
def _onPropertyChanged(self, key: str, property_name: str) -> None:
|
|
if property_name == "validationState":
|
|
# self._error_check_timer.start()
|
|
return
|
|
|
|
def _updateStacksHaveErrors(self) -> None:
|
|
return
|
|
# self._checkStacksHaveErrors()
|
|
|
|
|
|
def _checkStacksHaveErrors(self):
|
|
|
|
for node in DepthFirstIterator(self._scene.getRoot()):
|
|
|
|
# valdiate only objects which can be selected because the settings per object
|
|
# can be applied only for them
|
|
if not node.isSelectable():
|
|
continue
|
|
|
|
hasErrors = self._checkStackForErrors(node.callDecoration("getStack"))
|
|
Application.getInstance().getObjectsModel().setStacksHaveErrors(hasErrors)
|
|
|
|
#If any of models has an error then no reason check next objects on the build plate
|
|
if hasErrors:
|
|
break
|
|
|
|
|
|
def _checkStackForErrors(self, stack):
|
|
print("checking for errors")
|
|
if stack is None:
|
|
return False
|
|
|
|
for key in stack.getAllKeys():
|
|
validation_state = stack.getProperty(key, "validationState")
|
|
if validation_state in (ValidatorState.Exception, ValidatorState.MaximumError, ValidatorState.MinimumError):
|
|
Logger.log("w", "Setting Per Object %s is not valid.", key)
|
|
return True
|
|
return False
|
|
|
|
def subscribeForSettingValidation(self, setting_name):
|
|
selected_object = Selection.getSelectedObject(0)
|
|
stack = selected_object.callDecoration("getStack") # Don't try to get the active extruder since it may be None anyway.
|
|
if not stack:
|
|
return ""
|
|
|
|
settings = stack.getTop()
|
|
setting_instance = settings.getInstance(setting_name)
|
|
if setting_instance:
|
|
setting_instance.propertyChanged.connect(self._onPropertyChanged)
|
|
|
|
def unsubscribeForSettingValidation(self, setting_name):
|
|
selected_object = Selection.getSelectedObject(0)
|
|
stack = selected_object.callDecoration("getStack") # Don't try to get the active extruder since it may be None anyway.
|
|
if not stack:
|
|
return ""
|
|
|
|
settings = stack.getTop()
|
|
setting_instance = settings.getInstance(setting_name)
|
|
if setting_instance:
|
|
setting_instance.propertyChanged.disconnect(self._onPropertyChanged)
|