Python gooey界面运行程序后卡顿严重

DAIJINGDEA 2023-09-01 20:46:55

以下代码运行后总是卡，有说关于线程问题，仍然搞不懂，求大神指教！！！！！！

from gooey import Gooey, GooeyParser
import os
import pandas as pd
import matplotlib.pyplot as plt
import PIL.Image as Image
import PIL.ImageDraw as ImageDraw
import PIL.ImageFont as ImageFont

## pre process for txt file
def pre_process_for_txt_file(file_path):
# 读取文件并备份内容
with open(file_path, 'r', encoding='utf-8') as file:
lines = file.readlines()

# 处理每一行
with open(file_path, 'w', encoding='utf-8') as file:
for line in lines:
# 如果一行全部是空格，则忽略
if line.strip() == '':
continue

# 替换空格为'.'
modified_line = line.replace(' ', '.')
file.write(modified_line)

# 读取文本文件内容
def process_txt_to_image(input_filename):
with open(input_filename, 'r') as f:
lines = f.readlines()

first_valid_raw = 1
for idx, line in enumerate(lines, start=1):
if line.strip() and all(char in ['.', 'A', 'X', 'x', 'S', '1', '2', '6'] for char in line.strip()):
first_valid_raw = int((idx - 1) / 2)
# print("============" + str(first_valid_raw)) ## TODO_ZENGXIANG: find a feature to diff old 2 txt && 3 new txt
break

color_mapping = {
' ': (255, 255, 255), # 白色
'.': (255, 255, 255), # 白色
'A': (0, 255, 0), # 绿色
'1': (0, 255, 0), # 绿色
'X': (255, 0, 0), # 红色
'x': (255, 0, 0), # 红色
'2': (255, 0, 0), # 红色
'6': (0, 0, 255), # 蓝色
'S': (255, 255, 0), # 黄色
}

# 去除每行末尾的换行符
lines = [line.strip() for line in lines]

# 创建新图像
image_height = len([line for line in lines if line]) * 40 # 只计算非空行的高度，每行对应一个40x40的方框
max_line_length = max(len(line) for line in lines if line)
image_width = max_line_length * 40 # 每个字符对应一个40x40的方框

image = Image.new('RGB', (image_width, image_height), color=(255, 255, 255))
draw = ImageDraw.Draw(image)
font = ImageFont.load_default() # 使用默认字体

# 绘制方框、字符和坐标
y = 0
for line in lines:
if line:
x = 0
has_special_char = False
for idx, char in enumerate(line):
if char in color_mapping:
color = color_mapping[char]
draw.rectangle([x, y, x+40, y+40], fill=color, outline=(0, 0, 0))
char_width, char_height = draw.textsize(char, font=font)
draw.text((x+(40-char_width)//2, y+(40-char_height)//2), char, fill=(0, 0, 0), font=font)
draw.text((x, y+5), f"({idx+1},{y//40+1-first_valid_raw})", fill=(0, 0, 0), font=font)
else:
has_special_char = True
break
x += 40

if has_special_char:
draw.text((x+5, y+5), line, fill=(0, 0, 0), font=font)
y += 40
else:
y += 40

# 保存图像
output_filename = input_filename.replace('txt', 'png')
output_filename = input_filename.replace('TXT', 'png')
image.save(output_filename)
print("图像已保存为: " + output_filename)

# read excel file context
def process_excel_to_image(input_filename):

df = pd.read_excel(input_filename, skiprows=11)
# 读取Excel数据
df = pd.read_excel(input_filename)

# 提取数据列
x_values = df['X']
y_values = df['Y']
rdson_values = df['Vth_RT (V)']

# 绘制散点图
plt.figure(figsize=(8, 6))
plt.scatter(x_values, y_values, c=rdson_values, cmap='viridis', s=50, marker='o')

# 添加颜色标注
cbar = plt.colorbar()
cbar.set_label('Rdson_RT (Ω)')

# 添加标题和坐标轴标签
plt.title('Scatter Plot of Rdson_RT')
plt.xlabel('X')
plt.ylabel('Y')

# 保存图像
print(input_filename)
output_filename = input_filename.replace('excel', 'png')
output_filename = output_filename.replace('xlsx', 'png')
plt.savefig(output_filename)
print("图像已保存为: " + output_filename)

@Gooey(program_name="WaferAnalysis_GUI", language="chinese")
def main():
parser = GooeyParser(description="Convert test file to image")

group = parser.add_mutually_exclusive_group(required=True)
group.add_argument("-m", type=int, choices=[1, 2], help="Mode: 1 for convert txt to png, 2 for convert excel to png")

parser.add_argument("-i", help="Path to the folder containing txt files.")
parser.add_argument("-o", help="Path to the folder to save png files.")

args = parser.parse_args()
txt_path = args.i
input_filename = os.listdir(txt_path)

mode = args.m

if mode == 1:
for file_name in input_filename:
absolute_path = os.path.join(txt_path, file_name)
pre_process_for_txt_file(txt_path + "/" + file_name)
process_txt_to_image(txt_path + "/" + file_name)
elif mode == 2:
for file_name in input_filename:
absolute_path = os.path.join(txt_path, file_name)
process_excel_to_image(txt_path + "/" + file_name)
print("TODO.............")

if __name__ == "__main__":
main()

...全文