1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
|
from PyQt5.QtCore import Qt, QSize, QRect, QPoint, QAbstractTableModel, pyqtSignal, QModelIndex
from PyQt5.QtGui import QBrush, QPen, QColor, QCursor
from PyQt5.QtWidgets import QTableView, QMenu, QGraphicsRectItem, QInputDialog, QStyledItemDelegate, QHeaderView
from collections import defaultdict
class streamTableModel(QAbstractTableModel):
updateEvent = pyqtSignal()
def __init__(self, parent, list, header, *args):
super(streamTableModel, self).__init__(parent, *args)
self.list = list
self.header = header
def columnCount(self, parent=None):
return len(self.list)
def rowCount(self, parent=None):
return len(self.header)
def data(self, index, role):
if role == Qt.TextAlignmentRole:
return Qt.AlignHCenter
if role == Qt.BackgroundColorRole:
return Qt.white
if not index.isValid():
return None
elif role != Qt.DisplayRole:
return None
if index.row() == 0:
return self.list[index.column()].toPlainText()
else:
return self.list[index.column()].values[self.header[index.row()]]
def setData(self, index, value, role):
if not index.isValid():
return False
elif role != Qt.EditRole:
return False
if index.row() == 0:
self.list[index.column()].setPlainText(value)
else:
self.list[index.column()].values[self.header[index.row()]] = value
return True
def insertColumn(self, int=None, item=None):
int = int if int else self.rowCount()+1
self.beginInsertColumns(QModelIndex(), int, int)
self.list.insert(int, item)
self.endInsertColumns()
self.updateEvent.emit()
def insertRow(self, int=None, name="newVal"):
self.beginInsertRows(QModelIndex(), int, int)
self.header.insert(int, name)
self.endInsertRows()
self.updateEvent.emit()
def deleteRow(self, row):
self.beginRemoveRows(QModelIndex(), row, row)
valName = self.header.pop(row)
self.endRemoveRows()
for i in self.list:
i.values.pop(valName)
self.updateEvent.emit()
def headerData(self, col, orientation, role):
if orientation == Qt.Vertical and role == Qt.DisplayRole:
return self.header[col]
return None
def flags(self, index):
return (super(streamTableModel, self).flags(index) | Qt.ItemIsEditable)
class streamTable(QTableView):
def __init__(self, itemLabels=[], canvas=None, parent=None):
super(streamTable, self).__init__(parent=parent)
self.canvas = canvas
for i in itemLabels:
i.nameChanged.connect(self.repaint)
header = ["name", "val1", "val2", "val3", "val4", "val5"]
self.model = streamTableModel(self, itemLabels, header)
self.setShowGrid(False)
self.horizontalHeader().hide()
header = verticalHeader(Qt.Vertical, self)
self.setVerticalHeader(header)
header.labelChangeRequested.connect(self.labelChange)
self.setModel(self.model)
self.borderThickness = defaultdict(lambda: False)
self.model.updateEvent.connect(self.resizeHandler)
self.setItemDelegateForRow(0, drawBorderDelegate(self))
self.borderThickness[0] = True
def mousePressEvent(self, event):
if event.button() == Qt.RightButton:
point = event.pos()
# col = self.getCol(point.x())
index = self.indexAt(point)
menu = QMenu("Context Menu", self)
menu.addAction("Toggle bottom border thickness", lambda x=index.row(): self.changeRowBorder(x))
menu.addAction("Insert Row to bottom", lambda x=index.row(): self.insertRowBottom(x))
menu.addAction("Delete row", lambda x=index.row(): self.model.deleteRow(x))
menu.exec_(self.mapToGlobal(point)+ QPoint(20, 25))
event.accept()
return super(streamTable, self).mousePressEvent(event)
def changeRowBorder(self, row):
if self.borderThickness[row]:
self.borderThickness.pop(row)
self.setItemDelegateForRow(row, QStyledItemDelegate(self))
else:
self.borderThickness[row] = True
self.setItemDelegateForRow(row, drawBorderDelegate(self))
self.verticalHeader().repaint()
def labelChange(self, index):
newName, bool = QInputDialog.getText(self, "Change Property Name", "Enter new name",
text = self.model.header[index])
if bool:
for i in self.model.list:
i.values[newName] = i.values.pop(self.model.header[index])
self.model.header[index] = newName
self.repaint()
def insertRowBottom(self, row):
name, bool = QInputDialog.getText(self, "New Property", "Enter name",
text = "newVal")
if bool:
self.model.insertRow(row + 1, name)
self.repaint()
def resizeHandler(self):
self.resize(self.sizeHint())
def sizeHint(self):
return self.rect().size()
def rect(self):
w = self.verticalHeader().width() + 4
for i in range(self.model.columnCount()):
w += self.columnWidth(i)
h = 0
for i in range(self.model.rowCount()):
h += self.rowHeight(i)
return QRect(0, 0, w, h)
class drawBorderDelegate(QStyledItemDelegate):
# def __init__(self, parent):
# super(drawBorderDelegate, self).__init__(parent)
def paint(self, painter, option, index):
rect = option.rect
painter.drawLine(rect.bottomLeft(), rect.bottomRight())
painter.setPen(QPen(Qt.black, 1, Qt.SolidLine))
super(drawBorderDelegate, self).paint(painter, option, index)
class moveRect(QGraphicsRectItem):
def __init__(self, sideLength = 15, *args):
super(moveRect, self).__init__(-sideLength, -sideLength, sideLength, sideLength)
self.setBrush(Qt.transparent)
self.setPen(QPen(Qt.transparent))
self.setCursor(QCursor(Qt.SizeAllCursor))
self.setAcceptHoverEvents(True)
def hoverEnterEvent(self, event):
self.setBrush(QBrush(QColor(0, 0, 0, 120)))
return super(moveRect, self).hoverEnterEvent(event)
def hoverLeaveEvent(self, event):
self.setBrush(QBrush(QColor(0, 0, 0, 0)))
return super(moveRect, self).hoverLeaveEvent(event)
class verticalHeader(QHeaderView):
labelChangeRequested = pyqtSignal(int)
def mouseDoubleClickEvent(self, event):
index = self.logicalIndexAt(event.pos())
self.labelChangeRequested.emit(index)
return super().mouseDoubleClickEvent(event)
def paintSection(self, painter, option, index):
painter.save()
super(verticalHeader, self).paintSection(painter, option, index)
painter.restore()
if self.parentWidget().borderThickness[index]:
rect = option
painter.drawLine(rect.bottomLeft(), rect.bottomRight())
painter.setPen(QPen(Qt.black, 1, Qt.SolidLine))
|
