219 lines
7.6 KiB
Python
219 lines
7.6 KiB
Python
#!/usr/bin/env python3
|
||
# -*- coding: utf-8 -*-
|
||
|
||
import argparse
|
||
import os
|
||
from pathlib import Path
|
||
from datetime import datetime, timezone, timedelta
|
||
from typing import Optional, Tuple, List
|
||
|
||
from PIL import Image, ExifTags
|
||
import xml.etree.ElementTree as ET
|
||
|
||
# Mapy tagów EXIF
|
||
TAGS = {v: k for k, v in ExifTags.TAGS.items()}
|
||
GPSTAGS = {v: k for k, v in ExifTags.GPSTAGS.items()}
|
||
|
||
# Akceptowane rozszerzenia zdjęć
|
||
IMAGE_EXTS = {".jpg", ".jpeg", ".JPG", ".JPEG"}
|
||
|
||
def dms_to_decimal(dms, ref) -> Optional[float]:
|
||
"""
|
||
Konwersja DMS (na ułamki) -> stopnie dziesiętne.
|
||
dms: np. ((52,1), (13,1), (1234,100)) => 52° 13' 12.34"
|
||
ref: 'N'/'S' lub 'E'/'W'
|
||
"""
|
||
try:
|
||
deg = dms[0][0] / dms[0][1]
|
||
minutes = dms[1][0] / dms[1][1]
|
||
seconds = dms[2][0] / dms[2][1]
|
||
dec = deg + minutes / 60.0 + seconds / 3600.0
|
||
if ref in ("S", "W"):
|
||
dec = -dec
|
||
return dec
|
||
except Exception:
|
||
return None
|
||
|
||
def parse_exif_datetime(raw_dt: Optional[str], subsec: Optional[str]) -> Optional[datetime]:
|
||
"""
|
||
Parsuje DateTimeOriginal w formacie 'YYYY:MM:DD HH:MM:SS' i składa z sub-sekundami.
|
||
Zwraca naive datetime (bez strefy) – później ewentualnie przeliczane wg --tz.
|
||
"""
|
||
if not raw_dt:
|
||
return None
|
||
try:
|
||
base = datetime.strptime(raw_dt, "%Y:%m:%d %H:%M:%S")
|
||
if subsec and subsec.isdigit():
|
||
# Subsec może mieć różną długość – skracamy/padding do mikrosekund
|
||
micro = int((subsec + "000000")[:6])
|
||
base = base.replace(microsecond=micro)
|
||
return base
|
||
except Exception:
|
||
return None
|
||
|
||
def apply_tz_and_to_utc(dt: datetime, tz_offset: Optional[str]) -> Optional[datetime]:
|
||
"""
|
||
Jeśli podano --tz w formacie +HH:MM lub -HH:MM, interpretuj naive dt w tej strefie,
|
||
a następnie zwróć dt w UTC.
|
||
Jeśli --tz nie podano, zwróć None (żeby pominąć <time> albo zapisać bez gwarancji poprawnej TZ).
|
||
"""
|
||
if tz_offset is None:
|
||
return None
|
||
try:
|
||
sign = 1 if tz_offset.startswith("+") else -1
|
||
hh, mm = tz_offset[1:].split(":")
|
||
offset = timedelta(hours=int(hh), minutes=int(mm))
|
||
tzinfo = timezone(sign * offset)
|
||
aware = dt.replace(tzinfo=tzinfo)
|
||
return aware.astimezone(timezone.utc)
|
||
except Exception:
|
||
return None
|
||
|
||
def extract_exif_gps_and_time(img_path: Path) -> Tuple[Optional[float], Optional[float], Optional[float], Optional[datetime], Optional[datetime]]:
|
||
"""
|
||
Zwraca: (lat, lon, ele, naive_time, utc_time_if_available)
|
||
"""
|
||
try:
|
||
with Image.open(img_path) as im:
|
||
exif = im._getexif() or {}
|
||
except Exception:
|
||
return None, None, None, None, None
|
||
|
||
# Zamiana kluczy numerycznych na nazwy
|
||
exif_readable = {}
|
||
for tag, val in exif.items():
|
||
name = ExifTags.TAGS.get(tag, tag)
|
||
exif_readable[name] = val
|
||
|
||
gps_info = exif_readable.get("GPSInfo")
|
||
lat = lon = ele = None
|
||
if isinstance(gps_info, dict):
|
||
gps = {}
|
||
for k, v in gps_info.items():
|
||
gps_name = ExifTags.GPSTAGS.get(k, k)
|
||
gps[gps_name] = v
|
||
|
||
lat = dms_to_decimal(gps.get("GPSLatitude"), gps.get("GPSLatitudeRef", "N"))
|
||
lon = dms_to_decimal(gps.get("GPSLongitude"), gps.get("GPSLongitudeRef", "E"))
|
||
|
||
# Wysokość (opcjonalna)
|
||
if gps.get("GPSAltitude") and isinstance(gps["GPSAltitude"], tuple):
|
||
try:
|
||
ele = gps["GPSAltitude"][0] / gps["GPSAltitude"][1]
|
||
if gps.get("GPSAltitudeRef") == 1:
|
||
ele = -ele
|
||
except Exception:
|
||
ele = None
|
||
|
||
# Czas
|
||
raw_dt = exif_readable.get("DateTimeOriginal") or exif_readable.get("DateTime") or exif_readable.get("DateTimeDigitized")
|
||
# Subsekundy bywają w różnych tagach:
|
||
subsec = exif_readable.get("SubsecTimeOriginal") or exif_readable.get("SubsecTime") or exif_readable.get("SubSecTimeOriginal")
|
||
|
||
naive_dt = parse_exif_datetime(raw_dt, subsec)
|
||
return lat, lon, ele, naive_dt, None # utc_time wypełnimy później wg --tz
|
||
|
||
def collect_images(input_path: Path) -> List[Path]:
|
||
files = []
|
||
if input_path.is_dir():
|
||
for root, _, fns in os.walk(input_path):
|
||
for fn in fns:
|
||
p = Path(root) / fn
|
||
if p.suffix in IMAGE_EXTS:
|
||
files.append(p)
|
||
else:
|
||
if input_path.suffix in IMAGE_EXTS:
|
||
files.append(input_path)
|
||
return files
|
||
|
||
def generate_gpx(points: List[dict], creator: str = "exif2gpx") -> ET.ElementTree:
|
||
ET.register_namespace("", "http://www.topografix.com/GPX/1/1")
|
||
ET.register_namespace("xsi", "http://www.w3.org/2001/XMLSchema-instance")
|
||
gpx = ET.Element(
|
||
"gpx",
|
||
{
|
||
"version": "1.1",
|
||
"creator": creator,
|
||
"xmlns": "http://www.topografix.com/GPX/1/1",
|
||
"xmlns:xsi": "http://www.w3.org/2001/XMLSchema-instance",
|
||
"xsi:schemaLocation": "http://www.topografix.com/GPX/1/1 http://www.topografix.com/GPX/1/1/gpx.xsd",
|
||
},
|
||
)
|
||
trk = ET.SubElement(gpx, "trk")
|
||
name = ET.SubElement(trk, "name")
|
||
name.text = "Track from EXIF"
|
||
seg = ET.SubElement(trk, "trkseg")
|
||
|
||
for p in points:
|
||
trkpt = ET.SubElement(seg, "trkpt", {"lat": f"{p['lat']:.7f}", "lon": f"{p['lon']:.7f}"})
|
||
if p.get("ele") is not None:
|
||
ele = ET.SubElement(trkpt, "ele")
|
||
ele.text = f"{p['ele']:.2f}"
|
||
if p.get("time"):
|
||
# GPX wymaga ISO 8601 w UTC z 'Z'
|
||
t = ET.SubElement(trkpt, "time")
|
||
t.text = p["time"].strftime("%Y-%m-%dT%H:%M:%S.%fZ").rstrip("0").rstrip(".") + "Z"
|
||
# Dodatkowo można dodać nazwę pliku jako <desc>
|
||
desc = ET.SubElement(trkpt, "desc")
|
||
desc.text = p.get("source", "")
|
||
|
||
return ET.ElementTree(gpx)
|
||
|
||
def main():
|
||
ap = argparse.ArgumentParser(description="Z EXIF zdjęć (JPG) generuje GPX z trackiem.")
|
||
ap.add_argument("input", help="Plik JPG lub folder ze zdjęciami")
|
||
ap.add_argument("output", help="Ścieżka wyjściowa pliku GPX")
|
||
ap.add_argument("--tz", help="Strefa czasowa EXIF w formacie +HH:MM lub -HH:MM (np. +02:00)", default=None)
|
||
args = ap.parse_args()
|
||
|
||
src = Path(args.input)
|
||
out = Path(args.output)
|
||
|
||
files = collect_images(src)
|
||
if not files:
|
||
print("Brak zdjęć JPG/JPEG w podanej lokalizacji.")
|
||
return
|
||
|
||
points = []
|
||
for fp in files:
|
||
lat, lon, ele, naive_dt, _ = extract_exif_gps_and_time(fp)
|
||
if lat is None or lon is None:
|
||
continue
|
||
|
||
# Wyznacz czas w UTC jeśli podano --tz
|
||
time_utc = None
|
||
if naive_dt is not None and args.tz:
|
||
time_utc = apply_tz_and_to_utc(naive_dt, args.tz)
|
||
|
||
points.append({
|
||
"lat": lat,
|
||
"lon": lon,
|
||
"ele": ele,
|
||
"naive_time": naive_dt,
|
||
"time": time_utc, # tylko jeśli mamy --tz
|
||
"source": fp.name,
|
||
})
|
||
|
||
if not points:
|
||
print("Nie znaleziono zdjęć z danymi GPS.")
|
||
return
|
||
|
||
# Sortowanie: najpierw po czasie jeśli dostępny; potem po nazwie pliku
|
||
def sort_key(p):
|
||
# preferuj p['time'] (UTC) gdy jest; w przeciwnym razie naive_time; w ostateczności nazwa
|
||
return (
|
||
p["time"] if p["time"] is not None else
|
||
p["naive_time"] if p["naive_time"] is not None else
|
||
datetime.max
|
||
, p["source"])
|
||
|
||
points.sort(key=sort_key)
|
||
|
||
tree = generate_gpx(points)
|
||
out.parent.mkdir(parents=True, exist_ok=True)
|
||
tree.write(out, encoding="utf-8", xml_declaration=True)
|
||
print(f"Zapisano GPX: {out}")
|
||
|
||
if __name__ == "__main__":
|
||
main()
|