mirror of
https://git.mirrors.martin98.com/https://github.com/Ultimaker/Cura
synced 2025-04-29 07:14:24 +08:00
dd3b49e80d
The filter on type=variant is right above it in the same function, 20 lines of code earlier. I think this check is superfluous then. Contributes to issue CURA-5045.
143 lines
7.0 KiB
Python
143 lines
7.0 KiB
Python
# Copyright (c) 2018 Ultimaker B.V.
|
|
# Cura is released under the terms of the LGPLv3 or higher.
|
|
|
|
from enum import Enum
|
|
from collections import OrderedDict
|
|
from typing import Optional, TYPE_CHECKING
|
|
|
|
from UM.ConfigurationErrorMessage import ConfigurationErrorMessage
|
|
from UM.Logger import Logger
|
|
from UM.Settings.ContainerRegistry import ContainerRegistry
|
|
from UM.Util import parseBool
|
|
|
|
from cura.Machines.ContainerNode import ContainerNode
|
|
from cura.Settings.GlobalStack import GlobalStack
|
|
|
|
if TYPE_CHECKING:
|
|
from UM.Settings.DefinitionContainer import DefinitionContainer
|
|
|
|
|
|
class VariantType(Enum):
|
|
BUILD_PLATE = "buildplate"
|
|
NOZZLE = "nozzle"
|
|
|
|
|
|
ALL_VARIANT_TYPES = (VariantType.BUILD_PLATE, VariantType.NOZZLE)
|
|
|
|
|
|
#
|
|
# VariantManager is THE place to look for a specific variant. It maintains two variant lookup tables with the following
|
|
# structure:
|
|
#
|
|
# [machine_definition_id] -> [variant_type] -> [variant_name] -> ContainerNode(metadata / container)
|
|
# Example: "ultimaker3" -> "buildplate" -> "Glass" (if present) -> ContainerNode
|
|
# -> ...
|
|
# -> "nozzle" -> "AA 0.4"
|
|
# -> "BB 0.8"
|
|
# -> ...
|
|
#
|
|
# [machine_definition_id] -> [machine_buildplate_type] -> ContainerNode(metadata / container)
|
|
# Example: "ultimaker3" -> "glass" (this is different from the variant name) -> ContainerNode
|
|
#
|
|
# Note that the "container" field is not loaded in the beginning because it would defeat the purpose of lazy-loading.
|
|
# A container is loaded when getVariant() is called to load a variant InstanceContainer.
|
|
#
|
|
class VariantManager:
|
|
|
|
def __init__(self, container_registry):
|
|
self._container_registry = container_registry # type: ContainerRegistry
|
|
|
|
self._machine_to_variant_dict_map = dict() # <machine_type> -> <variant_dict>
|
|
self._machine_to_buildplate_dict_map = dict()
|
|
|
|
self._exclude_variant_id_list = ["empty_variant"]
|
|
|
|
#
|
|
# Initializes the VariantManager including:
|
|
# - initializing the variant lookup table based on the metadata in ContainerRegistry.
|
|
#
|
|
def initialize(self):
|
|
self._machine_to_variant_dict_map = OrderedDict()
|
|
self._machine_to_buildplate_dict_map = OrderedDict()
|
|
|
|
# Cache all variants from the container registry to a variant map for better searching and organization.
|
|
variant_metadata_list = self._container_registry.findContainersMetadata(type = "variant")
|
|
for variant_metadata in variant_metadata_list:
|
|
if variant_metadata["id"] in self._exclude_variant_id_list:
|
|
Logger.log("d", "Exclude variant [%s]", variant_metadata["id"])
|
|
continue
|
|
|
|
variant_name = variant_metadata["name"]
|
|
variant_definition = variant_metadata["definition"]
|
|
if variant_definition not in self._machine_to_variant_dict_map:
|
|
self._machine_to_variant_dict_map[variant_definition] = OrderedDict()
|
|
for variant_type in ALL_VARIANT_TYPES:
|
|
self._machine_to_variant_dict_map[variant_definition][variant_type] = dict()
|
|
|
|
variant_type = variant_metadata["hardware_type"]
|
|
variant_type = VariantType(variant_type)
|
|
variant_dict = self._machine_to_variant_dict_map[variant_definition][variant_type]
|
|
if variant_name in variant_dict:
|
|
# ERROR: duplicated variant name.
|
|
ConfigurationErrorMessage.getInstance().addFaultyContainers(variant_metadata["id"])
|
|
continue #Then ignore this variant. This now chooses one of the two variants arbitrarily and deletes the other one! No guarantees!
|
|
|
|
variant_dict[variant_name] = ContainerNode(metadata = variant_metadata)
|
|
|
|
# If the variant is a buildplate then fill also the buildplate map
|
|
if variant_type == VariantType.BUILD_PLATE:
|
|
if variant_definition not in self._machine_to_buildplate_dict_map:
|
|
self._machine_to_buildplate_dict_map[variant_definition] = OrderedDict()
|
|
|
|
variant_container = self._container_registry.findContainers(type = "variant", id = variant_metadata["id"])[0]
|
|
buildplate_type = variant_container.getProperty("machine_buildplate_type", "value")
|
|
if buildplate_type not in self._machine_to_buildplate_dict_map[variant_definition]:
|
|
self._machine_to_variant_dict_map[variant_definition][buildplate_type] = dict()
|
|
|
|
self._machine_to_buildplate_dict_map[variant_definition][buildplate_type] = variant_dict[variant_name]
|
|
|
|
#
|
|
# Gets the variant InstanceContainer with the given information.
|
|
# Almost the same as getVariantMetadata() except that this returns an InstanceContainer if present.
|
|
#
|
|
def getVariantNode(self, machine_definition_id: str, variant_name: str,
|
|
variant_type: Optional["VariantType"] = None) -> Optional["ContainerNode"]:
|
|
if variant_type is None:
|
|
variant_node = None
|
|
variant_type_dict = self._machine_to_variant_dict_map[machine_definition_id]
|
|
for variant_dict in variant_type_dict.values():
|
|
if variant_name in variant_dict:
|
|
variant_node = variant_dict[variant_name]
|
|
break
|
|
return variant_node
|
|
return self._machine_to_variant_dict_map[machine_definition_id].get(variant_type, {}).get(variant_name)
|
|
|
|
def getVariantNodes(self, machine: "GlobalStack",
|
|
variant_type: Optional["VariantType"] = None) -> dict:
|
|
machine_definition_id = machine.definition.getId()
|
|
return self._machine_to_variant_dict_map.get(machine_definition_id, {}).get(variant_type, {})
|
|
|
|
#
|
|
# Gets the default variant for the given machine definition.
|
|
#
|
|
def getDefaultVariantNode(self, machine_definition: "DefinitionContainer",
|
|
variant_type: VariantType) -> Optional["ContainerNode"]:
|
|
machine_definition_id = machine_definition.getId()
|
|
preferred_variant_name = None
|
|
if variant_type == VariantType.BUILD_PLATE:
|
|
if parseBool(machine_definition.getMetaDataEntry("has_variant_buildplates", False)):
|
|
preferred_variant_name = machine_definition.getMetaDataEntry("preferred_variant_buildplate_name")
|
|
else:
|
|
if parseBool(machine_definition.getMetaDataEntry("has_variants", False)):
|
|
preferred_variant_name = machine_definition.getMetaDataEntry("preferred_variant_name")
|
|
|
|
node = None
|
|
if preferred_variant_name:
|
|
node = self.getVariantNode(machine_definition_id, preferred_variant_name, variant_type)
|
|
return node
|
|
|
|
def getBuildplateVariantNode(self, machine_definition_id: str, buildplate_type: str) -> Optional["ContainerNode"]:
|
|
if machine_definition_id in self._machine_to_buildplate_dict_map:
|
|
return self._machine_to_buildplate_dict_map[machine_definition_id].get(buildplate_type)
|
|
return None
|