class QtImg(QtWidgets.QWidget, Ui_Img):
def __init__(self):
super(self.__class__, self).__init__()
Ui_Img.__init__(self)
self.setupUi(self)
self.bookId = ""
self.epsId = 0
self.curIndex = 0
self.setWindowTitle("图片查看")
self.setWindowModality(QtCore.Qt.ApplicationModal)
self.resize(800, 900)
self.radioButton.setChecked(True)
self.index = 1
self.comboBox.setCurrentIndex(self.index)
# self.setWindowFlags(Qt.FramelessWindowHint)
self.graphicsView.setFrameStyle(QFrame.NoFrame)
self.graphicsView.setObjectName("graphicsView")
self.graphicsView.setBackgroundBrush(QColor(Qt.white))
self.graphicsView.setCursor(Qt.OpenHandCursor)
self.graphicsView.setHorizontalScrollBarPolicy(Qt.ScrollBarAlwaysOff)
self.graphicsView.setVerticalScrollBarPolicy(Qt.ScrollBarAlwaysOff)
self.graphicsView.setRenderHints(QPainter.Antialiasing | QPainter.HighQualityAntialiasing |
QPainter.SmoothPixmapTransform)
self.graphicsView.setCacheMode(self.graphicsView.CacheBackground)
self.graphicsView.setViewportUpdateMode(self.graphicsView.SmartViewportUpdate)
self.graphicsItem = QGraphicsPixmapItem()
self.graphicsItem.setFlags(QGraphicsPixmapItem.ItemIsFocusable |
QGraphicsPixmapItem.ItemIsMovable)
self.setContextMenuPolicy(Qt.CustomContextMenu)
self.customContextMenuRequested.connect(self.CopyPicture)
self.graphicsScene = QGraphicsScene(self) # 场景
self.graphicsView.setScene(self.graphicsScene)
self.graphicsScene.addItem(self.graphicsItem)
self.graphicsView.setMinimumSize(10, 10)
self.pixMap = QPixmap("加载中")
self.graphicsItem.setPixmap(self.pixMap)
# self.radioButton.setChecked(True)
self.isStripModel = False
# self.radioButton.installEventFilter(self)
# self.radioButton_2.installEventFilter(self)
self.graphicsView.installEventFilter(self)
self.graphicsView.setWindowFlag(Qt.FramelessWindowHint)
self._delta = 0.1
self.scaleCnt = 0
def ShowImg(self, data):
self.scaleCnt = 0
self.pixMap = QPixmap()
self.pixMap.loadFromData(data)
self.show()
self.graphicsItem.setPixmap(self.pixMap)
self.graphicsView.setSceneRect(QRectF(QPointF(0, 0), QPointF(self.pixMap.width(), self.pixMap.height())))
self.ScalePicture()
def ScalePicture(self):
rect = QRectF(self.graphicsItem.pos(), QSizeF(
self.pixMap.size()))
unity = self.graphicsView.transform().mapRect(QRectF(0, 0, 1, 1))
width = unity.width()
height = unity.height()
if width <= 0 or height <= 0:
return
self.graphicsView.scale(1 / width, 1 / height)
viewRect = self.graphicsView.viewport().rect()
sceneRect = self.graphicsView.transform().mapRect(rect)
if sceneRect.width() <= 0 or sceneRect.height() <= 0:
return
x_ratio = viewRect.width() / sceneRect.width()
y_ratio = viewRect.height() / sceneRect.height()
x_ratio = y_ratio = min(x_ratio, y_ratio)
self.graphicsView.scale(x_ratio, y_ratio)
# if self.readImg.isStripModel:
# height2 = self.pixMap.size().height() / 2
# height3 = self.graphicsView.size().height()/2
# height3 = height3/x_ratio
# p = self.graphicsItem.pos()
# self.graphicsItem.setPos(p.x(), p.y()+height2-height3)
self.graphicsView.centerOn(rect.center())
for _ in range(abs(self.scaleCnt)):
if self.scaleCnt > 0:
self.graphicsView.scale(1.1, 1.1)
else:
self.graphicsView.scale(1/1.1, 1/1.1)
def resizeEvent(self, event) -> None:
super(self.__class__, self).resizeEvent(event)
self.ScalePicture()
def eventFilter(self, obj, ev):
if ev.type() == QEvent.KeyPress:
return True
else:
return super(self.__class__, self).eventFilter(obj, ev)
def wheelEvent(self, event):
if event.angleDelta().y() > 0:
self.zoomIn()
else:
self.zoomOut()
def zoomIn(self):
"""放大"""
self.zoom(1.1)
def zoomOut(self):
"""缩小"""
self.zoom(1/1.1)
def zoom(self, factor):
"""缩放
:param factor: 缩放的比例因子
"""
_factor = self.graphicsView.transform().scale(
factor, factor).mapRect(QRectF(0, 0, 1, 1)).width()
if _factor < 0.07 or _factor > 100:
# 防止过大过小
return
if factor >= 1:
self.scaleCnt += 1
else:
self.scaleCnt -= 1
self.graphicsView.scale(factor, factor)
def CopyPicture(self):
clipboard = QApplication.clipboard()
clipboard.setPixmap(self.pixMap)
QtBubbleLabel.ShowMsgEx(self, "复制成功")
return
def ReduceScalePic(self):
self.zoom(1/1.1)
return
def AddScalePic(self):
self.zoom(1.1)
return
def OpenPicture(self):
try:
filename = QFileDialog.getOpenFileName(self, "Open Image", ".", "Image Files(*.jpg *.png)")
if filename and len(filename) >= 1:
name = filename[0]
if os.path.isfile(name):
f = open(name, "rb")
data = f.read()
f.close()
QtImgMgr().ShowImg(data)
except Exception as ex:
Log.Error(ex)
return
def StartWaifu2x(self):
if not QtImgMgr().data:
return
self.comboBox.setEnabled(False)
self.changeButton.setEnabled(False)
self.index = self.comboBox.currentIndex()
index = self.comboBox.currentIndex()
model = ToolUtil.GetModelByIndex(index)
QtTask().AddConvertTask("", QtImgMgr().data, model, self.AddConvertBack,
cleanFlag="QtImg")
self.changeButton.setText("正在转换")
return
def AddConvertBack(self, data, waifuId, backParam, tick):
if data:
QtImgMgr().waifu2xData = data
if self.radioButton.isChecked():
self.ShowImg(data)
self.changeButton.setText("已转换")
self.changeButton.setEnabled(False)
else:
self.changeButton.setEnabled(True)
self.comboBox.setEnabled(True)
return
def SavePicture(self):
data = QtImgMgr().waifu2xData if QtImgMgr().waifu2xData else QtImgMgr().data
if not data:
return
try:
today = time.strftime("%Y%m%d%H%M%S", time.localtime(time.time()))
filepath = QFileDialog.getSaveFileName(self, "保存", "{}.jpg".format(today))
if filepath and len(filepath) >= 1:
name = filepath[0]
f = open(name, "wb")
f.write(data)
f.close()
except Exception as es:
Log.Error(es)
return
def SwithPicture(self):
if self.radioButton.isChecked() and QtImgMgr().waifu2xData:
self.ShowImg(QtImgMgr().waifu2xData)
else:
self.ShowImg(QtImgMgr().data)
return
def ChangeModel(self, index):
if self.comboBox.currentIndex() == self.index:
return
# self.index = self.comboBox.currentIndex()
self.changeButton.setText("转换")
self.changeButton.setEnabled(True)
return
class QtImg(QtWidgets.QWidget):
def __init__(self):
super(self.__class__, self).__init__()
self.bookId = ""
self.epsId = 0
self.curIndex = 0
self.gridLayout = QtWidgets.QGridLayout(self)
self.setWindowTitle("图片查看")
self.setWindowModality(QtCore.Qt.ApplicationModal)
self.resize(402, 509)
# self.setWindowFlags(Qt.FramelessWindowHint)
self.graphicsView = QtWidgets.QGraphicsView(self)
self.graphicsView.setFrameStyle(QFrame.NoFrame)
self.graphicsView.setObjectName("graphicsView")
self.graphicsView.setBackgroundBrush(QColor(Qt.white))
self.graphicsView.setCursor(Qt.OpenHandCursor)
self.graphicsView.setHorizontalScrollBarPolicy(Qt.ScrollBarAlwaysOff)
self.graphicsView.setVerticalScrollBarPolicy(Qt.ScrollBarAlwaysOff)
self.graphicsView.setRenderHints(QPainter.Antialiasing
| QPainter.HighQualityAntialiasing
| QPainter.SmoothPixmapTransform)
self.graphicsView.setCacheMode(self.graphicsView.CacheBackground)
self.graphicsView.setViewportUpdateMode(
self.graphicsView.SmartViewportUpdate)
self.graphicsItem = QGraphicsPixmapItem()
self.graphicsItem.setFlags(QGraphicsPixmapItem.ItemIsFocusable
| QGraphicsPixmapItem.ItemIsMovable)
self.setContextMenuPolicy(Qt.CustomContextMenu)
self.customContextMenuRequested.connect(self.CopyPicture)
self.graphicsScene = QGraphicsScene(self) # 场景
self.graphicsView.setScene(self.graphicsScene)
self.graphicsScene.addItem(self.graphicsItem)
self.graphicsView.setMinimumSize(10, 10)
self.pixMap = QPixmap("加载中")
self.graphicsItem.setPixmap(self.pixMap)
# self.radioButton.setChecked(True)
self.isStripModel = False
# self.radioButton.installEventFilter(self)
# self.radioButton_2.installEventFilter(self)
self.graphicsView.installEventFilter(self)
self.graphicsView.setWindowFlag(Qt.FramelessWindowHint)
self.gridLayout.addWidget(self.graphicsView)
self._delta = 0.1
self.scaleCnt = 0
def ShowImg(self, pixMap):
self.scaleCnt = 0
self.pixMap = QPixmap(pixMap)
self.show()
self.graphicsItem.setPixmap(self.pixMap)
self.graphicsView.setSceneRect(
QRectF(QPointF(0, 0),
QPointF(self.pixMap.width(), self.pixMap.height())))
self.ScalePicture()
def ScalePicture(self):
rect = QRectF(self.graphicsItem.pos(), QSizeF(self.pixMap.size()))
unity = self.graphicsView.transform().mapRect(QRectF(0, 0, 1, 1))
width = unity.width()
height = unity.height()
if width <= 0 or height <= 0:
return
self.graphicsView.scale(1 / width, 1 / height)
viewRect = self.graphicsView.viewport().rect()
sceneRect = self.graphicsView.transform().mapRect(rect)
if sceneRect.width() <= 0 or sceneRect.height() <= 0:
return
x_ratio = viewRect.width() / sceneRect.width()
y_ratio = viewRect.height() / sceneRect.height()
x_ratio = y_ratio = min(x_ratio, y_ratio)
self.graphicsView.scale(x_ratio, y_ratio)
# if self.readImg.isStripModel:
# height2 = self.pixMap.size().height() / 2
# height3 = self.graphicsView.size().height()/2
# height3 = height3/x_ratio
# p = self.graphicsItem.pos()
# self.graphicsItem.setPos(p.x(), p.y()+height2-height3)
self.graphicsView.centerOn(rect.center())
for _ in range(abs(self.scaleCnt)):
if self.scaleCnt > 0:
self.graphicsView.scale(1.1, 1.1)
else:
self.graphicsView.scale(1 / 1.1, 1 / 1.1)
def resizeEvent(self, event) -> None:
super(self.__class__, self).resizeEvent(event)
self.ScalePicture()
def eventFilter(self, obj, ev):
if ev.type() == QEvent.KeyPress:
return True
else:
return super(self.__class__, self).eventFilter(obj, ev)
def wheelEvent(self, event):
if event.angleDelta().y() > 0:
self.zoomIn()
else:
self.zoomOut()
def zoomIn(self):
"""放大"""
self.zoom(1.1)
def zoomOut(self):
"""缩小"""
self.zoom(1 / 1.1)
def zoom(self, factor):
"""缩放
:param factor: 缩放的比例因子
"""
_factor = self.graphicsView.transform().scale(factor, factor).mapRect(
QRectF(0, 0, 1, 1)).width()
if _factor < 0.07 or _factor > 100:
# 防止过大过小
return
if factor >= 1:
self.scaleCnt += 1
else:
self.scaleCnt -= 1
self.graphicsView.scale(factor, factor)
def CopyPicture(self):
clipboard = QApplication.clipboard()
clipboard.setPixmap(self.pixMap)
QtBubbleLabel.ShowMsgEx(self, "复制成功")
return
class QtImgFrame(QFrame):
def __init__(self, readImg):
QFrame.__init__(self)
self._readImg = weakref.ref(readImg)
self.graphicsView = QtWidgets.QGraphicsView(self)
self.graphicsView.setTransformationAnchor(
self.graphicsView.AnchorUnderMouse)
self.graphicsView.setResizeAnchor(self.graphicsView.AnchorUnderMouse)
self.graphicsView.setFrameStyle(QFrame.NoFrame)
self.graphicsView.setObjectName("graphicsView")
self.qtTool = QtImgTool(self)
self.graphicsView.setBackgroundBrush(QColor(Qt.white))
self.graphicsView.setCursor(Qt.OpenHandCursor)
self.graphicsView.setResizeAnchor(self.graphicsView.AnchorViewCenter)
self.graphicsView.setTransformationAnchor(
self.graphicsView.AnchorViewCenter)
self.graphicsView.setHorizontalScrollBarPolicy(Qt.ScrollBarAlwaysOff)
self.graphicsView.setVerticalScrollBarPolicy(Qt.ScrollBarAlwaysOff)
self.graphicsView.setRenderHints(QPainter.Antialiasing
| QPainter.HighQualityAntialiasing
| QPainter.SmoothPixmapTransform)
self.graphicsView.setCacheMode(self.graphicsView.CacheBackground)
self.graphicsView.setViewportUpdateMode(
self.graphicsView.SmartViewportUpdate)
self.graphicsItem = QGraphicsPixmapItem()
self.graphicsItem.setFlags(QGraphicsPixmapItem.ItemIsFocusable
| QGraphicsPixmapItem.ItemIsMovable)
self.graphicsScene = QGraphicsScene(self) # 场景
self.graphicsView.setScene(self.graphicsScene)
self.graphicsScene.addItem(self.graphicsItem)
self.graphicsView.setMinimumSize(10, 10)
self.graphicsView.installEventFilter(self)
self.graphicsView.setWindowFlag(Qt.FramelessWindowHint)
self.pixMap = QPixmap()
self.scaleCnt = 0
@property
def readImg(self):
return self._readImg()
def resizeEvent(self, event) -> None:
super(self.__class__, self).resizeEvent(event)
self.ScaleFrame()
self.ScalePicture()
def ScaleFrame(self):
size = self.size()
w = size.width()
h = size.height()
self.graphicsView.setGeometry(0, 0, w, h)
h = min(700, h)
self.qtTool.setGeometry(w - 220, 0, 220, h)
return
def ScalePicture(self):
if self.readImg.isStripModel:
self.graphicsItem.setPos(0, 0)
rect = QRectF(self.graphicsItem.pos(), QSizeF(self.pixMap.size()))
unity = self.graphicsView.transform().mapRect(QRectF(0, 0, 1, 1))
width = unity.width()
height = unity.height()
if width <= 0 or height <= 0:
return
self.graphicsView.scale(1 / width, 1 / height)
viewRect = self.graphicsView.viewport().rect()
sceneRect = self.graphicsView.transform().mapRect(rect)
if sceneRect.width() <= 0 or sceneRect.height() <= 0:
return
x_ratio = viewRect.width() / sceneRect.width()
y_ratio = viewRect.height() / sceneRect.height()
if not self.readImg.isStripModel:
x_ratio = y_ratio = min(x_ratio, y_ratio)
else:
x_ratio = y_ratio = max(x_ratio, y_ratio)
self.graphicsView.scale(x_ratio, y_ratio)
if self.readImg.isStripModel:
height2 = self.pixMap.size().height() / 2
height3 = self.graphicsView.size().height() / 2
height3 = height3 / x_ratio
p = self.graphicsItem.pos()
self.graphicsItem.setPos(p.x(), p.y() + height2 - height3)
self.graphicsView.centerOn(rect.center())
for _ in range(abs(self.scaleCnt)):
if self.scaleCnt > 0:
self.graphicsView.scale(1.1, 1.1)
else:
self.graphicsView.scale(1 / 1.1, 1 / 1.1)