mapillary_download/find_france_photos_and_move.py

###
# Filtrage de photos situées en France dans un dossier contenant des séquences de photo géolocalisées
# Rechercher et déplacer automatiquement les photos géolocalisées dans une certaine bounding box et les déplacer dans le dossier de destination.
#
# utilisation:
# python find_france_photos_and_move.py --source_dir /chemin/du/répertoire/source --destination_dir /chemin/du/répertoire/destination
#
###
import argparse
import os
import shutil
import exifread

# Définition du rectangle entourant la France métropolitaine et un peu autour
france_bbox: tuple[float, float, float, float] = (
    42.0,
    -5.0,
    51.0,
    10.0,
)  # (lat_min, lon_min, lat_max, lon_max)
# Définition du rectangle entourant la France métropolitaine et un peu autour
france_bbox: tuple[float, float, float, float] = (
    42.0,
    -5.0,
    51.0,
    10.0,
)  # (lat_min, lon_min, lat_max, lon_max)

# Définition du rectangle entourant la Guadeloupe
guadeloupe_bbox: tuple[float, float, float, float] = (15.8, -61.8, 17.3, -59.3)

# Définition du rectangle entourant la Martinique
martinique_bbox: tuple[float, float, float, float] = (14.3, -61.3, 15.1, -59.3)

# Définition du rectangle entourant la Guyane française
guyane_bbox: tuple[float, float, float, float] = (2.0, -54.5, 6.5, -51.5)

# Définition du rectangle entourant La Réunion
reunion_bbox: tuple[float, float, float, float] = (-21.3, 55.2, -20.8, 55.8)

# Définition du rectangle entourant Mayotte
mayotte_bbox: tuple[float, float, float, float] = (-13.0, 45.0, -12.5, 45.5)

# Définition du rectangle entourant Saint-Pierre-et-Miquelon
spm_bbox: tuple[float, float, float, float] = (46.7, -56.2, 47.1, -55.6)

# Définition du rectangle entourant les îles de Saint-Martin et Saint-Barthélemy
stm_sbh_bbox: tuple[float, float, float, float] = (18.0, -64.5, 18.5, -62.5)

# Définition du rectangle entourant Wallis-et-Futuna
wf_bbox: tuple[float, float, float, float] = (-13.3, -176.2, -13.1, -175.8)

# Définition du rectangle entourant la Nouvelle-Calédonie
nc_bbox: tuple[float, float, float, float] = (-22.5, 165.5, -18.5, 169.5)

# Définition du rectangle entourant la Polynésie française
pf_bbox: tuple[float, float, float, float] = (-27.5, -140.0, -7.5, -134.0)

# Définition du rectangle entourant les Terres australes et antarctiques françaises
taaf_bbox: tuple[float, float, float, float] = (-49.5, 68.5, -37.5, 77.5)

# Chemin du répertoire source
source_dir: str = "/home/cipherbliss/Téléchargements/FIBRELAND/TEST_IN_FR/"

# Chemin du répertoire destination
destination_dir: str = "/home/cipherbliss/Téléchargements/FIBRELAND/IN_FRANCE/"
sequence_folder: str = "principale_sequence"
count_files_all: int = 0
count_files_moved: int = 0
# Crée le répertoire destination si il n'existe pas
if not os.path.exists(destination_dir):
    os.makedirs(destination_dir)


# Fonction pour déplacer un fichier si il est dans le rectangle de la France
def move_file_if_in_france(filepath, sequence_folder):
    global count_files_all
    global count_files_moved

    # Ouvre le fichier image et lit les informations EXIF
    latitude, longitude = get_gps_info(filepath)

    if latitude and longitude:
        print(f"Latitude: {latitude}, Longitude: {longitude}")
        if are_lat_lon_in_france(latitude, longitude):
            move_file_in_destination(filepath, sequence_folder)
    else:
        print("Informations GPS non trouvées")


def move_file_in_destination(filepath, sequence_folder):
    global count_files_moved
    # Déplace le fichier dans le sous-répertoire "photos_in_france"
    dest_subdir = os.path.join(
        destination_dir, sequence_folder, os.path.basename(os.path.dirname(filepath))
    )
    if not os.path.exists(dest_subdir):
        os.makedirs(dest_subdir)
    shutil.move(filepath, os.path.join(dest_subdir, filepath))
    count_files_moved += 1
    print(f"Moved {filepath} to {dest_subdir}")
    return True


