mirror of
https://git.mirrors.martin98.com/https://github.com/Ultimaker/Cura
synced 2025-04-28 23:04:34 +08:00
42058a2e8f
This reverts commit 85f10e7c4f9bc48afd2fe43903e565b128e03189. The second time you start Cura, it won't slice due to the number of extruders being wrong.
260 lines
12 KiB
Python
260 lines
12 KiB
Python
# Copyright (c) 2017 Ultimaker B.V.
|
|
# Cura is released under the terms of the LGPLv3 or higher.
|
|
from typing import List, Optional, TYPE_CHECKING
|
|
|
|
from PyQt5.QtCore import QObject, QTimer, pyqtProperty, pyqtSignal
|
|
from UM.FlameProfiler import pyqtSlot
|
|
from UM.Application import Application
|
|
from UM.Logger import Logger
|
|
|
|
|
|
## The settingInheritance manager is responsible for checking each setting in order to see if one of the "deeper"
|
|
# containers has a setting function and the topmost one with a value has a value. We need to have this check
|
|
# because some profiles tend to have 'hardcoded' values that break our inheritance. A good example of that are the
|
|
# speed settings. If all the children of print_speed have a single value override, changing the speed won't
|
|
# actually do anything, as only the 'leaf' settings are used by the engine.
|
|
from UM.Settings.ContainerStack import ContainerStack
|
|
from UM.Settings.Interfaces import ContainerInterface
|
|
from UM.Settings.SettingFunction import SettingFunction
|
|
from UM.Settings.SettingInstance import InstanceState
|
|
|
|
from cura.Settings.ExtruderManager import ExtruderManager
|
|
|
|
if TYPE_CHECKING:
|
|
from cura.Settings.ExtruderStack import ExtruderStack
|
|
from UM.Settings.SettingDefinition import SettingDefinition
|
|
|
|
|
|
class SettingInheritanceManager(QObject):
|
|
def __init__(self, parent = None) -> None:
|
|
super().__init__(parent)
|
|
Application.getInstance().globalContainerStackChanged.connect(self._onGlobalContainerChanged)
|
|
self._global_container_stack = None # type: Optional[ContainerStack]
|
|
self._settings_with_inheritance_warning = [] # type: List[str]
|
|
self._active_container_stack = None # type: Optional[ExtruderStack]
|
|
self._onGlobalContainerChanged()
|
|
|
|
ExtruderManager.getInstance().activeExtruderChanged.connect(self._onActiveExtruderChanged)
|
|
self._onActiveExtruderChanged()
|
|
|
|
self._update_timer = QTimer()
|
|
self._update_timer.setInterval(500)
|
|
self._update_timer.setSingleShot(True)
|
|
self._update_timer.timeout.connect(self._update)
|
|
|
|
settingsWithIntheritanceChanged = pyqtSignal()
|
|
|
|
## Get the keys of all children settings with an override.
|
|
@pyqtSlot(str, result = "QStringList")
|
|
def getChildrenKeysWithOverride(self, key: str) -> List[str]:
|
|
if self._global_container_stack is None:
|
|
return []
|
|
definitions = self._global_container_stack.definition.findDefinitions(key=key)
|
|
if not definitions:
|
|
Logger.log("w", "Could not find definition for key [%s]", key)
|
|
return []
|
|
result = []
|
|
for key in definitions[0].getAllKeys():
|
|
if key in self._settings_with_inheritance_warning:
|
|
result.append(key)
|
|
return result
|
|
|
|
@pyqtSlot(str, str, result = "QStringList")
|
|
def getOverridesForExtruder(self, key: str, extruder_index: str) -> List[str]:
|
|
if self._global_container_stack is None:
|
|
return []
|
|
|
|
result = [] # type: List[str]
|
|
extruder_stack = ExtruderManager.getInstance().getExtruderStack(extruder_index)
|
|
if not extruder_stack:
|
|
Logger.log("w", "Unable to find extruder for current machine with index %s", extruder_index)
|
|
return result
|
|
|
|
definitions = self._global_container_stack.definition.findDefinitions(key = key)
|
|
if not definitions:
|
|
Logger.log("w", "Could not find definition for key [%s] (2)", key)
|
|
return result
|
|
|
|
for key in definitions[0].getAllKeys():
|
|
if self._settingIsOverwritingInheritance(key, extruder_stack):
|
|
result.append(key)
|
|
|
|
return result
|
|
|
|
@pyqtSlot(str)
|
|
def manualRemoveOverride(self, key: str) -> None:
|
|
if key in self._settings_with_inheritance_warning:
|
|
self._settings_with_inheritance_warning.remove(key)
|
|
self.settingsWithIntheritanceChanged.emit()
|
|
|
|
@pyqtSlot()
|
|
def forceUpdate(self) -> None:
|
|
self._update()
|
|
|
|
def _onActiveExtruderChanged(self) -> None:
|
|
new_active_stack = ExtruderManager.getInstance().getActiveExtruderStack()
|
|
if not new_active_stack:
|
|
self._active_container_stack = None
|
|
return
|
|
|
|
if new_active_stack != self._active_container_stack: # Check if changed
|
|
if self._active_container_stack: # Disconnect signal from old container (if any)
|
|
self._active_container_stack.propertyChanged.disconnect(self._onPropertyChanged)
|
|
self._active_container_stack.containersChanged.disconnect(self._onContainersChanged)
|
|
|
|
self._active_container_stack = new_active_stack
|
|
if self._active_container_stack is not None:
|
|
self._active_container_stack.propertyChanged.connect(self._onPropertyChanged)
|
|
self._active_container_stack.containersChanged.connect(self._onContainersChanged)
|
|
self._update() # Ensure that the settings_with_inheritance_warning list is populated.
|
|
|
|
def _onPropertyChanged(self, key: str, property_name: str) -> None:
|
|
if (property_name == "value" or property_name == "enabled") and self._global_container_stack:
|
|
definitions = self._global_container_stack.definition.findDefinitions(key = key) # type: List["SettingDefinition"]
|
|
if not definitions:
|
|
return
|
|
|
|
has_overwritten_inheritance = self._settingIsOverwritingInheritance(key)
|
|
|
|
settings_with_inheritance_warning_changed = False
|
|
|
|
# Check if the setting needs to be in the list.
|
|
if key not in self._settings_with_inheritance_warning and has_overwritten_inheritance:
|
|
self._settings_with_inheritance_warning.append(key)
|
|
settings_with_inheritance_warning_changed = True
|
|
elif key in self._settings_with_inheritance_warning and not has_overwritten_inheritance:
|
|
self._settings_with_inheritance_warning.remove(key)
|
|
settings_with_inheritance_warning_changed = True
|
|
|
|
parent = definitions[0].parent
|
|
# Find the topmost parent (Assumed to be a category)
|
|
if parent is not None:
|
|
while parent.parent is not None:
|
|
parent = parent.parent
|
|
else:
|
|
parent = definitions[0] # Already at a category
|
|
|
|
if parent.key not in self._settings_with_inheritance_warning and has_overwritten_inheritance:
|
|
# Category was not in the list yet, so needs to be added now.
|
|
self._settings_with_inheritance_warning.append(parent.key)
|
|
settings_with_inheritance_warning_changed = True
|
|
|
|
elif parent.key in self._settings_with_inheritance_warning and not has_overwritten_inheritance:
|
|
# Category was in the list and one of it's settings is not overwritten.
|
|
if not self._recursiveCheck(parent): # Check if any of it's children have overwritten inheritance.
|
|
self._settings_with_inheritance_warning.remove(parent.key)
|
|
settings_with_inheritance_warning_changed = True
|
|
|
|
# Emit the signal if there was any change to the list.
|
|
if settings_with_inheritance_warning_changed:
|
|
self.settingsWithIntheritanceChanged.emit()
|
|
|
|
def _recursiveCheck(self, definition: "SettingDefinition") -> bool:
|
|
for child in definition.children:
|
|
if child.key in self._settings_with_inheritance_warning:
|
|
return True
|
|
if child.children:
|
|
if self._recursiveCheck(child):
|
|
return True
|
|
return False
|
|
|
|
@pyqtProperty("QVariantList", notify = settingsWithIntheritanceChanged)
|
|
def settingsWithInheritanceWarning(self) -> List[str]:
|
|
return self._settings_with_inheritance_warning
|
|
|
|
## Check if a setting has an inheritance function that is overwritten
|
|
def _settingIsOverwritingInheritance(self, key: str, stack: ContainerStack = None) -> bool:
|
|
has_setting_function = False
|
|
if not stack:
|
|
stack = self._active_container_stack
|
|
if not stack: # No active container stack yet!
|
|
return False
|
|
|
|
if self._active_container_stack is None:
|
|
return False
|
|
all_keys = self._active_container_stack.getAllKeys()
|
|
|
|
containers = [] # type: List[ContainerInterface]
|
|
|
|
## Check if the setting has a user state. If not, it is never overwritten.
|
|
has_user_state = stack.getProperty(key, "state") == InstanceState.User
|
|
if not has_user_state:
|
|
return False
|
|
|
|
## If a setting is not enabled, don't label it as overwritten (It's never visible anyway).
|
|
if not stack.getProperty(key, "enabled"):
|
|
return False
|
|
|
|
## Also check if the top container is not a setting function (this happens if the inheritance is restored).
|
|
user_container = stack.getTop()
|
|
if user_container and isinstance(user_container.getProperty(key, "value"), SettingFunction):
|
|
return False
|
|
|
|
## Mash all containers for all the stacks together.
|
|
while stack:
|
|
containers.extend(stack.getContainers())
|
|
stack = stack.getNextStack()
|
|
has_non_function_value = False
|
|
for container in containers:
|
|
try:
|
|
value = container.getProperty(key, "value")
|
|
except AttributeError:
|
|
continue
|
|
if value is not None:
|
|
# If a setting doesn't use any keys, it won't change it's value, so treat it as if it's a fixed value
|
|
has_setting_function = isinstance(value, SettingFunction)
|
|
if has_setting_function:
|
|
for setting_key in value.getUsedSettingKeys():
|
|
if setting_key in all_keys:
|
|
break # We found an actual setting. So has_setting_function can remain true
|
|
else:
|
|
# All of the setting_keys turned out to not be setting keys at all!
|
|
# This can happen due enum keys also being marked as settings.
|
|
has_setting_function = False
|
|
|
|
if has_setting_function is False:
|
|
has_non_function_value = True
|
|
continue
|
|
|
|
if has_setting_function:
|
|
break # There is a setting function somewhere, stop looking deeper.
|
|
return has_setting_function and has_non_function_value
|
|
|
|
def _update(self) -> None:
|
|
self._settings_with_inheritance_warning = [] # Reset previous data.
|
|
|
|
# Make sure that the GlobalStack is not None. sometimes the globalContainerChanged signal gets here late.
|
|
if self._global_container_stack is None:
|
|
return
|
|
|
|
# Check all setting keys that we know of and see if they are overridden.
|
|
for setting_key in self._global_container_stack.getAllKeys():
|
|
override = self._settingIsOverwritingInheritance(setting_key)
|
|
if override:
|
|
self._settings_with_inheritance_warning.append(setting_key)
|
|
|
|
# Check all the categories if any of their children have their inheritance overwritten.
|
|
for category in self._global_container_stack.definition.findDefinitions(type = "category"):
|
|
if self._recursiveCheck(category):
|
|
self._settings_with_inheritance_warning.append(category.key)
|
|
|
|
# Notify others that things have changed.
|
|
self.settingsWithIntheritanceChanged.emit()
|
|
|
|
def _onGlobalContainerChanged(self) -> None:
|
|
if self._global_container_stack:
|
|
self._global_container_stack.propertyChanged.disconnect(self._onPropertyChanged)
|
|
self._global_container_stack.containersChanged.disconnect(self._onContainersChanged)
|
|
self._global_container_stack = Application.getInstance().getGlobalContainerStack()
|
|
if self._global_container_stack:
|
|
self._global_container_stack.containersChanged.connect(self._onContainersChanged)
|
|
self._global_container_stack.propertyChanged.connect(self._onPropertyChanged)
|
|
self._onActiveExtruderChanged()
|
|
|
|
def _onContainersChanged(self, container):
|
|
self._update_timer.start()
|
|
|
|
@staticmethod
|
|
def createSettingInheritanceManager(engine=None, script_engine=None):
|
|
return SettingInheritanceManager()
|