CuraFormulaFunctions.py 8.4 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177
  1. # Copyright (c) 2018 Ultimaker B.V.
  2. # Cura is released under the terms of the LGPLv3 or higher.
  3. from typing import Any, List, Optional, TYPE_CHECKING
  4. from UM.Settings.PropertyEvaluationContext import PropertyEvaluationContext
  5. from UM.Settings.SettingFunction import SettingFunction
  6. from UM.Logger import Logger
  7. if TYPE_CHECKING:
  8. from cura.CuraApplication import CuraApplication
  9. from cura.Settings.CuraContainerStack import CuraContainerStack
  10. #
  11. # This class contains all Cura-related custom functions that can be used in formulas. Some functions requires
  12. # information such as the currently active machine, so this is made into a class instead of standalone functions.
  13. #
  14. class CuraFormulaFunctions:
  15. def __init__(self, application: "CuraApplication") -> None:
  16. self._application = application
  17. # ================
  18. # Custom Functions
  19. # ================
  20. # Gets the default extruder position of the currently active machine.
  21. def getDefaultExtruderPosition(self) -> str:
  22. machine_manager = self._application.getMachineManager()
  23. return machine_manager.defaultExtruderPosition
  24. # Gets the given setting key from the given extruder position.
  25. def getValueInExtruder(self, extruder_position: int, property_key: str,
  26. context: Optional["PropertyEvaluationContext"] = None) -> Any:
  27. machine_manager = self._application.getMachineManager()
  28. if extruder_position == -1:
  29. extruder_position = int(machine_manager.defaultExtruderPosition)
  30. global_stack = machine_manager.activeMachine
  31. try:
  32. extruder_stack = global_stack.extruderList[int(extruder_position)]
  33. except IndexError:
  34. if extruder_position != 0:
  35. Logger.log("w", "Value for %s of extruder %s was requested, but that extruder is not available. Returning the result from extruder 0 instead" % (property_key, extruder_position))
  36. # This fixes a very specific fringe case; If a profile was created for a custom printer and one of the
  37. # extruder settings has been set to non zero and the profile is loaded for a machine that has only a single extruder
  38. # it would cause all kinds of issues (and eventually a crash).
  39. # See https://github.com/Ultimaker/Cura/issues/5535
  40. return self.getValueInExtruder(0, property_key, context)
  41. Logger.log("w", "Value for %s of extruder %s was requested, but that extruder is not available. " % (property_key, extruder_position))
  42. return None
  43. value = extruder_stack.getRawProperty(property_key, "value", context = context)
  44. if isinstance(value, SettingFunction):
  45. value = value(extruder_stack, context = context)
  46. return value
  47. # Gets all extruder values as a list for the given property.
  48. def getValuesInAllExtruders(self, property_key: str,
  49. context: Optional["PropertyEvaluationContext"] = None) -> List[Any]:
  50. machine_manager = self._application.getMachineManager()
  51. extruder_manager = self._application.getExtruderManager()
  52. global_stack = machine_manager.activeMachine
  53. result = []
  54. for extruder in extruder_manager.getActiveExtruderStacks():
  55. if not extruder.isEnabled:
  56. continue
  57. # only include values from extruders that are "active" for the current machine instance
  58. if int(extruder.getMetaDataEntry("position")) >= global_stack.getProperty("machine_extruder_count", "value", context = context):
  59. continue
  60. value = extruder.getRawProperty(property_key, "value", context = context)
  61. if value is None:
  62. continue
  63. if isinstance(value, SettingFunction):
  64. value = value(extruder, context = context)
  65. result.append(value)
  66. if not result:
  67. result.append(global_stack.getProperty(property_key, "value", context = context))
  68. return result
  69. # Get the resolve value or value for a given key.
  70. def getResolveOrValue(self, property_key: str, context: Optional["PropertyEvaluationContext"] = None) -> Any:
  71. machine_manager = self._application.getMachineManager()
  72. global_stack = machine_manager.activeMachine
  73. resolved_value = global_stack.getProperty(property_key, "value", context = context)
  74. return resolved_value
  75. # Gets the default setting value from given extruder position. The default value is what excludes the values in
  76. # the user_changes container.
  77. def getDefaultValueInExtruder(self, extruder_position: int, property_key: str) -> Any:
  78. machine_manager = self._application.getMachineManager()
  79. global_stack = machine_manager.activeMachine
  80. try:
  81. extruder_stack = global_stack.extruderList[extruder_position]
  82. except IndexError:
  83. Logger.log("w", "Unable to find extruder on in index %s", extruder_position)
  84. else:
  85. context = self.createContextForDefaultValueEvaluation(extruder_stack)
  86. return self.getValueInExtruder(extruder_position, property_key, context = context)
  87. # Gets all default setting values as a list from all extruders of the currently active machine.
  88. # The default values are those excluding the values in the user_changes container.
  89. def getDefaultValuesInAllExtruders(self, property_key: str) -> List[Any]:
  90. machine_manager = self._application.getMachineManager()
  91. global_stack = machine_manager.activeMachine
  92. context = self.createContextForDefaultValueEvaluation(global_stack)
  93. return self.getValuesInAllExtruders(property_key, context = context)
  94. # Gets the resolve value or value for a given key without looking the first container (user container).
  95. def getDefaultResolveOrValue(self, property_key: str) -> Any:
  96. machine_manager = self._application.getMachineManager()
  97. global_stack = machine_manager.activeMachine
  98. context = self.createContextForDefaultValueEvaluation(global_stack)
  99. return self.getResolveOrValue(property_key, context = context)
  100. # Gets the value for the given setting key starting from the given container index.
  101. def getValueFromContainerAtIndex(self, property_key: str, container_index: int,
  102. context: Optional["PropertyEvaluationContext"] = None) -> Any:
  103. machine_manager = self._application.getMachineManager()
  104. global_stack = machine_manager.activeMachine
  105. context = self.createContextForDefaultValueEvaluation(global_stack)
  106. context.context["evaluate_from_container_index"] = container_index
  107. return global_stack.getProperty(property_key, "value", context = context)
  108. # Gets the extruder value for the given setting key starting from the given container index.
  109. def getValueFromContainerAtIndexInExtruder(self, extruder_position: int, property_key: str, container_index: int,
  110. context: Optional["PropertyEvaluationContext"] = None) -> Any:
  111. machine_manager = self._application.getMachineManager()
  112. global_stack = machine_manager.activeMachine
  113. if extruder_position == -1:
  114. extruder_position = int(machine_manager.defaultExtruderPosition)
  115. global_stack = machine_manager.activeMachine
  116. try:
  117. extruder_stack = global_stack.extruderList[int(extruder_position)]
  118. except IndexError:
  119. Logger.log("w", "Value for %s of extruder %s was requested, but that extruder is not available. " % (property_key, extruder_position))
  120. return None
  121. context = self.createContextForDefaultValueEvaluation(extruder_stack)
  122. context.context["evaluate_from_container_index"] = container_index
  123. return self.getValueInExtruder(extruder_position, property_key, context)
  124. # Creates a context for evaluating default values (skip the user_changes container).
  125. def createContextForDefaultValueEvaluation(self, source_stack: "CuraContainerStack") -> "PropertyEvaluationContext":
  126. context = PropertyEvaluationContext(source_stack)
  127. context.context["evaluate_from_container_index"] = 1 # skip the user settings container
  128. context.context["override_operators"] = {
  129. "extruderValue": self.getDefaultValueInExtruder,
  130. "extruderValues": self.getDefaultValuesInAllExtruders,
  131. "resolveOrValue": self.getDefaultResolveOrValue,
  132. }
  133. return context