set -x

# n_proc=4
# n_points=4096
# focal=525

# pre-requisites: kaolin and nvdiffrast
# pip install kaolin==0.16.0 -f https://nvidia-kaolin.s3.us-east-2.amazonaws.com/torch-2.3.0_cu118.html

n_proc=1
n_points=4096
focal=24 # as in blender config 


# shape_root=/mnt/lustre/share/fzhong/shapenet/ShapeNetCore.v2/02958343/fffb1660a38af30ba4cf3601fb6b2442/models/
# shape_root=/mnt/lustre/share/fzhong/shapenet/ShapeNetCore.v2/02958343/
# save_root=/mnt/lustre/yslan/3D_Dataset/shapenet/pcd/car

# shape_root=/mnt/lustre/share/fzhong/shapenet/ShapeNetCore.v2/03001627/
# save_root=/mnt/lustre/yslan/3D_Dataset/shapenet/pcd-full/chair

# shape_root=/mnt/lustre/share/fzhong/shapenet/ShapeNetCore.v2/02691156/
# save_root=/mnt/lustre/yslan/3D_Dataset/shapenet/pcd/plane

# shape_root=/mnt/lustre/share/fzhong/shapenet/ShapeNetCore.v2/03001627/
# save_root=/mnt/lustre/yslan/3D_Dataset/shapenet/pcd-full/chair

# ! render gt
# shape_root=/mnt/sfs-common/yslan/Dataset/GSO/gso-unzip/gso-unzip
# save_root=/mnt/sfs-common/yslan/Repo/3dgen/FID-KID-Outputdir/3D-metrics/gso

# ! use FPS later:
# splatter-img LGM

# for method_name in ln3diff One-2-3-45 OpenLRM shape-e Lara CRM splatter-img
# for method_name in ln3diff One-2-3-45 OpenLRM
# for method_name in shape-e Lara CRM splatter-img
# for method_name in CRM splatter-img

# for method_name in splatter-img LGM

# for method_name in splatter-img/Animals LGM/Animals
# for method_name in splatter-img LGM

for method_name in LGM_fixpose

do

# shape_root=/mnt/sfs-common/yslan/Repo/3dgen/FID-KID-Outputdir/${method_name}
# save_root=/mnt/sfs-common/yslan/Repo/3dgen/FID-KID-Outputdir/3D-metrics/${method_name}

shape_root=/mnt/sfs-common/yslan/Repo/3dgen/FID-KID-Outputdir-objv/${method_name}
save_root=/mnt/sfs-common/yslan/Repo/3dgen/FID-KID-Outputdir-objv/3D-metrics-fps/${method_name}

python fps_gs.py \
    --shape_root ${shape_root} \
    --save_root ${save_root} \
    --n_proc ${n_proc} \
    --n_points ${n_points} \
    --image_height 512 \
    --image_width 512 \
    --focal_length_x ${focal} \
    --focal_length_y ${focal} \
    --principal_point_x 256 \
    --principal_point_y 256 \

done