Cura/cura/Machines/Models/CompatibleMachineModel.py

# Copyright (c) 2022 Ultimaker B.V.
# Cura is released under the terms of the LGPLv3 or higher.

from typing import Optional

from PyQt6.QtCore import Qt, QObject, pyqtSlot, pyqtProperty, pyqtSignal

from UM.Logger import Logger
from UM.Qt.ListModel import ListModel
from UM.i18n import i18nCatalog


class CompatibleMachineModel(ListModel):
    NameRole = Qt.ItemDataRole.UserRole + 1
    UniqueIdRole = Qt.ItemDataRole.UserRole + 2
    ExtrudersRole = Qt.ItemDataRole.UserRole + 3

    def __init__(self, parent: Optional[QObject] = None) -> None:
        super().__init__(parent)

        self._catalog = i18nCatalog("cura")

        self.addRoleName(self.NameRole, "name")
        self.addRoleName(self.UniqueIdRole, "unique_id")
        self.addRoleName(self.ExtrudersRole, "extruders")

        self._update()

        from cura.CuraApplication import CuraApplication
        machine_manager = CuraApplication.getInstance().getMachineManager()
        machine_manager.globalContainerChanged.connect(self._update)
        machine_manager.outputDevicesChanged.connect(self._update)

    @pyqtSlot()
    def forceUpdate(self):
        self._update()

    def _update(self) -> None:
        self.clear()

        def _makeMaterial(brand, name, color):
            if name.lower() in ["", "empty"]:
                return {"brand": "", "name": "(empty)", "hexcolor": "#ffffff"}
            else:
                return {"brand": brand, "name": name, "hexcolor": color}

        from cura.CuraApplication import CuraApplication
        machine_manager = CuraApplication.getInstance().getMachineManager()

        # Loop over the output-devices, not the stacks; need all applicable configurations, not just the current loaded one.
        for output_device in machine_manager.printerOutputDevices:
            for printer in output_device.printers:
                extruder_configs = dict()

                # initialize & add current active material:
                for extruder in printer.extruders:
                    materials = [_makeMaterial(
                        extruder.activeMaterial.brand, extruder.activeMaterial.name, extruder.activeMaterial.color)]
                    extruder_configs[extruder.getPosition()] = {
                        "position": extruder.getPosition(),
                        "core": extruder.hotendID,
                        "materials": materials
                    }

                # add currently inactive, but possible materials:
                for configuration in printer.availableConfigurations:
                    for extruder in configuration.extruderConfigurations:
                        if not extruder.position in extruder_configs:
                            Logger.log("w", f"No active extruder for position {extruder.position}.")
                            continue

                        entry = _makeMaterial(extruder.material.brand, extruder.material.name, extruder.material.color)
                        if entry not in extruder_configs[extruder.position]["materials"]:
                            extruder_configs[extruder.position]["materials"].append(entry)

                if any([len(extruder["materials"]) > 0 for extruder in extruder_configs.values()]):
                    self.appendItem({
                        "name": printer.name,
                        "unique_id": printer.name,  # <- Can assume the cloud doesn't have duplicate names?
                        "extruders": list(extruder_configs.values())
                    })