Files
RoadTripsGenerator/Scripts/extraction_infos.sh
T
2026-08-17 23:50:11 +02:00

243 lines
10 KiB
Bash
Executable File

#!/bin/bash
# =============================================================================
# EXTRACTION DES MÉTADONNÉES — photos EXIF + traces GPS embarquées GoPro
#
# Produit deux CSV sans en-tête, consommés par genere_carte.py :
# export_photos.csv : fichier, date, lat, lon, alt
# export_videos.csv : fichier, date GPS, lat, lon, alt, vitesse
#
# Usage : ./extraction_infos.sh [dossier_voyage]
#
# Variables :
# VIDEO_GPS=first (défaut) un seul point GPS par rush — RAPIDE.
# C'est tout ce que genere_carte.py utilise pour poser
# le marqueur : `folium.Marker(location=pts_v[0])`.
# VIDEO_GPS=full déroule toute la trace GPS de chaque rush.
# Lit l'intégralité de chaque fichier : compter plusieurs
# minutes par rush sur un disque externe.
# VARIANTE=D|A|B|C force le mode d'extraction vidéo (voir plus bas)
# =============================================================================
set -uo pipefail
shopt -s nocaseglob nullglob
BASE_DIR="${1:-.}"
PHOTOS_DIR="$BASE_DIR/photos"
RUSHS_DIR="$BASE_DIR/rushs"
CSV_PHOTOS="$BASE_DIR/export_photos.csv"
CSV_VIDEOS="$BASE_DIR/export_videos.csv"
VIDEO_GPS="${VIDEO_GPS:-first}"
VARIANTE="${VARIANTE:-}"
if ! command -v exiftool >/dev/null 2>&1; then
echo "❌ exiftool introuvable. Installe-le : brew install exiftool"
exit 1
fi
echo "--- EXTRACTION DES MÉTADONNÉES ---"
echo "📂 Répertoire : $BASE_DIR"
TMPD="$(mktemp -d "${TMPDIR:-/tmp}/extr.XXXXXX")"
ERRLOG="$TMPD/err.log"
: > "$ERRLOG"
trap 'rm -rf "$TMPD"' EXIT
nlines() { awk 'END{print NR+0}' "$1" 2>/dev/null || echo 0; }
# =============================================================================
# FORMATS
#
# `exiftool -p` supprime la LIGNE ENTIÈRE dès qu'UN SEUL tag du format est
# absent. Sur un rush GoPro, [Main] porte GPSLatitude/GPSLongitude mais PAS
# GPSDateTime, alors que [Doc1] porte tout. Un format à six tags peut donc ne
# rien produire sur un fichier parfaitement géolocalisé.
#
# ⚠️ NE JAMAIS ajouter -f pour « forcer » l'affichage. Vérifié en test :
# `-ee -p '$FileName,$SampleRate'` sur un fichier multi-documents donne
# 0 ligne sans -f, mais UNE SEULE ligne « fichier,- » AVEC -f — l'option fait
# perdre à -ee son itération par document. C'est pire que le mal.
#
# La parade : trois modes, essayés sur les premiers rushs ; on garde celui
# qui produit quelque chose.
# A = format complet (date, lat, lon, alt, vitesse)
# B = format réduit (date, lat, lon) — tolère l'absence d'alt/vitesse
# C = format réduit + -ee3, qui détaille davantage les sous-documents
# =============================================================================
FMT_PHOTOS='"$FileName",$DateTimeOriginal,$GPSLatitude,$GPSLongitude,$GPSAltitude'
FMT_PHOTOS_MIN='"$FileName",$DateTimeOriginal,$GPSLatitude,$GPSLongitude'
FMT_V_COMPLET='"$FileName",$GPSDateTime,$GPSLatitude,$GPSLongitude,$GPSAltitude,$GPSSpeed'
FMT_V_MINIMAL='"$FileName",$GPSDateTime,$GPSLatitude,$GPSLongitude'
PARSE_GPS='
# Reconstruit des lignes CSV depuis la sortie « Tag : valeur » de exiftool.
# Robuste au format long ("GPS Date Time") comme court ("GPSDateTime"), et au
# prefixe de groupe [Main] / [Doc1].
{
ligne = $0
sub(/^\[[^]]*\][ \t]*/, "", ligne)
if (index(ligne, ":") == 0) next
tag = ligne; sub(/[ \t]*:.*/, "", tag); gsub(/[ \t]/, "", tag)
val = ligne; sub(/^[^:]*:[ \t]*/, "", val); gsub(/[ \t]+$/, "", val)
if (tag == "GPSDateTime" && dt == "") dt = val
else if (tag == "CreateDate" && cd == "") cd = val
else if (tag == "GPSLatitude") lat = val
else if (tag == "GPSLongitude") {
if (lat != "") {
n++
if (mode == "full") { plat[n] = lat; plon[n] = val }
else if (flat == "") { flat = lat; flon = val }
lat = ""
}
}
else if (tag == "GPSAltitude") { if (mode == "full" && n > 0) palt[n] = val }
else if (tag == "GPSSpeed") { if (mode == "full" && n > 0) pspd[n] = val }
if (mode != "full" && flat != "" && dt != "") {
print f "," dt "," flat "," flon ",,"; emis = 1; exit
}
}
END {
d = (dt != "" ? dt : cd)
if (d == "") exit
if (mode == "full") { for (i = 1; i <= n; i++) print f "," d "," plat[i] "," plon[i] "," palt[i] "," pspd[i] }
else if (!emis && flat != "") print f "," d "," flat "," flon ",,"
}
'
# extrait_video <mode> <fichier> -> lignes CSV sur stdout
#
# Mode D (par defaut) : on demande les tags un par un et on parse la sortie
# « Tag : valeur ». C est la seule methode qui atteint reellement [Doc1] :
# `-p` ne voit que [Main], lequel porte la position mais PAS GPSDateTime, si
# bien qu un format reclamant la date ne produit jamais rien. Verifie sur une
# sortie exiftool reelle de GoPro Hero 9 (GX011569.MP4).
# En mode rapide, awk sort des qu il a une date et une position : exiftool
# recoit SIGPIPE et cesse de lire le fichier.
extrait_video() {
local m="$1" f="$2" nom
nom=$(basename "$f")
case "$m" in
D) exiftool -ee -n -q -s -api largefilesupport=1 \
-GPSDateTime -GPSLatitude -GPSLongitude -GPSAltitude -GPSSpeed -CreateDate \
"$f" 2>>"$ERRLOG" \
| awk -v f="\"$nom\"" -v mode="$VIDEO_GPS" "$PARSE_GPS" ;;
A) exiftool -ee -n -q -api largefilesupport=1 -p "$FMT_V_COMPLET" "$f" 2>>"$ERRLOG" ;;
B) exiftool -ee -n -q -api largefilesupport=1 -p "$FMT_V_MINIMAL" "$f" 2>>"$ERRLOG" | awk '{print $0 ",,"}' ;;
C) exiftool -ee3 -n -q -api largefilesupport=1 -p "$FMT_V_MINIMAL" "$f" 2>>"$ERRLOG" | awk '{print $0 ",,"}' ;;
esac
}
# --- 1. PHOTOS ---------------------------------------------------------------
: > "$CSV_PHOTOS"
if [ -d "$PHOTOS_DIR" ]; then
echo "📸 Photos : analyse en cours…"
# UN SEUL appel exiftool pour tout le dossier : chaque lancement du binaire
# coûte 150-250 ms (Perl). Mesuré sur 150 photos : 29,3 s → 0,97 s.
exiftool -n -q -ext jpg -ext jpeg -p "$FMT_PHOTOS" "$PHOTOS_DIR" \
2>>"$ERRLOG" > "$TMPD/p1.csv"
# Seconde passe, format réduit : rattrape les photos écartées par la
# première faute d'altitude. Deux appels batch restent sous la seconde.
exiftool -n -q -ext jpg -ext jpeg -p "$FMT_PHOTOS_MIN" "$PHOTOS_DIR" \
2>>"$ERRLOG" | awk '{print $0 ","}' > "$TMPD/p2.csv"
# Fusion : tout p1, plus les fichiers de p2 absents de p1.
cat "$TMPD/p1.csv" > "$CSV_PHOTOS"
awk -F, 'NR==FNR {vu[$1]=1; next} !($1 in vu)' "$TMPD/p1.csv" "$TMPD/p2.csv" >> "$CSV_PHOTOS"
n1=$(nlines "$TMPD/p1.csv"); nt=$(nlines "$CSV_PHOTOS")
printf "📸 Photos : %d ligne(s)" "$nt"
[ "$nt" -gt "$n1" ] && printf " (dont %d récupérée(s) sans altitude)" "$((nt - n1))"
echo
else
echo "⚠️ Dossier photos non trouvé, étape ignorée."
fi
# --- 2. VIDÉOS ---------------------------------------------------------------
: > "$CSV_VIDEOS"
if [ ! -d "$RUSHS_DIR" ]; then
echo "⚠️ Dossier rushs non trouvé, étape ignorée."
else
rushs=()
for f in "$RUSHS_DIR"/*.mp4 "$RUSHS_DIR"/*.mov; do
[ -e "$f" ] || continue
# macOS depose un fichier AppleDouble « ._nom » a cote de chaque
# fichier sur les disques non-HFS (exFAT). Ce ne sont pas des videos.
case "$(basename "$f")" in ._*) continue ;; esac
rushs+=("$f")
done
total="${#rushs[@]}"
if [ "$total" -eq 0 ]; then
echo "⚠️ Aucun rush trouvé dans $RUSHS_DIR"
else
if [ "$VIDEO_GPS" = "full" ]; then
echo "🎬 Vidéos : $total rush(s), trace GPS COMPLÈTE"
echo " ⏳ Chaque fichier est lu intégralement."
else
echo "🎬 Vidéos : $total rush(s), premier point GPS seulement"
echo " (VIDEO_GPS=full pour dérouler la trace entière)"
fi
# --- Choix du mode d'extraction ---
# Testé sur 3 rushs au maximum : inutile de parcourir 552 fichiers
# pour découvrir qu'un mode ne marche pas.
if [ -z "$VARIANTE" ]; then
echo " 🔎 recherche du mode d'extraction…"
for essai in D A B C; do
k=0
for f in "${rushs[@]}"; do
k=$((k + 1)); [ "$k" -gt 3 ] && break
if [ -n "$(extrait_video "$essai" "$f" | head -n 1)" ]; then
VARIANTE="$essai"; break
fi
done
[ -n "$VARIANTE" ] && break
done
fi
case "${VARIANTE:-}" in
D) echo " ✔ mode D : lecture directe des tags (atteint [Doc1])" ;;
A) echo " ✔ mode A : format complet (date, lat, lon, alt, vitesse)" ;;
B) echo " ✔ mode B : format réduit (date, lat, lon)" ;;
C) echo " ✔ mode C : format réduit + -ee3 (sous-documents détaillés)" ;;
*) echo " ⚠️ Aucun mode ne donne de résultat sur les 3 premiers rushs."
echo " Diagnostic : exiftool -ee -G3 -n <un_rush> | grep -i gps"
VARIANTE=A ;;
esac
i=0; avec=0; sans=0
T0=$(date +%s)
for f in "${rushs[@]}"; do
i=$((i + 1))
printf " [%3d/%3d] %-28s " "$i" "$total" "$(basename "$f")"
avant=$(nlines "$CSV_VIDEOS")
if [ "$VIDEO_GPS" = "full" ]; then
extrait_video "$VARIANTE" "$f" >> "$CSV_VIDEOS"
else
# `head -n 1` ferme le tuyau : exiftool reçoit SIGPIPE et
# s'arrête sans lire la fin du fichier.
extrait_video "$VARIANTE" "$f" | head -n 1 >> "$CSV_VIDEOS"
fi
ajoutes=$(( $(nlines "$CSV_VIDEOS") - avant ))
if [ "$ajoutes" -gt 0 ]; then
if [ "$VIDEO_GPS" = "full" ]; then printf "✓ %d points\n" "$ajoutes"; else echo "✓"; fi
avec=$((avec + 1))
else
echo "— pas de GPS"; sans=$((sans + 1))
fi
done
T1=$(date +%s); EL=$((T1 - T0))
printf "🎬 Vidéos : %d point(s) GPS, %d rush(s) localisé(s)" "$(nlines "$CSV_VIDEOS")" "$avec"
[ "$sans" -gt 0 ] && printf ", %d sans GPS" "$sans"
printf " — %d min %02d s\n" $((EL / 60)) $((EL % 60))
fi
fi
echo "✅ Générés : $CSV_PHOTOS et $CSV_VIDEOS"