Split material models into separate files

cura/Machines/Models/BaseMaterialsModel.py

@@ -0,0 +1,71 @@
+# Copyright (c) 2018 Ultimaker B.V.
+# Cura is released under the terms of the LGPLv3 or higher.
+
+from typing import Optional
+from PyQt5.QtCore import Qt, pyqtSignal, pyqtProperty
+
+from UM.Logger import Logger
+from UM.Qt.ListModel import ListModel
+
+
+def getAvailableMaterials(extruder_position: Optional[int] = None):
+    from cura.CuraApplication import CuraApplication
+    machine_manager = CuraApplication.getInstance().getMachineManager()
+    extruder_manager = CuraApplication.getInstance().getExtruderManager()
+
+    material_manager = CuraApplication.getInstance()._material_manager
+
+    active_global_stack = machine_manager._global_container_stack
+    extruder_stack = extruder_manager.getActiveExtruderStack()
+    if extruder_position is not None:
+        if active_global_stack is not None:
+            extruder_stack = active_global_stack.extruders.get(str(extruder_position))
+
+    if active_global_stack is None or extruder_stack is None:
+        Logger.log("d", "Active global stack [%s] or extruder stack [%s] is None, setting material list to empty.",
+                   active_global_stack, extruder_stack)
+        return
+
+    machine_definition_id = active_global_stack.definition.getId()
+    variant_name = None
+    if extruder_stack.variant.getId() != "empty_variant":
+        variant_name = extruder_stack.variant.getName()
+    diameter = extruder_stack.approximateMaterialDiameter
+
+    # Fetch the available materials (ContainerNode) for the current active machine and extruder setup.
+    result_dict = material_manager.getAvailableMaterials(machine_definition_id, variant_name, diameter)
+    return result_dict
+
+
+class BaseMaterialsModel(ListModel):
+    RootMaterialIdRole = Qt.UserRole + 1
+    IdRole = Qt.UserRole + 2
+    NameRole = Qt.UserRole + 3
+    BrandRole = Qt.UserRole + 4
+    MaterialRole = Qt.UserRole + 5
+    ColorRole = Qt.UserRole + 6
+    ContainerNodeRole = Qt.UserRole + 7
+
+    extruderPositionChanged = pyqtSignal()
+
+    def __init__(self, parent = None):
+        super().__init__(parent)
+
+        self.addRoleName(self.RootMaterialIdRole, "root_material_id")
+        self.addRoleName(self.IdRole, "id")
+        self.addRoleName(self.NameRole, "name")
+        self.addRoleName(self.BrandRole, "brand")
+        self.addRoleName(self.MaterialRole, "material")
+        self.addRoleName(self.ColorRole, "color_name")
+        self.addRoleName(self.ContainerNodeRole, "container_node")
+
+        self._extruder_position = 0
+
+    def setExtruderPosition(self, position: int):
+        if self._extruder_position != position:
+            self._extruder_position = position
+            self.extruderPositionChanged.emit()
+
+    @pyqtProperty(int, fset = setExtruderPosition, notify = extruderPositionChanged)
+    def extruderPosition(self) -> int:
+        return self._extruder_positoin