#compdef ns-download-data

# AUTOMATICALLY GENERATED by `shtab`


_shtab_tyro_ns_download_data_commands() {
  local _commands=(
    "blender:Download the blender dataset."
    "dnerf:Download the dnerf dataset."
    "eyefultower:Download the EyefulTower dataset. Use the --help flag with the \`eyefultower\` subcommand to see all available datasets. Find more information about the dataset at https\:\/\/github.com\/facebookresearch\/EyefulTower."
    "mill19:Download the Mill 19 dataset."
    "nerfosr:Download the NeRF-OSR dataset."
    "nerfstudio:Download data in the Nerfstudio format. If you are interested in the Nerfstudio Dataset subset from the SIGGRAPH 2023 paper, you can obtain that by using --capture-name nerfstudio-dataset or by visiting Google Drive directly at\: https\:\/\/drive.google.com\/drive\/folders\/19TV6kdVGcmg3cGZ1bNIUnBBMD-iQjRbG\?usp\=drive_link."
    "phototourism:Download the phototourism dataset."
    "record3d:Download the record3d dataset."
    "sdfstudio:Download the sdfstudio dataset."
    "sitcoms3d:Download the sitcoms3D dataset."
  )
  _describe 'ns-download-data commands' _commands
}

_shtab_tyro_ns_download_data_options=(
  {-h,--help}"[show this help message and exit]:help:"
)

_shtab_tyro_ns_download_data_blender_options=(
  {-h,--help}"[show this help message and exit]:help:"
  "--save-dir[The directory to save the dataset to (default\: data)]:save-dir:_files -/"
)

_shtab_tyro_ns_download_data_dnerf_options=(
  {-h,--help}"[show this help message and exit]:help:"
  "--save-dir[The directory to save the dataset to (default\: data)]:save-dir:_files -/"
)

_shtab_tyro_ns_download_data_eyefultower_options=(
  {-h,--help}"[show this help message and exit]:help:"
  "--save-dir[The directory to save the dataset to (default\: data)]:save-dir:_files -/"
  "--capture-name[(default\: )]:capture-name:(all apartment kitchen office1a office1b office2 office_view1 office_view2 riverview seating_area table workshop raf_emptyroom raf_furnishedroom)"
  "--resolution-name[(default\: )]:resolution-name:(all jpeg_1k jpeg_2k jpeg_4k jpeg_8k exr_1k exr_2k)"
)

_shtab_tyro_ns_download_data_mill19_options=(
  {-h,--help}"[show this help message and exit]:help:"
  "--save-dir[The directory to save the dataset to (default\: data)]:save-dir:_files -/"
  "--capture-name[(default\: building)]:capture-name:(building rubble all)"
)

_shtab_tyro_ns_download_data_nerfosr_options=(
  {-h,--help}"[show this help message and exit]:help:"
  "--save-dir[The directory to save the dataset to (default\: data)]:save-dir:_files -/"
  "--capture-name[(default\: europa)]:capture-name:(europa lk2 lwp rathaus schloss st stjacob stjohann trevi all)"
)

_shtab_tyro_ns_download_data_nerfstudio_options=(
  {-h,--help}"[show this help message and exit]:help:"
  "--save-dir[The directory to save the dataset to (default\: data)]:save-dir:_files -/"
  "--capture-name[(default\: bww_entrance)]:capture-name:(bww_entrance campanile desolation library poster redwoods2 storefront vegetation Egypt person kitchen plane dozer floating-tree aspen stump sculpture Giannini-Hall all nerfstudio-dataset)"
)

_shtab_tyro_ns_download_data_phototourism_options=(
  {-h,--help}"[show this help message and exit]:help:"
  "--save-dir[The directory to save the dataset to (default\: data)]:save-dir:_files -/"
  "--capture-name[(default\: brandenburg-gate)]:capture-name:(brandenburg-gate buckingham-palace colosseum-exterior grand-palace-brussels notre-dame-facade westminster-palace pantheon-exterior taj-mahal temple-nara trevi-fountain all)"
)

_shtab_tyro_ns_download_data_record3d_options=(
  {-h,--help}"[show this help message and exit]:help:"
  "--save-dir[The directory to save the dataset to (default\: data)]:save-dir:_files -/"
  "--capture-name[(default\: bear)]:capture-name:(bear)"
)

_shtab_tyro_ns_download_data_sdfstudio_options=(
  {-h,--help}"[show this help message and exit]:help:"
  "--save-dir[The directory to save the dataset to (default\: data)]:save-dir:_files -/"
  "--dataset-name[(default\: sdfstudio-demo-data)]:dataset-name:(sdfstudio-demo-data dtu replica scannet tanks-and-temple tanks-and-temple-highres heritage neural-rgbd-data all)"
)

_shtab_tyro_ns_download_data_sitcoms3d_options=(
  {-h,--help}"[show this help message and exit]:help:"
  "--save-dir[The directory to save the dataset to (default\: data)]:save-dir:_files -/"
)


_shtab_tyro_ns_download_data() {
  local context state line curcontext="$curcontext" one_or_more='(-)*' remainder='(*)'

  if ((${_shtab_tyro_ns_download_data_options[(I)${(q)one_or_more}*]} + ${_shtab_tyro_ns_download_data_options[(I)${(q)remainder}*]} == 0)); then  # noqa: E501
    _shtab_tyro_ns_download_data_options+=(': :_shtab_tyro_ns_download_data_commands' '*::: :->ns-download-data')
  fi
  _arguments -C -s $_shtab_tyro_ns_download_data_options

  case $state in
    ns-download-data)
      words=($line[1] "${words[@]}")
      (( CURRENT += 1 ))
      curcontext="${curcontext%:*:*}:_shtab_tyro_ns_download_data-$line[1]:"
      case $line[1] in
        blender) _arguments -C -s $_shtab_tyro_ns_download_data_blender_options ;;
        dnerf) _arguments -C -s $_shtab_tyro_ns_download_data_dnerf_options ;;
        eyefultower) _arguments -C -s $_shtab_tyro_ns_download_data_eyefultower_options ;;
        mill19) _arguments -C -s $_shtab_tyro_ns_download_data_mill19_options ;;
        nerfosr) _arguments -C -s $_shtab_tyro_ns_download_data_nerfosr_options ;;
        nerfstudio) _arguments -C -s $_shtab_tyro_ns_download_data_nerfstudio_options ;;
        phototourism) _arguments -C -s $_shtab_tyro_ns_download_data_phototourism_options ;;
        record3d) _arguments -C -s $_shtab_tyro_ns_download_data_record3d_options ;;
        sdfstudio) _arguments -C -s $_shtab_tyro_ns_download_data_sdfstudio_options ;;
        sitcoms3d) _arguments -C -s $_shtab_tyro_ns_download_data_sitcoms3d_options ;;
      esac
  esac
}



typeset -A opt_args

if [[ $zsh_eval_context[-1] == eval ]]; then
  # eval/source/. command, register function for later
  compdef _shtab_tyro_ns_download_data -N ns-download-data
else
  # autoload from fpath, call function directly
  _shtab_tyro_ns_download_data "$@"
fi

