# -*- coding: utf-8 -*-

"""
@File    : 15_掩膜提取.py
@Author  : fungis@163.com
@notice  :
"""
import os
from osgeo import gdal

# 单波段图像
raster_path = r'E:\gisData\raster\LC08_L1TP_123037_20200209_20200211_01_T1\LC08_L1TP_123037_20200209_20200211_01_T1_B1.TIF'
output_file_path = r'G:\temp\test.tif'
boundary_shp = r'E:\gisData\polygon\xuchang_polygon.shp'

# 判断文件是否存在
if os.path.exists(raster_path) is False:
    raise Exception('[Errno 2] 该文件不存在: \'' + raster_path + '\'')
# clip
gdal.Warp(output_file_path,  # 输出栅格
          raster_path,  # 输入栅格
          format='GTiff',
          dstSRS='EPSG:4326',
          cutlineDSName=boundary_shp,  # shp边界
          cropToCutline=True,  # 按掩膜图层范围裁剪
          outputType=gdal.GDT_Float32)
print('----处理完成！----')