Неожиданное заполнение pyqt в ячейке qtableview

Question

Я пытаюсь создать пользовательский класс TableModel для QTableView. Ячейки, содержащие 1 в качестве данных, должны иметь красный контур. Выделение выполняется путем возврата растрового изображения (с красными полями и текстом, нарисованным сверху) из TableModel вместо возврата простой строки.

Проблема в неожиданном заполнении растрового изображения, которое я возвращаю как DecorationRole. Я проверил, правильно ли отображается растровое изображение (и оно на самом деле имеет размер 21x21 пикселей с хорошо выполненным контуром, без отступов, как и планировалось) прямо перед линией return pixmap.

Вот правильное нарисованное растровое изображение, которое было сохранено непосредственно перед return из TableModel:

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

Есть идеи как это исправить? Спасибо.

Вот образец моего TableModel:

class TableModel(QtCore.QAbstractTableModel):
def __init__(self, topology=None):
    super().__init__()
    ...
    # Hardcode cell size and path to rectangle image
    self.cell_width, self.cell_height = 21, 21
    self.fpath_red_rect = './path/to/red_rect.png'

def rowCount(self, parent=QtCore.QModelIndex()):
    return self.data.shape[0]

def columnCount(self, parent=QtCore.QModelIndex()):
    return self.data.shape[1]

def headerData(self, section, orientation, role=QtCore.Qt.DisplayRole):
    ...

def size(self):
    return QtCore.QSize((self.columnCount() + 1) * self.cell_width,
                        (self.rowCount() + 1) * self.cell_height)

def data(self, index, role=QtCore.Qt.DisplayRole):
    if not index.isValid():
        return QtCore.QVariant()

    i = index.row()
    j = index.column()

    if role == QtCore.Qt.DisplayRole:
        if self.data[i, j] == 0:      # empty
            return ''
        elif self.data[i, j] == 1:    # cell with red rectangle
            # the text will be drawn on pixmap manually later
            return None
        else:
            return '{0}'.format(self.data[i, j])    # display default data

    if role == QtCore.Qt.DecorationRole:
        # Create pixmap, draw the rectangle on it and then draw text on top
        pixmap = QtGui.QPixmap(self.cell_width, self.cell_height)
        image = QtGui.QImage(self.fpath_red_rect).scaled(self.cell_width, self.cell_height)
        painter = QtGui.QPainter(pixmap)
        painter.drawImage(pixmap.rect().topLeft(), image)
        painter.drawText(pixmap.rect(), QtCore.Qt.AlignCenter, '{0}'.format(self.data[i, j]))
        painter.end()

        # If we save the pixmap to PNG image here (see the link above),
        # we get the expected 21 x 21 px image, with nice
        # and properly drawn rectangle and centered text.
        # But something goes wrong after returning

        return pixmap

    if role == QtCore.Qt.BackgroundRole:
        return QtGui.QBrush(self.getQtColor(self.data[i, j]))

    if role == QtCore.Qt.TextAlignmentRole:
        return QtCore.Qt.AlignCenter

    return QtCore.QVariant()

Answer 1

DecorationRole используется для рисования иконки по той причине, что вы наблюдаете смещение, в вашем случае вы не должны использовать эту роль, кроме того, задача рисования не должна выполняться в модели, поскольку он должен только предоставить данные, если вы хотите изменить чертеж, лучше использовать делегата, как показано ниже:

import sys

import numpy as np

from PyQt5 import QtCore, QtGui, QtWidgets

ValueRole = QtCore.Qt.UserRole + 1
max_val = 4

colors = [QtGui.QColor(*np.random.randint(255, size=3)) for i in range(max_val)]


class TableModel(QtCore.QAbstractTableModel):
    def __init__(self, parent=None):
        QtCore.QAbstractTableModel.__init__(self, parent)
        self.data = np.random.randint(max_val, size=(10, 10))

    def rowCount(self, parent=QtCore.QModelIndex()):
        return self.data.shape[0]

    def columnCount(self, parent=QtCore.QModelIndex()):
        return self.data.shape[1]

    def data(self, index, role=QtCore.Qt.DisplayRole):
        if not index.isValid():
            return QtCore.QVariant()

        i = index.row()
        j = index.column()
        val = self.data[i, j]

        if role == QtCore.Qt.DisplayRole:
            return str(val)

        elif role == QtCore.Qt.TextAlignmentRole:
            return QtCore.Qt.AlignCenter

        elif role == QtCore.Qt.BackgroundRole:
            return colors[val]

        if role == ValueRole:
            return val

        return QtCore.QVariant()

class Delegate(QtWidgets.QStyledItemDelegate):
    def paint(self, painter, option, index):
        QtWidgets.QStyledItemDelegate.paint(self, painter, option, index)
        if index.data(ValueRole) == 1:
            painter.save()
            pen = painter.pen()
            pen.setColor(QtCore.Qt.red)
            painter.setPen(pen)
            r = QtCore.QRect(option.rect)
            r.adjust(0, 0, -pen.width(), -pen.width())
            painter.drawRect(r)
            painter.restore()

if __name__ == '__main__':
    app = QtWidgets.QApplication(sys.argv)
    w = QtWidgets.QTableView()
    w.setItemDelegate(Delegate(w))
    model = TableModel()
    w.setModel(model)

    w.verticalHeader().setSectionResizeMode(QtWidgets.QHeaderView.Fixed)
    w.horizontalHeader().setSectionResizeMode(QtWidgets.QHeaderView.Fixed)

    for i in range(model.rowCount()):
        w.verticalHeader().resizeSection(i, 21)

    for j in range(model.columnCount()):
        w.horizontalHeader().resizeSection(j, 21)
    w.show()
    sys.exit(app.exec_())