train_data_path=data/sl_data/train.json
test_data_path=data/sl_data/test.json
run_name=$1
temperature=0.9

if [ -z "$2" ]
  then
    data_path=$test_data_path
else
    if [ $2 = "train" ]; then
        data_path=$train_data_path
        run_name=$1-train
    else
        data_path=$test_data_path
        temperature=0.3
    fi
fi

model_path=exp/model_ckpt/$1
inference_path=exp/model_generation/$run_name.jsonl
visual_obj_path=exp/visual_objects/$run_name
output_figure_path=exp/figures/$run_name
log_path=exp/logs/$run_name

mkdir -p $log_path
mkdir -p exp/model_generation

echo "--------------------Inferencing--------------------" > $log_path/inference.txt
rm $inference_path
python3 src/test/inference.py --pretrained-path $model_path --in-path $data_path --out-path $inference_path --num-samples 5 --temperature $temperature --model-name llama3 > $log_path/inference.txt $3

echo "--------------------Parsing CAD objects--------------------" > $log_path/parsing_cad.txt
rm -rf $visual_obj_path
python3 src/rendering_utils/parser.py --in-path $inference_path --out-path $visual_obj_path > $log_path/parsing_cad.txt

echo "--------------------Parsing visual objects--------------------" > $log_path/parsing_visual.txt
python3 src/rendering_utils/parser_visual.py --data_folder $visual_obj_path > $log_path/parsing_visual.txt
python3 src/rendering_utils/ptl_sampler.py --in_dir $visual_obj_path --out_dir ptl > $log_path/sampling_ptl.out

echo "--------------------Rendering--------------------" > $log_path/rendering.txt
rm -rf $output_figure_path
export DISPLAY=:99
Xvfb :99 -screen 0 640x480x24 &
python3 src/rendering_utils/img_renderer.py --input_dir $visual_obj_path --output_dir $output_figure_path > $log_path/rendering.txt