label_company

import re
from numpy import *
import cv2
import numpy as np
import os
import json
import sys
from label_json import Label_Json

register_name = []
register_hsi = []
register_rgb = []


file_path = r'/pytorch-3dunet-master-zr_rgb_and_hsi/pytorch3dunet/datasets/data_label_company'

class Label_Company:

    def getFiles(file_path, suffix):
        res = []
        for root, directory, files in os.walk(file_path):
            for filename in files:
                name, suf = os.path.splitext(filename)
                if suf == suffix:
                    res.append(os.path.join(root, filename))
        return res

    for file in getFiles("./",'.json'):

        if 'jpg.json' in file:
            with open(file, encoding='utf-8') as f:

                contents = json.load(f)
                dict = contents['markResult']['objects']
                dict = dict[0]
                dict = dict['features']
                dict = dict[0]
                dict = dict['geometry']['coordinates']
                register_rgb = []
                register_rgb.append(dict)
                # register_rgb = register_rgb[0]
                # print(len(register_rgb))
                fw = open('data_coordinate.txt', 'a+')
                for i in range(len(register_rgb)):
                    s = " ".join([",".join([str(n) for n in c]) for c in register_rgb[i]]) + '\n'
                    # s = " ".join([",".join([str(n) for n in register_rgb[i]])])
                    file = file.split('.json')[-2].split('/')[2]
                    print(file + '\n' + "RGB:" + s)
                    fw.write(file + "..RGB:" + s)
                fw.close()

        else:

            with open(file, encoding='utf-8') as f:

                contents = json.load(f)
                # print(contents)
                dict = contents['markResult']['objects']
                dict = dict[0]
                dict = dict['features']
                dict = dict[0]
                dict = dict['geometry']['coordinates']
                register_hsi = []
                register_hsi.append(dict)
                # print(len(register_hsi))
                fw = open('data_coordinate.txt', 'a+')
                for i in range(len(register_hsi)):
                    s = " ".join([",".join([str(n) for n in c]) for c in register_hsi[i]]) + '\n'
                    file = file.split('.json')[-2].split('/')[2]
                    print(file + '\n' + "HSI:" + s)
                    fw.write(file + "..HSI:" + s)
                fw.close()

    with open('data_coordinate.txt', 'r') as ins:
        ff = ins.readlines()
        ff.sort()

    write = open('data_coordinate.txt','w')
    write.write(''.join(ff).replace('..', '\n'))
    write.close()


if __name__ == '__main__':
    Label_Company()
    Label_Json()