mirror of
https://github.com/Ultimaker/Cura.git
synced 2025-07-07 06:57:28 -06:00
04cd76cb3b
When choosing the printer, all possible (single) extruder configs need to be shown for that printer. CURA-9278
84 lines
3.4 KiB
Python
84 lines
3.4 KiB
Python
# Copyright (c) 2022 Ultimaker B.V.
|
|
# Cura is released under the terms of the LGPLv3 or higher.
|
|
|
|
# TODO?: documentation
|
|
|
|
from typing import Optional, Dict, cast
|
|
|
|
from PyQt6.QtCore import Qt, QObject, pyqtSlot, pyqtProperty, pyqtSignal
|
|
|
|
from UM.Qt.ListModel import ListModel
|
|
from UM.Settings.ContainerStack import ContainerStack
|
|
from UM.i18n import i18nCatalog
|
|
from UM.Util import parseBool
|
|
|
|
from cura.Settings.CuraContainerRegistry import CuraContainerRegistry
|
|
from cura.Settings.ExtruderStack import ExtruderStack
|
|
|
|
|
|
class CompatibleMachineModel(ListModel):
|
|
NameRole = Qt.ItemDataRole.UserRole + 1
|
|
IdRole = 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.IdRole, "id")
|
|
self.addRoleName(self.ExtrudersRole, "extruders")
|
|
|
|
self._update()
|
|
|
|
from cura.CuraApplication import CuraApplication
|
|
machine_manager = CuraApplication.getInstance().getMachineManager()
|
|
machine_manager.globalContainerChanged.connect(self._update)
|
|
|
|
def _update(self) -> None:
|
|
self.clear()
|
|
|
|
from cura.CuraApplication import CuraApplication
|
|
machine_manager = CuraApplication.getInstance().getMachineManager()
|
|
|
|
# Need to loop over the output-devices, not the stacks, since we 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 = [] if not extruder.activeMaterial else [{
|
|
"brand": extruder.activeMaterial.brand,
|
|
"name": extruder.activeMaterial.name,
|
|
"hexcolor": 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:
|
|
print(" CONFIG !")
|
|
for extruder in configuration.extruderConfigurations:
|
|
|
|
if not extruder.position in extruder_configs:
|
|
# TODO: log -- all extruders should be present in the init round, regardless of if a material was active
|
|
continue
|
|
|
|
extruder_configs[extruder.position]["materials"].append({
|
|
"brand": extruder.material.brand,
|
|
"name": extruder.material.name,
|
|
"hexcolor": extruder.material.color
|
|
})
|
|
|
|
self.appendItem({
|
|
"name": printer.name,
|
|
"id": printer.uniqueName,
|
|
"extruders": [extruder for extruder in extruder_configs.values()]
|
|
})
|
|
|
|
# TODO: Handle 0 compatible machines -> option to close window? Message in card? (remember the design has a refresh button!)
|