mirror of
https://git.mirrors.martin98.com/https://github.com/Ultimaker/Cura
synced 2025-05-05 14:54:33 +08:00
4d0bb217af
The fix is likely not what was intended by the author of the offending code (and does not seem to be functional), but at least it fixes the hard crash outlined in CURA-1666
56 lines
2.0 KiB
Python
56 lines
2.0 KiB
Python
# Copyright (c) 2016 Ultimaker B.V.
|
|
# Cura is released under the terms of the AGPLv3 or higher.
|
|
|
|
from PyQt5.QtCore import Qt
|
|
|
|
import cura.ExtruderManager
|
|
import UM.Qt.ListModel
|
|
|
|
## Model that holds extruders.
|
|
#
|
|
# This model is designed for use by any list of extruders, but specifically
|
|
# intended for drop-down lists of extruders in place of settings.
|
|
class ExtrudersModel(UM.Qt.ListModel.ListModel):
|
|
## Human-readable name of the extruder.
|
|
NameRole = Qt.UserRole + 1
|
|
|
|
## Colour of the material loaded in the extruder.
|
|
ColourRole = Qt.UserRole + 2
|
|
|
|
## Index of the extruder, which is also the value of the setting itself.
|
|
#
|
|
# An index of 0 indicates the first extruder, an index of 1 the second
|
|
# one, and so on. This is the value that will be saved in instance
|
|
# containers.
|
|
IndexRole = Qt.UserRole + 3
|
|
|
|
## Initialises the extruders model, defining the roles and listening for
|
|
# changes in the data.
|
|
#
|
|
# \param parent Parent QtObject of this list.
|
|
def __init__(self, parent = None):
|
|
super().__init__(parent)
|
|
|
|
self.addRoleName(self.NameRole, "name")
|
|
self.addRoleName(self.ColourRole, "colour")
|
|
self.addRoleName(self.IndexRole, "index")
|
|
|
|
#Listen to changes.
|
|
manager = cura.ExtruderManager.ExtruderManager.getInstance()
|
|
manager.extrudersChanged.connect(self._updateExtruders)
|
|
self._updateExtruders()
|
|
|
|
## Update the list of extruders.
|
|
#
|
|
# This should be called whenever the list of extruders changes.
|
|
def _updateExtruders(self):
|
|
self.clear()
|
|
manager = cura.ExtruderManager.ExtruderManager.getInstance()
|
|
for index, extruder in enumerate(manager._extruder_trains):
|
|
item = { #Construct an item with only the relevant information.
|
|
"name": extruder.name,
|
|
"colour": extruder.material.getMetaDataEntry("color_code", default = "#FFFF00"),
|
|
"index": index
|
|
}
|
|
self.appendItem(item)
|
|
self.sort(lambda item: item["index"]) |