你已经派生过 gprMax
镜像自地址
https://gitee.com/sunhf/gprMax.git
已同步 2025-08-07 04:56:51 +08:00
49 行
2.0 KiB
Python
49 行
2.0 KiB
Python
# Copyright (C) 2015: The University of Edinburgh
|
|
# Authors: Craig Warren and Antonis Giannopoulos
|
|
#
|
|
# This file is part of gprMax.
|
|
#
|
|
# gprMax is free software: you can redistribute it and/or modify
|
|
# it under the terms of the GNU General Public License as published by
|
|
# the Free Software Foundation, either version 3 of the License, or
|
|
# (at your option) any later version.
|
|
#
|
|
# gprMax is distributed in the hope that it will be useful,
|
|
# but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
# GNU General Public License for more details.
|
|
#
|
|
# You should have received a copy of the GNU General Public License
|
|
# along with gprMax. If not, see <http://www.gnu.org/licenses/>.
|
|
|
|
import os, argparse
|
|
import h5py
|
|
import numpy as np
|
|
from .plot_fields import plot_Ascan
|
|
|
|
"""Plots electric and magnetic fields from all receiver points in the given output file. Each receiver point is plotted in a new figure window."""
|
|
|
|
# Parse command line arguments
|
|
parser = argparse.ArgumentParser(description='Plots electric and magnetic fields from all receiver points in the given output file. Each receiver point is plotted in a new figure window.', usage='cd gprMax; python -m tools.plot_Ascan_hdf5 outputfile')
|
|
parser.add_argument('outputfile', help='name of output file including path')
|
|
args = parser.parse_args()
|
|
|
|
file = args.outputfile
|
|
f = h5py.File(file, 'r')
|
|
nrx = f.attrs['nrx']
|
|
time = np.arange(0, f.attrs['dt'] * f.attrs['Iterations'], f.attrs['dt'])
|
|
time = time / 1E-9
|
|
|
|
for rx in range(1, nrx + 1):
|
|
path = '/rxs/rx' + str(rx) + '/'
|
|
Ex = f[path + 'Ex'][:]
|
|
Ey = f[path + 'Ey'][:]
|
|
Ez = f[path + 'Ez'][:]
|
|
Hx = f[path + 'Hx'][:]
|
|
Hy = f[path + 'Hy'][:]
|
|
Hz = f[path + 'Hz'][:]
|
|
fig, plt = plot_Ascan('rx' + str(rx), time, Ex, Ey, Ez, Hx, Hy, Hz)
|
|
#fig.savefig(os.path.splitext(os.path.abspath(file))[0] + '.pdf', dpi=None, format='pdf', bbox_inches='tight', pad_inches=0.1)
|
|
plt.show()
|
|
|
|
f.close() |