search.py

#!/usr/bin/env python3
# -*- coding: utf-8 -*-
__author__ = 'jinmu333'

import threading
import tkinter as tk
from tkinter import ttk
from tkinter.filedialog import *
import tkinter.messagebox
from PIL import Image, ImageTk, ImageGrab
import requests
from hyperlpr import *
import cv2
from threading import Thread
from lib import img_math
import lib.img_function as predict
from lib.img_api import api_pic
import lib.screencut as screencut
from time import sleep
import time


class ThreadWithReturnValue(Thread):
    def __init__(self, group=None, target=None, name=None, args=(), kwargs=None, *, daemon=None):
        Thread.__init__(self, group, target, name, args, kwargs, daemon=daemon)
        self._return1 = None
        self._return2 = None
        self._return3 = None

    def run(self):
        if self._target is not None:
            try:
                self._return1, self._return2, self._return3 = self._target(*self._args, **self._kwargs)
            except:
                pass

    def join(self):
        Thread.join(self)
        return self._return1, self._return2, self._return3


class Login(ttk.Frame):
    def __init__(self, win):
        ttk.Frame.__init__(self, win)
        frame0 = ttk.Frame(self)
        frame1 = ttk.Frame(self)
        frame2 = ttk.Frame(self)
        frame3 = ttk.Frame(self)
        frame4 = ttk.Frame(self)
        win.title("车牌搜索系统")
        win.minsize(850, 600)
        self.center_window()
        self.s1 = StringVar()
        self.s2 = StringVar()
        self.matchflag = 0
        self.stopflag = 1
        self.pic_path = ""
        self.pic_path3 = ""
        self.thread_run2 = False

        self.pilImage = Image.open("pic/searchl.png")
        self.tkImage = ImageTk.PhotoImage(image=self.pilImage)
        self.image_ctl = tk.Label(frame0, image=self.tkImage)
        self.image_ctl.pack(side=LEFT)

        self.pilImage2 = Image.open("pic/searchr.png")
        self.tkImage2 = ImageTk.PhotoImage(image=self.pilImage2)
        self.image_ctl2 = tk.Label(frame0, image=self.tkImage2)
        self.image_ctl2.pack(side=RIGHT)

        frame0.pack(side=TOP, fill=tk.Y, expand=1)
        frame4.pack(side=TOP, fill=tk.Y, expand=1)
        frame1.pack(side=TOP, fill=tk.Y, expand=1)
        frame2.pack(side=TOP, fill=tk.Y, expand=1)
        frame3.pack(side=TOP, fill=tk.Y, expand=1)

        self.label = ttk.Label(frame1, text='要查找的车牌')
        self.label.pack(side=LEFT)
        self.input1 = ttk.Entry(frame1, textvariable=self.s1, width=30)
        self.input1.pack(side=LEFT)
        self.face_button1 = ttk.Button(frame1, text="选择文件", width=15, command=self.file1)
        self.face_button1.pack(side=RIGHT)
        self.cut_ctrl2 = ttk.Button(frame1, text="截图选取", width=15, command=self.cut_pic1)
        self.cut_ctrl2.pack(side=RIGHT)
        self.label2 = ttk.Label(frame2, text='查找路径: ')
        self.label2.pack(side=LEFT)
        self.input2 = ttk.Entry(frame2, textvariable=self.s2, width=30)
        self.input2.pack(side=LEFT)
        self.face_button2 = ttk.Button(frame2, text="选择文件夹", width=15, command=self.file2)
        self.face_button2.pack(side=RIGHT)
        self.clean_button = ttk.Button(frame3, text="重置显示信息", width=15, command=self.cut_clean)
        self.clean_button.pack(side=LEFT)
        self.file_pic2_button = ttk.Button(frame3, text="快速识别", width=15, command=self.file_pic2)
        self.file_pic2_button.pack(side=LEFT)
        self.url_face_button = ttk.Button(frame3, text="停止识别", width=15, command=self.stop)
        self.url_face_button.pack(side=LEFT)
        self.file_pic_button = ttk.Button(frame3, text="开始识别", width=15, command=self.file_pic)
        self.file_pic_button.pack(side=RIGHT)

        self.match = ttk.Label(frame4, text='', font=('Times', '20'))
        self.match.pack()

        self.pack(fill=tk.BOTH, expand=tk.YES, padx="10", pady="10")

        self.predictor = predict.CardPredictor()
        self.predictor.train_svm()

    def cut_pic1(self):
        log.state('icon')
        sleep(0.2)
        filename = "tmp/cut1.gif"
        im =ImageGrab.grab()
        im.save(filename)
        im.close()
        w = screencut.MyCapture(log, filename)
        self.cut_ctrl2.wait_window(w.top)
        log.state('normal')
        os.remove(filename)
        self.pic_path = "tmp/cut.png"
        self.pic_cut = Image.open(self.pic_path)
        self.pic_cut.save("tmp/cut1.png")
        self.pic_path = "tmp/cut1.png"
        self.s1.set(self.pic_path)
        self.pilImage3 = Image.open(self.pic_path)
        w, h = self.pilImage3.size
        pil_image_resized = self.resize(w, h, self.pilImage3)
        self.tkImage3 = ImageTk.PhotoImage(image=pil_image_resized)
        self.image_ctl.configure(image=self.tkImage3)

    def center_window(self):
        screenwidth = log.winfo_screenwidth()
        screenheight = log.winfo_screenheight()
        log.update()
        width = log.winfo_width()
        height = log.winfo_height()
        size = '+%d+%d' % ((screenwidth - width)/2, (screenheight - height)/2)
        log.geometry(size)

    def get_img_list(self, images_path):
        self.count = 0
        self.array_of_img = []
        for filename in os.listdir(images_path):
            #print(filename)
            try:
                self.pilImage3 = Image.open(images_path + "/" + filename)
                self.array_of_img.append(images_path + "/" + filename)
                self.count = self.count + 1
                # images_path2 = images_path + "/" + filename
                # self.pic_path2 = images_path2
            except:
                pass
        self.pic_pathstart = self.array_of_img[self.count-1]
        self.countstart = self.count
        print(self.array_of_img)

    def file1(self):
        self.pic_path = askopenfilename(title="选择识别图片", filetypes=[("jpeg图片", "*.jpeg"), ("jpg图片", "*.jpg"), ("png图片", "*.png")])
        self.s1.set(self.pic_path)
        self.pilImage3 = Image.open(self.pic_path)
        w, h = self.pilImage3.size
        pil_image_resized = self.resize(w, h, self.pilImage3)
        self.tkImage3 = ImageTk.PhotoImage(image=pil_image_resized)
        self.image_ctl.configure(image=self.tkImage3)

    def file2(self):
        self.pic_path3 = askdirectory(title="选择识别路径")
        self.s2.set(self.pic_path3)
        self.get_img_list(self.pic_path3)

    def stop(self):
        self.stopflag = 0
        print("stop")

    def file_pic(self):
        # print("file_pic")
        if (self.pic_path3 == ""):
            tkinter.messagebox.showinfo(title='车牌对比识别系统', message='路径不能为空')
            return
        if (self.pic_path == ""):
            if self.input1.get() == "":
                tkinter.messagebox.showinfo(title='车牌对比识别系统', message='图片1不能为空')
                return
            else:
                self.matchstr1 = self.input1.get()
                print(self.matchstr1)
        else:
            imagepath1 = os.path.exists(self.pic_path)
            # print(imagepath1)
            if not imagepath1:
                return
            else:
                self.matchstr1 = self.match_path(self.pic_path)
        # print("file_pic off")
        self.matchflag = 0
        self.stopflag = 1
        self.thread = threading.Thread(target=self.pic_search, args=(self,))
        self.thread.setDaemon(True)
        self.thread.start()
        self.thread_run = True

    def pic_search(self, self2):
        self.thread_run = True
        print("开始查找")
        wait_time = time.time()
        while self.thread_run:
            while self.count:
                if self.stopflag==1:
                    self.pic_path2 = self.array_of_img[self.count-1]
                    # print(self.pic_path2)
                    if time.time()-wait_time > 2:
                        wait_time = time.time()
                        print("正在查找", self.count)
                        try:
                            self.match_pic()
                        except:
                            pass
                        self.count = self.count - 1

                    if self.matchflag == 1:
                        print(self.pic_path2)
                        self.thread_run = False
                        self.show_pic2()
                        self.cut_clean2()
                        print("查找结束")
                        return
                if self.stopflag==0:
                    self.thread_run = False
                    return
                if self.count == 0:
                    self.thread_run = False
                    self.show_pic2()
                    self.cut_clean2()
                    print("查找结束")
        self.show_pic2()

    def pic_search2(self, self2):
        self.thread_run2 = True
        print("开始查找")
        while self.thread_run2:
            while self.count:
                if self.stopflag==1:
                    self.pic_path2 = self.array_of_img[self.count-1]
                    # print(self.pic_path2)
                    print("正在查找", self.count)
                    try:
                        self.match_pic()
                    except:
                        pass
                    self.count = self.count - 1
                    if self.matchflag == 1:
                        print(self.pic_path2)
                        self.thread_run2 = False
                        self.show_pic2()
                        self.cut_clean2()
                        print("查找结束")
                        return
                if self.stopflag==0:
                    self.thread_run2 = False
                    return
                if self.count == 0:
                    self.thread_run2 = False
                    self.show_pic2()
                    self.cut_clean2()
                    print("查找结束")

    def file_pic2(self):
        # print("file_pic")
        if (self.pic_path3 == ""):
            tkinter.messagebox.showinfo(title='车牌对比识别系统', message='路径不能为空')
            return
        if (self.pic_path == ""):
            if self.input1.get()=="":
                tkinter.messagebox.showinfo(title='车牌对比识别系统', message='图片1不能为空')
                return
            else:
                self.matchstr1 = self.input1.get()
                print(self.matchstr1)
        else:
            imagepath1 = os.path.exists(self.pic_path)
            # print(imagepath1)
            if not imagepath1:
                return
            else:
                self.matchstr1 = self.match_path(self.pic_path)
        # print("file_pic off")
        self.matchflag = 0
        self.stopflag = 1
        self.thread2 = threading.Thread(target=self.pic_search2, args=(self,))
        self.thread2.setDaemon(True)
        self.thread2.start()
        self.thread_run2 = True

    def show_pic2(self):
        self.pilImage4 = Image.open(self.pic_path2)
        w, h = self.pilImage4.size
        pil_image_resized2 = self.resize(w, h, self.pilImage4)
        self.tkImage4 = ImageTk.PhotoImage(image=pil_image_resized2)
        self.image_ctl2.configure(image=self.tkImage4)

    def match_pic(self):
        if self.thread_run2 == False:
            self.show_pic2()
        # matchstr1 = self.match_path(self.pic_path)
        matchstr2 = self.match_path(self.pic_path2)
        print(matchstr2)
        if self.matchstr1==matchstr2:
            self.matchflag = 1
            matchstr3 = "        车牌相符        "
        else:
            matchstr3 = "        车牌不符        "
        matchstr = self.matchstr1 + matchstr3 + matchstr2
        self.match.configure(text=str(matchstr))

    def match_path(self, pic_path):
        r_c = None
        r_color = None
        textstr = None
        img_bgr = img_math.img_read(pic_path)
        first_img, oldimg = self.predictor.img_first_pre(img_bgr)
        th1 = ThreadWithReturnValue(target=self.predictor.img_color_contours, args=(first_img, oldimg))
        th2 = ThreadWithReturnValue(target=self.predictor.img_only_color, args=(oldimg, oldimg, first_img))
        th1.start()
        th2.start()
        r_c, roi_c, color_c = th1.join()
        r_color, roi_color, color_color = th2.join()
        try:
            Plate = HyperLPR_PlateRecogntion(img_bgr)
            # print(Plate[0][0])
            r_c = Plate[0][0]
            r_color = Plate[0][0]
        except:
            pass
        if r_c:
            textstr = r_c
        if r_color:
            textstr = r_color
        if not (r_color or r_c):
            textstr = self.api_ctl(pic_path)
        return str(textstr)

    def api_ctl(self, pic_path):
        colorstr, textstr = api_pic(pic_path)
        # print(colorstr, textstr)
        return textstr

    def resize(self, w, h, pil_image):
        w_box = 400
        h_box = 400
        f1 = 1.0*w_box/w
        f2 = 1.0*h_box/h
        factor = min([f1, f2])
        width = int(w*factor)
        height = int(h*factor)
        return pil_image.resize((width, height), Image.ANTIALIAS)

    def cut_clean(self):
        self.pilImage3 = Image.open("pic/searchl.png")
        self.tkImage3 = ImageTk.PhotoImage(image=self.pilImage3)
        self.image_ctl.configure(image=self.tkImage3)
        self.pilImage4 = Image.open("pic/searchr.png")
        self.tkImage4 = ImageTk.PhotoImage(image=self.pilImage4)
        self.image_ctl2.configure(image=self.tkImage4)
        self.stopflag = 0
        self.count = 0
        self.thread_run = False
        self.thread_run2 = False
        self.s1.set("")
        self.s2.set("")
        self.pic_path = ""
        self.pic_path2 = ""
        self.match.configure(text="")

    def cut_clean2(self):
        self.stopflag = 0
        # self.count = 0
        self.s1.set("")
        self.s2.set("")
        self.pic_path = ""
        self.pic_path2 = ""


def close_window():
    print("log destroy")
    log.destroy()


if __name__ == '__main__':
    log = tk.Tk()

    login = Login(log)
    # close,退出输出destroy
    log.protocol('WM_DELETE_WINDOW', close_window)
    # 进入消息循环
    log.mainloop()