import os

from PIL import Image


def main(input_path, output_path):
    num = len(input_path) + 1

    for root, dirs, files in os.walk(input_path):
        for file in files:
            if file.endswith('tif'):
                im = Image.open(os.path.join(root, file))
                try:
                    im = im.transpose(Image.Transpose.FLIP_TOP_BOTTOM)  # 旋转图片
                    if not os.path.exists(os.path.join(output_path, root[num:])):
                        os.makedirs(os.path.join(output_path, root[num:]))
                    im.save(os.path.join(output_path, root[num:], file.replace('tif', 'png')))
                except OSError:
                    print(os.path.join(output_path, root[num:], file))
                    img = im.convert("RGB")  # 4通道转化为rgb三通道
                    img.save(os.path.join(output_path, root[num:], file.replace('tif', 'png')))


if __name__ == '__main__':
    # if len(sys.argv) == 1:
    #     print("""
    #     功能：dds翻转输出png
    #     usge：python dds_to_png.py -i input_path -o output_path
    #     eg: python 4s_script_20220311.py -i /data/slice/data/inputpath -o /data/slice/data/inputpath
    #     -i --input_path dds输入目录
    #     -o --output_path png输出目录
    #     """)
    #
    # args = argparse.ArgumentParser()
    # args.add_argument('-i', '--input_path', required=True, type=str, help='dds输入目录')
    # args.add_argument('-o', '--output_path', required=True, type=str, help='png输出目录')
    #
    # # 参数获取
    # parse = args.parse_args()
    # input_path = parse.input_path
    # output_path = parse.output_path
    input_path = '/Volumes/data/model_data/750/huaihaicunju/HHC_n'
    output_path = '/Volumes/data/model_data/750/huaihaicunju/HHC_n'
    main(input_path, output_path)