imagesplit.py

#!/usr/bin/env python3
# -*- coding: utf-8 -*-
# @Date     : 2020-2-4 10:54:44
# @Author   : Albert Shi
# @Link     : http://blog.csdn.net/albertsh
# @Github   : https://github.com/AlbertGithubHome
__author__ = 'AlbertS'
# @Subject  : split images

import os
import tkinter as tk
from PIL import Image, ImageTk, ImageDraw
from tkinter.filedialog import askopenfilename

def imsplit(input_file_name, row_part, col_part):
    img = Image.open(input_file_name)
    w, h = img.size
    print('width = %d, height = %d' % (w, h))

    file_path,temp_file_name = os.path.split(input_file_name)
    file_name,ext = os.path.splitext(temp_file_name)

    avg_width_pixel = w // col_part
    avg_height_pixel = h // row_part

    for r in range(row_part):
        for c in range(col_part):
            end_width_pixel = (c + 1) * avg_width_pixel if c < col_part - 1 else w
            end_height_pixel = (r + 1) * avg_height_pixel if r < row_part - 1 else h
            box = (c * avg_width_pixel, r * avg_height_pixel, end_width_pixel, end_height_pixel)
            output_file = os.path.join(file_path, file_name + '_' + str(r) + '_' + str(c) + ext)
            print(output_file)
            img.crop(box).save(output_file, 'JPEG' if ext.lower() == '.jpg' else ext.upper()[1:])


def main():
    root = Tk()
    #设置窗口的大小宽x高+偏移量
    root.geometry('500x300+500+200')
    #设置窗口标题
    root.title('图片切割程序')
    # root.overrideredirect(1)
    #设置窗口图标
    # root.iconbitmap('spider_128px_1169260_easyicon.net.ico')
    root.mainloop()

main_w = 800
main_h = 600
class App(tk.Frame):
    def __init__(self, master=None):
        super().__init__(master, width=main_w, height=main_h)
        self.main_w = main_w
        self.main_h = main_h
        self.pack()

        self.path=tk.StringVar()
        self.row_part = tk.StringVar(value='3')
        self.col_part = tk.StringVar(value='1')
        tk.Entry(self,state='readonly',text=self.path,width=90).grid(row=0,column=0,columnspan=4)#,sticky=tk.E+tk.W)
        tk.Button(self,text='选择图片',command=self.choose_pic).grid(row=0,column=4,sticky=tk.W)

        spilt_num_changed = self.register(self.spilt_num_changed)
        tk.Label(self, text='切割行数：').grid(row=1,column=0,sticky=tk.E)
        self.entry_row=tk.Entry(self,text=self.path, textvariable=self.row_part,
              validate='focusout',
              validatecommand=(spilt_num_changed, '%P', '%v', '%W')
              # %P表示 当输入框的值允许改变，该值有效。该值为当前文本框内容
              # %v(小写大写不一样的)，当前validate的值
              # %W表示该组件的名字)
              )#,width=10)

        self.entry_row.grid(row=1,column=1,sticky=tk.W)
        tk.Label(self, text='切割列数：').grid(row=1,column=2,sticky=tk.E)
        self.entry_col=tk.Entry(self,text=self.path, textvariable=self.col_part,
              validate='focusout',
              validatecommand=(spilt_num_changed, '%P', '%v', '%W')
              )
        self.entry_col.grid(row=1,column=3,sticky=tk.W)
        tk.Button(self,text='进行切割',command=self.img_split).grid(row=1,column=4,sticky=tk.W)

    def spilt_num_changed(self, content, reason, name):
        # 光标失去时，这个函数就会执行
        #print(content, reason, name)
        try:
            self.show_image();
        except:
            pass
        return True

    def choose_pic(self):
        self.img_path=askopenfilename()
        try:
            self.show_image();
            self.path.set(self.img_path)
        except:
            self.path.set('please select a image!')

    def resize_and_drawline(self, pil_image):
        w, h = pil_image.size
        padding = 20
        f1 = (1.0*self.main_w-padding)/w
        f2 = (1.0*self.main_h-padding)/h
        factor = min([f1, f2])
        width = int(w*factor)
        height = int(h*factor)
        #print(factor)
        pil_image=pil_image.resize((width, height), Image.ANTIALIAS)

        self.row_part = self.entry_row.get();
        self.col_part = self.entry_col.get();
        row_part = int(self.row_part)
        col_part = int(self.col_part)

        avg_width_pixel = width // col_part
        avg_height_pixel = height // row_part
        pil_image = pil_image.convert('RGB')
        draw =ImageDraw.Draw(pil_image)

        for r in range(row_part):
            draw.line((0,(r+1)*avg_height_pixel) + (width,(r+1)*avg_height_pixel), fill='red')
        for c in range(col_part):
            draw.line(((c+1)*avg_width_pixel,0) + ((c+1)*avg_width_pixel,height), fill='red')

        #draw.line((0,0) + (width,height), fill='red')
        return pil_image

    def show_image(self):
        src_image = Image.open(self.img_path)
        resized_image = self.resize_and_drawline(src_image);
        self.tk_image = ImageTk.PhotoImage(image=resized_image)
        tk.Label(self, image=self.tk_image).grid(row=2,column=0,columnspan=5,ipady=15)

    def img_split(self):
        self.row_part = self.entry_row.get();
        self.col_part = self.entry_col.get();
        print(self.row_part)
        try:
            imsplit(self.img_path, int(self.row_part), int(self.col_part))
        except:
            self.path.set('please select a image!')


def mainApp(w, h):
    root = tk.Tk()
    root.geometry('{0}x{1}+400+200'.format(w, h))
    root.resizable(0,0)
    root.title('图片切割器')
    #root.overrideredirect(1)
    root.iconbitmap('icon.ico')
    app = App(root)
    app.mainloop()

if __name__ == '__main__':
    # imsplit('./20200204080200.jpg', 3, 1);
    mainApp(main_w, main_h)