import os, sys, shutil, json
from osgeo import gdal
import matplotlib.pyplot as plt


# 判断路径是否是目录
def isDir(dirPath: str):
    return os.path.exists(dirPath) and os.path.isdir(dirPath)


# 对参数进行去重
def are_equal(*args):
    # 使用集合去除重复值，如果长度为1，则说明所有元素相等
    return len(set(args)) == 1


if are_equal(17136, 17136, 17136, 17136, 17136, 17136):
    print("ok")


# 删除一个目录或者文件
def rmPath(fileOrDirPath: str):
    if os.path.exists(fileOrDirPath):
        if os.path.isdir(fileOrDirPath):
            shutil.rmtree(fileOrDirPath)
        elif os.path.isfile(fileOrDirPath):
            os.remove(fileOrDirPath)


# 删除目录
def rmDir(dirPath: str) -> bool:
    if not os.path.isdir(dirPath):
        return True
    # 删除本地的input目录
    try:
        # 尝试递归删除目录及其内容
        shutil.rmtree(dirPath)
        # print(f"Directory {dirPath} successfully removed.")
    except OSError as e:
        # 如果目录无法删除，则捕获异常
        print(f"Failed to remove directory {dirPath} :", e)
        return False

    return True


# 创建目录
def mkDir(dirPath: str) -> bool:
    try:
        # 尝试递归创建目录及其父目录
        os.makedirs(dirPath)
        # print(f"Directory {dirPath} created successfully.")
    except OSError as e:
        # 如果创建目录失败，则捕获异常
        print(f"Failed to create directory {dirPath} :", e)
        return False
    return True


# 拷贝文件
def copyFile(sourceFile: str, targetFile: str) -> bool:
    try:
        # 尝试复制文件
        shutil.copy(sourceFile, targetFile)
        # print(f"File {sourceFile} copied successfully.")
    except IOError as e:
        # 如果复制文件失败，则捕获异常
        print(f"Failed to copy file {sourceFile}:", e)
        return False
    return True


# 将内容保持为json文件
def saveDataToJSON(content: list, fileName):
    with open(fileName, "w") as f:
        json.dump(content, f, indent=4)
    pass


# 判断文件是否被其他程序访问
def isFileOpen(filePath):
    try:
        thefile = open(filePath, "r+")
        thefile.close()
    except IOError as e:
        # 如果是Permission denied错误，通常意味着文件正在被其他程序访问
        if "Permission denied" in str(e):
            return True
    return False


def saveENVIImgToPng(imgFilePath: str, pngFilePath: str):
    # 打开 ENVI 格式的图像文件
    # image_path = "postProcessing/Case1/reflectance/08.REF.img"
    # 如果文件不存在进行报错并退出
    if not os.path.isfile(imgFilePath):
        print("img file not exist!")
        return None
    dataset = gdal.Open(imgFilePath, gdal.GA_ReadOnly)
    if dataset is None:
        print("Failed to open image")
        return None

    # 读取图像数据
    band = dataset.GetRasterBand(1)  # 仅读取第一个波段
    image_array = band.ReadAsArray()

    # 显示图像
    plt.imshow(image_array)  # 使用灰度色彩映射
    # plt.imshow(image_array, cmap="gray")  # 使用灰度色彩映射
    # plt.colorbar()
    # plt.show()

    plt.savefig(pngFilePath)

    # 关闭数据集
    dataset = None
    # 发货文件名
    return pngFilePath

def printWithFlush(*args, sep=' ', end='\n', file=None, flush=True):
    print(*args, sep=sep, end=end, file=file, flush=flush)
    # sys.stdout.flush()