#!/bin/sh

pwd=`pwd`

img=../input/img_003.png

# img=/Users/skal/black_512x512_rgba.png
#img=/Users/skal/rgba_0_0_0_0.png
check_pt=../checkpoints/checkpoint_epoch_10000.pth
#check_pt=../chk_5000_3x3x3.pt

#../train_cnn.py --layers 3 --kernel_sizes 3,3,3 --epochs 10000 --batch_size 8 --input  ../input/ --target ../target_2/ --checkpoint-every 1000
#../train_cnn.py --export-only ${check_pt}
#../train_cnn.py --export-only ${check_pt} --infer ${img} --output test/toto.png

#../train_cnn.py --layers 2 --kernel_sizes 1,1 --epochs 10 --batch_size 5 --input ../input/ --target ../target_2/ --checkpoint-every 10
#../train_cnn.py --export-only ${check_pt}
#../train_cnn.py --export-only ${check_pt} --infer ${img} --output test/toto.png

## XXX uncomment!
../train_cnn.py --export-only ${check_pt} \
  --infer ${img} \
  --output ref/toto.png --save-intermediates ref/ # --debug-hex

echo "== GENERATE SHADERS =="
echo
cd ../../
./training/train_cnn.py --export-only ${pwd}/${check_pt}

echo "== COMPILE =="
echo
cmake --build build -j4 --target cnn_test
cd ${pwd}

echo "== RUN =="
echo
rm -f cur/toto.png
../../build/cnn_test ${img} cur/toto.png --save-intermediates cur/ --layers 3 # --debug-hex

open cur/*.png ref/*.png

echo "open cur/*.png ref/*.png"

#pngcrush -rem gAMA -rem sRGB cur/toto.png toto.png && mv toto.png cur/toto.png
#pngcrush -rem gAMA -rem sRGB cur/layer_0.png toto.png && mv toto.png cur/layer_0.png