diff --git a/run.sh b/run.sh index 9dec617..08ab1cc 100755 --- a/run.sh +++ b/run.sh @@ -5,7 +5,7 @@ # Options: # -r RESOLUTION Résolution en m/px, ou multiples séparées par virgules (défaut: 0.5, ex: 0.5,0.2) # -w WORKERS Nombre de workers parallèles (défaut: 1) -# -g Activer l'accélération GPU +# -g [LISTE] Sélectionner le(s) GPU(s) : -g (auto), -g 0, -g 0,2, -g all # -v Mode verbeux # --debug Mode debug # -f / --force Régénérer tous les fichiers @@ -37,26 +37,38 @@ if [ $# -eq 0 ]; then echo "Options:" echo " -r RESOLUTION Résolution en m/px, ou multiples (défaut: 0.5, ex: 0.5,0.2)" echo " -w WORKERS Nombre de workers CPU parallèles (défaut: 1)" - echo " -g Activer l'accélération GPU NVIDIA" + echo " -g [LISTE] Sélectionner le(s) GPU(s) : -g (auto), -g 0, -g 0,2, -g all" echo " -v Mode verbeux (timestamps + niveaux)" echo " --debug Mode debug (détails internes fichier:ligne)" - echo " -f / --force Régénérer tous les fichiers même si les WebP existent" + echo " -f / --force Régénérer tous les fichiers même si les AVIF existent" echo " --force-classification" echo " Reclassifier le sol même si le fichier .las existe" - echo " --keep-tif Conserver les TIFF pour régénérer les WebP" + echo " --keep-tif Conserver les TIFF pour régénérer les AVIF" echo " --ground-classification {auto,smrf,csf}" echo " Méthode de classification du sol (défaut: auto)" + echo " --quality N Qualité image 1-100 (défaut: 98, 100=lossless)" + echo " --lossless Compression lossless (équivalent à --quality 100)" + echo " --format FMT Format de sortie : avif (défaut) ou webp" + echo " --only VIZ... Générer uniquement ces visualisations" + echo " --skip VIZ... Exclure ces visualisations" echo " --file NOM... Traiter un ou plusieurs fichiers LAZ (nom complet sans .laz/.las)" echo " --test Exécuter les tests unitaires" echo " -h Afficher cette aide" echo "" + echo "Visualisations disponibles:" + echo " hillshade slope mslrm sailore pos_open neg_open" + echo " svf aniso_open roughness wavelet flow_acc" + echo " ortho topo" + echo "" echo "Exemples:" - echo " $0 -g # GPU, auto" + echo " $0 -g # GPU auto (premier disponible)" + echo " $0 -g 0 # GPU index 0" + echo " $0 -g 0,2 # GPUs 0 et 2" + echo " $0 -g all # Tous les GPUs" echo " $0 -g -w 4 # GPU + 4 workers" - echo " $0 -g -v # GPU + verbeux" echo " $0 -g -r 0.2 # Haute résolution" echo " $0 -g -r 0.5,0.2 # Multi-résolution (0.5m + 0.2m)" - echo " $0 -g --force # Régénérer WebP (DTM conservé si --keep-tif)" + echo " $0 -g --force # Régénérer AVIF (DTM conservé si --keep-tif)" echo " $0 -g --force-classification # Reclassifier le sol seulement" echo " $0 -g --ground-classification csf # Forcer CSF (terrain complexe)" echo " $0 -g --file LHD_...IGN69.copc # Un fichier" @@ -66,6 +78,7 @@ fi RESOLUTION=0.5 WORKERS=1 GPU_FLAG="" +GPU_ARG="" VERBOSE_FLAG="" FORCE_FLAG="" FILE_ARGS="" @@ -76,13 +89,24 @@ QUALITY="" FORMAT_FLAG="" ONLY_FLAG="" SKIP_FLAG="" +TEST_FLAG=0 # Parse arguments manually (more robust than getopts for mixed short/long options) while [ $# -gt 0 ]; do case "$1" in -r) RESOLUTION="$2"; shift 2 ;; -w) WORKERS="$2"; shift 2 ;; - -g) GPU_FLAG="--gpus all"; shift ;; + -g) + GPU_FLAG="--gpus all" + # Check if next arg is a GPU selector (not another flag) + if [ $# -gt 1 ] && [[ ! "$2" =~ ^- ]] && [[ "$2" != "all" || "$2" == "all" ]]; then + GPU_ARG="$2" + GPU_FLAG="--gpus all" + shift 2 + else + shift + fi + ;; -v) VERBOSE_FLAG="-v"; shift ;; -f) FORCE_FLAG="--force"; shift ;; --debug) VERBOSE_FLAG="--debug"; shift ;; @@ -97,7 +121,7 @@ while [ $# -gt 0 ]; do --only) shift; ONLY_FLAG="--only"; while [ $# -gt 0 ] && [[ ! "$1" =~ ^- ]]; do ONLY_FLAG="$ONLY_FLAG $1"; shift; done ;; --skip) shift; SKIP_FLAG="--skip"; while [ $# -gt 0 ] && [[ ! "$1" =~ ^- ]]; do SKIP_FLAG="$SKIP_FLAG $1"; shift; done ;; --file) shift; while [ $# -gt 0 ] && [[ ! "$1" =~ ^- ]]; do FILE_ARGS="$FILE_ARGS $1"; shift; done ;; - --test) ;; # Handled below + --test) TEST_FLAG=1 ;; -h|--help|-help) echo "Pipeline LiDAR Archéologique" echo "" @@ -106,13 +130,13 @@ while [ $# -gt 0 ]; do echo "Options:" echo " -r RESOLUTION Résolution en m/px (défaut: 0.5)" echo " -w WORKERS Nombre de workers CPU parallèles (défaut: 1)" - echo " -g Activer l'accélération GPU NVIDIA" + echo " -g [LISTE] Sélectionner le(s) GPU(s) : -g (auto), -g 0, -g 0,2, -g all" echo " -v Mode verbeux (timestamps + niveaux)" echo " --debug Mode debug (détails internes fichier:ligne)" - echo " -f / --force Régénérer tous les fichiers même si les WebP existent" + echo " -f / --force Régénérer tous les fichiers même si les AVIF existent" echo " --force-classification" echo " Reclassifier le sol même si le fichier .las existe" - echo " --keep-tif Conserver les TIFF pour régénérer les WebP" + echo " --keep-tif Conserver les TIFF pour régénérer les AVIF" echo " --ground-classification {auto,smrf,csf}" echo " Méthode de classification du sol (défaut: auto)" echo " --quality N Qualité image 1-100 (défaut: 98, 100=lossless)" @@ -125,20 +149,24 @@ while [ $# -gt 0 ]; do echo " -h Afficher cette aide" echo "" echo "Visualisations disponibles:" - echo " hillshade slope aspect curvature lrm pos_open neg_open" - echo " mslrm tpi sailore roughness anomalies wavelet flow ortho topo" + echo " hillshade slope mslrm sailore pos_open neg_open" + echo " svf aniso_open roughness wavelet flow_acc" + echo " ortho topo" echo "" echo "Exemples:" - echo " $0 -g # GPU, auto" + echo " $0 -g # GPU auto (premier disponible)" + echo " $0 -g 0 # GPU index 0" + echo " $0 -g 0,2 # GPUs 0 et 2" + echo " $0 -g all # Tous les GPUs" echo " $0 -g -w 4 # GPU + 4 workers" echo " $0 -g -v # GPU + verbeux" echo " $0 -g -r 0.2 # Haute résolution" - echo " $0 -g -r 0.5,0.2 # Multi-résolution (0.5m + 0.2m)" - echo " $0 -g --force # Régénérer WebP" - echo " $0 -g --only hillshade svf lrm # Seulement 3 visualisations" + echo " $0 -g -r 0.5,0.2 # Multi-résolution (0.5m + 0.2m)" + echo " $0 -g --force # Régénérer AVIF" + echo " $0 -g --only hillshade svf # Seulement 2 visualisations" echo " $0 -g --skip ortho topo # Sans les overlays IGN" - echo " $0 -g --lossless # WebP lossless" - echo " $0 -g --quality 90 # WebP qualité 90" + echo " $0 -g --lossless # AVIF lossless" + echo " $0 -g --quality 90 # AVIF qualité 90" echo " $0 -g --ground-classification csf # Forcer CSF (terrain complexe)" echo " $0 -g --file LHD_...IGN69.copc # Un fichier" exit 0 @@ -151,7 +179,7 @@ done FILE_ARGS=$(echo "$FILE_ARGS" | xargs) # Check for --test flag first -if [[ " $* " == *" --test "* ]]; then +if [ "$TEST_FLAG" -eq 1 ]; then # Build l'image si elle n'existe pas if ! docker image inspect "$IMAGE_NAME" >/dev/null 2>&1; then echo "Build de l'image Docker..." @@ -181,7 +209,7 @@ echo " Pipeline LiDAR Archéologique" echo "============================================" echo " Résolution : ${RESOLUTION}m/px" echo " Workers : ${WORKERS}" -echo " GPU : $([ -n "$GPU_FLAG" ] && echo 'OUI' || echo 'non')" +echo " GPU : $([ -n "$GPU_FLAG" ] && echo "OUI${GPU_ARG:+ ($GPU_ARG)}" || echo 'non')" echo " Verbeux : $([ -n "$VERBOSE_FLAG" ] && echo 'OUI' || echo 'non')" echo " Force : $([ -n "$FORCE_FLAG" ] && echo 'OUI' || echo 'non')" echo " Force classif.: $([ -n "$FORCE_CLASSIFY_FLAG" ] && echo 'OUI' || echo 'non')" @@ -200,6 +228,12 @@ fi echo "============================================" CMD_ARGS="-o /data/output -r $RESOLUTION -w $WORKERS $VERBOSE_FLAG $FORCE_FLAG $FORCE_CLASSIFY_FLAG $KEEP_TIF_FLAG $QUALITY $FORMAT_FLAG" +if [ -n "$GPU_ARG" ]; then + CMD_ARGS="$CMD_ARGS -g $GPU_ARG" +fi +if [ -z "$GPU_ARG" ] && [ -n "$GPU_FLAG" ]; then + CMD_ARGS="$CMD_ARGS -g" +fi if [ -n "$GROUND_METHOD" ]; then CMD_ARGS="$CMD_ARGS --ground-classification $GROUND_METHOD" fi @@ -219,4 +253,4 @@ docker run --rm --init $GPU_FLAG \ -v "${OUTPUT_DIR}:/data/output" \ "$IMAGE_NAME" \ python3 -m lidar_pipeline /data/input \ - $CMD_ARGS \ No newline at end of file + $CMD_ARGS