ThreeMFWorkspaceReader.py 58 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374757677787980818283848586878889909192939495969798991001011021031041051061071081091101111121131141151161171181191201211221231241251261271281291301311321331341351361371381391401411421431441451461471481491501511521531541551561571581591601611621631641651661671681691701711721731741751761771781791801811821831841851861871881891901911921931941951961971981992002012022032042052062072082092102112122132142152162172182192202212222232242252262272282292302312322332342352362372382392402412422432442452462472482492502512522532542552562572582592602612622632642652662672682692702712722732742752762772782792802812822832842852862872882892902912922932942952962972982993003013023033043053063073083093103113123133143153163173183193203213223233243253263273283293303313323333343353363373383393403413423433443453463473483493503513523533543553563573583593603613623633643653663673683693703713723733743753763773783793803813823833843853863873883893903913923933943953963973983994004014024034044054064074084094104114124134144154164174184194204214224234244254264274284294304314324334344354364374384394404414424434444454464474484494504514524534544554564574584594604614624634644654664674684694704714724734744754764774784794804814824834844854864874884894904914924934944954964974984995005015025035045055065075085095105115125135145155165175185195205215225235245255265275285295305315325335345355365375385395405415425435445455465475485495505515525535545555565575585595605615625635645655665675685695705715725735745755765775785795805815825835845855865875885895905915925935945955965975985996006016026036046056066076086096106116126136146156166176186196206216226236246256266276286296306316326336346356366376386396406416426436446456466476486496506516526536546556566576586596606616626636646656666676686696706716726736746756766776786796806816826836846856866876886896906916926936946956966976986997007017027037047057067077087097107117127137147157167177187197207217227237247257267277287297307317327337347357367377387397407417427437447457467477487497507517527537547557567577587597607617627637647657667677687697707717727737747757767777787797807817827837847857867877887897907917927937947957967977987998008018028038048058068078088098108118128138148158168178188198208218228238248258268278288298308318328338348358368378388398408418428438448458468478488498508518528538548558568578588598608618628638648658668678688698708718728738748758768778788798808818828838848858868878888898908918928938948958968978988999009019029039049059069079089099109119129139149159169179189199209219229239249259269279289299309319329339349359369379389399409419429439449459469479489499509519529539549559569579589599609619629639649659669679689699709719729739749759769779789799809819829839849859869879889899909919929939949959969979989991000100110021003100410051006100710081009101010111012101310141015101610171018101910201021102210231024102510261027102810291030103110321033103410351036103710381039104010411042104310441045104610471048104910501051105210531054105510561057105810591060106110621063106410651066106710681069107010711072107310741075107610771078107910801081108210831084108510861087108810891090
  1. # Copyright (c) 2019 Ultimaker B.V.
  2. # Cura is released under the terms of the LGPLv3 or higher.
  3. from configparser import ConfigParser
  4. import zipfile
  5. import os
  6. from typing import cast, Dict, List, Optional, Tuple
  7. import xml.etree.ElementTree as ET
  8. from UM.Workspace.WorkspaceReader import WorkspaceReader
  9. from UM.Application import Application
  10. from UM.Logger import Logger
  11. from UM.Message import Message
  12. from UM.i18n import i18nCatalog
  13. from UM.Settings.ContainerFormatError import ContainerFormatError
  14. from UM.Settings.ContainerStack import ContainerStack
  15. from UM.Settings.DefinitionContainer import DefinitionContainer
  16. from UM.Settings.InstanceContainer import InstanceContainer
  17. from UM.Settings.ContainerRegistry import ContainerRegistry
  18. from UM.MimeTypeDatabase import MimeTypeDatabase, MimeType
  19. from UM.Job import Job
  20. from UM.Preferences import Preferences
  21. from cura.Machines.ContainerTree import ContainerTree
  22. from cura.Machines.VariantType import VariantType
  23. from cura.Settings.CuraStackBuilder import CuraStackBuilder
  24. from cura.Settings.ExtruderManager import ExtruderManager
  25. from cura.Settings.ExtruderStack import ExtruderStack
  26. from cura.Settings.GlobalStack import GlobalStack
  27. from cura.Settings.CuraContainerStack import _ContainerIndexes
  28. from cura.CuraApplication import CuraApplication
  29. from cura.Utils.Threading import call_on_qt_thread
  30. from PyQt5.QtCore import QCoreApplication
  31. from .WorkspaceDialog import WorkspaceDialog
  32. i18n_catalog = i18nCatalog("cura")
  33. class ContainerInfo:
  34. def __init__(self, file_name: Optional[str], serialized: Optional[str], parser: Optional[ConfigParser]) -> None:
  35. self.file_name = file_name
  36. self.serialized = serialized
  37. self.parser = parser
  38. self.container = None
  39. self.definition_id = None
  40. class QualityChangesInfo:
  41. def __init__(self) -> None:
  42. self.name = None
  43. self.global_info = None
  44. self.extruder_info_dict = {} # type: Dict[str, ContainerInfo]
  45. class MachineInfo:
  46. def __init__(self) -> None:
  47. self.container_id = None
  48. self.name = None
  49. self.definition_id = None
  50. self.metadata_dict = {} # type: Dict[str, str]
  51. self.quality_type = None
  52. self.custom_quality_name = None
  53. self.quality_changes_info = None
  54. self.variant_info = None
  55. self.definition_changes_info = None
  56. self.user_changes_info = None
  57. self.extruder_info_dict = {} # type: Dict[str, ExtruderInfo]
  58. class ExtruderInfo:
  59. def __init__(self) -> None:
  60. self.position = None
  61. self.enabled = True
  62. self.variant_info = None
  63. self.root_material_id = None
  64. self.definition_changes_info = None
  65. self.user_changes_info = None
  66. ## Base implementation for reading 3MF workspace files.
  67. class ThreeMFWorkspaceReader(WorkspaceReader):
  68. def __init__(self) -> None:
  69. super().__init__()
  70. self._supported_extensions = [".3mf"]
  71. self._dialog = WorkspaceDialog()
  72. self._3mf_mesh_reader = None
  73. self._container_registry = ContainerRegistry.getInstance()
  74. # suffixes registered with the MimeTypes don't start with a dot '.'
  75. self._definition_container_suffix = "." + cast(MimeType, ContainerRegistry.getMimeTypeForContainer(DefinitionContainer)).preferredSuffix
  76. self._material_container_suffix = None # We have to wait until all other plugins are loaded before we can set it
  77. self._instance_container_suffix = "." + cast(MimeType, ContainerRegistry.getMimeTypeForContainer(InstanceContainer)).preferredSuffix
  78. self._container_stack_suffix = "." + cast(MimeType, ContainerRegistry.getMimeTypeForContainer(ContainerStack)).preferredSuffix
  79. self._extruder_stack_suffix = "." + cast(MimeType, ContainerRegistry.getMimeTypeForContainer(ExtruderStack)).preferredSuffix
  80. self._global_stack_suffix = "." + cast(MimeType, ContainerRegistry.getMimeTypeForContainer(GlobalStack)).preferredSuffix
  81. # Certain instance container types are ignored because we make the assumption that only we make those types
  82. # of containers. They are:
  83. # - quality
  84. # - variant
  85. self._ignored_instance_container_types = {"quality", "variant"}
  86. self._resolve_strategies = {} # type: Dict[str, str]
  87. self._id_mapping = {} # type: Dict[str, str]
  88. # In Cura 2.5 and 2.6, the empty profiles used to have those long names
  89. self._old_empty_profile_id_dict = {"empty_%s" % k: "empty" for k in ["material", "variant"]}
  90. self._is_same_machine_type = False
  91. self._old_new_materials = {} # type: Dict[str, str]
  92. self._machine_info = None
  93. def _clearState(self):
  94. self._is_same_machine_type = False
  95. self._id_mapping = {}
  96. self._old_new_materials = {}
  97. self._machine_info = None
  98. ## Get a unique name based on the old_id. This is different from directly calling the registry in that it caches results.
  99. # This has nothing to do with speed, but with getting consistent new naming for instances & objects.
  100. def getNewId(self, old_id: str):
  101. if old_id not in self._id_mapping:
  102. self._id_mapping[old_id] = self._container_registry.uniqueName(old_id)
  103. return self._id_mapping[old_id]
  104. ## Separates the given file list into a list of GlobalStack files and a list of ExtruderStack files.
  105. #
  106. # In old versions, extruder stack files have the same suffix as container stack files ".stack.cfg".
  107. #
  108. def _determineGlobalAndExtruderStackFiles(self, project_file_name: str, file_list: List[str]) -> Tuple[str, List[str]]:
  109. archive = zipfile.ZipFile(project_file_name, "r")
  110. global_stack_file_list = [name for name in file_list if name.endswith(self._global_stack_suffix)]
  111. extruder_stack_file_list = [name for name in file_list if name.endswith(self._extruder_stack_suffix)]
  112. # separate container stack files and extruder stack files
  113. files_to_determine = [name for name in file_list if name.endswith(self._container_stack_suffix)]
  114. for file_name in files_to_determine:
  115. # FIXME: HACK!
  116. # We need to know the type of the stack file, but we can only know it if we deserialize it.
  117. # The default ContainerStack.deserialize() will connect signals, which is not desired in this case.
  118. # Since we know that the stack files are INI files, so we directly use the ConfigParser to parse them.
  119. serialized = archive.open(file_name).read().decode("utf-8")
  120. stack_config = ConfigParser(interpolation = None)
  121. stack_config.read_string(serialized)
  122. # sanity check
  123. if not stack_config.has_option("metadata", "type"):
  124. Logger.log("e", "%s in %s doesn't seem to be valid stack file", file_name, project_file_name)
  125. continue
  126. stack_type = stack_config.get("metadata", "type")
  127. if stack_type == "extruder_train":
  128. extruder_stack_file_list.append(file_name)
  129. elif stack_type == "machine":
  130. global_stack_file_list.append(file_name)
  131. else:
  132. Logger.log("w", "Unknown container stack type '%s' from %s in %s",
  133. stack_type, file_name, project_file_name)
  134. if len(global_stack_file_list) > 1:
  135. Logger.log("e", "More than one global stack file found: [{file_list}]".format(file_list = global_stack_file_list))
  136. #But we can recover by just getting the first global stack file.
  137. if len(global_stack_file_list) == 0:
  138. Logger.log("e", "No global stack file found!")
  139. raise FileNotFoundError("No global stack file found!")
  140. return global_stack_file_list[0], extruder_stack_file_list
  141. ## read some info so we can make decisions
  142. # \param file_name
  143. # \param show_dialog In case we use preRead() to check if a file is a valid project file, we don't want to show a dialog.
  144. def preRead(self, file_name, show_dialog=True, *args, **kwargs):
  145. self._clearState()
  146. self._3mf_mesh_reader = Application.getInstance().getMeshFileHandler().getReaderForFile(file_name)
  147. if self._3mf_mesh_reader and self._3mf_mesh_reader.preRead(file_name) == WorkspaceReader.PreReadResult.accepted:
  148. pass
  149. else:
  150. Logger.log("w", "Could not find reader that was able to read the scene data for 3MF workspace")
  151. return WorkspaceReader.PreReadResult.failed
  152. self._machine_info = MachineInfo()
  153. machine_type = ""
  154. variant_type_name = i18n_catalog.i18nc("@label", "Nozzle")
  155. # Check if there are any conflicts, so we can ask the user.
  156. archive = zipfile.ZipFile(file_name, "r")
  157. cura_file_names = [name for name in archive.namelist() if name.startswith("Cura/")]
  158. resolve_strategy_keys = ["machine", "material", "quality_changes"]
  159. self._resolve_strategies = {k: None for k in resolve_strategy_keys}
  160. containers_found_dict = {k: False for k in resolve_strategy_keys}
  161. #
  162. # Read definition containers
  163. #
  164. machine_definition_id = None
  165. machine_definition_container_count = 0
  166. extruder_definition_container_count = 0
  167. definition_container_files = [name for name in cura_file_names if name.endswith(self._definition_container_suffix)]
  168. for definition_container_file in definition_container_files:
  169. container_id = self._stripFileToId(definition_container_file)
  170. definitions = self._container_registry.findDefinitionContainersMetadata(id = container_id)
  171. serialized = archive.open(definition_container_file).read().decode("utf-8")
  172. if not definitions:
  173. definition_container = DefinitionContainer.deserializeMetadata(serialized, container_id)[0]
  174. else:
  175. definition_container = definitions[0]
  176. definition_container_type = definition_container.get("type")
  177. if definition_container_type == "machine":
  178. machine_definition_id = container_id
  179. machine_type = definition_container["name"]
  180. variant_type_name = definition_container.get("variants_name", variant_type_name)
  181. machine_definition_container_count += 1
  182. elif definition_container_type == "extruder":
  183. extruder_definition_container_count += 1
  184. else:
  185. Logger.log("w", "Unknown definition container type %s for %s",
  186. definition_container_type, definition_container_file)
  187. QCoreApplication.processEvents() # Ensure that the GUI does not freeze.
  188. Job.yieldThread()
  189. if machine_definition_container_count != 1:
  190. return WorkspaceReader.PreReadResult.failed # Not a workspace file but ordinary 3MF.
  191. material_labels = []
  192. material_conflict = False
  193. xml_material_profile = self._getXmlProfileClass()
  194. reverse_material_id_dict = {}
  195. if self._material_container_suffix is None:
  196. self._material_container_suffix = ContainerRegistry.getMimeTypeForContainer(xml_material_profile).preferredSuffix
  197. if xml_material_profile:
  198. material_container_files = [name for name in cura_file_names if name.endswith(self._material_container_suffix)]
  199. for material_container_file in material_container_files:
  200. container_id = self._stripFileToId(material_container_file)
  201. serialized = archive.open(material_container_file).read().decode("utf-8")
  202. metadata_list = xml_material_profile.deserializeMetadata(serialized, container_id)
  203. reverse_map = {metadata["id"]: container_id for metadata in metadata_list}
  204. reverse_material_id_dict.update(reverse_map)
  205. material_labels.append(self._getMaterialLabelFromSerialized(serialized))
  206. if self._container_registry.findContainersMetadata(id = container_id): #This material already exists.
  207. containers_found_dict["material"] = True
  208. if not self._container_registry.isReadOnly(container_id): # Only non readonly materials can be in conflict
  209. material_conflict = True
  210. QCoreApplication.processEvents() # Ensure that the GUI does not freeze.
  211. Job.yieldThread()
  212. # Check if any quality_changes instance container is in conflict.
  213. instance_container_files = [name for name in cura_file_names if name.endswith(self._instance_container_suffix)]
  214. quality_name = ""
  215. custom_quality_name = ""
  216. num_settings_overridden_by_quality_changes = 0 # How many settings are changed by the quality changes
  217. num_user_settings = 0
  218. quality_changes_conflict = False
  219. self._machine_info.quality_changes_info = QualityChangesInfo()
  220. quality_changes_info_list = []
  221. instance_container_info_dict = {} # id -> parser
  222. for instance_container_file_name in instance_container_files:
  223. container_id = self._stripFileToId(instance_container_file_name)
  224. serialized = archive.open(instance_container_file_name).read().decode("utf-8")
  225. # Qualities and variants don't have upgrades, so don't upgrade them
  226. parser = ConfigParser(interpolation = None)
  227. parser.read_string(serialized)
  228. container_type = parser["metadata"]["type"]
  229. if container_type not in ("quality", "variant"):
  230. serialized = InstanceContainer._updateSerialized(serialized, instance_container_file_name)
  231. parser = ConfigParser(interpolation = None)
  232. parser.read_string(serialized)
  233. container_info = ContainerInfo(instance_container_file_name, serialized, parser)
  234. instance_container_info_dict[container_id] = container_info
  235. container_type = parser["metadata"]["type"]
  236. if container_type == "quality_changes":
  237. quality_changes_info_list.append(container_info)
  238. if not parser.has_option("metadata", "position"):
  239. self._machine_info.quality_changes_info.name = parser["general"]["name"]
  240. self._machine_info.quality_changes_info.global_info = container_info
  241. else:
  242. position = parser["metadata"]["position"]
  243. self._machine_info.quality_changes_info.extruder_info_dict[position] = container_info
  244. custom_quality_name = parser["general"]["name"]
  245. values = parser["values"] if parser.has_section("values") else dict()
  246. num_settings_overridden_by_quality_changes += len(values)
  247. # Check if quality changes already exists.
  248. quality_changes = self._container_registry.findInstanceContainers(name = custom_quality_name,
  249. type = "quality_changes")
  250. if quality_changes:
  251. containers_found_dict["quality_changes"] = True
  252. # Check if there really is a conflict by comparing the values
  253. instance_container = InstanceContainer(container_id)
  254. try:
  255. instance_container.deserialize(serialized, file_name = instance_container_file_name)
  256. except ContainerFormatError:
  257. Logger.logException("e", "Failed to deserialize InstanceContainer %s from project file %s",
  258. instance_container_file_name, file_name)
  259. return ThreeMFWorkspaceReader.PreReadResult.failed
  260. if quality_changes[0] != instance_container:
  261. quality_changes_conflict = True
  262. elif container_type == "quality":
  263. if not quality_name:
  264. quality_name = parser["general"]["name"]
  265. elif container_type == "user":
  266. num_user_settings += len(parser["values"])
  267. elif container_type in self._ignored_instance_container_types:
  268. # Ignore certain instance container types
  269. Logger.log("w", "Ignoring instance container [%s] with type [%s]", container_id, container_type)
  270. continue
  271. QCoreApplication.processEvents() # Ensure that the GUI does not freeze.
  272. Job.yieldThread()
  273. if self._machine_info.quality_changes_info.global_info is None:
  274. self._machine_info.quality_changes_info = None
  275. # Load ContainerStack files and ExtruderStack files
  276. try:
  277. global_stack_file, extruder_stack_files = self._determineGlobalAndExtruderStackFiles(
  278. file_name, cura_file_names)
  279. except FileNotFoundError:
  280. return WorkspaceReader.PreReadResult.failed
  281. machine_conflict = False
  282. # Because there can be cases as follows:
  283. # - the global stack exists but some/all of the extruder stacks DON'T exist
  284. # - the global stack DOESN'T exist but some/all of the extruder stacks exist
  285. # To simplify this, only check if the global stack exists or not
  286. global_stack_id = self._stripFileToId(global_stack_file)
  287. serialized = archive.open(global_stack_file).read().decode("utf-8")
  288. machine_name = self._getMachineNameFromSerializedStack(serialized)
  289. self._machine_info.metadata_dict = self._getMetaDataDictFromSerializedStack(serialized)
  290. stacks = self._container_registry.findContainerStacks(name = machine_name, type = "machine")
  291. self._is_same_machine_type = True
  292. existing_global_stack = None
  293. if stacks:
  294. global_stack = stacks[0]
  295. existing_global_stack = global_stack
  296. containers_found_dict["machine"] = True
  297. # Check if there are any changes at all in any of the container stacks.
  298. id_list = self._getContainerIdListFromSerialized(serialized)
  299. for index, container_id in enumerate(id_list):
  300. # take into account the old empty container IDs
  301. container_id = self._old_empty_profile_id_dict.get(container_id, container_id)
  302. if global_stack.getContainer(index).getId() != container_id:
  303. machine_conflict = True
  304. break
  305. self._is_same_machine_type = global_stack.definition.getId() == machine_definition_id
  306. # Get quality type
  307. parser = ConfigParser(interpolation = None)
  308. parser.read_string(serialized)
  309. index_map_version = _ContainerIndexes.getIndexMapping(int(parser["metadata"]["setting_version"]))
  310. quality_container_id = parser["containers"][str(index_map_version[_ContainerIndexes.Quality])]
  311. quality_type = "empty_quality"
  312. if quality_container_id not in ("empty", "empty_quality"):
  313. quality_type = instance_container_info_dict[quality_container_id].parser["metadata"]["quality_type"]
  314. # Get machine info
  315. serialized = archive.open(global_stack_file).read().decode("utf-8")
  316. serialized = GlobalStack._updateSerialized(serialized, global_stack_file)
  317. parser = ConfigParser(interpolation = None)
  318. parser.read_string(serialized)
  319. index_map_version = _ContainerIndexes.getIndexMapping(int(parser["metadata"]["setting_version"]))
  320. definition_changes_id = parser["containers"][str(index_map_version[_ContainerIndexes.DefinitionChanges])]
  321. if definition_changes_id not in ("empty", "empty_definition_changes"):
  322. self._machine_info.definition_changes_info = instance_container_info_dict[definition_changes_id]
  323. user_changes_id = parser["containers"][str(index_map_version[_ContainerIndexes.UserChanges])]
  324. if user_changes_id not in ("empty", "empty_user_changes"):
  325. self._machine_info.user_changes_info = instance_container_info_dict[user_changes_id]
  326. # Also check variant and material in case it doesn't have extruder stacks
  327. if not extruder_stack_files:
  328. position = "0"
  329. extruder_info = ExtruderInfo()
  330. extruder_info.position = position
  331. variant_id = parser["containers"][str(index_map_version[_ContainerIndexes.Variant])]
  332. material_id = parser["containers"][str(index_map_version[_ContainerIndexes.Material])]
  333. if variant_id not in ("empty", "empty_variant"):
  334. extruder_info.variant_info = instance_container_info_dict[variant_id]
  335. if material_id not in ("empty", "empty_material"):
  336. root_material_id = reverse_material_id_dict[material_id]
  337. extruder_info.root_material_id = root_material_id
  338. self._machine_info.extruder_info_dict[position] = extruder_info
  339. else:
  340. variant_id = parser["containers"][str(index_map_version[_ContainerIndexes.Variant])]
  341. if variant_id not in ("empty", "empty_variant"):
  342. self._machine_info.variant_info = instance_container_info_dict[variant_id]
  343. QCoreApplication.processEvents() # Ensure that the GUI does not freeze.
  344. Job.yieldThread()
  345. # if the global stack is found, we check if there are conflicts in the extruder stacks
  346. for extruder_stack_file in extruder_stack_files:
  347. serialized = archive.open(extruder_stack_file).read().decode("utf-8")
  348. serialized = ExtruderStack._updateSerialized(serialized, extruder_stack_file)
  349. parser = ConfigParser(interpolation = None)
  350. parser.read_string(serialized)
  351. index_map_version = _ContainerIndexes.getIndexMapping(int(parser["metadata"]["setting_version"]))
  352. # The check should be done for the extruder stack that's associated with the existing global stack,
  353. # and those extruder stacks may have different IDs.
  354. # So we check according to the positions
  355. position = parser["metadata"]["position"]
  356. variant_id = parser["containers"][str(index_map_version[_ContainerIndexes.Variant])]
  357. material_id = parser["containers"][str(index_map_version[_ContainerIndexes.Material])]
  358. extruder_info = ExtruderInfo()
  359. extruder_info.position = position
  360. if parser.has_option("metadata", "enabled"):
  361. extruder_info.enabled = parser["metadata"]["enabled"]
  362. if variant_id not in ("empty", "empty_variant"):
  363. if variant_id in instance_container_info_dict:
  364. extruder_info.variant_info = instance_container_info_dict[variant_id]
  365. if material_id not in ("empty", "empty_material"):
  366. root_material_id = reverse_material_id_dict[material_id]
  367. extruder_info.root_material_id = root_material_id
  368. definition_changes_id = parser["containers"][str(index_map_version[_ContainerIndexes.DefinitionChanges])]
  369. if definition_changes_id not in ("empty", "empty_definition_changes"):
  370. extruder_info.definition_changes_info = instance_container_info_dict[definition_changes_id]
  371. user_changes_id = parser["containers"][str(index_map_version[_ContainerIndexes.UserChanges])]
  372. if user_changes_id not in ("empty", "empty_user_changes"):
  373. extruder_info.user_changes_info = instance_container_info_dict[user_changes_id]
  374. self._machine_info.extruder_info_dict[position] = extruder_info
  375. if not machine_conflict and containers_found_dict["machine"]:
  376. if position not in global_stack.extruders:
  377. continue
  378. existing_extruder_stack = global_stack.extruders[position]
  379. # check if there are any changes at all in any of the container stacks.
  380. id_list = self._getContainerIdListFromSerialized(serialized)
  381. for index, container_id in enumerate(id_list):
  382. # take into account the old empty container IDs
  383. container_id = self._old_empty_profile_id_dict.get(container_id, container_id)
  384. if existing_extruder_stack.getContainer(index).getId() != container_id:
  385. machine_conflict = True
  386. break
  387. num_visible_settings = 0
  388. try:
  389. temp_preferences = Preferences()
  390. serialized = archive.open("Cura/preferences.cfg").read().decode("utf-8")
  391. temp_preferences.deserialize(serialized)
  392. visible_settings_string = temp_preferences.getValue("general/visible_settings")
  393. has_visible_settings_string = visible_settings_string is not None
  394. if visible_settings_string is not None:
  395. num_visible_settings = len(visible_settings_string.split(";"))
  396. active_mode = temp_preferences.getValue("cura/active_mode")
  397. if not active_mode:
  398. active_mode = Application.getInstance().getPreferences().getValue("cura/active_mode")
  399. except KeyError:
  400. # If there is no preferences file, it's not a workspace, so notify user of failure.
  401. Logger.log("w", "File %s is not a valid workspace.", file_name)
  402. return WorkspaceReader.PreReadResult.failed
  403. # Check if the machine definition exists. If not, indicate failure because we do not import definition files.
  404. def_results = self._container_registry.findDefinitionContainersMetadata(id = machine_definition_id)
  405. if not def_results:
  406. message = Message(i18n_catalog.i18nc("@info:status Don't translate the XML tags <filename> or <message>!",
  407. "Project file <filename>{0}</filename> contains an unknown machine type"
  408. " <message>{1}</message>. Cannot import the machine."
  409. " Models will be imported instead.", file_name, machine_definition_id),
  410. title = i18n_catalog.i18nc("@info:title", "Open Project File"))
  411. message.show()
  412. Logger.log("i", "Could unknown machine definition %s in project file %s, cannot import it.",
  413. self._machine_info.definition_id, file_name)
  414. return WorkspaceReader.PreReadResult.failed
  415. # In case we use preRead() to check if a file is a valid project file, we don't want to show a dialog.
  416. if not show_dialog:
  417. return WorkspaceReader.PreReadResult.accepted
  418. # prepare data for the dialog
  419. num_extruders = extruder_definition_container_count
  420. if num_extruders == 0:
  421. num_extruders = 1 # No extruder stacks found, which means there is one extruder
  422. extruders = num_extruders * [""]
  423. quality_name = custom_quality_name if custom_quality_name else quality_name
  424. self._machine_info.container_id = global_stack_id
  425. self._machine_info.name = machine_name
  426. self._machine_info.definition_id = machine_definition_id
  427. self._machine_info.quality_type = quality_type
  428. self._machine_info.custom_quality_name = quality_name
  429. if machine_conflict and not self._is_same_machine_type:
  430. machine_conflict = False
  431. is_printer_group = False
  432. if machine_conflict:
  433. group_name = existing_global_stack.getMetaDataEntry("group_name")
  434. if group_name is not None:
  435. is_printer_group = True
  436. machine_name = group_name
  437. # Show the dialog, informing the user what is about to happen.
  438. self._dialog.setMachineConflict(machine_conflict)
  439. self._dialog.setIsPrinterGroup(is_printer_group)
  440. self._dialog.setQualityChangesConflict(quality_changes_conflict)
  441. self._dialog.setMaterialConflict(material_conflict)
  442. self._dialog.setHasVisibleSettingsField(has_visible_settings_string)
  443. self._dialog.setNumVisibleSettings(num_visible_settings)
  444. self._dialog.setQualityName(quality_name)
  445. self._dialog.setQualityType(quality_type)
  446. self._dialog.setNumSettingsOverriddenByQualityChanges(num_settings_overridden_by_quality_changes)
  447. self._dialog.setNumUserSettings(num_user_settings)
  448. self._dialog.setActiveMode(active_mode)
  449. self._dialog.setMachineName(machine_name)
  450. self._dialog.setMaterialLabels(material_labels)
  451. self._dialog.setMachineType(machine_type)
  452. self._dialog.setExtruders(extruders)
  453. self._dialog.setVariantType(variant_type_name)
  454. self._dialog.setHasObjectsOnPlate(Application.getInstance().platformActivity)
  455. self._dialog.show()
  456. # Block until the dialog is closed.
  457. self._dialog.waitForClose()
  458. if self._dialog.getResult() == {}:
  459. return WorkspaceReader.PreReadResult.cancelled
  460. self._resolve_strategies = self._dialog.getResult()
  461. #
  462. # There can be 3 resolve strategies coming from the dialog:
  463. # - new: create a new container
  464. # - override: override the existing container
  465. # - None: There is no conflict, which means containers with the same IDs may or may not be there already.
  466. # If there is an existing container, there is no conflict between them, and default to "override"
  467. # If there is no existing container, default to "new"
  468. #
  469. # Default values
  470. for key, strategy in self._resolve_strategies.items():
  471. if key not in containers_found_dict or strategy is not None:
  472. continue
  473. self._resolve_strategies[key] = "override" if containers_found_dict[key] else "new"
  474. return WorkspaceReader.PreReadResult.accepted
  475. ## Read the project file
  476. # Add all the definitions / materials / quality changes that do not exist yet. Then it loads
  477. # all the stacks into the container registry. In some cases it will reuse the container for the global stack.
  478. # It handles old style project files containing .stack.cfg as well as new style project files
  479. # containing global.cfg / extruder.cfg
  480. #
  481. # \param file_name
  482. @call_on_qt_thread
  483. def read(self, file_name):
  484. application = CuraApplication.getInstance()
  485. archive = zipfile.ZipFile(file_name, "r")
  486. cura_file_names = [name for name in archive.namelist() if name.startswith("Cura/")]
  487. # Create a shadow copy of the preferences (we don't want all of the preferences, but we do want to re-use its
  488. # parsing code.
  489. temp_preferences = Preferences()
  490. serialized = archive.open("Cura/preferences.cfg").read().decode("utf-8")
  491. temp_preferences.deserialize(serialized)
  492. # Copy a number of settings from the temp preferences to the global
  493. global_preferences = application.getInstance().getPreferences()
  494. visible_settings = temp_preferences.getValue("general/visible_settings")
  495. if visible_settings is None:
  496. Logger.log("w", "Workspace did not contain visible settings. Leaving visibility unchanged")
  497. else:
  498. global_preferences.setValue("general/visible_settings", visible_settings)
  499. global_preferences.setValue("cura/active_setting_visibility_preset", "custom")
  500. categories_expanded = temp_preferences.getValue("cura/categories_expanded")
  501. if categories_expanded is None:
  502. Logger.log("w", "Workspace did not contain expanded categories. Leaving them unchanged")
  503. else:
  504. global_preferences.setValue("cura/categories_expanded", categories_expanded)
  505. application.expandedCategoriesChanged.emit() # Notify the GUI of the change
  506. # If a machine with the same name is of a different type, always create a new one.
  507. if not self._is_same_machine_type or self._resolve_strategies["machine"] != "override":
  508. # We need to create a new machine
  509. machine_name = self._container_registry.uniqueName(self._machine_info.name)
  510. global_stack = CuraStackBuilder.createMachine(machine_name, self._machine_info.definition_id)
  511. if global_stack: #Only switch if creating the machine was successful.
  512. extruder_stack_dict = global_stack.extruders
  513. self._container_registry.addContainer(global_stack)
  514. else:
  515. # Find the machine
  516. global_stack = self._container_registry.findContainerStacks(name = self._machine_info.name, type = "machine")[0]
  517. extruder_stacks = self._container_registry.findContainerStacks(machine = global_stack.getId(),
  518. type = "extruder_train")
  519. extruder_stack_dict = {stack.getMetaDataEntry("position"): stack for stack in extruder_stacks}
  520. # Make sure that those extruders have the global stack as the next stack or later some value evaluation
  521. # will fail.
  522. for stack in extruder_stacks:
  523. stack.setNextStack(global_stack, connect_signals = False)
  524. Logger.log("d", "Workspace loading is checking definitions...")
  525. # Get all the definition files & check if they exist. If not, add them.
  526. definition_container_files = [name for name in cura_file_names if name.endswith(self._definition_container_suffix)]
  527. for definition_container_file in definition_container_files:
  528. container_id = self._stripFileToId(definition_container_file)
  529. definitions = self._container_registry.findDefinitionContainersMetadata(id = container_id)
  530. if not definitions:
  531. definition_container = DefinitionContainer(container_id)
  532. try:
  533. definition_container.deserialize(archive.open(definition_container_file).read().decode("utf-8"),
  534. file_name = definition_container_file)
  535. except ContainerFormatError:
  536. # We cannot just skip the definition file because everything else later will just break if the
  537. # machine definition cannot be found.
  538. Logger.logException("e", "Failed to deserialize definition file %s in project file %s",
  539. definition_container_file, file_name)
  540. definition_container = self._container_registry.findDefinitionContainers(id = "fdmprinter")[0] #Fall back to defaults.
  541. self._container_registry.addContainer(definition_container)
  542. Job.yieldThread()
  543. QCoreApplication.processEvents() # Ensure that the GUI does not freeze.
  544. Logger.log("d", "Workspace loading is checking materials...")
  545. # Get all the material files and check if they exist. If not, add them.
  546. xml_material_profile = self._getXmlProfileClass()
  547. if self._material_container_suffix is None:
  548. self._material_container_suffix = ContainerRegistry.getMimeTypeForContainer(xml_material_profile).suffixes[0]
  549. if xml_material_profile:
  550. material_container_files = [name for name in cura_file_names if name.endswith(self._material_container_suffix)]
  551. for material_container_file in material_container_files:
  552. to_deserialize_material = False
  553. container_id = self._stripFileToId(material_container_file)
  554. need_new_name = False
  555. materials = self._container_registry.findInstanceContainers(id = container_id)
  556. if not materials:
  557. # No material found, deserialize this material later and add it
  558. to_deserialize_material = True
  559. else:
  560. material_container = materials[0]
  561. old_material_root_id = material_container.getMetaDataEntry("base_file")
  562. if old_material_root_id is not None and not self._container_registry.isReadOnly(old_material_root_id): # Only create new materials if they are not read only.
  563. to_deserialize_material = True
  564. if self._resolve_strategies["material"] == "override":
  565. # Remove the old materials and then deserialize the one from the project
  566. root_material_id = material_container.getMetaDataEntry("base_file")
  567. application.getContainerRegistry().removeContainer(root_material_id)
  568. elif self._resolve_strategies["material"] == "new":
  569. # Note that we *must* deserialize it with a new ID, as multiple containers will be
  570. # auto created & added.
  571. container_id = self.getNewId(container_id)
  572. self._old_new_materials[old_material_root_id] = container_id
  573. need_new_name = True
  574. if to_deserialize_material:
  575. material_container = xml_material_profile(container_id)
  576. try:
  577. material_container.deserialize(archive.open(material_container_file).read().decode("utf-8"),
  578. file_name = container_id + "." + self._material_container_suffix)
  579. except ContainerFormatError:
  580. Logger.logException("e", "Failed to deserialize material file %s in project file %s",
  581. material_container_file, file_name)
  582. continue
  583. if need_new_name:
  584. new_name = ContainerRegistry.getInstance().uniqueName(material_container.getName())
  585. material_container.setName(new_name)
  586. material_container.setDirty(True)
  587. self._container_registry.addContainer(material_container)
  588. Job.yieldThread()
  589. QCoreApplication.processEvents() # Ensure that the GUI does not freeze.
  590. # Handle quality changes if any
  591. self._processQualityChanges(global_stack)
  592. # Prepare the machine
  593. self._applyChangesToMachine(global_stack, extruder_stack_dict)
  594. Logger.log("d", "Workspace loading is notifying rest of the code of changes...")
  595. # Actually change the active machine.
  596. #
  597. # This is scheduled for later is because it depends on the Variant/Material/Qualitiy Managers to have the latest
  598. # data, but those managers will only update upon a container/container metadata changed signal. Because this
  599. # function is running on the main thread (Qt thread), although those "changed" signals have been emitted, but
  600. # they won't take effect until this function is done.
  601. # To solve this, we schedule _updateActiveMachine() for later so it will have the latest data.
  602. self._updateActiveMachine(global_stack)
  603. # Load all the nodes / mesh data of the workspace
  604. nodes = self._3mf_mesh_reader.read(file_name)
  605. if nodes is None:
  606. nodes = []
  607. base_file_name = os.path.basename(file_name)
  608. self.setWorkspaceName(base_file_name)
  609. return nodes
  610. def _processQualityChanges(self, global_stack):
  611. if self._machine_info.quality_changes_info is None:
  612. return
  613. # If we have custom profiles, load them
  614. quality_changes_name = self._machine_info.quality_changes_info.name
  615. if self._machine_info.quality_changes_info is not None:
  616. Logger.log("i", "Loading custom profile [%s] from project file",
  617. self._machine_info.quality_changes_info.name)
  618. # Get the correct extruder definition IDs for quality changes
  619. machine_definition_id_for_quality = ContainerTree.getInstance().machines[global_stack.definition.getId()].quality_definition
  620. machine_definition_for_quality = self._container_registry.findDefinitionContainers(id = machine_definition_id_for_quality)[0]
  621. quality_changes_info = self._machine_info.quality_changes_info
  622. quality_changes_quality_type = quality_changes_info.global_info.parser["metadata"]["quality_type"]
  623. quality_changes_intent_category_per_extruder = {position: info.parser["metadata"].get("intent_category", "default") for position, info in quality_changes_info.extruder_info_dict.items()}
  624. quality_changes_name = quality_changes_info.name
  625. create_new = self._resolve_strategies.get("quality_changes") != "override"
  626. if create_new:
  627. container_info_dict = {None: self._machine_info.quality_changes_info.global_info}
  628. container_info_dict.update(quality_changes_info.extruder_info_dict)
  629. quality_changes_name = self._container_registry.uniqueName(quality_changes_name)
  630. for position, container_info in container_info_dict.items():
  631. extruder_stack = None
  632. intent_category = None # type: Optional[str]
  633. if position is not None:
  634. extruder_stack = global_stack.extruders[position]
  635. intent_category = quality_changes_intent_category_per_extruder[position]
  636. container = self._createNewQualityChanges(quality_changes_quality_type, intent_category, quality_changes_name, global_stack, extruder_stack)
  637. container_info.container = container
  638. self._container_registry.addContainer(container)
  639. Logger.log("d", "Created new quality changes container [%s]", container.getId())
  640. else:
  641. # Find the existing containers
  642. quality_changes_containers = self._container_registry.findInstanceContainers(name = quality_changes_name,
  643. type = "quality_changes")
  644. for container in quality_changes_containers:
  645. extruder_position = container.getMetaDataEntry("position")
  646. if extruder_position is None:
  647. quality_changes_info.global_info.container = container
  648. else:
  649. if extruder_position not in quality_changes_info.extruder_info_dict:
  650. quality_changes_info.extruder_info_dict[extruder_position] = ContainerInfo(None, None, None)
  651. container_info = quality_changes_info.extruder_info_dict[extruder_position]
  652. container_info.container = container
  653. # If there is no quality changes for any extruder, create one.
  654. if not quality_changes_info.extruder_info_dict:
  655. container_info = ContainerInfo(None, None, None)
  656. quality_changes_info.extruder_info_dict["0"] = container_info
  657. # If the global stack we're "targeting" has never been active, but was updated from Cura 3.4,
  658. # it might not have it's extruders set properly.
  659. if not global_stack.extruders:
  660. ExtruderManager.getInstance().fixSingleExtrusionMachineExtruderDefinition(global_stack)
  661. extruder_stack = global_stack.extruders["0"]
  662. intent_category = quality_changes_intent_category_per_extruder["0"]
  663. container = self._createNewQualityChanges(quality_changes_quality_type, intent_category, quality_changes_name, global_stack, extruder_stack)
  664. container_info.container = container
  665. self._container_registry.addContainer(container)
  666. Logger.log("d", "Created new quality changes container [%s]", container.getId())
  667. # Clear all existing containers
  668. quality_changes_info.global_info.container.clear()
  669. for container_info in quality_changes_info.extruder_info_dict.values():
  670. if container_info.container:
  671. container_info.container.clear()
  672. # Loop over everything and override the existing containers
  673. global_info = quality_changes_info.global_info
  674. global_info.container.clear() # Clear all
  675. for key, value in global_info.parser["values"].items():
  676. if not machine_definition_for_quality.getProperty(key, "settable_per_extruder"):
  677. global_info.container.setProperty(key, "value", value)
  678. else:
  679. quality_changes_info.extruder_info_dict["0"].container.setProperty(key, "value", value)
  680. for position, container_info in quality_changes_info.extruder_info_dict.items():
  681. if container_info.parser is None:
  682. continue
  683. if container_info.container is None:
  684. extruder_stack = global_stack.extruders[position]
  685. intent_category = quality_changes_intent_category_per_extruder[position]
  686. container = self._createNewQualityChanges(quality_changes_quality_type, intent_category, quality_changes_name, global_stack, extruder_stack)
  687. container_info.container = container
  688. self._container_registry.addContainer(container)
  689. for key, value in container_info.parser["values"].items():
  690. container_info.container.setProperty(key, "value", value)
  691. self._machine_info.quality_changes_info.name = quality_changes_name
  692. ## Helper class to create a new quality changes profile.
  693. #
  694. # This will then later be filled with the appropriate data.
  695. # \param quality_type The quality type of the new profile.
  696. # \param intent_category The intent category of the new profile.
  697. # \param name The name for the profile. This will later be made unique so
  698. # it doesn't need to be unique yet.
  699. # \param global_stack The global stack showing the configuration that the
  700. # profile should be created for.
  701. # \param extruder_stack The extruder stack showing the configuration that
  702. # the profile should be created for. If this is None, it will be created
  703. # for the global stack.
  704. def _createNewQualityChanges(self, quality_type: str, intent_category: Optional[str], name: str, global_stack: GlobalStack, extruder_stack: Optional[ExtruderStack]) -> InstanceContainer:
  705. container_registry = CuraApplication.getInstance().getContainerRegistry()
  706. base_id = global_stack.definition.getId() if extruder_stack is None else extruder_stack.getId()
  707. new_id = base_id + "_" + name
  708. new_id = new_id.lower().replace(" ", "_")
  709. new_id = container_registry.uniqueName(new_id)
  710. # Create a new quality_changes container for the quality.
  711. quality_changes = InstanceContainer(new_id)
  712. quality_changes.setName(name)
  713. quality_changes.setMetaDataEntry("type", "quality_changes")
  714. quality_changes.setMetaDataEntry("quality_type", quality_type)
  715. if intent_category is not None:
  716. quality_changes.setMetaDataEntry("intent_category", intent_category)
  717. # If we are creating a container for an extruder, ensure we add that to the container.
  718. if extruder_stack is not None:
  719. quality_changes.setMetaDataEntry("position", extruder_stack.getMetaDataEntry("position"))
  720. # If the machine specifies qualities should be filtered, ensure we match the current criteria.
  721. machine_definition_id = ContainerTree.getInstance().machines[global_stack.definition.getId()].quality_definition
  722. quality_changes.setDefinition(machine_definition_id)
  723. quality_changes.setMetaDataEntry("setting_version", CuraApplication.getInstance().SettingVersion)
  724. quality_changes.setDirty(True)
  725. return quality_changes
  726. @staticmethod
  727. def _clearStack(stack):
  728. application = CuraApplication.getInstance()
  729. stack.definitionChanges.clear()
  730. stack.variant = application.empty_variant_container
  731. stack.material = application.empty_material_container
  732. stack.quality = application.empty_quality_container
  733. stack.qualityChanges = application.empty_quality_changes_container
  734. stack.userChanges.clear()
  735. def _applyDefinitionChanges(self, global_stack, extruder_stack_dict):
  736. values_to_set_for_extruders = {}
  737. if self._machine_info.definition_changes_info is not None:
  738. parser = self._machine_info.definition_changes_info.parser
  739. for key, value in parser["values"].items():
  740. if global_stack.getProperty(key, "settable_per_extruder"):
  741. values_to_set_for_extruders[key] = value
  742. else:
  743. global_stack.definitionChanges.setProperty(key, "value", value)
  744. for position, extruder_stack in extruder_stack_dict.items():
  745. if position not in self._machine_info.extruder_info_dict:
  746. continue
  747. extruder_info = self._machine_info.extruder_info_dict[position]
  748. if extruder_info.definition_changes_info is None:
  749. continue
  750. parser = extruder_info.definition_changes_info.parser
  751. for key, value in values_to_set_for_extruders.items():
  752. extruder_stack.definitionChanges.setProperty(key, "value", value)
  753. if parser is not None:
  754. for key, value in parser["values"].items():
  755. extruder_stack.definitionChanges.setProperty(key, "value", value)
  756. def _applyUserChanges(self, global_stack, extruder_stack_dict):
  757. values_to_set_for_extruder_0 = {}
  758. if self._machine_info.user_changes_info is not None:
  759. parser = self._machine_info.user_changes_info.parser
  760. for key, value in parser["values"].items():
  761. if global_stack.getProperty(key, "settable_per_extruder"):
  762. values_to_set_for_extruder_0[key] = value
  763. else:
  764. global_stack.userChanges.setProperty(key, "value", value)
  765. for position, extruder_stack in extruder_stack_dict.items():
  766. if position not in self._machine_info.extruder_info_dict:
  767. continue
  768. extruder_info = self._machine_info.extruder_info_dict[position]
  769. if extruder_info.user_changes_info is not None:
  770. parser = self._machine_info.extruder_info_dict[position].user_changes_info.parser
  771. if position == "0":
  772. for key, value in values_to_set_for_extruder_0.items():
  773. extruder_stack.userChanges.setProperty(key, "value", value)
  774. if parser is not None:
  775. for key, value in parser["values"].items():
  776. extruder_stack.userChanges.setProperty(key, "value", value)
  777. def _applyVariants(self, global_stack, extruder_stack_dict):
  778. application = CuraApplication.getInstance()
  779. variant_manager = application.getVariantManager()
  780. if self._machine_info.variant_info is not None:
  781. parser = self._machine_info.variant_info.parser
  782. variant_name = parser["general"]["name"]
  783. variant_type = VariantType.BUILD_PLATE
  784. node = variant_manager.getVariantNode(global_stack.definition.getId(), variant_name, variant_type)
  785. if node is not None and node.container is not None:
  786. global_stack.variant = node.container
  787. for position, extruder_stack in extruder_stack_dict.items():
  788. if position not in self._machine_info.extruder_info_dict:
  789. continue
  790. extruder_info = self._machine_info.extruder_info_dict[position]
  791. if extruder_info.variant_info is None:
  792. # If there is no variant_info, try to use the default variant. Otherwise, leave it be.
  793. machine_node = ContainerTree.getInstance().machines[global_stack.definition.getId()]
  794. node = machine_node.variants[machine_node.preferred_variant_name]
  795. if node is not None and node.container is not None:
  796. extruder_stack.variant = node.container
  797. continue
  798. parser = extruder_info.variant_info.parser
  799. variant_name = parser["general"]["name"]
  800. variant_type = VariantType.NOZZLE
  801. node = ContainerTree.getInstance().machines[global_stack.definition.getId()].variants[variant_name]
  802. if node is not None and node.container is not None:
  803. extruder_stack.variant = node.container
  804. def _applyMaterials(self, global_stack, extruder_stack_dict):
  805. machine_node = ContainerTree.getInstance().machines[global_stack]
  806. for position, extruder_stack in extruder_stack_dict.items():
  807. if position not in self._machine_info.extruder_info_dict:
  808. continue
  809. extruder_info = self._machine_info.extruder_info_dict[position]
  810. if extruder_info.root_material_id is None:
  811. continue
  812. root_material_id = extruder_info.root_material_id
  813. root_material_id = self._old_new_materials.get(root_material_id, root_material_id)
  814. material_node = machine_node.variants[extruder_stack.variant.getName()].materials[root_material_id]
  815. extruder_stack.material = material_node.container # type: InstanceContainer
  816. def _applyChangesToMachine(self, global_stack, extruder_stack_dict):
  817. # Clear all first
  818. self._clearStack(global_stack)
  819. for extruder_stack in extruder_stack_dict.values():
  820. self._clearStack(extruder_stack)
  821. self._applyDefinitionChanges(global_stack, extruder_stack_dict)
  822. self._applyUserChanges(global_stack, extruder_stack_dict)
  823. self._applyVariants(global_stack, extruder_stack_dict)
  824. self._applyMaterials(global_stack, extruder_stack_dict)
  825. # prepare the quality to select
  826. self._quality_changes_to_apply = None
  827. self._quality_type_to_apply = None
  828. if self._machine_info.quality_changes_info is not None:
  829. self._quality_changes_to_apply = self._machine_info.quality_changes_info.name
  830. else:
  831. self._quality_type_to_apply = self._machine_info.quality_type
  832. # Set enabled/disabled for extruders
  833. for position, extruder_stack in extruder_stack_dict.items():
  834. extruder_info = self._machine_info.extruder_info_dict.get(position)
  835. if not extruder_info:
  836. continue
  837. if "enabled" not in extruder_stack.getMetaData():
  838. extruder_stack.setMetaDataEntry("enabled", "True")
  839. extruder_stack.setMetaDataEntry("enabled", str(extruder_info.enabled))
  840. # Set metadata fields that are missing from the global stack
  841. for key, value in self._machine_info.metadata_dict.items():
  842. if key not in global_stack.getMetaData():
  843. global_stack.setMetaDataEntry(key, value)
  844. def _updateActiveMachine(self, global_stack):
  845. # Actually change the active machine.
  846. machine_manager = Application.getInstance().getMachineManager()
  847. container_tree = ContainerTree.getInstance()
  848. machine_manager.setActiveMachine(global_stack.getId())
  849. # Set metadata fields that are missing from the global stack
  850. for key, value in self._machine_info.metadata_dict.items():
  851. if key not in global_stack.getMetaData():
  852. global_stack.setMetaDataEntry(key, value)
  853. if self._quality_changes_to_apply:
  854. quality_changes_group_list = container_tree.getCurrentQualityChangesGroups()
  855. quality_changes_group = next((qcg for qcg in quality_changes_group_list if qcg.name == self._quality_changes_to_apply), None)
  856. if not quality_changes_group:
  857. Logger.log("e", "Could not find quality_changes [%s]", self._quality_changes_to_apply)
  858. return
  859. machine_manager.setQualityChangesGroup(quality_changes_group, no_dialog = True)
  860. else:
  861. self._quality_type_to_apply = self._quality_type_to_apply.lower()
  862. quality_group_dict = container_tree.getCurrentQualityGroups()
  863. if self._quality_type_to_apply in quality_group_dict:
  864. quality_group = quality_group_dict[self._quality_type_to_apply]
  865. else:
  866. Logger.log("i", "Could not find quality type [%s], switch to default", self._quality_type_to_apply)
  867. preferred_quality_type = global_stack.getMetaDataEntry("preferred_quality_type")
  868. quality_group = quality_group_dict.get(preferred_quality_type)
  869. if quality_group is None:
  870. Logger.log("e", "Could not get preferred quality type [%s]", preferred_quality_type)
  871. if quality_group is not None:
  872. machine_manager.setQualityGroup(quality_group, no_dialog = True)
  873. # Notify everything/one that is to notify about changes.
  874. global_stack.containersChanged.emit(global_stack.getTop())
  875. @staticmethod
  876. def _stripFileToId(file):
  877. mime_type = MimeTypeDatabase.getMimeTypeForFile(file)
  878. file = mime_type.stripExtension(file)
  879. return file.replace("Cura/", "")
  880. def _getXmlProfileClass(self):
  881. return self._container_registry.getContainerForMimeType(MimeTypeDatabase.getMimeType("application/x-ultimaker-material-profile"))
  882. ## Get the list of ID's of all containers in a container stack by partially parsing it's serialized data.
  883. @staticmethod
  884. def _getContainerIdListFromSerialized(serialized):
  885. parser = ConfigParser(interpolation = None, empty_lines_in_values = False)
  886. parser.read_string(serialized)
  887. container_ids = []
  888. if "containers" in parser:
  889. for index, container_id in parser.items("containers"):
  890. container_ids.append(container_id)
  891. elif parser.has_option("general", "containers"):
  892. container_string = parser["general"].get("containers", "")
  893. container_list = container_string.split(",")
  894. container_ids = [container_id for container_id in container_list if container_id != ""]
  895. # HACK: there used to be 6 containers numbering from 0 to 5 in a stack,
  896. # now we have 7: index 5 becomes "definition_changes"
  897. if len(container_ids) == 6:
  898. # Hack; We used to not save the definition changes. Fix this.
  899. container_ids.insert(5, "empty")
  900. return container_ids
  901. @staticmethod
  902. def _getMachineNameFromSerializedStack(serialized):
  903. parser = ConfigParser(interpolation = None, empty_lines_in_values = False)
  904. parser.read_string(serialized)
  905. return parser["general"].get("name", "")
  906. @staticmethod
  907. def _getMetaDataDictFromSerializedStack(serialized: str) -> Dict[str, str]:
  908. parser = ConfigParser(interpolation = None, empty_lines_in_values = False)
  909. parser.read_string(serialized)
  910. return dict(parser["metadata"])
  911. @staticmethod
  912. def _getMaterialLabelFromSerialized(serialized):
  913. data = ET.fromstring(serialized)
  914. metadata = data.iterfind("./um:metadata/um:name/um:label", {"um": "http://www.ultimaker.com/material"})
  915. for entry in metadata:
  916. return entry.text