CuraFormulaFunctions.py 6.5 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142
  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.extruders[str(extruder_position)]
  33. except KeyError:
  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 form 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. extruder_stack = global_stack.extruders[str(extruder_position)]
  81. context = self.createContextForDefaultValueEvaluation(extruder_stack)
  82. return self.getValueInExtruder(extruder_position, property_key, context = context)
  83. # Gets all default setting values as a list from all extruders of the currently active machine.
  84. # The default values are those excluding the values in the user_changes container.
  85. def getDefaultValuesInAllExtruders(self, property_key: str) -> List[Any]:
  86. machine_manager = self._application.getMachineManager()
  87. global_stack = machine_manager.activeMachine
  88. context = self.createContextForDefaultValueEvaluation(global_stack)
  89. return self.getValuesInAllExtruders(property_key, context = context)
  90. # Gets the resolve value or value for a given key without looking the first container (user container).
  91. def getDefaultResolveOrValue(self, property_key: str) -> Any:
  92. machine_manager = self._application.getMachineManager()
  93. global_stack = machine_manager.activeMachine
  94. context = self.createContextForDefaultValueEvaluation(global_stack)
  95. return self.getResolveOrValue(property_key, context = context)
  96. # Creates a context for evaluating default values (skip the user_changes container).
  97. def createContextForDefaultValueEvaluation(self, source_stack: "CuraContainerStack") -> "PropertyEvaluationContext":
  98. context = PropertyEvaluationContext(source_stack)
  99. context.context["evaluate_from_container_index"] = 1 # skip the user settings container
  100. context.context["override_operators"] = {
  101. "extruderValue": self.getDefaultValueInExtruder,
  102. "extruderValues": self.getDefaultValuesInAllExtruders,
  103. "resolveOrValue": self.getDefaultResolveOrValue,
  104. }
  105. return context