publicImage/gprMax
1
0
派生 0
镜像自地址 https://gitee.com/sunhf/gprMax.git 已同步 2025-08-07 04:56:51 +08:00
代码 工单 软件包 项目 版本发布 百科 动态

Updated so they can be imported or run as scripts from the command line.

浏览代码
这个提交包含在：
Craig Warren
2016-04-26 17:24:53 +01:00
父节点 0e0d842556
当前提交 0c3e98f02f
共有 3 个文件被更改，包括 240 次插入 和 217 次删除
下载 Patch 文件 下载 Diff 文件 展开所有文件 折叠所有文件

57
tools/plot_Ascan.py
查看文件

@@ -25,17 +25,18 @@ import matplotlib.gridspec as gridspec
from gprMax.exceptions import CmdInputError
from gprMax.receivers import Rx
"""Plots electric and magnetic fields and currents 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 and currents 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 outputfile')
parser.add_argument('outputfile', help='name of output file including path')
parser.add_argument('--outputs', help='outputs to be plotted', choices='Ex, Ey, Ez, Hx, Hy, Hz, Ix, Iy, Iz', default=Rx.availableoutputs, nargs='+')
parser.add_argument('-fft', action='store_true', default=False, help='plot FFT (single output must be specified)')
args = parser.parse_args()
def make_plot(filename, outputs=Rx.availableoutputs, fft=False):
"""Plots electric and magnetic fields and currents from all receiver points in the given output file. Each receiver point is plotted in a new figure window.
Args:
filename (string): Filename (including path) of output file.
outputs (list): List of field/current components to plot.
fft (boolean): Plot FFT switch.
"""
# Open output file and read some attributes
f = h5py.File(args.outputfile, 'r')
f = h5py.File(filename, 'r')
nrx = f.attrs['nrx']
dt = f.attrs['dt']
iterations = f.attrs['Iterations']
@@ -44,11 +45,11 @@ time *= (iterations * dt)
# Check there are any receivers
if nrx == 0:
raise CmdInputError('No receivers found in {}'.format(args.outputfile))
raise CmdInputError('No receivers found in {}'.format(filename))
# Check for single output component when doing a FFT
if args.fft:
if not len(args.outputs) == 1:
if fft:
if not len(outputs) == 1:
raise CmdInputError('A single output must be specified when using the -fft option')
# New plot for each receiver
@@ -57,17 +58,17 @@ for rx in range(1, nrx + 1):
availableoutputs = list(f[path].keys())
# If only a single output is required, create one subplot
if len(args.outputs) == 1:
if len(outputs) == 1:
# Check for polarity of output and if requested output is in file
if args.outputs[0][0] == 'm':
if outputs[0][0] == 'm':
polarity = -1
outputtext = '-' + args.outputs[0][1:]
output = args.outputs[0][1:]
outputtext = '-' + outputs[0][1:]
output = outputs[0][1:]
else:
polarity = 1
outputtext = args.outputs[0]
output = args.outputs[0]
outputtext = outputs[0]
output = outputs[0]
if output not in availableoutputs:
raise CmdInputError('{} output requested to plot, but the available output for receiver 1 is {}'.format(output, ', '.join(availableoutputs)))
@@ -75,7 +76,7 @@ for rx in range(1, nrx + 1):
outputdata = f[path + output][:] * polarity
# Plotting if FFT required
if args.fft:
if fft:
# Calculate magnitude of frequency spectra of waveform
power = 10 * np.log10(np.abs(np.fft.fft(outputdata))**2)
freqs = np.fft.fftfreq(power.size, d=dt)
@@ -108,13 +109,13 @@ for rx in range(1, nrx + 1):
ax2.grid()
# Change colours and labels for magnetic field components or currents
if 'H' in args.outputs[0]:
if 'H' in outputs[0]:
plt.setp(line1, color='g')
plt.setp(line2, color='g')
plt.setp(ax1, ylabel=outputtext + ' field strength [A/m]')
plt.setp(stemlines, 'color', 'g')
plt.setp(markerline, 'markerfacecolor', 'g', 'markeredgecolor', 'g')
elif 'I' in args.outputs[0]:
elif 'I' in outputs[0]:
plt.setp(line1, color='b')
plt.setp(line2, color='b')
plt.setp(ax1, ylabel=outputtext + ' current [A]')
@@ -142,7 +143,7 @@ for rx in range(1, nrx + 1):
else:
fig, ax = plt.subplots(subplot_kw=dict(xlabel='Time [s]'), num='rx' + str(rx), figsize=(20, 10), facecolor='w', edgecolor='w')
gs = gridspec.GridSpec(3, 3, hspace=0.3, wspace=0.3)
for output in args.outputs:
for output in outputs:
# Check for polarity of output and if requested output is in file
if output[0] == 'm':
@@ -155,7 +156,7 @@ for rx in range(1, nrx + 1):
# Check if requested output is in file
if output not in availableoutputs:
raise CmdInputError('Output(s) requested to plot: {}, but available output(s) for receiver {} in the file: {}'.format(', '.join(args.outputs), rx, ', '.join(availableoutputs)))
raise CmdInputError('Output(s) requested to plot: {}, but available output(s) for receiver {} in the file: {}'.format(', '.join(outputs), rx, ', '.join(availableoutputs)))
outputdata = f[path + output][:] * polarity
@@ -210,3 +211,15 @@ for rx in range(1, nrx + 1):
#fig.savefig(os.path.splitext(os.path.abspath(file))[0] + '_rx' + str(rx) + '.png', dpi=150, format='png', bbox_inches='tight', pad_inches=0.1)
plt.show()
if __name__ == "__main__":
# Parse command line arguments
parser = argparse.ArgumentParser(description='Plots electric and magnetic fields and currents 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 outputfile')
parser.add_argument('outputfile', help='name of output file including path')
parser.add_argument('--outputs', help='outputs to be plotted', default=Rx.availableoutputs, choices='Ex, Ey, Ez, Hx, Hy, Hz, Ix, Iy, Iz', nargs='+')
parser.add_argument('-fft', action='store_true', help='plot FFT (single output must be specified)', default=False)
args = parser.parse_args()
make_plot(args.outputfile, args.outputs, fft=args.fft)

36
tools/plot_Bscan.py
查看文件

@@ -23,35 +23,36 @@ import matplotlib.pyplot as plt
from gprMax.exceptions import CmdInputError
"""Plots a B-scan image."""
# Parse command line arguments
parser = argparse.ArgumentParser(description='Plots a B-scan image.', usage='cd gprMax; python -m tools.plot_Bscan outputfile output')
parser.add_argument('outputfile', help='name of output file including path')
parser.add_argument('output', help='name of output component to be plotted', choices='Ex, Ey, Ez, Hx, Hy, Hz, Ix, Iy or Iz')
args = parser.parse_args()
def make_plot(filename, output):
"""Plots a B-scan image.
Args:
filename (string): Filename (including path) of output file.
output (string): Field/current component to plot.
"""
# Open output file and read some attributes
f = h5py.File(args.outputfile, 'r')
f = h5py.File(filename, 'r')
nrx = f.attrs['nrx']
# Check there are any receivers
if nrx == 0:
raise CmdInputError('No receivers found in {}'.format(args.outputfile))
raise CmdInputError('No receivers found in {}'.format(filename))
for rx in range(1, nrx + 1):
path = '/rxs/rx' + str(rx) + '/'
availableoutputs = list(f[path].keys())
# Check if requested output is in file
if args.output not in availableoutputs:
raise CmdInputError('{} output requested to plot, but the available output for receiver 1 is {}'.format(args.output, ', '.join(availableoutputs)))
if output not in availableoutputs:
raise CmdInputError('{} output requested to plot, but the available output for receiver 1 is {}'.format(output, ', '.join(availableoutputs)))
outputdata = f[path + '/' + args.output]
outputdata = f[path + '/' + output]
# Check that there is more than one A-scan present
if outputdata.shape[1] == 1:
raise CmdInputError('{} contains only a single A-scan.'.format(args.outputfile))
raise CmdInputError('{} contains only a single A-scan.'.format(filename))
# Plot B-scan image
fig = plt.figure(num='rx' + str(rx), figsize=(20, 10), facecolor='w', edgecolor='w')
@@ -72,3 +73,14 @@ for rx in range(1, nrx + 1):
#fig.savefig(os.path.splitext(os.path.abspath(args.outputfile))[0] + '.png', dpi=150, format='png', bbox_inches='tight', pad_inches=0.1)
plt.show()
if __name__ == "__main__":
# Parse command line arguments
parser = argparse.ArgumentParser(description='Plots a B-scan image.', usage='cd gprMax; python -m tools.plot_Bscan outputfile output')
parser.add_argument('outputfile', help='name of output file including path')
parser.add_argument('output', help='name of output component to be plotted', choices='Ex, Ey, Ez, Hx, Hy, Hz, Ix, Iy or Iz')
args = parser.parse_args()
make_plot(args.outputfile, args.output)

12
tools/plot_builtin_wave.py
查看文件

@@ -25,8 +25,6 @@ from gprMax.exceptions import CmdInputError
from gprMax.utilities import round_value
from gprMax.waveforms import Waveform
"""Plot built-in waveforms that can be used with sources."""
def check_timewindow(timewindow, dt):
"""Checks and sets time window and number of iterations.
@@ -58,7 +56,7 @@ def check_timewindow(timewindow, dt):
return timewindow, iterations
def plot_waveform(w, timewindow, dt, iterations, fft=False):
def make_plot(w, timewindow, dt, iterations, fft=False):
"""Plots waveform and prints useful information about its properties.
Args:
@@ -66,7 +64,7 @@ def plot_waveform(w, timewindow, dt, iterations, fft=False):
timewindow (float): Time window.
dt (float): Time discretisation.
iterations (int): Number of iterations.
fft (boolean): Plot FFT of waveform.
fft (boolean): Plot FFT switch.
"""
time = np.linspace(0, 1, iterations)
@@ -118,7 +116,7 @@ def plot_waveform(w, timewindow, dt, iterations, fft=False):
plt.setp(baseline, 'linewidth', 0)
plt.setp(stemlines, 'color', 'r')
plt.setp(markerline, 'markerfacecolor', 'r', 'markeredgecolor', 'r')
ax2.plot(freqs[pltrange]/1e9, power[pltrange], 'r', lw=2)
ax2.plot(freqs[pltrange], power[pltrange], 'r', lw=2)
ax2.set_xlabel('Frequency [Hz]')
ax2.set_ylabel('Power [dB]')
@@ -148,7 +146,7 @@ if __name__ == "__main__":
parser.add_argument('freq', type=float, help='centre frequency of waveform')
parser.add_argument('timewindow', help='time window to view waveform')
parser.add_argument('dt', type=float, help='time step to view waveform')
parser.add_argument('-fft', action='store_true', default=False, help='plot FFT of waveform')
parser.add_argument('-fft', action='store_true', help='plot FFT of waveform', default=False)
args = parser.parse_args()
# Check waveform parameters
@@ -164,7 +162,7 @@ if __name__ == "__main__":
w.freq = args.freq
timewindow, iterations = check_timewindow(args.timewindow, args.dt)
plot_waveform(w, timewindow, args.dt, iterations, args.fft)
make_plot(w, timewindow, args.dt, iterations, args.fft)