# -*- coding:utf-8 -*-

'''
Date:2020.07.03
author: tim
describe: this script is used for transform the other label format(some giver dataset) to the format of ourselves
'''

import os
import json

labelothersPath = "/home/liu/work/dataset/RS_datasets_test" # your original label file path

othersFolderNameList = os.listdir(labelothersPath)

for name in othersFolderNameList:

    labelothersNameList = os.listdir(os.path.join(labelothersPath,name,"label"))
    for labelFileName in labelothersNameList:
        labelInfoNew = []
        labelTransedPath = os.path.join(labelothersPath,name,"label_transed")
        if not os.path.exists(labelTransedPath):
            os.makedirs(labelTransedPath)
        with open(os.path.join(labelothersPath,name,"label",labelFileName),'r',encoding='utf8') as fp:
           # labelInfo = {'labels':[
           # {"centers":{'x':float,"y":float,"z":float},# for ours position
           #  "size":{'x':float,'y':float,'z'：float}, # for ours scale
           #  "rotation":{"pitch":float,"roll":float,"yaw":float}, # for ours rotation
           #  "tracker_id" : int  # for ours obj_id
           #  "type": str # for ours obj_type
           # }]}
            labelInfo = json.load(fp)
            infoList = labelInfo["labels"]
           # transform the label format to ours
            for info in infoList:
                infoNew = {}
                centers = info["center"]
                size = info["size"]
                rotation = {}
                rotation["x"] = info["rotation"]["pitch"]
                rotation["y"] = info["rotation"]["roll"]
                rotation["z"] = info["rotation"]["yaw"]
                trackerId = info["tracker_id"]
                type = info["type"]
                infoNew["psr"] = {"position": centers,"scale":size,"rotation":rotation}
                infoNew["obj_type"] = type
                infoNew["obj_id"] = str(trackerId)
                labelInfoNew.append(infoNew)
        # write the new info into the new json file
        with open (os.path.join(labelTransedPath,labelFileName),"a+") as f :
            json.dump(labelInfoNew,f)