# Copyright (c) 2017 Ultimaker B.V. # Cura is released under the terms of the LGPLv3 or higher. from typing import List, Optional, Set, TYPE_CHECKING from PyQt6.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) 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._update_timer = QTimer() self._update_timer.setInterval(500) self._update_timer.setSingleShot(True) self._update_timer.timeout.connect(self._update) Application.getInstance().globalContainerStackChanged.connect(self._onGlobalContainerChanged) ExtruderManager.getInstance().activeExtruderChanged.connect(self._onActiveExtruderChanged) self._onGlobalContainerChanged() self._onActiveExtruderChanged() settingsWithIntheritanceChanged = pyqtSignal() @pyqtSlot(str, result = "QStringList") def getChildrenKeysWithOverride(self, key: str) -> List[str]: """Get the keys of all children settings with an override.""" 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 = bool) def hasOverrides(self, key: str, extruder_index: str): return key in self.getOverridesForExtruder(key, extruder_index) @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 scheduleUpdate(self) -> None: self._update_timer.start() 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_timer.start() # 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 def _userSettingIsOverwritingInheritance(self, key: str, stack: ContainerStack, all_keys: Set[str] = set()) -> bool: """Check if a setting known as having a User state has an inheritance function that is overwritten""" has_setting_function = False containers = [] # type: List[ContainerInterface] # 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 user_container = stack.getTop() # Also check if the top container is not a setting function (this happens if the inheritance is restored). if user_container and isinstance(user_container.getProperty(key, "value"), SettingFunction): return False if not all_keys: all_keys = self._active_container_stack.getAllKeys() ## 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 _settingIsOverwritingInheritance(self, key: str, stack: ContainerStack = None) -> bool: """Check if a setting has an inheritance function that is overwritten""" 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 has_user_state = stack.getProperty(key, "state") == InstanceState.User if not has_user_state: return False return self._userSettingIsOverwritingInheritance(key, stack) 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 or self._active_container_stack is None: return # Check all user setting keys that we know of and see if they are overridden. all_keys = self._active_container_stack.getAllKeys() for setting_key in self._active_container_stack.getAllKeysWithUserState(): if self._userSettingIsOverwritingInheritance(setting_key, self._active_container_stack, all_keys): 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()