From f9277c0de84e325c3909726e4854d64a8740e097 Mon Sep 17 00:00:00 2001 From: QwejayHuang Date: Sun, 29 Dec 2024 10:08:59 +0800 Subject: [PATCH] Update ImageSplitter.py --- ImageSplitter.py | 256 ++++++++++++++++++++++++++++++++++------------- 1 file changed, 184 insertions(+), 72 deletions(-) diff --git a/ImageSplitter.py b/ImageSplitter.py index 4837ef7..5d88ce4 100644 --- a/ImageSplitter.py +++ b/ImageSplitter.py @@ -1,5 +1,5 @@ import tkinter as tk -from tkinter import filedialog, messagebox +from tkinter import filedialog from PIL import Image, ImageTk import fitz # PyMuPDF import os @@ -9,7 +9,6 @@ import webbrowser import ttkbootstrap as ttk from ttkbootstrap.constants import * -from PIL import Image import pillow_heif # 全局变量 @@ -23,29 +22,39 @@ total_pages = 1 split_percentage_label = None window_resize_id = None -grid_entry = None # 修复 grid_entry 未定义的问题 +grid_row_entry = None +grid_col_entry = None img_display_x = 0 img_display_y = 0 img_display_width = 0 img_display_height = 0 +split_color_var = None + +# 颜色映射字典 +color_mapping = { + '红色': 'red', + '蓝色': 'blue', + '绿色': 'green', + '黑色': 'black', + '黄色': 'yellow' +} # 函数定义 -def split_image(imgs, direction, grid_num=1): +def split_image(imgs, direction, grid_row=1, grid_col=1): """根据方向分割图像""" split_imgs = [] if direction == '多宫格': try: - grid_num = int(grid_num) - if grid_num < 1: + grid_row = int(grid_row_entry.get()) + grid_col = int(grid_col_entry.get()) + if grid_row < 1 or grid_col < 1: raise ValueError - rows = int(math.ceil(math.sqrt(grid_num))) - cols = int(math.ceil(grid_num / rows)) for img in imgs: img_width, img_height = img.size - cell_width = img_width // cols - cell_height = img_height // rows - for r in range(rows): - for c in range(cols): + cell_width = img_width // grid_col + cell_height = img_height // grid_row + for r in range(grid_row): + for c in range(grid_col): left = c * cell_width upper = r * cell_height right = left + cell_width @@ -53,20 +62,46 @@ def split_image(imgs, direction, grid_num=1): split_img = img.crop((left, upper, right, lower)) split_imgs.append(split_img) except ValueError: - set_status("宫格数必须为大于等于1的整数。", "danger") + set_status("行数和列数必须为大于等于1的整数。", "danger") return split_imgs elif direction == '垂直': - for img in imgs: - split_width = int(img.width * split_position) - left_img = img.crop((0, 0, split_width, img.height)) - right_img = img.crop((split_width, 0, img.width, img.height)) - split_imgs.extend([left_img, right_img]) + try: + grid_col = int(grid_col_entry.get()) + if grid_col < 1: + raise ValueError + for img in imgs: + img_width, img_height = img.size + cell_width = img_width // grid_col + cell_height = img_height + for c in range(grid_col): + left = c * cell_width + upper = 0 + right = left + cell_width + lower = img_height + split_img = img.crop((left, upper, right, lower)) + split_imgs.append(split_img) + except ValueError: + set_status("列数必须为大于等于1的整数。", "danger") + return split_imgs elif direction == '水平': - for img in imgs: - split_height = int(img.height * split_position) - top_img = img.crop((0, 0, img.width, split_height)) - bottom_img = img.crop((0, split_height, img.width, img.height)) - split_imgs.extend([top_img, bottom_img]) + try: + grid_row = int(grid_row_entry.get()) + if grid_row < 1: + raise ValueError + for img in imgs: + img_width, img_height = img.size + cell_width = img_width + cell_height = img_height // grid_row + for r in range(grid_row): + left = 0 + upper = r * cell_height + right = img_width + lower = upper + cell_height + split_img = img.crop((left, upper, right, lower)) + split_imgs.append(split_img) + except ValueError: + set_status("行数必须为大于等于1的整数。", "danger") + return split_imgs elif direction == '不分割': split_imgs.extend(imgs) return split_imgs @@ -123,16 +158,34 @@ def save_file(): return save_extension = save_format_var.get() direction = split_direction_var.get() - grid_num = grid_entry.get() + grid_row = grid_row_entry.get() + grid_col = grid_col_entry.get() if direction == '多宫格': try: - grid_num = int(grid_num) - if grid_num < 1: + grid_row = int(grid_row) + grid_col = int(grid_col) + if grid_row < 1 or grid_col < 1: raise ValueError except ValueError: - set_status("宫格数必须为大于等于1的整数。", "danger") + set_status("行数和列数必须为大于等于1的整数。", "danger") return - imgs_to_save = split_image(imgs, direction, grid_num) + elif direction == '垂直': + try: + grid_col = int(grid_col) + if grid_col < 1: + raise ValueError + except ValueError: + set_status("列数必须为大于等于1的整数。", "danger") + return + elif direction == '水平': + try: + grid_row = int(grid_row) + if grid_row < 1: + raise ValueError + except ValueError: + set_status("行数必须为大于等于1的整数。", "danger") + return + imgs_to_save = split_image(imgs, direction, grid_row, grid_col) save_images(imgs_to_save, save_extension) def load_file_in_background(file_path, file_extension): @@ -199,27 +252,48 @@ def draw_split_line(): image_canvas.delete("split_line") if imgs: direction = split_direction_var.get() + selected_color = split_color_var.get() # 获取选择的中文颜色 + color = color_mapping.get(selected_color, 'red') # 获取对应的英文颜色 if direction == '多宫格': try: - grid_num = int(grid_entry.get()) - rows = int(math.ceil(math.sqrt(grid_num))) - cols = int(math.ceil(grid_num / rows)) - cell_width = img_display_width / cols - cell_height = img_display_height / rows - for i in range(1, cols): + grid_row = int(grid_row_entry.get()) + grid_col = int(grid_col_entry.get()) + if grid_row < 1 or grid_col < 1: + raise ValueError + cell_width = img_display_width / grid_col + cell_height = img_display_height / grid_row + for i in range(1, grid_col): split_x = img_display_x + cell_width * i - image_canvas.create_line(split_x, img_display_y, split_x, img_display_y + img_display_height, fill='red', width=2, tags="split_line") - for j in range(1, rows): + image_canvas.create_line(split_x, img_display_y, split_x, img_display_y + img_display_height, fill=color, width=2, tags="split_line") + for j in range(1, grid_row): split_y = img_display_y + cell_height * j - image_canvas.create_line(img_display_x, split_y, img_display_x + img_display_width, split_y, fill='red', width=2, tags="split_line") + image_canvas.create_line(img_display_x, split_y, img_display_x + img_display_width, split_y, fill=color, width=2, tags="split_line") except ValueError: pass elif direction == '垂直': - split_x = img_display_x + img_display_width * split_position - image_canvas.create_line(split_x, img_display_y, split_x, img_display_y + img_display_height, fill='red', width=2, tags="split_line") + try: + grid_col = int(grid_col_entry.get()) + if grid_col < 1: + raise ValueError + cell_width = img_display_width / grid_col + for i in range(1, grid_col): + split_x = img_display_x + cell_width * i + image_canvas.create_line(split_x, img_display_y, split_x, img_display_y + img_display_height, fill=color, width=2, tags="split_line") + except ValueError: + pass elif direction == '水平': - split_y = img_display_y + img_display_height * split_position - image_canvas.create_line(img_display_x, split_y, img_display_x + img_display_width, split_y, fill='red', width=2, tags="split_line") + try: + grid_row = int(grid_row_entry.get()) + if grid_row < 1: + raise ValueError + cell_height = img_display_height / grid_row + for j in range(1, grid_row): + split_y = img_display_y + cell_height * j + image_canvas.create_line(img_display_x, split_y, img_display_x + img_display_width, split_y, fill=color, width=2, tags="split_line") + except ValueError: + pass + elif direction == '不分割': + pass def set_status(message, color="secondary"): """设置状态栏信息""" @@ -228,19 +302,35 @@ def set_status(message, color="secondary"): def update_split_direction(*args): """更新分割方向""" - global image_canvas, grid_entry # 声明全局变量 + global image_canvas, grid_row_entry, grid_col_entry direction = split_direction_var.get() - if direction in ['垂直', '水平']: - if grid_entry: - grid_entry['state'] = 'disabled' + if direction == '垂直': + grid_row_entry.delete(0, tk.END) + grid_row_entry.insert(0, "1") + grid_row_entry['state'] = 'disabled' + grid_col_entry.delete(0, tk.END) + grid_col_entry.insert(0, "2") + grid_col_entry['state'] = 'normal' + image_canvas.bind("", drag_split_line) + elif direction == '水平': + grid_row_entry.delete(0, tk.END) + grid_row_entry.insert(0, "2") + grid_row_entry['state'] = 'normal' + grid_col_entry.delete(0, tk.END) + grid_col_entry.insert(0, "1") + grid_col_entry['state'] = 'disabled' image_canvas.bind("", drag_split_line) elif direction == '多宫格': - if grid_entry: - grid_entry['state'] = 'normal' + grid_row_entry.delete(0, tk.END) + grid_row_entry.insert(0, "3") + grid_row_entry['state'] = 'normal' + grid_col_entry.delete(0, tk.END) + grid_col_entry.insert(0, "3") + grid_col_entry['state'] = 'normal' image_canvas.unbind("") else: - if grid_entry: - grid_entry['state'] = 'disabled' + grid_row_entry['state'] = 'disabled' + grid_col_entry['state'] = 'disabled' image_canvas.unbind("") draw_split_line() @@ -259,7 +349,6 @@ def drag_split_line(event): def on_drop(event): """处理拖放文件""" global file_path, file_extension - # 移除 split()，直接使用 strip('{}') file_path = event.data.strip('{}') file_extension = os.path.splitext(file_path)[1].lower() if file_extension in ['.pdf', '.jpg', '.jpeg', '.png', '.bmp', '.webp', '.heic']: @@ -270,12 +359,13 @@ def on_drop(event): def update_grid_lines(event): """当宫格数量变化时，更新分割线""" try: - grid_num = int(grid_entry.get()) - if grid_num < 1: + grid_row = int(grid_row_entry.get()) + grid_col = int(grid_col_entry.get()) + if grid_row < 1 or grid_col < 1: raise ValueError - draw_split_line() # 重新绘制分割线 + draw_split_line() except ValueError: - set_status("宫格数必须为大于等于1的整数。", "danger") + set_status("行数和列数必须为大于等于1的整数。", "danger") def get_original_image(file_path, file_extension): """加载原始图像或 PDF 页面""" @@ -340,8 +430,8 @@ def vertical_flip(): # 创建主窗口 root = TkinterDnD.Tk() -root.geometry("820x640") -root.title("Image Splitter v1.5 —— QwejayHuang") +root.geometry("880x680") +root.title("Image Splitter 2.0 —— QwejayHuang") # 初始化 ttkbootstrap 样式 style = ttk.Style("litera") @@ -356,26 +446,27 @@ def vertical_flip(): split_direction_var = tk.StringVar() split_direction_var.set('不分割') -# 先不设置 trace direction_menu_label = ttk.Label(top_menu, text="分割类型:", bootstyle="secondary") direction_menu_label.pack(side=LEFT, padx=5) direction_menu = ttk.OptionMenu(top_menu, split_direction_var, '不分割', '不分割', '垂直', '水平', '多宫格') direction_menu.pack(side=LEFT, padx=10, pady=5) -vcmd = (root.register(lambda value: value.isdigit() or value == ""), '%P') -grid_entry_label = ttk.Label(top_menu, text="宫格数:", bootstyle="secondary") -grid_entry_label.pack(side=LEFT, padx=5) -grid_entry = ttk.Entry(top_menu, width=5, validate='key', validatecommand=vcmd) -grid_entry.pack(side=LEFT, padx=5) -grid_entry.insert(0, "9") -grid_entry['state'] = 'disabled' +split_direction_var.trace('w', update_split_direction) -save_format_var = tk.StringVar() -save_format_var.set('.jpg') -save_format_label = ttk.Label(top_menu, text="保存格式:", bootstyle="secondary") -save_format_label.pack(side=LEFT, padx=5) -save_format_menu = ttk.OptionMenu(top_menu, save_format_var, '.JPG', '.JPG', '.PDF', '.PNG', '.BMP', '.WEBP', '.HEIC') -save_format_menu.pack(side=LEFT, padx=10, pady=5) +vcmd = (root.register(lambda value: value.isdigit() or value == ""), '%P') +grid_row_entry_label = ttk.Label(top_menu, text="行数:", bootstyle="secondary") +grid_row_entry_label.pack(side=LEFT, padx=5) +grid_row_entry = ttk.Entry(top_menu, width=5, validate='key', validatecommand=vcmd) +grid_row_entry.pack(side=LEFT, padx=5) +grid_row_entry.insert(0, "3") +grid_row_entry['state'] = 'disabled' # 默认不分割，行输入框变灰色 + +grid_col_entry_label = ttk.Label(top_menu, text="列数:", bootstyle="secondary") +grid_col_entry_label.pack(side=LEFT, padx=5) +grid_col_entry = ttk.Entry(top_menu, width=5, validate='key', validatecommand=vcmd) +grid_col_entry.pack(side=LEFT, padx=5) +grid_col_entry.insert(0, "3") +grid_col_entry['state'] = 'disabled' # 默认不分割，列输入框变灰色 # 创建dpi变量 dpi_var = tk.StringVar() @@ -391,6 +482,13 @@ def vertical_flip(): dpi_combobox.bind("<>", lambda event: set_status(f"DPI设置为: {dpi_var.get()}", "info")) dpi_combobox.bind("", lambda event: set_status(f"DPI设置为: {dpi_var.get()}", "info")) +save_format_var = tk.StringVar() +save_format_var.set('.jpg') +save_format_label = ttk.Label(top_menu, text="保存格式:", bootstyle="secondary") +save_format_label.pack(side=LEFT, padx=5) +save_format_menu = ttk.OptionMenu(top_menu, save_format_var, '.JPG', '.JPG', '.PDF', '.PNG', '.BMP', '.WEBP', '.HEIC') +save_format_menu.pack(side=LEFT, padx=10, pady=5) + save_button = ttk.Button(top_menu, text="保存文件", command=save_file, bootstyle="primary") save_button.pack(side=LEFT, padx=10, pady=5) @@ -433,6 +531,19 @@ def on_resize(event): button_inner_frame = ttk.Frame(button_frame) button_inner_frame.grid(row=0, column=2, padx=5, sticky='e') +# 新增颜色选择器 +split_color_var = tk.StringVar() +split_color_var.set('红色') # 默认颜色 +split_color_var.trace('w', lambda *args: draw_split_line()) # 绑定变量变化事件 + +# 添加标签 +color_label = ttk.Label(button_inner_frame, text="分割线颜色：", bootstyle="secondary") +color_label.pack(side=LEFT, padx=5) + +color_options = ['红色', '蓝色', '绿色', '黑色', '黄色'] +color_menu = ttk.OptionMenu(button_inner_frame, split_color_var, '红色', *color_options) +color_menu.pack(side=LEFT, padx=5) + rotate_button = ttk.Button(button_inner_frame, text="旋转", command=lambda: rotate_image(), bootstyle="secondary") rotate_button.pack(side=LEFT, padx=5) @@ -457,8 +568,9 @@ def on_resize(event): root.drop_target_register(DND_FILES) root.dnd_bind('<>', on_drop) -# 绑定宫格数量输入框的值变化事件 -grid_entry.bind('', update_grid_lines) +# 绑定行数和列数输入框的值变化事件 +grid_row_entry.bind('', update_grid_lines) +grid_col_entry.bind('', update_grid_lines) # 启动主循环 root.mainloop() \ No newline at end of file