Arcpy 多线程批量重采样脚本

Arcpy 多线程批量重采样脚本

import arcpy
import os
import multiprocessingdef resample_tifs(input_folder, output_folder, cell_size=0.05, resampling_type="BILINEAR"):"""将指定文件夹下的所有 TIFF 文件重采样到指定分辨率，并输出到新文件夹中。参数：input_folder (str): 输入 TIFF 文件所在的文件夹路径。output_folder (str): 重采样后的 TIFF 文件输出到的文件夹路径。cell_size (float): 重采样后的分辨率大小。默认为 symbol。resampling_type (str): 重采样方法。可选值包括 "NEAREST_NEIGHBOR", "BILINEAR", "CUBIC", "MAJORITY"。默认为 "BILINEAR"。"""# 检查输入文件夹是否存在if not os.path.exists(input_folder):raise FileNotFoundError(f"输入文件夹不存在: {input_folder}")# 如果输出文件夹不存在，则创建if not os.path.exists(output_folder):os.makedirs(output_folder)print(f"已创建输出文件夹: {output_folder}")# 设置环境工作空间arcpy.env.workspace = input_folder# 获取所有 TIFF 文件tif_list = arcpy.ListRasters("*", "TIF")if not tif_list:print("没有找到 TIFF 文件。")returnprint(f"找到 {len(tif_list)} 个 TIFF 文件。开始重采样...")# 遍历每个 TIFF 文件并进行重采样for tif in tif_list:input_path = os.path.join(input_folder, tif)output_path = os.path.join(output_folder, tif)try:print(f"正在处理: {tif}")# 使用 Resample_management 工具进行重采样arcpy.management.Resample(in_raster=input_path,out_raster=output_path,cell_size=cell_size,resampling_type=resampling_type)print(f"成功重采样并保存到: {output_path}")except arcpy.ExecuteError:print(f"ArcPy 错误处理文件 {tif}: {arcpy.GetMessages(2)}")except Exception as e:print(f"错误处理文件 {tif}: {e}")print("所有文件重采样完成。")def main():output_dir2 = r"./010_世界人口数据 _0.1/"    #保存new重采样文件symbol = 0.25 # 重采样大小tasks = [# # ("./01_land_cover", "./China_01_landcover"),# ("./Global_tif_0.1_LC", output_dir2, symbol, "MAJORITY"),("010_世界人口数据", output_dir2, symbol, "NEAREST"),] # 创建进程池，进程数设为CPU核心数pool = multiprocessing.Pool(processes=multiprocessing.cpu_count())# 准备并行执行的任务results = []print(tasks)for input_dir11, output_dir22,cell_size ,resampling_type in tasks:result = pool.apply_async(resample_tifs, args=(input_dir11, output_dir22,cell_size ,resampling_type))results.append(result)# 关闭进程池，等待所有任务完成pool.close()pool.join()print("所有任务已完成。")# 示例用法
if __name__ == "__main__":main()