From fb3ce29de06bb20438940ae8297abaaf16c25eb7 Mon Sep 17 00:00:00 2001 From: hcy <202152320217@stu.zzut.edu.cn> Date: Sun, 23 Jun 2024 14:20:55 +0000 Subject: [PATCH] update app.py. Signed-off-by: hcy <202152320217@stu.zzut.edu.cn> --- app.py | 24 ++++++++++++------------ 1 file changed, 12 insertions(+), 12 deletions(-) diff --git a/app.py b/app.py index 68a8a9d..66e60c7 100644 --- a/app.py +++ b/app.py @@ -24,24 +24,24 @@ class DocScannerApp: # 在窗口中创建一个宽600，高400画布 # 用于把图片和裁剪框绘制上去 - self.canvas = #TODO + self.canvas = tk.Canvas(self.root, width=600, height=400) # 将画布添加到窗口中 - #TODO + self.canvas.pack() # 为画布绑定鼠标事件 # 鼠标左键按下,释放和移动事件都绑定到方法：mouse_callback 上 # 注意绑定的事是实例方法，不要忘了self - #TODO # 鼠标按下事件 - #TODO # 鼠标释放事件 - #TODO # 鼠标移动事件 + self.canvas.bind(", self.mouse_callback") # 鼠标按下事件 + self.canvas.bind(", self.mouse_callback") # 鼠标释放事件 + self.canvas.bind(", self.mouse_callback") # 鼠标移动事件 # 创建一个 "Select Image" 按钮，点击时执行 self.select_image 函数 btn_select = tk.Button(self.root, text="Select Image", command=self.select_image) btn_select.pack(side=tk.LEFT) # 将按钮添加到窗口左侧 - self.show_mouse_move = #TODO # 创建一个 BooleanVar 实例来保存复选框的状态 + self.show_mouse_move = tk.BooleanVar() # 创建一个 BooleanVar 实例来保存复选框的状态 # 创建一个复选框，文本为 "Show Mouse Move"，把状态与 self.show_mouse_move 绑定 - chk_show_mouse_move = #TODO + chk_show_mouse_move = tk.Checkbutton(self.root, text="Show Mouse Move", variable=self.show_mouse_move) chk_show_mouse_move.pack(side=tk.LEFT) # 将复选框添加到窗口左侧 # 创建一个 "Crop" 按钮，点击时执行 self.crop 函数 @@ -55,9 +55,9 @@ class DocScannerApp: img_height, img_width, _ = self.img.shape # 获取图片的高度和宽度 if img_height > 600 or img_width > 800: # 如果图片的高度大于600或宽度大于800 - scale = #TODO # 计算缩放比例 - self.img = #TODO # 缩放图片 - self.corners = #TODO # 缩放角落 + scale = min(600 / img_height, 800 / img_width) # 计算缩放比例 + self.img = cv.resize(self.img, (int(img_width * scale), int(img_height * scale))) # 缩放图片 + self.corners = self.corners * scale # 缩放角落 img_height, img_width, _ = self.img.shape # 更新图片的高度和宽度 self.canvas.config(width=img_width, height=img_height) # 调整画布的大小以适应图片 @@ -71,7 +71,7 @@ class DocScannerApp: if event.type == tk.EventType.ButtonPress: # 如果是鼠标按下事件 print(f"鼠标在({x},{y})按下") for idx, corner in enumerate(self.corners): # 遍历每个角落点 - #TODO: # 如果鼠标位置和角落的距离小于10 + if np.linalg.norm(corner - np.array([x, y])) < 10: # 如果鼠标位置和角落的距离小于10 self.dragging_idx = idx # 设置正在拖动的角落索引值为当前的角落的索引值 print(f"要开始拖动角落: corners[{idx}]={corner} , 所以变量: dragging_idx 别设置为 {idx}") break # 已经找到了要拖动的点，所以跳出循环 @@ -92,7 +92,7 @@ class DocScannerApp: # 在图片上绘制一个绿色的圆形标记角落的位置 cv.circle(img_copy, tuple(corner.astype(int)), 5, (0, 255, 0), -1) # 在图片上绘制一个绿色的，宽度为2的多边形连接所有角落 - #TODO + cv.drawContours(img_copy, [self.corners.astype(int)], -1, (0, 255, 0), 2) img_tk = self.cv2image_to_tkinter_image(img_copy) # 将 OpenCV 图片转换为 Tkinter 图片 self.canvas.create_image(0, 0, anchor=tk.NW, image=img_tk) # 在画布上创建图片 self.canvas.image = img_tk # 保存图片，防止被垃圾回收 -- Gitee