rscder-pub/rscder/gui/layercombox.py
2022-05-12 18:57:30 +08:00

27 lines
749 B
Python

from PyQt5.QtWidgets import QComboBox
from PyQt5.QtGui import QIcon
from rscder.utils.project import Project
class LayerCombox(QComboBox):
def __init__(self, parent=None):
super().__init__(parent)
self.addItem('---', None)
for layer in Project().layers.values():
self.addItem(layer.name, layer.id)
for i in range(self.count() - 1):
self.setItemIcon(i + 1, QIcon(':/icons/layer.png'))
self.currentIndexChanged.connect(self.on_changed)
self.current_layer = None
def on_changed(self, index):
if index == 0:
self.current_layer = None
else:
self.current_layer = Project().layers[self.itemData(index)]