from glob import glob
import os
from shutil import copy

# sample keypoint filename: HICO_test2015_00009750_keypoints.json
test_keypoints = '/media/mona/easystore/human_pose_estimation/data/smplify-x/djrn_test_data/keypoints/'
train_keypoints = '/media/mona/easystore/human_pose_estimation/data/smplify-x/djrn_train_data/keypoints/'

# sample image name: HICO_test2015_00000330.jpg
ride_bike_test = '/home/mona/research/code/ConsNet/data/ride_bike_djrn/test/ride_bicycle/'
ride_bike_train = '/home/mona/research/code/ConsNet/data/ride_bike_djrn/train/ride_bicycle/'

test_keypoints_ride_bike = 'ride_bike_keypoints/test/keypoints'
train_keypoints_ride_bike = 'ride_bike_keypoints/train/keypoints'
if not os.path.exists(test_keypoints_ride_bike):
    os.makedirs(test_keypoints_ride_bike)
if not os.path.exists(train_keypoints_ride_bike):
    os.makedirs(train_keypoints_ride_bike)    
    
for filename in glob(ride_bike_test + "*.jpg"):
    for keypoint_f in glob(test_keypoints + "*.json"):
        if os.path.basename(filename)[:-4] in keypoint_f:
            copy(keypoint_f, test_keypoints_ride_bike)
            
#import pdb
for filename in glob(ride_bike_train + "*.jpg"):
    for keypoint_f in glob(train_keypoints + "*.json"):
        if os.path.basename(filename)[:-4] in keypoint_f:
            #pdb.set_trace()
            copy(keypoint_f, train_keypoints_ride_bike)