import json
import os
import csv
# 定义文件所在的目录
input_dir = 'E:/python-testn/pythonProject3/hh_1/evaluate_result'

# 获取目录中的所有文件
files = os.listdir(input_dir)

# with open("token_counts_QS.csv","w", newline='') as csvfile:
#     writer = csv.writer(csvfile)
#     writer.writerow(["Model", "token_subset_1", "token_subset_2","token_subset_3","token_subset_4","token_subset_5","token_subset_6"])


with open("line_counts_QS.csv","w", newline='') as csvfile:
    writer = csv.writer(csvfile)
    writer.writerow(["Model", "line_subset_1", "line_subset_2","line_subset_3","line_subset_4","line_subset_5","line_subset_6"])

# with open("CC_QS.csv", "w", newline='') as csvfile:
#     writer = csv.writer(csvfile)
#     writer.writerow(["Model", "CC_subset_1", "CC_subset_2","CC_subset_3","CC_subset_4","CC_subset_5","CC_subset_6"])



for file_name in files:
    # 构建完整的文件路径
    input_file_path = os.path.join(input_dir, file_name)
    first_underscore_index = file_name.find('_')

    # 找到最后一个 - 的位置
    last_dash_index = file_name.rfind('-')
    model_name = file_name[first_underscore_index + 1:last_dash_index]
    print(model_name)
    with open(input_file_path,"r",encoding="utf-8") as file:
        data1=json.load(file)

    with open("QS.json", "r", encoding="utf-8") as file:
        data2=json.load(file)
    sum0=0
    count0=0
    sum1=0
    count1=0
    sum2=0
    count2=0
    sum3 = 0
    count3 = 0
    sum4=0
    count4=0
    sum5 = 0
    count5 = 0



    for (item1,item2) in zip(data1["humaneval"]["pass@1"],data2):
    #     #按照token个数划分后的评估结果
    #     if item2["token_diff"] == 0:
    #         index, value = item1
    #         print(item2["token_diff"],index,value)
    #         sum0=sum0+value
    #         count0=count0+1
    #     if item2["token_diff"] == 1:
    #         index, value = item1
    #         print(item2["token_diff"], index, value)
    #         sum1=sum1+value
    #         count1=count1+1
    #     if item2["token_diff"] == 2:
    #         index, value = item1
    #         print(item2["token_diff"], index, value)
    #         sum2=sum2+value
    #         count2=count2+1
    #     if item2["token_diff"] == 3:
    #         index, value = item1
    #         print(item2["token_diff"], index, value)
    #         sum3=sum3+value
    #         count3=count3+1
    #     if item2["token_diff"] == 4:
    #         index, value = item1
    #         print(item2["token_diff"], index, value)
    #         sum4 = sum4 + value
    #         count4 = count4 + 1
    #     if item2["token_diff"] ==5:
    #         index, value = item1
    #         print(item2["token_diff"], index, value)
    #         sum5 = sum5 + value
    #         count5 = count5 + 1


        #按照行数划分后的评估结果
        if item2["line_diff"] == 0:
            index, value = item1
            print(item2["line_diff"],index,value)
            sum0=sum0+value
            count0=count0+1
        if item2["line_diff"] == 1:
            index, value = item1
            print(item2["line_diff"], index, value)
            sum1=sum1+value
            count1=count1+1
        if item2["line_diff"] == 2:
            index, value = item1
            print(item2["line_diff"], index, value)
            sum2=sum2+value
            count2=count2+1
        if item2["line_diff"] == 3:
            index, value = item1
            print(item2["line_diff"], index, value)
            sum3=sum3+value
            count3=count3+1
        if item2["line_diff"] == 4:
            index, value = item1
            print(item2["line_diff"], index, value)
            sum4=sum4+value
            count4=count4+1
        if item2["line_diff"] == 5:
            index, value = item1
            print(item2["line_diff"], index, value)
            sum5 = sum5 + value
            count5 = count5 + 1

        #按照圈复杂度划分后的评估结果
        # if item2["CC_diff"] == 0:
        #     index, value = item1
        #     print(item2["CC_diff"],index,value)
        #     sum0=sum0+value
        #     count0=count0+1
        # if item2["CC_diff"] == 1:
        #     index, value = item1
        #     print(item2["CC_diff"], index, value)
        #     sum1=sum1+value
        #     count1=count1+1
        # if item2["CC_diff"] == 2:
        #     index, value = item1
        #     print(item2["CC_diff"], index, value)
        #     sum2=sum2+value
        #     count2=count2+1
        # if item2["CC_diff"] == 3 :
        #     index, value = item1
        #     print(item2["CC_diff"], index, value)
        #     sum3=sum3+value
        #     count3=count3+1
        # if item2["CC_diff"] == 4 :
        #     index, value = item1
        #     print(item2["CC_diff"], index, value)
        #     sum4=sum4+value
        #     count4=count4+1
        # if item2["CC_diff"] == 5 :
        #     index, value = item1
        #     print(item2["CC_diff"], index, value)
        #     sum5=sum5+value
        #     count5=count5+1



    mean0=round(sum0/count0*100,2)

    mean1=round(sum1/count1*100,2)
    mean2=round(sum2/count2*100,2)
    mean3=round(sum3/count3*100,2)
    mean4=round(sum4/count4*100,2)
    mean5 = round(sum5 / count5 * 100, 2)
    print("count_result!!")
    print(count0,count1,count2,count3,count4,count5)
    print(mean0,mean1,mean2,mean3,count4,mean5)
    # with open("token_counts_QS.csv", mode='a', newline='', encoding='utf-8') as file:
    #     writer = csv.writer(file)
    #     writer.writerow([model_name,mean0,mean1,mean2,mean3,mean4,mean5])

    with open("line_counts_QS.csv", mode='a', newline='', encoding='utf-8') as file:
        writer = csv.writer(file)
        writer.writerow([model_name,mean0,mean1,mean2,mean3,mean4,mean5])
    #
    # with open("CC_QS.csv", mode='a', newline='', encoding='utf-8') as file:
    #     writer = csv.writer(file)
    #     writer.writerow([model_name,mean0,mean1,mean2,mean3,mean4,mean5])
    #