51 行
2.0 KiB
Python
51 行
2.0 KiB
Python
# -*- coding: utf-8 -*-
|
|
import warnings
|
|
warnings.filterwarnings('ignore')
|
|
import argparse, json
|
|
from common import NoIndent, MyEncoder
|
|
|
|
def _argparse():
|
|
parser = argparse.ArgumentParser(description='tl client')
|
|
parser.add_argument('-a', '--argfile', action='store', dest='argfile', required=True, help='name of arguments file')
|
|
return parser.parse_args()
|
|
|
|
def main():
|
|
parser = _argparse()
|
|
argfile = parser.argfile
|
|
if argfile[-5:] != '.json':
|
|
print('输入的参数文件名有误')
|
|
return
|
|
with open(argfile,'r') as jsonFile:
|
|
paramData = json.load(jsonFile)
|
|
if ("ETOPO1Path" not in paramData) or ("STPath" not in paramData) or\
|
|
("freq" not in paramData) or ("receiveDepth" not in paramData) or\
|
|
("lat" not in paramData) or ("lon" not in paramData) or\
|
|
("timeType" not in paramData) or\
|
|
("NThreads" not in paramData) or ("outParam" not in paramData):
|
|
print(f'{argfile}中的参数有误，请核实！')
|
|
return
|
|
ETOPO1Path = paramData["ETOPO1Path"]
|
|
STPath = paramData["STPath"]
|
|
freq = float(paramData["freq"])
|
|
receiveDepth = float(paramData["receiveDepth"])
|
|
lat = float(paramData["lat"])
|
|
lon = float(paramData["lon"])
|
|
timeType = paramData["timeType"]
|
|
NThreads = int(paramData["NThreads"])
|
|
outParam = paramData["outParam"]
|
|
|
|
from pyat.RANDIcore import call_RANDI
|
|
NL = call_RANDI(ETOPO1Path, STPath, outParam["outPath"], freq, receiveDepth,
|
|
lon, lat, timeType, NThreads)
|
|
print(round(NL,2))
|
|
outData = {
|
|
"param": paramData,
|
|
"noiseLevel": round(NL,2)
|
|
}
|
|
with open(("" if outParam["outPath"]=="" else outParam["outPath"] + "/")+"RANDIOut.json", 'w') as jsonFile:
|
|
json_data = json.dumps(outData, cls=MyEncoder, indent=2, ensure_ascii=False)
|
|
jsonFile.write(json_data)
|
|
|
|
|
|
if __name__ == '__main__':
|
|
main() |