mirror of
https://git.mirrors.martin98.com/https://github.com/Ultimaker/Cura
synced 2025-05-01 16:24:41 +08:00
110 lines
4.5 KiB
Python
110 lines
4.5 KiB
Python
#Copyright (c) 2019 Ultimaker B.V.
|
|
#Cura is released under the terms of the LGPLv3 or higher.
|
|
|
|
from PyQt5.QtCore import Qt, QTimer
|
|
import collections
|
|
from typing import TYPE_CHECKING, Optional, Dict
|
|
|
|
from cura.Machines.Models.IntentModel import IntentModel
|
|
from cura.Settings.IntentManager import IntentManager
|
|
from UM.Qt.ListModel import ListModel
|
|
from UM.Settings.ContainerRegistry import ContainerRegistry #To update the list if anything changes.
|
|
from PyQt5.QtCore import pyqtProperty, pyqtSignal
|
|
import cura.CuraApplication
|
|
if TYPE_CHECKING:
|
|
from UM.Settings.ContainerRegistry import ContainerInterface
|
|
|
|
from UM.i18n import i18nCatalog
|
|
catalog = i18nCatalog("cura")
|
|
|
|
|
|
## Lists the intent categories that are available for the current printer
|
|
# configuration.
|
|
class IntentCategoryModel(ListModel):
|
|
NameRole = Qt.UserRole + 1
|
|
IntentCategoryRole = Qt.UserRole + 2
|
|
WeightRole = Qt.UserRole + 3
|
|
QualitiesRole = Qt.UserRole + 4
|
|
DescriptionRole = Qt.UserRole + 5
|
|
|
|
modelUpdated = pyqtSignal()
|
|
|
|
# Translations to user-visible string. Ordered by weight.
|
|
# TODO: Create a solution for this name and weight to be used dynamically.
|
|
_translations = collections.OrderedDict() # type: "collections.OrderedDict[str,Dict[str,Optional[str]]]"
|
|
_translations["default"] = {
|
|
"name": catalog.i18nc("@label", "Default")
|
|
}
|
|
_translations["engineering"] = {
|
|
"name": catalog.i18nc("@label", "Engineering"),
|
|
"description": catalog.i18nc("@text", "Suitable for engineering work")
|
|
|
|
}
|
|
_translations["smooth"] = {
|
|
"name": catalog.i18nc("@label", "Smooth"),
|
|
"description": catalog.i18nc("@text", "Optimized for a smooth surfaces")
|
|
}
|
|
|
|
## Creates a new model for a certain intent category.
|
|
# \param The category to list the intent profiles for.
|
|
def __init__(self, intent_category: str) -> None:
|
|
super().__init__()
|
|
self._intent_category = intent_category
|
|
|
|
self.addRoleName(self.NameRole, "name")
|
|
self.addRoleName(self.IntentCategoryRole, "intent_category")
|
|
self.addRoleName(self.WeightRole, "weight")
|
|
self.addRoleName(self.QualitiesRole, "qualities")
|
|
self.addRoleName(self.DescriptionRole, "description")
|
|
|
|
application = cura.CuraApplication.CuraApplication.getInstance()
|
|
|
|
ContainerRegistry.getInstance().containerAdded.connect(self._onContainerChange)
|
|
ContainerRegistry.getInstance().containerRemoved.connect(self._onContainerChange)
|
|
machine_manager = cura.CuraApplication.CuraApplication.getInstance().getMachineManager()
|
|
machine_manager.activeMaterialChanged.connect(self.update)
|
|
machine_manager.activeVariantChanged.connect(self.update)
|
|
machine_manager.extruderChanged.connect(self.update)
|
|
|
|
extruder_manager = application.getExtruderManager()
|
|
extruder_manager.extrudersChanged.connect(self.update)
|
|
|
|
self._update_timer = QTimer()
|
|
self._update_timer.setInterval(500)
|
|
self._update_timer.setSingleShot(True)
|
|
self._update_timer.timeout.connect(self._update)
|
|
|
|
self.update()
|
|
|
|
## Updates the list of intents if an intent profile was added or removed.
|
|
def _onContainerChange(self, container: "ContainerInterface") -> None:
|
|
if container.getMetaDataEntry("type") == "intent":
|
|
self.update()
|
|
|
|
def update(self):
|
|
self._update_timer.start()
|
|
|
|
## Updates the list of intents.
|
|
def _update(self) -> None:
|
|
available_categories = IntentManager.getInstance().currentAvailableIntentCategories()
|
|
result = []
|
|
for category in available_categories:
|
|
qualities = IntentModel()
|
|
qualities.setIntentCategory(category)
|
|
result.append({
|
|
"name": IntentCategoryModel.translation(category, "name", catalog.i18nc("@label", "Unknown")),
|
|
"description": IntentCategoryModel.translation(category, "description", None),
|
|
"intent_category": category,
|
|
"weight": list(self._translations.keys()).index(category),
|
|
"qualities": qualities
|
|
})
|
|
result.sort(key = lambda k: k["weight"])
|
|
self.setItems(result)
|
|
|
|
## Get a display value for a category. See IntenCategoryModel._translations
|
|
## for categories and keys
|
|
@staticmethod
|
|
def translation(category: str, key: str, default: Optional[str] = None):
|
|
display_strings = IntentCategoryModel._translations.get(category, {})
|
|
return display_strings.get(key, default)
|