from PIL import Image as Im
from PIL import ImageTk
import tkinter as tk  # 导入 Tkinter 库
from tkinter.filedialog import askopenfilename
import tkinter.messagebox
import tkinter.ttk
import tkinter.messagebox
from PIL import ImageDraw
from PIL import ImageFont
from PIL import ImageEnhance
from hig_class import *
from extra import wlw
import numpy as np
import tempfile
import os

class Win(tk.Tk):
    def __init__(self):
        super().__init__()
        self.title('图像处理工具')
        self.geometry('1080x720')
        self.picture = None  # self.picture将作为picture类的实例化对象
        self.img = None  # self.img将作为窗口类中一直变动的PIL对象图片
        self.file_entry = None
        self.setupUI()

    def setupUI(self):
        # 右边菜单栏
        right_f = tk.Frame(self, height=720, width=360, bg="#F8F8FF")
        right_f.pack(side=tk.RIGHT)

        # 各种功能按钮名称及位置
        btn1 = tk.Button(right_f, text='打开图像', bg="#BFEFFF", command=self.openToshow)
        btn1.place(y=25, x=30, width=300, height=40)
        btn2 = tk.Button(right_f, text='截图', bg="#BFEFFF", command=self.window_cut)
        btn2.place(y=85, x=30, width=144, height=40)
        btn3 = tk.Button(right_f, text='大小', bg="#BFEFFF", command=self.window_size)
        btn3.place(y=85, x=186, width=144, height=40)
        btn4 = tk.Button(right_f, text='旋转', bg='#B0E2FF', command=self.window_rotate)
        btn4.place(y=145, x=30, width=144, height=40)
        btn5 = tk.Button(right_f, text='翻转', bg='#B0E2FF', command=self.window_mirror)
        btn5.place(y=145, x=186, width=144, height=40)
        btn6 = tk.Button(right_f, text='灰度变换', bg='#B0E2FF', command=self.window_word)
        btn6.place(y=205, x=30, width=144, height=40)
        btn7 = tk.Button(right_f, text='滤镜', bg='#B0E2FF', command=self.window_style)
        btn7.place(y=205, x=186, width=144, height=40)
        btn8 = tk.Button(right_f, text='显示直方图', bg='#63B8FF', command=self.toshowhist)
        btn8.place(y=265, x=30, width=144, height=40)
        btn9 = tk.Button(right_f, text='图像相加', bg='#63B8FF', command=self.openToshow2)
        btn9.place(y=265, x=186, width=144, height=40)
        btn10 = tk.Button(right_f, text='滤波', bg='#63B8FF', command=self.openfilter)
        btn10.place(y=325, x=30, width=144, height=40)
        btn11 = tk.Button(right_f, text='Sobel锐化', bg='#63B8FF', command=self.opensharpen)
        btn11.place(y=325, x=186, width=144, height=40)

        # 各种调整名称及位置
        lb1 = tk.Label(right_f, text='亮 度:', bg="#F8F8FF")
        lb1.place(y=395, x=55, width=60, height=30)
        self.inp1 = tk.Entry(right_f)
        self.inp1.place(y=395, x=105, width=90, height=30)
        lb2 = tk.Label(right_f, text='%', bg="#F8F8FF")
        lb2.place(y=395, x=205, width=13, height=30)
        b1 = tk.Button(right_f, text='确定', fg='#F8F8FF', width=8, command=self.brightnessPic, bg="#7EC0EE")
        b1.place(y=395, x=240)

        lb3 = tk.Label(right_f, text='色彩度:', bg="#F8F8FF")
        lb3.place(y=440, x=55, width=60, height=30)
        self.inp2 = tk.Entry(right_f)
        self.inp2.place(y=440, x=105, width=90, height=30)
        lb4 = tk.Label(right_f, text='%', bg="#F8F8FF")
        lb4.place(y=440, x=205, width=13, height=30)
        b2 = tk.Button(right_f, text='确定', fg='#F8F8FF', width=8, command=self.coolorPic, bg="#7EC0EE")
        b2.place(y=444, x=240)

        lb5 = tk.Label(right_f, text='对比度:', bg="#F8F8FF")
        lb5.place(y=485, x=55, width=60, height=30)
        self.inp3 = tk.Entry(right_f)
        self.inp3.place(y=485, x=105, width=90, height=30)
        lb6 = tk.Label(right_f, text='%', bg="#F8F8FF")
        lb6.place(y=485, x=205, width=13, height=30)
        b3 = tk.Button(right_f, text='确定', fg='#F8F8FF', width=8, command=self.contrastPic, bg="#7EC0EE")
        b3.place(y=489, x=240)

        lb7 = tk.Label(right_f, text='锐 度:', bg="#F8F8FF")
        lb7.place(y=530, x=55, width=60, height=30)
        self.inp4 = tk.Entry(right_f)
        self.inp4.place(y=530, x=105, width=90, height=30)
        lb8 = tk.Label(right_f, text='%', bg="#F8F8FF")
        lb8.place(y=530, x=205, width=13, height=30)
        b4 = tk.Button(right_f, text='确定', fg='#F8F8FF', width=8, command=self.sharpnessPic, bg="#7EC0EE")
        b4.place(y=534, x=240)

        # 底部恢复、保存、对比、人脸识别
        btn18 = tk.Button(right_f, text='保存图像', fg='#F8F8FF', command=self.save_pic, bg="#1E90FF")
        btn18.place(y=590, x=200, width=100, height=30)
        btn19 = tk.Button(right_f, text='恢复图像', fg='#F8F8FF', command=self.replay, bg="#1E90FF")
        btn19.place(y=590, x=30)
        btn20 = tk.Button(right_f, text='对比图像', fg='#F8F8FF', command=self.compare, bg="#1E90FF")
        btn20.place(y=590, x=120)
        btn21 = tk.Button(right_f, text='人脸识别', fg='#F8F8FF', command=self.facedetec, bg="#1874CD")
        btn21.place(y=640, x=30)

        # 右侧图像显示栏
        right_f = tk.Frame(self, height=720, width=720)
        right_f.pack(side=tk.RIGHT)
        self.image_l = tk.Label(right_f, relief='ridge')
        self.image_l.place(x=0, y=0, width=720, height=720)



    def openfilter(self):
        filter_win = tk.Toplevel()
        filter_win.title('滤波选择')
        filter_win.geometry('150x150')
        b1 = tk.Button(filter_win, text='均值滤波', command=self.filter1)
        b1.place(y=30, x=35, width=75)
        b2 = tk.Button(filter_win, text='中值滤波', command=self.filter2)
        b2.place(y=60, x=35, width=75)
        b3 = tk.Button(filter_win, text='完成', command=filter_win.destroy)
        b3.place(y=110, x=80, width=40)

    def opensharpen(self):
        sharpen_pic = self.img
        sharpen_pic = wlw.sharpen(sharpen_pic)
        self.show_img(sharpen_pic)

    # 打开图片时使用，传值（图）给展示函数
    def openToshow(self):
        address = self.getAddress()
        self.open_picToimg = self.picture.open_pic(address)
        self.firstPic(self.open_picToimg)
        self.show_img(self.open_picToimg)

    # 打开图片时使用，传值（图）给展示函数
    def openToshow2(self):
        address2 = self.getAddress2()
        self.open_picToimg2 = self.picture.open_pic(address2)
        self.firstPic2(self.open_picToimg2)
        self.show_img2(self.open_picToimg2)

    def filter1(self):
        img_show = wlw.filter1(self.img)
        self.image_l.config(image=img_show)
        self.image_l.image = img_show
        return self.img

    def filter2(self):
        img_show = wlw.filter2(self.img)
        self.image_l.config(image=img_show)
        self.image_l.image = img_show
        return self.img

    # 打开图片时使用，获得地址
    def getAddress(self):
        path = tk.StringVar()
        self.file_entry = tk.Entry(self, state='readonly', text=path)
        path_ = askopenfilename()
        path.set(path_)
        self.picture = picture()
        return self.file_entry.get()

    # 打开图片时使用，获得地址
    def getAddress2(self):
        path2 = tk.StringVar()
        file_entry2 = tk.Entry(self, state='readonly', text=path2)
        path_2 = askopenfilename()
        path2.set(path_2)
        self.picture2 = picture()
        return file_entry2.get()

    # 展示函数
    def show_img(self, n_img):
        self.img = n_img  # self.img PIL对象方便传值给picture类以及本类中其他需要使用PIL图像的地方
        img_show = ImageTk.PhotoImage(self.img)
        self.image_l.config(image=img_show)
        self.image_l.image = img_show
        return self.img

    def show_img2(self, n_img):
        self.img2 = n_img  # self.img PIL对象方便传值给picture类以及本类中其他需要使用PIL图像的地方
        img_show2 = self.img2

        img_show2 = wlw.Add(img_show2, self.Fpic)
        self.image_l.config(image=img_show2)
        self.image_l.image = img_show2
        return self.img2

    # 保存函数
    def save_pic(self):
        fname = tkinter.filedialog.asksaveasfilename(title='保存文件', filetypes=[("PNG", ".png")])
        self.img.save(str(fname))  # PIL保存

    # 打开图像调用
    def open_pic(self, address):
        self.pic_get = Im.open(address).convert('RGBA')
        wid, hei = self.pic_get.size
        if wid > 600 or hei > 400:
            if tk.messagebox.askokcancel('提示', '图片可能过大，是否压缩？'):
                needShow_pic = self.openResize()
                return needShow_pic
            return self.pic_get
        else:
            return self.pic_get

    # 原图储存
    def firstPic(self, pic):
        self.Fpic = pic
        return self.Fpic

    # 原图储存
    def firstPic2(self, pic):
        self.Fpic2 = pic
        return self.Fpic2

    # 打开图像时的图像压缩展示
    def openResize(self):
        w, h = self.pic_get.size
        w_hope = 500
        h_hope = 300
        f1 = 1.0 * w_hope / w
        f2 = 1.0 * h_hope / h
        factor = min([f1, f2])
        width = int(w * factor)
        height = int(h * factor)
        pic_show = self.pic_get.resize((width, height))
        return pic_show

    # 截图操作页面
    def window_cut(self):
        Cut_win = tk.Toplevel()
        Cut_win.title('截图操作')
        Cut_win.geometry('220x380')
        nowPic = self.img
        if self.img == None:
            lNone = tk.Label(Cut_win, text='请先打开一张图片')
            lNone.place(y=55, x=50)
        else:
            wNow, hNow = nowPic.size
            l1 = tk.Label(Cut_win, text="此时图片尺寸:")
            l1.place(y=30, x=25)
            l2 = tk.Label(Cut_win, text=wNow)
            l2.place(y=55, x=25)
            l3 = tk.Label(Cut_win, text="X")
            l3.place(y=55, x=65)
            l4 = tk.Label(Cut_win, text=hNow)
            l4.place(y=55, x=85)
            l5 = tk.Label(Cut_win, text="截图区域")
            l5.place(y=85, x=25)
            l6 = tk.Label(Cut_win, text="起始点：（左上为（0，0））")
            l6.place(y=115, x=25)
            l7 = tk.Label(Cut_win, text="x：")
            l7.place(y=145, x=25)
            self.e1 = tk.Entry(Cut_win, width=10)
            self.e1.place(y=145, x=55)
            l8 = tk.Label(Cut_win, text="y：")
            l8.place(y=180, x=25)
            self.e2 = tk.Entry(Cut_win, width=10)
            self.e2.place(y=180, x=55)
            l9 = tk.Label(Cut_win, text="终止点：")
            l9.place(y=220, x=25)
            l10 = tk.Label(Cut_win, text="x：")
            l10.place(y=250, x=25)
            self.e3 = tk.Entry(Cut_win, width=10)
            self.e3.place(y=250, x=55)
            l11 = tk.Label(Cut_win, text="y：")
            l11.place(y=285, x=25)
            self.e4 = tk.Entry(Cut_win, width=10)
            self.e4.place(y=285, x=55)
            b1 = tk.Button(Cut_win, text='确定', command=self.getCutpart)
            b1.place(y=320, x=80, width=40)
            b2 = tk.Button(Cut_win, text='完成', command=Cut_win.destroy)
            b2.place(y=350, x=150, width=40)

    # 大小尺寸操作窗口
    def window_size(self):
        Size_win = tk.Toplevel()
        Size_win.title('尺寸操作')
        Size_win.geometry('200x180')
        l1 = tk.Label(Size_win, text="宽:")
        l1.place(y=30, x=25)
        self.text1 = tk.Entry(Size_win, width=10)
        self.text1.place(y=25, x=50)
        l1_1 = tk.Label(Size_win, text="px")
        l1_1.place(y=28, x=150)
        l2 = tk.Label(Size_win, text="高:")
        l2.place(y=60, x=25)
        self.text2 = tk.Entry(Size_win, width=10)
        self.text2.place(y=55, x=50)
        l2_1 = tk.Label(Size_win, text="px")
        l2_1.place(y=58, x=150)
        b1 = tk.Button(Size_win, text='确定', command=self.getSize_change)
        b1.place(y=100, x=80, width=40)
        b2 = tk.Button(Size_win, text='完成', command=Size_win.destroy)
        b2.place(y=145, x=140, width=40)

    # 接收截图区域,传送展示
    def getCutpart(self):
        nCut_pic = self.img
        x = int(self.e1.get())
        y = int(self.e2.get())
        xl = int(self.e3.get())
        yl = int(self.e4.get())
        self.picture = picture()
        showCut_pic = self.picture.Cutpic(nCut_pic, x, y, xl, yl)
        self.show_img(showCut_pic)

    # 获得输入尺寸
    def getSize_change(self):
        sizeNum_w = int(self.text1.get())
        sizeNum_h = int(self.text2.get())
        # print('1')
        self.showSize_change(sizeNum_w, sizeNum_h)
        # print(sizeNum_w, sizeNum_h)

    # 尺寸修改并展示图片
    def showSize_change(self, renewSize_w, renewSize_h):
        # print('2')
        self.picture = picture()
        needResize_pic = self.img
        show_resizePic = self.picture.changeResize(needResize_pic, renewSize_w, renewSize_h)
        self.show_img(show_resizePic)

    # 灰度转换功能窗口
    def window_word(self):
        Word_win = tk.Toplevel()
        Word_win.title('灰度变化')
        Word_win.geometry('300x400')
        l1 = tk.Label(Word_win, text='请选择灰度转换方法：')
        l1.place(y=10, x=20, width=130)
        l0 = tk.Label(Word_win, text="n（n值）：")
        l0.place(y=40, x=20)
        self.i0 = tk.Entry(Word_win, width=13)  # 输入参数n
        self.i0.place(y=40, x=100)
        bt1 = tk.Button(Word_win, text='n值化', command=self.getn)
        bt1.place(y=70, x=20, width=80)

        l2 = tk.Label(Word_win, text="k（对比度）：")
        l2.place(y=110, x=20)
        self.i1 = tk.Entry(Word_win, width=13)  # 输入线性化参数k
        self.i1.place(y=110, x=100)
        l3 = tk.Label(Word_win, text="b（亮度）：")
        l3.place(y=130, x=20)
        self.i2 = tk.Entry(Word_win, width=13)  # 输入线性化参数b
        self.i2.place(y=130, x=100)
        bt2 = tk.Button(Word_win, text='线性化', command=self.getkb)
        bt2.place(y=160, x=20, width=80)

        l3 = tk.Label(Word_win, text="c（尺度比较常数）：")
        l3.place(y=200, x=20)
        self.i3 = tk.Entry(Word_win, width=13)  # 输入参数c
        self.i3.place(y=200, x=140)
        bt3 = tk.Button(Word_win, text='非线性化（对数转换)', command=self.getc)
        bt3.place(y=230, x=20, width=120)
        b7 = tk.Button(Word_win, text='完成', command=Word_win.destroy)
        b7.place(y=300, x=210)

    # 获得输入的n值转换值
    def getn(self):
        npic = self.img
        n = float(self.i0.get())
        # 图像线性化获取展示
        npic = wlw.tonpic(npic, n)
        self.show_img(npic)

    # 获得输入的线性转换值
    def getkb(self):
        linearpic = self.img
        k = float(self.i1.get())
        b = float(self.i2.get())
        # 图像线性化获取展示
        linearpic = wlw.linearization(linearpic, k, b)
        self.show_img(linearpic)

    # 获得输入的对数转换值
    def getc(self):
        logpic = self.img
        c = float(self.i3.get())
        # 图像对数化获取展示
        logpic = wlw.tologpic(logpic, c)
        self.show_img(logpic)

    # 图像旋转操作窗口
    def window_rotate(self):
        Rot_win = tk.Toplevel()
        Rot_win.title('旋转')
        Rot_win.geometry('225x220')
        l1 = tk.Label(Rot_win, text="角度：")
        l1.place(y=20, x=10)
        self.inpt = tk.Entry(Rot_win, width=13)
        self.inpt.place(y=15, x=50)
        b0 = tk.Button(Rot_win, text='确定', command=lambda: self.getDegree('1'))
        b0.place(y=55, x=160, width=40)
        b1 = tk.Button(Rot_win, text='+90', command=lambda: self.getDegree('2'))
        b1.place(y=85, x=60, width=95)
        b2 = tk.Button(Rot_win, text='-90', command=lambda: self.getDegree('3'))
        b2.place(y=115, x=60, width=95)
        b3 = tk.Button(Rot_win, text='180', command=lambda: self.getDegree('4'))
        b3.place(y=145, x=60, width=95)
        b4 = tk.Button(Rot_win, text='完成', command=Rot_win.destroy)
        b4.place(y=180, x=160, width=40)

    # 旋转角度获取
    def getDegree(self, n):
        self.picture = picture()
        needRotate_pic = self.img
        # print('99')
        # print(n)
        if n == '1':
            shouldDegree = float(self.inpt.get())
            # print('36')
            # print(shouldDegree)
            showRotate_pic = self.picture.rotatePic(needRotate_pic, shouldDegree)
        elif n == '2':
            # print('34')
            showRotate_pic = self.picture.rotatePic(needRotate_pic, +90)
        elif n == '3':
            showRotate_pic = self.picture.rotatePic(needRotate_pic, -90)
        elif n == '4':
            showRotate_pic = self.picture.rotatePic(needRotate_pic, 180)
        else:
            return 0
        self.show_img(showRotate_pic)

    # 滤镜选择窗口
    def window_style(self):
        Sty_win = tk.Toplevel()
        Sty_win.title('滤镜选择')
        Sty_win.geometry('230x180')
        bt1 = tk.Button(Sty_win, text='图像模糊', command=self.sty_1)
        bt1.place(y=25, x=25, width=80)
        bt2 = tk.Button(Sty_win, text='轮廓滤波', command=self.sty_2)
        bt2.place(y=25, x=115, width=80)
        bt3 = tk.Button(Sty_win, text='高斯模糊', command=self.sty_3)
        bt3.place(y=65, x=25, width=80)
        bt4 = tk.Button(Sty_win, text='浮雕滤镜', command=self.sty_4)
        bt4.place(y=65, x=115, width=80)
        bt6 = tk.Button(Sty_win, text='完成', command=Sty_win.destroy)
        bt6.place(y=140, x=160)

    # 图像模糊获取展示
    def sty_1(self):
        sty_1_pic = self.img
        relSty_1 = wlw.blurPic(sty_1_pic)
        self.show_img(relSty_1)

    # 边界增强获取展示
    def sty_2(self):
        sty_2_pic = self.img
        reSty_2 = wlw.edge(sty_2_pic)
        self.show_img(reSty_2)

    # 高斯模糊获取展示
    def sty_3(self):
        sty_3_pic = self.img
        reSty_3 = wlw.gaussianBlur(sty_3_pic)
        self.show_img(reSty_3)

    # 浮雕滤镜获取展示
    def sty_4(self):
        sty_4_pic = self.img
        reSty_4 = wlw.emboss(sty_4_pic)
        self.show_img(reSty_4)

    # 亮度调整
    def brightnessPic(self):
        self.picture = picture()
        needBright_pic = self.img
        b_num = float(self.inp1.get())
        briNum = b_num / 100
        showBright_pic = self.picture.brightPic(needBright_pic, briNum)
        self.show_img(showBright_pic)

    # 色彩度调整
    def coolorPic(self):
        self.picture = picture()
        needColor_pic = self.img
        co_num = float(self.inp2.get())
        colNum = co_num / 100
        showColor_pic = self.picture.colornPic(needColor_pic, colNum)
        self.show_img(showColor_pic)

    # 对比度调整
    def contrastPic(self):
        self.picture = picture()
        needCon_pic = self.img
        c_num = float(self.inp3.get())
        ConNum = c_num / 100
        showContrast_pic = self.picture.constractPic(needCon_pic, ConNum)
        self.show_img(showContrast_pic)

    # 锐度调整
    def sharpnessPic(self):
        self.picture = picture()
        needSharp_pic = self.img
        s_num = float(self.inp4.get())
        ShNum = s_num / 100
        showSharp_pic = self.picture.constractPic(needSharp_pic, ShNum)
        self.show_img(showSharp_pic)

    # 镜像操作窗口
    def window_mirror(self):
        Mir_win = tk.Toplevel()
        Mir_win.title('镜像操作')
        Mir_win.geometry('150x150')
        b1 = tk.Button(Mir_win, text='左右', command=self.MirrorImg_lr)
        b1.place(y=30, x=35, width=75)
        b2 = tk.Button(Mir_win, text='上下', command=self.MirrorImg_tb)
        b2.place(y=60, x=35, width=75)
        b3 = tk.Button(Mir_win, text='完成', command=Mir_win.destroy)
        b3.place(y=110, x=80, width=40)

    # 镜像左右调用展示
    def MirrorImg_lr(self):
        self.picture = picture()
        Mirror_img_lr = self.img
        MittotImg_lrFinish = self.picture.MirrorPic_leftOrright(Mirror_img_lr)
        self.show_img(MittotImg_lrFinish)

    # 镜像上下调用展示
    def MirrorImg_tb(self):
        self.picture = picture()
        Mirror_img_tb = self.img
        MittotImg_tbFinish = self.picture.MirrorPic_topOrbuttom(Mirror_img_tb)
        self.show_img(MittotImg_tbFinish)

    # 恢复图像
    def replay(self):
        self.show_img(self.Fpic)

    # 对比图像
    def compare(self):
        Im._show(self.Fpic)

    def toshowhist(self):
        # gethist = wlw.showhist(self.img)
        # self.show_img(gethist)
        wlw.showhist(self.img)

    #人脸识别
    def facedetec(self):
        minc=self.img
        path=self.file_entry.get()
        minc = wlw.face_detect(minc, path)
        self.show_img(minc)
        print("图片路径：", path)

if __name__ == '__main__':
    root = Win()
    # 窗体主循环
    root.mainloop()