diff options
Diffstat (limited to '')
| -rw-r--r-- | src/worldeditor.py | 163 |
1 files changed, 159 insertions, 4 deletions
diff --git a/src/worldeditor.py b/src/worldeditor.py index a2262ae..35e42a6 100644 --- a/src/worldeditor.py +++ b/src/worldeditor.py @@ -1,17 +1,172 @@ from common import * +import re + +def editableColourStr(array): + return '#%02X%02X%02X (%d)' % tuple(array) + +NICE_STR_RE = re.compile('^#([0-9a-fA-F]{2})([0-9a-fA-F]{2})([0-9a-fA-F]{2})\s*(?:\((\d+)\))?$') +def colourFromNiceStr(thing): + match = NICE_STR_RE.match(thing) + try: + if match: + r,g,b,a = match.groups() + return (int(r,16), int(g,16), int(b,16), int(a) if a != None else 255) + except: + pass + return None + +class KPWorldTableModel(QtCore.QAbstractTableModel): + FIELDS = ('Name', 'Track ID', + 'FS Text 1', 'FS Text 2', + 'FS Hint 1', 'FS Hint 2', + 'HUD Text 1', 'HUD Text 2', + 'HUD Hue', 'HUD Saturation', 'HUD Lightness') + + def __init__(self, kpmap, parent=None): + QtCore.QAbstractTableModel.__init__(self, parent) + + self.currentMap = kpmap + self.worlds = kpmap.worlds + + def columnCount(self, parent): + return len(self.FIELDS) + def headerData(self, section, orientation, role): + if orientation == Qt.Horizontal: + if role == Qt.DisplayRole: + return self.FIELDS[section] + else: + if role == Qt.DisplayRole: + return str(self.worlds[section].uniqueKey) + + return QtCore.QVariant() + + def rowCount(self, parent): + if parent.isValid(): + return 0 + else: + return len(self.worlds) + + def data(self, index, role): + if index.isValid(): + entry = self.worlds[index.row()] + col = index.column() + + if role == Qt.DisplayRole or role == Qt.EditRole: + if col == 0: + return entry.name + elif col == 1: + return entry.musicTrackID + elif col == 2 or col == 3: + return editableColourStr(entry.fsTextColours[col - 2]) + elif col == 4 or col == 5: + return editableColourStr(entry.fsHintColours[col - 4]) + elif col == 6 or col == 7: + return editableColourStr(entry.hudTextColours[col - 6]) + elif col >= 8 and col <= 10: + return entry.hudHintTransform[col - 8] + + if role == Qt.DecorationRole: + if col == 2 or col == 3: + return QtGui.QColor(*entry.fsTextColours[col - 2]) + elif col == 4 or col == 5: + return QtGui.QColor(*entry.fsHintColours[col - 4]) + elif col == 6 or col == 7: + return QtGui.QColor(*entry.hudTextColours[col - 6]) + + return QtCore.QVariant() + + def flags(self, index): + return Qt.ItemIsEnabled | Qt.ItemIsSelectable | Qt.ItemIsEditable + + def setData(self, index, value, role): + if index.isValid(): + if role == Qt.EditRole: + success = False + + entry = self.worlds[index.row()] + col = index.column() + + if col == 0: + entry.name = str(value.toString()) + success = True + elif col == 1: + v,ok = value.toInt() + if ok: + entry.musicTrackID = v + success = True + elif col >= 2 and col <= 7: + newCol = colourFromNiceStr(str(value.toString())) + if newCol: + success = True + if col == 2: + entry.fsTextColours = (newCol, entry.fsTextColours[1]) + elif col == 3: + entry.fsTextColours = (entry.fsTextColours[0], newCol) + elif col == 4: + entry.fsHintColours = (newCol, entry.fsHintColours[1]) + elif col == 5: + entry.fsHintColours = (entry.fsHintColours[0], newCol) + elif col == 6: + entry.hudTextColours = (newCol, entry.hudTextColours[1]) + elif col == 7: + entry.hudTextColours = (entry.hudTextColours[0], newCol) + elif col >= 8 and col <= 10: + v,ok = value.toInt() + if ok: + new = list(entry.hudHintTransform) + new[col - 8] = v + entry.hudHintTransform = new + success = True + + if success: + self.dataChanged.emit(index, index) + return success + + return False + + + def addEntryToEnd(self): + self.beginInsertRows(QtCore.QModelIndex(), len(self.worlds), len(self.worlds)) + entry = KPWorldDef() + entry.uniqueKey = self.currentMap.allocateWorldDefKey() + self.worlds.append(entry) + self.endInsertRows() + + def removeRows(self, row, count, parent): + if not parent.isValid(): + if row >= 0 and (row + count) <= len(self.worlds): + self.beginRemoveRows(parent, row, row+count-1) + for i in xrange(count): + del self.worlds[row] + self.endRemoveRows() + + class KPWorldEditor(QtGui.QWidget): - def __init__(self, parent=None): - QtGui.QWidget.__init__(self, parent) + def __init__(self, kpmap, parent=None): + QtGui.QWidget.__init__(self, parent, Qt.Window) self.setWindowTitle('World Editor') self.dataView = QtGui.QTableView(self) + self.addButton = QtGui.QPushButton('Add', self) - self.removedButton = QtGui.QPushButton('Remove', self) + self.removeButton = QtGui.QPushButton('Remove', self) layout = QtGui.QGridLayout(self) layout.addWidget(self.dataView, 0, 0, 1, 2) layout.addWidget(self.addButton, 1, 0, 1, 1) layout.addWidget(self.removeButton, 1, 1, 1, 1) - self.currentWorlds = None + self.model = KPWorldTableModel(kpmap, self) + self.dataView.setModel(self.model) + + self.addButton.clicked.connect(self.model.addEntryToEnd) + self.removeButton.clicked.connect(self.removeCurrentEntry) + + def removeCurrentEntry(self): + what = self.dataView.selectionModel().currentIndex() + if what.isValid(): + what = what.row() + key = self.model.worlds[what].uniqueKey + self.model.removeRows(what, 1, QtCore.QModelIndex()) + |