Image View
This commit is contained in:
Irony
parent
b81c8fcffe
commit
14a896a22e
1 changed files with 236 additions and 0 deletions
236
QListView/ImageView.py
Normal file
236
QListView/ImageView.py
Normal file
|
|
@ -0,0 +1,236 @@
|
|||
#!/usr/bin/env python
|
||||
# -*- coding: utf-8 -*-
|
||||
|
||||
"""
|
||||
Created on 2021/4/15
|
||||
@author: Irony
|
||||
@site: https://github.com/PyQt5
|
||||
@email: 892768447@qq.com
|
||||
@file: ImageView
|
||||
@description:
|
||||
"""
|
||||
import os
|
||||
|
||||
from PyQt5.QtCore import QPointF, Qt, QRectF, QSizeF
|
||||
from PyQt5.QtGui import QStandardItem, QStandardItemModel, QPainter, QColor, QImage, QPixmap
|
||||
from PyQt5.QtWidgets import QListView, QGraphicsView, QGraphicsPixmapItem, QGraphicsScene
|
||||
|
||||
ScrollPixel = 40
|
||||
|
||||
|
||||
class BigImageView(QGraphicsView):
|
||||
"""图片查看控件"""
|
||||
|
||||
def __init__(self, *args, **kwargs):
|
||||
image = kwargs.pop('image', None)
|
||||
background = kwargs.pop('background', None)
|
||||
super(BigImageView, self).__init__(*args, **kwargs)
|
||||
self.setCursor(Qt.OpenHandCursor)
|
||||
self.setBackground(background)
|
||||
self.setHorizontalScrollBarPolicy(Qt.ScrollBarAlwaysOff)
|
||||
self.setVerticalScrollBarPolicy(Qt.ScrollBarAlwaysOff)
|
||||
self.setRenderHints(QPainter.Antialiasing | QPainter.HighQualityAntialiasing |
|
||||
QPainter.SmoothPixmapTransform)
|
||||
self.setCacheMode(self.CacheBackground)
|
||||
self.setViewportUpdateMode(self.SmartViewportUpdate)
|
||||
self._item = QGraphicsPixmapItem() # 放置图像
|
||||
self._item.setFlags(QGraphicsPixmapItem.ItemIsFocusable |
|
||||
QGraphicsPixmapItem.ItemIsMovable)
|
||||
self._scene = QGraphicsScene(self) # 场景
|
||||
self.setScene(self._scene)
|
||||
self._scene.addItem(self._item)
|
||||
rect = QApplication.instance().desktop().availableGeometry()
|
||||
self.resize(int(rect.width() * 2 / 3), int(rect.height() * 2 / 3))
|
||||
|
||||
self.pixmap = None
|
||||
self._delta = 0.1 # 缩放
|
||||
self.setPixmap(image)
|
||||
|
||||
def setBackground(self, color):
|
||||
"""设置背景颜色
|
||||
:param color: 背景颜色
|
||||
:type color: QColor or str or GlobalColor
|
||||
"""
|
||||
if isinstance(color, QColor):
|
||||
self.setBackgroundBrush(color)
|
||||
elif isinstance(color, (str, Qt.GlobalColor)):
|
||||
color = QColor(color)
|
||||
if color.isValid():
|
||||
self.setBackgroundBrush(color)
|
||||
|
||||
def setPixmap(self, pixmap, fitIn=True):
|
||||
"""加载图片
|
||||
:param pixmap: 图片或者图片路径
|
||||
:param fitIn: 是否适应
|
||||
:type pixmap: QPixmap or QImage or str
|
||||
:type fitIn: bool
|
||||
"""
|
||||
if isinstance(pixmap, QPixmap):
|
||||
self.pixmap = pixmap
|
||||
elif isinstance(pixmap, QImage):
|
||||
self.pixmap = QPixmap.fromImage(pixmap)
|
||||
elif isinstance(pixmap, str) and os.path.isfile(pixmap):
|
||||
self.pixmap = QPixmap(pixmap)
|
||||
else:
|
||||
return
|
||||
self._item.setPixmap(self.pixmap)
|
||||
self._item.update()
|
||||
self.setSceneDims()
|
||||
if fitIn:
|
||||
self.fitInView(QRectF(self._item.pos(), QSizeF(
|
||||
self.pixmap.size())), Qt.KeepAspectRatio)
|
||||
self.update()
|
||||
|
||||
def setSceneDims(self):
|
||||
if not self.pixmap:
|
||||
return
|
||||
self.setSceneRect(QRectF(QPointF(0, 0), QPointF(self.pixmap.width(), self.pixmap.height())))
|
||||
|
||||
def fitInView(self, rect, flags=Qt.IgnoreAspectRatio):
|
||||
"""剧中适应
|
||||
:param rect: 矩形范围
|
||||
:param flags:
|
||||
:return:
|
||||
"""
|
||||
if not self.scene() or rect.isNull():
|
||||
return
|
||||
unity = self.transform().mapRect(QRectF(0, 0, 1, 1))
|
||||
self.scale(1 / unity.width(), 1 / unity.height())
|
||||
viewRect = self.viewport().rect()
|
||||
sceneRect = self.transform().mapRect(rect)
|
||||
x_ratio = viewRect.width() / sceneRect.width()
|
||||
y_ratio = viewRect.height() / sceneRect.height()
|
||||
if flags == Qt.KeepAspectRatio:
|
||||
x_ratio = y_ratio = min(x_ratio, y_ratio)
|
||||
elif flags == Qt.KeepAspectRatioByExpanding:
|
||||
x_ratio = y_ratio = max(x_ratio, y_ratio)
|
||||
self.scale(x_ratio, y_ratio)
|
||||
self.centerOn(rect.center())
|
||||
|
||||
def wheelEvent(self, event):
|
||||
if event.angleDelta().y() > 0:
|
||||
self.zoomIn()
|
||||
else:
|
||||
self.zoomOut()
|
||||
|
||||
def zoomIn(self):
|
||||
"""放大"""
|
||||
self.zoom(1 + self._delta)
|
||||
|
||||
def zoomOut(self):
|
||||
"""缩小"""
|
||||
self.zoom(1 - self._delta)
|
||||
|
||||
def zoom(self, factor):
|
||||
"""缩放
|
||||
:param factor: 缩放的比例因子
|
||||
"""
|
||||
_factor = self.transform().scale(
|
||||
factor, factor).mapRect(QRectF(0, 0, 1, 1)).width()
|
||||
if _factor < 0.07 or _factor > 100:
|
||||
# 防止过大过小
|
||||
return
|
||||
self.scale(factor, factor)
|
||||
|
||||
|
||||
class ImageView(QListView):
|
||||
|
||||
def __init__(self, *args, **kwargs):
|
||||
super(ImageView, self).__init__(*args, **kwargs)
|
||||
self.setFrameShape(self.NoFrame)
|
||||
self.setHorizontalScrollBarPolicy(Qt.ScrollBarAlwaysOff)
|
||||
self.setEditTriggers(self.NoEditTriggers)
|
||||
self.setDropIndicatorShown(True)
|
||||
self.setDragDropMode(self.DragDrop)
|
||||
self.setDefaultDropAction(Qt.IgnoreAction)
|
||||
self.setSelectionMode(self.ExtendedSelection)
|
||||
self.setVerticalScrollMode(self.ScrollPerPixel)
|
||||
self.setHorizontalScrollMode(self.ScrollPerPixel)
|
||||
self.setFlow(self.LeftToRight)
|
||||
self.setWrapping(True)
|
||||
self.setResizeMode(self.Adjust)
|
||||
self.setSpacing(6)
|
||||
self.setViewMode(self.IconMode)
|
||||
self.setWordWrap(True)
|
||||
self.setSelectionRectVisible(True)
|
||||
self.setContextMenuPolicy(Qt.CustomContextMenu)
|
||||
# 解决拖动到顶部或者底部自动滚动
|
||||
self.setAutoScrollMargin(150)
|
||||
self.verticalScrollBar().setSingleStep(ScrollPixel)
|
||||
# 设置model
|
||||
self.dmodel = QStandardItemModel(self)
|
||||
self.setModel(self.dmodel)
|
||||
|
||||
# 大图控件
|
||||
self.bigView = BigImageView(background='#323232')
|
||||
|
||||
def addItem(self, image):
|
||||
if isinstance(image, str):
|
||||
image = QPixmap(image)
|
||||
# 添加一个item
|
||||
item = QStandardItem()
|
||||
# 记录原始图片
|
||||
item.setData(image, Qt.UserRole + 1) # 用于双击的时候取出来
|
||||
# 缩放成小图并显示
|
||||
item.setData(image.scaled(60, 60, Qt.IgnoreAspectRatio, Qt.SmoothTransformation), Qt.DecorationRole)
|
||||
# 添加item到界面中
|
||||
self.dmodel.appendRow(item)
|
||||
|
||||
def count(self):
|
||||
return self.dmodel.rowCount()
|
||||
|
||||
def setCurrentRow(self, row):
|
||||
self.setCurrentIndex(self.dmodel.index(row, 0))
|
||||
|
||||
def currentRow(self):
|
||||
return self.currentIndex().row()
|
||||
|
||||
def updateGeometries(self):
|
||||
# 一次滑动20px
|
||||
super(ImageView, self).updateGeometries()
|
||||
self.verticalScrollBar().setSingleStep(ScrollPixel)
|
||||
|
||||
def closeEvent(self, event):
|
||||
# 关闭预览窗口
|
||||
self.bigView.close()
|
||||
super(ImageView, self).closeEvent(event)
|
||||
|
||||
def wheelEvent(self, event):
|
||||
# 修复滑动bug
|
||||
if self.flow() == QListView.LeftToRight:
|
||||
bar = self.horizontalScrollBar()
|
||||
value = ScrollPixel if event.angleDelta().y() < 0 else (0 - ScrollPixel)
|
||||
bar.setSliderPosition(bar.value() + value)
|
||||
else:
|
||||
super(ImageView, self).wheelEvent(event)
|
||||
|
||||
def mouseDoubleClickEvent(self, event):
|
||||
# 列表双击,如果有item则进入item处理流程,否则调用打开图片功能
|
||||
index = self.indexAt(event.pos())
|
||||
if index and index.isValid():
|
||||
item = self.dmodel.itemFromIndex(index)
|
||||
if item:
|
||||
# 取出原图用来新窗口显示
|
||||
image = item.data(Qt.UserRole + 1)
|
||||
self.bigView.setPixmap(image)
|
||||
self.bigView.show()
|
||||
return
|
||||
super(ImageView, self).mouseDoubleClickEvent(event)
|
||||
|
||||
|
||||
if __name__ == '__main__':
|
||||
import sys
|
||||
import cgitb
|
||||
|
||||
cgitb.enable(format='text')
|
||||
from PyQt5.QtWidgets import QApplication
|
||||
|
||||
app = QApplication(sys.argv)
|
||||
w = ImageView()
|
||||
w.show()
|
||||
|
||||
# 添加模拟图片
|
||||
for i in range(3):
|
||||
for name in os.listdir('ScreenShot'):
|
||||
w.addItem(os.path.join('ScreenShot', name))
|
||||
sys.exit(app.exec_())
|
||||
Loading…
Reference in a new issue