def __paint_image_array(
self,
raw_data,
painter: "QPainter",
option: "QStyleOptionViewItem",
index: "QModelIndex",
):
"""Paints a pixmap centered on the cell.
Generated pixmaps are saved on cache by the name "dataset_row_col"
"""
# Load Qt pixap from array
pix = QPixmap()
pix_name = str(id(raw_data))
if not QPixmapCache.find(pix_name, pix):
# Load pix from raw array
pix = QPixmap.fromImage(qimage2ndarray.array2qimage(raw_data))
# Save pix on cache
QPixmapCache.insert(pix_name, pix)
pix = pix.scaled(option.rect.width(), option.rect.height(),
Qt.KeepAspectRatio)
# Calculate central position
x_coord = option.rect.center().x() - pix.width() / 2
y_coord = option.rect.center().y() - pix.height() / 2
draw_rect = QRect(x_coord, y_coord, pix.width(), pix.height())
# Draw pixm
painter.drawPixmap(draw_rect, pix)
def testWithKey(self):
pm1 = QPixmap()
ok = QPixmapCache.find(QPixmapCache.Key(), pm1)
self.assertFalse(ok)
self.assertEqual(QPixmapCache.find(QPixmapCache.Key()), None)
pm2 = QPixmap()
key = QPixmapCache.insert(pm2)
pm3 = QPixmap()
ok = QPixmapCache.find(key, pm3)
self.assertTrue(ok)
self.assertEqual(QPixmapCache.find(key).toImage().bits(), pm3.toImage().bits())
def testWithKey(self):
pm1 = QPixmap()
ok = QPixmapCache.find(QPixmapCache.Key(), pm1)
self.assertFalse(ok)
self.assertEqual(QPixmapCache.find(QPixmapCache.Key()), None)
pm2 = QPixmap()
key = QPixmapCache.insert(pm2)
pm3 = QPixmap()
ok = QPixmapCache.find(key, pm3)
self.assertTrue(ok)
self.assertEqual(QPixmapCache.find(key).toImage().bits(), pm3.toImage().bits())
def testWithString(self):
pm1 = QPixmap()
ok = QPixmapCache.find('img', pm1)
self.assertFalse(ok)
self.assertEqual(QPixmapCache.find('img'), None)
pm2 = QPixmap()
ok = QPixmapCache.insert('img', pm2)
self.assertTrue(ok)
pm3 = QPixmap()
ok = QPixmapCache.find('img', pm3)
self.assertTrue(ok)
b1 = QPixmapCache.find('img').toImage().bits()
b2 = pm3.toImage().bits()
self.assertEqual(QPixmapCache.find('img').toImage().bits(), pm3.toImage().bits())
def testWithString(self):
pm1 = QPixmap()
ok = QPixmapCache.find('img', pm1)
self.assertFalse(ok)
self.assertEqual(QPixmapCache.find('img'), None)
pm2 = QPixmap()
ok = QPixmapCache.insert('img', pm2)
self.assertTrue(ok)
pm3 = QPixmap()
ok = QPixmapCache.find('img', pm3)
self.assertTrue(ok)
b1 = QPixmapCache.find('img').toImage().bits()
b2 = pm3.toImage().bits()
self.assertEqual(QPixmapCache.find('img').toImage().bits(), pm3.toImage().bits())
label.setPixmap(pix)
label.show()
def slotPrint(self):
printer = QPrinter(QPrinter.HighResolution)
pd = QPrintDialog(printer, self)
if pd.exec_() == QDialog.Accepted:
r = self.view.graphicsView().mapToScene(
self.view.graphicsView().viewport().rect()).boundingRect()
self.view.print(printer, r.left(), r.right())
def slotHeaderMenu(self, pt):
menu = QMenu()
menu.addAction("This")
menu.addAction("is")
menu.addAction("just")
menu.addAction("a")
menu.addAction("test")
menu.exec_(pt)
if __name__ == '__main__':
app = QApplication(sys.argv)
QPixmapCache.setCacheLimit(30 * 1024)
w = MyWidget()
w.show()
sys.exit(app.exec_())
def clear(self):
QPixmapCache.clear()
self._cached_data.clear()
self._cached_data = [[], []]
self.modelReset.emit()