CuraContainerRegistry.py 45 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765
  1. # Copyright (c) 2019 Ultimaker B.V.
  2. # Cura is released under the terms of the LGPLv3 or higher.
  3. import os
  4. import re
  5. import configparser
  6. from typing import Any, cast, Dict, Optional, List, Union
  7. from PyQt5.QtWidgets import QMessageBox
  8. from UM.Decorators import override
  9. from UM.Settings.ContainerFormatError import ContainerFormatError
  10. from UM.Settings.Interfaces import ContainerInterface
  11. from UM.Settings.ContainerRegistry import ContainerRegistry
  12. from UM.Settings.ContainerStack import ContainerStack
  13. from UM.Settings.InstanceContainer import InstanceContainer
  14. from UM.Settings.SettingInstance import SettingInstance
  15. from UM.Logger import Logger
  16. from UM.Message import Message
  17. from UM.Platform import Platform
  18. from UM.PluginRegistry import PluginRegistry # For getting the possible profile writers to write with.
  19. from UM.Resources import Resources
  20. from UM.Util import parseBool
  21. from cura.ReaderWriters.ProfileWriter import ProfileWriter
  22. from . import ExtruderStack
  23. from . import GlobalStack
  24. import cura.CuraApplication
  25. from cura.Settings.cura_empty_instance_containers import empty_quality_container
  26. from cura.Machines.ContainerTree import ContainerTree
  27. from cura.ReaderWriters.ProfileReader import NoProfileException, ProfileReader
  28. from UM.i18n import i18nCatalog
  29. catalog = i18nCatalog("cura")
  30. class CuraContainerRegistry(ContainerRegistry):
  31. def __init__(self, *args, **kwargs):
  32. super().__init__(*args, **kwargs)
  33. # We don't have all the machines loaded in the beginning, so in order to add the missing extruder stack
  34. # for single extrusion machines, we subscribe to the containerAdded signal, and whenever a global stack
  35. # is added, we check to see if an extruder stack needs to be added.
  36. self.containerAdded.connect(self._onContainerAdded)
  37. ## Overridden from ContainerRegistry
  38. #
  39. # Adds a container to the registry.
  40. #
  41. # This will also try to convert a ContainerStack to either Extruder or
  42. # Global stack based on metadata information.
  43. @override(ContainerRegistry)
  44. def addContainer(self, container: ContainerInterface) -> None:
  45. # Note: Intentional check with type() because we want to ignore subclasses
  46. if type(container) == ContainerStack:
  47. container = self._convertContainerStack(cast(ContainerStack, container))
  48. if isinstance(container, InstanceContainer) and type(container) != type(self.getEmptyInstanceContainer()):
  49. # Check against setting version of the definition.
  50. required_setting_version = cura.CuraApplication.CuraApplication.SettingVersion
  51. actual_setting_version = int(container.getMetaDataEntry("setting_version", default = 0))
  52. if required_setting_version != actual_setting_version:
  53. Logger.log("w", "Instance container {container_id} is outdated. Its setting version is {actual_setting_version} but it should be {required_setting_version}.".format(container_id = container.getId(), actual_setting_version = actual_setting_version, required_setting_version = required_setting_version))
  54. return # Don't add.
  55. super().addContainer(container)
  56. ## Create a name that is not empty and unique
  57. # \param container_type \type{string} Type of the container (machine, quality, ...)
  58. # \param current_name \type{} Current name of the container, which may be an acceptable option
  59. # \param new_name \type{string} Base name, which may not be unique
  60. # \param fallback_name \type{string} Name to use when (stripped) new_name is empty
  61. # \return \type{string} Name that is unique for the specified type and name/id
  62. def createUniqueName(self, container_type: str, current_name: str, new_name: str, fallback_name: str) -> str:
  63. new_name = new_name.strip()
  64. num_check = re.compile(r"(.*?)\s*#\d+$").match(new_name)
  65. if num_check:
  66. new_name = num_check.group(1)
  67. if new_name == "":
  68. new_name = fallback_name
  69. unique_name = new_name
  70. i = 1
  71. # In case we are renaming, the current name of the container is also a valid end-result
  72. while self._containerExists(container_type, unique_name) and unique_name != current_name:
  73. i += 1
  74. unique_name = "%s #%d" % (new_name, i)
  75. return unique_name
  76. ## Check if a container with of a certain type and a certain name or id exists
  77. # Both the id and the name are checked, because they may not be the same and it is better if they are both unique
  78. # \param container_type \type{string} Type of the container (machine, quality, ...)
  79. # \param container_name \type{string} Name to check
  80. def _containerExists(self, container_type: str, container_name: str):
  81. container_class = ContainerStack if container_type == "machine" else InstanceContainer
  82. return self.findContainersMetadata(container_type = container_class, id = container_name, type = container_type, ignore_case = True) or \
  83. self.findContainersMetadata(container_type = container_class, name = container_name, type = container_type)
  84. ## Exports an profile to a file
  85. #
  86. # \param container_list \type{list} the containers to export. This is not
  87. # necessarily in any order!
  88. # \param file_name \type{str} the full path and filename to export to.
  89. # \param file_type \type{str} the file type with the format "<description> (*.<extension>)"
  90. # \return True if the export succeeded, false otherwise.
  91. def exportQualityProfile(self, container_list: List[InstanceContainer], file_name: str, file_type: str) -> bool:
  92. # Parse the fileType to deduce what plugin can save the file format.
  93. # fileType has the format "<description> (*.<extension>)"
  94. split = file_type.rfind(" (*.") # Find where the description ends and the extension starts.
  95. if split < 0: # Not found. Invalid format.
  96. Logger.log("e", "Invalid file format identifier %s", file_type)
  97. return False
  98. description = file_type[:split]
  99. extension = file_type[split + 4:-1] # Leave out the " (*." and ")".
  100. if not file_name.endswith("." + extension): # Auto-fill the extension if the user did not provide any.
  101. file_name += "." + extension
  102. # On Windows, QML FileDialog properly asks for overwrite confirm, but not on other platforms, so handle those ourself.
  103. if not Platform.isWindows():
  104. if os.path.exists(file_name):
  105. result = QMessageBox.question(None, catalog.i18nc("@title:window", "File Already Exists"),
  106. catalog.i18nc("@label Don't translate the XML tag <filename>!", "The file <filename>{0}</filename> already exists. Are you sure you want to overwrite it?").format(file_name))
  107. if result == QMessageBox.No:
  108. return False
  109. profile_writer = self._findProfileWriter(extension, description)
  110. try:
  111. if profile_writer is None:
  112. raise Exception("Unable to find a profile writer")
  113. success = profile_writer.write(file_name, container_list)
  114. except Exception as e:
  115. Logger.log("e", "Failed to export profile to %s: %s", file_name, str(e))
  116. m = Message(catalog.i18nc("@info:status Don't translate the XML tags <filename> or <message>!", "Failed to export profile to <filename>{0}</filename>: <message>{1}</message>", file_name, str(e)),
  117. lifetime = 0,
  118. title = catalog.i18nc("@info:title", "Error"))
  119. m.show()
  120. return False
  121. if not success:
  122. Logger.log("w", "Failed to export profile to %s: Writer plugin reported failure.", file_name)
  123. m = Message(catalog.i18nc("@info:status Don't translate the XML tag <filename>!", "Failed to export profile to <filename>{0}</filename>: Writer plugin reported failure.", file_name),
  124. lifetime = 0,
  125. title = catalog.i18nc("@info:title", "Error"))
  126. m.show()
  127. return False
  128. m = Message(catalog.i18nc("@info:status Don't translate the XML tag <filename>!", "Exported profile to <filename>{0}</filename>", file_name),
  129. title = catalog.i18nc("@info:title", "Export succeeded"))
  130. m.show()
  131. return True
  132. ## Gets the plugin object matching the criteria
  133. # \param extension
  134. # \param description
  135. # \return The plugin object matching the given extension and description.
  136. def _findProfileWriter(self, extension: str, description: str) -> Optional[ProfileWriter]:
  137. plugin_registry = PluginRegistry.getInstance()
  138. for plugin_id, meta_data in self._getIOPlugins("profile_writer"):
  139. for supported_type in meta_data["profile_writer"]: # All file types this plugin can supposedly write.
  140. supported_extension = supported_type.get("extension", None)
  141. if supported_extension == extension: # This plugin supports a file type with the same extension.
  142. supported_description = supported_type.get("description", None)
  143. if supported_description == description: # The description is also identical. Assume it's the same file type.
  144. return cast(ProfileWriter, plugin_registry.getPluginObject(plugin_id))
  145. return None
  146. ## Imports a profile from a file
  147. #
  148. # \param file_name The full path and filename of the profile to import.
  149. # \return Dict with a 'status' key containing the string 'ok' or 'error',
  150. # and a 'message' key containing a message for the user.
  151. def importProfile(self, file_name: str) -> Dict[str, str]:
  152. Logger.log("d", "Attempting to import profile %s", file_name)
  153. if not file_name:
  154. return { "status": "error", "message": catalog.i18nc("@info:status Don't translate the XML tags <filename>!", "Failed to import profile from <filename>{0}</filename>: {1}", file_name, "Invalid path")}
  155. global_stack = cura.CuraApplication.CuraApplication.getInstance().getGlobalContainerStack()
  156. if not global_stack:
  157. return {"status": "error", "message": catalog.i18nc("@info:status Don't translate the XML tags <filename>!", "Can't import profile from <filename>{0}</filename> before a printer is added.", file_name)}
  158. container_tree = ContainerTree.getInstance()
  159. machine_extruders = []
  160. for position in sorted(global_stack.extruders):
  161. machine_extruders.append(global_stack.extruders[position])
  162. plugin_registry = PluginRegistry.getInstance()
  163. extension = file_name.split(".")[-1]
  164. for plugin_id, meta_data in self._getIOPlugins("profile_reader"):
  165. if meta_data["profile_reader"][0]["extension"] != extension:
  166. continue
  167. profile_reader = cast(ProfileReader, plugin_registry.getPluginObject(plugin_id))
  168. try:
  169. profile_or_list = profile_reader.read(file_name) # Try to open the file with the profile reader.
  170. except NoProfileException:
  171. return { "status": "ok", "message": catalog.i18nc("@info:status Don't translate the XML tags <filename>!", "No custom profile to import in file <filename>{0}</filename>", file_name)}
  172. except Exception as e:
  173. # Note that this will fail quickly. That is, if any profile reader throws an exception, it will stop reading. It will only continue reading if the reader returned None.
  174. Logger.log("e", "Failed to import profile from %s: %s while using profile reader. Got exception %s", file_name, profile_reader.getPluginId(), str(e))
  175. return { "status": "error", "message": catalog.i18nc("@info:status Don't translate the XML tags <filename>!", "Failed to import profile from <filename>{0}</filename>:", file_name) + "\n<message>" + str(e) + "</message>"}
  176. if profile_or_list:
  177. # Ensure it is always a list of profiles
  178. if not isinstance(profile_or_list, list):
  179. profile_or_list = [profile_or_list]
  180. # First check if this profile is suitable for this machine
  181. global_profile = None
  182. extruder_profiles = []
  183. if len(profile_or_list) == 1:
  184. global_profile = profile_or_list[0]
  185. else:
  186. for profile in profile_or_list:
  187. if not profile.getMetaDataEntry("position"):
  188. global_profile = profile
  189. else:
  190. extruder_profiles.append(profile)
  191. extruder_profiles = sorted(extruder_profiles, key = lambda x: int(x.getMetaDataEntry("position")))
  192. profile_or_list = [global_profile] + extruder_profiles
  193. if not global_profile:
  194. Logger.log("e", "Incorrect profile [%s]. Could not find global profile", file_name)
  195. return { "status": "error",
  196. "message": catalog.i18nc("@info:status Don't translate the XML tags <filename>!", "This profile <filename>{0}</filename> contains incorrect data, could not import it.", file_name)}
  197. profile_definition = global_profile.getMetaDataEntry("definition")
  198. # Make sure we have a profile_definition in the file:
  199. if profile_definition is None:
  200. break
  201. machine_definitions = self.findContainers(id = profile_definition)
  202. if not machine_definitions:
  203. Logger.log("e", "Incorrect profile [%s]. Unknown machine type [%s]", file_name, profile_definition)
  204. return {"status": "error",
  205. "message": catalog.i18nc("@info:status Don't translate the XML tags <filename>!", "This profile <filename>{0}</filename> contains incorrect data, could not import it.", file_name)
  206. }
  207. machine_definition = machine_definitions[0]
  208. # Get the expected machine definition.
  209. # i.e.: We expect gcode for a UM2 Extended to be defined as normal UM2 gcode...
  210. has_machine_quality = parseBool(machine_definition.getMetaDataEntry("has_machine_quality", "false"))
  211. profile_definition = machine_definition.getMetaDataEntry("quality_definition", machine_definition.getId()) if has_machine_quality else "fdmprinter"
  212. expected_machine_definition = container_tree.machines[global_stack.definition.getId()].quality_definition
  213. # And check if the profile_definition matches either one (showing error if not):
  214. if profile_definition != expected_machine_definition:
  215. Logger.log("d", "Profile {file_name} is for machine {profile_definition}, but the current active machine is {expected_machine_definition}. Changing profile's definition.".format(file_name = file_name, profile_definition = profile_definition, expected_machine_definition = expected_machine_definition))
  216. global_profile.setMetaDataEntry("definition", expected_machine_definition)
  217. for extruder_profile in extruder_profiles:
  218. extruder_profile.setMetaDataEntry("definition", expected_machine_definition)
  219. quality_name = global_profile.getName()
  220. quality_type = global_profile.getMetaDataEntry("quality_type")
  221. name_seed = os.path.splitext(os.path.basename(file_name))[0]
  222. new_name = self.uniqueName(name_seed)
  223. # Ensure it is always a list of profiles
  224. if type(profile_or_list) is not list:
  225. profile_or_list = [profile_or_list]
  226. # Make sure that there are also extruder stacks' quality_changes, not just one for the global stack
  227. if len(profile_or_list) == 1:
  228. global_profile = profile_or_list[0]
  229. extruder_profiles = []
  230. for idx, extruder in enumerate(global_stack.extruders.values()):
  231. profile_id = ContainerRegistry.getInstance().uniqueName(global_stack.getId() + "_extruder_" + str(idx + 1))
  232. profile = InstanceContainer(profile_id)
  233. profile.setName(quality_name)
  234. profile.setMetaDataEntry("setting_version", cura.CuraApplication.CuraApplication.SettingVersion)
  235. profile.setMetaDataEntry("type", "quality_changes")
  236. profile.setMetaDataEntry("definition", expected_machine_definition)
  237. profile.setMetaDataEntry("quality_type", quality_type)
  238. profile.setDirty(True)
  239. if idx == 0:
  240. # Move all per-extruder settings to the first extruder's quality_changes
  241. for qc_setting_key in global_profile.getAllKeys():
  242. settable_per_extruder = global_stack.getProperty(qc_setting_key, "settable_per_extruder")
  243. if settable_per_extruder:
  244. setting_value = global_profile.getProperty(qc_setting_key, "value")
  245. setting_definition = global_stack.getSettingDefinition(qc_setting_key)
  246. if setting_definition is not None:
  247. new_instance = SettingInstance(setting_definition, profile)
  248. new_instance.setProperty("value", setting_value)
  249. new_instance.resetState() # Ensure that the state is not seen as a user state.
  250. profile.addInstance(new_instance)
  251. profile.setDirty(True)
  252. global_profile.removeInstance(qc_setting_key, postpone_emit = True)
  253. extruder_profiles.append(profile)
  254. for profile in extruder_profiles:
  255. profile_or_list.append(profile)
  256. # Import all profiles
  257. profile_ids_added = [] # type: List[str]
  258. for profile_index, profile in enumerate(profile_or_list):
  259. if profile_index == 0:
  260. # This is assumed to be the global profile
  261. profile_id = (cast(ContainerInterface, global_stack.getBottom()).getId() + "_" + name_seed).lower().replace(" ", "_")
  262. elif profile_index < len(machine_extruders) + 1:
  263. # This is assumed to be an extruder profile
  264. extruder_id = machine_extruders[profile_index - 1].definition.getId()
  265. extruder_position = str(profile_index - 1)
  266. if not profile.getMetaDataEntry("position"):
  267. profile.setMetaDataEntry("position", extruder_position)
  268. else:
  269. profile.setMetaDataEntry("position", extruder_position)
  270. profile_id = (extruder_id + "_" + name_seed).lower().replace(" ", "_")
  271. else: # More extruders in the imported file than in the machine.
  272. continue # Delete the additional profiles.
  273. result = self._configureProfile(profile, profile_id, new_name, expected_machine_definition)
  274. if result is not None:
  275. # Remove any profiles that did got added.
  276. for profile_id in profile_ids_added:
  277. self.removeContainer(profile_id)
  278. return {"status": "error", "message": catalog.i18nc(
  279. "@info:status Don't translate the XML tag <filename>!",
  280. "Failed to import profile from <filename>{0}</filename>:",
  281. file_name) + " " + result}
  282. profile_ids_added.append(profile.getId())
  283. return {"status": "ok", "message": catalog.i18nc("@info:status", "Successfully imported profile {0}", profile_or_list[0].getName())}
  284. # This message is throw when the profile reader doesn't find any profile in the file
  285. return {"status": "error", "message": catalog.i18nc("@info:status", "File {0} does not contain any valid profile.", file_name)}
  286. # If it hasn't returned by now, none of the plugins loaded the profile successfully.
  287. return {"status": "error", "message": catalog.i18nc("@info:status", "Profile {0} has an unknown file type or is corrupted.", file_name)}
  288. @override(ContainerRegistry)
  289. def load(self) -> None:
  290. super().load()
  291. self._registerSingleExtrusionMachinesExtruderStacks()
  292. self._connectUpgradedExtruderStacksToMachines()
  293. ## Check if the metadata for a container is okay before adding it.
  294. #
  295. # This overrides the one from UM.Settings.ContainerRegistry because we
  296. # also require that the setting_version is correct.
  297. @override(ContainerRegistry)
  298. def _isMetadataValid(self, metadata: Optional[Dict[str, Any]]) -> bool:
  299. if metadata is None:
  300. return False
  301. if "setting_version" not in metadata:
  302. return False
  303. try:
  304. if int(metadata["setting_version"]) != cura.CuraApplication.CuraApplication.SettingVersion:
  305. return False
  306. except ValueError: #Not parsable as int.
  307. return False
  308. return True
  309. ## Update an imported profile to match the current machine configuration.
  310. #
  311. # \param profile The profile to configure.
  312. # \param id_seed The base ID for the profile. May be changed so it does not conflict with existing containers.
  313. # \param new_name The new name for the profile.
  314. #
  315. # \return None if configuring was successful or an error message if an error occurred.
  316. def _configureProfile(self, profile: InstanceContainer, id_seed: str, new_name: str, machine_definition_id: str) -> Optional[str]:
  317. profile.setDirty(True) # Ensure the profiles are correctly saved
  318. new_id = self.createUniqueName("quality_changes", "", id_seed, catalog.i18nc("@label", "Custom profile"))
  319. profile.setMetaDataEntry("id", new_id)
  320. profile.setName(new_name)
  321. # Set the unique Id to the profile, so it's generating a new one even if the user imports the same profile
  322. # It also solves an issue with importing profiles from G-Codes
  323. profile.setMetaDataEntry("id", new_id)
  324. profile.setMetaDataEntry("definition", machine_definition_id)
  325. if "type" in profile.getMetaData():
  326. profile.setMetaDataEntry("type", "quality_changes")
  327. else:
  328. profile.setMetaDataEntry("type", "quality_changes")
  329. quality_type = profile.getMetaDataEntry("quality_type")
  330. if not quality_type:
  331. return catalog.i18nc("@info:status", "Profile is missing a quality type.")
  332. global_stack = cura.CuraApplication.CuraApplication.getInstance().getGlobalContainerStack()
  333. if global_stack is None:
  334. return None
  335. definition_id = ContainerTree.getInstance().machines[global_stack.definition.getId()].quality_definition
  336. profile.setDefinition(definition_id)
  337. # Check to make sure the imported profile actually makes sense in context of the current configuration.
  338. # This prevents issues where importing a "draft" profile for a machine without "draft" qualities would report as
  339. # successfully imported but then fail to show up.
  340. quality_group_dict = ContainerTree.getInstance().getCurrentQualityGroups()
  341. # "not_supported" profiles can be imported.
  342. if quality_type != empty_quality_container.getMetaDataEntry("quality_type") and quality_type not in quality_group_dict:
  343. return catalog.i18nc("@info:status", "Could not find a quality type {0} for the current configuration.", quality_type)
  344. ContainerRegistry.getInstance().addContainer(profile)
  345. return None
  346. @override(ContainerRegistry)
  347. def saveDirtyContainers(self) -> None:
  348. # Lock file for "more" atomically loading and saving to/from config dir.
  349. with self.lockFile():
  350. # Save base files first
  351. for instance in self.findDirtyContainers(container_type=InstanceContainer):
  352. if instance.getMetaDataEntry("removed"):
  353. continue
  354. if instance.getId() == instance.getMetaData().get("base_file"):
  355. self.saveContainer(instance)
  356. for instance in self.findDirtyContainers(container_type=InstanceContainer):
  357. if instance.getMetaDataEntry("removed"):
  358. continue
  359. self.saveContainer(instance)
  360. for stack in self.findContainerStacks():
  361. self.saveContainer(stack)
  362. ## Gets a list of profile writer plugins
  363. # \return List of tuples of (plugin_id, meta_data).
  364. def _getIOPlugins(self, io_type):
  365. plugin_registry = PluginRegistry.getInstance()
  366. active_plugin_ids = plugin_registry.getActivePlugins()
  367. result = []
  368. for plugin_id in active_plugin_ids:
  369. meta_data = plugin_registry.getMetaData(plugin_id)
  370. if io_type in meta_data:
  371. result.append( (plugin_id, meta_data) )
  372. return result
  373. ## Convert an "old-style" pure ContainerStack to either an Extruder or Global stack.
  374. def _convertContainerStack(self, container: ContainerStack) -> Union[ExtruderStack.ExtruderStack, GlobalStack.GlobalStack]:
  375. assert type(container) == ContainerStack
  376. container_type = container.getMetaDataEntry("type")
  377. if container_type not in ("extruder_train", "machine"):
  378. # It is not an extruder or machine, so do nothing with the stack
  379. return container
  380. Logger.log("d", "Converting ContainerStack {stack} to {type}", stack = container.getId(), type = container_type)
  381. if container_type == "extruder_train":
  382. new_stack = ExtruderStack.ExtruderStack(container.getId())
  383. else:
  384. new_stack = GlobalStack.GlobalStack(container.getId())
  385. container_contents = container.serialize()
  386. new_stack.deserialize(container_contents)
  387. # Delete the old configuration file so we do not get double stacks
  388. if os.path.isfile(container.getPath()):
  389. os.remove(container.getPath())
  390. return new_stack
  391. def _registerSingleExtrusionMachinesExtruderStacks(self) -> None:
  392. machines = self.findContainerStacks(type = "machine", machine_extruder_trains = {"0": "fdmextruder"})
  393. for machine in machines:
  394. extruder_stacks = self.findContainerStacks(type = "extruder_train", machine = machine.getId())
  395. if not extruder_stacks:
  396. self.addExtruderStackForSingleExtrusionMachine(machine, "fdmextruder")
  397. def _onContainerAdded(self, container: ContainerInterface) -> None:
  398. # We don't have all the machines loaded in the beginning, so in order to add the missing extruder stack
  399. # for single extrusion machines, we subscribe to the containerAdded signal, and whenever a global stack
  400. # is added, we check to see if an extruder stack needs to be added.
  401. if not isinstance(container, ContainerStack) or container.getMetaDataEntry("type") != "machine":
  402. return
  403. machine_extruder_trains = container.getMetaDataEntry("machine_extruder_trains")
  404. if machine_extruder_trains is not None and machine_extruder_trains != {"0": "fdmextruder"}:
  405. return
  406. extruder_stacks = self.findContainerStacks(type = "extruder_train", machine = container.getId())
  407. if not extruder_stacks:
  408. self.addExtruderStackForSingleExtrusionMachine(container, "fdmextruder")
  409. #
  410. # new_global_quality_changes is optional. It is only used in project loading for a scenario like this:
  411. # - override the current machine
  412. # - create new for custom quality profile
  413. # new_global_quality_changes is the new global quality changes container in this scenario.
  414. # create_new_ids indicates if new unique ids must be created
  415. #
  416. def addExtruderStackForSingleExtrusionMachine(self, machine, extruder_id, new_global_quality_changes = None, create_new_ids = True):
  417. new_extruder_id = extruder_id
  418. application = cura.CuraApplication.CuraApplication.getInstance()
  419. extruder_definitions = self.findDefinitionContainers(id = new_extruder_id)
  420. if not extruder_definitions:
  421. Logger.log("w", "Could not find definition containers for extruder %s", new_extruder_id)
  422. return
  423. extruder_definition = extruder_definitions[0]
  424. unique_name = self.uniqueName(machine.getName() + " " + new_extruder_id) if create_new_ids else machine.getName() + " " + new_extruder_id
  425. extruder_stack = ExtruderStack.ExtruderStack(unique_name)
  426. extruder_stack.setName(extruder_definition.getName())
  427. extruder_stack.setDefinition(extruder_definition)
  428. extruder_stack.setMetaDataEntry("position", extruder_definition.getMetaDataEntry("position"))
  429. # create a new definition_changes container for the extruder stack
  430. definition_changes_id = self.uniqueName(extruder_stack.getId() + "_settings") if create_new_ids else extruder_stack.getId() + "_settings"
  431. definition_changes_name = definition_changes_id
  432. definition_changes = InstanceContainer(definition_changes_id, parent = application)
  433. definition_changes.setName(definition_changes_name)
  434. definition_changes.setMetaDataEntry("setting_version", application.SettingVersion)
  435. definition_changes.setMetaDataEntry("type", "definition_changes")
  436. definition_changes.setMetaDataEntry("definition", extruder_definition.getId())
  437. # move definition_changes settings if exist
  438. for setting_key in definition_changes.getAllKeys():
  439. if machine.definition.getProperty(setting_key, "settable_per_extruder"):
  440. setting_value = machine.definitionChanges.getProperty(setting_key, "value")
  441. if setting_value is not None:
  442. # move it to the extruder stack's definition_changes
  443. setting_definition = machine.getSettingDefinition(setting_key)
  444. new_instance = SettingInstance(setting_definition, definition_changes)
  445. new_instance.setProperty("value", setting_value)
  446. new_instance.resetState() # Ensure that the state is not seen as a user state.
  447. definition_changes.addInstance(new_instance)
  448. definition_changes.setDirty(True)
  449. machine.definitionChanges.removeInstance(setting_key, postpone_emit = True)
  450. self.addContainer(definition_changes)
  451. extruder_stack.setDefinitionChanges(definition_changes)
  452. # create empty user changes container otherwise
  453. user_container_id = self.uniqueName(extruder_stack.getId() + "_user") if create_new_ids else extruder_stack.getId() + "_user"
  454. user_container_name = user_container_id
  455. user_container = InstanceContainer(user_container_id, parent = application)
  456. user_container.setName(user_container_name)
  457. user_container.setMetaDataEntry("type", "user")
  458. user_container.setMetaDataEntry("machine", machine.getId())
  459. user_container.setMetaDataEntry("setting_version", application.SettingVersion)
  460. user_container.setDefinition(machine.definition.getId())
  461. user_container.setMetaDataEntry("position", extruder_stack.getMetaDataEntry("position"))
  462. if machine.userChanges:
  463. # For the newly created extruder stack, we need to move all "per-extruder" settings to the user changes
  464. # container to the extruder stack.
  465. for user_setting_key in machine.userChanges.getAllKeys():
  466. settable_per_extruder = machine.getProperty(user_setting_key, "settable_per_extruder")
  467. if settable_per_extruder:
  468. setting_value = machine.getProperty(user_setting_key, "value")
  469. setting_definition = machine.getSettingDefinition(user_setting_key)
  470. new_instance = SettingInstance(setting_definition, definition_changes)
  471. new_instance.setProperty("value", setting_value)
  472. new_instance.resetState() # Ensure that the state is not seen as a user state.
  473. user_container.addInstance(new_instance)
  474. user_container.setDirty(True)
  475. machine.userChanges.removeInstance(user_setting_key, postpone_emit = True)
  476. self.addContainer(user_container)
  477. extruder_stack.setUserChanges(user_container)
  478. empty_variant = application.empty_variant_container
  479. empty_material = application.empty_material_container
  480. empty_quality = application.empty_quality_container
  481. if machine.variant.getId() not in ("empty", "empty_variant"):
  482. variant = machine.variant
  483. else:
  484. variant = empty_variant
  485. extruder_stack.variant = variant
  486. if machine.material.getId() not in ("empty", "empty_material"):
  487. material = machine.material
  488. else:
  489. material = empty_material
  490. extruder_stack.material = material
  491. if machine.quality.getId() not in ("empty", "empty_quality"):
  492. quality = machine.quality
  493. else:
  494. quality = empty_quality
  495. extruder_stack.quality = quality
  496. machine_quality_changes = machine.qualityChanges
  497. if new_global_quality_changes is not None:
  498. machine_quality_changes = new_global_quality_changes
  499. if machine_quality_changes.getId() not in ("empty", "empty_quality_changes"):
  500. extruder_quality_changes_container = self.findInstanceContainers(name = machine_quality_changes.getName(), extruder = extruder_id)
  501. if extruder_quality_changes_container:
  502. extruder_quality_changes_container = extruder_quality_changes_container[0]
  503. quality_changes_id = extruder_quality_changes_container.getId()
  504. extruder_stack.qualityChanges = self.findInstanceContainers(id = quality_changes_id)[0]
  505. else:
  506. # Some extruder quality_changes containers can be created at runtime as files in the qualities
  507. # folder. Those files won't be loaded in the registry immediately. So we also need to search
  508. # the folder to see if the quality_changes exists.
  509. extruder_quality_changes_container = self._findQualityChangesContainerInCuraFolder(machine_quality_changes.getName())
  510. if extruder_quality_changes_container:
  511. quality_changes_id = extruder_quality_changes_container.getId()
  512. extruder_quality_changes_container.setMetaDataEntry("position", extruder_definition.getMetaDataEntry("position"))
  513. extruder_stack.qualityChanges = self.findInstanceContainers(id = quality_changes_id)[0]
  514. else:
  515. # If we still cannot find a quality changes container for the extruder, create a new one
  516. container_name = machine_quality_changes.getName()
  517. container_id = self.uniqueName(extruder_stack.getId() + "_qc_" + container_name)
  518. extruder_quality_changes_container = InstanceContainer(container_id, parent = application)
  519. extruder_quality_changes_container.setName(container_name)
  520. extruder_quality_changes_container.setMetaDataEntry("type", "quality_changes")
  521. extruder_quality_changes_container.setMetaDataEntry("setting_version", application.SettingVersion)
  522. extruder_quality_changes_container.setMetaDataEntry("position", extruder_definition.getMetaDataEntry("position"))
  523. extruder_quality_changes_container.setMetaDataEntry("quality_type", machine_quality_changes.getMetaDataEntry("quality_type"))
  524. extruder_quality_changes_container.setMetaDataEntry("intent_category", "default") # Intent categories weren't a thing back then.
  525. extruder_quality_changes_container.setDefinition(machine_quality_changes.getDefinition().getId())
  526. self.addContainer(extruder_quality_changes_container)
  527. extruder_stack.qualityChanges = extruder_quality_changes_container
  528. if not extruder_quality_changes_container:
  529. Logger.log("w", "Could not find quality_changes named [%s] for extruder [%s]",
  530. machine_quality_changes.getName(), extruder_stack.getId())
  531. else:
  532. # Move all per-extruder settings to the extruder's quality changes
  533. for qc_setting_key in machine_quality_changes.getAllKeys():
  534. settable_per_extruder = machine.getProperty(qc_setting_key, "settable_per_extruder")
  535. if settable_per_extruder:
  536. setting_value = machine_quality_changes.getProperty(qc_setting_key, "value")
  537. setting_definition = machine.getSettingDefinition(qc_setting_key)
  538. new_instance = SettingInstance(setting_definition, definition_changes)
  539. new_instance.setProperty("value", setting_value)
  540. new_instance.resetState() # Ensure that the state is not seen as a user state.
  541. extruder_quality_changes_container.addInstance(new_instance)
  542. extruder_quality_changes_container.setDirty(True)
  543. machine_quality_changes.removeInstance(qc_setting_key, postpone_emit=True)
  544. else:
  545. extruder_stack.qualityChanges = self.findInstanceContainers(id = "empty_quality_changes")[0]
  546. self.addContainer(extruder_stack)
  547. # Also need to fix the other qualities that are suitable for this machine. Those quality changes may still have
  548. # per-extruder settings in the container for the machine instead of the extruder.
  549. if machine_quality_changes.getId() not in ("empty", "empty_quality_changes"):
  550. quality_changes_machine_definition_id = machine_quality_changes.getDefinition().getId()
  551. else:
  552. whole_machine_definition = machine.definition
  553. machine_entry = machine.definition.getMetaDataEntry("machine")
  554. if machine_entry is not None:
  555. container_registry = ContainerRegistry.getInstance()
  556. whole_machine_definition = container_registry.findDefinitionContainers(id = machine_entry)[0]
  557. quality_changes_machine_definition_id = "fdmprinter"
  558. if whole_machine_definition.getMetaDataEntry("has_machine_quality"):
  559. quality_changes_machine_definition_id = machine.definition.getMetaDataEntry("quality_definition",
  560. whole_machine_definition.getId())
  561. qcs = self.findInstanceContainers(type = "quality_changes", definition = quality_changes_machine_definition_id)
  562. qc_groups = {} # map of qc names -> qc containers
  563. for qc in qcs:
  564. qc_name = qc.getName()
  565. if qc_name not in qc_groups:
  566. qc_groups[qc_name] = []
  567. qc_groups[qc_name].append(qc)
  568. # Try to find from the quality changes cura directory too
  569. quality_changes_container = self._findQualityChangesContainerInCuraFolder(machine_quality_changes.getName())
  570. if quality_changes_container:
  571. qc_groups[qc_name].append(quality_changes_container)
  572. for qc_name, qc_list in qc_groups.items():
  573. qc_dict = {"global": None, "extruders": []}
  574. for qc in qc_list:
  575. extruder_position = qc.getMetaDataEntry("position")
  576. if extruder_position is not None:
  577. qc_dict["extruders"].append(qc)
  578. else:
  579. qc_dict["global"] = qc
  580. if qc_dict["global"] is not None and len(qc_dict["extruders"]) == 1:
  581. # Move per-extruder settings
  582. for qc_setting_key in qc_dict["global"].getAllKeys():
  583. settable_per_extruder = machine.getProperty(qc_setting_key, "settable_per_extruder")
  584. if settable_per_extruder:
  585. setting_value = qc_dict["global"].getProperty(qc_setting_key, "value")
  586. setting_definition = machine.getSettingDefinition(qc_setting_key)
  587. new_instance = SettingInstance(setting_definition, definition_changes)
  588. new_instance.setProperty("value", setting_value)
  589. new_instance.resetState() # Ensure that the state is not seen as a user state.
  590. qc_dict["extruders"][0].addInstance(new_instance)
  591. qc_dict["extruders"][0].setDirty(True)
  592. qc_dict["global"].removeInstance(qc_setting_key, postpone_emit=True)
  593. # Set next stack at the end
  594. extruder_stack.setNextStack(machine)
  595. return extruder_stack
  596. def _findQualityChangesContainerInCuraFolder(self, name: str) -> Optional[InstanceContainer]:
  597. quality_changes_dir = Resources.getPath(cura.CuraApplication.CuraApplication.ResourceTypes.QualityChangesInstanceContainer)
  598. instance_container = None
  599. for item in os.listdir(quality_changes_dir):
  600. file_path = os.path.join(quality_changes_dir, item)
  601. if not os.path.isfile(file_path):
  602. continue
  603. parser = configparser.ConfigParser(interpolation = None)
  604. try:
  605. parser.read([file_path])
  606. except Exception:
  607. # Skip, it is not a valid stack file
  608. continue
  609. if not parser.has_option("general", "name"):
  610. continue
  611. if parser["general"]["name"] == name:
  612. # Load the container
  613. container_id = os.path.basename(file_path).replace(".inst.cfg", "")
  614. if self.findInstanceContainers(id = container_id):
  615. # This container is already in the registry, skip it
  616. continue
  617. instance_container = InstanceContainer(container_id)
  618. with open(file_path, "r", encoding = "utf-8") as f:
  619. serialized = f.read()
  620. try:
  621. instance_container.deserialize(serialized, file_path)
  622. except ContainerFormatError:
  623. Logger.logException("e", "Unable to deserialize InstanceContainer %s", file_path)
  624. continue
  625. self.addContainer(instance_container)
  626. break
  627. return instance_container
  628. # Fix the extruders that were upgraded to ExtruderStack instances during addContainer.
  629. # The stacks are now responsible for setting the next stack on deserialize. However,
  630. # due to problems with loading order, some stacks may not have the proper next stack
  631. # set after upgrading, because the proper global stack was not yet loaded. This method
  632. # makes sure those extruders also get the right stack set.
  633. def _connectUpgradedExtruderStacksToMachines(self) -> None:
  634. extruder_stacks = self.findContainers(container_type = ExtruderStack.ExtruderStack)
  635. for extruder_stack in extruder_stacks:
  636. if extruder_stack.getNextStack():
  637. # Has the right next stack, so ignore it.
  638. continue
  639. machines = ContainerRegistry.getInstance().findContainerStacks(id = extruder_stack.getMetaDataEntry("machine", ""))
  640. if machines:
  641. extruder_stack.setNextStack(machines[0])
  642. else:
  643. Logger.log("w", "Could not find machine {machine} for extruder {extruder}", machine = extruder_stack.getMetaDataEntry("machine"), extruder = extruder_stack.getId())
  644. # Override just for the type.
  645. @classmethod
  646. @override(ContainerRegistry)
  647. def getInstance(cls, *args, **kwargs) -> "CuraContainerRegistry":
  648. return cast(CuraContainerRegistry, super().getInstance(*args, **kwargs))