import os
import time
import subprocess

start_time = time.time()

parser = argparse.ArgumentParser()
parser.add_argument('--folder', required=True)
parser.add_argument('--out_fn', required=True)

os.makedirs('./uprez', exist_ok=True)

print(">>> {}".format(dataset))

in_dir = os.path.abspath('./recursive/{}'.format(dataset))
vsr_duf_inputs_dir = os.path.abspath('../vsr-duf/inputs')
vsr_duf_results_dir = os.path.abspath('../vsr-duf/results/{}'.format(dataset))
out_dir = os.path.abspath('./uprez')

subprocess.call([
  '/home/lens/miniconda3/envs/tf/bin/python',
  '../vsr-duf/crop_frames.py',
  '--in_dir', in_dir,
  '--out_dir', vsr_duf_inputs_dir,
  '--ratio', '6.0'
])

subprocess.call([
  '/home/lens/miniconda3/envs/tf/bin/python',
  '../vsr-duf/test.py',
  '--in_dir', vsr_duf_inputs_dir,
  '--out_dir', vsr_duf_results_dir,
  '--network_dir', os.path.abspath('../vsr-duf'),
])

subprocess.call([
  '/home/lens/miniconda3/envs/tf/bin/python',
  '../vsr-duf/merge_frames.py',
  '--in_dir', vsr_duf_results_dir,
  '--render_dir', out_dir,
  '--out_fn', opt.out_fn,
  '--mov',
])

finish_time = time.time()

print("took %02f s.".format(finish_time - start_time))