#!/bin/bash
set -e
R=/c/Users/Demo/tools/ai-video; W=$R/work; cd $R
ES="$R/esrgan/realesrgan-ncnn-vulkan.exe"; RF="$R/rife-ncnn-vulkan-20221029-windows/rife-ncnn-vulkan.exe"
declare -A SLOW=([a1]=3 [a2]=3 [a3]=3 [a4]=3 [a5]=3 [b1]=2)
for s in a1 a2 a3 a4 a5 b1; do
  echo "[$(date +%T)] $s upscale"; mkdir -p $W/$s/x4 $W/$s/hd $W/$s/slow
  "$ES" -i $W/$s/src -o $W/$s/x4 -n realesrgan-x4plus -m $R/esrgan/models -f png >/dev/null 2>&1
  ffmpeg -v error -y -i $W/$s/x4/%05d.png -vf "scale=1920:1080:flags=lanczos" $W/$s/hd/%05d.png
  rm -rf $W/$s/x4
  n=$(( $(ls $W/$s/hd | wc -l) * ${SLOW[$s]} ))
  echo "[$(date +%T)] $s rife -> $n frames"
  "$RF" -i $W/$s/hd -o $W/$s/slow -m $R/rife-ncnn-vulkan-20221029-windows/rife-v4.6 -n $n -f %05d.png >/dev/null 2>&1
done
mkdir -p $W/img
for f in f3557d7c-e4fa-45fe-9925-4fa301e1944d 476dce33-89cd-4a59-b7b0-a7df779f126b f27c81d7-6978-4c57-8e56-537fa96c53c5 e3af2663-05d6-4ee1-8d27-6a87866b4237 89b40766-a7a1-4bc7-b7e4-e672ec36d130; do
  echo "[$(date +%T)] img $f"; "$ES" -i /c/Users/Demo/Downloads/$f.png -o $W/img/$f.png -n realesrgan-x4plus -m $R/esrgan/models >/dev/null 2>&1
done
cp /c/Users/Demo/Downloads/6a99d399-e4a8-46a7-82e9-74b4c21eaddb.png $W/img/
echo "[$(date +%T)] DONE"