mirror of
https://git.mirrors.martin98.com/https://github.com/Ultimaker/Cura
synced 2025-04-24 06:39:38 +08:00
82 lines
3.9 KiB
Python
82 lines
3.9 KiB
Python
# Copyright (c) 2019 Ultimaker B.V.
|
|
# Cura is released under the terms of the LGPLv3 or higher.
|
|
|
|
from UM.Logger import Logger
|
|
from UM.Settings.ContainerRegistry import ContainerRegistry # To listen to containers being added.
|
|
from UM.Settings.DefinitionContainer import DefinitionContainer
|
|
from UM.Settings.Interfaces import ContainerInterface
|
|
import cura.CuraApplication # Imported like this to prevent circular dependencies.
|
|
from UM.Signal import Signal
|
|
from cura.Machines.MachineNode import MachineNode
|
|
|
|
from typing import Dict, List, TYPE_CHECKING
|
|
import time
|
|
|
|
if TYPE_CHECKING:
|
|
from cura.Machines.QualityGroup import QualityGroup
|
|
|
|
## This class contains a look-up tree for which containers are available at
|
|
# which stages of configuration.
|
|
#
|
|
# The tree starts at the machine definitions. For every distinct definition
|
|
# there will be one machine node here.
|
|
#
|
|
# All of the fallbacks for material choices, quality choices, etc. should be
|
|
# encoded in this tree. There must always be at least one child node (for
|
|
# nodes that have children) but that child node may be a node representing the
|
|
# empty instance container.
|
|
class ContainerTree:
|
|
__instance = None
|
|
|
|
@classmethod
|
|
def getInstance(cls):
|
|
if cls.__instance is None:
|
|
cls.__instance = ContainerTree()
|
|
return cls.__instance
|
|
|
|
def __init__(self) -> None:
|
|
self.machines = {} # type: Dict[str, MachineNode] # Mapping from definition ID to machine nodes.
|
|
self.materialsChanged = Signal() # Emitted when any of the material nodes in the tree got changed.
|
|
|
|
container_registry = ContainerRegistry.getInstance()
|
|
container_registry.containerAdded.connect(self._machineAdded)
|
|
self._loadAll()
|
|
|
|
## Get the quality groups available for the currently activated printer.
|
|
#
|
|
# This contains all quality groups, enabled or disabled. To check whether
|
|
# the quality group can be activated, test for the
|
|
# ``QualityGroup.is_available`` property.
|
|
# \return For every quality type, one quality group.
|
|
def getCurrentQualityGroups(self) -> Dict[str, "QualityGroup"]:
|
|
global_stack = cura.CuraApplication.CuraApplication.getInstance().getGlobalContainerStack()
|
|
if global_stack is None:
|
|
return {}
|
|
variant_names = [extruder.variant.getName() for extruder in global_stack.extruders.values()]
|
|
material_bases = [extruder.material.getMetaDataEntry("base_file") for extruder in global_stack.extruders.values()]
|
|
extruder_enabled = [extruder.isEnabled for extruder in global_stack.extruders.values()]
|
|
return self.machines[global_stack.definition.getId()].getQualityGroups(variant_names, material_bases, extruder_enabled)
|
|
|
|
## Builds the initial container tree.
|
|
def _loadAll(self):
|
|
Logger.log("i", "Building container tree.")
|
|
start_time = time.time()
|
|
all_stacks = ContainerRegistry.getInstance().findContainerStacks()
|
|
for stack in all_stacks:
|
|
definition_id = stack.definition.getId()
|
|
if definition_id not in self.machines:
|
|
self.machines[definition_id] = MachineNode(definition_id)
|
|
self.machines[definition_id].materialsChanged.connect(self.materialsChanged)
|
|
|
|
Logger.log("d", "Building the container tree took %s seconds", time.time() - start_time)
|
|
|
|
## When a printer gets added, we need to build up the tree for that container.
|
|
def _machineAdded(self, definition_container: ContainerInterface):
|
|
if not isinstance(definition_container, DefinitionContainer):
|
|
return # Not our concern.
|
|
definition_id = definition_container.getId()
|
|
if definition_id in self.machines:
|
|
return # Already have this definition ID.
|
|
|
|
self.machines[definition_id] = MachineNode(definition_id)
|
|
self.machines[definition_id].materialsChanged.connect(self.materialsChanged) |