prepare: add index module, update pipeline, tests, Dockerfile, and run.sh

lidar_pipeline/tests/test_index.py

@@ -0,0 +1,205 @@
+"""Tests pour la carte globale interactive (index.py)."""
+
+import json
+from pathlib import Path
+
+
+def test_parse_basename_coords_valid():
+    """Parse les coordonnées d'un basename LHD valide."""
+    from lidar_pipeline.index import parse_basename_coords
+    assert parse_basename_coords("LHD_FXX_1000_6881_PTS_LAMB93_IGN69") == (1000, 6881)
+    assert parse_basename_coords("LHD_FXX_1049_6895_PTS_LAMB93_IGN69") == (1049, 6895)
+
+
+def test_parse_basename_coords_with_res_suffix():
+    """Les noms de dossier avec suffixe résolution sont aussi parsables."""
+    from lidar_pipeline.index import parse_basename_coords
+    assert parse_basename_coords("LHD_FXX_1000_6881_PTS_LAMB93_IGN69_r0p2") == (1000, 6881)
+
+
+def test_parse_basename_coords_invalid():
+    """Les noms non-LHD retournent None."""
+    from lidar_pipeline.index import parse_basename_coords
+    assert parse_basename_coords("random_dir") is None
+    assert parse_basename_coords("DTM") is None
+    assert parse_basename_coords("") is None
+
+
+def test_strip_res_suffix_primary():
+    """Dossier sans suffixe = résolution primaire (0.5)."""
+    from lidar_pipeline.index import _strip_res_suffix
+    base, res = _strip_res_suffix("LHD_FXX_1000_6881_PTS_LAMB93_IGN69")
+    assert base == "LHD_FXX_1000_6881_PTS_LAMB93_IGN69"
+    assert res == 0.5
+
+
+def test_strip_res_suffix_multi():
+    """Dossier avec suffixe _r0p2 = résolution 0.2."""
+    from lidar_pipeline.index import _strip_res_suffix
+    base, res = _strip_res_suffix("LHD_FXX_1000_6881_PTS_LAMB93_IGN69_r0p2")
+    assert base == "LHD_FXX_1000_6881_PTS_LAMB93_IGN69"
+    assert res == 0.2
+
+
+def test_compute_bbox():
+    """Calcule la bounding box d'un ensemble de tuiles."""
+    from lidar_pipeline.index import compute_bbox
+    tiles = [
+        {'col': 1000, 'row': 6881},
+        {'col': 1001, 'row': 6882},
+        {'col': 1002, 'row': 6880},
+    ]
+    bbox = compute_bbox(tiles)
+    assert bbox == {'min_col': 1000, 'max_col': 1002, 'min_row': 6880, 'max_row': 6882}
+
+
+def test_compute_bbox_empty():
+    """Aucune tuile → None."""
+    from lidar_pipeline.index import compute_bbox
+    assert compute_bbox([]) is None
+
+
+def _make_fake_viz_dir(vis_dir, basename, col, row, viz_keys=('hillshade_multi', 'svf'), ext='webp', res_suffix=''):
+    """Crée un faux dossier de visualisations avec de petites images."""
+    from PIL import Image as PILImage
+    import numpy as np
+
+    dir_name = f"LHD_FXX_{col}_{row}_PTS_LAMB93_IGN69{res_suffix}"
+    tile_dir = Path(vis_dir) / dir_name
+    tile_dir.mkdir(parents=True, exist_ok=True)
+    for v in viz_keys:
+        arr = np.random.randint(0, 255, (50, 50, 3), dtype=np.uint8)
+        img = PILImage.fromarray(arr)
+        fname = f"LHD_FXX_{col}_{row}_PTS_LAMB93_IGN69{res_suffix}_{v}.{ext}"
+        img.save(str(tile_dir / fname), format='WEBP', quality=80)
+    return tile_dir
+
+
+def test_scan_tiles(tmp_path):
+    """scan_tiles détecte les dossiers de tuiles et leurs visualisations."""
+    from lidar_pipeline.index import scan_tiles
+
+    vis_dir = tmp_path / "visualisations"
+    vis_dir.mkdir()
+    _make_fake_viz_dir(vis_dir, "a", 1000, 6881, ('hillshade_multi', 'svf'))
+    _make_fake_viz_dir(vis_dir, "b", 1001, 6881, ('hillshade_multi',))
+
+    tiles = scan_tiles(vis_dir)
+    assert len(tiles) == 2
+    names = sorted(t['dir_name'] for t in tiles)
+    assert "LHD_FXX_1000_6881_PTS_LAMB93_IGN69" in names
+    assert "LHD_FXX_1001_6881_PTS_LAMB93_IGN69" in names
+    # Vérifie que les viz sont détectées
+    t0 = next(t for t in tiles if t['col'] == 1000)
+    assert 'hillshade_multi' in t0['viz']
+    assert 'svf' in t0['viz']
+
+
+def test_scan_tiles_ignores_non_lhd(tmp_path):
+    """Les dossiers non-LHD (ex: temp, DTM) sont ignorés."""
+    from lidar_pipeline.index import scan_tiles
+
+    vis_dir = tmp_path / "visualisations"
+    vis_dir.mkdir()
+    (vis_dir / "random_folder").mkdir()
+    _make_fake_viz_dir(vis_dir, "a", 1000, 6881)
+
+    tiles = scan_tiles(vis_dir)
+    assert len(tiles) == 1
+    assert tiles[0]['col'] == 1000
+
+
+def test_scan_tiles_multi_resolution(tmp_path):
+    """Les dossiers avec suffixe résolution sont correctement décodés."""
+    from lidar_pipeline.index import scan_tiles
+
+    vis_dir = tmp_path / "visualisations"
+    vis_dir.mkdir()
+    _make_fake_viz_dir(vis_dir, "a", 1000, 6881, res_suffix='')
+    _make_fake_viz_dir(vis_dir, "a", 1000, 6881, res_suffix='_r0p2')
+
+    tiles = scan_tiles(vis_dir)
+    assert len(tiles) == 2
+    resolutions = sorted(t['resolution'] for t in tiles)
+    assert resolutions == [0.2, 0.5]
+
+
+def test_build_index_generates_html(tmp_path):
+    """build_index génère index.html et les vignettes."""
+    from lidar_pipeline.index import build_index
+
+    output_dir = tmp_path / "output"
+    vis_dir = output_dir / "visualisations"
+    vis_dir.mkdir(parents=True)
+    _make_fake_viz_dir(vis_dir, "a", 1000, 6881, ('hillshade_multi', 'svf'))
+    _make_fake_viz_dir(vis_dir, "b", 1001, 6881, ('hillshade_multi',))
+
+    result = build_index(output_dir)
+    assert result is not None
+    html_path = Path(result)
+    assert html_path.exists()
+    assert html_path.name == "index.html"
+
+    content = html_path.read_text(encoding='utf-8')
+    # Vérifie la présence des éléments clés
+    assert "Carte globale LiDAR" in content
+    assert "LHD_FXX_1000_6881" in content
+    assert "LHD_FXX_1001_6881" in content
+    # Vérifie que le JSON intégré est valide
+    assert "const DATA" in content
+    # Vérifie les vignettes générées
+    thumb_dir = output_dir / "index_thumbs"
+    assert thumb_dir.is_dir()
+    thumbs = list(thumb_dir.glob("*.jpg"))
+    assert len(thumbs) >= 2  # au moins hillshade pour chaque tuile
+
+
+def test_build_index_empty_returns_none(tmp_path):
+    """Aucune tuile → build_index retourne None sans crash."""
+    from lidar_pipeline.index import build_index
+
+    output_dir = tmp_path / "output"
+    (output_dir / "visualisations").mkdir(parents=True)
+    result = build_index(output_dir)
+    assert result is None
+
+
+def test_build_index_embeds_valid_json(tmp_path):
+    """Le JSON embarqué dans le HTML est valide et contient les tuiles."""
+    from lidar_pipeline.index import build_index
+
+    output_dir = tmp_path / "output"
+    vis_dir = output_dir / "visualisations"
+    vis_dir.mkdir(parents=True)
+    _make_fake_viz_dir(vis_dir, "a", 1000, 6881, ('hillshade_multi',))
+
+    build_index(output_dir)
+    content = (output_dir / "index.html").read_text(encoding='utf-8')
+    # Extrait le JSON entre "const DATA = " et ";"
+    start = content.index("const DATA = ") + len("const DATA = ")
+    # Trouve le ; de fin de déclaration
+    depth = 0
+    end = start
+    for i, ch in enumerate(content[start:], start):
+        if ch == '{':
+            depth += 1
+        elif ch == '}':
+            depth -= 1
+            if depth == 0:
+                end = i + 1
+                break
+    data = json.loads(content[start:end])
+    assert 'tiles' in data
+    assert 'bbox' in data
+    assert 'vizList' in data
+    assert len(data['tiles']) == 1
+    assert data['tiles'][0]['col'] == 1000
+    assert data['tiles'][0]['row'] == 6881
+
+
+def test_pick_display_viz_prefers_hillshade():
+    """Le choix de viz par défaut privilégie hillshade_multi."""
+    from lidar_pipeline.index import _pick_display_viz
+    assert _pick_display_viz(['svf', 'hillshade_multi', 'slope']) == 'hillshade_multi'
+    assert _pick_display_viz(['svf', 'slope']) == 'svf'
+    assert _pick_display_viz(['topo']) == 'topo'