-
Notifications
You must be signed in to change notification settings - Fork 2
/
Copy pathGUI_main.py
132 lines (121 loc) · 6.98 KB
/
GUI_main.py
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
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
"""
图形界面
"""
# TODO:
# 插帧功能：
# Paper: https://arxiv.org/abs/2011.06294
# Code: https://github.com/hzwer/RIFE
import tkinter as tk # 使用Tkinter_GUI包
from tkinter import StringVar
import GUI_func as gui
from tkinter_stdout import myStdout
""" 窗口初始化 """
window = tk.Tk() # 实例化object，建立窗口window
window.title('Super-resolution 超分辨率工具箱 v1.0_demo') # 窗口名
window.geometry('740x690') # 设定窗口的大小(长x宽)
window.resizable(0, 0) # 设置窗口大小不可变
myfront = ['Consolas', 11] # 字体参数
infofront = ['Consolas', 10]
""" 框架组件 用于布局"""
# 父组件 边框像素(1凹陷感，2框线感)->简写bd 边线风格 垂直边距 高 宽
f1 = tk.Frame(window, borderwidth=2, relief="groove", height=100, width=700)
f2 = tk.Frame(window, borderwidth=2, relief="groove", height=110, width=700)
f3 = tk.Frame(window, borderwidth=2, relief="groove", height=400, width=700)
# grid布局 所在行 所在列 (跨越行数) 水平边距 垂直边距
f1.grid(row=0, column=0, padx=20, pady=20) # 700+20+20=740 边框长+边距*2=窗口长 因此边框居中
f2.grid(row=1, column=0, padx=20, pady=0)
f3.grid(row=2, column=0, padx=20, pady=20)
# 固定组件大小,防止子组件大小影响
f1.grid_propagate(0) # 子组件为grid布局
f2.grid_propagate(0)
f3.grid_propagate(0)
""" 标签 输入输出"""
title_output = tk.Label(f1, text='输出路径：', font=myfront, width=0, height=0)
title_input = tk.Label(f1, text='输入路径：', font=myfront, width=0, height=0)
# font为字体 width为长 height为高 这里的是字符的长和高(字符间距)
""" 输入路径 """
text_input = tk.Entry(f1, width=65, bd=2, show=None, state='normal', exportselection=False) # 无密文 可写
# 此处双击文本 与list选择模型(line:70)事件冲突 双击后list失去选择报错 指定ex=f后解决(选中文本不可复制到剪贴板)
# 可双击选中 ctrl+C复制
text_input.insert(0, "未选择...") # 写入文本
text_input.config(state='readonly') # 写入后设为只读
""" 输出路径 """
text_output = tk.Entry(f1, width=65, bd=2, show=None, state='normal', exportselection=False)
text_output.insert(0, "未选择...")
text_output.config(state='readonly')
""" 按钮 选择文件 """
button_chooseFile = tk.Button(f1, text='选择文件', font=myfront, width=10, height=1, bd=4,
command=lambda: gui.func_chooseFile(entryInput=text_input, entryOutput=text_output))
# command是按钮回调函数 lambda:传递参数
""" 按钮 开始 """
button_start = tk.Button(f1, text='开始', font=myfront, width=10, height=1, bd=4,
command=lambda: gui.func_start())
""" 布局 """
title_input.grid(row=0, column=0, padx=10, pady=15)
title_output.grid(row=1, column=0, padx=10, pady=0)
text_input.grid(row=0, column=1, padx=0, pady=0)
text_output.grid(row=1, column=1, padx=0, pady=0)
button_chooseFile.grid(row=0, column=2, padx=15, pady=0)
button_start.grid(row=1, column=2, padx=15, pady=0)
""" 标签 参数选择"""
title_chvi = tk.Label(f2, text='视频/图像：', font=myfront, width=0, height=0)
title_chways = tk.Label(f2, text='选择算法：', font=myfront, width=0, height=0)
title_model = tk.Label(f2, text='训练模型：', font=myfront, width=0, height=0)
""" 列表 模型选择 """
fm1 = tk.Frame(f2, borderwidth=2, relief="groove", height=100, width=160) # 滚动条和显示框 组合框 用于布局
scroll_model = tk.Scrollbar(fm1, orient='vertical') # 用于list的垂直滚动条
list_model = tk.Listbox(fm1, height=5, yscrollcommand=scroll_model.set) # 创建list组件 height指定显示行数 yscrollcommand设置滚动条
list_model.bind('<<ListboxSelect>>', lambda event: gui.func_chModel(listbox=list_model)) # 列表框绑定函数 参数传递
gui.renewList(list_model, gui.model_srcnn) # 更新list
scroll_model.config(command=list_model.yview) # 操作滚动条 调用list显示
""" 单选 算法选择 """
# 创建n个radiobutton 其中variable=var
# value='A'：当选中其中一个选项 将value赋值到variable的参数 variable的参数变量控制绘制的显示状态
r_way = StringVar() # py变量值相同 变量名不同 他们的id内存地址也是一样的 因此两组选择控件会冲突 不建议参数变量直接用int型 而是StringVar
r_way.set('SRCNN')
radio_way = tk.Radiobutton(f2, text='SRCNN', variable=r_way, value='SRCNN',
command=lambda: gui.func_Ways(ch='SRCNN', listbox=list_model))
radio_way2 = tk.Radiobutton(f2, text='FSRCNN', variable=r_way, value='FSRCNN',
command=lambda: gui.func_Ways(ch='FSRCNN', listbox=list_model))
""" 单选 视频/图像 """
r_vi = StringVar()
r_vi.set('image')
radio_vi = tk.Radiobutton(f2, text='视频', variable=r_vi, value='video',
command=lambda: gui.func_VideoImage(
ch='video', radio1=radio_way, radio2=radio_way2, listbox=list_model, r_way=r_way))
radio_vi2 = tk.Radiobutton(f2, text='图像', variable=r_vi, value='image',
command=lambda: gui.func_VideoImage(
ch='image', radio1=radio_way, radio2=radio_way2, listbox=list_model, r_way=r_way))
""" 布局 """
title_chvi.grid(row=0, column=0, padx=40, pady=10)
radio_vi.grid(row=2, column=0, padx=0, pady=0)
radio_vi2.grid(row=1, column=0, padx=0, pady=0)
title_chways.grid(row=0, column=1, padx=80, pady=0)
radio_way.grid(row=1, column=1, padx=0, pady=0)
radio_way2.grid(row=2, column=1, padx=0, pady=0)
title_model.grid(row=0, column=2, padx=0, pady=0)
fm1.grid(row=0, rowspan=3, column=3, padx=0, pady=4) # 组合框在f2的grid布局
list_model.pack(side="left", fill="y") # 列表/滚动条在fm1的pack布局 放置方向 填充方向
scroll_model.pack(side="right", fill="both")
""" 标签 """
title_inform = tk.Label(f3, text='控制台信息', font=myfront, width=0, height=0)
""" 文本框 控制台信息 """
fm2 = tk.Frame(f3, borderwidth=2, relief="groove", height=355, width=680) # 滚动条和信息框 组合框 用于布局
scroll_inform = tk.Scrollbar(fm2, orient='vertical', width=21) # 垂直滚动条
text_inform = tk.Text(fm2, yscrollcommand=scroll_inform.set, font=infofront, width=93, exportselection=False)
# 宽高单位为字符数 不可复制剪切板原因见line39
scroll_inform.config(command=text_inform.yview) # 垂直_操作滚动条
""" 布局 """
title_inform.grid(row=0, column=0, padx=0, pady=4)
fm2.grid(row=1, column=0, padx=8, pady=0) # 组合框在 f2 的grid布局
fm2.pack_propagate(0) # 固定pack布局大小
text_inform.pack(side="left", fill="both") # 信息/滚动条在 fm2 的pack布局
scroll_inform.pack(side="right", fill="both")
""" 重定向标准输出"""
mystd = myStdout(text_inform) # 实例化重定向类
""" 主窗口循环显示 """
window.mainloop()
# 没有mainloop 则是静态window 传入值不会有循环
# window.mainloop会让window不断刷新
# mainloop相当很大的while循环 点击一次更新一次
mystd.restoreStd() # 恢复标准输出