123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126 |
- # Copyright (c) 2017 Ultimaker B.V.
- # Cura is released under the terms of the LGPLv3 or higher.
- from typing import Any, TYPE_CHECKING, Optional
- from UM.Decorators import override
- from UM.MimeTypeDatabase import MimeType, MimeTypeDatabase
- from UM.Settings.ContainerStack import ContainerStack
- from UM.Settings.ContainerRegistry import ContainerRegistry
- from UM.Settings.Interfaces import ContainerInterface, PropertyEvaluationContext
- from . import Exceptions
- from .CuraContainerStack import CuraContainerStack
- from .ExtruderManager import ExtruderManager
- if TYPE_CHECKING:
- from cura.Settings.GlobalStack import GlobalStack
- ## Represents an Extruder and its related containers.
- #
- #
- class ExtruderStack(CuraContainerStack):
- def __init__(self, container_id: str, *args, **kwargs):
- super().__init__(container_id, *args, **kwargs)
- self.addMetaDataEntry("type", "extruder_train") # For backward compatibility
- self.propertiesChanged.connect(self._onPropertiesChanged)
- ## Overridden from ContainerStack
- #
- # This will set the next stack and ensure that we register this stack as an extruder.
- @override(ContainerStack)
- def setNextStack(self, stack: ContainerStack) -> None:
- super().setNextStack(stack)
- stack.addExtruder(self)
- self.addMetaDataEntry("machine", stack.id)
- # For backward compatibility: Register the extruder with the Extruder Manager
- ExtruderManager.getInstance().registerExtruder(self, stack.id)
- @override(ContainerStack)
- def getNextStack(self) -> Optional["GlobalStack"]:
- return super().getNextStack()
- @classmethod
- def getLoadingPriority(cls) -> int:
- return 3
- ## Overridden from ContainerStack
- #
- # It will perform a few extra checks when trying to get properties.
- #
- # The two extra checks it currently does is to ensure a next stack is set and to bypass
- # the extruder when the property is not settable per extruder.
- #
- # \throws Exceptions.NoGlobalStackError Raised when trying to get a property from an extruder without
- # having a next stack set.
- @override(ContainerStack)
- def getProperty(self, key: str, property_name: str, context: Optional[PropertyEvaluationContext] = None) -> Any:
- if not self._next_stack:
- raise Exceptions.NoGlobalStackError("Extruder {id} is missing the next stack!".format(id = self.id))
- if context is None:
- context = PropertyEvaluationContext()
- context.pushContainer(self)
- if not super().getProperty(key, "settable_per_extruder", context):
- result = self.getNextStack().getProperty(key, property_name, context)
- context.popContainer()
- return result
- limit_to_extruder = super().getProperty(key, "limit_to_extruder", context)
- if limit_to_extruder is not None:
- limit_to_extruder = str(limit_to_extruder)
- if (limit_to_extruder is not None and limit_to_extruder != "-1") and self.getMetaDataEntry("position") != str(limit_to_extruder):
- if str(limit_to_extruder) in self.getNextStack().extruders:
- result = self.getNextStack().extruders[str(limit_to_extruder)].getProperty(key, property_name, context)
- if result is not None:
- context.popContainer()
- return result
- result = super().getProperty(key, property_name, context)
- context.popContainer()
- return result
- @override(CuraContainerStack)
- def _getMachineDefinition(self) -> ContainerInterface:
- if not self.getNextStack():
- raise Exceptions.NoGlobalStackError("Extruder {id} is missing the next stack!".format(id = self.id))
- return self.getNextStack()._getMachineDefinition()
- @override(CuraContainerStack)
- def deserialize(self, contents: str) -> None:
- super().deserialize(contents)
- stacks = ContainerRegistry.getInstance().findContainerStacks(id=self.getMetaDataEntry("machine", ""))
- if stacks:
- self.setNextStack(stacks[0])
- def _onPropertiesChanged(self, key, properties):
- # When there is a setting that is not settable per extruder that depends on a value from a setting that is,
- # we do not always get properly informed that we should re-evaluate the setting. So make sure to indicate
- # something changed for those settings.
- if not self.getNextStack():
- return #There are no global settings to depend on.
- definitions = self.getNextStack().definition.findDefinitions(key = key)
- if definitions:
- has_global_dependencies = False
- for relation in definitions[0].relations:
- if not getattr(relation.target, "settable_per_extruder", True):
- has_global_dependencies = True
- break
- if has_global_dependencies:
- self.getNextStack().propertiesChanged.emit(key, properties)
- extruder_stack_mime = MimeType(
- name = "application/x-cura-extruderstack",
- comment = "Cura Extruder Stack",
- suffixes = ["extruder.cfg"]
- )
- MimeTypeDatabase.addMimeType(extruder_stack_mime)
- ContainerRegistry.addContainerTypeByName(ExtruderStack, "extruder_stack", extruder_stack_mime.name)
|