MachineManager.py 39 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837
  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.Application import Application
  6. from UM.Preferences import Preferences
  7. from UM.Logger import Logger
  8. import UM.Settings
  9. from cura.PrinterOutputDevice import PrinterOutputDevice
  10. from . import ExtruderManager
  11. from UM.i18n import i18nCatalog
  12. catalog = i18nCatalog("cura")
  13. import time
  14. class MachineManager(QObject):
  15. def __init__(self, parent = None):
  16. super().__init__(parent)
  17. self._active_container_stack = None
  18. self._global_container_stack = None
  19. Application.getInstance().globalContainerStackChanged.connect(self._onGlobalContainerChanged)
  20. ## When the global container is changed, active material probably needs to be updated.
  21. self.globalContainerChanged.connect(self.activeMaterialChanged)
  22. self.globalContainerChanged.connect(self.activeVariantChanged)
  23. self.globalContainerChanged.connect(self.activeQualityChanged)
  24. self._active_stack_valid = None
  25. self._onGlobalContainerChanged()
  26. ExtruderManager.getInstance().activeExtruderChanged.connect(self._onActiveExtruderStackChanged)
  27. self._onActiveExtruderStackChanged()
  28. ExtruderManager.getInstance().activeExtruderChanged.connect(self.activeMaterialChanged)
  29. ExtruderManager.getInstance().activeExtruderChanged.connect(self.activeVariantChanged)
  30. ExtruderManager.getInstance().activeExtruderChanged.connect(self.activeQualityChanged)
  31. self.globalContainerChanged.connect(self.activeStackChanged)
  32. self.globalValueChanged.connect(self.activeStackChanged)
  33. ExtruderManager.getInstance().activeExtruderChanged.connect(self.activeStackChanged)
  34. self._empty_variant_container = UM.Settings.ContainerRegistry.getInstance().findInstanceContainers(id = "empty_variant")[0]
  35. self._empty_material_container = UM.Settings.ContainerRegistry.getInstance().findInstanceContainers(id = "empty_material")[0]
  36. self._empty_quality_container = UM.Settings.ContainerRegistry.getInstance().findInstanceContainers(id = "empty_quality")[0]
  37. self._empty_quality_changes_container = UM.Settings.ContainerRegistry.getInstance().findInstanceContainers(id = "empty_quality_changes")[0]
  38. Preferences.getInstance().addPreference("cura/active_machine", "")
  39. self._global_event_keys = set()
  40. active_machine_id = Preferences.getInstance().getValue("cura/active_machine")
  41. self._printer_output_devices = []
  42. Application.getInstance().getOutputDeviceManager().outputDevicesChanged.connect(self._onOutputDevicesChanged)
  43. if active_machine_id != "":
  44. # An active machine was saved, so restore it.
  45. self.setActiveMachine(active_machine_id)
  46. if self._global_container_stack and self._global_container_stack.getProperty("machine_extruder_count", "value") > 1:
  47. # Make sure _active_container_stack is properly initiated
  48. ExtruderManager.getInstance().setActiveExtruderIndex(0)
  49. self._auto_materials_changed = {}
  50. self._auto_hotends_changed = {}
  51. globalContainerChanged = pyqtSignal()
  52. activeMaterialChanged = pyqtSignal()
  53. activeVariantChanged = pyqtSignal()
  54. activeQualityChanged = pyqtSignal()
  55. activeStackChanged = pyqtSignal()
  56. globalValueChanged = pyqtSignal() # Emitted whenever a value inside global container is changed.
  57. activeValidationChanged = pyqtSignal() # Emitted whenever a validation inside active container is changed
  58. blurSettings = pyqtSignal() # Emitted to force fields in the advanced sidebar to un-focus, so they update properly
  59. outputDevicesChanged = pyqtSignal()
  60. def _onOutputDevicesChanged(self):
  61. for printer_output_device in self._printer_output_devices:
  62. printer_output_device.hotendIdChanged.disconnect(self._onHotendIdChanged)
  63. printer_output_device.materialIdChanged.disconnect(self._onMaterialIdChanged)
  64. self._printer_output_devices.clear()
  65. for printer_output_device in Application.getInstance().getOutputDeviceManager().getOutputDevices():
  66. if isinstance(printer_output_device, PrinterOutputDevice):
  67. self._printer_output_devices.append(printer_output_device)
  68. printer_output_device.hotendIdChanged.connect(self._onHotendIdChanged)
  69. printer_output_device.materialIdChanged.connect(self._onMaterialIdChanged)
  70. self.outputDevicesChanged.emit()
  71. @pyqtProperty("QVariantList", notify = outputDevicesChanged)
  72. def printerOutputDevices(self):
  73. return self._printer_output_devices
  74. def _onHotendIdChanged(self, index, hotend_id):
  75. if not self._global_container_stack:
  76. return
  77. containers = UM.Settings.ContainerRegistry.getInstance().findInstanceContainers(type="variant", definition=self._global_container_stack.getBottom().getId(), name=hotend_id)
  78. if containers: # New material ID is known
  79. extruder_manager = ExtruderManager.getInstance()
  80. extruders = list(extruder_manager.getMachineExtruders(self.activeMachineId))
  81. matching_extruder = None
  82. for extruder in extruders:
  83. if str(index) == extruder.getMetaDataEntry("position"):
  84. matching_extruder = extruder
  85. break
  86. if matching_extruder and matching_extruder.findContainer({"type": "variant"}).getName() != hotend_id:
  87. # Save the material that needs to be changed. Multiple changes will be handled by the callback.
  88. self._auto_hotends_changed[str(index)] = containers[0].getId()
  89. self._printer_output_devices[0].materialHotendChangedMessage(self._materialHotendChangedCallback)
  90. else:
  91. Logger.log("w", "No variant found for printer definition %s with id %s" % (self._global_container_stack.getBottom().getId(), hotend_id))
  92. def _autoUpdateHotends(self):
  93. extruder_manager = ExtruderManager.getInstance()
  94. for position in self._auto_hotends_changed:
  95. hotend_id = self._auto_hotends_changed[position]
  96. old_index = extruder_manager.activeExtruderIndex
  97. if old_index != int(position):
  98. extruder_manager.setActiveExtruderIndex(int(position))
  99. else:
  100. old_index = None
  101. Logger.log("d", "Setting hotend variant of hotend %s to %s" % (position, hotend_id))
  102. self.setActiveVariant(hotend_id)
  103. if old_index is not None:
  104. extruder_manager.setActiveExtruderIndex(old_index)
  105. def _onMaterialIdChanged(self, index, material_id):
  106. if not self._global_container_stack:
  107. return
  108. definition_id = "fdmprinter"
  109. if self._global_container_stack.getMetaDataEntry("has_machine_materials", False):
  110. definition_id = self._global_container_stack.getBottom().getId()
  111. extruder_manager = ExtruderManager.getInstance()
  112. containers = UM.Settings.ContainerRegistry.getInstance().findInstanceContainers(type = "material", definition = definition_id, GUID = material_id)
  113. if containers: # New material ID is known
  114. extruders = list(extruder_manager.getMachineExtruders(self.activeMachineId))
  115. matching_extruder = None
  116. for extruder in extruders:
  117. if str(index) == extruder.getMetaDataEntry("position"):
  118. matching_extruder = extruder
  119. break
  120. if matching_extruder and matching_extruder.findContainer({"type":"material"}).getMetaDataEntry("GUID") != material_id:
  121. # Save the material that needs to be changed. Multiple changes will be handled by the callback.
  122. self._auto_materials_changed[str(index)] = containers[0].getId()
  123. self._printer_output_devices[0].materialHotendChangedMessage(self._materialHotendChangedCallback)
  124. else:
  125. Logger.log("w", "No material definition found for printer definition %s and GUID %s" % (definition_id, material_id))
  126. def _materialHotendChangedCallback(self, button):
  127. if button == QMessageBox.No:
  128. self._auto_materials_changed = {}
  129. self._auto_hotends_changed = {}
  130. return
  131. self._autoUpdateMaterials()
  132. self._autoUpdateHotends()
  133. def _autoUpdateMaterials(self):
  134. extruder_manager = ExtruderManager.getInstance()
  135. for position in self._auto_materials_changed:
  136. material_id = self._auto_materials_changed[position]
  137. old_index = extruder_manager.activeExtruderIndex
  138. if old_index != int(position):
  139. extruder_manager.setActiveExtruderIndex(int(position))
  140. else:
  141. old_index = None
  142. Logger.log("d", "Setting material of hotend %s to %s" % (position, material_id))
  143. self.setActiveMaterial(material_id)
  144. if old_index is not None:
  145. extruder_manager.setActiveExtruderIndex(old_index)
  146. def _onGlobalContainerChanged(self):
  147. if self._global_container_stack:
  148. self._global_container_stack.nameChanged.disconnect(self._onMachineNameChanged)
  149. self._global_container_stack.containersChanged.disconnect(self._onInstanceContainersChanged)
  150. self._global_container_stack.propertyChanged.disconnect(self._onPropertyChanged)
  151. material = self._global_container_stack.findContainer({"type": "material"})
  152. material.nameChanged.disconnect(self._onMaterialNameChanged)
  153. quality = self._global_container_stack.findContainer({"type": "quality"})
  154. quality.nameChanged.disconnect(self._onQualityNameChanged)
  155. self._global_container_stack = Application.getInstance().getGlobalContainerStack()
  156. self._active_container_stack = self._global_container_stack
  157. self.globalContainerChanged.emit()
  158. if self._global_container_stack:
  159. Preferences.getInstance().setValue("cura/active_machine", self._global_container_stack.getId())
  160. self._global_container_stack.nameChanged.connect(self._onMachineNameChanged)
  161. self._global_container_stack.containersChanged.connect(self._onInstanceContainersChanged)
  162. self._global_container_stack.propertyChanged.connect(self._onPropertyChanged)
  163. material = self._global_container_stack.findContainer({"type": "material"})
  164. material.nameChanged.connect(self._onMaterialNameChanged)
  165. quality = self._global_container_stack.findContainer({"type": "quality"})
  166. quality.nameChanged.connect(self._onQualityNameChanged)
  167. def _onActiveExtruderStackChanged(self):
  168. self.blurSettings.emit() # Ensure no-one has focus.
  169. if self._active_container_stack and self._active_container_stack != self._global_container_stack:
  170. self._active_container_stack.containersChanged.disconnect(self._onInstanceContainersChanged)
  171. self._active_container_stack.propertyChanged.disconnect(self._onPropertyChanged)
  172. self._active_container_stack = ExtruderManager.getInstance().getActiveExtruderStack()
  173. if self._active_container_stack:
  174. self._active_container_stack.containersChanged.connect(self._onInstanceContainersChanged)
  175. self._active_container_stack.propertyChanged.connect(self._onPropertyChanged)
  176. else:
  177. self._active_container_stack = self._global_container_stack
  178. self._active_stack_valid = not self._checkStackForErrors(self._active_container_stack)
  179. self.activeValidationChanged.emit()
  180. def _onInstanceContainersChanged(self, container):
  181. container_type = container.getMetaDataEntry("type")
  182. if container_type == "material":
  183. self.activeMaterialChanged.emit()
  184. elif container_type == "variant":
  185. self.activeVariantChanged.emit()
  186. elif container_type == "quality":
  187. self.activeQualityChanged.emit()
  188. def _onPropertyChanged(self, key, property_name):
  189. if property_name == "validationState":
  190. if self._active_stack_valid:
  191. if self._active_container_stack.getProperty(key, "settable_per_extruder"):
  192. changed_validation_state = self._active_container_stack.getProperty(key, property_name)
  193. else:
  194. changed_validation_state = self._global_container_stack.getProperty(key, property_name)
  195. if changed_validation_state in (UM.Settings.ValidatorState.Exception, UM.Settings.ValidatorState.MaximumError, UM.Settings.ValidatorState.MinimumError):
  196. self._active_stack_valid = False
  197. self.activeValidationChanged.emit()
  198. else:
  199. if not self._checkStackForErrors(self._active_container_stack) and not self._checkStackForErrors(self._global_container_stack):
  200. self._active_stack_valid = True
  201. self.activeValidationChanged.emit()
  202. self.activeStackChanged.emit()
  203. @pyqtSlot(str)
  204. def setActiveMachine(self, stack_id):
  205. containers = UM.Settings.ContainerRegistry.getInstance().findContainerStacks(id = stack_id)
  206. if containers:
  207. Application.getInstance().setGlobalContainerStack(containers[0])
  208. @pyqtSlot(str, str)
  209. def addMachine(self, name, definition_id):
  210. container_registry = UM.Settings.ContainerRegistry.getInstance()
  211. definitions = container_registry.findDefinitionContainers(id = definition_id)
  212. if definitions:
  213. definition = definitions[0]
  214. name = self._createUniqueName("machine", "", name, definition.getName())
  215. new_global_stack = UM.Settings.ContainerStack(name)
  216. new_global_stack.addMetaDataEntry("type", "machine")
  217. container_registry.addContainer(new_global_stack)
  218. variant_instance_container = self._updateVariantContainer(definition)
  219. material_instance_container = self._updateMaterialContainer(definition, variant_instance_container)
  220. quality_instance_container = self._updateQualityContainer(definition, variant_instance_container, material_instance_container)
  221. current_settings_instance_container = UM.Settings.InstanceContainer(name + "_current_settings")
  222. current_settings_instance_container.addMetaDataEntry("machine", name)
  223. current_settings_instance_container.addMetaDataEntry("type", "user")
  224. current_settings_instance_container.setDefinition(definitions[0])
  225. container_registry.addContainer(current_settings_instance_container)
  226. new_global_stack.addContainer(definition)
  227. if variant_instance_container:
  228. new_global_stack.addContainer(variant_instance_container)
  229. if material_instance_container:
  230. new_global_stack.addContainer(material_instance_container)
  231. if quality_instance_container:
  232. new_global_stack.addContainer(quality_instance_container)
  233. new_global_stack.addContainer(self._empty_quality_changes_container)
  234. new_global_stack.addContainer(current_settings_instance_container)
  235. ExtruderManager.getInstance().addMachineExtruders(definition, new_global_stack.getId())
  236. Application.getInstance().setGlobalContainerStack(new_global_stack)
  237. ## Create a name that is not empty and unique
  238. # \param container_type \type{string} Type of the container (machine, quality, ...)
  239. # \param current_name \type{} Current name of the container, which may be an acceptable option
  240. # \param new_name \type{string} Base name, which may not be unique
  241. # \param fallback_name \type{string} Name to use when (stripped) new_name is empty
  242. # \return \type{string} Name that is unique for the specified type and name/id
  243. def _createUniqueName(self, container_type, current_name, new_name, fallback_name):
  244. return UM.Settings.ContainerRegistry.getInstance().createUniqueName(container_type, current_name, new_name, fallback_name)
  245. ## Convenience function to check if a stack has errors.
  246. def _checkStackForErrors(self, stack):
  247. if stack is None:
  248. return False
  249. for key in stack.getAllKeys():
  250. validation_state = stack.getProperty(key, "validationState")
  251. if validation_state in (UM.Settings.ValidatorState.Exception, UM.Settings.ValidatorState.MaximumError, UM.Settings.ValidatorState.MinimumError):
  252. return True
  253. return False
  254. ## Remove all instances from the top instanceContainer (effectively removing all user-changed settings)
  255. @pyqtSlot()
  256. def clearUserSettings(self):
  257. if not self._active_container_stack:
  258. return
  259. self.blurSettings.emit()
  260. user_settings = self._active_container_stack.getTop()
  261. user_settings.clear()
  262. ## Check if the global_container has instances in the user container
  263. @pyqtProperty(bool, notify = activeStackChanged)
  264. def hasUserSettings(self):
  265. if not self._global_container_stack:
  266. return False
  267. if self._global_container_stack.getTop().findInstances():
  268. return True
  269. for stack in ExtruderManager.getInstance().getMachineExtruders(self._global_container_stack.getId()):
  270. if stack.getTop().findInstances():
  271. return True
  272. return False
  273. ## Check if the global profile does not contain error states
  274. # Note that the _active_stack_valid is cached due to performance issues
  275. # Calling _checkStackForErrors on every change is simply too expensive
  276. @pyqtProperty(bool, notify = activeValidationChanged)
  277. def isActiveStackValid(self):
  278. return bool(self._active_stack_valid)
  279. @pyqtProperty(str, notify = activeStackChanged)
  280. def activeUserProfileId(self):
  281. if self._active_container_stack:
  282. return self._active_container_stack.getTop().getId()
  283. return ""
  284. @pyqtProperty(str, notify = globalContainerChanged)
  285. def activeMachineName(self):
  286. if self._global_container_stack:
  287. return self._global_container_stack.getName()
  288. return ""
  289. @pyqtProperty(str, notify = globalContainerChanged)
  290. def activeMachineId(self):
  291. if self._global_container_stack:
  292. return self._global_container_stack.getId()
  293. return ""
  294. @pyqtProperty(str, notify = activeStackChanged)
  295. def activeStackId(self):
  296. if self._active_container_stack:
  297. return self._active_container_stack.getId()
  298. return ""
  299. @pyqtProperty(str, notify = activeMaterialChanged)
  300. def activeMaterialName(self):
  301. if self._active_container_stack:
  302. material = self._active_container_stack.findContainer({"type":"material"})
  303. if material:
  304. return material.getName()
  305. return ""
  306. @pyqtProperty(str, notify=activeMaterialChanged)
  307. def activeMaterialId(self):
  308. if self._active_container_stack:
  309. material = self._active_container_stack.findContainer({"type": "material"})
  310. if material:
  311. return material.getId()
  312. return ""
  313. @pyqtProperty("QVariantMap", notify = activeMaterialChanged)
  314. def allActiveMaterialIds(self):
  315. if not self._global_container_stack:
  316. return {}
  317. result = {}
  318. for stack in ExtruderManager.getInstance().getActiveGlobalAndExtruderStacks():
  319. material_container = stack.findContainer(type = "material")
  320. if not material_container:
  321. continue
  322. result[stack.getId()] = material_container.getId()
  323. return result
  324. @pyqtProperty(str, notify=activeQualityChanged)
  325. def activeQualityMaterialId(self):
  326. if self._active_container_stack:
  327. quality = self._active_container_stack.findContainer({"type": "quality"})
  328. if quality:
  329. return quality.getMetaDataEntry("material")
  330. return ""
  331. @pyqtProperty(str, notify=activeQualityChanged)
  332. def activeQualityName(self):
  333. if self._active_container_stack:
  334. quality = self._active_container_stack.findContainer({"type": "quality_changes"})
  335. if quality and quality != self._empty_quality_changes_container:
  336. return quality.getName()
  337. quality = self._active_container_stack.findContainer({"type": "quality"})
  338. if quality:
  339. return quality.getName()
  340. return ""
  341. @pyqtProperty(str, notify=activeQualityChanged)
  342. def activeQualityId(self):
  343. if self._global_container_stack:
  344. quality = self._global_container_stack.findContainer({"type": "quality_changes"})
  345. if quality and quality != self._empty_quality_changes_container:
  346. return quality.getId()
  347. quality = self._global_container_stack.findContainer({"type": "quality"})
  348. if quality:
  349. return quality.getId()
  350. return ""
  351. @pyqtProperty(str, notify = activeQualityChanged)
  352. def activeQualityType(self):
  353. if self._global_container_stack:
  354. quality = self._global_container_stack.findContainer(type = "quality")
  355. if quality:
  356. return quality.getMetaDataEntry("quality_type")
  357. return ""
  358. @pyqtProperty(str, notify = activeQualityChanged)
  359. def activeQualityChangesId(self):
  360. if self._global_container_stack:
  361. changes = self._global_container_stack.findContainer(type = "quality_changes")
  362. if changes:
  363. return changes.getId()
  364. return ""
  365. ## Check if a container is read_only
  366. @pyqtSlot(str, result = bool)
  367. def isReadOnly(self, container_id):
  368. containers = UM.Settings.ContainerRegistry.getInstance().findInstanceContainers(id = container_id)
  369. if not containers or not self._active_container_stack:
  370. return True
  371. return containers[0].isReadOnly()
  372. ## Copy the value of the setting of the current extruder to all other extruders as well as the global container.
  373. @pyqtSlot(str)
  374. def copyValueToExtruders(self, key):
  375. if not self._active_container_stack or self._global_container_stack.getProperty("machine_extruder_count", "value") <= 1:
  376. return
  377. new_value = self._active_container_stack.getProperty(key, "value")
  378. stacks = [stack for stack in ExtruderManager.getInstance().getMachineExtruders(self._global_container_stack.getId())]
  379. stacks.append(self._global_container_stack)
  380. for extruder_stack in stacks:
  381. if extruder_stack != self._active_container_stack and extruder_stack.getProperty(key, "value") != new_value:
  382. extruder_stack.getTop().setProperty(key, "value", new_value)
  383. @pyqtSlot(str)
  384. def setActiveMaterial(self, material_id):
  385. containers = UM.Settings.ContainerRegistry.getInstance().findInstanceContainers(id = material_id)
  386. if not containers or not self._active_container_stack:
  387. return
  388. old_variant = self._active_container_stack.findContainer({"type":"variant"})
  389. old_material = self._active_container_stack.findContainer({"type":"material"})
  390. old_quality = self._active_container_stack.findContainer({"type": "quality"})
  391. if old_material:
  392. old_material.nameChanged.disconnect(self._onMaterialNameChanged)
  393. material_index = self._active_container_stack.getContainerIndex(old_material)
  394. self._active_container_stack.replaceContainer(material_index, containers[0])
  395. containers[0].nameChanged.connect(self._onMaterialNameChanged)
  396. preferred_quality_name = None
  397. if old_quality:
  398. preferred_quality_name = old_quality.getName()
  399. self.setActiveQuality(self._updateQualityContainer(self._global_container_stack.getBottom(), old_variant, containers[0], preferred_quality_name).id)
  400. else:
  401. Logger.log("w", "While trying to set the active material, no material was found to replace.")
  402. @pyqtSlot(str)
  403. def setActiveVariant(self, variant_id):
  404. containers = UM.Settings.ContainerRegistry.getInstance().findInstanceContainers(id = variant_id)
  405. if not containers or not self._active_container_stack:
  406. return
  407. old_variant = self._active_container_stack.findContainer({"type": "variant"})
  408. old_material = self._active_container_stack.findContainer({"type": "material"})
  409. if old_variant:
  410. variant_index = self._active_container_stack.getContainerIndex(old_variant)
  411. self._active_container_stack.replaceContainer(variant_index, containers[0])
  412. preferred_material = None
  413. if old_material:
  414. preferred_material_name = old_material.getName()
  415. self.setActiveMaterial(self._updateMaterialContainer(self._global_container_stack.getBottom(), containers[0], preferred_material_name).id)
  416. else:
  417. Logger.log("w", "While trying to set the active variant, no variant was found to replace.")
  418. @pyqtSlot(str)
  419. def setActiveQuality(self, quality_id):
  420. containers = UM.Settings.ContainerRegistry.getInstance().findInstanceContainers(id = quality_id)
  421. if not containers or not self._global_container_stack:
  422. return
  423. quality_container = None
  424. quality_changes_container = self._empty_quality_changes_container
  425. container_type = containers[0].getMetaDataEntry("type")
  426. if container_type == "quality":
  427. quality_container = containers[0]
  428. elif container_type == "quality_changes":
  429. quality_changes_container = containers[0]
  430. containers = UM.Settings.ContainerRegistry.getInstance().findInstanceContainers(
  431. quality_type = quality_changes_container.getMetaDataEntry("quality"))
  432. if not containers:
  433. Logger.log("e", "Could not find quality %s for changes %s, not changing quality", quality_changes_container.getMetaDataEntry("quality"), quality_changes_container.getId())
  434. return
  435. quality_container = containers[0]
  436. else:
  437. Logger.log("e", "Tried to set quality to a container that is not of the right type")
  438. return
  439. quality_type = quality_container.getMetaDataEntry("quality_type")
  440. if not quality_type:
  441. quality_type = quality_changes_container.getName()
  442. for stack in ExtruderManager.getInstance().getActiveGlobalAndExtruderStacks():
  443. extruder_id = stack.getId() if stack != self._global_container_stack else None
  444. criteria = { "quality_type": quality_type, "extruder": extruder_id }
  445. if self._global_container_stack.getMetaDataEntry("has_machine_quality"):
  446. material = stack.findContainer(type = "material")
  447. criteria["material"] = material.getId()
  448. stack_quality = UM.Settings.ContainerRegistry.getInstance().findInstanceContainers(**criteria)
  449. if not stack_quality:
  450. criteria.pop("extruder")
  451. stack_quality = UM.Settings.ContainerRegistry.getInstance().findInstanceContainers(**criteria)
  452. if not stack_quality:
  453. stack_quality = quality_container
  454. else:
  455. stack_quality = stack_quality[0]
  456. else:
  457. stack_quality = stack_quality[0]
  458. if quality_changes_container != self._empty_quality_changes_container:
  459. stack_quality_changes = UM.Settings.ContainerRegistry.getInstance().findInstanceContainers(name = quality_changes_container.getName(), extruder = extruder_id)[0]
  460. else:
  461. stack_quality_changes = self._empty_quality_changes_container
  462. old_quality = stack.findContainer(type = "quality")
  463. old_quality.nameChanged.disconnect(self._onQualityNameChanged)
  464. old_changes = stack.findContainer(type = "quality_changes")
  465. old_changes.nameChanged.disconnect(self._onQualityNameChanged)
  466. stack.replaceContainer(stack.getContainerIndex(old_quality), stack_quality)
  467. stack.replaceContainer(stack.getContainerIndex(old_changes), stack_quality_changes)
  468. stack_quality.nameChanged.connect(self._onQualityNameChanged)
  469. stack_quality_changes.nameChanged.connect(self._onQualityNameChanged)
  470. if self.hasUserSettings and Preferences.getInstance().getValue("cura/active_mode") == 1:
  471. # Ask the user if the user profile should be cleared or not (discarding the current settings)
  472. # In Simple Mode we assume the user always wants to keep the (limited) current settings
  473. details = catalog.i18nc("@label", "You made changes to the following setting(s):")
  474. user_settings = self._active_container_stack.getTop().findInstances(**{})
  475. for setting in user_settings:
  476. details = details + "\n " + setting.definition.label
  477. Application.getInstance().messageBox(catalog.i18nc("@window:title", "Switched profiles"), catalog.i18nc("@label", "Do you want to transfer your changed settings to this profile?"),
  478. catalog.i18nc("@label", "If you transfer your settings they will override settings in the profile."), details,
  479. buttons = QMessageBox.Yes + QMessageBox.No, icon = QMessageBox.Question, callback = self._keepUserSettingsDialogCallback)
  480. self.activeQualityChanged.emit()
  481. def _keepUserSettingsDialogCallback(self, button):
  482. if button == QMessageBox.Yes:
  483. # Yes, keep the settings in the user profile with this profile
  484. pass
  485. elif button == QMessageBox.No:
  486. # No, discard the settings in the user profile
  487. global_stack = Application.getInstance().getGlobalContainerStack()
  488. for extruder in ExtruderManager.getInstance().getMachineExtruders(global_stack.getId()):
  489. extruder.getTop().clear()
  490. global_stack.getTop().clear()
  491. @pyqtProperty(str, notify = activeVariantChanged)
  492. def activeVariantName(self):
  493. if self._active_container_stack:
  494. variant = self._active_container_stack.findContainer({"type": "variant"})
  495. if variant:
  496. return variant.getName()
  497. return ""
  498. @pyqtProperty(str, notify = activeVariantChanged)
  499. def activeVariantId(self):
  500. if self._active_container_stack:
  501. variant = self._active_container_stack.findContainer({"type": "variant"})
  502. if variant:
  503. return variant.getId()
  504. return ""
  505. @pyqtProperty(str, notify = globalContainerChanged)
  506. def activeDefinitionId(self):
  507. if self._global_container_stack:
  508. definition = self._global_container_stack.getBottom()
  509. if definition:
  510. return definition.id
  511. return ""
  512. ## Gets how the active definition calls variants
  513. # Caveat: per-definition-variant-title is currently not translated (though the fallback is)
  514. @pyqtProperty(str, notify = globalContainerChanged)
  515. def activeDefinitionVariantsName(self):
  516. fallback_title = catalog.i18nc("@label", "Nozzle")
  517. if self._global_container_stack:
  518. return self._global_container_stack.getBottom().getMetaDataEntry("variants_name", fallback_title)
  519. return fallback_title
  520. @pyqtSlot(str, str)
  521. def renameMachine(self, machine_id, new_name):
  522. containers = UM.Settings.ContainerRegistry.getInstance().findContainerStacks(id = machine_id)
  523. if containers:
  524. new_name = self._createUniqueName("machine", containers[0].getName(), new_name, containers[0].getBottom().getName())
  525. containers[0].setName(new_name)
  526. self.globalContainerChanged.emit()
  527. @pyqtSlot(str)
  528. def removeMachine(self, machine_id):
  529. # If the machine that is being removed is the currently active machine, set another machine as the active machine.
  530. activate_new_machine = (self._global_container_stack and self._global_container_stack.getId() == machine_id)
  531. ExtruderManager.getInstance().removeMachineExtruders(machine_id)
  532. containers = UM.Settings.ContainerRegistry.getInstance().findInstanceContainers(type = "user", machine = machine_id)
  533. for container in containers:
  534. UM.Settings.ContainerRegistry.getInstance().removeContainer(container.getId())
  535. UM.Settings.ContainerRegistry.getInstance().removeContainer(machine_id)
  536. if activate_new_machine:
  537. stacks = UM.Settings.ContainerRegistry.getInstance().findContainerStacks(type = "machine")
  538. if stacks:
  539. Application.getInstance().setGlobalContainerStack(stacks[0])
  540. @pyqtProperty(bool, notify = globalContainerChanged)
  541. def hasMaterials(self):
  542. if self._global_container_stack:
  543. return bool(self._global_container_stack.getMetaDataEntry("has_materials", False))
  544. return False
  545. @pyqtProperty(bool, notify = globalContainerChanged)
  546. def hasVariants(self):
  547. if self._global_container_stack:
  548. return bool(self._global_container_stack.getMetaDataEntry("has_variants", False))
  549. return False
  550. ## Property to indicate if a machine has "specialized" material profiles.
  551. # Some machines have their own material profiles that "override" the default catch all profiles.
  552. @pyqtProperty(bool, notify = globalContainerChanged)
  553. def filterMaterialsByMachine(self):
  554. if self._global_container_stack:
  555. return bool(self._global_container_stack.getMetaDataEntry("has_machine_materials", False))
  556. return False
  557. ## Property to indicate if a machine has "specialized" quality profiles.
  558. # Some machines have their own quality profiles that "override" the default catch all profiles.
  559. @pyqtProperty(bool, notify = globalContainerChanged)
  560. def filterQualityByMachine(self):
  561. if self._global_container_stack:
  562. return bool(self._global_container_stack.getMetaDataEntry("has_machine_quality", False))
  563. return False
  564. ## Get the Definition ID of a machine (specified by ID)
  565. # \param machine_id string machine id to get the definition ID of
  566. # \returns DefinitionID (string) if found, None otherwise
  567. @pyqtSlot(str, result = str)
  568. def getDefinitionByMachineId(self, machine_id):
  569. containers = UM.Settings.ContainerRegistry.getInstance().findContainerStacks(id=machine_id)
  570. if containers:
  571. return containers[0].getBottom().getId()
  572. @staticmethod
  573. def createMachineManager(engine=None, script_engine=None):
  574. return MachineManager()
  575. def _updateVariantContainer(self, definition):
  576. if not definition.getMetaDataEntry("has_variants"):
  577. return self._empty_variant_container
  578. containers = []
  579. preferred_variant = definition.getMetaDataEntry("preferred_variant")
  580. if preferred_variant:
  581. containers = UM.Settings.ContainerRegistry.getInstance().findInstanceContainers(type = "variant", definition = definition.id, id = preferred_variant)
  582. if not containers:
  583. containers = UM.Settings.ContainerRegistry.getInstance().findInstanceContainers(type = "variant", definition = definition.id)
  584. if containers:
  585. return containers[0]
  586. return self._empty_variant_container
  587. def _updateMaterialContainer(self, definition, variant_container = None, preferred_material_name = None):
  588. if not definition.getMetaDataEntry("has_materials"):
  589. return self._empty_material_container
  590. search_criteria = { "type": "material" }
  591. if definition.getMetaDataEntry("has_machine_materials"):
  592. search_criteria["definition"] = definition.id
  593. if definition.getMetaDataEntry("has_variants") and variant_container:
  594. search_criteria["variant"] = variant_container.id
  595. else:
  596. search_criteria["definition"] = "fdmprinter"
  597. if preferred_material_name:
  598. search_criteria["name"] = preferred_material_name
  599. else:
  600. preferred_material = definition.getMetaDataEntry("preferred_material")
  601. if preferred_material:
  602. search_criteria["id"] = preferred_material
  603. containers = UM.Settings.ContainerRegistry.getInstance().findInstanceContainers(**search_criteria)
  604. if containers:
  605. return containers[0]
  606. if "name" in search_criteria or "id" in search_criteria:
  607. # If a material by this name can not be found, try a wider set of search criteria
  608. search_criteria.pop("name", None)
  609. search_criteria.pop("id", None)
  610. containers = UM.Settings.ContainerRegistry.getInstance().findInstanceContainers(**search_criteria)
  611. if containers:
  612. return containers[0]
  613. return self._empty_material_container
  614. def _updateQualityContainer(self, definition, variant_container, material_container = None, preferred_quality_name = None):
  615. container_registry = UM.Settings.ContainerRegistry.getInstance()
  616. search_criteria = { "type": "quality" }
  617. if definition.getMetaDataEntry("has_machine_quality"):
  618. search_criteria["definition"] = definition.id
  619. if definition.getMetaDataEntry("has_materials") and material_container:
  620. search_criteria["material"] = material_container.id
  621. else:
  622. search_criteria["definition"] = "fdmprinter"
  623. if preferred_quality_name and preferred_quality_name != "empty":
  624. search_criteria["name"] = preferred_quality_name
  625. else:
  626. preferred_quality = definition.getMetaDataEntry("preferred_quality")
  627. if preferred_quality:
  628. search_criteria["id"] = preferred_quality
  629. containers = container_registry.findInstanceContainers(**search_criteria)
  630. if containers:
  631. return containers[0]
  632. if "material" in search_criteria:
  633. # If a quality for this specific material cannot be found, try finding qualities for a generic version of the material
  634. material_search_criteria = { "type": "material", "material": material_container.getMetaDataEntry("material"), "color_name": "Generic" }
  635. if definition.getMetaDataEntry("has_machine_quality"):
  636. material_search_criteria["definition"] = definition.id
  637. if definition.getMetaDataEntry("has_variants") and variant_container:
  638. material_search_criteria["variant"] = variant_container.id
  639. else:
  640. material_search_criteria["definition"] = "fdmprinter"
  641. material_containers = container_registry.findInstanceContainers(**material_search_criteria)
  642. if material_containers:
  643. search_criteria["material"] = material_containers[0].getId()
  644. containers = container_registry.findInstanceContainers(**search_criteria)
  645. if containers:
  646. return containers[0]
  647. if "name" in search_criteria or "id" in search_criteria:
  648. # If a quality by this name can not be found, try a wider set of search criteria
  649. search_criteria.pop("name", None)
  650. search_criteria.pop("id", None)
  651. containers = container_registry.findInstanceContainers(**search_criteria)
  652. if containers:
  653. return containers[0]
  654. return self._empty_quality_container
  655. def _onMachineNameChanged(self):
  656. self.globalContainerChanged.emit()
  657. def _onMaterialNameChanged(self):
  658. self.activeMaterialChanged.emit()
  659. def _onQualityNameChanged(self):
  660. self.activeQualityChanged.emit()