import numpy as np
import os
 
# modify gt labels
dir_txt = "model/ship_gt/labels_test/"
save_dir_ = "model/ship_gt/labels_test_5type/"

list_txt = os.listdir(dir_txt)
# labels from float to int
# print(list_txt)
# import pdb
# pdb.set_trace()

for txt in list_txt:
    # print('{}{}'.format(dir_txt,txt))
    if txt.split('_')[0] == 'render':
        with open('{}{}'.format(dir_txt,txt), 'r') as fr:
            # import pdb
            # pdb.set_trace()
            line = fr.readlines()[0]

            # print(line.split(','))
            data = line.split(',')
            # print(data)
            
            # print('{}{}'.format(save_dir_,txt))
            # print('{},{},{},{},{},{},{},{},{}\n'.format(data[0],data[1],data[2],data[3],data[4],data[5],data[6],data[7],data[8]))
            # if len(data) == 5:
            
            if len(data) == 9:
                with open('{}{}'.format(save_dir_,txt),'w') as f:
                    # f.write('{},{},{},{},{}'.format('RENDER' if (data[0]=='CVN') else data[0],data[1],data[2],data[3],data[4]))
                    f.write('{},{},{},{},{},{},{},{},{}'.format(data[0],data[1],data[2],data[3],data[4],data[5],data[6],data[7],'render\n' if (data[8]=='cvn\n') else data[8]))