mirror of
https://github.com/Ultimaker/Cura.git
synced 2025-07-07 23:17:32 -06:00
8f075b644d
This means that the parent of the quality node could be one of two types. A bit confusing. Contributes to issue CURA-6600.
69 lines
No EOL
3.7 KiB
Python
69 lines
No EOL
3.7 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.Util import parseBool
|
|
from UM.Settings.ContainerRegistry import ContainerRegistry # To find all the variants for this machine.
|
|
from UM.Settings.Interfaces import ContainerInterface
|
|
from cura.Machines.ContainerNode import ContainerNode
|
|
from cura.Machines.QualityNode import QualityNode
|
|
from cura.Machines.VariantNode import VariantNode
|
|
|
|
if TYPE_CHECKING:
|
|
from typing import Dict
|
|
|
|
## This class represents a machine in the container tree.
|
|
#
|
|
# The subnodes of these nodes are variants.
|
|
class MachineNode(ContainerNode):
|
|
def __init__(self, container_id: str) -> None:
|
|
super().__init__(container_id)
|
|
self.variants = {} # type: Dict[str, VariantNode] # Mapping variant names to their nodes.
|
|
self.global_qualities = {} # type: Dict[str, QualityNode] # Mapping quality types to the global quality for those types.
|
|
container_registry = ContainerRegistry.getInstance()
|
|
|
|
my_metadata = container_registry.findContainersMetadata(id = container_id)[0]
|
|
# Some of the metadata is cached upon construction here.
|
|
# ONLY DO THAT FOR METADATA THAT DOESN'T CHANGE DURING RUNTIME!
|
|
# Otherwise you need to keep it up-to-date during runtime.
|
|
self.has_machine_materials = parseBool(my_metadata.get("has_machine_materials", "false"))
|
|
self.has_machine_quality = parseBool(my_metadata.get("has_machine_quality", "false"))
|
|
self.quality_definition = my_metadata.get("quality_definition", container_id)
|
|
self.exclude_materials = my_metadata.get("exclude_materials", [])
|
|
self.preferred_variant_name = my_metadata.get("preferred_variant_name", "")
|
|
|
|
container_registry.containerAdded.connect(self._variantAdded)
|
|
self._loadAll()
|
|
|
|
## (Re)loads all variants under this printer.
|
|
def _loadAll(self):
|
|
# Find all the variants for this definition ID.
|
|
container_registry = ContainerRegistry.getInstance()
|
|
variants = container_registry.findInstanceContainersMetadata(type = "variant", definition = self.container_id, hardware_type = "nozzle")
|
|
for variant in variants:
|
|
variant_name = variant["name"]
|
|
if variant_name not in self.variants:
|
|
self.variants[variant_name] = VariantNode(variant["id"], machine = self)
|
|
|
|
# Find the global qualities for this printer.
|
|
global_qualities = container_registry.findInstanceContainersMetadata(type = "quality", definition = self.container_id, global_quality = True) # First try specific to this printer.
|
|
if len(global_qualities) == 0: # This printer doesn't override the global qualities.
|
|
global_qualities = container_registry.findInstanceContainersMetadata(type = "quality", definition = "fdmprinter", global_quality = True) # Otherwise pick the global global qualities.
|
|
for global_quality in global_qualities:
|
|
self.global_qualities[global_quality["quality_type"]] = QualityNode(global_quality["id"], parent = self)
|
|
|
|
## When a variant gets added to the set of profiles, we need to update our
|
|
# tree here.
|
|
def _variantAdded(self, container: ContainerInterface):
|
|
if container.getMetaDataEntry("type") != "variant":
|
|
return # Not interested.
|
|
name = container.getMetaDataEntry("name")
|
|
if name in self.variants:
|
|
return # Already have this one.
|
|
if container.getMetaDataEntry("hardware_type") != "nozzle":
|
|
return # Only want nozzles in my tree.
|
|
if container.getMetaDataEntry("definition") != self.container_id:
|
|
return # Not a nozzle that fits in my machine.
|
|
|
|
self.variants[name] = VariantNode(container.getId(), machine = self) |