import subprocess, os, time, shutil, configparser, sys # 获取当前文件的父目录的绝对路径 current_dir = os.path.dirname(os.path.abspath(__file__)) # 获取父目录的绝对路径 parent_dir = os.path.abspath(os.path.join(current_dir, os.pardir)) # 将父目录添加到 Python 模块搜索路径中 sys.path.append(parent_dir) from Utils import utils as globalUtils # python.exe main.py E:\01-Projects\01-CAE-FullLink\output\workingDir\test06\SurfaceCharacteristicSimulation inputDir = "input/" inputDistDir = "Dir_REF/" outputSourceDir = "Dir_REF/" outputDistDir = "output/" def readParamValueList(paramFilePath: str): paramValueList = [] with open(paramFilePath, "r", encoding="utf-8") as file: for line in file: paramValueList = paramValueList + [float(x) for x in line.split(",")] return paramValueList # 读取输入文件，生成所有组合的值 def readInput() -> list: # 读取input_surface.txt，获取每个参数文件公共内容部分 inputfilePath = inputDir + "input_surface.txt" paramNameList = "" publicParamValueList = [] with open(inputfilePath, "r", encoding="utf-8") as file: lines = file.readlines() paramNameList = lines[0] publicParamValueList = [int(x) for x in lines[1].split(",")] globalUtils.printWithFlush(paramNameList) globalUtils.printWithFlush(publicParamValueList) # 依次读取 Lat、Lon、SA、SZ、VA、VZ latFilePath = inputDir + "Lat.txt" latValueList = readParamValueList(latFilePath) lonFilePath = inputDir + "Lon.txt" lonValueList = readParamValueList(lonFilePath) saFilePath = inputDir + "SA.txt" saValueList = readParamValueList(saFilePath) szFilePath = inputDir + "SZ.txt" szValueList = readParamValueList(szFilePath) vaFilePath = inputDir + "VA.txt" vaValueList = readParamValueList(vaFilePath) vzFilePath = inputDir + "VZ.txt" vzValueList = readParamValueList(vzFilePath) globalUtils.printWithFlush("经度参数个数 " + str(len(latValueList))) globalUtils.printWithFlush("纬度参数个数 " + str(len(lonValueList))) globalUtils.printWithFlush("太阳天顶角参数个数 " + str(len(saValueList))) globalUtils.printWithFlush("观察天顶角参数个数 " + str(len(szValueList))) globalUtils.printWithFlush("太阳方位角参数个数 " + str(len(vaValueList))) globalUtils.printWithFlush("观测方位角参数个数 " + str(len(vzValueList))) if not globalUtils.are_equal( len(latValueList), len(lonValueList), len(saValueList), len(szValueList), len(vaValueList), len(vzValueList), ): globalUtils.printWithFlush( "经度、纬度、太阳天顶角、观察天顶角、太阳方位角、观测方位角的参数个数不一致，请查看参数文件。" ) return [] # 参数内容列表 paramContentList = [] for index, lat in enumerate(latValueList): lon = lonValueList[index] sa = saValueList[index] sz = szValueList[index] va = vaValueList[index] vz = vzValueList[index] paramContentList.append( [ publicParamValueList[0], lon, lat, sz, vz, sa, va, publicParamValueList[1], publicParamValueList[2], ] ) # globalUtils.printWithFlush(latValueList) # return [paramNameList, paramValueList.split(",")] return [paramNameList, paramContentList] # 从项目目录下拷贝参数文件过来 def copyInputFiles(projectDir: str) -> bool: # 先删除，再创建 globalUtils.rmPath(inputDir) os.mkdir(inputDir) inputFileList = [ "input_surface.txt", "Lat.txt", "Lon.txt", "SA.txt", "SZ.txt", "VA.txt", "VZ.txt", ] for inputFile in inputFileList: inputFileProjectPath = os.path.join(projectDir, inputFile) if not os.path.isfile(inputFileProjectPath): globalUtils.printWithFlush(f"{inputFileProjectPath} is not file or exist") return False inputFileSolverPath = os.path.join(inputDir, inputFile) shutil.copy(inputFileProjectPath, inputFileSolverPath) return True # 执行一次求解 def executeCompute( inputFile: str, paramNameList, paramContent, outputSourceFile, outputDistFile ): # 先删除旧的input.txt globalUtils.rmPath(inputFile) # 生成新的input.txt with open(inputFile, "w") as file: file.write(paramNameList) file.write(paramContent) # 先删除旧的output.txt globalUtils.rmPath(outputSourceFile) # 启动求解任务 result = subprocess.run(["Cal_REF/Cal_REF.exe", "Dir_REF/input.txt"], cwd="Cal_REF") # 循环等待结果文件的生成，然后进行移动 while 1: time.sleep(3) # 如果文件生成，且没有被其他程序打开（被其他程序访问，例如：正在往里写文件） if os.path.isfile(outputSourceFile) and not globalUtils.isFileOpen( outputSourceFile ): os.rename(outputSourceFile, outputDistFile) break return result # 其它模块调用本模块的求解（大气模块调用） def toaRunModel(surFileDir: str): # 提供的目录下有 input.txt，没有output.txt inputSourceFile = os.path.join(surFileDir, "input.txt") outputDistFile = os.path.join(surFileDir, "output.txt") # 求解时的 input.txt和output.txt inputFilePath = os.path.join(inputDistDir, "input.txt") outputSourceFile = os.path.join(inputDistDir, "output.txt") # 第1步，拷贝inputFile到 with open(inputSourceFile, "r", encoding="utf-8") as file: lines = file.readlines() if len(lines) < 2: return None return executeCompute( inputFilePath, lines[0], lines[1], outputSourceFile, outputDistFile ) def test(testDir: str): currentExeDir = os.getcwd() globalUtils.printWithFlush("execute dir: ", currentExeDir) # 获取当前文件的绝对路径 currentFile = os.path.realpath(__file__) # 获取当前文件的目录 currentDir = os.path.dirname(currentFile) # 切换到'/home' 目录 os.chdir(currentDir) globalUtils.printWithFlush("surface dir: ", currentDir) inputFilePath = os.path.join(inputDistDir, "input.txt") globalUtils.printWithFlush("surface input.txt: ", inputFilePath) globalUtils.printWithFlush(os.getcwd()) # 执行一次求解 result = toaRunModel(testDir) globalUtils.printWithFlush("surface run ok:", result) # 切换回之前的目录 os.chdir(currentExeDir) pass def generateImgPng(): imgFileDir = "postProcessing" pngFileDir = "output" caseList = ["Case1", "Case2"] for case in caseList: imgList = ["07.REF.img", "08.REF.img", "09.REF.img", "10.REF.img"] for imgFile in imgList: imgFilePath = os.path.join(imgFileDir, case + "/reflectance/" + imgFile) pngFilePath = os.path.join( pngFileDir, case + "-" + imgFile.replace(".img", ".png") ) pngFilePath = globalUtils.saveENVIImgToPng(imgFilePath, pngFilePath) pass def main(): # 项目目录的output，求解结束后传回到项目目录下 projectDir = sys.argv[1] globalUtils.printWithFlush(projectDir) # 从项目目录下拷贝参数文件过来 if not copyInputFiles(projectDir): globalUtils.printWithFlush("参数文件不完整") return # 读取参数文件 paramContentValue = readInput() if len(paramContentValue) == 0: return paramNameList = paramContentValue[0] paramContentList = paramContentValue[1] # 读取配置文件 # 创建 ConfigParser 对象 config = configparser.ConfigParser() config.read("config.ini") taskNum = int(config.get("settings", "taskNum")) # 生成的结果文件 outputSourceFile = outputSourceDir + "output.txt" inputFile = inputDistDir + "input.txt" # 删除之前的输出文件目录，创建新的目录 globalUtils.rmPath(outputDistDir) os.mkdir(outputDistDir) # 获取当前时间戳 timestamp = time.time() # 根据任务数进行求解 for index, paramContent in enumerate(paramContentList): # 只计算200个 if index == taskNum: break result = executeCompute( inputFile, paramNameList, ",".join(map(str, paramContent)), outputSourceFile, outputDistDir + "output" + str(index) + ".txt", ) runTimestamp = time.time() globalUtils.printWithFlush(result, index, runTimestamp - timestamp) timestamp = runTimestamp # 生成后处理图片 generateImgPng() # 项目目录下的output目录 projectOutputDir = os.path.join(projectDir, "output") # 删除项目目录下旧的结果文件 globalUtils.rmPath(projectOutputDir) # 将结果拷贝到目的目录 shutil.copytree(outputDistDir, projectOutputDir) pass # 主函数 if __name__ == "__main__": try: # 尝试执行的代码块 main() except Exception as e: # 捕获除了ZeroDivisionError外的所有异常 globalUtils.printWithFlush("发生异常:", str(e)) finally: # 无论是否发生异常，最终都会执行该代码块 globalUtils.printWithFlush("求解执行完毕！")