123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362 |
- from collections import defaultdict
- import threading
- from typing import Any, Dict, Optional, Set, TYPE_CHECKING, List
- import uuid
- from PyQt6.QtCore import pyqtProperty, pyqtSlot, pyqtSignal
- from UM.Decorators import override
- from UM.MimeTypeDatabase import MimeType, MimeTypeDatabase
- from UM.Settings.ContainerStack import ContainerStack
- from UM.Settings.ContainerRegistry import ContainerRegistry
- from UM.Settings.Interfaces import PropertyEvaluationContext
- from UM.Logger import Logger
- from UM.Resources import Resources
- from UM.Platform import Platform
- from UM.Util import parseBool
- import cura.CuraApplication
- from cura.PrinterOutput.PrinterOutputDevice import ConnectionType
- from . import Exceptions
- from .CuraContainerStack import CuraContainerStack
- if TYPE_CHECKING:
- from cura.Settings.ExtruderStack import ExtruderStack
- class GlobalStack(CuraContainerStack):
- """Represents the Global or Machine stack and its related containers."""
- def __init__(self, container_id: str) -> None:
- super().__init__(container_id)
- self.setMetaDataEntry("type", "machine")
-
-
-
-
- self.setMetaDataEntry("group_id", str(uuid.uuid4()))
- self._extruders = {}
-
-
-
-
- self._resolving_settings = defaultdict(set)
-
-
- self.metaDataChanged.connect(self.configuredConnectionTypesChanged)
- self.setDirty(False)
- extrudersChanged = pyqtSignal()
- configuredConnectionTypesChanged = pyqtSignal()
- @pyqtProperty("QVariantList", notify = extrudersChanged)
- def extruderList(self) -> List["ExtruderStack"]:
- result_tuple_list = sorted(list(self._extruders.items()), key=lambda x: int(x[0]))
- result_list = [item[1] for item in result_tuple_list]
- machine_extruder_count = self.getProperty("machine_extruder_count", "value")
- return result_list[:machine_extruder_count]
- @pyqtProperty(int, constant = True)
- def maxExtruderCount(self):
- return len(self.getMetaDataEntry("machine_extruder_trains"))
- @pyqtProperty(bool, notify=configuredConnectionTypesChanged)
- def supportsNetworkConnection(self):
- return self.getMetaDataEntry("supports_network_connection", False)
- @pyqtProperty(bool, constant = True)
- def supportsMaterialExport(self):
- """
- Whether the printer supports Cura's export format of material profiles.
- :return: ``True`` if it supports it, or ``False`` if not.
- """
- return self.getMetaDataEntry("supports_material_export", False)
- @classmethod
- def getLoadingPriority(cls) -> int:
- return 2
- @pyqtProperty("QVariantList", notify=configuredConnectionTypesChanged)
- def configuredConnectionTypes(self) -> List[int]:
- """The configured connection types can be used to find out if the global
- stack is configured to be connected with a printer, without having to
- know all the details as to how this is exactly done (and without
- actually setting the stack to be active).
- This data can then in turn also be used when the global stack is active;
- If we can't get a network connection, but it is configured to have one,
- we can display a different icon to indicate the difference.
- """
-
-
- connection_types = self.getMetaDataEntry("connection_type", "").split(",")
- result = []
- for connection_type in connection_types:
- if connection_type != "":
- try:
- result.append(int(connection_type))
- except ValueError:
-
- pass
- return result
-
-
-
-
- @pyqtProperty(bool, notify = configuredConnectionTypesChanged)
- def hasRemoteConnection(self) -> bool:
- has_remote_connection = False
- for connection_type in self.configuredConnectionTypes:
- has_remote_connection |= connection_type in [ConnectionType.NetworkConnection.value,
- ConnectionType.CloudConnection.value]
- return has_remote_connection
- def addConfiguredConnectionType(self, connection_type: int) -> None:
- """:sa configuredConnectionTypes"""
- configured_connection_types = self.configuredConnectionTypes
- if connection_type not in configured_connection_types:
-
- configured_connection_types.append(connection_type)
- self.setMetaDataEntry("connection_type", ",".join([str(c_type) for c_type in configured_connection_types]))
- def removeConfiguredConnectionType(self, connection_type: int) -> None:
- """:sa configuredConnectionTypes"""
- configured_connection_types = self.configuredConnectionTypes
- if connection_type in configured_connection_types:
-
- configured_connection_types.remove(connection_type)
- self.setMetaDataEntry("connection_type", ",".join([str(c_type) for c_type in configured_connection_types]))
- @classmethod
- def getConfigurationTypeFromSerialized(cls, serialized: str) -> Optional[str]:
- configuration_type = super().getConfigurationTypeFromSerialized(serialized)
- if configuration_type == "machine":
- return "machine_stack"
- return configuration_type
- def getIntentCategory(self) -> str:
- intent_category = "default"
- for extruder in self.extruderList:
- category = extruder.intent.getMetaDataEntry("intent_category", "default")
- if category != "default" and category != intent_category:
- intent_category = category
- return intent_category
- def getBuildplateName(self) -> Optional[str]:
- name = None
- if self.variant.getId() != "empty_variant":
- name = self.variant.getName()
- return name
- @pyqtProperty(str, constant = True)
- def preferred_output_file_formats(self) -> str:
- return self.getMetaDataEntry("file_formats")
- def addExtruder(self, extruder: ContainerStack) -> None:
- """Add an extruder to the list of extruders of this stack.
- :param extruder: The extruder to add.
- :raise Exceptions.TooManyExtrudersError: Raised when trying to add an extruder while we
- already have the maximum number of extruders.
- """
- position = extruder.getMetaDataEntry("position")
- if position is None:
- Logger.log("w", "No position defined for extruder {extruder}, cannot add it to stack {stack}", extruder = extruder.id, stack = self.id)
- return
- if any(item.getId() == extruder.id for item in self._extruders.values()):
- Logger.log("w", "Extruder [%s] has already been added to this stack [%s]", extruder.id, self.getId())
- return
- self._extruders[position] = extruder
- self.extrudersChanged.emit()
- Logger.log("i", "Extruder[%s] added to [%s] at position [%s]", extruder.id, self.id, position)
- @override(ContainerStack)
- def getProperty(self, key: str, property_name: str, context: Optional[PropertyEvaluationContext] = None) -> Any:
- """Overridden from ContainerStack
- This will return the value of the specified property for the specified setting,
- unless the property is "value" and that setting has a "resolve" function set.
- When a resolve is set, it will instead try and execute the resolve first and
- then fall back to the normal "value" property.
- :param key: The setting key to get the property of.
- :param property_name: The property to get the value of.
- :return: The value of the property for the specified setting, or None if not found.
- """
- if not self.definition.findDefinitions(key = key):
- return None
- if context:
- context.pushContainer(self)
-
-
-
-
-
- if self._shouldResolve(key, property_name, context):
- current_thread = threading.current_thread()
- self._resolving_settings[current_thread.name].add(key)
- resolve = super().getProperty(key, "resolve", context)
- self._resolving_settings[current_thread.name].remove(key)
- if resolve is not None:
- return resolve
-
- limit_to_extruder = super().getProperty(key, "limit_to_extruder", context)
- if limit_to_extruder is not None:
- limit_to_extruder = str(limit_to_extruder)
- if limit_to_extruder is not None and limit_to_extruder != "-1" and limit_to_extruder in self._extruders:
- if super().getProperty(key, "settable_per_extruder", context):
- result = self._extruders[str(limit_to_extruder)].getProperty(key, property_name, context)
- if result is not None:
- if context:
- context.popContainer()
- return result
- else:
- Logger.log("e", "Setting {setting} has limit_to_extruder but is not settable per extruder!", setting = key)
- result = super().getProperty(key, property_name, context)
- if context:
- context.popContainer()
- return result
- @override(ContainerStack)
- def setNextStack(self, stack: CuraContainerStack, connect_signals: bool = True) -> None:
- """Overridden from ContainerStack
- This will simply raise an exception since the Global stack cannot have a next stack.
- """
- raise Exceptions.InvalidOperationError("Global stack cannot have a next stack!")
-
-
- def _shouldResolve(self, key: str, property_name: str, context: Optional[PropertyEvaluationContext] = None) -> bool:
- if property_name != "value":
-
- return False
- if not self.definition.getProperty(key, "resolve"):
-
- return False
- current_thread = threading.current_thread()
- if key in self._resolving_settings[current_thread.name]:
-
-
-
-
- return False
- if self.hasUserValue(key):
-
- return False
- return True
- def isValid(self) -> bool:
- """Perform some sanity checks on the global stack
- Sanity check for extruders; they must have positions 0 and up to machine_extruder_count - 1
- """
- container_registry = ContainerRegistry.getInstance()
- extruder_trains = container_registry.findContainerStacks(type = "extruder_train", machine = self.getId())
- machine_extruder_count = self.getProperty("machine_extruder_count", "value")
- extruder_check_position = set()
- for extruder_train in extruder_trains:
- extruder_position = extruder_train.getMetaDataEntry("position")
- extruder_check_position.add(extruder_position)
- for check_position in range(machine_extruder_count):
- if str(check_position) not in extruder_check_position:
- return False
- return True
- def getHeadAndFansCoordinates(self):
- return self.getProperty("machine_head_with_fans_polygon", "value")
- @pyqtProperty(bool, constant = True)
- def hasMaterials(self) -> bool:
- return parseBool(self.getMetaDataEntry("has_materials", False))
- @pyqtProperty(bool, constant = True)
- def hasVariants(self) -> bool:
- return parseBool(self.getMetaDataEntry("has_variants", False))
- @pyqtProperty(bool, constant = True)
- def hasVariantBuildplates(self) -> bool:
- return parseBool(self.getMetaDataEntry("has_variant_buildplates", False))
- @pyqtSlot(result = str)
- def getDefaultFirmwareName(self) -> str:
- """Get default firmware file name if one is specified in the firmware"""
- machine_has_heated_bed = self.getProperty("machine_heated_bed", "value")
- baudrate = 250000
- if Platform.isLinux():
-
-
- baudrate = 115200
-
- hex_file = self.getMetaDataEntry("firmware_file", None)
- if machine_has_heated_bed:
- hex_file = self.getMetaDataEntry("firmware_hbk_file", hex_file)
- if not hex_file:
- Logger.log("w", "There is no firmware for machine %s.", self.getBottom().id)
- return ""
- try:
- return Resources.getPath(cura.CuraApplication.CuraApplication.ResourceTypes.Firmware, hex_file.format(baudrate=baudrate))
- except FileNotFoundError:
- Logger.log("w", "Firmware file %s not found.", hex_file)
- return ""
- def getName(self) -> str:
- return self._metadata.get("group_name", self._metadata.get("name", ""))
- def setName(self, name: str) -> None:
- super().setName(name)
- nameChanged = pyqtSignal()
- name = pyqtProperty(str, fget=getName, fset=setName, notify=nameChanged)
- def hasNetworkedConnection(self) -> bool:
- has_connection = False
- for connection_type in [ConnectionType.NetworkConnection.value, ConnectionType.CloudConnection.value]:
- has_connection |= connection_type in self.configuredConnectionTypes
- return has_connection
- global_stack_mime = MimeType(
- name = "application/x-cura-globalstack",
- comment = "Cura Global Stack",
- suffixes = ["global.cfg"]
- )
- MimeTypeDatabase.addMimeType(global_stack_mime)
- ContainerRegistry.addContainerTypeByName(GlobalStack, "global_stack", global_stack_mime.name)
|