Rename quality drop down menu models

cura/Machines/Models/QualityProfilesDropDownMenuModel.py

@@ -0,0 +1,105 @@
+# Copyright (c) 2018 Ultimaker B.V.
+# Cura is released under the terms of the LGPLv3 or higher.
+
+from PyQt5.QtCore import Qt
+
+from UM.Application import Application
+from UM.Logger import Logger
+from UM.Qt.ListModel import ListModel
+
+from cura.Machines.QualityManager import QualityGroup
+
+
+#
+# QML Model for all built-in quality profiles. This model is used for the drop-down quality menu.
+#
+class QualityProfilesDropDownMenuModel(ListModel):
+    NameRole = Qt.UserRole + 1
+    QualityTypeRole = Qt.UserRole + 2
+    LayerHeightRole = Qt.UserRole + 3
+    LayerHeightWithoutUnitRole = Qt.UserRole + 4
+    AvailableRole = Qt.UserRole + 5
+    QualityGroupRole = Qt.UserRole + 6
+    QualityChangesGroupRole = Qt.UserRole + 7
+
+    def __init__(self, parent = None):
+        super().__init__(parent)
+
+        self.addRoleName(self.NameRole, "name")
+        self.addRoleName(self.QualityTypeRole, "quality_type")
+        self.addRoleName(self.LayerHeightRole, "layer_height")
+        self.addRoleName(self.LayerHeightWithoutUnitRole, "layer_height_without_unit")
+        self.addRoleName(self.AvailableRole, "available")
+        self.addRoleName(self.QualityGroupRole, "quality_group")
+        self.addRoleName(self.QualityChangesGroupRole, "quality_changes_group")
+
+        # connect signals
+        Application.getInstance().globalContainerStackChanged.connect(self._update)
+        Application.getInstance().getMachineManager().activeQualityGroupChanged.connect(self._update)
+
+        self._quality_manager = Application.getInstance()._quality_manager
+        self._quality_manager.qualitiesUpdated.connect(self._update)
+
+        self._layer_height_unit = ""  # This is cached
+
+        self._update()
+
+    def _update(self):
+        Logger.log("d", "Updating quality profile model ...")
+
+        machine_manager = Application.getInstance().getMachineManager()
+        global_stack = machine_manager._global_container_stack
+        if global_stack is None:
+            self.setItems([])
+            Logger.log("d", "No active GlobalStack, set quality profile model as empty.")
+            return
+
+        # Check for material compatibility
+        if not machine_manager.activeMaterialsCompatible():
+            self.setItems([])
+            return
+
+        quality_group_dict = self._quality_manager.getQualityGroups(global_stack)
+
+        item_list = []
+        for key in sorted(quality_group_dict):
+            quality_group = quality_group_dict[key]
+
+            layer_height = self._fetchLayerHeight(quality_group)
+
+            item = {"name": quality_group.name,
+                    "quality_type": quality_group.quality_type,
+                    "layer_height": layer_height + self._layer_height_unit,
+                    "layer_height_without_unit": layer_height,
+                    "available": quality_group.is_available,
+                    "quality_group": quality_group}
+
+            item_list.append(item)
+
+        # Sort items based on layer_height
+        item_list = sorted(item_list, key = lambda x: float(x["layer_height_without_unit"]))
+
+        self.setItems(item_list)
+
+    def _fetchLayerHeight(self, quality_group: "QualityGroup"):
+        global_stack = Application.getInstance().getMachineManager()._global_container_stack
+        if not self._layer_height_unit:
+            unit = global_stack.definition.getProperty("layer_height", "unit")
+            if not unit:
+                unit = ""
+            self._layer_height_unit = unit
+
+        default_layer_height = global_stack.definition.getProperty("layer_height", "value")
+
+        # Get layer_height from the quality profile for the GlobalStack
+        container = quality_group.node_for_global.getContainer()
+
+        layer_height = default_layer_height
+        if container.hasProperty("layer_height", "value"):
+            layer_height = str(container.getProperty("layer_height", "value"))
+        else:
+            # Look for layer_height in the GlobalStack from material -> definition
+            container = global_stack.definition
+            if container.hasProperty("layer_height", "value"):
+                layer_height = container.getProperty("layer_height", "value")
+        return str(layer_height)