# create a folder for the registered subject

import os

import shutil

import pickle

SUBJECT_NAME = 'benny_cai'

SCAN_POSE = 'side'

# SCAN_POSE = 'front'

if __name__ == '__main__':

    folder_path = 'data/' + SUBJECT_NAME

    if not os.path.isdir(folder_path):

        os.mkdir(folder_path)

    os.chdir(folder_path)

    if not os.path.isdir(SCAN_POSE):

        os.mkdir(SCAN_POSE)

    os.chdir(SCAN_POSE)

    for subfolder in ['color_images',

                      'depth_images',

                      'intrinsics',

                      'OpenPose',

                      'ViTPose_base',

                      'ViTPose_large']:

        if not os.path.isdir(subfolder):

            os.mkdir(subfolder)

    # copy odometry.log from 'data' folder

    src = '../../odometry.log'

    dst = 'odometry.log'

    shutil.copyfile(src, dst)

    file1 = open("../../subject_name.txt", "a")

    file1.write(SUBJECT_NAME + '_' + SCAN_POSE + "\n")

    file1.close()

    for subfolder in ['OpenPose',

                      'ViTPose_base',

                      'ViTPose_large']:

        for subsubfolder in ['keypoints', 'output_images']:

            if not os.path.isdir(subfolder + '/' + subsubfolder):

                os.mkdir(subfolder + '/' + subsubfolder)