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
|
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
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 not index.isValid():
return None
elif role != Qt.DisplayRole:
return None
if index.row() == 0:
return self.list[index.column()][index.row()].toPlainText()
else:
return self.list[index.column()][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()][index.row()].setPlainText(value)
else:
self.list[index.column()][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] + [0 for _ in range(self.rowCount()-1)])
self.endInsertColumns()
self.updateEvent.emit()
def insertRow(self, int=None, name="newVal"):
self.beginInsertRows(QModelIndex(), int, int)
for item in self.list:
item.insert(int, 0)
self.header.insert(int, name)
self.endInsertRows()
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
list = []
for i, item in enumerate(itemLabels):
list.append([item] + [0 for _ in range(5)])
header = ["name", "val1", "val2", "val3", "val4", "val5"]
self.model = streamTableModel(self, list, header)
self.setShowGrid(False)
self.horizontalHeader().hide()
self.setModel(self.model)
self.borderThickness = defaultdict(lambda: 1)
self.model.updateEvent.connect(self.refresh)
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("Change bottom border thickness", lambda x=index.row(): self.changeRowBorder(x))
menu.addAction("Insert Row to bottom", lambda x=index.row(): self.insertRowBottom(x))
menu.exec_(self.mapToGlobal(point)+ QPoint(20, 25))
event.accept()
return super(streamTable, self).mousePressEvent(event)
def changeRowBorder(self, row):
newWidth, bool = QInputDialog.getInt(self, "Change Horizontal Border Width", "Enter new Width in pixels", self.borderThickness[row], 0, 10, step=1)
if bool:
self.setItemDelegateForRow(row, drawBorderDelegate(self, newWidth, True))
def changeColBorder(self, col):
newWidth, bool = QInputDialog.getInt(self, "Change Vertical Border Width", "Enter new Width in pixels", self.borderThickness[-col], 0, 10, step=1)
if bool:
self.setItemDelegateForColumn(col, drawBorderDelegate(self, newWidth, False))
def insertRowBottom(self, row):
self.model.insertRow(row + 1)
def refresh(self):
self.resizeHandler()
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 = 4
for i in range(self.model.rowCount()):
h += self.rowHeight(i)
return QRect(0, 0, w, h)
class drawBorderDelegate(QStyledItemDelegate):
def __init__(self, parent, bool1, bool2 = True):
super(drawBorderDelegate, self).__init__(parent)
self.horizontal = bool1
self.bottom = bool2
def paint(self, painter, option, index):
rect = option.rect
if self.horizontal:
painter.drawLine(rect.bottomLeft(), rect.bottomRight())
else:
painter.drawLine(rect.topRight(), rect.bottomRight())
painter.setPen(QPen(Qt.black, 2, 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)
|
