'''
现场1:N测试，确定阈值
'''
import os
import numpy as np
import matplotlib.pyplot as plt


def showHist(filtered_data):
    Same = filtered_data[:, 1].astype(np.float32)
    Cross = filtered_data[:, 2].astype(np.float32)

    fig, axs = plt.subplots(2, 1)
    axs[0].hist(Same, bins=50, edgecolor='black')
    axs[0].set_xlim([-0.1, 1])
    axs[0].set_title('first')

    axs[1].hist(Cross, bins=50, edgecolor='black')
    axs[1].set_xlim([-0.1, 1])
    axs[1].set_title('second')
    # plt.savefig('plot.png')
    plt.show()


def get_tartget_list(nested_list):
    filtered_list = np.array(list(filter(lambda x: len(x) >= 2, nested_list)))  # 去除无轨迹的数据
    filtered_correct = filtered_list[filtered_list[:, 0] != 'wrong']  # 获取比对正确的时项
    filtered_wrong = filtered_list[filtered_list[:, 0] == 'wrong']  # 获取比对错误的时项
    showHist(filtered_correct)
    # showHist(filtered_wrong)
    print(filtered_list)


def deal_process(file_pth):
    flag = False
    event = file_pth.split('\\')[-2]
    target_barcode = file_pth.split('\\')[-2].split('_')[-1]
    temp_list = []

    with open(file_pth, 'r') as f:
        for line in f:
            if 'oneToOne' in line:
                flag = True
                continue
            if flag:
                line = line.replace('\n', '')
                comparison_data = line.split(',')
                forecast_barcode = comparison_data[0]
                value = comparison_data[-1].split(':')[-1]
                if value == '':
                    break
                if len(temp_list) == 0:
                    if forecast_barcode == target_barcode:
                        temp_list.append('correct')
                    else:
                        temp_list.append('wrong')
                temp_list.append(float(value))
        temp_list.append(event)
    return temp_list


def anaylze_scratch(scratch_pth):
    purchase, back = [], []
    for root, dirs, files in os.walk(scratch_pth):
        if len(root) > 0:
            if len(root.split('_')) == 4:  # 加购
                process = os.path.join(root, 'process.data')
                if not os.path.exists(process):
                    continue
                purchase.append(deal_process(process))
            elif len(root.split('_')) == 3:
                process = os.path.join(root, 'process.data')
                if not os.path.exists(process):
                    continue
                back.append(deal_process(process))
    # get_tartget_list(purchase)
    get_tartget_list(back)
    print(purchase)


if __name__ == '__main__':
    # scratch_pth = r'\\192.168.1.28\\share\\测试视频数据以及日志\\各模块测试记录\\展厅测试\\1108_展厅模型v800测试\\'
    scratch_pth = r'\\192.168.1.28\\share\\测试视频数据以及日志\\各模块测试记录\\展厅测试\\1120_展厅模型v801测试\\扫A放A\\'
    anaylze_scratch(scratch_pth)