# -*- coding: utf-8 -*-
import warnings
warnings.filterwarnings('ignore')
import argparse, json
from common import NoIndent, MyEncoder

def _argparse():
    parser = argparse.ArgumentParser(description='cs 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 ("m" not in paramData) or ("d" not in paramData) or\
            ("f" not in paramData) or ("c" not in paramData) or\
            ("outParam" not in paramData):
            print(f'{argfile}中的参数有误，请核实！')
            return
        m = int(paramData["m"])
        d = float(paramData["d"])
        f = float(paramData["f"])
        c = float(paramData["c"])
        outParam = paramData["outParam"]
    
        from pyat.cscore import np, dis_mats, pn_funcs, ag_funcs
        import sys
        print(sys.path)
        
        
        # 输出：
        # 所有阵元的间距的集合
        dis_m = dis_mats(m, d)
        print('所有阵元的间距的集合：', dis_m)

        # 相邻阵元的噪声相关性
        pn_res = pn_funcs(f, c, dis_m)
        print('相邻阵元的噪声相关性：pn=', pn_res[0, 1])

        # 阵增益
        ag_res = ag_funcs(m, pn_res)
        print('阵增益：ag=', ag_res)
        outData = {
            "param": paramData,
            "dis": np.round(dis_m,2).tolist(),
            "pn": np.round(pn_res,2).tolist(),
            "ag": np.round(ag_res,2).tolist()
        }
        with open(("" if outParam["outPath"]=="" else outParam["outPath"] + "/")+"csOut.json", 'w') as jsonFile:
            json_data = json.dumps(outData, cls=MyEncoder, indent=2, ensure_ascii=False)
            jsonFile.write(json_data)
        # # write data to file                
        # # 保存地形数据
        # bathym.r = np.around(bathym.r, decimals=0)
        # bathym.d = np.around(bathym.d, decimals=0)
        # bathymData = {
        #     "param": paramData,
        #     "count": len(bathym.d),
        #     "descript": "距离、深度",
        #     "data": NoIndent(list(zip(bathym.r, bathym.d)))
        # }
        # with open(("" if outParam["outPath"]=="" else outParam["outPath"] + "/")+"tlOut_bathym.json", 'w') as jsonFile:
        #     json_data = json.dumps(bathymData, cls=MyEncoder, indent=2, ensure_ascii=False)
        #     jsonFile.write(json_data)

        # # 保存传播损失数据
        # tlpoint = []
        # depths = np.around(depths, decimals=0)
        # ranges = np.around(ranges, decimals=0)
        # TL = np.around(TL, decimals=2)
        # tlpoint = list(zip(np.tile(depths.reshape(len(depths),1),len(ranges)).flatten(), np.tile(ranges,(len(depths),1)).flatten(), TL.flatten()))
        # tlData = {
        #     "param": paramData,
        #     "count": len(ranges)*len(depths),
        #     "descript": "距离、深度、损失",
        #     "data": NoIndent(tlpoint)
        # }
        # with open(("" if outParam["outPath"]=="" else outParam["outPath"] + "/")+"tlOut_tl.json", 'w') as jsonFile:
        #     json_data = json.dumps(tlData, cls=MyEncoder, indent=2, ensure_ascii=False)
        #     jsonFile.write(json_data)
    
if __name__ == '__main__':
    main()