fulllink-solver/GeometricSightTracking/main.py

import matlab.engine
import sys
import os
import json
import mainMapEcharts
import shutil

# 获取当前文件的父目录的绝对路径
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\test1\PlatformPerformance\a_sat.txt E:\01-Projects\01-CAE-FullLink\output\workingDir\test2\GeometricSightTracking\output
outputDir = "output"


# 执行一次求解
def calculation(eng: matlab.engine):
    # 执行一次求解
    eng.main(nargout=0)


def readInputGeo():
    filePath = "input_geo.txt"
    paramNameList = ""
    paramValueList = []
    with open(filePath, "r", encoding="utf-8") as file:
        lines = file.readlines()
        paramNameList = lines[0]
        paramValueList = lines[1]

    return [paramNameList, paramValueList.split(",")]


# 生成参数组合
def genParamList(filePath: str):
    paramList = []
    with open(filePath, "r", encoding="utf-8") as file:
        for line in file:
            paramList.append(line.split()[1:])
    return paramList


# 生成一个求解参数文件
def genParamFile(paramVector: list, geoParamNameList: list, geoParamValueList: list):
    paramValueList = [geoParamValueList[0]] + paramVector + geoParamValueList[1:]
    globalUtils.printWithFlush(paramValueList)
    globalUtils.printWithFlush(geoParamNameList)

    with open("input.txt", "w") as file:
        file.write(geoParamNameList)
        file.write(",".join(map(str, paramValueList)))
    pass


# 将输出文件转换为json格式
def genGeoJsonFileFromOutput(outputFilePath: str):
    if not os.path.exists(outputFilePath):
        return
    jsonFilePath = outputFilePath.replace(".txt", ".json")
    geoDataDict = {}
    i = 1
    with open(outputFilePath, "r", encoding="utf-8") as file:
        for line in file:
            geoDataDict[str(i)] = [float(x) for x in line.split()]
            i += 1
    with open(jsonFilePath, "w") as file:
        file.write(json.dumps(geoDataDict))
    pass


def isDir(dirPath: str):
    return os.path.exists(dirPath) and os.path.isdir(dirPath)


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 main():
    # 平台性能模块参数部分
    aSatFilePath = sys.argv[1]
    # 结果文件的目录地址
    outputDistDir = sys.argv[2]

    # 读取参数组合
    paramList = genParamList(aSatFilePath)

    # TODO，检查，是否有 input_geo.txt、平台性能模块文件有参数

    # 读取 geo模块的参数
    paramDict = readInputGeo()
    paramNameList = paramDict[0]
    paramValueList = paramDict[1]

    # 判断output目录是否存在，不存在在创建
    rmPath(outputDir)
    os.mkdir(outputDir)

    eng = matlab.engine.start_matlab()
    for index, paramVector in enumerate(paramList):
        # 生成求解参数文件
        genParamFile(paramVector, paramNameList, paramValueList)
        # 调用求解器求解
        calculation(eng)
        # 将结果文件移动到output目录中
        outputFileName = outputDir + "/output" + str(index) + ".txt"
        if os.path.exists(outputFileName):
            os.remove(outputFileName)
        os.rename("output.txt", outputFileName)
        # 将结果数据生成html
        mainMapEcharts.rendOutputGeo(outputFileName)
        # 将结果数据生成为json文件
        # genGeoJsonFileFromOutput(outputFilePath=outputFileName)
        # break

    # 删除旧的目录
    rmPath(outputDistDir)

    # 将结果拷贝的目标目录下
    shutil.copytree(outputDir, outputDistDir)


if __name__ == "__main__":
    try:
        # 尝试执行的代码块
        main()
    except Exception as e:
        # 捕获除了ZeroDivisionError外的所有异常
        globalUtils.printWithFlush("发生异常:", str(e))
    finally:
        # 无论是否发生异常，最终都会执行该代码块
        globalUtils.printWithFlush("求解执行完毕！")