mirror of
https://git.mirrors.martin98.com/https://github.com/Ultimaker/Cura
synced 2025-04-23 22:29:41 +08:00
d8853b8a98
Cast every sorting key to uppercase before doing this. Don't cast to lowercase or there will be problems with characters that don't have lowercase and with Turkish dotted i vs. undotted i. Fixes #3460.
123 lines
5.2 KiB
Python
123 lines
5.2 KiB
Python
# Copyright (c) 2018 Ultimaker B.V.
|
|
# Cura is released under the terms of the LGPLv3 or higher.
|
|
|
|
from PyQt5.QtCore import Qt, pyqtSlot
|
|
|
|
from UM.Qt.ListModel import ListModel
|
|
|
|
|
|
#
|
|
# This the QML model for the quality management page.
|
|
#
|
|
class QualityManagementModel(ListModel):
|
|
NameRole = Qt.UserRole + 1
|
|
IsReadOnlyRole = Qt.UserRole + 2
|
|
QualityGroupRole = Qt.UserRole + 3
|
|
QualityChangesGroupRole = Qt.UserRole + 4
|
|
|
|
def __init__(self, parent = None):
|
|
super().__init__(parent)
|
|
|
|
self.addRoleName(self.NameRole, "name")
|
|
self.addRoleName(self.IsReadOnlyRole, "is_read_only")
|
|
self.addRoleName(self.QualityGroupRole, "quality_group")
|
|
self.addRoleName(self.QualityChangesGroupRole, "quality_changes_group")
|
|
|
|
from cura.CuraApplication import CuraApplication
|
|
self._container_registry = CuraApplication.getInstance().getContainerRegistry()
|
|
self._machine_manager = CuraApplication.getInstance().getMachineManager()
|
|
self._extruder_manager = CuraApplication.getInstance().getExtruderManager()
|
|
self._quality_manager = CuraApplication.getInstance().getQualityManager()
|
|
|
|
self._machine_manager.globalContainerChanged.connect(self._update)
|
|
self._quality_manager.qualitiesUpdated.connect(self._update)
|
|
|
|
self._update()
|
|
|
|
def _update(self):
|
|
global_stack = self._machine_manager.activeMachine
|
|
|
|
quality_group_dict = self._quality_manager.getQualityGroups(global_stack)
|
|
quality_changes_group_dict = self._quality_manager.getQualityChangesGroups(global_stack)
|
|
|
|
available_quality_types = set(quality_type for quality_type, quality_group in quality_group_dict.items()
|
|
if quality_group.is_available)
|
|
if not available_quality_types and not quality_changes_group_dict:
|
|
# Nothing to show
|
|
self.setItems([])
|
|
return
|
|
|
|
item_list = []
|
|
# Create quality group items
|
|
for quality_group in quality_group_dict.values():
|
|
if not quality_group.is_available:
|
|
continue
|
|
|
|
item = {"name": quality_group.name,
|
|
"is_read_only": True,
|
|
"quality_group": quality_group,
|
|
"quality_changes_group": None}
|
|
item_list.append(item)
|
|
# Sort by quality names
|
|
item_list = sorted(item_list, key = lambda x: x["name"].upper())
|
|
|
|
# Create quality_changes group items
|
|
quality_changes_item_list = []
|
|
for quality_changes_group in quality_changes_group_dict.values():
|
|
if quality_changes_group.quality_type not in available_quality_types:
|
|
continue
|
|
quality_group = quality_group_dict[quality_changes_group.quality_type]
|
|
item = {"name": quality_changes_group.name,
|
|
"is_read_only": False,
|
|
"quality_group": quality_group,
|
|
"quality_changes_group": quality_changes_group}
|
|
quality_changes_item_list.append(item)
|
|
|
|
# Sort quality_changes items by names and append to the item list
|
|
quality_changes_item_list = sorted(quality_changes_item_list, key = lambda x: x["name"].upper())
|
|
item_list += quality_changes_item_list
|
|
|
|
self.setItems(item_list)
|
|
|
|
# TODO: Duplicated code here from InstanceContainersModel. Refactor and remove this later.
|
|
#
|
|
## Gets a list of the possible file filters that the plugins have
|
|
# registered they can read or write. The convenience meta-filters
|
|
# "All Supported Types" and "All Files" are added when listing
|
|
# readers, but not when listing writers.
|
|
#
|
|
# \param io_type \type{str} name of the needed IO type
|
|
# \return A list of strings indicating file name filters for a file
|
|
# dialog.
|
|
@pyqtSlot(str, result = "QVariantList")
|
|
def getFileNameFilters(self, io_type):
|
|
from UM.i18n import i18nCatalog
|
|
catalog = i18nCatalog("uranium")
|
|
#TODO: This function should be in UM.Resources!
|
|
filters = []
|
|
all_types = []
|
|
for plugin_id, meta_data in self._getIOPlugins(io_type):
|
|
for io_plugin in meta_data[io_type]:
|
|
filters.append(io_plugin["description"] + " (*." + io_plugin["extension"] + ")")
|
|
all_types.append("*.{0}".format(io_plugin["extension"]))
|
|
|
|
if "_reader" in io_type:
|
|
# if we're listing readers, add the option to show all supported files as the default option
|
|
filters.insert(0, catalog.i18nc("@item:inlistbox", "All Supported Types ({0})", " ".join(all_types)))
|
|
filters.append(catalog.i18nc("@item:inlistbox", "All Files (*)")) # Also allow arbitrary files, if the user so prefers.
|
|
return filters
|
|
|
|
## Gets a list of profile reader or writer plugins
|
|
# \return List of tuples of (plugin_id, meta_data).
|
|
def _getIOPlugins(self, io_type):
|
|
from UM.PluginRegistry import PluginRegistry
|
|
pr = PluginRegistry.getInstance()
|
|
active_plugin_ids = pr.getActivePlugins()
|
|
|
|
result = []
|
|
for plugin_id in active_plugin_ids:
|
|
meta_data = pr.getMetaData(plugin_id)
|
|
if io_type in meta_data:
|
|
result.append( (plugin_id, meta_data) )
|
|
return result
|