forked from yuka-friends/Windrecorder
-
Notifications
You must be signed in to change notification settings - Fork 0
/
Copy pathonboard_setting.py
253 lines (202 loc) · 8.23 KB
/
onboard_setting.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
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
211
212
213
214
215
216
217
218
219
220
221
222
223
224
225
226
227
228
229
230
231
232
233
234
235
236
237
238
239
240
241
242
243
244
245
246
247
248
249
250
251
252
253
import getpass
import os
import subprocess
import time
import windrecorder.maintainManager as maintainManager
import windrecorder.record as record
import windrecorder.utils as utils
from windrecorder.config import config
from windrecorder.utils import get_text as _t
# 全部向导的步骤数
ALLSTEPS = 5
# 清理缓存
if os.path.exists("cache"):
utils.empty_directory("cache")
# 画分割线的
def divider():
print("\n--------------------------------------------------------------------\n")
# 画抬头的
def print_header(step=1, toast=""):
subprocess.run("cls", shell=True)
print("Weclome to Windrecorder | 欢迎使用捕风记录仪\n")
print("Thanks for downloading! This Quick Wizard will help you set it up. \n感谢下载使用！本向导将协助你完成基础配置项。不用担心，所有选项之后都可以再次调整。")
divider()
print(step, "/", ALLSTEPS, toast)
print("\n")
# 配置文件已有选项的指示器
def config_indicator(config_element, expect_result):
if config_element == expect_result:
return _t("qs_config_indicator")
else:
return ""
# ========================================================
subprocess.run("color 06", shell=True)
# 设置语言
while True:
print_header(step=1)
print("First, please choose your interface language. (Enter the number option and press Enter to confirm.)")
print("首先，请设置你的界面语言。（输入数字项后回车确认）")
divider()
print(
f"""
1. English {config_indicator(config.lang,"en")}
2. 简体中文 {config_indicator(config.lang,"sc")}
3. 日本語 {config_indicator(config.lang,"ja")}
"""
)
input_lang_num = input("> ")
if input_lang_num == "1":
config.set_and_save_config("lang", "en")
lang = "en"
print("The interface language is set to English")
break
if input_lang_num == "2":
config.set_and_save_config("lang", "sc")
lang = "sc"
print("界面语言已设定为：简体中文")
break
if input_lang_num == "3":
config.set_and_save_config("lang", "ja")
lang = "ja"
print("インターフェース言語は日本語に設定されています。")
break
else:
print(_t("qs_la_text_same_as_previous"))
break
divider()
subprocess.run("pause", shell=True)
# 设置用户名
while True:
print_header(step=2)
if config.user_name == "default":
sys_username = getpass.getuser() # 如果为默认用户名，获取当前系统的用户名
else:
sys_username = config.user_name # 如果配置文件已有自定义用户名，读取之前的用户名
print(_t("qs_un_set_your_username"))
print(_t("qs_un_describe").format(sys_username=sys_username))
divider()
your_username = input("> ")
if len(your_username) > 20:
print(_t("qs_un_longer_than_expect"))
divider()
subprocess.run("pause", shell=True)
elif len(your_username) == 0:
print(_t("qs_un_use_current_name").format(sys_username=sys_username))
config.set_and_save_config("user_name", sys_username)
break
else:
print(_t("qs_un_use_custom_name").format(your_username=your_username))
config.set_and_save_config("user_name", your_username)
break
divider()
subprocess.run("pause", shell=True)
# 选择可ocr的语言
os_support_lang = utils.get_os_support_lang()
if len(os_support_lang) > 1: # 如果系统安装了超过一种语言
while True:
print_header(step=3)
print(_t("qs_olang_intro"))
utils.print_numbered_list(os_support_lang)
divider()
try:
input_ocr_lang_num = int(input("> "))
if input_ocr_lang_num <= len(os_support_lang):
config.set_and_save_config("ocr_lang", os_support_lang[input_ocr_lang_num - 1])
print(
_t("qs_olang_ocrlang_set_to"),
os_support_lang[input_ocr_lang_num - 1],
)
subprocess.run("pause", shell=True)
break
except ValueError:
print(_t("qs_olang_error"))
subprocess.run("pause", shell=True)
else: # 如果系统只安装了一种语言，自动选择
print_header(step=3)
print(_t("qs_olang_one_choice_default_set").format(os_support_lang=os_support_lang[0]))
subprocess.run("pause", shell=True)
# 测试与设置 ocr 引擎
test_img_filepath = "__assets__\\OCR_test_1080_" + config.ocr_lang + ".png" # 读取测试图像
if not os.path.exists(test_img_filepath):
test_img_filepath = "OCR_test_1080_en-US.png" # fallback 读取为英文图像
with open("__assets__\\OCR_test_1080_words_" + config.ocr_lang + ".txt", encoding="utf-8") as f: # 读取比对参考文本
ocr_text_refer = f.read()
ocr_text_refer = utils.wrap_text_by_remove_break(ocr_text_refer)
# 测试COL - 已废弃
# if config.enable_ocr_chineseocr_lite_onnx:
# try:
# time_cost_col = time.time()
# ocr_result_col = maintainManager.ocr_image_col(test_img_filepath)
# time_cost_col = time.time() - time_cost_col
# ocr_result_col = utils.wrap_text_by_remove_break(ocr_result_col)
# _, ocr_correct_col = maintainManager.compare_strings(ocr_result_col, ocr_text_refer)
# except Exception as e:
# ocr_result_col = ""
# print(e)
# else:
# print("enable_ocr_chineseocr_lite_onnx disabled.")
# ocr_result_col = ""
# 测试ms ocr
time_cost_ms = time.time()
ocr_result_ms = maintainManager.ocr_image_ms(test_img_filepath)
time_cost_ms = time.time() - time_cost_ms
ocr_result_ms = utils.wrap_text_by_remove_break(ocr_result_ms)
_, ocr_correct_ms = maintainManager.compare_strings(ocr_result_ms, ocr_text_refer)
while True:
print_header(step=3)
print(_t("qs_ocr_title"))
if ocr_result_ms:
print("- Windows.Media.Ocr.Cli OCR languages: ", config.ocr_lang)
print(
_t("qs_ocr_result_describe").format(
accuracy=ocr_correct_ms,
timecost=time_cost_ms,
timecost_15=utils.convert_seconds_to_hhmmss(int(time_cost_ms * 350)),
)
)
if ocr_correct_ms < 50:
print(_t("qs_ocr_tips_low_accuracy"))
break
# if ocr_result_col:
# print("- chineseocr_lite_onnx")
# # print("准确率：", ocr_correct_col, "，识别时间：", time_cost_col, "，索引15分钟视频约用时：", utils.convert_seconds_to_hhmmss(int(time_cost_col*350)))
# print(_t("qs_ocr_result_describe").format(accuracy=ocr_correct_col, timecost=time_cost_col , timecost_15=utils.convert_seconds_to_hhmmss(int(time_cost_col*350))))
# else:
# print(_t("qs_ocr_describe_disable_clo"))
# divider()
# print(_t("qs_ocr_describe"))
# print(_t("qs_ocr_cta"))
# print("1. Windows.Media.Ocr.Cli", _t("qs_ocr_option_recommand"), "\n2. chineseocr_lite_onnx")
# input_lang_num = input('> ')
# if input_lang_num == '2':
# if config.enable_ocr_chineseocr_lite_onnx:
# config.set_and_save_config("ocr_engine", "chineseocr_lite_onnx")
# print(_t("qs_ocr_option_recommand"), "chineseocr_lite_onnx")
# else:
# config.set_and_save_config("ocr_engine", "Windows.Media.Ocr.Cli")
# print(_t("qs_ocr_engine_chosen"), "Windows.Media.Ocr.Cli")
# break
# else:
# config.set_and_save_config("ocr_engine", "Windows.Media.Ocr.Cli")
# print(_t("qs_ocr_engine_chosen"), "Windows.Media.Ocr.Cli")
# break
divider()
subprocess.run("pause", shell=True)
# 设置显示器录制选项
while True:
print_header(step=4)
print(_t("qs_mo_describe"))
monitor_width = utils.get_screen_resolution().width
monitor_height = utils.get_screen_resolution().height
scale_width, scale_height = record.get_scale_screen_res_strategy(origin_width=monitor_width, origin_height=monitor_height)
print(_t("qs_mo_detect").format(monitor_width=monitor_width, monitor_height=monitor_height))
print(_t("qs_mo_cta"))
break
divider()
subprocess.run("pause", shell=True)
# 完成初始化设定
print_header(step=5)
print(_t("qs_end_describe"))
print(_t("qs_end_slogan"))
print(_t("qs_end_feedback"))
divider()