import os
import shutil
import tqdm

if __name__ == '__main__':
    folder = '/home/workspace/data/dataset/local-cache/train_skirt'
    is_skirt = True 
    skirt_modifier = '_skirt' if is_skirt else ''
    frames_file = 'frames' + skirt_modifier + '_10p.txt'
    neighbourhood = 10

    frames = []
    with open(os.path.join(folder, frames_file), 'r') as f:
        lines = f.readlines()
        lines = [line.strip() for line in lines]
    frames = lines

    os.makedirs(os.path.join(folder, f'uv_curvature_{neighbourhood}'), exist_ok=True)
    for frame in tqdm.tqdm(frames):
        id = frame.split('_')[0]
        src = os.path.join(folder, f'{id}', 'curvature_uv_10', f'{frame}.npz')
        dst = os.path.join(folder, f'uv_curvature_{neighbourhood}', f'{frame}.npz')
        shutil.move(src, dst)