#遍歷文件夾中的所有txt文件
def scanner_txt(inputSrc,txt_list):
    file_list = os.listdir(inputSrc)
    for file in file_list:
        curr_file=path.join(inputSrc,file)
        #遞歸實(shí)現(xiàn)
        if(path.isdir(curr_file)):
            scanner_txt(curr_file,txt_list)
        else:
            curr_file_name=curr_file.split(".")
            curr_file_type=curr_file_name[len(curr_file_name)-1]
            if curr_file_type=="txt":
                txt_list.append(curr_file)
    return txt_list

def readTxtDataProprocessing(filepath,kss7sd,kss8sd,kss9sd):
    resFrame=[]
    print(filepath)
    with open(filepath,encoding='utf-8') as f:
        for line in f:
            frame_info=line.split(',')
            if(len(frame_info)<3):
                continue
            framenum=frame_info[0]
            frametime=frame_info[2].split('\n')[0]
            resFrame.append([int(framenum),int(frametime)])
    f.close()
    #文件內(nèi)閉眼時(shí)長(zhǎng)為空的不處理
    if(len(resFrame)==0):
        return
    #去掉重復(fù)的值，歸并成統(tǒng)一的值
    totalSD=deduplicationData(resFrame)
    filename=filepath.split('\\')[-1]
    kssLevel=filename.split('-')[2]
    print("文件:{}  總體閉眼時(shí)長(zhǎng)標(biāo)準(zhǔn)差為:{}".format(filename,totalSD))
    print()
    if(kssLevel=='KSS7'):
        kss7sd.append(totalSD)
    elif(kssLevel=='KSS8'):
        kss8sd.append(totalSD)
    elif(kssLevel=='KSS9'):
        kss9sd.append(totalSD)

def deduplicationData(resFrame):
    uniqueRes = []
    framelength = len(resFrame)
    i = 0
    while (i < framelength):
        if (resFrame[i][0] == 17924):
            t = 1;
        count = 1
        if (i + 1 < framelength):
            if (resFrame[i][1] == resFrame[i + 1][1]):
                count += 1
        if (i + 1 >= framelength):
            break
        if (i + 2 < framelength):
            if (resFrame[i][1] == resFrame[i + 2][1]):
                count += 1

        if (count != 1):
            uniqueRes.append([resFrame[i][0], resFrame[i][1]])
        i += count
        #計(jì)算閉眼時(shí)長(zhǎng)標(biāo)準(zhǔn)差
    return closeEyeTimeSD(uniqueRes)
    # for i in range(len(uniqueRes)):
    #     print("frameNum:", uniqueRes[i][0], "frameTime:", uniqueRes[i][1], "ms")

#data 經(jīng)過(guò)去重和選幀后得到的列表
def closeEyeTimeSD(data):
    #按每一分鐘的時(shí)間間隔劃分區(qū)間：1秒鐘30幀，1分鐘1800幀
    #建立分割數(shù)組保存第n分鐘內(nèi)的睜閉眼數(shù)據(jù),劃分成12段
    dividArr=[]
    for i in range(12):
        dividArr.append([])
    for i in range(len(data)):
        index=int(data[i][0]/1800)
        dividArr[index].append(data[i][1])
    #獲取標(biāo)準(zhǔn)差
    sdArr=[]
    for i in range(len(dividArr)):
        avergeTime,sdArray=countTotalSD(dividArr[i])
        if(dividArr[i]!=[]):
            sdArr.append(sdArray)
            print("第{}分鐘平均眨眼時(shí)長(zhǎng)為:{}，閉眼時(shí)長(zhǎng)標(biāo)準(zhǔn)差為:{}".format(i+1,avergeTime,sdArray))
    totalSD=countTotalSD(sdArr)[1]
    return totalSD

def countTotalSD(sdArray):
    if(len(sdArray)==0):
        return [0,0]
    sumArray = 0
    for i in sdArray:
        sumArray += i
    # 計(jì)算全部的標(biāo)準(zhǔn)差
    averageSD = sumArray / len(sdArray)
    sumSD = 0
    for i in sdArray:
        sumSD += pow((i - averageSD), 2)
    totalSD = math.sqrt(sumSD / len(sdArray))
    return [averageSD,totalSD]