feat: 添加TIFF处理类

Author: CheaterScript

main.py

@@ -8,14 +8,15 @@
 
 from PyQt6 import QtWidgets, QtCore
 from PyQt6.QtGui import QGuiApplication
+from tools.tiff_processer import TIFF
 from ui.Ui_MainWindow import Ui_MainWindow
 
 from tools.raw_processer import Raw
 from tools.png_processer import PNG
 from tools.jpg_processer import JPG
 
 # 支持的格式, 不在其中的会跳过处理
-FILE_TYPES = ('.jpg', '.jpge', '.png', '.raw')
+FILE_TYPES = ('.jpg', '.jpge', '.png', '.raw', '.tif', '.tiff')
 
 
 class ToolWindow(QtWidgets.QMainWindow, Ui_MainWindow):
@@ -24,7 +25,7 @@ def __init__(self, parent=None):
 
     def openFile(self):
         fileName, _ = QtWidgets.QFileDialog.getOpenFileName(self, "打开文件", os.getcwd(
-        ), "All Images(*.jpg;*jpge;*.png;*.raw);;PNG(*.png);;RAW(*.raw);;JEPE(*.jpg;*jpge)")
+        ), "All Images(*.jpg;*jpge;*.png;*.raw;*.tiff;*.tif);;PNG(*.png);;Photoshop RAW(*.raw);;JEPE(*.jpg;*jpge);;TIFF(*.tif;*.tiff)")
 
         self.lineEdit_input.setText(fileName)
 
@@ -152,6 +153,8 @@ def handle(self):
             processer = PNG(inputPath, outputPath, row, col)
         if extension_name == '.jpg' or extension_name == '.jpge':
             processer = JPG(inputPath, outputPath, row, col)
+        if extension_name == '.tif' or extension_name == '.tiff':
+            processer = TIFF(inputPath, outputPath, row, col)
 
         if not processer:
             self.complete()

test.raw

@@ -37,7 +37,6 @@ def slice(self) -> None:
                 print("正在处理%d/%d张……" % (i*self.rows+j, total))
 
     def save(self, img, index) -> None:
-        print(img.shape)
         cv2.imencode(self.inputPath.suffix, img, [int(cv2.IMWRITE_PNG_COMPRESSION), 0])[
             1].tofile("%s/%s_%03d%s" %
                       (self.output, self.inputPath.stem, index, self.inputPath.suffix))

test.tif

@@ -0,0 +1,42 @@
+import cv2
+from tools.processer import Processer
+from pathlib import Path
+import math
+from PyQt6 import QtWidgets
+import numpy as np
+
+
+class TIFF(Processer):
+    def load(self) -> None:
+        self.img = cv2.imdecode(np.fromfile(
+            self.input, dtype=np.uint8), cv2.IMREAD_UNCHANGED)
+        self.inputPath = Path(self.input)
+
+    def slice(self) -> None:
+        self.load()
+        total = self.rows * self.cols
+        shape = self.img.shape
+        singleWidth = math.ceil(shape[1] / self.cols)
+        singleHeight = math.ceil(shape[0] / self.rows)
+        print(shape)
+        if singleWidth * self.cols != shape[0] or singleHeight * self.rows != shape[1]:
+            self.img = cv2.resize(
+                self.img, (singleWidth * self.cols, singleHeight * self.rows))
+
+        for i in range(self.rows):
+            for j in range(self.cols):
+                percent = int((i*self.rows+j + 1) / total * 100)
+                self.update.emit(percent)
+
+                x = j * singleWidth
+                y = i * singleHeight
+
+                new_img = self.img[y:y+singleHeight, x: x+singleWidth]
+
+                self.save(new_img, i*self.rows+j)
+                print("正在处理%d/%d张……" % (i*self.rows+j, total))
+
+    def save(self, img, index) -> None:
+        cv2.imencode(self.inputPath.suffix, img, [int(cv2.IMWRITE_JPEG_QUALITY), 100])[
+            1].tofile("%s/%s_%03d%s" %
+                      (self.output, self.inputPath.stem, index, '.jpg'))