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:
@ -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")
|
||||
|
||||
# 构建栅格化参数
|
||||
|
||||
Reference in New Issue
Block a user