Foo

gap_validation/fusion_masked.py

@@ -0,0 +1,113 @@
+"""EFAST with symlinked S2 dir (withhold one acquisition); outputs under validation/."""
+
+from __future__ import annotations
+
+from pathlib import Path
+from tempfile import TemporaryDirectory
+
+from fusion import run_efast, run_efast_itb
+from preparation import _get_base_dir, _get_itb_base_dir
+
+from gap_validation.s2_mask_dir import build_masked_s2_dir_bti, build_masked_s2_dir_itb
+
+
+def prepared_s3_dir(season: int, site_name: str, strategy: str) -> Path:
+    return _get_base_dir(season, site_name, strategy) / "s3"
+
+
+def validation_fusion_dir(
+    site_name: str,
+    season: int,
+    gap_days: int,
+    strategy: str,
+    sigma: int | None,
+    mode: str,
+) -> Path:
+    """``data/.../validation/fusion/gap_{n}/{strategy}_sigma{20|30}_{bti|itb}/``."""
+    sig = 30 if sigma == 30 else 20
+    return (
+        Path(f"data/{site_name}/{season}/validation")
+        / "fusion"
+        / f"gap_{gap_days}"
+        / f"{strategy}_sigma{sig}_{mode}"
+    )
+
+
+def run_masked_fusion_one_date(
+    season: int,
+    site_position: tuple[float, float],
+    site_name: str,
+    strategy: str,
+    sigma: int | None,
+    mode: str,
+    prediction_date_iso: str,
+    withheld_yyyymmdd: str,
+    fusion_output_dir: Path,
+) -> Path:
+    """Build temp masked S2 dir, run EFAST for ``prediction_date_iso`` only; return output dir."""
+    fusion_output_dir.mkdir(parents=True, exist_ok=True)
+    date_range = f"{prediction_date_iso[:10]}/{prediction_date_iso[:10]}"
+    s3_dir = prepared_s3_dir(season, site_name, strategy)
+
+    with TemporaryDirectory(prefix="gapval_s2_") as tmp:
+        tmp_s2 = Path(tmp) / "s2"
+        if mode == "bti":
+            prep_s2 = _get_base_dir(season, site_name, strategy) / "s2"
+            build_masked_s2_dir_bti(prep_s2, withheld_yyyymmdd, tmp_s2)
+            run_efast(
+                season,
+                site_position,
+                site_name,
+                cleaning_strategy=strategy,
+                sigma=sigma,
+                date_range=date_range,
+                s2_output_dir=tmp_s2,
+                s3_output_dir=s3_dir,
+                fusion_output_dir=fusion_output_dir,
+            )
+        elif mode == "itb":
+            prep_s2 = _get_itb_base_dir(season, site_name, strategy) / "s2"
+            s3_itb = _get_itb_base_dir(season, site_name, strategy) / "s3"
+            build_masked_s2_dir_itb(prep_s2, withheld_yyyymmdd, tmp_s2)
+            run_efast_itb(
+                season,
+                site_position,
+                site_name,
+                cleaning_strategy=strategy,
+                sigma=sigma,
+                date_range=date_range,
+                s2_output_dir=tmp_s2,
+                s3_output_dir=s3_itb,
+                fusion_output_dir=fusion_output_dir,
+            )
+        else:
+            raise ValueError(f"mode must be bti or itb, got {mode!r}")
+
+    return fusion_output_dir
+
+
+def production_fusion_path(
+    season: int,
+    site_name: str,
+    strategy: str,
+    sigma: int | None,
+    mode: str,
+    yyyymmdd: str,
+) -> Path:
+    """Single-date fused raster from the normal prepared tree (no-gap baseline)."""
+    if mode == "bti":
+        base = _get_base_dir(season, site_name, strategy)
+        sub = f"fusion_sigma{sigma}" if sigma else "fusion"
+        return base / sub / f"REFL_{yyyymmdd}.tif"
+    base = _get_itb_base_dir(season, site_name, strategy)
+    sub = f"fusion_sigma{sigma}" if sigma else "fusion"
+    return base / sub / f"GCC_{yyyymmdd}.tif"
+
+
+def withheld_s2_refl_path(
+    season: int, site_name: str, strategy: str, withheld_filename: str | None
+) -> Path | None:
+    if not withheld_filename:
+        return None
+    p = _get_base_dir(season, site_name, strategy) / "s2" / withheld_filename
+    return p if p.is_file() else None