From d687cf85874aa2e490ca216b64b6f7bd95ced8bf Mon Sep 17 00:00:00 2001 From: "mrsingorxing@gmail.com" Date: Thu, 3 Jul 2025 12:07:58 +0800 Subject: [PATCH] fixed transmission loss plotting --- arlpy/uwapm.py | 10 +++++----- 1 file changed, 5 insertions(+), 5 deletions(-) diff --git a/arlpy/uwapm.py b/arlpy/uwapm.py index eef226c..6aeaa5c 100644 --- a/arlpy/uwapm.py +++ b/arlpy/uwapm.py @@ -524,18 +524,18 @@ def plot_transmission_loss(tloss, env=None, **kwargs): max_angle=45 ) >>> tloss = pm.compute_transmission_loss(env) - >>> pm.plot_transmission_loss(tloss, width=1000) + >>> pm.plot_transmission_loss(tloss,env, width=1000) """ - xr = (min(tloss.columns), max(tloss.columns)) + if env is None: + raise ValueError("Environment is required") + xr = (env['rx_range'].min(), env['rx_range'].max()) yr = (-max(tloss.index), -min(tloss.index)) xlabel = 'Range (m)' if xr[1]-xr[0] > 10000: - xr = (min(tloss.columns)/1000, max(tloss.columns)/1000) + xr = (env['rx_range'].min()/1000, env['rx_range'].max()/1000) xlabel = 'Range (km)' oh = _plt.hold() _plt.image(20*_np.log10(_fi.epsilon+_np.abs(_np.flipud(_np.array(tloss)))), x=xr, y=yr, xlabel=xlabel, ylabel='Depth (m)', xlim=xr, ylim=yr, **kwargs) - if env is not None: - plot_env(env, rx_plot=False) _plt.hold(oh) def pyplot_env(env, surface_color='dodgerblue', bottom_color='peru', tx_color='orangered', rx_color='midnightblue',