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

"""
@File    : 10_文件转换之Map转TIF.py
@Author  : fungis@163.com
@Time    : 2023/05/31 21:30
@notice  :
"""
from osgeo import gdal


def convertToPCRaster(src_filename, dst_filename, ot, vs):
    """
    将tif文件转成PCRaster文件写入硬盘
    :param src_filename: tif文件
    :param dst_filename: PCRaster文件
    :param ot: 值类型  float or int
    :return:
    """
    # 打开源数据集
    src_ds = gdal.Open(src_filename)
    # 定义输出文件名和格式dst_filename
    dst_ds = gdal.Translate(dst_filename, src_ds, format='PCRaster', outputType=ot)

    dst_ds = None
    src_ds = None


def convertToTif(src_filename, dst_filename, ot, vs):
    """
       将PCRaster文件转成tif文件写入硬盘
       :param src_filename: PCRaster文件
       :param dst_filename: tif文件
       :param ot: 值类型  float or int
       :return:
       """
    src_ds = gdal.Open(src_filename)
    dst_ds = gdal.Translate(dst_filename, src_ds, format='GTiff', outputType=ot)

    dst_ds = None
    src_ds = None


if __name__ == '__main__':
    tif_path = r'/results/tif/coast_distance.tif'
    convertToPCRaster(tif_path, "./results/gwlevel.map", gdal.GDT_Float32, "VS_SCALAR")
    convertToTif(r'demo-raster/data-use/lddsound_05min.map', './results/lddsound_05min.tif', gdal.GDT_Float32,
                 "VS_SCALAR")