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

Fixed bug with overlapping text on plots

浏览代码
这个提交包含在：
Craig Warren
2025-07-17 12:33:35 +01:00
父节点 96790996e2
当前提交 225a6d3ce6
共有 1 个文件被更改，包括 42 次插入 和 54 次删除
下载 Patch 文件 下载 Diff 文件 展开所有文件 折叠所有文件

96
tools/plot_Ascan.py
查看文件

@@ -18,12 +18,10 @@
import argparse import argparse
import os import os
import sys
import h5py import h5py
import numpy as np import numpy as np
import matplotlib.pyplot as plt import matplotlib.pyplot as plt
import matplotlib.gridspec as gridspec
from gprMax.exceptions import CmdInputError from gprMax.exceptions import CmdInputError
from gprMax.receivers import Rx from gprMax.receivers import Rx
@@ -135,7 +133,6 @@ def mpl_plot(filename, outputs=Rx.defaultoutputs, fft=False):
fig, ax = plt.subplots(subplot_kw=dict(xlabel='Time [s]', ylabel=outputtext + ' field strength [V/m]'), num='rx' + str(rx), figsize=(20, 10), facecolor='w', edgecolor='w') fig, ax = plt.subplots(subplot_kw=dict(xlabel='Time [s]', ylabel=outputtext + ' field strength [V/m]'), num='rx' + str(rx), figsize=(20, 10), facecolor='w', edgecolor='w')
line = ax.plot(time, outputdata, 'r', lw=2, label=outputtext) line = ax.plot(time, outputdata, 'r', lw=2, label=outputtext)
ax.set_xlim([0, np.amax(time)]) ax.set_xlim([0, np.amax(time)])
# ax.set_ylim([-15, 20])
ax.grid(which='both', axis='both', linestyle='-.') ax.grid(which='both', axis='both', linestyle='-.')
if 'H' in output: if 'H' in output:
@@ -147,15 +144,21 @@ def mpl_plot(filename, outputs=Rx.defaultoutputs, fft=False):
# If multiple outputs required, create all nine subplots and populate only the specified ones # If multiple outputs required, create all nine subplots and populate only the specified ones
else: else:
fig, ax = plt.subplots(subplot_kw=dict(xlabel='Time [s]'), num='rx' + str(rx), figsize=(20, 10), facecolor='w', edgecolor='w') plt_cols = 3 if len(outputs) == 9 else 2
if len(outputs) == 9:
gs = gridspec.GridSpec(3, 3, hspace=0.3, wspace=0.3) fig, axs = plt.subplots(
else: subplot_kw=dict(xlabel="Time [s]"),
gs = gridspec.GridSpec(3, 2, hspace=0.3, wspace=0.3) num='rx' + str(rx),
figsize=(20, 10),
nrows = 3,
ncols = plt_cols,
facecolor="w",
edgecolor="w",
)
for output in outputs: for output in outputs:
# Check for polarity of output and if requested output is in file # Check for polarity of output and if requested output is in file
if output[-1] == 'm': if output[-1] == '-':
polarity = -1 polarity = -1
outputtext = '-' + output[0:-1] outputtext = '-' + output[0:-1]
output = output[0:-1] output = output[0:-1]
@@ -169,51 +172,36 @@ def mpl_plot(filename, outputs=Rx.defaultoutputs, fft=False):
outputdata = f[path + output][:] * polarity outputdata = f[path + output][:] * polarity
if output == 'Ex': if output == "Ex":
ax = plt.subplot(gs[0, 0]) axs[0, 0].plot(time, outputdata, "r", lw=2, label=outputtext)
ax.plot(time, outputdata, 'r', lw=2, label=outputtext) axs[0, 0].set_ylabel(outputtext + ", field strength [V/m]")
ax.set_ylabel(outputtext + ', field strength [V/m]') elif output == "Ey":
# ax.set_ylim([-15, 20]) axs[1, 0].plot(time, outputdata, "r", lw=2, label=outputtext)
elif output == 'Ey': axs[1, 0].set_ylabel(outputtext + ", field strength [V/m]")
ax = plt.subplot(gs[1, 0]) elif output == "Ez":
ax.plot(time, outputdata, 'r', lw=2, label=outputtext) axs[2, 0].plot(time, outputdata, "r", lw=2, label=outputtext)
ax.set_ylabel(outputtext + ', field strength [V/m]') axs[2, 0].set_ylabel(outputtext + ", field strength [V/m]")
# ax.set_ylim([-15, 20]) elif output == "Hx":
elif output == 'Ez': axs[0, 1].plot(time, outputdata, "g", lw=2, label=outputtext)
ax = plt.subplot(gs[2, 0]) axs[0, 1].set_ylabel(outputtext + ", field strength [A/m]")
ax.plot(time, outputdata, 'r', lw=2, label=outputtext) elif output == "Hy":
ax.set_ylabel(outputtext + ', field strength [V/m]') axs[1, 1].plot(time, outputdata, "g", lw=2, label=outputtext)
# ax.set_ylim([-15, 20]) axs[1, 1].set_ylabel(outputtext + ", field strength [A/m]")
elif output == 'Hx': elif output == "Hz":
ax = plt.subplot(gs[0, 1]) axs[2, 1].plot(time, outputdata, "g", lw=2, label=outputtext)
ax.plot(time, outputdata, 'g', lw=2, label=outputtext) axs[2, 1].set_ylabel(outputtext + ", field strength [A/m]")
ax.set_ylabel(outputtext + ', field strength [A/m]') elif output == "Ix":
# ax.set_ylim([-0.03, 0.03]) axs[0, 2].plot(time, outputdata, "b", lw=2, label=outputtext)
elif output == 'Hy': axs[0, 2].set_ylabel(outputtext + ", current [A]")
ax = plt.subplot(gs[1, 1]) elif output == "Iy":
ax.plot(time, outputdata, 'g', lw=2, label=outputtext) axs[1, 2].plot(time, outputdata, "b", lw=2, label=outputtext)
ax.set_ylabel(outputtext + ', field strength [A/m]') axs[1, 2].set_ylabel(outputtext + ", current [A]")
# ax.set_ylim([-0.03, 0.03]) elif output == "Iz":
elif output == 'Hz': axs[2, 2].plot(time, outputdata, "b", lw=2, label=outputtext)
ax = plt.subplot(gs[2, 1]) axs[2, 2].set_ylabel(outputtext + ", current [A]")
ax.plot(time, outputdata, 'g', lw=2, label=outputtext) for ax in fig.axes:
ax.set_ylabel(outputtext + ', field strength [A/m]') ax.set_xlim([0, np.amax(time)])
# ax.set_ylim([-0.03, 0.03]) ax.grid(which="both", axis="both", linestyle="-.")
elif output == 'Ix':
ax = plt.subplot(gs[0, 2])
ax.plot(time, outputdata, 'b', lw=2, label=outputtext)
ax.set_ylabel(outputtext + ', current [A]')
elif output == 'Iy':
ax = plt.subplot(gs[1, 2])
ax.plot(time, outputdata, 'b', lw=2, label=outputtext)
ax.set_ylabel(outputtext + ', current [A]')
elif output == 'Iz':
ax = plt.subplot(gs[2, 2])
ax.plot(time, outputdata, 'b', lw=2, label=outputtext)
ax.set_ylabel(outputtext + ', current [A]')
for ax in fig.axes:
ax.set_xlim([0, np.amax(time)])
ax.grid(which='both', axis='both', linestyle='-.')
# Save a PDF/PNG of the figure # Save a PDF/PNG of the figure
# fig.savefig(os.path.splitext(os.path.abspath(filename))[0] + '_rx' + str(rx) + '.pdf', dpi=None, format='pdf', bbox_inches='tight', pad_inches=0.1) # fig.savefig(os.path.splitext(os.path.abspath(filename))[0] + '_rx' + str(rx) + '.pdf', dpi=None, format='pdf', bbox_inches='tight', pad_inches=0.1)