MachineManager.py 57 KB

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