mirror of
https://github.com/olivierkes/manuskript.git
synced 2024-09-30 01:07:24 +13:00
141 lines
4.4 KiB
Python
141 lines
4.4 KiB
Python
|
#!/usr/bin/env python
|
||
|
#--!-- coding: utf8 --!--
|
||
|
|
||
|
from qt import *
|
||
|
from enums import *
|
||
|
|
||
|
from enum import Enum
|
||
|
from lxml import etree as ET
|
||
|
|
||
|
class plotsProxyModel(QSortFilterProxyModel):
|
||
|
|
||
|
newStatuses = pyqtSignal()
|
||
|
|
||
|
def __init__(self, parent=None):
|
||
|
QSortFilterProxyModel.__init__(self, parent)
|
||
|
|
||
|
#self.rootItem = QStandardItem()
|
||
|
self.p1 = QStandardItem(self.tr("Main"))
|
||
|
self.p2 = QStandardItem(self.tr("Secundary"))
|
||
|
self.p3 = QStandardItem(self.tr("Minors"))
|
||
|
|
||
|
self._cats = [
|
||
|
self.p1,
|
||
|
self.p2,
|
||
|
self.p3
|
||
|
]
|
||
|
|
||
|
def mapFromSource(self, sourceIndex):
|
||
|
if not sourceIndex.isValid():
|
||
|
return QModelIndex()
|
||
|
|
||
|
row = self._map.index(sourceIndex.row())
|
||
|
#item = sourceIndex.internalPointer()
|
||
|
item = self.sourceModel().itemFromIndex(sourceIndex)
|
||
|
|
||
|
return self.createIndex(row, sourceIndex.column(), item)
|
||
|
|
||
|
def flags(self, index):
|
||
|
if not index.isValid():
|
||
|
return Qt.NoItemFlags
|
||
|
|
||
|
if index.isValid() and not self.mapToSource(index).isValid():
|
||
|
return Qt.NoItemFlags#Qt.ItemIsEnabled
|
||
|
else:
|
||
|
return Qt.ItemIsEnabled | Qt.ItemIsSelectable
|
||
|
|
||
|
def mapToSource(self, proxyIndex):
|
||
|
if not proxyIndex.isValid():
|
||
|
return QModelIndex()
|
||
|
|
||
|
row = self._map[proxyIndex.row()]
|
||
|
|
||
|
if type(row) != int:
|
||
|
return QModelIndex()
|
||
|
|
||
|
#item = proxyIndex.internalPointer()
|
||
|
item = self.sourceModel().item(row, proxyIndex.column())
|
||
|
|
||
|
return self.sourceModel().indexFromItem(item)
|
||
|
|
||
|
def setSourceModel(self, model):
|
||
|
QSortFilterProxyModel.setSourceModel(self, model)
|
||
|
self.sourceModel().dataChanged.connect(self.mapModelMaybe)
|
||
|
self.sourceModel().rowsInserted.connect(self.mapModel)
|
||
|
self.sourceModel().rowsRemoved.connect(self.mapModel)
|
||
|
self.sourceModel().rowsMoved.connect(self.mapModel)
|
||
|
|
||
|
self.mapModel()
|
||
|
|
||
|
def mapModelMaybe(self, topLeft, bottomRight):
|
||
|
if topLeft.column() <= Plot.importance.value <= bottomRight.column():
|
||
|
self.mapModel()
|
||
|
|
||
|
def mapModel(self):
|
||
|
self.beginResetModel()
|
||
|
src = self.sourceModel()
|
||
|
|
||
|
self._map = []
|
||
|
|
||
|
for i in range(len(self._cats)):
|
||
|
self._map.append(self._cats[i])
|
||
|
|
||
|
for p in range(src.rowCount()):
|
||
|
item = src.item(p, Plot.importance.value)
|
||
|
|
||
|
if item:
|
||
|
imp = int(item.text())
|
||
|
else:
|
||
|
imp = 0
|
||
|
|
||
|
if 2-imp == i:
|
||
|
self._map.append(p)
|
||
|
|
||
|
self.endResetModel()
|
||
|
|
||
|
|
||
|
def data(self, index, role=Qt.DisplayRole):
|
||
|
|
||
|
if index.isValid() and not self.mapToSource(index).isValid():
|
||
|
row = index.row()
|
||
|
|
||
|
if role == Qt.DisplayRole:
|
||
|
return self._map[row].text()
|
||
|
|
||
|
elif role == Qt.ForegroundRole:
|
||
|
return QBrush(Qt.darkBlue)
|
||
|
elif role == Qt.BackgroundRole:
|
||
|
return QBrush(QColor(Qt.blue).lighter(190))
|
||
|
elif role == Qt.TextAlignmentRole:
|
||
|
return Qt.AlignCenter
|
||
|
elif role == Qt.FontRole:
|
||
|
f = QFont()
|
||
|
#f.setPointSize(f.pointSize() + 1)
|
||
|
f.setWeight(QFont.Bold)
|
||
|
return f
|
||
|
else:
|
||
|
#FIXME: sometimes, the name of the character is not displayed
|
||
|
return self.sourceModel().data(self.mapToSource(index), role)
|
||
|
|
||
|
def index(self, row, column, parent):
|
||
|
|
||
|
i = self._map[row]
|
||
|
|
||
|
if type(i) != int:
|
||
|
return self.createIndex(row, column, i)
|
||
|
|
||
|
else:
|
||
|
return self.mapFromSource(self.sourceModel().index(i, column, QModelIndex()))
|
||
|
|
||
|
def parent(self, index=QModelIndex()):
|
||
|
return QModelIndex()
|
||
|
|
||
|
def rowCount(self, parent=QModelIndex()):
|
||
|
return len(self._map)
|
||
|
|
||
|
def columnCount(self, parent=QModelIndex()):
|
||
|
return self.sourceModel().columnCount(QModelIndex())
|
||
|
|
||
|
def item(self, row, col, parent=QModelIndex()):
|
||
|
idx = self.mapToSource(self.index(row, col, parent))
|
||
|
return self.sourceModel().item(idx.row(), idx.column())
|