import subprocess
import sys
from pathlib import Path
import os
import re

output_folder_name ="/bulk_convert_out/"

droppedFile = sys.argv[1]
droppedName = Path(droppedFile).name
originalfiles = sys.orig_argv[2:]

#long path shenanigans 
joined = (' '.join(originalfiles))
re.sub (' +', ' ', joined)
listed = re.split(r'(.+?mkv|.+?mov|.+?mp4)', joined)
listed = list(filter(None, listed))
listed = [i.lstrip() for i in listed]

#print(droppedName)

#get path of script location and remove the tail
head_tail = os.path.split(sys.argv[0])
output_path = head_tail[0]


#skip first part of sys.argv since it points to script location
arguments = sys.argv[1:]

#check if output folder exists, otherwise create it
isExist = os.path.exists(output_path+output_folder_name)
if not isExist:
    os.makedirs(output_path+output_folder_name)


print("input 0: ", listed[0])
print("input 1: ", listed[1])
x=int(input("select reference file: "))
ref= listed.pop(x)
print(listed)
print("selected reference file is: ", ref)
print("1: vmaf_v0.6.1.json")
print("2: vmaf_4k_v0.6.1.json")
x=int(input("select vmaf model: "))

if x == 1:
    scale = '"scale=1920:1080"'
    model = "vmaf_v0.6.1"
else:
    scale = '"scale=3840:2160"'
    model = "vmaf_4k_v0.6.1"


#basename = os.path.basename(file)

args = 'ffmpeg.exe -i "{distorted}" -i "{reference}" -lavfi "[0:v]{scale}[distorted];[1:v]{scale}[reference]; [distorted][reference]libvmaf='"'"'model=version=vmaf_v0.6.1:feature=name=psnr|name=float_ssim:n_threads=12:log_path=log.txt:log_fmt=json'"'"'" -f null -'.format(distorted=listed[0],reference=ref,scale=scale, model=model)
print(args)
subprocess.run(args)



k=input("press close to exit")