123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891892893894895896897898899900901902903904905906907908909910911912913914915916917918919920921922923924925926927928929930931932933934935936937938939940 |
- # Copyright (c) 2017 Ultimaker B.V.
- # Cura is released under the terms of the LGPLv3 or higher.
- import copy
- import io
- import json #To parse the product-to-id mapping file.
- import os.path #To find the product-to-id mapping.
- import sys
- from typing import Any, Dict, List, Optional
- import xml.etree.ElementTree as ET
- from UM.Resources import Resources
- from UM.Logger import Logger
- from cura.CuraApplication import CuraApplication
- import UM.Dictionary
- from UM.Settings.InstanceContainer import InstanceContainer
- from UM.Settings.ContainerRegistry import ContainerRegistry
- ## Handles serializing and deserializing material containers from an XML file
- class XmlMaterialProfile(InstanceContainer):
- CurrentFdmMaterialVersion = "1.3"
- Version = 1
- def __init__(self, container_id, *args, **kwargs):
- super().__init__(container_id, *args, **kwargs)
- self._inherited_files = []
- ## Translates the version number in the XML files to the setting_version
- # metadata entry.
- #
- # Since the two may increment independently we need a way to say which
- # versions of the XML specification are compatible with our setting data
- # version numbers.
- #
- # \param xml_version: The version number found in an XML file.
- # \return The corresponding setting_version.
- @classmethod
- def xmlVersionToSettingVersion(cls, xml_version: str) -> int:
- if xml_version == "1.3":
- return CuraApplication.SettingVersion
- return 0 #Older than 1.3.
- def getInheritedFiles(self):
- return self._inherited_files
- ## Overridden from InstanceContainer
- # set the meta data for all machine / variant combinations
- def setMetaDataEntry(self, key, value):
- registry = ContainerRegistry.getInstance()
- if registry.isReadOnly(self.getId()):
- return
- super().setMetaDataEntry(key, value)
- basefile = self.getMetaDataEntry("base_file", self.getId()) #if basefile is self.getId, this is a basefile.
- # Update all containers that share basefile
- for container in registry.findInstanceContainers(base_file = basefile):
- if container.getMetaDataEntry(key, None) != value: # Prevent recursion
- container.setMetaDataEntry(key, value)
- ## Overridden from InstanceContainer, similar to setMetaDataEntry.
- # without this function the setName would only set the name of the specific nozzle / material / machine combination container
- # The function is a bit tricky. It will not set the name of all containers if it has the correct name itself.
- def setName(self, new_name):
- registry = ContainerRegistry.getInstance()
- if registry.isReadOnly(self.getId()):
- return
- # Not only is this faster, it also prevents a major loop that causes a stack overflow.
- if self.getName() == new_name:
- return
- super().setName(new_name)
- basefile = self.getMetaDataEntry("base_file", self.getId()) # if basefile is self.getId, this is a basefile.
- # Update the basefile as well, this is actually what we're trying to do
- # Update all containers that share GUID and basefile
- containers = registry.findInstanceContainers(base_file = basefile)
- for container in containers:
- container.setName(new_name)
- ## Overridden from InstanceContainer, to set dirty to base file as well.
- def setDirty(self, dirty):
- super().setDirty(dirty)
- base_file = self.getMetaDataEntry("base_file", None)
- registry = ContainerRegistry.getInstance()
- if base_file is not None and base_file != self.getId() and not registry.isReadOnly(base_file):
- containers = registry.findContainers(id = base_file)
- if containers:
- containers[0].setDirty(dirty)
- ## Overridden from InstanceContainer
- # base file: common settings + supported machines
- # machine / variant combination: only changes for itself.
- def serialize(self, ignored_metadata_keys: Optional[set] = None):
- registry = ContainerRegistry.getInstance()
- base_file = self.getMetaDataEntry("base_file", "")
- if base_file and self.getId() != base_file:
- # Since we create an instance of XmlMaterialProfile for each machine and nozzle in the profile,
- # we should only serialize the "base" material definition, since that can then take care of
- # serializing the machine/nozzle specific profiles.
- raise NotImplementedError("Ignoring serializing non-root XML materials, the data is contained in the base material")
- builder = ET.TreeBuilder()
- root = builder.start("fdmmaterial",
- {"xmlns": "http://www.ultimaker.com/material",
- "version": self.CurrentFdmMaterialVersion})
- ## Begin Metadata Block
- builder.start("metadata")
- metadata = copy.deepcopy(self.getMetaData())
- # setting_version is derived from the "version" tag in the schema, so don't serialize it into a file
- if ignored_metadata_keys is None:
- ignored_metadata_keys = set()
- ignored_metadata_keys |= {"setting_version"}
- # remove the keys that we want to ignore in the metadata
- for key in ignored_metadata_keys:
- if key in metadata:
- del metadata[key]
- properties = metadata.pop("properties", {})
- # Metadata properties that should not be serialized.
- metadata.pop("status", "")
- metadata.pop("variant", "")
- metadata.pop("type", "")
- metadata.pop("base_file", "")
- metadata.pop("approximate_diameter", "")
- metadata.pop("id", "")
- metadata.pop("container_type", "")
- metadata.pop("name", "")
- ## Begin Name Block
- builder.start("name")
- builder.start("brand")
- builder.data(metadata.pop("brand", ""))
- builder.end("brand")
- builder.start("material")
- builder.data(metadata.pop("material", ""))
- builder.end("material")
- builder.start("color")
- builder.data(metadata.pop("color_name", ""))
- builder.end("color")
- builder.start("label")
- builder.data(self.getName())
- builder.end("label")
- builder.end("name")
- ## End Name Block
- for key, value in metadata.items():
- builder.start(key)
- if value is not None: #Nones get handled well by the builder.
- #Otherwise the builder always expects a string.
- #Deserialize expects the stringified version.
- value = str(value)
- builder.data(value)
- builder.end(key)
- builder.end("metadata")
- ## End Metadata Block
- ## Begin Properties Block
- builder.start("properties")
- for key, value in properties.items():
- builder.start(key)
- builder.data(value)
- builder.end(key)
- builder.end("properties")
- ## End Properties Block
- ## Begin Settings Block
- builder.start("settings")
- if self.getDefinition().getId() == "fdmprinter":
- for instance in self.findInstances():
- self._addSettingElement(builder, instance)
- machine_container_map = {}
- machine_nozzle_map = {}
- all_containers = registry.findInstanceContainers(GUID = self.getMetaDataEntry("GUID"), base_file = self.getId())
- for container in all_containers:
- definition_id = container.getDefinition().getId()
- if definition_id == "fdmprinter":
- continue
- if definition_id not in machine_container_map:
- machine_container_map[definition_id] = container
- if definition_id not in machine_nozzle_map:
- machine_nozzle_map[definition_id] = {}
- variant = container.getMetaDataEntry("variant")
- if variant:
- machine_nozzle_map[definition_id][variant] = container
- continue
- machine_container_map[definition_id] = container
- # Map machine human-readable names to IDs
- product_id_map = self.getProductIdMap()
- for definition_id, container in machine_container_map.items():
- definition = container.getDefinition()
- product = definition_id
- for product_name, product_id_list in product_id_map.items():
- if definition_id in product_id_list:
- product = product_name
- break
- builder.start("machine")
- builder.start("machine_identifier", {
- "manufacturer": container.getMetaDataEntry("machine_manufacturer", definition.getMetaDataEntry("manufacturer", "Unknown")),
- "product": product
- })
- builder.end("machine_identifier")
- for instance in container.findInstances():
- if self.getDefinition().getId() == "fdmprinter" and self.getInstance(instance.definition.key) and self.getProperty(instance.definition.key, "value") == instance.value:
- # If the settings match that of the base profile, just skip since we inherit the base profile.
- continue
- self._addSettingElement(builder, instance)
- # Find all hotend sub-profiles corresponding to this material and machine and add them to this profile.
- for hotend_id, hotend in machine_nozzle_map[definition_id].items():
- variant_containers = registry.findInstanceContainersMetadata(id = hotend.getMetaDataEntry("variant"))
- if not variant_containers:
- continue
- # The hotend identifier is not the containers name, but its "name".
- builder.start("hotend", {"id": variant_containers[0]["name"]})
- # Compatible is a special case, as it's added as a meta data entry (instead of an instance).
- compatible = hotend.getMetaDataEntry("compatible")
- if compatible is not None:
- builder.start("setting", {"key": "hardware compatible"})
- if compatible:
- builder.data("yes")
- else:
- builder.data("no")
- builder.end("setting")
- for instance in hotend.findInstances():
- if container.getInstance(instance.definition.key) and container.getProperty(instance.definition.key, "value") == instance.value:
- # If the settings match that of the machine profile, just skip since we inherit the machine profile.
- continue
- self._addSettingElement(builder, instance)
- builder.end("hotend")
- builder.end("machine")
- builder.end("settings")
- ## End Settings Block
- builder.end("fdmmaterial")
- root = builder.close()
- _indent(root)
- stream = io.BytesIO()
- tree = ET.ElementTree(root)
- # this makes sure that the XML header states encoding="utf-8"
- tree.write(stream, encoding="utf-8", xml_declaration=True)
- return stream.getvalue().decode('utf-8')
- # Recursively resolve loading inherited files
- def _resolveInheritance(self, file_name):
- xml = self._loadFile(file_name)
- inherits = xml.find("./um:inherits", self.__namespaces)
- if inherits is not None:
- inherited = self._resolveInheritance(inherits.text)
- xml = self._mergeXML(inherited, xml)
- return xml
- def _loadFile(self, file_name):
- path = Resources.getPath(CuraApplication.getInstance().ResourceTypes.MaterialInstanceContainer, file_name + ".xml.fdm_material")
- with open(path, encoding="utf-8") as f:
- contents = f.read()
- self._inherited_files.append(path)
- return ET.fromstring(contents)
- # The XML material profile can have specific settings for machines.
- # Some machines share profiles, so they are only created once.
- # This function duplicates those elements so that each machine tag only has one identifier.
- def _expandMachinesXML(self, element):
- settings_element = element.find("./um:settings", self.__namespaces)
- machines = settings_element.iterfind("./um:machine", self.__namespaces)
- machines_to_add = []
- machines_to_remove = []
- for machine in machines:
- identifiers = list(machine.iterfind("./um:machine_identifier", self.__namespaces))
- has_multiple_identifiers = len(identifiers) > 1
- if has_multiple_identifiers:
- # Multiple identifiers found. We need to create a new machine element and copy all it's settings there.
- for identifier in identifiers:
- new_machine = copy.deepcopy(machine)
- # Create list of identifiers that need to be removed from the copied element.
- other_identifiers = [self._createKey(other_identifier) for other_identifier in identifiers if other_identifier is not identifier]
- # As we can only remove by exact object reference, we need to look through the identifiers of copied machine.
- new_machine_identifiers = list(new_machine.iterfind("./um:machine_identifier", self.__namespaces))
- for new_machine_identifier in new_machine_identifiers:
- key = self._createKey(new_machine_identifier)
- # Key was in identifiers to remove, so this element needs to be purged
- if key in other_identifiers:
- new_machine.remove(new_machine_identifier)
- machines_to_add.append(new_machine)
- machines_to_remove.append(machine)
- else:
- pass # Machine only has one identifier. Nothing to do.
- # Remove & add all required machines.
- for machine_to_remove in machines_to_remove:
- settings_element.remove(machine_to_remove)
- for machine_to_add in machines_to_add:
- settings_element.append(machine_to_add)
- return element
- def _mergeXML(self, first, second):
- result = copy.deepcopy(first)
- self._combineElement(self._expandMachinesXML(result), self._expandMachinesXML(second))
- return result
- def _createKey(self, element):
- key = element.tag.split("}")[-1]
- if "key" in element.attrib:
- key += " key:" + element.attrib["key"]
- if "manufacturer" in element.attrib:
- key += " manufacturer:" + element.attrib["manufacturer"]
- if "product" in element.attrib:
- key += " product:" + element.attrib["product"]
- if key == "machine":
- for item in element:
- if "machine_identifier" in item.tag:
- key += " " + item.attrib["product"]
- return key
- # Recursively merges XML elements. Updates either the text or children if another element is found in first.
- # If it does not exist, copies it from second.
- def _combineElement(self, first, second):
- # Create a mapping from tag name to element.
- mapping = {}
- for element in first:
- key = self._createKey(element)
- mapping[key] = element
- for element in second:
- key = self._createKey(element)
- if len(element): # Check if element has children.
- try:
- if "setting" in element.tag and not "settings" in element.tag:
- # Setting can have points in it. In that case, delete all values and override them.
- for child in list(mapping[key]):
- mapping[key].remove(child)
- for child in element:
- mapping[key].append(child)
- else:
- self._combineElement(mapping[key], element) # Multiple elements, handle those.
- except KeyError:
- mapping[key] = element
- first.append(element)
- else:
- try:
- mapping[key].text = element.text
- except KeyError: # Not in the mapping, so simply add it
- mapping[key] = element
- first.append(element)
- def clearData(self):
- self._metadata = {
- "id": self.getId(),
- "name": ""
- }
- self._definition = None
- self._instances = {}
- self._read_only = False
- self._dirty = False
- self._path = ""
- @classmethod
- def getConfigurationTypeFromSerialized(cls, serialized: str) -> Optional[str]:
- return "materials"
- @classmethod
- def getVersionFromSerialized(cls, serialized: str) -> Optional[int]:
- data = ET.fromstring(serialized)
- version = XmlMaterialProfile.Version
- # get setting version
- if "version" in data.attrib:
- setting_version = cls.xmlVersionToSettingVersion(data.attrib["version"])
- else:
- setting_version = cls.xmlVersionToSettingVersion("1.2")
- return version * 1000000 + setting_version
- ## Overridden from InstanceContainer
- def deserialize(self, serialized, file_name = None):
- containers_to_add = []
- # update the serialized data first
- from UM.Settings.Interfaces import ContainerInterface
- serialized = ContainerInterface.deserialize(self, serialized, file_name)
- try:
- data = ET.fromstring(serialized)
- except:
- Logger.logException("e", "An exception occurred while parsing the material profile")
- return
- # Reset previous metadata
- old_id = self.getId()
- self.clearData() # Ensure any previous data is gone.
- meta_data = {}
- meta_data["type"] = "material"
- meta_data["base_file"] = self.getId()
- meta_data["status"] = "unknown" # TODO: Add material verification
- meta_data["id"] = old_id
- meta_data["container_type"] = XmlMaterialProfile
- common_setting_values = {}
- inherits = data.find("./um:inherits", self.__namespaces)
- if inherits is not None:
- inherited = self._resolveInheritance(inherits.text)
- data = self._mergeXML(inherited, data)
- # set setting_version in metadata
- if "version" in data.attrib:
- meta_data["setting_version"] = self.xmlVersionToSettingVersion(data.attrib["version"])
- else:
- meta_data["setting_version"] = self.xmlVersionToSettingVersion("1.2") #1.2 and lower didn't have that version number there yet.
- meta_data["name"] = "Unknown Material" #In case the name tag is missing.
- for entry in data.iterfind("./um:metadata/*", self.__namespaces):
- tag_name = _tag_without_namespace(entry)
- if tag_name == "name":
- brand = entry.find("./um:brand", self.__namespaces)
- material = entry.find("./um:material", self.__namespaces)
- color = entry.find("./um:color", self.__namespaces)
- label = entry.find("./um:label", self.__namespaces)
- if label is not None:
- meta_data["name"] = label.text
- else:
- meta_data["name"] = self._profile_name(material.text, color.text)
- meta_data["brand"] = brand.text
- meta_data["material"] = material.text
- meta_data["color_name"] = color.text
- continue
- # setting_version is derived from the "version" tag in the schema earlier, so don't set it here
- if tag_name == "setting_version":
- continue
- meta_data[tag_name] = entry.text
- if tag_name in self.__material_metadata_setting_map:
- common_setting_values[self.__material_metadata_setting_map[tag_name]] = entry.text
- if "description" not in meta_data:
- meta_data["description"] = ""
- if "adhesion_info" not in meta_data:
- meta_data["adhesion_info"] = ""
- property_values = {}
- properties = data.iterfind("./um:properties/*", self.__namespaces)
- for entry in properties:
- tag_name = _tag_without_namespace(entry)
- property_values[tag_name] = entry.text
- if tag_name in self.__material_properties_setting_map:
- common_setting_values[self.__material_properties_setting_map[tag_name]] = entry.text
- meta_data["approximate_diameter"] = str(round(float(property_values.get("diameter", 2.85)))) # In mm
- meta_data["properties"] = property_values
- meta_data["definition"] = "fdmprinter"
- common_compatibility = True
- settings = data.iterfind("./um:settings/um:setting", self.__namespaces)
- for entry in settings:
- key = entry.get("key")
- if key in self.__material_settings_setting_map:
- common_setting_values[self.__material_settings_setting_map[key]] = entry.text
- elif key in self.__unmapped_settings:
- if key == "hardware compatible":
- common_compatibility = self._parseCompatibleValue(entry.text)
- self._cached_values = common_setting_values # from InstanceContainer ancestor
- meta_data["compatible"] = common_compatibility
- self.setMetaData(meta_data)
- self._dirty = False
- # Map machine human-readable names to IDs
- product_id_map = self.getProductIdMap()
- machines = data.iterfind("./um:settings/um:machine", self.__namespaces)
- for machine in machines:
- machine_compatibility = common_compatibility
- machine_setting_values = {}
- settings = machine.iterfind("./um:setting", self.__namespaces)
- for entry in settings:
- key = entry.get("key")
- if key in self.__material_settings_setting_map:
- machine_setting_values[self.__material_settings_setting_map[key]] = entry.text
- elif key in self.__unmapped_settings:
- if key == "hardware compatible":
- machine_compatibility = self._parseCompatibleValue(entry.text)
- else:
- Logger.log("d", "Unsupported material setting %s", key)
- cached_machine_setting_properties = common_setting_values.copy()
- cached_machine_setting_properties.update(machine_setting_values)
- identifiers = machine.iterfind("./um:machine_identifier", self.__namespaces)
- for identifier in identifiers:
- machine_id_list = product_id_map.get(identifier.get("product"), [])
- if not machine_id_list:
- machine_id_list = self.getPossibleDefinitionIDsFromName(identifier.get("product"))
- for machine_id in machine_id_list:
- definitions = ContainerRegistry.getInstance().findDefinitionContainersMetadata(id = machine_id)
- if not definitions:
- Logger.log("w", "No definition found for machine ID %s", machine_id)
- continue
- Logger.log("d", "Found definition for machine ID %s", machine_id)
- definition = definitions[0]
- machine_manufacturer = identifier.get("manufacturer", definition.get("manufacturer", "Unknown")) #If the XML material doesn't specify a manufacturer, use the one in the actual printer definition.
- if machine_compatibility:
- new_material_id = self.getId() + "_" + machine_id
- # The child or derived material container may already exist. This can happen when a material in a
- # project file and the a material in Cura have the same ID.
- # In the case if a derived material already exists, override that material container because if
- # the data in the parent material has been changed, the derived ones should be updated too.
- if ContainerRegistry.getInstance().isLoaded(new_material_id):
- new_material = ContainerRegistry.getInstance().findContainers(id = new_material_id)[0]
- is_new_material = False
- else:
- new_material = XmlMaterialProfile(new_material_id)
- is_new_material = True
- new_material.setMetaData(copy.deepcopy(self.getMetaData()))
- new_material.getMetaData()["id"] = new_material_id
- new_material.getMetaData()["name"] = self.getName()
- new_material.setDefinition(machine_id)
- # Don't use setMetadata, as that overrides it for all materials with same base file
- new_material.getMetaData()["compatible"] = machine_compatibility
- new_material.getMetaData()["machine_manufacturer"] = machine_manufacturer
- new_material.getMetaData()["definition"] = machine_id
- new_material.setCachedValues(cached_machine_setting_properties)
- new_material._dirty = False
- if is_new_material:
- containers_to_add.append(new_material)
- hotends = machine.iterfind("./um:hotend", self.__namespaces)
- for hotend in hotends:
- hotend_id = hotend.get("id")
- if hotend_id is None:
- continue
- variant_containers = ContainerRegistry.getInstance().findInstanceContainersMetadata(id = hotend_id)
- if not variant_containers:
- # It is not really properly defined what "ID" is so also search for variants by name.
- variant_containers = ContainerRegistry.getInstance().findInstanceContainersMetadata(definition = machine_id, name = hotend_id)
- if not variant_containers:
- continue
- hotend_compatibility = machine_compatibility
- hotend_setting_values = {}
- settings = hotend.iterfind("./um:setting", self.__namespaces)
- for entry in settings:
- key = entry.get("key")
- if key in self.__material_settings_setting_map:
- hotend_setting_values[self.__material_settings_setting_map[key]] = entry.text
- elif key in self.__unmapped_settings:
- if key == "hardware compatible":
- hotend_compatibility = self._parseCompatibleValue(entry.text)
- else:
- Logger.log("d", "Unsupported material setting %s", key)
- new_hotend_id = self.getId() + "_" + machine_id + "_" + hotend_id.replace(" ", "_")
- # Same as machine compatibility, keep the derived material containers consistent with the parent
- # material
- if ContainerRegistry.getInstance().isLoaded(new_hotend_id):
- new_hotend_material = ContainerRegistry.getInstance().findContainers(id = new_hotend_id)[0]
- is_new_material = False
- else:
- new_hotend_material = XmlMaterialProfile(new_hotend_id)
- is_new_material = True
- new_hotend_material.setMetaData(copy.deepcopy(self.getMetaData()))
- new_hotend_material.getMetaData()["id"] = new_hotend_id
- new_hotend_material.getMetaData()["name"] = self.getName()
- new_hotend_material.getMetaData()["variant"] = variant_containers[0]["id"]
- new_hotend_material.setDefinition(machine_id)
- # Don't use setMetadata, as that overrides it for all materials with same base file
- new_hotend_material.getMetaData()["compatible"] = hotend_compatibility
- new_hotend_material.getMetaData()["machine_manufacturer"] = machine_manufacturer
- new_hotend_material.getMetaData()["definition"] = machine_id
- cached_hotend_setting_properties = cached_machine_setting_properties.copy()
- cached_hotend_setting_properties.update(hotend_setting_values)
- new_hotend_material.setCachedValues(cached_hotend_setting_properties)
- new_hotend_material._dirty = False
- if is_new_material:
- containers_to_add.append(new_hotend_material)
- # there is only one ID for a machine. Once we have reached here, it means we have already found
- # a workable ID for that machine, so there is no need to continue
- break
- for container_to_add in containers_to_add:
- ContainerRegistry.getInstance().addContainer(container_to_add)
- @classmethod
- def deserializeMetadata(cls, serialized: str, container_id: str) -> List[Dict[str, Any]]:
- result_metadata = [] #All the metadata that we found except the base (because the base is returned).
- #Update the serialized data to the latest version.
- serialized = cls._updateSerialized(serialized)
- base_metadata = {
- "type": "material",
- "status": "unknown", #TODO: Add material verification.
- "container_type": XmlMaterialProfile,
- "id": container_id,
- "base_file": container_id
- }
- try:
- data = ET.fromstring(serialized)
- except:
- Logger.logException("e", "An exception occurred while parsing the material profile")
- return []
- #TODO: Implement the <inherits> tag. It's unused at the moment though.
- if "version" in data.attrib:
- base_metadata["setting_version"] = cls.xmlVersionToSettingVersion(data.attrib["version"])
- else:
- base_metadata["setting_version"] = cls.xmlVersionToSettingVersion("1.2") #1.2 and lower didn't have that version number there yet.
- for entry in data.iterfind("./um:metadata/*", cls.__namespaces):
- tag_name = _tag_without_namespace(entry)
- if tag_name == "name":
- brand = entry.find("./um:brand", cls.__namespaces)
- material = entry.find("./um:material", cls.__namespaces)
- color = entry.find("./um:color", cls.__namespaces)
- label = entry.find("./um:label", cls.__namespaces)
- if label is not None:
- base_metadata["name"] = label.text
- else:
- base_metadata["name"] = cls._profile_name(material.text, color.text)
- base_metadata["brand"] = brand.text
- base_metadata["material"] = material.text
- base_metadata["color_name"] = color.text
- continue
- #Setting_version is derived from the "version" tag in the schema earlier, so don't set it here.
- if tag_name == "setting_version":
- continue
- base_metadata[tag_name] = entry.text
- if "description" not in base_metadata:
- base_metadata["description"] = ""
- if "adhesion_info" not in base_metadata:
- base_metadata["adhesion_info"] = ""
- property_values = {}
- properties = data.iterfind("./um:properties/*", cls.__namespaces)
- for entry in properties:
- tag_name = _tag_without_namespace(entry)
- property_values[tag_name] = entry.text
- base_metadata["approximate_diameter"] = str(round(float(property_values.get("diameter", 2.85)))) # In mm
- base_metadata["properties"] = property_values
- base_metadata["definition"] = "fdmprinter"
- compatible_entries = data.iterfind("./um:settings/um:setting[@key='hardware compatible']", cls.__namespaces)
- try:
- common_compatibility = cls._parseCompatibleValue(next(compatible_entries).text)
- except StopIteration: #No 'hardware compatible' setting.
- common_compatibility = True
- base_metadata["compatible"] = common_compatibility
- result_metadata.append(base_metadata)
- # Map machine human-readable names to IDs
- product_id_map = cls.getProductIdMap()
- for machine in data.iterfind("./um:settings/um:machine", cls.__namespaces):
- machine_compatibility = common_compatibility
- for entry in machine.iterfind("./um:setting", cls.__namespaces):
- key = entry.get("key")
- if key == "hardware compatible":
- machine_compatibility = cls._parseCompatibleValue(entry.text)
- for identifier in machine.iterfind("./um:machine_identifier", cls.__namespaces):
- machine_id_list = product_id_map.get(identifier.get("product"), [])
- if not machine_id_list:
- machine_id_list = cls.getPossibleDefinitionIDsFromName(identifier.get("product"))
- for machine_id in machine_id_list:
- definition_metadata = ContainerRegistry.getInstance().findDefinitionContainersMetadata(id = machine_id)
- if not definition_metadata:
- Logger.log("w", "No definition found for machine ID %s", machine_id)
- continue
- definition_metadata = definition_metadata[0]
- machine_manufacturer = identifier.get("manufacturer", definition_metadata.get("manufacturer", "Unknown")) #If the XML material doesn't specify a manufacturer, use the one in the actual printer definition.
- if machine_compatibility:
- new_material_id = container_id + "_" + machine_id
- # The child or derived material container may already exist. This can happen when a material in a
- # project file and the a material in Cura have the same ID.
- # In the case if a derived material already exists, override that material container because if
- # the data in the parent material has been changed, the derived ones should be updated too.
- found_materials = ContainerRegistry.getInstance().findInstanceContainersMetadata(id = new_material_id)
- if found_materials:
- new_material_metadata = found_materials[0]
- else:
- new_material_metadata = {}
- new_material_metadata.update(base_metadata)
- new_material_metadata["id"] = new_material_id
- new_material_metadata["compatible"] = machine_compatibility
- new_material_metadata["machine_manufacturer"] = machine_manufacturer
- new_material_metadata["definition"] = machine_id
- if len(found_materials) == 0: #This is a new material.
- result_metadata.append(new_material_metadata)
- for hotend in machine.iterfind("./um:hotend", cls.__namespaces):
- hotend_id = hotend.get("id")
- if hotend_id is None:
- continue
- variant_containers = ContainerRegistry.getInstance().findInstanceContainersMetadata(id = hotend_id)
- if not variant_containers:
- # It is not really properly defined what "ID" is so also search for variants by name.
- variant_containers = ContainerRegistry.getInstance().findInstanceContainersMetadata(definition = machine_id, name = hotend_id)
- hotend_compatibility = machine_compatibility
- for entry in hotend.iterfind("./um:setting", cls.__namespaces):
- key = entry.get("key")
- if key == "hardware compatible":
- hotend_compatibility = cls._parseCompatibleValue(entry.text)
- new_hotend_id = container_id + "_" + machine_id + "_" + hotend_id.replace(" ", "_")
- # Same as machine compatibility, keep the derived material containers consistent with the parent
- # material
- found_materials = ContainerRegistry.getInstance().findInstanceContainersMetadata(id = new_hotend_id)
- if found_materials:
- new_hotend_material_metadata = found_materials[0]
- else:
- new_hotend_material_metadata = {}
- new_hotend_material_metadata.update(base_metadata)
- if variant_containers:
- new_hotend_material_metadata["variant"] = variant_containers[0]["id"]
- else:
- new_hotend_material_metadata["variant"] = hotend_id
- _with_missing_variants.append(new_hotend_material_metadata)
- new_hotend_material_metadata["compatible"] = hotend_compatibility
- new_hotend_material_metadata["machine_manufacturer"] = machine_manufacturer
- new_hotend_material_metadata["id"] = new_hotend_id
- new_hotend_material_metadata["definition"] = machine_id
- if len(found_materials) == 0:
- result_metadata.append(new_hotend_material_metadata)
- # there is only one ID for a machine. Once we have reached here, it means we have already found
- # a workable ID for that machine, so there is no need to continue
- break
- return result_metadata
- def _addSettingElement(self, builder, instance):
- try:
- key = UM.Dictionary.findKey(self.__material_settings_setting_map, instance.definition.key)
- except ValueError:
- return
- builder.start("setting", { "key": key })
- builder.data(str(instance.value))
- builder.end("setting")
- @classmethod
- def _profile_name(cls, material_name, color_name):
- if color_name != "Generic":
- return "%s %s" % (color_name, material_name)
- else:
- return material_name
- @classmethod
- def getPossibleDefinitionIDsFromName(cls, name):
- name_parts = name.lower().split(" ")
- merged_name_parts = []
- for part in name_parts:
- if len(part) == 0:
- continue
- if len(merged_name_parts) == 0:
- merged_name_parts.append(part)
- continue
- if part.isdigit():
- # for names with digit(s) such as Ultimaker 3 Extended, we generate an ID like
- # "ultimaker3_extended", ignoring the space between "Ultimaker" and "3".
- merged_name_parts[-1] = merged_name_parts[-1] + part
- else:
- merged_name_parts.append(part)
- id_list = [name.lower().replace(" ", ""), # simply removing all spaces
- name.lower().replace(" ", "_"), # simply replacing all spaces with underscores
- "_".join(merged_name_parts),
- ]
- return id_list
- ## Gets a mapping from product names in the XML files to their definition
- # IDs.
- #
- # This loads the mapping from a file.
- @classmethod
- def getProductIdMap(cls) -> Dict[str, List[str]]:
- product_to_id_file = os.path.join(os.path.dirname(sys.modules[cls.__module__].__file__), "product_to_id.json")
- with open(product_to_id_file) as f:
- product_to_id_map = json.load(f)
- product_to_id_map = {key: [value] for key, value in product_to_id_map.items()}
- return product_to_id_map
- ## Parse the value of the "material compatible" property.
- @classmethod
- def _parseCompatibleValue(cls, value: str):
- return value in {"yes", "unknown"}
- ## Small string representation for debugging.
- def __str__(self):
- return "<XmlMaterialProfile '{my_id}' ('{name}') from base file '{base_file}'>".format(my_id = self.getId(), name = self.getName(), base_file = self.getMetaDataEntry("base_file"))
- # Map XML file setting names to internal names
- __material_settings_setting_map = {
- "print temperature": "default_material_print_temperature",
- "heated bed temperature": "material_bed_temperature",
- "standby temperature": "material_standby_temperature",
- "processing temperature graph": "material_flow_temp_graph",
- "print cooling": "cool_fan_speed",
- "retraction amount": "retraction_amount",
- "retraction speed": "retraction_speed",
- "adhesion tendency": "material_adhesion_tendency",
- "surface energy": "material_surface_energy"
- }
- __unmapped_settings = [
- "hardware compatible"
- ]
- __material_properties_setting_map = {
- "diameter": "material_diameter"
- }
- __material_metadata_setting_map = {
- "GUID": "material_guid"
- }
- # Map of recognised namespaces with a proper prefix.
- __namespaces = {
- "um": "http://www.ultimaker.com/material"
- }
- ## Helper function for pretty-printing XML because ETree is stupid
- def _indent(elem, level = 0):
- i = "\n" + level * " "
- if len(elem):
- if not elem.text or not elem.text.strip():
- elem.text = i + " "
- if not elem.tail or not elem.tail.strip():
- elem.tail = i
- for elem in elem:
- _indent(elem, level + 1)
- if not elem.tail or not elem.tail.strip():
- elem.tail = i
- else:
- if level and (not elem.tail or not elem.tail.strip()):
- elem.tail = i
- # The namespace is prepended to the tag name but between {}.
- # We are only interested in the actual tag name, so discard everything
- # before the last }
- def _tag_without_namespace(element):
- return element.tag[element.tag.rfind("}") + 1:]
- #While loading XML profiles, some of these profiles don't know what variant
- #they belong to. We'd like to search by the machine ID and the variant's
- #name, but we don't know the variant's ID. Not all variants have been loaded
- #yet so we can't run a filter on the name and machine. The ID is unknown
- #so we can't lazily load the variant either. So we have to wait until all
- #the rest is loaded properly and then assign the correct variant to the
- #material files that were missing it.
- _with_missing_variants = []
- def _fillMissingVariants():
- registry = ContainerRegistry.getInstance()
- for variant_metadata in _with_missing_variants:
- variants = registry.findContainersMetadata(definition = variant_metadata["definition"], name = variant_metadata["variant"])
- if not variants:
- Logger.log("w", "Could not find variant for variant-specific material {material_id}.".format(material_id = variant_metadata["id"]))
- continue
- variant_metadata["variant"] = variants[0]["id"]
- ContainerRegistry.allMetadataLoaded.connect(_fillMissingVariants)
|