#½øÈëÏß³ÌÏȰѱê־λÖÃΪTrue
# from System_I import Detection
# import os,torch
# os.environ['CUDA_VISIBLE_DEVICES'] = '1'
# devices = torch.device('cuda:0' if torch.cuda.is_available() else 'cpu')
# root = "/home/wanghc/output"
import  os,torch, threading,argparse
class global_para():
    #check()
    parser1 = argparse.ArgumentParser(description="CUI")
    parser1.add_argument("video_num", default='/data1/Hatcher/model/project/data_MPKFNet/rawdata/竹林嘴-3月3日候鸟视频.mp4', type=str,
                        help='Video name.')
    parser1.add_argument("GPU", default='2', type=str,
                        help='GPU.')
    parser1.add_argument("--thresh", default=0.4, type=float,
                        help='GPU.')
    parser1.add_argument("--red", default=0, type=int,
                        help='red.')
    parser1.add_argument("--add", default=0, type=str,
                        help='add.')

    args = parser1.parse_args()
    #print(args)
    root = "/home/gyd/output1"
    # os.environ['CUDA_VISIBLE_DEVICES'] = '0'
    os.environ['CUDA_VISIBLE_DEVICES'] = args.GPU
    devices = torch.device('cuda:0' if torch.cuda.is_available() else 'cpu')
    camera_address = args.add
    VideoPath = args.video_num
    red_flag = args.red
    detect_thresh = args.thresh

    #fine_thresh2 = torch.tensor(fine_thresh2).cuda()
    birdIndex = 99999
    #birdIndex = torch.tensor(birdIndex).cuda()
    is_ex_bird = False  # Ä¿±ê¼ì²â±ê־λ£¬Ã¿¸öÒ»Ãë¼ì²âÒ»´ÎÄ¿±ê¼ì²â£¬Èç¹ûÓÐÄñ£¬ÔòÖÃΪTrue£¬ÎªTrueµÄʱºò˵Ã÷£¬ ´¦Àí30sµÄÏ̻߳¹ÔÚ¼ÌÐø£¬Ö±µ½±ê־λ±äΪÁËFalseÔÙ¿ªÆôеÄÏß³Ì
    Density_thread_flag = False  # ÃܶȹÀ¼ÆÏ̱߳ê־λ£¬ÎªTrue˵Ã÷¿ªÆôÁËÃܶȹÀ¼ÆỊ̈߳¬´¦ÀíÍê750֡Ϊֹ£¬±ê־λ±êΪFalse£¬
    Detection_thread_flag = False  # Ä¿±ê¼ì²âºÍϸÁ£¶Èʶ±ðÏ̱߳ê־λ£¬ÎªTrue˵Ã÷¿ªÆôÁËÃܶȹÀ¼ÆỊ̈߳¬´¦ÀíÍê750֡Ϊֹ£¬±ê־λ±êΪFalse£¬
    Vedio_thread_flag = False  # ´æ´¢ÊÓƵÏ̱߳ê־룬´æ´¢30s
    SaveImage_thread_flag = False  # ±£´æԭͼÏ̱߳ê־룬´æ´¢30ÕÅԭͼ
    ALL_thread_flag = False  # ËùÓÐÏ̵߳ıê־룬µ¥¶À¿ªÉèÒ»¸öỊ̈߳¬¼à¿ØËùÓÐÏß³ÌÊÇ·ñ¶¼½áÊø£¬Èç¹û¶¼½áÊø£¬ÔòÖÃΪFalse£¬ºóÐøÌí¼ÓÏ̹߳ÜÀí£¬¿ÉÒÔ·ÅÔÚÕâÀïÃæ

    ShowAllNumFlag = False


    DebugPrint_flag = False  # ´òÓ¡±ê־λ
    Species = torch.zeros(8).cuda()
    num_count_old =0
    boxes_fine = None
    labels_fine = None

    showThresh = 25
    destory_image = False
    Camera_device = args.video_num  # ÉãÏñÍ·±àºÅ£¬0001
    image_time = None  # µ±Ç°´¦ÀíµÄ30ÊÓƵµÄ±àºÅ
    ImgDealQueue = []  # 750Ö¡ÊÓƵ£¬»º´æ¿Õ¼ä
    img_show = None


    #event = threading.Event()
    eventDensity = threading.Event()
    eventDetection = threading.Event()
    eventSaveimage = threading.Event()
    max_age = 2
    min_hits = 3
    sortiou_threshold = 0.3
    Finerestartboxes = []
    IndexMax = 0
    radioShowX = 2560.0
    radioShowY = 1440.0
    radioShowtext = radioShowX
    SortLabel_Index = None   #±êÇ©ºÍË÷Òý¶ÔÓ¦¹Øϵ[[4,1],[3,2]]
    SortLabel_Indexpre = [] #±êÇ©ºÍË÷ÒýµÄ¶ÔÓ¦¹Øϵ£¬ÓÃÔÚ×·×Ù¹ý³ÌÖÐгöÏÖµÄÄ¿±ê£¬[[label,Index,Count]]
    # Detection2 = Detection(devices,root)
    FineCountThresh = 5

    img_show = None

    Fine_thresh = 0.4

    # DetWeight = "/data1/Hatcher/model/Detection/yolov5/runs/train/exp21/weights/best.pt"
    # FineWeight = "/data/Hatcher/data/weigth/areaproject/resnet50-JXPYH-Bird-3-4-20220429/_42911_PYH-3/best.pth"
    DetWeight = "/data1/Hatcher/model/Detection/yolov5/runs/train/KunMingCuiHu-Fish2/weights/best.pt"
    FineWeight = "/data/Hatcher/data/weigth/areaproject/resnet50-KMDC-Fish-5-1-20220505/_5510_DianChiFish-4/best.pth"
    # DetWeight = "/data1/Hatcher/model/Detection/yolov5/runs/train/BeijingCuiHu-Bird26/weights/best.pt"
    # FineWeight = "/data/Hatcher/data/weigth/areaproject/resnet50-cui90_1model0.7/_42111_CUI-94-1/best.pth"
    #BHG_chineseName = ["kong", '白鹤', '白头鹤','白枕鹤']
    BHG_chineseName = ["kong","鲫鱼","麦穗鱼","鳑鲏","鲈鱼"]

    # BHG_chineseName = ['Kong', '白额雁', '白骨顶', '苍鹭', '苍鹭', '白秋沙鸭', '白腰草鹬', '斑头雁', '斑嘴鸭', '苍鹭', '赤膀鸭', '赤颈鸭', '苍鹭', '大麻鳽',
    #                    '豆雁', '反嘴鹬', '凤头鸊鷉', '凤头麦鸡', '凤头潜鸭', '黑鹳', '黑水鸡', '红嘴鸥', '鸿雁', '灰鹤', '灰雁', '绿翅鸭', '绿头鸭', '琵嘴鸭',
    #                    '普通鸬鹚', '普通秋沙鸭', '翘鼻麻鸭', '鹊鸭', '扇尾沙锥', '小白额雁', '小天鹅', '针尾鸭', '苍鹭', '暗绿绣眼鸟', '八哥', '白腹鹞', '白鹡鸰',
    #                    '白眼潜鸭', '草鹭', '池鹭', '戴胜', '黑翅长脚鹬', '黑鸢', '红隼', '花脸鸭', '环颈雉', '灰头绿啄木鸟', '灰头麦鸡', '矶鹬', '栗苇鳽',
    #                    '罗纹鸭', '牛背鹭', '普通翠鸟', '雀鹰', '喜鹊', '苍鹭', '鸳鸯', '棕背伯劳', '红胸秋沙鸭', '普通燕鸥', '小白额燕鸥', '丘鹬', '凤头蜂鹰',
    #                    '大天鹅', '大鸨', '小田鸡', '山斑鸠', '帆背潜鸭', '日本松雀鹰', '普通鵟', '松雀鹰', '毛脚鵟', '灰斑鸠', '白头鹞', '白尾鹞', '白眉鸭',
    #                    '白鹈鹕', '红头潜鸭', '苍鹰', '蒙古银鸥', '西方秧鸡', '赤嘴潜鸭', '阔嘴鹬', '雕鸮', '鹌鹑', '鹗', '黄苇鳽', '黑颈鹤', '小䴙䴘', '赤麻鸭']
    # BHG_chineseName =["kong","绿头鸭","斑嘴鸭","灰雁","鸿雁","kong","小鷿鷈"]
    spiec_number_old = [0 for i in range(len(BHG_chineseName))]
    FineClassNum = 5
    color_show = (255, 255, 255)
    color_box = (255, 0, 0)
    NoneLabel = "鱼"#"鹤类" # "鱼"
    fine_thresh1 = 6000
    fine_thresh2 = 6000
    ShowAllNumFlag = True
    fontScale = 1
"""
1.change weight
2.change BHG_chineseName
3.change FineClassNum
4.change color_box  (0, 115, 51)green  (255, 0, 0) red
5.change parser1
6.change NoneLabel
7. change main
8. change fine_thresh2
9. GPU
10. change ShowAllNumFlag
fontScale  plot file para to adjust chinese   400W ---2  
                                              200W ---1  
                                              800W ---3
"""