chore: initialize insar management system v2

2026-04-14 13:16:01 +08:00
commit ecc72ec9cd
361 changed files with 2142522 additions and 0 deletions
@@ -0,0 +1,180 @@
+#!/usr/bin/env python3
+from __future__ import annotations
+
+import argparse
+from pathlib import Path
+
+import numpy as np
+from osgeo import gdal
+
+gdal.UseExceptions()
+
+DEFAULT_WAVELENGTH = 0.23793052222222222
+DEFAULT_NODATA = -9999.0
+
+
+def parse_args() -> argparse.Namespace:
+    parser = argparse.ArgumentParser(
+        description="Export ISCE2 geocoded displacement/coherence products to GeoTIFF."
+    )
+    parser.add_argument(
+        "work_dir",
+        type=Path,
+        help="ISCE2 work directory containing interferogram/*.geo outputs",
+    )
+    parser.add_argument(
+        "--output-dir",
+        type=Path,
+        default=None,
+        help="Output directory for GeoTIFF files. Default: work_dir",
+    )
+    parser.add_argument(
+        "--prefix",
+        type=str,
+        default=None,
+        help="Output filename prefix. Default: work_dir basename",
+    )
+    parser.add_argument(
+        "--wavelength",
+        type=float,
+        default=DEFAULT_WAVELENGTH,
+        help="Radar wavelength in meters",
+    )
+    parser.add_argument(
+        "--coh-threshold",
+        type=float,
+        default=0.05,
+        help="Mask pixels with coherence below this threshold in *_disp.tif",
+    )
+    parser.add_argument(
+        "--include-disp-full",
+        action="store_true",
+        help="Also export the unmasked displacement GeoTIFF for debugging",
+    )
+    return parser.parse_args()
+
+
+def write_geotiff(array: np.ndarray, ref_ds: gdal.Dataset, out_path: Path, nodata: float) -> None:
+    out_path.parent.mkdir(parents=True, exist_ok=True)
+    if out_path.exists():
+        out_path.unlink()
+
+    driver = gdal.GetDriverByName("GTiff")
+    ds = driver.Create(
+        str(out_path),
+        ref_ds.RasterXSize,
+        ref_ds.RasterYSize,
+        1,
+        gdal.GDT_Float32,
+        options=["COMPRESS=LZW", "TILED=YES"],
+    )
+    ds.SetGeoTransform(ref_ds.GetGeoTransform())
+    ds.SetProjection(ref_ds.GetProjection())
+    band = ds.GetRasterBand(1)
+    band.SetNoDataValue(nodata)
+    band.WriteArray(array.astype(np.float32))
+    ds.FlushCache()
+    ds = None
+
+
+def export_products(
+    work_dir: Path,
+    output_dir: Path,
+    prefix: str,
+    wavelength: float,
+    coh_threshold: float,
+    include_disp_full: bool = False,
+    nodata: float = DEFAULT_NODATA,
+) -> dict[str, Path]:
+    unw_path = work_dir / "interferogram" / "filt_topophase.unw.geo.vrt"
+    cor_path = work_dir / "interferogram" / "topophase.cor.geo.vrt"
+
+    if not unw_path.exists():
+        raise FileNotFoundError(f"Missing unwrapped product: {unw_path}")
+    if not cor_path.exists():
+        raise FileNotFoundError(f"Missing coherence product: {cor_path}")
+
+    unw_ds = gdal.Open(str(unw_path))
+    cor_ds = gdal.Open(str(cor_path))
+    if unw_ds is None or cor_ds is None:
+        raise RuntimeError("Failed to open ISCE2 geo products with GDAL.")
+
+    amp = unw_ds.GetRasterBand(1).ReadAsArray().astype(np.float32)
+    phase = unw_ds.GetRasterBand(2).ReadAsArray().astype(np.float32)
+
+    coh_band = 2 if cor_ds.RasterCount >= 2 else 1
+    coh = cor_ds.GetRasterBand(coh_band).ReadAsArray().astype(np.float32)
+    coh_valid = coh > 0
+
+    disp_m = phase * wavelength / (4.0 * np.pi)
+    disp_m_full = disp_m.copy()
+
+    mask = (coh < coh_threshold) | (amp == 0)
+    disp_m[mask] = nodata
+    disp_m_full[amp == 0] = nodata
+
+    coh_out = coh.copy()
+    coh_out[~coh_valid] = nodata
+
+    output_dir.mkdir(parents=True, exist_ok=True)
+    out_disp = output_dir / f"{prefix}_disp.tif"
+    out_coh = output_dir / f"{prefix}_coh.tif"
+
+    write_geotiff(disp_m, unw_ds, out_disp, nodata)
+    write_geotiff(coh_out, cor_ds, out_coh, nodata)
+    out_disp_full = None
+    if include_disp_full:
+        out_disp_full = output_dir / f"{prefix}_disp_full.tif"
+        write_geotiff(disp_m_full, unw_ds, out_disp_full, nodata)
+
+    valid_disp = disp_m[disp_m != nodata]
+    valid_coh = coh_out[coh_out != nodata]
+    valid_full = disp_m_full[disp_m_full != nodata] if include_disp_full else np.array([], dtype=np.float32)
+
+    print(f"Work dir:                {work_dir}")
+    print(f"Output prefix:           {prefix}")
+    print(f"Coherence threshold:     {coh_threshold}")
+    print(f"Unwrap support ratio:    {(amp != 0).mean()*100:.2f}%")
+    print(f"Coherence support ratio: {coh_valid.mean()*100:.2f}%")
+    print(f"Masked disp ratio:       {(disp_m != nodata).mean()*100:.2f}%")
+    if valid_disp.size:
+        print(f"Masked disp range:       [{valid_disp.min():.4f}, {valid_disp.max():.4f}] m")
+    if include_disp_full and valid_full.size:
+        print(f"Full disp range:         [{valid_full.min():.4f}, {valid_full.max():.4f}] m")
+    if valid_coh.size:
+        print(f"Coherence range:         [{valid_coh.min():.4f}, {valid_coh.max():.4f}]")
+    print(f"Wrote:                   {out_disp}")
+    if out_disp_full is not None:
+        print(f"Wrote:                   {out_disp_full}")
+    print(f"Wrote:                   {out_coh}")
+
+    unw_ds = None
+    cor_ds = None
+    outputs: dict[str, Path] = {
+        "disp": out_disp,
+        "coh": out_coh,
+    }
+    if out_disp_full is not None:
+        outputs["disp_full"] = out_disp_full
+    return outputs
+
+
+def main() -> int:
+    args = parse_args()
+    work_dir = args.work_dir.resolve()
+    output_dir = args.output_dir.resolve() if args.output_dir else work_dir
+    prefix = args.prefix or work_dir.name
+
+    export_products(
+        work_dir=work_dir,
+        output_dir=output_dir,
+        prefix=prefix,
+        wavelength=args.wavelength,
+        coh_threshold=args.coh_threshold,
+        include_disp_full=args.include_disp_full,
+    )
+    return 0
+
+
+if __name__ == "__main__":
+    raise SystemExit(main())