mirror of
https://github.com/Ultimaker/Cura.git
synced 2025-07-08 23:46:22 -06:00
b1fb843f09
This fallback with the GUID makes this part a lot more complex, but in theory it should work. I hope that we can get some tests to debug this because it's 90% made from the top of my head now. Contributes to issue CURA-6600.
83 lines
No EOL
5.2 KiB
Python
83 lines
No EOL
5.2 KiB
Python
# Copyright (c) 2019 Ultimaker B.V.
|
|
# Cura is released under the terms of the LGPLv3 or higher.
|
|
|
|
from typing import TYPE_CHECKING
|
|
|
|
from UM.Settings.ContainerRegistry import ContainerRegistry
|
|
from UM.Settings.Interfaces import ContainerInterface
|
|
from cura.Machines.ContainerNode import ContainerNode
|
|
from cura.Machines.QualityNode import QualityNode
|
|
|
|
if TYPE_CHECKING:
|
|
from typing import Dict
|
|
from cura.Machines.VariantNode import VariantNode
|
|
|
|
## Represents a material in the container tree.
|
|
#
|
|
# Its subcontainers are quality profiles.
|
|
class MaterialNode(ContainerNode):
|
|
def __init__(self, container_id, variant: "VariantNode") -> None:
|
|
super().__init__(container_id)
|
|
self.variant = variant
|
|
self.qualities = {} # type: Dict[str, QualityNode] # Mapping container IDs to quality profiles.
|
|
container_registry = ContainerRegistry.getInstance()
|
|
my_metadata = container_registry.findContainersMetadata(id = container_id)[0]
|
|
self.base_file = my_metadata["base_file"]
|
|
container_registry.containerAdded.connect(self._qualityAdded)
|
|
self._loadAll()
|
|
|
|
def _loadAll(self) -> None:
|
|
container_registry = ContainerRegistry.getInstance()
|
|
# Find all quality profiles that fit on this material.
|
|
if not self.variant.machine.has_machine_quality: # Need to find the global qualities.
|
|
qualities = container_registry.findInstanceContainersMetadata(type = "quality", definition = "fdmprinter")
|
|
else:
|
|
# Need to find the qualities that specify a material profile with the same material type.
|
|
my_metadata = container_registry.findInstanceContainersMetadata(id = self.container_id)[0]
|
|
my_material_type = my_metadata.get("material")
|
|
qualities = []
|
|
for material_metadata in container_registry.findInstanceContainersMetadata(type = "material", material = my_material_type):
|
|
qualities.extend(container_registry.findInstanceContainersMetadata(type = "quality", definition = self.variant.machine.quality_definition, variant = self.variant.variant_name, material = material_metadata["id"]))
|
|
if not qualities: # No quality profiles found. Go by GUID then.
|
|
my_guid = my_metadata.get("material")
|
|
for material_metadata in container_registry.findInstanceContainersMetadata(type = "material", guid = my_guid):
|
|
qualities.extend(container_registry.findInstanceContainersMetadata(type = "quality", definition = self.variant.machine.quality_definition, variant = self.variant.variant_name, material = material_metadata["id"]))
|
|
|
|
for quality in qualities:
|
|
quality_id = quality["id"]
|
|
if quality_id not in self.qualities:
|
|
self.qualities[quality_id] = QualityNode(quality_id, parent = self)
|
|
|
|
def _qualityAdded(self, container: ContainerInterface) -> None:
|
|
if container.getMetaDataEntry("type") != "quality":
|
|
return # Not interested.
|
|
if not self.variant.machine.has_machine_quality:
|
|
if container.getMetaDataEntry("definition") != "fdmprinter":
|
|
return # Only want global qualities.
|
|
else:
|
|
if container.getMetaDataEntry("definition") != self.variant.machine.quality_definition:
|
|
return # Doesn't match the machine.
|
|
if container.getMetaDataEntry("variant") != self.variant.variant_name:
|
|
return # Doesn't match the variant.
|
|
# Detect if we're falling back to matching via GUID.
|
|
# If so, we might need to erase the current list and put just this one in (i.e. no longer use the fallback).
|
|
container_registry = ContainerRegistry.getInstance()
|
|
my_metadata = container_registry.findInstanceContainersMetadata(id = self.container_id)[0]
|
|
my_material_type = my_metadata.get("material")
|
|
allowed_material_ids = {metadata["id"] for metadata in container_registry.findInstanceContainersMetadata(type = "material", material = my_material_type)}
|
|
is_fallback_guid = len(self.qualities) == 0 or next(iter(self.qualities.values())).getMetaDataEntry("material") not in allowed_material_ids # Select any quality profile; if the material is not matching by material type, we've been falling back to GUID.
|
|
|
|
if is_fallback_guid and container.getMetaDataEntry("material") in allowed_material_ids: # So far we needed the fallback, but no longer!
|
|
self.qualities.clear()
|
|
else:
|
|
if not is_fallback_guid:
|
|
if container.getMetaDataEntry("material") not in allowed_material_ids:
|
|
return # Doesn't match the material type.
|
|
else:
|
|
my_material_guid = my_metadata["guid"]
|
|
allowed_material_ids = {metadata["id"] for metadata in container_registry.findInstanceContainersMetadata(type = "material", guid = my_material_guid)}
|
|
if container.getMetaDataEntry("material") not in allowed_material_ids:
|
|
return # Doesn't match the material GUID.
|
|
|
|
quality_id = container.getId()
|
|
self.qualities[quality_id] = QualityNode(quality_id, parent = self) |