帮我增加一个功能，框选好的尺寸可以进行保存在主界面，方便下次点击就能调用 import os import tkinter as tk from tkinter import filedialog, messagebox, ttk from PIL import Image, ImageTk, UnidentifiedImageError class ImageCropperApp: def __init__(self, root): self.root = root self.root.title("批量图片裁剪工具") self.root.geometry("900x600") self.root.minsize(800, 500) # 设置中文字体支持 self.style = ttk.Style() self.style.configure('.', font=('SimHei', 10)) # 初始化变量 self.images = [] # 存储所有待处理图片路径 self.current_image_index = 0 self.current_image = None self.current_photo = None self.canvas_image = None self.selection_start = None self.selection_rect = None self.crop_box = None self.output_dir = "" # 创建界面 self.create_widgets() def create_widgets(self): # 顶部控制区 top_frame = ttk.Frame(self.root, padding=10) top_frame.pack(fill=tk.X) ttk.Button(top_frame, text="选择图片文件夹", command=self.select_image_folder).pack(side=tk.LEFT, padx=5) ttk.Button(top_frame, text="设置输出文件夹", command=self.set_output_folder).pack(side=tk.LEFT, padx=5) self.status_var = tk.StringVar(value="请选择图片文件夹") ttk.Label(top_frame, textvariable=self.status_var).pack(side=tk.LEFT, padx=10) # 中部画布区 self.canvas_frame = ttk.Frame(self.root) self.canvas_frame.pack(fill=tk.BOTH, expand=True, padx=10, pady=10) self.canvas = tk.Canvas(self.canvas_frame, bg="white", highlightthickness=1) self.canvas.pack(fill=tk.BOTH, expand=True) # 底部控制区 bottom_frame = ttk.Frame(self.root, padding=10) bottom_frame.pack(fill=tk.X) ttk.Button(bottom_frame, text="上一张", command=self.prev_image).pack(side=tk.LEFT, padx=5) ttk.Button(bottom_frame, text="下一张", command=self.next_image).pack(side=tk.LEFT, padx=5) ttk.Label(bottom_frame, text="裁剪尺寸:").pack(side=tk.LEFT, padx=5) self.width_var = tk.StringVar(value="200") self.height_var = tk.StringVar(value="200") ttk.Entry(bottom_frame, textvariable=self.width_var, width=5).pack(side=tk.LEFT, padx=2) ttk.Label(bottom_frame, text="x").pack(side=tk.LEFT, padx=2) ttk.Entry(bottom_frame, textvariable=self.height_var, width=5).pack(side=tk.LEFT, padx=2) ttk.Button(bottom_frame, text="应用裁剪尺寸", command=self.apply_custom_crop_size).pack(side=tk.LEFT, padx=5) ttk.Button(bottom_frame, text="批量裁剪全部图片", command=self.batch_crop_images).pack(side=tk.LEFT, padx=5) self.image_counter = tk.StringVar(value="0/0") ttk.Label(bottom_frame, textvariable=self.image_counter).pack(side=tk.RIGHT, padx=10) # 绑定画布事件 self.canvas.bind("<Button-1>", self.on_canvas_click) self.canvas.bind("<B1-Motion>", self.on_canvas_drag) self.canvas.bind("<ButtonRelease-1>", self.on_canvas_release) # 滚动支持 self.canvas_frame.bind("<Configure>", self.on_canvas_configure) self.canvas.bind("<MouseWheel>", self.on_mousewheel) def select_image_folder(self): folder = filedialog.askdirectory(title="选择图片文件夹") if folder: self.images = [] for filename in os.listdir(folder): if filename.lower().endswith(('.png', '.jpg', '.jpeg', '.bmp', '.gif')): self.images.append(os.path.join(folder, filename)) if self.images: self.current_image_index = 0 self.load_image() self.status_var.set(f"已加载 {len(self.images)} 张图片") self.image_counter.set(f"{self.current_image_index + 1}/{len(self.images)}") else: messagebox.showinfo("提示", "所选文件夹中没有找到支持的图片格式") def set_output_folder(self): folder = filedialog.askdirectory(title="设置输出文件夹") if folder: self.output_dir = folder self.status_var.set(f"输出文件夹已设置: {folder}") def load_image(self): if 0 <= self.current_image_index < len(self.images): try: self.current_image = Image.open(self.images[self.current_image_index]) self.display_image() self.crop_box = None self.selection_start = None self.selection_rect = None except UnidentifiedImageError: messagebox.showerror("错误", f"无法打开图片: {self.images[self.current_image_index]}") self.next_image() def display_image(self): # 清除画布 self.canvas.delete("all") # 调整图片大小以适应画布 canvas_width = self.canvas.winfo_width() canvas_height = self.canvas.winfo_height() if canvas_width <= 1 or canvas_height <= 1: # 画布尚未完全初始化 return img_width, img_height = self.current_image.size ratio = min(canvas_width / img_width, canvas_height / img_height) display_width = int(img_width * ratio) display_height = int(img_height * ratio) self.current_photo = ImageTk.PhotoImage( self.current_image.resize((display_width, display_height), Image.LANCZOS)) self.canvas_image = self.canvas.create_image(canvas_width // 2, canvas_height // 2, image=self.current_photo) # 如果有裁剪框，重新绘制 if self.crop_box: x1, y1, x2, y2 = self.crop_box # 转换为画布坐标 x1_canvas = x1 * ratio + canvas_width // 2 - display_width // 2 y1_canvas = y1 * ratio + canvas_height // 2 - display_height // 2 x2_canvas = x2 * ratio + canvas_width // 2 - display_width // 2 y2_canvas = y2 * ratio + canvas_height // 2 - display_height // 2 self.selection_rect = self.canvas.create_rectangle( x1_canvas, y1_canvas, x2_canvas, y2_canvas, outline="red", width=2 ) def on_canvas_click(self, event): if not self.current_image: return self.selection_start = (event.x, event.y) # 如果已经有选择框，删除它 if self.selection_rect: self.canvas.delete(self.selection_rect) self.selection_rect = None def on_canvas_drag(self, event): if not self.current_image or not self.selection_start: return # 如果已经有选择框，删除它 if self.selection_rect: self.canvas.delete(self.selection_rect) # 创建新的选择框 x1, y1 = self.selection_start x2, y2 = event.x, event.y self.selection_rect = self.canvas.create_rectangle(x1, y1, x2, y2, outline="red", width=2) def on_canvas_release(self, event): if not self.current_image or not self.selection_start: return # 获取选择框的坐标 x1, y1 = self.selection_start x2, y2 = event.x, event.y # 确保坐标按正确顺序排列 x1, x2 = min(x1, x2), max(x1, x2) y1, y2 = min(y1, y2), max(y1, y2) # 转换为图像坐标 canvas_width = self.canvas.winfo_width() canvas_height = self.canvas.winfo_height() img_width, img_height = self.current_image.size ratio = min(canvas_width / img_width, canvas_height / img_height) img_x1 = int((x1 - canvas_width // 2 + img_width * ratio // 2) / ratio) img_y1 = int((y1 - canvas_height // 2 + img_height * ratio // 2) / ratio) img_x2 = int((x2 - canvas_width // 2 + img_width * ratio // 2) / ratio) img_y2 = int((y2 - canvas_height // 2 + img_height * ratio // 2) / ratio) # 确保坐标在图像范围内 img_x1 = max(0, min(img_x1, img_width)) img_y1 = max(0, min(img_y1, img_height)) img_x2 = max(0, min(img_x2, img_width)) img_y2 = max(0, min(img_y2, img_height)) # 保存裁剪框 self.crop_box = (img_x1, img_y1, img_x2, img_y2) # 显示裁剪尺寸信息 width = img_x2 - img_x1 height = img_y2 - img_y1 self.status_var.set(f"已选择裁剪区域: {width}x{height} 像素") def apply_custom_crop_size(self): try: width = int(self.width_var.get()) height = int(self.height_var.get()) if width <= 0 or height <= 0: messagebox.showerror("错误", "裁剪尺寸必须为正整数") return if not self.crop_box: messagebox.showinfo("提示", "请先在图片上选择一个裁剪区域") return # 获取当前裁剪框的中心点 x1, y1, x2, y2 = self.crop_box center_x = (x1 + x2) // 2 center_y = (y1 + y2) // 2 # 计算新的裁剪框坐标 new_x1 = max(0, center_x - width // 2) new_y1 = max(0, center_y - height // 2) new_x2 = min(self.current_image.width, new_x1 + width) new_y2 = min(self.current_image.height, new_y1 + height) # 调整裁剪框大小，保持中心点不变 if new_x2 - new_x1 < width: diff = width - (new_x2 - new_x1) new_x1 = max(0, new_x1 - diff // 2) new_x2 = new_x1 + width if new_y2 - new_y1 < height: diff = height - (new_y2 - new_y1) new_y1 = max(0, new_y1 - diff // 2) new_y2 = new_y1 + height self.crop_box = (new_x1, new_y1, new_x2, new_y2) # 刷新显示 self.display_image() self.status_var.set(f"已应用自定义裁剪尺寸: {width}x{height} 像素") except ValueError: messagebox.showerror("错误", "请输入有效的整数尺寸") def crop_current_image(self): if not self.crop_box: messagebox.showinfo("提示", "请先选择裁剪区域") return if not self.output_dir: messagebox.showinfo("提示", "请先设置输出文件夹") return x1, y1, x2, y2 = self.crop_box cropped_img = self.current_image.crop((x1, y1, x2, y2)) # 保存裁剪后的图片 filename = os.path.basename(self.images[self.current_image_index]) name, ext = os.path.splitext(filename) output_path = os.path.join(self.output_dir, f"{name}_cropped{ext}") try: # 保存时保持原画质 if ext.lower() in ('.jpg', '.jpeg'): cropped_img.save(output_path, quality=100, subsampling=0) else: cropped_img.save(output_path) self.status_var.set(f"已裁剪并保存图片: {output_path}") return True except Exception as e: messagebox.showerror("错误", f"保存图片时出错: {str(e)}") return False def batch_crop_images(self): if not self.crop_box: messagebox.showinfo("提示", "请先在第一张图片上选择裁剪区域") return if not self.output_dir: messagebox.showinfo("提示", "请先设置输出文件夹") return total = len(self.images) success_count = 0 error_count = 0 # 保存当前状态 current_index = self.current_image_index original_crop_box = self.crop_box # 批量处理所有图片 for i, img_path in enumerate(self.images): try: # 更新状态 self.status_var.set(f"正在处理图片 {i + 1}/{total}...") self.root.update() # 打开图片 self.current_image = Image.open(img_path) # 使用相同的裁剪框 self.crop_box = original_crop_box # 裁剪并保存 x1, y1, x2, y2 = self.crop_box cropped_img = self.current_image.crop((x1, y1, x2, y2)) filename = os.path.basename(img_path) name, ext = os.path.splitext(filename) output_path = os.path.join(self.output_dir, f"{name}_cropped{ext}") if ext.lower() in ('.jpg', '.jpeg'): cropped_img.save(output_path, quality=100, subsampling=0) else: cropped_img.save(output_path) success_count += 1 except Exception as e: error_count += 1 print(f"处理图片 {img_path} 时出错: {str(e)}") # 恢复状态 self.current_image_index = current_index self.load_image() # 显示结果 result_msg = f"批量裁剪完成！\n成功: {success_count} 张\n失败: {error_count} 张" self.status_var.set(result_msg) messagebox.showinfo("批量裁剪结果", result_msg) def prev_image(self): if self.images and self.current_image_index > 0: self.current_image_index -= 1 self.load_image() self.image_counter.set(f"{self.current_image_index + 1}/{len(self.images)}") def next_image(self): if self.images and self.current_image_index < len(self.images) - 1: self.current_image_index += 1 self.load_image() self.image_counter.set(f"{self.current_image_index + 1}/{len(self.images)}") def on_canvas_configure(self, event): # 画布大小改变时重绘图像 if self.current_image: self.display_image() def on_mousewheel(self, event): # 处理鼠标滚轮缩放 if self.current_image: # 缩放逻辑可以在这里实现 pass if __name__ == "__main__": root = tk.Tk() app = ImageCropperApp(root) root.mainloop()