SettingOverrideDecorator.py 4.9 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116
  1. # Copyright (c) 2016 Ultimaker B.V.
  2. # Cura is released under the terms of the AGPLv3 or higher.
  3. import copy
  4. from UM.Scene.SceneNodeDecorator import SceneNodeDecorator
  5. from UM.Signal import Signal, signalemitter
  6. from UM.Settings.ContainerStack import ContainerStack
  7. from UM.Settings.InstanceContainer import InstanceContainer
  8. from UM.Settings.ContainerRegistry import ContainerRegistry
  9. from UM.Logger import Logger
  10. from UM.Application import Application
  11. from cura.Settings.ExtruderManager import ExtruderManager
  12. ## A decorator that adds a container stack to a Node. This stack should be queried for all settings regarding
  13. # the linked node. The Stack in question will refer to the global stack (so that settings that are not defined by
  14. # this stack still resolve.
  15. @signalemitter
  16. class SettingOverrideDecorator(SceneNodeDecorator):
  17. ## Event indicating that the user selected a different extruder.
  18. activeExtruderChanged = Signal()
  19. def __init__(self):
  20. super().__init__()
  21. self._stack = ContainerStack(stack_id = id(self))
  22. self._stack.setDirty(False) # This stack does not need to be saved.
  23. self._stack.addContainer(InstanceContainer(container_id = "SettingOverrideInstanceContainer"))
  24. if ExtruderManager.getInstance().extruderCount > 1:
  25. self._extruder_stack = ExtruderManager.getInstance().getExtruderStack(0).getId()
  26. else:
  27. self._extruder_stack = None
  28. self._stack.propertyChanged.connect(self._onSettingChanged)
  29. ContainerRegistry.getInstance().addContainer(self._stack)
  30. Application.getInstance().globalContainerStackChanged.connect(self._updateNextStack)
  31. self.activeExtruderChanged.connect(self._updateNextStack)
  32. self._updateNextStack()
  33. def __deepcopy__(self, memo):
  34. ## Create a fresh decorator object
  35. deep_copy = SettingOverrideDecorator()
  36. ## Copy the instance
  37. instance_container = copy.deepcopy(self._stack.getContainer(0), memo)
  38. ## Set the copied instance as the first (and only) instance container of the stack.
  39. deep_copy._stack.replaceContainer(0, instance_container)
  40. # Properly set the right extruder on the copy
  41. deep_copy.setActiveExtruder(self._extruder_stack)
  42. return deep_copy
  43. ## Gets the currently active extruder to print this object with.
  44. #
  45. # \return An extruder's container stack.
  46. def getActiveExtruder(self):
  47. return self._extruder_stack
  48. ## Gets the signal that emits if the active extruder changed.
  49. #
  50. # This can then be accessed via a decorator.
  51. def getActiveExtruderChangedSignal(self):
  52. return self.activeExtruderChanged
  53. ## Gets the currently active extruders position
  54. #
  55. # \return An extruder's position, or None if no position info is available.
  56. def getActiveExtruderPosition(self):
  57. containers = ContainerRegistry.getInstance().findContainers(id = self.getActiveExtruder())
  58. if containers:
  59. container_stack = containers[0]
  60. return container_stack.getMetaDataEntry("position", default=None)
  61. def _onSettingChanged(self, instance, property_name): # Reminder: 'property' is a built-in function
  62. # Trigger slice/need slicing if the value has changed.
  63. if property_name == "value":
  64. Application.getInstance().getBackend().needsSlicing()
  65. Application.getInstance().getBackend().tickle()
  66. ## Makes sure that the stack upon which the container stack is placed is
  67. # kept up to date.
  68. def _updateNextStack(self):
  69. if self._extruder_stack:
  70. extruder_stack = ContainerRegistry.getInstance().findContainerStacks(id = self._extruder_stack)
  71. if extruder_stack:
  72. if self._stack.getNextStack():
  73. old_extruder_stack_id = self._stack.getNextStack().getId()
  74. else:
  75. old_extruder_stack_id = ""
  76. self._stack.setNextStack(extruder_stack[0])
  77. # Trigger slice/need slicing if the extruder changed.
  78. if self._stack.getNextStack().getId() != old_extruder_stack_id:
  79. Application.getInstance().getBackend().needsSlicing()
  80. Application.getInstance().getBackend().tickle()
  81. else:
  82. Logger.log("e", "Extruder stack %s below per-object settings does not exist.", self._extruder_stack)
  83. else:
  84. self._stack.setNextStack(Application.getInstance().getGlobalContainerStack())
  85. ## Changes the extruder with which to print this node.
  86. #
  87. # \param extruder_stack_id The new extruder stack to print with.
  88. def setActiveExtruder(self, extruder_stack_id):
  89. self._extruder_stack = extruder_stack_id
  90. self._updateNextStack()
  91. ExtruderManager.getInstance().resetSelectedObjectExtruders()
  92. self.activeExtruderChanged.emit()
  93. def getStack(self):
  94. return self._stack