117 lines
4.6 KiB
Python
117 lines
4.6 KiB
Python
#!/usr/bin/env python3
|
|
"""Analyze first-pass water and shoreline applicability for the Ground Zero tile."""
|
|
|
|
from __future__ import annotations
|
|
|
|
import json
|
|
from datetime import datetime, timezone
|
|
from pathlib import Path
|
|
|
|
import numpy as np
|
|
from osgeo import gdal
|
|
|
|
gdal.UseExceptions()
|
|
|
|
|
|
PROJECT_ROOT = Path(__file__).resolve().parents[1]
|
|
TILE_ID = "gz_us_ca_pacifica_utm10n_e544_n4160"
|
|
DEM_PATH = PROJECT_ROOT / "Data" / "Terrain" / "Extracted" / TILE_ID / f"{TILE_ID}_1m_dem_subset.tif"
|
|
OUTPUT_DIR = PROJECT_ROOT / "Data" / "Terrain" / "Analysis" / TILE_ID
|
|
OUTPUT_PATH = OUTPUT_DIR / f"{TILE_ID}_water_shoreline_analysis.json"
|
|
|
|
SEA_LEVEL_M = 0.0
|
|
SHORELINE_BUFFER_M = 2.0
|
|
LOW_COASTAL_BUFFER_M = 5.0
|
|
|
|
|
|
def summarize_edge(values: np.ndarray) -> dict[str, float | int]:
|
|
finite = values[np.isfinite(values)]
|
|
return {
|
|
"sample_count": int(finite.size),
|
|
"min_elevation_m": float(np.min(finite)) if finite.size else None,
|
|
"max_elevation_m": float(np.max(finite)) if finite.size else None,
|
|
"mean_elevation_m": float(np.mean(finite)) if finite.size else None,
|
|
"sea_level_or_below_samples": int(np.count_nonzero(finite <= SEA_LEVEL_M)),
|
|
"near_sea_level_samples": int(np.count_nonzero(finite <= SHORELINE_BUFFER_M)),
|
|
"low_coastal_samples": int(np.count_nonzero(finite <= LOW_COASTAL_BUFFER_M)),
|
|
}
|
|
|
|
|
|
def main() -> None:
|
|
dataset = gdal.Open(str(DEM_PATH), gdal.GA_ReadOnly)
|
|
if dataset is None:
|
|
raise RuntimeError(f"Could not open DEM: {DEM_PATH}")
|
|
|
|
band = dataset.GetRasterBand(1)
|
|
data = band.ReadAsArray().astype(float)
|
|
nodata = band.GetNoDataValue()
|
|
if nodata is not None:
|
|
data[data == nodata] = np.nan
|
|
|
|
finite = data[np.isfinite(data)]
|
|
if finite.size == 0:
|
|
raise RuntimeError(f"DEM has no finite elevation samples: {DEM_PATH}")
|
|
|
|
total_samples = int(finite.size)
|
|
sea_or_below = int(np.count_nonzero(finite <= SEA_LEVEL_M))
|
|
near_sea = int(np.count_nonzero(finite <= SHORELINE_BUFFER_M))
|
|
low_coastal = int(np.count_nonzero(finite <= LOW_COASTAL_BUFFER_M))
|
|
|
|
edge_width = 10
|
|
edges = {
|
|
"north": summarize_edge(data[:edge_width, :].reshape(-1)),
|
|
"south": summarize_edge(data[-edge_width:, :].reshape(-1)),
|
|
"west": summarize_edge(data[:, :edge_width].reshape(-1)),
|
|
"east": summarize_edge(data[:, -edge_width:].reshape(-1)),
|
|
}
|
|
|
|
has_open_water = sea_or_below > 0
|
|
has_shoreline = near_sea > 0
|
|
|
|
result = {
|
|
"schema_version": 1,
|
|
"tile_id": TILE_ID,
|
|
"generated_at_utc": datetime.now(timezone.utc).replace(microsecond=0).isoformat().replace("+00:00", "Z"),
|
|
"source_dem": str(DEM_PATH.relative_to(PROJECT_ROOT)),
|
|
"thresholds": {
|
|
"sea_level_m": SEA_LEVEL_M,
|
|
"shoreline_buffer_m": SHORELINE_BUFFER_M,
|
|
"low_coastal_buffer_m": LOW_COASTAL_BUFFER_M,
|
|
},
|
|
"elevation_summary": {
|
|
"sample_count": total_samples,
|
|
"min_elevation_m": float(np.min(finite)),
|
|
"max_elevation_m": float(np.max(finite)),
|
|
"mean_elevation_m": float(np.mean(finite)),
|
|
"sea_level_or_below_samples": sea_or_below,
|
|
"near_sea_level_samples": near_sea,
|
|
"low_coastal_samples": low_coastal,
|
|
"sea_level_or_below_percent": sea_or_below / total_samples * 100.0,
|
|
"near_sea_level_percent": near_sea / total_samples * 100.0,
|
|
"low_coastal_percent": low_coastal / total_samples * 100.0,
|
|
},
|
|
"edge_summary": edges,
|
|
"classification": {
|
|
"contains_open_water": has_open_water,
|
|
"contains_shoreline": has_shoreline,
|
|
"water_depth_required_for_mvp_tile": False,
|
|
"shoreline_mesh_required_for_mvp_tile": False,
|
|
"freshwater_source_required_for_gameplay": True,
|
|
"recommended_first_pass": "No ocean/shoreline actor required inside the current Ground Zero tile. Add a small freshwater source for gameplay separately, and defer ocean/bathymetry work to west/southwest neighbor coastal tiles.",
|
|
},
|
|
"notes": [
|
|
"This analysis uses the extracted 1-meter USGS DEM for the selected 1 km tile.",
|
|
"The selected Ground Zero tile is coastal-influenced but the DEM minimum is above sea level.",
|
|
"Ocean depth/bathymetry remains required for adjacent coastal/ocean tiles before multi-tile expansion.",
|
|
],
|
|
}
|
|
|
|
OUTPUT_DIR.mkdir(parents=True, exist_ok=True)
|
|
OUTPUT_PATH.write_text(json.dumps(result, indent=2) + "\n", encoding="utf-8")
|
|
print(f"Wrote {OUTPUT_PATH.relative_to(PROJECT_ROOT)}")
|
|
print(json.dumps(result["classification"], indent=2))
|
|
|
|
|
|
if __name__ == "__main__":
|
|
main()
|