import os

new_line=[]
with open('trainlist01.txt',"r") as f:
    for line in f:
        new_line.append('/your_path/Datasets/UCF-101/'+line)
        
with open('trainlist01.txt',"w") as f:
    for i in new_line:
        f.write(i)

import os 
import fnmatch


class_ind = [x.strip().split() for x in open('/home/xu/datasets/ucf101/ucfTrainTestlist/classInd.txt')]
class_mapping = {x[1]:int(x[0]) for x in class_ind}
#print(class_mapping)

def make_list(read_list, write_list):
    new_line=[]
    with open(read_list,'r') as f:
        for line in f:
            video_frame_path=os.path.join('/home/xu/datasets/ucf101/jpg', line.split()[0].split('.')[0])
            video_frame_number=len(fnmatch.filter(os.listdir(video_frame_path), 'image'+'*'))
            video_groundtruth_class=class_mapping[line.split()[0].split('.')[0].split('/')[0]]
            new_line.append('{} {} {}\n'.format(video_frame_path, video_frame_number, video_groundtruth_class))
            
    with open(write_list,'w') as f:
        for i in new_line:
            f.write(i)
    
        
#read_list='/home/xu/datasets/ucf101/ucfTrainTestlist/trainlist01.txt'
#write_list='/home/xu/datasets/ucf101/ucfTrainTestlist/ucf_rgb_train_split_1.txt'
for i in range(3):
    read_train_list='/home/xu/datasets/ucf101/ucfTrainTestlist/trainlist0{}.txt'.format(i+1)
    write_train_list='/home/xu/datasets/ucf101/ucfTrainTestlist/ucf_rgb_train_split_{}.txt'.format(i+1)
    make_list(read_train_list, write_train_list)
    read_val_list='/home/xu/datasets/ucf101/ucfTrainTestlist/testlist0{}.txt'.format(i+1)
    write_val_list='/home/xu/datasets/ucf101/ucfTrainTestlist/ucf_rgb_val_split_{}.txt'.format(i+1)
    make_list(read_val_list, write_val_list)