def are_lat_lon_in_france(gps_lat, gps_lon):
    """
    recherche d'une zone du territoire français


    France métropolitaine : 551 695 km²
    Terres australes et antarctiques françaises : 432 000 km²
    Guyane française : 83 534 km²
    Nouvelle-Calédonie : 18 575 km²
    Polynésie française : 4 167 km²
    La Réunion : 2 512 km²
    Martinique : 1 128 km²
    Guadeloupe : 1 628 km²
    Mayotte : 374 km²
    Saint-Pierre-et-Miquelon : 242 km²
    Wallis-et-Futuna : 142 km²
    Saint-Martin et Saint-Barthélemy : 53 km²


    :param gps_lat:
    :param gps_lon:
    :return:
    """
    global france_bbox, guyane_bbox, nc_bbox, pf_bbox, reunion_bbox, guadeloupe_bbox, martinique_bbox, mayotte_bbox, spm_bbox, stm_sbh_bbox, wf_bbox, taaf_bbox

    print("lat lon :", gps_lat, gps_lon)

    if (
        france_bbox[0] <= gps_lat <= france_bbox[2]
        and france_bbox[1] <= gps_lon <= france_bbox[3]
    ):
        return "France métropolitaine"
    elif (
        taaf_bbox[0] <= gps_lat <= taaf_bbox[2]
        and taaf_bbox[1] <= gps_lon <= taaf_bbox[3]
    ):
        return "Terres australes et antarctiques françaises"
    elif (
        guyane_bbox[0] <= gps_lat <= guyane_bbox[2]
        and guyane_bbox[1] <= gps_lon <= guyane_bbox[3]
    ):
        return "Guyane française"
    elif (
        reunion_bbox[0] <= gps_lat <= reunion_bbox[2]
        and reunion_bbox[1] <= gps_lon <= reunion_bbox[3]
    ):
        return "La Réunion"
    elif wf_bbox[0] <= gps_lat <= wf_bbox[2] and wf_bbox[1] <= gps_lon <= wf_bbox[3]:
        return "Wallis-et-Futuna"
    elif (
        stm_sbh_bbox[0] <= gps_lat <= stm_sbh_bbox[2]
        and stm_sbh_bbox[1] <= gps_lon <= stm_sbh_bbox[3]
    ):
        return "Saint-Martin et Saint-Barthélemy"
    elif (
        spm_bbox[0] <= gps_lat <= spm_bbox[2] and spm_bbox[1] <= gps_lon <= spm_bbox[3]
    ):
        return "Saint-Pierre-et-Miquelon"
    elif (
        mayotte_bbox[0] <= gps_lat <= mayotte_bbox[2]
        and mayotte_bbox[1] <= gps_lon <= mayotte_bbox[3]
    ):
        return "Mayotte"
    elif (
        martinique_bbox[0] <= gps_lat <= martinique_bbox[2]
        and martinique_bbox[1] <= gps_lon <= martinique_bbox[3]
    ):
        return "Martinique"
    elif (
        guadeloupe_bbox[0] <= gps_lat <= guadeloupe_bbox[2]
        and guadeloupe_bbox[1] <= gps_lon <= guadeloupe_bbox[3]
    ):
        return "Guadeloupe"

    elif pf_bbox[0] <= gps_lat <= pf_bbox[2] and pf_bbox[1] <= gps_lon <= pf_bbox[3]:
        return "Polynésie française"
    elif nc_bbox[0] <= gps_lat <= nc_bbox[2] and nc_bbox[1] <= gps_lon <= nc_bbox[3]:
        return "Nouvelle-Calédonie"
    else:
        return None  # "Hors de France"


def get_gps_info(filepath):
    with open(filepath, "rb") as f:
        tags = exifread.process_file(f)
        gps_info = {}

        # Recherche les informations GPS dans les informations EXIF

        # print("clés exif ", tags.keys())
        for tag in tags.keys():
            if tag.startswith("GPS"):
                gps_info[tag] = tags[tag]

        # Extraction des informations de latitude et de longitude
        gps_latitude = convert_rational_to_float(gps_info.get("GPS GPSLatitude"))
        gps_longitude = convert_rational_to_float(gps_info.get("GPS GPSLongitude"))

        if gps_latitude and gps_longitude:
            return gps_latitude, gps_longitude
        else:
            return None, None


def convert_rational_to_float(rational):
    return float(rational.values[0].num) / float(rational.values[0].den)


if __name__ == "__main__":
    parser = argparse.ArgumentParser()
    parser.add_argument(
        "--source_dir",
        default="/home/cipherbliss/Téléchargements/FIBRELAND/TEST_IN_FR/",
        help="Chemin du répertoire source",
    )
    parser.add_argument(
        "--destination_dir",
        default="/home/cipherbliss/Téléchargements/FIBRELAND/IN_FRANCE/",
        help="Chemin du répertoire destination",
    )
    parser.add_argument(
        "--sequence_folder",
        default="principale_sequence",
        help="Nom du dossier de séquence",
    )
    args = parser.parse_args()

    # Parcourt tous les fichiers dans le répertoire source et ses sous-répertoires
    for root, dirs, files in os.walk(args.source_dir):
        for filename in files:
            # Vérifie si le fichier est une image
            if filename.lower().endswith(
                (".png", ".jpg", ".jpeg", ".gif", ".bmp", ".tif")
            ):
                filepath = os.path.join(root, filename)
                move_file_if_in_france(filepath, sequence_folder)

    print(
        "fichiers se situant en france déplacés: ",
        count_files_moved,
        " / ",
        count_files_all,
    )