MachineManager.py 62 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891892893894895896897898899900901902903904905906907908909910911912913914915916917918919920921922923924925926927928929930931932933934935936937938939940941942943944945946947948949950951952953954955956957958959960961962963964965966967968969970971972973974975976977978979980981982983984985986987988989990991992993994995996997998999100010011002100310041005100610071008100910101011101210131014101510161017101810191020102110221023102410251026102710281029103010311032103310341035103610371038103910401041104210431044104510461047104810491050105110521053105410551056105710581059106010611062106310641065106610671068106910701071107210731074107510761077107810791080108110821083108410851086108710881089109010911092109310941095109610971098109911001101110211031104110511061107110811091110111111121113111411151116111711181119112011211122112311241125112611271128112911301131113211331134113511361137113811391140114111421143114411451146114711481149115011511152115311541155115611571158115911601161116211631164116511661167116811691170117111721173117411751176117711781179118011811182118311841185118611871188118911901191119211931194119511961197119811991200120112021203120412051206120712081209121012111212121312141215121612171218121912201221122212231224122512261227122812291230123112321233123412351236123712381239124012411242124312441245124612471248124912501251125212531254125512561257125812591260126112621263126412651266126712681269127012711272127312741275127612771278127912801281128212831284128512861287128812891290129112921293
  1. # Copyright (c) 2017 Ultimaker B.V.
  2. # Cura is released under the terms of the AGPLv3 or higher.
  3. from typing import Union
  4. from PyQt5.QtCore import QObject, pyqtProperty, pyqtSignal, QTimer
  5. from UM.FlameProfiler import pyqtSlot
  6. from PyQt5.QtWidgets import QMessageBox
  7. from UM import Util
  8. from UM.Application import Application
  9. from UM.Preferences import Preferences
  10. from UM.Logger import Logger
  11. from UM.Message import Message
  12. from UM.Decorators import deprecated
  13. from UM.Settings.ContainerRegistry import ContainerRegistry
  14. from UM.Settings.ContainerStack import ContainerStack
  15. from UM.Settings.InstanceContainer import InstanceContainer
  16. from UM.Settings.SettingDefinition import SettingDefinition
  17. from UM.Settings.SettingFunction import SettingFunction
  18. from UM.Settings.Validator import ValidatorState
  19. from UM.Signal import postponeSignals
  20. import UM.FlameProfiler
  21. from cura.QualityManager import QualityManager
  22. from cura.PrinterOutputDevice import PrinterOutputDevice
  23. from cura.Settings.ExtruderManager import ExtruderManager
  24. from .GlobalStack import GlobalStack
  25. from .CuraStackBuilder import CuraStackBuilder
  26. from UM.i18n import i18nCatalog
  27. catalog = i18nCatalog("cura")
  28. from typing import TYPE_CHECKING, Optional
  29. if TYPE_CHECKING:
  30. from UM.Settings.DefinitionContainer import DefinitionContainer
  31. import os
  32. class MachineManager(QObject):
  33. def __init__(self, parent = None):
  34. super().__init__(parent)
  35. self._active_container_stack = None # type: ContainerStack
  36. self._global_container_stack = None # type: ContainerStack
  37. self._error_check_timer = QTimer()
  38. self._error_check_timer.setInterval(250)
  39. self._error_check_timer.setSingleShot(True)
  40. self._error_check_timer.timeout.connect(self._updateStacksHaveErrors)
  41. self._instance_container_timer = QTimer()
  42. self._instance_container_timer.setInterval(250)
  43. self._instance_container_timer.setSingleShot(True)
  44. self._instance_container_timer.timeout.connect(self.__onInstanceContainersChanged)
  45. Application.getInstance().globalContainerStackChanged.connect(self._onGlobalContainerChanged)
  46. ## When the global container is changed, active material probably needs to be updated.
  47. self.globalContainerChanged.connect(self.activeMaterialChanged)
  48. self.globalContainerChanged.connect(self.activeVariantChanged)
  49. self.globalContainerChanged.connect(self.activeQualityChanged)
  50. self._stacks_have_errors = None
  51. self._empty_variant_container = ContainerRegistry.getInstance().getEmptyInstanceContainer()
  52. self._empty_material_container = ContainerRegistry.getInstance().getEmptyInstanceContainer()
  53. self._empty_quality_container = ContainerRegistry.getInstance().getEmptyInstanceContainer()
  54. self._empty_quality_changes_container = ContainerRegistry.getInstance().getEmptyInstanceContainer()
  55. self._onGlobalContainerChanged()
  56. ExtruderManager.getInstance().activeExtruderChanged.connect(self._onActiveExtruderStackChanged)
  57. self._onActiveExtruderStackChanged()
  58. ExtruderManager.getInstance().activeExtruderChanged.connect(self.activeMaterialChanged)
  59. ExtruderManager.getInstance().activeExtruderChanged.connect(self.activeVariantChanged)
  60. ExtruderManager.getInstance().activeExtruderChanged.connect(self.activeQualityChanged)
  61. self.globalContainerChanged.connect(self.activeStackChanged)
  62. self.globalValueChanged.connect(self.activeStackValueChanged)
  63. ExtruderManager.getInstance().activeExtruderChanged.connect(self.activeStackChanged)
  64. self.activeStackChanged.connect(self.activeStackValueChanged)
  65. Preferences.getInstance().addPreference("cura/active_machine", "")
  66. self._global_event_keys = set()
  67. active_machine_id = Preferences.getInstance().getValue("cura/active_machine")
  68. self._printer_output_devices = []
  69. Application.getInstance().getOutputDeviceManager().outputDevicesChanged.connect(self._onOutputDevicesChanged)
  70. if active_machine_id != "":
  71. # An active machine was saved, so restore it.
  72. self.setActiveMachine(active_machine_id)
  73. if self._global_container_stack and self._global_container_stack.getProperty("machine_extruder_count", "value") > 1:
  74. # Make sure _active_container_stack is properly initiated
  75. ExtruderManager.getInstance().setActiveExtruderIndex(0)
  76. self._auto_materials_changed = {}
  77. self._auto_hotends_changed = {}
  78. self._material_incompatible_message = Message(catalog.i18nc("@info:status",
  79. "The selected material is incompatible with the selected machine or configuration."))
  80. globalContainerChanged = pyqtSignal() # Emitted whenever the global stack is changed (ie: when changing between printers, changing a global profile, but not when changing a value)
  81. activeMaterialChanged = pyqtSignal()
  82. activeVariantChanged = pyqtSignal()
  83. activeQualityChanged = pyqtSignal()
  84. activeStackChanged = pyqtSignal() # Emitted whenever the active stack is changed (ie: when changing between extruders, changing a profile, but not when changing a value)
  85. globalValueChanged = pyqtSignal() # Emitted whenever a value inside global container is changed.
  86. activeStackValueChanged = pyqtSignal() # Emitted whenever a value inside the active stack is changed.
  87. activeStackValidationChanged = pyqtSignal() # Emitted whenever a validation inside active container is changed
  88. stacksValidationChanged = pyqtSignal() # Emitted whenever a validation is changed
  89. blurSettings = pyqtSignal() # Emitted to force fields in the advanced sidebar to un-focus, so they update properly
  90. outputDevicesChanged = pyqtSignal()
  91. def _onOutputDevicesChanged(self) -> None:
  92. for printer_output_device in self._printer_output_devices:
  93. printer_output_device.hotendIdChanged.disconnect(self._onHotendIdChanged)
  94. printer_output_device.materialIdChanged.disconnect(self._onMaterialIdChanged)
  95. self._printer_output_devices.clear()
  96. for printer_output_device in Application.getInstance().getOutputDeviceManager().getOutputDevices():
  97. if isinstance(printer_output_device, PrinterOutputDevice):
  98. self._printer_output_devices.append(printer_output_device)
  99. printer_output_device.hotendIdChanged.connect(self._onHotendIdChanged)
  100. printer_output_device.materialIdChanged.connect(self._onMaterialIdChanged)
  101. self.outputDevicesChanged.emit()
  102. @pyqtProperty("QVariantList", notify = outputDevicesChanged)
  103. def printerOutputDevices(self):
  104. return self._printer_output_devices
  105. @pyqtProperty(int, constant=True)
  106. def totalNumberOfSettings(self):
  107. return len(ContainerRegistry.getInstance().findDefinitionContainers(id="fdmprinter")[0].getAllKeys())
  108. def _onHotendIdChanged(self, index: Union[str, int], hotend_id: str) -> None:
  109. if not self._global_container_stack:
  110. return
  111. containers = ContainerRegistry.getInstance().findInstanceContainers(type="variant", definition=self._global_container_stack.getBottom().getId(), name=hotend_id)
  112. if containers: # New material ID is known
  113. extruder_manager = ExtruderManager.getInstance()
  114. machine_id = self.activeMachineId
  115. extruders = extruder_manager.getMachineExtruders(machine_id)
  116. matching_extruder = None
  117. for extruder in extruders:
  118. if str(index) == extruder.getMetaDataEntry("position"):
  119. matching_extruder = extruder
  120. break
  121. if matching_extruder and matching_extruder.findContainer({"type": "variant"}).getName() != hotend_id:
  122. # Save the material that needs to be changed. Multiple changes will be handled by the callback.
  123. self._auto_hotends_changed[str(index)] = containers[0].getId()
  124. self._printer_output_devices[0].materialHotendChangedMessage(self._materialHotendChangedCallback)
  125. else:
  126. Logger.log("w", "No variant found for printer definition %s with id %s" % (self._global_container_stack.getBottom().getId(), hotend_id))
  127. def _onMaterialIdChanged(self, index, material_id):
  128. if not self._global_container_stack:
  129. return
  130. definition_id = "fdmprinter"
  131. if self._global_container_stack.getMetaDataEntry("has_machine_materials", False):
  132. definition_id = self.activeQualityDefinitionId
  133. extruder_manager = ExtruderManager.getInstance()
  134. containers = ContainerRegistry.getInstance().findInstanceContainers(type = "material", definition = definition_id, GUID = material_id)
  135. if containers: # New material ID is known
  136. extruders = list(extruder_manager.getMachineExtruders(self.activeMachineId))
  137. matching_extruder = None
  138. for extruder in extruders:
  139. if str(index) == extruder.getMetaDataEntry("position"):
  140. matching_extruder = extruder
  141. break
  142. if matching_extruder and matching_extruder.findContainer({"type": "material"}).getMetaDataEntry("GUID") != material_id:
  143. # Save the material that needs to be changed. Multiple changes will be handled by the callback.
  144. variant_container = matching_extruder.findContainer({"type": "variant"})
  145. if self._global_container_stack.getBottom().getMetaDataEntry("has_variants") and variant_container:
  146. variant_id = self.getQualityVariantId(self._global_container_stack.getBottom(), variant_container)
  147. for container in containers:
  148. if container.getMetaDataEntry("variant") == variant_id:
  149. self._auto_materials_changed[str(index)] = container.getId()
  150. break
  151. else:
  152. # Just use the first result we found.
  153. self._auto_materials_changed[str(index)] = containers[0].getId()
  154. self._printer_output_devices[0].materialHotendChangedMessage(self._materialHotendChangedCallback)
  155. else:
  156. Logger.log("w", "No material definition found for printer definition %s and GUID %s" % (definition_id, material_id))
  157. def _materialHotendChangedCallback(self, button):
  158. if button == QMessageBox.No:
  159. self._auto_materials_changed = {}
  160. self._auto_hotends_changed = {}
  161. return
  162. self._autoUpdateMaterials()
  163. self._autoUpdateHotends()
  164. def _autoUpdateMaterials(self):
  165. extruder_manager = ExtruderManager.getInstance()
  166. for position in self._auto_materials_changed:
  167. material_id = self._auto_materials_changed[position]
  168. old_index = extruder_manager.activeExtruderIndex
  169. if old_index != int(position):
  170. extruder_manager.setActiveExtruderIndex(int(position))
  171. else:
  172. old_index = None
  173. Logger.log("d", "Setting material of hotend %s to %s" % (position, material_id))
  174. self.setActiveMaterial(material_id)
  175. if old_index is not None:
  176. extruder_manager.setActiveExtruderIndex(old_index)
  177. def _autoUpdateHotends(self):
  178. extruder_manager = ExtruderManager.getInstance()
  179. for position in self._auto_hotends_changed:
  180. hotend_id = self._auto_hotends_changed[position]
  181. old_index = extruder_manager.activeExtruderIndex
  182. if old_index != int(position):
  183. extruder_manager.setActiveExtruderIndex(int(position))
  184. else:
  185. old_index = None
  186. Logger.log("d", "Setting hotend variant of hotend %s to %s" % (position, hotend_id))
  187. self.setActiveVariant(hotend_id)
  188. if old_index is not None:
  189. extruder_manager.setActiveExtruderIndex(old_index)
  190. def _onGlobalContainerChanged(self):
  191. if self._global_container_stack:
  192. try:
  193. self._global_container_stack.nameChanged.disconnect(self._onMachineNameChanged)
  194. except TypeError: #pyQtSignal gives a TypeError when disconnecting from something that was already disconnected.
  195. pass
  196. try:
  197. self._global_container_stack.containersChanged.disconnect(self._onInstanceContainersChanged)
  198. except TypeError:
  199. pass
  200. try:
  201. self._global_container_stack.propertyChanged.disconnect(self._onPropertyChanged)
  202. except TypeError:
  203. pass
  204. material = self._global_container_stack.material
  205. material.nameChanged.disconnect(self._onMaterialNameChanged)
  206. quality = self._global_container_stack.quality
  207. quality.nameChanged.disconnect(self._onQualityNameChanged)
  208. if self._global_container_stack.getProperty("machine_extruder_count", "value") > 1:
  209. for extruder_stack in ExtruderManager.getInstance().getActiveExtruderStacks():
  210. extruder_stack.propertyChanged.disconnect(self._onPropertyChanged)
  211. extruder_stack.containersChanged.disconnect(self._onInstanceContainersChanged)
  212. self._global_container_stack = Application.getInstance().getGlobalContainerStack()
  213. self._active_container_stack = self._global_container_stack
  214. self.globalContainerChanged.emit()
  215. if self._global_container_stack:
  216. Preferences.getInstance().setValue("cura/active_machine", self._global_container_stack.getId())
  217. self._global_container_stack.nameChanged.connect(self._onMachineNameChanged)
  218. self._global_container_stack.containersChanged.connect(self._onInstanceContainersChanged)
  219. self._global_container_stack.propertyChanged.connect(self._onPropertyChanged)
  220. if self._global_container_stack.getProperty("machine_extruder_count", "value") > 1:
  221. # For multi-extrusion machines, we do not want variant or material profiles in the stack,
  222. # because these are extruder specific and may cause wrong values to be used for extruders
  223. # that did not specify a value in the extruder.
  224. global_variant = self._global_container_stack.variant
  225. if global_variant != self._empty_variant_container:
  226. self._global_container_stack.setVariant(self._empty_variant_container)
  227. global_material = self._global_container_stack.material
  228. if global_material != self._empty_material_container:
  229. self._global_container_stack.setMaterial(self._empty_material_container)
  230. for extruder_stack in ExtruderManager.getInstance().getActiveExtruderStacks(): #Listen for changes on all extruder stacks.
  231. extruder_stack.propertyChanged.connect(self._onPropertyChanged)
  232. extruder_stack.containersChanged.connect(self._onInstanceContainersChanged)
  233. else:
  234. material = self._global_container_stack.material
  235. material.nameChanged.connect(self._onMaterialNameChanged)
  236. quality = self._global_container_stack.quality
  237. quality.nameChanged.connect(self._onQualityNameChanged)
  238. self._error_check_timer.start()
  239. ## Update self._stacks_valid according to _checkStacksForErrors and emit if change.
  240. def _updateStacksHaveErrors(self):
  241. old_stacks_have_errors = self._stacks_have_errors
  242. self._stacks_have_errors = self._checkStacksHaveErrors()
  243. if old_stacks_have_errors != self._stacks_have_errors:
  244. self.stacksValidationChanged.emit()
  245. def _onActiveExtruderStackChanged(self):
  246. self.blurSettings.emit() # Ensure no-one has focus.
  247. old_active_container_stack = self._active_container_stack
  248. self._active_container_stack = ExtruderManager.getInstance().getActiveExtruderStack()
  249. if not self._active_container_stack:
  250. self._active_container_stack = self._global_container_stack
  251. self._error_check_timer.start()
  252. if old_active_container_stack != self._active_container_stack:
  253. # Many methods and properties related to the active quality actually depend
  254. # on _active_container_stack. If it changes, then the properties change.
  255. self.activeQualityChanged.emit()
  256. def __onInstanceContainersChanged(self):
  257. self.activeQualityChanged.emit()
  258. self.activeVariantChanged.emit()
  259. self.activeMaterialChanged.emit()
  260. self._error_check_timer.start()
  261. def _onInstanceContainersChanged(self, container):
  262. self._instance_container_timer.start()
  263. def _onPropertyChanged(self, key, property_name):
  264. if property_name == "value":
  265. # Notify UI items, such as the "changed" star in profile pull down menu.
  266. self.activeStackValueChanged.emit()
  267. elif property_name == "validationState":
  268. self._error_check_timer.start()
  269. @pyqtSlot(str)
  270. def setActiveMachine(self, stack_id: str) -> None:
  271. self.blurSettings.emit() # Ensure no-one has focus.
  272. containers = ContainerRegistry.getInstance().findContainerStacks(id = stack_id)
  273. if containers:
  274. Application.getInstance().setGlobalContainerStack(containers[0])
  275. @pyqtSlot(str, str)
  276. def addMachine(self, name: str, definition_id: str) -> None:
  277. new_stack = CuraStackBuilder.createMachine(name, definition_id)
  278. if new_stack:
  279. Application.getInstance().setGlobalContainerStack(new_stack)
  280. else:
  281. Logger.log("w", "Failed creating a new machine!")
  282. ## Create a name that is not empty and unique
  283. # \param container_type \type{string} Type of the container (machine, quality, ...)
  284. # \param current_name \type{} Current name of the container, which may be an acceptable option
  285. # \param new_name \type{string} Base name, which may not be unique
  286. # \param fallback_name \type{string} Name to use when (stripped) new_name is empty
  287. # \return \type{string} Name that is unique for the specified type and name/id
  288. def _createUniqueName(self, container_type: str, current_name: str, new_name: str, fallback_name: str) -> str:
  289. return ContainerRegistry.getInstance().createUniqueName(container_type, current_name, new_name, fallback_name)
  290. def _checkStacksHaveErrors(self):
  291. if self._global_container_stack is None: #No active machine.
  292. return False
  293. if self._global_container_stack.hasErrors():
  294. return True
  295. for stack in ExtruderManager.getInstance().getMachineExtruders(self._global_container_stack.getId()):
  296. if stack.hasErrors():
  297. return True
  298. return False
  299. ## Remove all instances from the top instanceContainer (effectively removing all user-changed settings)
  300. @pyqtSlot()
  301. def clearUserSettings(self):
  302. if not self._active_container_stack:
  303. return
  304. self.blurSettings.emit()
  305. user_settings = self._active_container_stack.getTop()
  306. user_settings.clear()
  307. ## Check if the global_container has instances in the user container
  308. @pyqtProperty(bool, notify = activeStackValueChanged)
  309. def hasUserSettings(self):
  310. if not self._global_container_stack:
  311. return False
  312. if self._global_container_stack.getTop().findInstances():
  313. return True
  314. stacks = list(ExtruderManager.getInstance().getMachineExtruders(self._global_container_stack.getId()))
  315. for stack in stacks:
  316. if stack.getTop().findInstances():
  317. return True
  318. return False
  319. @pyqtProperty(int, notify = activeStackValueChanged)
  320. def numUserSettings(self):
  321. if not self._global_container_stack:
  322. return 0
  323. num_user_settings = 0
  324. num_user_settings += len(self._global_container_stack.getTop().findInstances())
  325. stacks = list(ExtruderManager.getInstance().getMachineExtruders(self._global_container_stack.getId()))
  326. for stack in stacks:
  327. num_user_settings += len(stack.getTop().findInstances())
  328. return num_user_settings
  329. ## Delete a user setting from the global stack and all extruder stacks.
  330. # \param key \type{str} the name of the key to delete
  331. @pyqtSlot(str)
  332. def clearUserSettingAllCurrentStacks(self, key):
  333. if not self._global_container_stack:
  334. return
  335. send_emits_containers = []
  336. top_container = self._global_container_stack.getTop()
  337. top_container.removeInstance(key, postpone_emit=True)
  338. send_emits_containers.append(top_container)
  339. linked = not self._global_container_stack.getProperty(key, "settable_per_extruder") or \
  340. self._global_container_stack.getProperty(key, "limit_to_extruder") != "-1"
  341. if not linked:
  342. stack = ExtruderManager.getInstance().getActiveExtruderStack()
  343. stacks = [stack]
  344. else:
  345. stacks = ExtruderManager.getInstance().getMachineExtruders(self._global_container_stack.getId())
  346. for stack in stacks:
  347. if stack is not None:
  348. container = stack.getTop()
  349. container.removeInstance(key, postpone_emit=True)
  350. send_emits_containers.append(container)
  351. for container in send_emits_containers:
  352. container.sendPostponedEmits()
  353. ## Check if none of the stacks contain error states
  354. # Note that the _stacks_have_errors is cached due to performance issues
  355. # Calling _checkStack(s)ForErrors on every change is simply too expensive
  356. @pyqtProperty(bool, notify = stacksValidationChanged)
  357. def stacksHaveErrors(self):
  358. return bool(self._stacks_have_errors)
  359. @pyqtProperty(str, notify = activeStackChanged)
  360. def activeUserProfileId(self) -> str:
  361. if self._active_container_stack:
  362. return self._active_container_stack.getTop().getId()
  363. return ""
  364. @pyqtProperty(str, notify = globalContainerChanged)
  365. def activeMachineName(self) -> str:
  366. if self._global_container_stack:
  367. return self._global_container_stack.getName()
  368. return ""
  369. @pyqtProperty(str, notify = globalContainerChanged)
  370. def activeMachineId(self) -> str:
  371. if self._global_container_stack:
  372. return self._global_container_stack.getId()
  373. return ""
  374. @pyqtProperty("QObject", notify = globalContainerChanged)
  375. def activeMachine(self) -> GlobalStack:
  376. return self._global_container_stack
  377. @pyqtProperty(str, notify = activeStackChanged)
  378. def activeStackId(self) -> str:
  379. if self._active_container_stack:
  380. return self._active_container_stack.getId()
  381. return ""
  382. @pyqtProperty(str, notify = activeMaterialChanged)
  383. def activeMaterialName(self) -> str:
  384. if self._active_container_stack:
  385. material = self._active_container_stack.material
  386. if material:
  387. return material.getName()
  388. return ""
  389. @pyqtProperty("QVariantList", notify=activeVariantChanged)
  390. def activeVariantNames(self):
  391. result = []
  392. if ExtruderManager.getInstance().getActiveGlobalAndExtruderStacks() is not None:
  393. for stack in ExtruderManager.getInstance().getActiveGlobalAndExtruderStacks():
  394. variant_container = stack.variant
  395. if variant_container and variant_container != self._empty_variant_container:
  396. result.append(variant_container.getName())
  397. return result
  398. @pyqtProperty("QVariantList", notify = activeVariantChanged)
  399. def activeMaterialIds(self):
  400. result = []
  401. if ExtruderManager.getInstance().getActiveGlobalAndExtruderStacks() is not None:
  402. for stack in ExtruderManager.getInstance().getActiveGlobalAndExtruderStacks():
  403. variant_container = stack.findContainer({"type": "variant"})
  404. if variant_container and variant_container != self._empty_variant_container:
  405. result.append(variant_container.getId())
  406. return result
  407. @pyqtProperty("QVariantList", notify = activeMaterialChanged)
  408. def activeMaterialNames(self):
  409. result = []
  410. if ExtruderManager.getInstance().getActiveGlobalAndExtruderStacks() is not None:
  411. for stack in ExtruderManager.getInstance().getActiveGlobalAndExtruderStacks():
  412. material_container = stack.material
  413. if material_container and material_container != self._empty_material_container:
  414. result.append(material_container.getName())
  415. return result
  416. @pyqtProperty(str, notify=activeMaterialChanged)
  417. def activeMaterialId(self) -> str:
  418. if self._active_container_stack:
  419. material = self._active_container_stack.material
  420. if material:
  421. return material.getId()
  422. return ""
  423. @pyqtProperty("QVariantMap", notify = activeVariantChanged)
  424. def allActiveVariantIds(self):
  425. if not self._global_container_stack:
  426. return {}
  427. result = {}
  428. for stack in ExtruderManager.getInstance().getActiveGlobalAndExtruderStacks():
  429. variant_container = stack.variant
  430. if not variant_container:
  431. continue
  432. result[stack.getId()] = variant_container.getId()
  433. return result
  434. @pyqtProperty("QVariantMap", notify = activeMaterialChanged)
  435. def allActiveMaterialIds(self):
  436. if not self._global_container_stack:
  437. return {}
  438. result = {}
  439. for stack in ExtruderManager.getInstance().getActiveGlobalAndExtruderStacks():
  440. material_container = stack.material
  441. if not material_container:
  442. continue
  443. result[stack.getId()] = material_container.getId()
  444. return result
  445. ## Gets the layer height of the currently active quality profile.
  446. #
  447. # This is indicated together with the name of the active quality profile.
  448. #
  449. # \return The layer height of the currently active quality profile. If
  450. # there is no quality profile, this returns 0.
  451. @pyqtProperty(float, notify=activeQualityChanged)
  452. def activeQualityLayerHeight(self):
  453. if not self._global_container_stack:
  454. return 0
  455. quality_changes = self._global_container_stack.qualityChanges
  456. if quality_changes:
  457. value = self._global_container_stack.getRawProperty("layer_height", "value", skip_until_container = quality_changes.getId())
  458. if isinstance(value, SettingFunction):
  459. value = value(self._global_container_stack)
  460. return value
  461. quality = self._global_container_stack.quality
  462. if quality:
  463. value = self._global_container_stack.getRawProperty("layer_height", "value", skip_until_container = quality.getId())
  464. if isinstance(value, SettingFunction):
  465. value = value(self._global_container_stack)
  466. return value
  467. return 0 #No quality profile.
  468. ## Get the Material ID associated with the currently active material
  469. # \returns MaterialID (string) if found, empty string otherwise
  470. @pyqtProperty(str, notify=activeQualityChanged)
  471. def activeQualityMaterialId(self) -> str:
  472. if self._active_container_stack:
  473. quality = self._active_container_stack.quality
  474. if quality:
  475. material_id = quality.getMetaDataEntry("material")
  476. if material_id:
  477. # if the currently active machine inherits its qualities from a different machine
  478. # definition, make sure to return a material that is relevant to that machine definition
  479. definition_id = self.activeDefinitionId
  480. quality_definition_id = self.activeQualityDefinitionId
  481. if definition_id != quality_definition_id:
  482. material_id = material_id.replace(definition_id, quality_definition_id, 1)
  483. return material_id
  484. return ""
  485. @pyqtProperty(str, notify=activeQualityChanged)
  486. def activeQualityName(self):
  487. if self._active_container_stack and self._global_container_stack:
  488. quality = self._global_container_stack.qualityChanges
  489. if quality and not isinstance(quality, type(self._empty_quality_changes_container)):
  490. return quality.getName()
  491. quality = self._active_container_stack.quality
  492. if quality:
  493. return quality.getName()
  494. return ""
  495. @pyqtProperty(str, notify=activeQualityChanged)
  496. def activeQualityId(self):
  497. if self._active_container_stack:
  498. quality = self._active_container_stack.qualityChanges
  499. if quality and not isinstance(quality, type(self._empty_quality_changes_container)):
  500. return quality.getId()
  501. quality = self._active_container_stack.quality
  502. if quality:
  503. return quality.getId()
  504. return ""
  505. @pyqtProperty(str, notify=activeQualityChanged)
  506. def globalQualityId(self):
  507. if self._global_container_stack:
  508. quality = self._global_container_stack.qualityChanges
  509. if quality and not isinstance(quality, type(self._empty_quality_changes_container)):
  510. return quality.getId()
  511. quality = self._global_container_stack.quality
  512. if quality:
  513. return quality.getId()
  514. return ""
  515. @pyqtProperty(str, notify = activeQualityChanged)
  516. def activeQualityType(self):
  517. if self._active_container_stack:
  518. quality = self._active_container_stack.quality
  519. if quality:
  520. return quality.getMetaDataEntry("quality_type")
  521. return ""
  522. @pyqtProperty(bool, notify = activeQualityChanged)
  523. def isActiveQualitySupported(self):
  524. if self._active_container_stack:
  525. quality = self._active_container_stack.quality
  526. if quality:
  527. return Util.parseBool(quality.getMetaDataEntry("supported", True))
  528. return False
  529. ## Get the Quality ID associated with the currently active extruder
  530. # Note that this only returns the "quality", not the "quality_changes"
  531. # \returns QualityID (string) if found, empty string otherwise
  532. # \sa activeQualityId()
  533. # \todo Ideally, this method would be named activeQualityId(), and the other one
  534. # would be named something like activeQualityOrQualityChanges() for consistency
  535. @pyqtProperty(str, notify = activeQualityChanged)
  536. def activeQualityContainerId(self):
  537. # We're using the active stack instead of the global stack in case the list of qualities differs per extruder
  538. if self._global_container_stack:
  539. quality = self._active_container_stack.quality
  540. if quality:
  541. return quality.getId()
  542. return ""
  543. @pyqtProperty(str, notify = activeQualityChanged)
  544. def activeQualityChangesId(self):
  545. if self._active_container_stack:
  546. changes = self._active_container_stack.qualityChanges
  547. if changes and changes.getId() != "empty":
  548. return changes.getId()
  549. return ""
  550. ## Check if a container is read_only
  551. @pyqtSlot(str, result = bool)
  552. def isReadOnly(self, container_id) -> bool:
  553. containers = ContainerRegistry.getInstance().findInstanceContainers(id = container_id)
  554. if not containers or not self._active_container_stack:
  555. return True
  556. return containers[0].isReadOnly()
  557. ## Copy the value of the setting of the current extruder to all other extruders as well as the global container.
  558. @pyqtSlot(str)
  559. def copyValueToExtruders(self, key):
  560. if not self._active_container_stack or self._global_container_stack.getProperty("machine_extruder_count", "value") <= 1:
  561. return
  562. new_value = self._active_container_stack.getProperty(key, "value")
  563. stacks = [stack for stack in ExtruderManager.getInstance().getMachineExtruders(self._global_container_stack.getId())]
  564. stacks.append(self._global_container_stack)
  565. for extruder_stack in stacks:
  566. if extruder_stack != self._active_container_stack and extruder_stack.getProperty(key, "value") != new_value:
  567. extruder_stack.getTop().setProperty(key, "value", new_value)
  568. ## Set the active material by switching out a container
  569. # Depending on from/to material+current variant, a quality profile is chosen and set.
  570. @pyqtSlot(str)
  571. def setActiveMaterial(self, material_id):
  572. with postponeSignals(*self._getContainerChangedSignals(), compress = True):
  573. containers = ContainerRegistry.getInstance().findInstanceContainers(id = material_id)
  574. if not containers or not self._active_container_stack:
  575. return
  576. material_container = containers[0]
  577. Logger.log("d", "Attempting to change the active material to %s", material_id)
  578. old_material = self._active_container_stack.material
  579. old_quality = self._active_container_stack.quality
  580. old_quality_changes = self._active_container_stack.qualityChanges
  581. if not old_material:
  582. Logger.log("w", "While trying to set the active material, no material was found to replace it.")
  583. return
  584. if old_quality_changes and old_quality_changes.getId() == "empty_quality_changes":
  585. old_quality_changes = None
  586. self.blurSettings.emit()
  587. old_material.nameChanged.disconnect(self._onMaterialNameChanged)
  588. self._active_container_stack.material = material_container
  589. Logger.log("d", "Active material changed")
  590. material_container.nameChanged.connect(self._onMaterialNameChanged)
  591. if material_container.getMetaDataEntry("compatible") == False:
  592. self._material_incompatible_message.show()
  593. else:
  594. self._material_incompatible_message.hide()
  595. quality_type = None
  596. new_quality_id = None
  597. if old_quality:
  598. new_quality_id = old_quality.getId()
  599. quality_type = old_quality.getMetaDataEntry("quality_type")
  600. if old_quality_changes:
  601. quality_type = old_quality_changes.getMetaDataEntry("quality_type")
  602. new_quality_id = old_quality_changes.getId()
  603. # See if the requested quality type is available in the new situation.
  604. machine_definition = self._active_container_stack.getBottom()
  605. quality_manager = QualityManager.getInstance()
  606. candidate_quality = None
  607. if quality_type:
  608. candidate_quality = quality_manager.findQualityByQualityType(quality_type,
  609. quality_manager.getWholeMachineDefinition(machine_definition),
  610. [material_container])
  611. if not candidate_quality or candidate_quality.getId() == "empty_quality":
  612. # Fall back to a quality (which must be compatible with all other extruders)
  613. new_qualities = quality_manager.findAllUsableQualitiesForMachineAndExtruders(
  614. self._global_container_stack, ExtruderManager.getInstance().getExtruderStacks())
  615. if new_qualities:
  616. new_quality_id = new_qualities[0].getId() # Just pick the first available one
  617. else:
  618. Logger.log("w", "No quality profile found that matches the current machine and extruders.")
  619. else:
  620. if not old_quality_changes:
  621. new_quality_id = candidate_quality.getId()
  622. self.setActiveQuality(new_quality_id)
  623. @pyqtSlot(str)
  624. def setActiveVariant(self, variant_id):
  625. with postponeSignals(*self._getContainerChangedSignals(), compress = True):
  626. containers = ContainerRegistry.getInstance().findInstanceContainers(id = variant_id)
  627. if not containers or not self._active_container_stack:
  628. return
  629. Logger.log("d", "Attempting to change the active variant to %s", variant_id)
  630. old_variant = self._active_container_stack.variant
  631. old_material = self._active_container_stack.material
  632. if old_variant:
  633. self.blurSettings.emit()
  634. self._active_container_stack.variant = containers[0]
  635. Logger.log("d", "Active variant changed to {active_variant_id}".format(active_variant_id = containers[0].getId()))
  636. preferred_material = None
  637. if old_material:
  638. preferred_material_name = old_material.getName()
  639. self.setActiveMaterial(self._updateMaterialContainer(self._global_container_stack.getBottom(), self._global_container_stack, containers[0], preferred_material_name).id)
  640. else:
  641. Logger.log("w", "While trying to set the active variant, no variant was found to replace.")
  642. ## set the active quality
  643. # \param quality_id The quality_id of either a quality or a quality_changes
  644. @pyqtSlot(str)
  645. def setActiveQuality(self, quality_id):
  646. with postponeSignals(*self._getContainerChangedSignals(), compress = True):
  647. self.blurSettings.emit()
  648. containers = ContainerRegistry.getInstance().findInstanceContainers(id = quality_id)
  649. if not containers or not self._global_container_stack:
  650. return
  651. Logger.log("d", "Attempting to change the active quality to %s", quality_id)
  652. # Quality profile come in two flavours: type=quality and type=quality_changes
  653. # If we found a quality_changes profile then look up its parent quality profile.
  654. container_type = containers[0].getMetaDataEntry("type")
  655. quality_name = containers[0].getName()
  656. quality_type = containers[0].getMetaDataEntry("quality_type")
  657. # Get quality container and optionally the quality_changes container.
  658. if container_type == "quality":
  659. new_quality_settings_list = self.determineQualityAndQualityChangesForQualityType(quality_type)
  660. elif container_type == "quality_changes":
  661. new_quality_settings_list = self._determineQualityAndQualityChangesForQualityChanges(quality_name)
  662. else:
  663. Logger.log("e", "Tried to set quality to a container that is not of the right type")
  664. return
  665. # Check if it was at all possible to find new settings
  666. if new_quality_settings_list is None:
  667. return
  668. name_changed_connect_stacks = [] # Connect these stacks to the name changed callback
  669. for setting_info in new_quality_settings_list:
  670. stack = setting_info["stack"]
  671. stack_quality = setting_info["quality"]
  672. stack_quality_changes = setting_info["quality_changes"]
  673. name_changed_connect_stacks.append(stack_quality)
  674. name_changed_connect_stacks.append(stack_quality_changes)
  675. self._replaceQualityOrQualityChangesInStack(stack, stack_quality, postpone_emit=True)
  676. self._replaceQualityOrQualityChangesInStack(stack, stack_quality_changes, postpone_emit=True)
  677. # Send emits that are postponed in replaceContainer.
  678. # Here the stacks are finished replacing and every value can be resolved based on the current state.
  679. for setting_info in new_quality_settings_list:
  680. setting_info["stack"].sendPostponedEmits()
  681. # Connect to onQualityNameChanged
  682. for stack in name_changed_connect_stacks:
  683. stack.nameChanged.connect(self._onQualityNameChanged)
  684. if self.hasUserSettings and Preferences.getInstance().getValue("cura/active_mode") == 1:
  685. self._askUserToKeepOrClearCurrentSettings()
  686. self.activeQualityChanged.emit()
  687. ## Determine the quality and quality changes settings for the current machine for a quality name.
  688. #
  689. # \param quality_name \type{str} the name of the quality.
  690. # \return \type{List[Dict]} with keys "stack", "quality" and "quality_changes".
  691. @UM.FlameProfiler.profile
  692. def determineQualityAndQualityChangesForQualityType(self, quality_type):
  693. quality_manager = QualityManager.getInstance()
  694. result = []
  695. empty_quality_changes = self._empty_quality_changes_container
  696. global_container_stack = self._global_container_stack
  697. if not global_container_stack:
  698. return []
  699. global_machine_definition = quality_manager.getParentMachineDefinition(global_container_stack.getBottom())
  700. extruder_stacks = ExtruderManager.getInstance().getActiveExtruderStacks()
  701. if extruder_stacks:
  702. stacks = extruder_stacks
  703. else:
  704. stacks = [global_container_stack]
  705. for stack in stacks:
  706. material = stack.material
  707. quality = quality_manager.findQualityByQualityType(quality_type, global_machine_definition, [material])
  708. if not quality: #No quality profile is found for this quality type.
  709. quality = self._empty_quality_container
  710. result.append({"stack": stack, "quality": quality, "quality_changes": empty_quality_changes})
  711. if extruder_stacks:
  712. # Add an extra entry for the global stack.
  713. global_quality = quality_manager.findQualityByQualityType(quality_type, global_machine_definition, [], global_quality = "True")
  714. if not global_quality:
  715. global_quality = self._empty_quality_container
  716. result.append({"stack": global_container_stack, "quality": global_quality, "quality_changes": empty_quality_changes})
  717. return result
  718. ## Determine the quality and quality changes settings for the current machine for a quality changes name.
  719. #
  720. # \param quality_changes_name \type{str} the name of the quality changes.
  721. # \return \type{List[Dict]} with keys "stack", "quality" and "quality_changes".
  722. def _determineQualityAndQualityChangesForQualityChanges(self, quality_changes_name):
  723. result = []
  724. quality_manager = QualityManager.getInstance()
  725. global_container_stack = self._global_container_stack
  726. global_machine_definition = quality_manager.getParentMachineDefinition(global_container_stack.getBottom())
  727. quality_changes_profiles = quality_manager.findQualityChangesByName(quality_changes_name,
  728. global_machine_definition)
  729. global_quality_changes = [qcp for qcp in quality_changes_profiles if qcp.getMetaDataEntry("extruder") is None]
  730. if global_quality_changes:
  731. global_quality_changes = global_quality_changes[0]
  732. else:
  733. Logger.log("e", "Could not find the global quality changes container with name %s", quality_changes_name)
  734. return None
  735. material = global_container_stack.material
  736. # For the global stack, find a quality which matches the quality_type in
  737. # the quality changes profile and also satisfies any material constraints.
  738. quality_type = global_quality_changes.getMetaDataEntry("quality_type")
  739. if global_container_stack.getProperty("machine_extruder_count", "value") > 1:
  740. global_quality = quality_manager.findQualityByQualityType(quality_type, global_machine_definition, [], global_quality = True)
  741. else:
  742. global_quality = quality_manager.findQualityByQualityType(quality_type, global_machine_definition, [material])
  743. # Find the values for each extruder.
  744. extruder_stacks = ExtruderManager.getInstance().getActiveExtruderStacks()
  745. for stack in extruder_stacks:
  746. extruder_definition = quality_manager.getParentMachineDefinition(stack.getBottom())
  747. quality_changes_list = [qcp for qcp in quality_changes_profiles
  748. if qcp.getMetaDataEntry("extruder") == extruder_definition.getId()]
  749. if quality_changes_list:
  750. quality_changes = quality_changes_list[0]
  751. else:
  752. quality_changes = global_quality_changes
  753. material = stack.material
  754. quality = quality_manager.findQualityByQualityType(quality_type, global_machine_definition, [material])
  755. if not quality: #No quality profile found for this quality type.
  756. quality = self._empty_quality_container
  757. result.append({"stack": stack, "quality": quality, "quality_changes": quality_changes})
  758. if extruder_stacks:
  759. global_quality = quality_manager.findQualityByQualityType(quality_type, global_machine_definition, [material], global_quality = "True")
  760. if not global_quality:
  761. global_quality = self._empty_quality_container
  762. result.append({"stack": global_container_stack, "quality": global_quality, "quality_changes": global_quality_changes})
  763. else:
  764. result.append({"stack": global_container_stack, "quality": global_quality, "quality_changes": global_quality_changes})
  765. return result
  766. def _replaceQualityOrQualityChangesInStack(self, stack, container, postpone_emit = False):
  767. # Disconnect the signal handling from the old container.
  768. container_type = container.getMetaDataEntry("type")
  769. if container_type == "quality":
  770. stack.quality.nameChanged.disconnect(self._onQualityNameChanged)
  771. stack.setQuality(container, postpone_emit = postpone_emit)
  772. stack.qualityChanges.nameChanged.connect(self._onQualityNameChanged)
  773. elif container_type == "quality_changes" or container_type is None:
  774. # If the container is an empty container, we need to change the quality_changes.
  775. # Quality can never be set to empty.
  776. stack.qualityChanges.nameChanged.disconnect(self._onQualityNameChanged)
  777. stack.setQualityChanges(container, postpone_emit = postpone_emit)
  778. stack.qualityChanges.nameChanged.connect(self._onQualityNameChanged)
  779. self._onQualityNameChanged()
  780. def _askUserToKeepOrClearCurrentSettings(self):
  781. Application.getInstance().discardOrKeepProfileChanges()
  782. @pyqtProperty(str, notify = activeVariantChanged)
  783. def activeVariantName(self):
  784. if self._active_container_stack:
  785. variant = self._active_container_stack.variant
  786. if variant:
  787. return variant.getName()
  788. return ""
  789. @pyqtProperty(str, notify = activeVariantChanged)
  790. def activeVariantId(self):
  791. if self._active_container_stack:
  792. variant = self._active_container_stack.variant
  793. if variant:
  794. return variant.getId()
  795. return ""
  796. @pyqtProperty(str, notify = globalContainerChanged)
  797. def activeDefinitionId(self):
  798. if self._global_container_stack:
  799. definition = self._global_container_stack.getBottom()
  800. if definition:
  801. return definition.id
  802. return ""
  803. @pyqtProperty(str, notify=globalContainerChanged)
  804. def activeDefinitionName(self):
  805. if self._global_container_stack:
  806. definition = self._global_container_stack.getBottom()
  807. if definition:
  808. return definition.getName()
  809. return ""
  810. ## Get the Definition ID to use to select quality profiles for the currently active machine
  811. # \returns DefinitionID (string) if found, empty string otherwise
  812. # \sa getQualityDefinitionId
  813. @pyqtProperty(str, notify = globalContainerChanged)
  814. def activeQualityDefinitionId(self):
  815. if self._global_container_stack:
  816. return self.getQualityDefinitionId(self._global_container_stack.getBottom())
  817. return ""
  818. ## Get the Definition ID to use to select quality profiles for machines of the specified definition
  819. # This is normally the id of the definition itself, but machines can specify a different definition to inherit qualities from
  820. # \param definition (DefinitionContainer) machine definition
  821. # \returns DefinitionID (string) if found, empty string otherwise
  822. def getQualityDefinitionId(self, definition):
  823. return QualityManager.getInstance().getParentMachineDefinition(definition).getId()
  824. ## Get the Variant ID to use to select quality profiles for the currently active variant
  825. # \returns VariantID (string) if found, empty string otherwise
  826. # \sa getQualityVariantId
  827. @pyqtProperty(str, notify = activeVariantChanged)
  828. def activeQualityVariantId(self):
  829. if self._active_container_stack:
  830. variant = self._active_container_stack.variant
  831. if variant:
  832. return self.getQualityVariantId(self._global_container_stack.getBottom(), variant)
  833. return ""
  834. ## Get the Variant ID to use to select quality profiles for variants of the specified definitions
  835. # This is normally the id of the variant itself, but machines can specify a different definition
  836. # to inherit qualities from, which has consequences for the variant to use as well
  837. # \param definition (DefinitionContainer) machine definition
  838. # \param variant (DefinitionContainer) variant definition
  839. # \returns VariantID (string) if found, empty string otherwise
  840. def getQualityVariantId(self, definition, variant):
  841. variant_id = variant.getId()
  842. definition_id = definition.getId()
  843. quality_definition_id = self.getQualityDefinitionId(definition)
  844. if definition_id != quality_definition_id:
  845. variant_id = variant_id.replace(definition_id, quality_definition_id, 1)
  846. return variant_id
  847. ## Gets how the active definition calls variants
  848. # Caveat: per-definition-variant-title is currently not translated (though the fallback is)
  849. @pyqtProperty(str, notify = globalContainerChanged)
  850. def activeDefinitionVariantsName(self):
  851. fallback_title = catalog.i18nc("@label", "Nozzle")
  852. if self._global_container_stack:
  853. return self._global_container_stack.getBottom().getMetaDataEntry("variants_name", fallback_title)
  854. return fallback_title
  855. @pyqtSlot(str, str)
  856. def renameMachine(self, machine_id, new_name):
  857. containers = ContainerRegistry.getInstance().findContainerStacks(id = machine_id)
  858. if containers:
  859. new_name = self._createUniqueName("machine", containers[0].getName(), new_name, containers[0].getBottom().getName())
  860. containers[0].setName(new_name)
  861. self.globalContainerChanged.emit()
  862. @pyqtSlot(str)
  863. def removeMachine(self, machine_id):
  864. # If the machine that is being removed is the currently active machine, set another machine as the active machine.
  865. activate_new_machine = (self._global_container_stack and self._global_container_stack.getId() == machine_id)
  866. ExtruderManager.getInstance().removeMachineExtruders(machine_id)
  867. containers = ContainerRegistry.getInstance().findInstanceContainers(type = "user", machine = machine_id)
  868. for container in containers:
  869. ContainerRegistry.getInstance().removeContainer(container.getId())
  870. ContainerRegistry.getInstance().removeContainer(machine_id)
  871. if activate_new_machine:
  872. stacks = ContainerRegistry.getInstance().findContainerStacks(type = "machine")
  873. if stacks:
  874. Application.getInstance().setGlobalContainerStack(stacks[0])
  875. @pyqtProperty(bool, notify = globalContainerChanged)
  876. def hasMaterials(self):
  877. if self._global_container_stack:
  878. return bool(self._global_container_stack.getMetaDataEntry("has_materials", False))
  879. return False
  880. @pyqtProperty(bool, notify = globalContainerChanged)
  881. def hasVariants(self):
  882. if self._global_container_stack:
  883. return bool(self._global_container_stack.getMetaDataEntry("has_variants", False))
  884. return False
  885. ## Property to indicate if a machine has "specialized" material profiles.
  886. # Some machines have their own material profiles that "override" the default catch all profiles.
  887. @pyqtProperty(bool, notify = globalContainerChanged)
  888. def filterMaterialsByMachine(self):
  889. if self._global_container_stack:
  890. return bool(self._global_container_stack.getMetaDataEntry("has_machine_materials", False))
  891. return False
  892. ## Property to indicate if a machine has "specialized" quality profiles.
  893. # Some machines have their own quality profiles that "override" the default catch all profiles.
  894. @pyqtProperty(bool, notify = globalContainerChanged)
  895. def filterQualityByMachine(self):
  896. if self._global_container_stack:
  897. return bool(self._global_container_stack.getMetaDataEntry("has_machine_quality", False))
  898. return False
  899. ## Get the Definition ID of a machine (specified by ID)
  900. # \param machine_id string machine id to get the definition ID of
  901. # \returns DefinitionID (string) if found, None otherwise
  902. @pyqtSlot(str, result = str)
  903. def getDefinitionByMachineId(self, machine_id):
  904. containers = ContainerRegistry.getInstance().findContainerStacks(id=machine_id)
  905. if containers:
  906. return containers[0].getBottom().getId()
  907. @staticmethod
  908. def createMachineManager(engine=None, script_engine=None):
  909. return MachineManager()
  910. def _updateVariantContainer(self, definition: "DefinitionContainer"):
  911. if not definition.getMetaDataEntry("has_variants"):
  912. return self._empty_variant_container
  913. machine_definition_id = Application.getInstance().getMachineManager().getQualityDefinitionId(definition)
  914. containers = []
  915. preferred_variant = definition.getMetaDataEntry("preferred_variant")
  916. if preferred_variant:
  917. containers = ContainerRegistry.getInstance().findInstanceContainers(type = "variant", definition = machine_definition_id, id = preferred_variant)
  918. if not containers:
  919. containers = ContainerRegistry.getInstance().findInstanceContainers(type = "variant", definition = machine_definition_id)
  920. if containers:
  921. return containers[0]
  922. return self._empty_variant_container
  923. def _updateMaterialContainer(self, definition: "DefinitionContainer", stack: "ContainerStack", variant_container: Optional["InstanceContainer"] = None, preferred_material_name: Optional[str] = None):
  924. if not definition.getMetaDataEntry("has_materials"):
  925. return self._empty_material_container
  926. approximate_material_diameter = round(stack.getProperty("material_diameter", "value"))
  927. search_criteria = { "type": "material", "approximate_diameter": approximate_material_diameter }
  928. if definition.getMetaDataEntry("has_machine_materials"):
  929. search_criteria["definition"] = self.getQualityDefinitionId(definition)
  930. if definition.getMetaDataEntry("has_variants") and variant_container:
  931. search_criteria["variant"] = self.getQualityVariantId(definition, variant_container)
  932. else:
  933. search_criteria["definition"] = "fdmprinter"
  934. if preferred_material_name:
  935. search_criteria["name"] = preferred_material_name
  936. else:
  937. preferred_material = definition.getMetaDataEntry("preferred_material")
  938. if preferred_material:
  939. search_criteria["id"] = preferred_material
  940. containers = ContainerRegistry.getInstance().findInstanceContainers(**search_criteria)
  941. if containers:
  942. return containers[0]
  943. if "variant" in search_criteria or "id" in search_criteria:
  944. # If a material by this name can not be found, try a wider set of search criteria
  945. search_criteria.pop("variant", None)
  946. search_criteria.pop("id", None)
  947. containers = ContainerRegistry.getInstance().findInstanceContainers(**search_criteria)
  948. if containers:
  949. return containers[0]
  950. Logger.log("w", "Unable to find a material container with provided criteria, returning an empty one instead.")
  951. return self._empty_material_container
  952. def _updateQualityContainer(self, definition: "DefinitionContainer", variant_container: "ContainerStack", material_container = None, preferred_quality_name: Optional[str] = None):
  953. container_registry = ContainerRegistry.getInstance()
  954. search_criteria = { "type": "quality" }
  955. if definition.getMetaDataEntry("has_machine_quality"):
  956. search_criteria["definition"] = self.getQualityDefinitionId(definition)
  957. if definition.getMetaDataEntry("has_materials") and material_container:
  958. search_criteria["material"] = material_container.id
  959. else:
  960. search_criteria["definition"] = "fdmprinter"
  961. if preferred_quality_name and preferred_quality_name != "empty":
  962. search_criteria["name"] = preferred_quality_name
  963. else:
  964. preferred_quality = definition.getMetaDataEntry("preferred_quality")
  965. if preferred_quality:
  966. search_criteria["id"] = preferred_quality
  967. containers = container_registry.findInstanceContainers(**search_criteria)
  968. if containers:
  969. return containers[0]
  970. if "material" in search_criteria:
  971. # First check if we can solve our material not found problem by checking if we can find quality containers
  972. # that are assigned to the parents of this material profile.
  973. try:
  974. inherited_files = material_container.getInheritedFiles()
  975. except AttributeError: # Material_container does not support inheritance.
  976. inherited_files = []
  977. if inherited_files:
  978. for inherited_file in inherited_files:
  979. # Extract the ID from the path we used to load the file.
  980. search_criteria["material"] = os.path.basename(inherited_file).split(".")[0]
  981. containers = container_registry.findInstanceContainers(**search_criteria)
  982. if containers:
  983. return containers[0]
  984. # We still weren't able to find a quality for this specific material.
  985. # Try to find qualities for a generic version of the material.
  986. material_search_criteria = { "type": "material", "material": material_container.getMetaDataEntry("material"), "color_name": "Generic"}
  987. if definition.getMetaDataEntry("has_machine_quality"):
  988. if material_container:
  989. material_search_criteria["definition"] = material_container.getDefinition().id
  990. if definition.getMetaDataEntry("has_variants"):
  991. material_search_criteria["variant"] = material_container.getMetaDataEntry("variant")
  992. else:
  993. material_search_criteria["definition"] = self.getQualityDefinitionId(definition)
  994. if definition.getMetaDataEntry("has_variants") and variant_container:
  995. material_search_criteria["variant"] = self.getQualityVariantId(definition, variant_container)
  996. else:
  997. material_search_criteria["definition"] = "fdmprinter"
  998. material_containers = container_registry.findInstanceContainers(**material_search_criteria)
  999. # Try all materials to see if there is a quality profile available.
  1000. for material_container in material_containers:
  1001. search_criteria["material"] = material_container.getId()
  1002. containers = container_registry.findInstanceContainers(**search_criteria)
  1003. if containers:
  1004. return containers[0]
  1005. if "name" in search_criteria or "id" in search_criteria:
  1006. # If a quality by this name can not be found, try a wider set of search criteria
  1007. search_criteria.pop("name", None)
  1008. search_criteria.pop("id", None)
  1009. containers = container_registry.findInstanceContainers(**search_criteria)
  1010. if containers:
  1011. return containers[0]
  1012. # Notify user that we were unable to find a matching quality
  1013. message = Message(catalog.i18nc("@info:status", "Unable to find a quality profile for this combination. Default settings will be used instead."))
  1014. message.show()
  1015. return self._empty_quality_container
  1016. ## Finds a quality-changes container to use if any other container
  1017. # changes.
  1018. #
  1019. # \param quality_type The quality type to find a quality-changes for.
  1020. # \param preferred_quality_changes_name The name of the quality-changes to
  1021. # pick, if any such quality-changes profile is available.
  1022. def _updateQualityChangesContainer(self, quality_type, preferred_quality_changes_name = None):
  1023. container_registry = ContainerRegistry.getInstance() # Cache.
  1024. search_criteria = { "type": "quality_changes" }
  1025. search_criteria["quality"] = quality_type
  1026. if preferred_quality_changes_name:
  1027. search_criteria["name"] = preferred_quality_changes_name
  1028. # Try to search with the name in the criteria first, since we prefer to have the correct name.
  1029. containers = container_registry.findInstanceContainers(**search_criteria)
  1030. if containers: # Found one!
  1031. return containers[0]
  1032. if "name" in search_criteria:
  1033. del search_criteria["name"] # Not found, then drop the name requirement (if we had one) and search again.
  1034. containers = container_registry.findInstanceContainers(**search_criteria)
  1035. if containers:
  1036. return containers[0]
  1037. return self._empty_quality_changes_container # Didn't find anything with the required quality_type.
  1038. def _onMachineNameChanged(self):
  1039. self.globalContainerChanged.emit()
  1040. def _onMaterialNameChanged(self):
  1041. self.activeMaterialChanged.emit()
  1042. def _onQualityNameChanged(self):
  1043. self.activeQualityChanged.emit()
  1044. def _getContainerChangedSignals(self):
  1045. stacks = ExtruderManager.getInstance().getActiveExtruderStacks()
  1046. stacks.append(self._global_container_stack)
  1047. return [ s.containersChanged for s in stacks ]