import aircv
import cv2


class image_util:

    # 判斷img_target是否在img_source內出現(xiàn)，若存在則提供在圖片img_source中的位置
    def match_pic(self,img_source, img_target):
        """
        進行圖片的匹配
        """
        imsrc = aircv.imread(img_source)
        imobj = aircv.imread(img_target)
        pos = aircv.find_template(imsrc, imobj, rgb=True, bgremove=True)
        if not pos:
            return None, None, None, None

        left_top = pos['rectangle'][0]  # 左上
        right_bottom = pos['rectangle'][3]  # 右下
        circle_center_pos = pos['result']  # 中心點坐標
        circle_center_pos = tuple(map(int, circle_center_pos))
        confidence = pos['confidence'] # 準確率
        print('準確率:', confidence)
        #print(circle_center_pos)  # 坐標位置
        return circle_center_pos, left_top, right_bottom, confidence

    # 在圖片img_source內標記出 中心點為pos 左上點為left_top 右下點為right_bottom 的空心圖形
    def draw_box(self,img_source, pos, left_top, right_bottom, circle_radius,  color, line_width):
        """
        將小圖在大圖中圈出來
        """
        if not pos:
            print('pos is null')
            return
        # 框出圓形
        cv2.circle(img_source, pos, circle_radius, color, line_width)
        # 框出矩形
        cv2.rectangle(img_source, left_top, right_bottom, color, line_width)  # 左上缺脉，右下
        # 創(chuàng)建窗口時候可以鼠標隨意拖動窗口改變大小
        cv2.namedWindow("objDetect", 0)
        # 設置長寬大小為640*480
        cv2.resizeWindow("objDetect", 640, 480)
        # 移動窗口到（200,200）坐標
        cv2.moveWindow("objDetect", 200, 200)
        cv2.imshow('objDetect', imsrc)
        cv2.waitKey(0)
        cv2.destroyAllWindows()



if __name__ == "__main__":
    dir="D:/Program/Python/wg/yaomo"
    iu = image_util()
    img_source = dir+"/base.png"
    img_target = dir+"/gj.png"
    circle_center_pos, left_top, right_bottom, pos = iu.match_pic(img_source, img_target)
    print(circle_center_pos[0],circle_center_pos[1])
    # 畫線的屬性
    circle_radius = 40
    color = (255, 0, 0)
    line_width = 5
    imsrc = aircv.imread(img_source)
    iu.draw_box(imsrc, circle_center_pos, left_top, right_bottom, circle_radius, color, line_width)