import sys
import os
from os import path
from shutil import copy2

input_path = sys.argv[1]
output_path = sys.argv[2]
multiplier = int(sys.argv[3])
image_path = path.join(input_path, 'JPEGImages')
gt_path = path.join(input_path, 'Annotations')

videos = sorted(os.listdir(image_path))

for vid in videos:
    os.makedirs(path.join(output_path, 'JPEGImages', vid), exist_ok=True)
    os.makedirs(path.join(output_path, 'Annotations', vid), exist_ok=True)
    frames = sorted(os.listdir(path.join(image_path, vid)))

    num_frames = len(frames)
    counter = 0
    output_counter = 0
    direction = 1
    for _ in range(multiplier):
        for _ in range(num_frames):
            copy2(path.join(image_path, vid, frames[counter]), 
                    path.join(output_path, 'JPEGImages', vid, f'{output_counter:05d}.jpg'))

            mask_path = path.join(gt_path, vid, frames[counter].replace('.jpg', '.png'))
            if path.exists(mask_path):
                copy2(mask_path, 
                    path.join(output_path, 'Annotations', vid, f'{output_counter:05d}.png'))

            counter += direction
            output_counter += 1
            if counter == 0 or counter == len(frames) - 1:
                direction *= -1