Python实现自动批量写gprMax3.0输入文件以及批量运行

之前实现了python运行gprMax'3.0,python运行gprMax3.0_zhoudapeng01的专栏-CSDN博客

此处利用python实现生成批量的gprMax3.0模型文件，并且进行批量的运行，参考了其他博主的批量运行python脚本运行gprMax3.0批量仿真GPR数据_LuoJiabin的博客-CSDN博客

需要注意的事项：

1、写完模型输入文件时，要关闭文件，否则后面调用api运行模型时会出错；

2、运行Bscan时，尽量不输出模型的vti文件，太占内存，geomery_view注释掉

批量生成gprMax3.0文件代码

import os
from gprMax.gprMax import api
import numpy as np
from tools.plot_Bscan import get_output_data, mpl_plot
from tools.outputfiles_merge import merge_files
import matplotlib.pyplot as plt
path = os.getcwd()  # 获得当前脚本所在的文件路径
for i in range(1,4):
    filename = str(i)+"th_model.in"
    f = open(filename, 'a')
    f.write("#title: the  {}th model\n".format(i))
    #保证模型的区域一致，便于做深度学习
    f.write("#domain: 8.4 14.4 0.02\n")
    #网格大小可以根据中心频率的需要进行调节
    f.write("#dx_dy_dz: 0.02 0.02 0.02\n")
    #时窗尽量保持一致
    f.write("#time_window: 50e-9\n")
    #材料可以设置为变化的
    f.write("#material: 6 0.01 1 0 car\n")
    #天线类型尽量保持一致，频率可以改变
    f.write("#waveform: ricker 1 100e6 my_ricker\n")
    #改变发射源的位置
    start_source_x=4.2
    start_source_y=0.2
    start_source_z=0
    source_x=start_source_x+i*0.5
    source_y=start_source_y+i*0.1
    source_z=start_source_z+i*0
    f.write("#hertzian_dipole: z {} {} {} my_ricker\n".format(source_x,source_y,source_z))
    #改变接收源的位置
    start_Rev_x=4.2
    start_Rev_y=0.75
    start_Rev_z=0
    Rev_x=start_Rev_x+i*0.5
    Rev_y=start_Rev_y+i+0.1
    Rev_z=start_Rev_z+i*0
    f.write("#rx: {} {} {}\n".format(Rev_x,Rev_y,Rev_z))
    #Bscan需要添加源移动步长和接收移动的步长
    step_source_x=0.02
    step_source_y=0
    step_source_z=0
    f.write("#src_steps: {} {} {}\n".format(step_source_x,step_source_y,step_source_z))
    step_Rev_x=0.02
    step_Rev_y=0
    step_Rev_z=0
    f.write("#rx_steps: {} {} {}\n".format(step_Rev_x,step_Rev_y,step_Rev_z))
    #模型可以根据需要进行改变,背景尽量设置为一个固定的，然后添加各种形状的异常
    f.write("#box: 0 0 0 8.4 14.4 0.02 car\n")
    f.write("#box: 4.7 11 0 6.7 12 0.02 free_space\n")
    f.write("#box: 4.7 6.5 0 5.7 7.5 0.02 free_space\n")
    f.write("#box: 4.7 2.5 0 5.2 3.5 0.02 free_space\n")
    #模型是否显示
    f.write("#geometry_view: 0 0 0 8.4 14.4 0.02 0.02 0.02 0.02 {} n \n".format(str(i)+"th_model"))
    #模型文件写完之后要关闭，免得影响后面的调用
    f.close()

批量运行代码（接上面代码）

#模型的运行
    api(filename, n=3, geometry_only=False)  # geometry_only：仅几何图形
    fi = filename[0:-3]
    merge_files(fi, removefiles=True)

    #保存运行数据为txt，以及进行绘图保存
    filename_b = fi + '_merged.out'
    rxnumber = 1
    rxcomponent = 'Ez'
    # 获取回波数据
    outputdata, dt = get_output_data(filename_b, rxnumber, rxcomponent)
    # 保存回波数据
    fi2 = fi.split('/')[-1]  # 得到文件名，去掉了前面的路径
    out_path = path + '/out_data'
    if (os.path.exists(out_path)):
        pass
    else:
        os.mkdir(out_path)
    np.savetxt(out_path + '/' + fi2 + '.txt', outputdata, delimiter=' ')  # 未去除直达波

    # 绘图
    plt.imshow(outputdata, extent=[0, outputdata.shape[1], outputdata.shape[0], 0], interpolation='nearest',
               aspect='auto', cmap='gray',
               vmin=-np.amax(np.abs(outputdata)), vmax=np.amax(np.abs(outputdata)))
    img_path = path + '/img_data'
    if (os.path.exists(img_path)):
        pass
    else:
        os.mkdir(img_path)
    plt.savefig(img_path + '/' + fi2 + '.png', dpi=300)  # 保存图片

运行完之后，会出现3个输入.in文件，3个合成后的输出.out文件，以及3个txt数据文件和3张图