fix: 防御性编程重构 — 栅格空间对齐 + NoData 处理 + 高危代码加固

**核心修复 (preview_generator.py):**
- 废弃 _align_mask_to_image() (numpy crop/pad 在地理空间上错误)
- 新增 _warp_mask_to_image(): 使用 gdal.Warp 将掩膜重采样到与底图
  完全一致的像素网格,处理旋转/偏移/投影差异
- 新增 _normalize_mask(nodata_value): 正确的掩膜值域自劢归一间 (0/1 vs 0/255)
- 修复 alpha = mask_data/255.0 → mask_data (掩膜是二值 0/1, 不是 0/255)
- 面积计算 valid_pixels 使用 warp 前数据排除 nodata 背景

**新增防御工具模块:**
- spatial_validator.py: SpatialAlignmentError 自定义异常 +
  validate_two_rasters() / validate_spatial_alignment() 强制空间一致性检查 +
  has_rotation() / get_pixel_resolution() 诊断工具
- nodata_handler.py: read_band_safe() / read_bands_safe() 自动 NoData→nan +
  read_band_masked() 返回 MaskedArray + create_valid_mask()

**P0 高危代码集成 (3处):**
- sampling.py: 耀斑掩膜与水体掩膜 bool 运算前验证 numpy shape 一致
- find_severe_glint_area.py: 栅格 mask 读取后验证 dims+GT+projection 对齐
- waterindex_inversion/__init__.py: mask 与 BSQ 维度验证, 不一致时优雅降级

**旋转影像兼容性修复:**
- extract_water_area.py: pixel_size = sqrt(gt[1]^2+gt[2]^2) 使用勾股定理
  正确计算旋转影像的像素分辨率
This commit is contained in:
duxin
2026-07-07 09:10:40 +08:00
parent c46f78e69d
commit 826f110894
7 changed files with 947 additions and 26 deletions

View File

@ -85,9 +85,10 @@ def rasterize_shp(shp_filepath, raster_fn_out, img_path, NoData_value=None):
layer = source_ds.GetLayer(0)
layer_name = layer.GetName()
# about 25 metres(ish) use 0.001 if you want roughly 100m
pixel_size_x = abs(geotransform[1]) # 像素宽度(X方向)
pixel_size_y = abs(geotransform[5]) # 像素高度(Y方向,通常是负值,需要取绝对值)
# 计算像素分辨率(考虑旋转参数)
# 对于旋转影像(gt[2] 或 gt[4] != 0),像素实际分辨率需用勾股定理
pixel_size_x = np.sqrt(geotransform[1]**2 + geotransform[2]**2)
pixel_size_y = np.sqrt(geotransform[4]**2 + geotransform[5]**2)
raster_fn_out_tmp = append2filename(raster_fn_out, "_tmp_delete")
# 构建栅格化参数