img2brailleGUI/GUI.py at main · wang11wei/img2brailleGUI · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
from PySide2.QtWidgets import QApplication, QMessageBox, QFileDialog
from PySide2.QtUiTools import QUiLoader
from PySide2.QtGui import QPixmap
from img2braille import *


class Stats:

    def __init__(self):
        # 从文件中加载UI定义
        self.ui = QUiLoader().load('transformGUI.ui')

        # 选择图像
        self.ui.button_selectPic.clicked.connect(self.openimage)
        self.ui.button_start.clicked.connect(self.start)

    def openimage(self):
        # getOpenFileName第一个参数为父类名，直接用最大的父类 self.ui
        imgName, imgType = QFileDialog.getOpenFileName(self.ui, "打开图片", "", "*.png;;*.jpg;;All Files(*)")
        # 文本输出
        self.ui.text_selectPic.setText(imgName)
        jpg = QPixmap(imgName).scaled(self.ui.showImage.width(), self.ui.showImage.height())
        self.ui.showImage.setPixmap(jpg)

    def start(self):
        if self.ui.text_selectPic.text():
            filename = self.ui.text_selectPic.text()
            print(filename)
            disable_smoothing = False
            no_resize = False
            width = int(self.ui.text_width.text())
            height = None
            invert = self.ui.button_invert.isChecked()

            if no_resize:
                resize_size = size_max
            elif width and not height:
                resize_size = size_from_width_ratio(width)
            elif height and not width:
                resize_size = size_from_height_ratio(height)
            else:
                resize_size = size_from_width_ratio(30)  # ！！！设置宽度，
            print(self.ui.spin_bsize.value())
            result = img2braille(
                filename=filename,
                bsize=int(self.ui.spin_bsize.value()),
                resize_size=resize_size,
                smoothing=not disable_smoothing,
                invert=invert
            )
            try:
                res = ""
                for c in result:
                    res += c
                self.ui.text_ouput.setPlainText(res)
            except BrokenPipeError:
                pass
        else:
            # 输出文本框
            QMessageBox.about(self.ui,
                              'error',
                              'select image first!'
                              )


app = QApplication([])
stats = Stats()
stats.ui.show()
app.exec_()