Растягиваемый QLinearGradient как BackgroundRole для ячеек QTableView с изменяемым размером в PyQt5? ⇐ Python

[Anonymous]

Рассмотрим этот пример, в котором я хочу применить «вертикальный» фон ко всем ячейкам в третьем столбце таблицы:

Код: Выделить всё

import sys
from PyQt5 import QtCore, QtGui, QtWidgets
from PyQt5.QtCore import (Qt, QPointF)
from PyQt5.QtGui import (QColor, QGradient, QLinearGradient, QBrush)
# starting point from https://www.pythonguis.com/tutorials/qtableview-modelviews-numpy-pandas/

class TableModel(QtCore.QAbstractTableModel):
def __init__(self, data):
super(TableModel, self).__init__()
self._data = data
self.bg_col1 = QColor("#A3A3FF")
self.bg_col2 = QColor("#FFFFA3")
self.bg_grad = QLinearGradient(QPointF(0.0, 0.0), QPointF(0.0, 1.0)) # setcolor 0 on top, 1 on bottom
self.bg_grad.setCoordinateMode(QGradient.ObjectMode) #StretchToDeviceMode) #ObjectBoundingMode)
self.bg_grad.setSpread(QGradient.PadSpread) #RepeatSpread) # PadSpread (default)
self.bg_grad.setColorAt(0.0, self.bg_col1)
self.bg_grad.setColorAt(1.0, self.bg_col2)
self.bg_grad_brush = QBrush(self.bg_grad)
#
def data(self, index, role):
if role == Qt.DisplayRole:
# See below for the nested-list data structure.
# .row() indexes into the outer list,
# .column() indexes into the sub-list
return self._data[index.row()][index.column()]
if role == Qt.BackgroundRole:
if index.column() == 2:
return self.bg_grad_brush
#
def rowCount(self, index):
# The length of the outer list.
return len(self._data)
#
def columnCount(self, index):
# The following takes the first sub-list, and returns
# the length (only works if all rows are an equal length)
return len(self._data[0])

class MainWindow(QtWidgets.QMainWindow):
def __init__(self):
super().__init__()
self.table = QtWidgets.QTableView()
data = [
[4, 9, 2],
[1, 0, 0],
[3, 5, 0],
[3, 3, 2],
[7, 8, 9],
]
self.model = TableModel(data)
self.table.setModel(self.model)
self.setCentralWidget(self.table)

app=QtWidgets.QApplication(sys.argv)
window=MainWindow()
window.show()
app.exec_()

Когда я запускаю этот код, сначала я получаю этот рисунок:
[img]https://i .sstatic.net/KnE9jt7G.png[/img]

Только ячейка в первой строке имеет градиент, как я и предполагал; но все остальные ниже отображаются плоскими цветами.
Я могу заметить, что если я изменю высоту второй строки, в соответствующей ячейке появится градиент, но это как-то неправильно - не растягивается пропорционально границам ячейки:


Если я впоследствии изменю высоту строки 1, то увижу, что ячейка с фоном в этой строке «растягивает» градиент фона в соответствии с моими ожиданиями, а тем временем ячейка под ней потеряла градиент, который она показывала ранее:

< /p>
Что я могу сделать, чтобы все ячейки с градиентной BackgroundRole вели себя так же, как ячейка в первой строке (отображается полный градиент и растягивается в соответствии с размером ячейки), и не «теряют " их рендеринг градиента, если другая ячейка меняет размер?

Спасибо @musicamante; помимо того, что изложено в ответе, также необходимо использовать QGradient.ObjectBoundingMode вместо QGradient.ObjectMode - вот исправленный код:

Код: Выделить всё

import sys
from PyQt5 import QtCore, QtGui, QtWidgets
from PyQt5.QtCore import (Qt, QPointF)
from PyQt5.QtGui import (QColor, QGradient, QLinearGradient, QBrush, QTransform)
from PyQt5.QtWidgets import (QStyledItemDelegate,)
# starting point from https://www.pythonguis.com/tutorials/qtableview-modelviews-numpy-pandas/

class GradientDelegate(QStyledItemDelegate):
def initStyleOption(self, opt, index):
super().initStyleOption(opt, index)
if (
isinstance(opt.backgroundBrush, QBrush)
and opt.backgroundBrush.gradient()
):
print(f"{opt.rect=}")
opt.backgroundBrush.setTransform(
QTransform.fromTranslate(-opt.rect.x(), -opt.rect.y()))

class TableModel(QtCore.QAbstractTableModel):
def __init__(self, data):
super(TableModel, self).__init__()
self._data = data
self.bg_col1 = QColor("#A3A3FF")
self.bg_col2 = QColor("#FFFFA3")
self.bg_grad = QLinearGradient(QPointF(0.0, 0.0), QPointF(0.0, 1.0)) # setcolor 0 on top, 1 on bottom
self.bg_grad.setCoordinateMode(QGradient.ObjectBoundingMode) #StretchToDeviceMode) #ObjectBoundingMode) # ObjectMode
self.bg_grad.setSpread(QGradient.PadSpread) #RepeatSpread) # PadSpread (default)
self.bg_grad.setColorAt(0.0, self.bg_col1)
self.bg_grad.setColorAt(1.0, self.bg_col2)
self.bg_grad_brush = QBrush(self.bg_grad)
#
def data(self, index, role):
if role == Qt.DisplayRole:
# See below for the nested-list data structure.
# .row() indexes into the outer list,
# .column() indexes into the sub-list
return self._data[index.row()][index.column()]
if role == Qt.BackgroundRole:
if index.column() == 2:
return QBrush(self.bg_grad) #self.bg_grad_brush
#
def rowCount(self, index):
# The length of the outer list.
return len(self._data)
#
def columnCount(self, index):
# The following takes the first sub-list, and returns
# the length (only works if all rows are an equal length)
return len(self._data[0])

class MainWindow(QtWidgets.QMainWindow):
def __init__(self):
super().__init__()
self.table = QtWidgets.QTableView()
data = [
[4, 9, 2],
[1, 0, 0],
[3, 5, 0],
[3, 3, 2],
[7, 8, 9],
]
self.model = TableModel(data)
self.table.setModel(self.model)
self.table.setItemDelegateForColumn(2, GradientDelegate(self.table))
self.setCentralWidget(self.table)

app=QtWidgets.QApplication(sys.argv)
window=MainWindow()
window.show()
app.exec_()

... и вот результаты - именно то, что я хотел получить для начала:



Подробнее здесь: https://stackoverflow.com/questions/791 ... w-cells-in