# ROOT_DIR="/scratch/izar/skorokho/data/recordings_09082025"

# TO_PROCESS=(
#     chair_living_room chair_my_room corridor dustbin 
#     fabric_bin_corridor fabric_bin_living_room living_room 
#     outdoor_container outdoor_green stone
# )

# for ((i=0; i<${#TO_PROCESS[@]}; i++)); do
# # for ((i=1; i<2; i++)); do
#     echo "processing ${TO_PROCESS[$i]}"
#     INPUT_PATH="${ROOT_DIR}/${TO_PROCESS[$i]}"
#     OUTPUT_PATH="${ROOT_DIR}/ns_${TO_PROCESS[$i]}"
#     sai-cli process $INPUT_PATH --key_frame_distance 0.01 $OUTPUT_PATH 
# done

# for ((i=0; i<${#TO_PROCESS[@]}; i++)); do
#     echo "processing ${TO_PROCESS[$i]}"
#     INPUT_DIR="${ROOT_DIR}/ns_${TO_PROCESS[$i]}"
#     OUTPUT_DIR="${ROOT_DIR}/ns_processed_${TO_PROCESS[$i]}"
#     python3 scripts/real_to_scenenet_before_3dgs.py \
#         --input_dir $INPUT_DIR \
#         --save_dir $OUTPUT_DIR
# done

# for i in "$@"; do
#     INPUT_DIR="${ROOT_DIR}/ns_processed_${TO_PROCESS[$i]}"
#     OUTPUT_DIR="${ROOT_DIR}/splat_${TO_PROCESS[$i]}"
#     ns-train splatfacto \
#         --data $INPUT_DIR \
#         --output-dir $OUTPUT_DIR \
#         --pipeline.model.background-color black \
#         --viewer.quit-on-train-completion True \
#         nerfstudio-data \
#         --orientation-method none --center-method none --auto-scale-poses False
# done

# python3 scripts/create_real_oos.py \
#     --obj_scene_dir "/scratch/izar/skorokho/data/recordings_09082025/ns_processed_fabric_bin_corridor/" \
#     --scene_dir  "/scratch/izar/skorokho/data/recordings_09082025/ns_processed_corridor/" \
#     --obj_dir "/scratch/izar/skorokho/data/recordings_09082025/ns_processed_fabric_bin_living_room/" \
#     --save_dir "/scratch/izar/skorokho/data/recordings_09082025/ns_processed_fabric_bin_in_corridor/"

# python3 scripts/create_real_oos.py \
#     --obj_scene_dir "/scratch/izar/skorokho/data/recordings_09082025/ns_processed_outdoor_green/" \
#     --scene_dir  "/scratch/izar/skorokho/data/recordings_09082025/ns_processed_outdoor_green/" \
#     --obj_dir "/scratch/izar/skorokho/data/recordings_09082025/ns_processed_stone/" \
#     --save_dir "/scratch/izar/skorokho/data/recordings_09082025/ns_processed_stone_in_garden/"

# python3 scripts/create_real_oos.py \
#     --obj_scene_dir "/scratch/izar/skorokho/data/recordings_09082025/ns_processed_outdoor_container/" \
#     --scene_dir  "/scratch/izar/skorokho/data/recordings_09082025/ns_processed_outdoor_container/" \
#     --obj_dir "/scratch/izar/skorokho/data/recordings_09082025/ns_processed_dustbin/" \
#     --save_dir "/scratch/izar/skorokho/data/recordings_09082025/ns_processed_dustbins_ifo_container/"

# python3 scripts/create_real_oos.py \
#     --obj_scene_dir "/scratch/izar/skorokho/data/recordings_09082025/ns_processed_chair_living_room" \
#     --scene_dir  "/scratch/izar/skorokho/data/recordings_09082025/ns_processed_living_room" \
#     --obj_dir "/scratch/izar/skorokho/data/recordings_09082025/ns_processed_chair_my_room" \
#     --save_dir "/scratch/izar/skorokho/data/recordings_09082025/ns_processed_chair_in_living_room"

# ROOT_DIR="/scratch/izar/skorokho/data/toster2"

# TO_PROCESS=(
#     dark_toster dark_notoster light_toster light_notoster
# )

# for i in "$@"; do
#     INPUT_DIR="${ROOT_DIR}/ns_processed_${TO_PROCESS[$i]}"
#     OUTPUT_DIR="${ROOT_DIR}/splat_${TO_PROCESS[$i]}"
#     ns-train splatfacto \
#         --data $INPUT_DIR \
#         --output-dir $OUTPUT_DIR \
#         --pipeline.model.background-color black \
#         --viewer.quit-on-train-completion True \
#         nerfstudio-data \
#         --orientation-method none --center-method none --auto-scale-poses False
# done

python3 scripts/create_real_oos.py \
    --obj_scene_dir "/scratch/izar/skorokho/data/toster2/ns_processed_light_toster" \
    --scene_dir  "/scratch/izar/skorokho/data/toster2/ns_processed_light_notoster" \
    --obj_dir "/scratch/izar/skorokho/data/toster2/ns_processed_dark_toster" \
    --save_dir "/scratch/izar/skorokho/data/toster2/dark_to_light"


python3 scripts/create_real_oos.py \
    --obj_scene_dir "/scratch/izar/skorokho/data/toster2/ns_processed_dark_toster" \
    --scene_dir  "/scratch/izar/skorokho/data/toster2/ns_processed_dark_notoster" \
    --obj_dir "/scratch/izar/skorokho/data/toster2/ns_processed_light_toster" \
    --save_dir "/scratch/izar/skorokho/data/toster2/light_to_dark"
