Apply current workspace changes

third_party/PyINT/pyint/test_srtm_manual.py

@@ -0,0 +1,119 @@
+#!/usr/bin/env python
+"""
+测试手动处理 SRTM .hgt 文件的脚本
+
+这个脚本演示如何在 Python < 3.12 环境下手动处理 SRTM 数据
+"""
+
+import os
+import sys
+
+print("=" * 80)
+print("SRTM 手动处理功能测试")
+print("=" * 80)
+
+# 检查Python版本
+print(f"\nPython版本: {sys.version_info.major}.{sys.version_info.minor}.{sys.version_info.micro}")
+
+# 导入处理函数
+from makedem import process_srtm_hgt_files, HAS_SRTM
+
+print(f"\nsrtm库状态: {'已安装' if HAS_SRTM else '未安装 (需要Python >= 3.12)'}")
+
+print("\n" + "=" * 80)
+print("使用说明")
+print("=" * 80)
+
+print("""
+步骤1: 下载SRTM .hgt文件
+---------------------------
+从以下网站下载所需区域的 .hgt 文件:
+
+1. CSI-CGIAR SRTM (推荐,无需注册)
+   https://srtm.csi.cgiar.org/
+   
+2. USGS EarthExplorer (需要注册)
+   https://earthexplorer.usgs.gov/
+
+文件命名规则:
+- N39E116.hgt → 北纬39-40°, 东经116-117°
+- N39W110.hgt → 北纬39-40°, 西经110-109°
+- S10E120.hgt → 南纬10-9°, 东经120-121°
+
+步骤2: 组织文件
+---------------------------
+将所有 .hgt 文件放在同一目录下:
+
+mkdir -p ~/data/SRTM
+mv N*.hgt S*.hgt ~/data/SRTM/
+
+步骤3: 运行命令
+---------------------------
+使用以下命令生成DEM:
+
+makedem.py -r 116/117/39/40 --dem-source srtm --srtm-data-dir ~/data/SRTM
+
+或者直接调用Python函数:
+
+from makedem import process_srtm_hgt_files
+
+dem_file = process_srtm_hgt_files(
+    west=116.0,
+    south=39.0,
+    east=117.0,
+    north=40.0,
+    srtm_data_dir='/path/to/srtm/data',
+    save_path='/path/to/output'
+)
+
+""")
+
+print("=" * 80)
+print("示例: 为北京地区(116-117°E, 39-40°N)下载SRTM数据")
+print("=" * 80)
+
+print("""
+需要下载的文件:
+1. 访问 https://srtm.csi.cgiar.org/
+2. 选择区域: 39-40°N, 116-117°E
+3. 下载文件: N39E116.hgt
+
+下载命令示例:
+-------------""")
+print(f"""
+# 创建数据目录
+mkdir -p ~/data/SRTM
+
+# 假设已下载 N39E116.hgt 到 ~/Downloads
+mv ~/Downloads/N39E116.hgt ~/data/SRTM/
+
+# 生成DEM
+makedem.py -r 116/117/39/40 --dem-source srtm --srtm-data-dir ~/data/SRTM
+
+# 输出文件将保存为: SRTM_116_117_39_40.tif
+""")
+
+print("=" * 80)
+print("注意事项")
+print("=" * 80)
+
+print("""
+1. SRTM文件必须覆盖整个目标区域
+   - 对于区域 116-117°E, 39-40°N
+   - 需要 N39E116.hgt 文件
+
+2. 文件格式
+   - 支持 .hgt 文件 (未压缩)
+   - 如果下载的是 .hgt.zip, 请先解压
+
+3. 空数据区域
+   - SRTM数据在海洋区域可能为空值
+   - 程序会自动处理空值区域
+
+4. Python版本兼容性
+   - Python >= 3.12: 可使用 srtm 库(自动查询)
+   - Python < 3.12: 使用手动处理方法(GDAL合并)
+   - 两种方法结果相同,只是处理方式不同
+""")
+
+print("=" * 80)