# -*- coding: UTF-8 -*- from common import * from editorui import * class KPLayerList(QtGui.QWidget): def __init__(self): QtGui.QWidget.__init__(self) self.layout = QtGui.QVBoxLayout() self.layout.setSpacing(0) self.model = KP.map.layerModel self.listView = QtGui.QListView() self.listView.setModel(self.model) self.listView.selectionModel().currentRowChanged.connect(self.handleRowChanged) self.layout.addWidget(self.listView) self.toolbar = QtGui.QToolBar() self.layout.addWidget(self.toolbar) self.setupToolbar(self.toolbar) self.setLayout(self.layout) def setupToolbar(self, tb): tb.addAction(QtGui.QIcon(), 'Add', self.addLayer) tb.addAction(QtGui.QIcon(), 'Remove', self.removeLayer) tb.addAction(QtGui.QIcon(), 'Move Up', self.moveUp) tb.addAction(QtGui.QIcon(), 'Move Down', self.moveDown) def selectedLayerIndex(self): return self.listView.selectionModel().currentIndex().row() def selectedLayer(self): return KP.map.layers[self.listView.selectionModel().currentIndex().row()] selectedLayerChanged = QtCore.pyqtSignal(KPLayer) @QtCore.pyqtSlot(QtCore.QModelIndex, QtCore.QModelIndex) def handleRowChanged(self, current, previous): self.selectedLayerChanged.emit(KP.map.layers[current.row()]) def addLayer(self): KP.map.appendLayer(KP.map.createNewLayer()) def removeLayer(self): KP.map.removeLayer(self.selectedLayer()) def moveUp(self): index = self.selectedLayerIndex() KP.map.moveLayer(index, index - 1) def moveDown(self): index = self.selectedLayerIndex() KP.map.moveLayer(index, index + 2) class KPObjectSelector(QtGui.QListView): def __init__(self): """Initialises the widget. Remember to call setModel() on it with a KPGroupModel whenever the layer changes.""" font = QtGui.QFont() font.setPixelSize(22) font.setBold(True) icon = QtGui.QIcon('icons/downArrow.png') self.sorterButton = QtGui.QToolButton() string = QtCore.QString(QtCore.QChar(0x25BE)) string.append(' All Groups') self.sorterButton.setText(string) self.sorterButton.setFont(font) self.sorterButton.setPopupMode(2) self.sorterButton.setSizePolicy(QtGui.QSizePolicy.Expanding, QtGui.QSizePolicy.Expanding) self.sorterMenu = QtGui.QMenu() self.sorterMenu.setFixedWidth(self.sorterButton.width()) action = self.sorterMenu.addAction("test 1") self.sorterButton.setMenu(self.sorterMenu) layout = QtGui.QBoxLayout(2) layout.setAlignment(Qt.AlignTop) layout.setContentsMargins(0,0,0,0) layout.setMargin(0) layout.setSpacing(0) thing = QtGui.QListView.__init__(self) self.setFlow(QtGui.QListView.LeftToRight) self.setLayoutMode(QtGui.QListView.SinglePass) self.setMovement(QtGui.QListView.Static) self.setResizeMode(QtGui.QListView.Adjust) self.setWrapping(True) self.toolbar = QtGui.QToolBar() self.toolbar.setFixedHeight(28) self.toolbar.addWidget(self.sorterButton) layout.addWidget(self.toolbar) self.setLayout(layout) # Borrowed the signals and junk from Reggie, figure we'll need em' self.clicked.connect(self.handleObjReplace) self.sorterMenu.triggered.connect(self.toggleTopLevel) def resizeEvent(self, event): QtGui.QListView.resizeEvent(self, event) self.sorterMenu.setFixedWidth(self.width()-self.verticalScrollBar().width()) def currentSelectedObject(self): """Returns the currently selected object reference, for painting purposes.""" index = self.currentIndex().row() object = self.model().groupItem().getItem(index) return object def setModel(self, model): """Sets the model and the menu sorting list""" QtGui.QListView.setModel(self, model) model.view = self menuList = model.groupItem().getGroupList() string = QtCore.QString(QtCore.QChar(0x25BE)) string.append(' All Groups') self.sorterButton.setText(string) self.sorterMenu.clear() for item in menuList: actionMan = self.sorterMenu.addAction(item[0]) actionMan.setData((item[1], item[2])) def toggleTopLevel(self, action): """Changes the top level group in the list view.""" name = str(action.text()).strip() startRow = action.data().toPyObject()[0] endRow = action.data().toPyObject()[1] for row in xrange(self.model().rowCount()): if (row < startRow) or (row > endRow): self.setRowHidden(row, True) else: self.setRowHidden(row, False) string = QtCore.QString(QtCore.QChar(0x25BE)) string.append(' ' + name) self.sorterButton.setText(string) @QtCore.pyqtSlot(QtCore.QModelIndex, QtCore.QModelIndex) def currentChanged(self, current, previous): """Throws a signal emitting the current object when changed""" i = current.row() object = self.model().groupItem().getItem(i) self.objChanged.emit(object) def handleObjReplace(self, index): """Throws a signal when the selected object is used as a replacement""" if QtGui.QApplication.keyboardModifiers() == QtCore.Qt.AltModifier: i = current.row() object = self.model().groupItem().getItem(i) self.objReplaced.emit(object) objChanged = QtCore.pyqtSignal(int) objReplaced = QtCore.pyqtSignal(KPTileObject) class KPMainWindow(QtGui.QMainWindow): def __init__(self): QtGui.QMainWindow.__init__(self) self.editor = KPEditorWidget() self.setCentralWidget(self.editor) self.scene = KPMapScene() self.editor.setScene(self.scene) self.setupMenuBar() self.setupDocks() def setupMenuBar(self): mb = self.menuBar() m = mb.addMenu('&File') # ... def setupDocks(self): self.layerList = KPLayerList() self.layerListDock = QtGui.QDockWidget('Layers') self.layerListDock.setWidget(self.layerList) self.layerList.selectedLayerChanged.connect(self.scene.setCurrentLayer) self.objectSelector = KPObjectSelector() self.updateObjectSelector() self.objectSelectorDock = QtGui.QDockWidget('Objects') self.objectSelectorDock.setWidget(self.objectSelector) self.addDockWidget(Qt.RightDockWidgetArea, self.layerListDock) self.addDockWidget(Qt.RightDockWidgetArea, self.objectSelectorDock) @QtCore.pyqtSlot() def updateObjectSelector(self): try: tileset = KPTileset.loadFromArc('/Users/Tempus/Dropbox/NEWERsmbw/Test3.arc') except IOError: tileset = KPTileset.loadFromArc('/home/me/Dropbox/NEWERsmbw/Test3.arc') self.objectSelector.setModel(tileset.getModel())