ADD file via upload

hahaMirror.py

@@ -0,0 +1,135 @@
+# -*- coding: utf-8 -*-
+import sys
+import cv2
+import math
+from PyQt5.QtCore import QRect, QCoreApplication, QMetaObject, Qt
+from PyQt5.QtGui import QImage, QPixmap
+from PyQt5.QtWidgets import QMainWindow, QApplication, QMenuBar, QWidget, QPushButton, QGraphicsView, QGraphicsScene, \
+    QFileDialog, QStatusBar, QMessageBox
+
+
+class Ui_MainWindow(object):
+    def setupUi(self, MainWindow):
+        if not MainWindow.objectName():
+            MainWindow.setObjectName("MainWindow")
+        MainWindow.resize(800, 600)
+        self.centralwidget = QWidget(MainWindow)
+        self.centralwidget.setObjectName("centralwidget")
+
+        self.pushButton_1 = QPushButton(self.centralwidget)
+        self.pushButton_1.setObjectName("pushButton_1")
+        self.pushButton_1.setGeometry(QRect(80, 470, 191, 71))
+
+        self.pushButton_2 = QPushButton(self.centralwidget)
+        self.pushButton_2.setObjectName("pushButton_2")
+        self.pushButton_2.setGeometry(QRect(300, 470, 201, 71))
+
+        self.pushButton_3 = QPushButton(self.centralwidget)
+        self.pushButton_3.setObjectName("pushButton_3")
+        self.pushButton_3.setGeometry(QRect(530, 470, 201, 71))
+
+        self.graphicsView = QGraphicsView(self.centralwidget)
+        self.graphicsView.setObjectName("graphicsView")
+        self.graphicsView.setGeometry(QRect(70, 60, 275, 330))
+
+        self.graphicsView_2 = QGraphicsView(self.centralwidget)
+        self.graphicsView_2.setObjectName("graphicsView_2")
+        self.graphicsView_2.setGeometry(QRect(450, 60, 280, 330))
+        MainWindow.setCentralWidget(self.centralwidget)
+        self.menubar = QMenuBar(MainWindow)
+        self.menubar.setObjectName("menubar")
+        self.menubar.setGeometry(QRect(0, 0, 800, 23))
+        MainWindow.setMenuBar(self.menubar)
+        self.statusbar = QStatusBar(MainWindow)
+        self.statusbar.setObjectName("statusbar")
+        MainWindow.setStatusBar(self.statusbar)
+
+        self.retranslateUi(MainWindow)
+        QMetaObject.connectSlotsByName(MainWindow)
+
+    def retranslateUi(self, MainWindow):
+        MainWindow.setWindowTitle(QCoreApplication.translate("MainWindow", "MainWindow", None))
+        self.pushButton_2.setText(QCoreApplication.translate("MainWindow", "Apply Funny Mirror", None))
+        self.pushButton_1.setText(QCoreApplication.translate("MainWindow", "Select Image", None))
+        self.pushButton_3.setText(QCoreApplication.translate("MainWindow", "Save Image", None))
+
+def enlarge_effect(img):
+    #获取图像的高度 h，宽度 w，和颜色通道数
+    h, w, n = img.shape
+    #计算图像的中心点坐标
+    cx = w / 2
+    cy = h / 2
+    #定义影响的区域
+    radius = min(h, w) // 2
+    r = int(radius / 2.0)
+    #创建原始图像的一个副本，在这个副本上应用效果
+    new_img = img.copy()
+    #用双重循环来遍历图像的每一个像素
+    for i in range(w):
+        for j in range(h):
+            tx = i - cx
+            ty = j - cy
+            distance = tx * tx + ty * ty
+            #判断当前像素是否在影响范围内
+            if distance < radius * radius:
+                #图像中心区域的像素进行重新映射，创建一种放大效果
+                x = int(int(tx / 2.0) * (math.sqrt(distance) / r) + cx)
+                y = int(int(ty / 2.0) * (math.sqrt(distance) / r) + cy)
+                if 0 <= x < w and 0 <= y < h:  # 添加边界检查
+                    new_img[j, i, 0] = img[y, x, 0]
+                    new_img[j, i, 1] = img[y, x, 1]
+                    new_img[j, i, 2] = img[y, x, 2]
+    return new_img
+
+class MainWindow(QMainWindow, Ui_MainWindow):
+    def __init__(self):
+        super(MainWindow, self).__init__()
+        self.setupUi(self)
+        self.pushButton_1.clicked.connect(self.select_image)
+        self.pushButton_2.clicked.connect(self.apply_effect)
+        self.pushButton_3.clicked.connect(self.save_image)
+        self.input_img = None
+        self.output_img = None
+
+    def select_image(self):
+        options = QFileDialog.Options()
+        fileName, _ = QFileDialog.getOpenFileName(self, "Select Image", "", "Image Files (*.png *.jpg *.bmp)", options=options)
+        if fileName:
+            self.input_img = cv2.imread(fileName)
+            if self.input_img is None:
+                print("Error: Unable to read image file")
+                return
+            self.show_image(self.input_img, self.graphicsView)
+
+    def apply_effect(self):
+        if self.input_img is None:
+            return
+        self.output_img = enlarge_effect(self.input_img)
+        self.show_image(self.output_img, self.graphicsView_2)
+
+    def show_image(self, img, view):
+        img_rgb = cv2.cvtColor(img, cv2.COLOR_BGR2RGB)
+        h, w, ch = img_rgb.shape
+        bytes_per_line = ch * w
+        qimg = QImage(img_rgb.data, w, h, bytes_per_line, QImage.Format_RGB888)
+        pixmap = QPixmap.fromImage(qimg)
+        scene = QGraphicsScene()
+        scene.addPixmap(pixmap)
+        view.setScene(scene)
+        view.fitInView(scene.itemsBoundingRect(), Qt.KeepAspectRatio)  # 修正了参数名称
+
+    def save_image(self):
+        if self.output_img is not None:
+            options = QFileDialog.Options()
+            options |= QFileDialog.ShowDirsOnly
+            file_name, _ = QFileDialog.getSaveFileName(self, "保存图片", "", "Images (*.png *.xpm *.jpg);;All Files (*)", options=options)
+            if file_name:
+                cv2.imwrite(file_name, self.output_img)
+        else:
+            QMessageBox.warning(self, "Warning", "没有转换后的图片可保存")
+
+if __name__ == "__main__":
+    app = QApplication(sys.argv)
+    mainWin = MainWindow()
+    mainWin.show()
+    sys.exit(app.exec_())