diff options
| -rw-r--r-- | src/main/python/utils/streamTable.py | 43 |
1 files changed, 35 insertions, 8 deletions
diff --git a/src/main/python/utils/streamTable.py b/src/main/python/utils/streamTable.py index c9faef7..01311e8 100644 --- a/src/main/python/utils/streamTable.py +++ b/src/main/python/utils/streamTable.py @@ -1,12 +1,15 @@ from PyQt5.QtCore import Qt, QSize, QRectF, QPoint -from PyQt5.QtGui import QBrush, QColor -from PyQt5.QtWidgets import QTableWidget, QMenu, QGraphicsRectItem +from PyQt5.QtGui import QBrush, QPen, QColor +from PyQt5.QtWidgets import QTableWidget, QMenu, QGraphicsRectItem, QInputDialog, QStyledItemDelegate + +from collections import defaultdict class streamTable(QTableWidget): def __init__(self, int, int_, canvas=None, parent=None): super(streamTable, self).__init__(int, int_, parent=parent) self.canvas = canvas + self.borderThickness = defaultdict(lambda: 1) def mousePressEvent(self, event): if event.button() == Qt.RightButton: @@ -14,18 +17,27 @@ class streamTable(QTableWidget): col = self.getCol(point.x()) row = self.getRow(point.y()) menu = QMenu("Context Menu", self) + menu.addAction("Change bottom border thickness", lambda x=row: self.changeRowBorder(x)) + menu.addAction("Change right border thickness", lambda x=col: self.changeColBorder(x)) menu.addAction("Insert Column to right", lambda x=col: self.insertColRight(col)) - menu.addAction("Insert Row to bottom", lambda x=point: self.insertRowBottom(row)) + menu.addAction("Insert Row to bottom", lambda x=row: self.insertRowBottom(row)) menu.exec_(self.mapToGlobal(point)+ QPoint(20, 25)) event.accept() return super(streamTable, self).mousePressEvent(event) - # def mouseDoubleClickEvent(self): - # return super(streamTable, self).mouseDoubleClickEvent() + 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 getRow(self, point): h = self.horizontalHeader().height() - i = 0 + i = -1 while (h < point): h += self.rowHeight(i) i += 1 @@ -33,7 +45,7 @@ class streamTable(QTableWidget): def getCol(self, point): w = self.verticalHeader().width() - i = 0 + i = -1 while (w < point): w += self.columnWidth(i) i += 1 @@ -49,7 +61,6 @@ class streamTable(QTableWidget): def resizeHandler(self): self.resize(self.sizeHint()) - # self.canvas.streamTableRect.setRect(streamTable.rect().adjusted(-10, -10, 10, 10)) def sizeHint(self): w = self.verticalHeader().width() + 4 @@ -72,6 +83,22 @@ class streamTable(QTableWidget): def cellEntered(self, row, column): print(row, column) +class drawBorderDelegate(QStyledItemDelegate): + def __init__(self, parent, width, bool1, bool2 = True): + super(drawBorderDelegate, self).__init__(parent) + self.horizontal = bool1 + self.bottom = bool2 + self.width = width + + 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, self.width, Qt.SolidLine)) + super(drawBorderDelegate, self).paint(painter, option, index) + class moveRect(QGraphicsRectItem): def __init__(self, *args): super(moveRect, self).__init__(*args) |