Corrected electric field to voltage conversion.

Corrected figure plot saving filenames.
Added better checking of available components from a receiver output.
这个提交包含在:
Craig Warren
2016-05-10 14:05:41 +01:00
父节点 1c5f02fc45
当前提交 bb306a60b7

查看文件

@@ -37,6 +37,7 @@ def plot_antenna_params(filename, tln=1, rxn=None, rx=None):
# Open output file and read some attributes # Open output file and read some attributes
f = h5py.File(filename, 'r') f = h5py.File(filename, 'r')
dxdydz = f.attrs['dx, dy, dz']
dt = f.attrs['dt'] dt = f.attrs['dt']
iterations = f.attrs['Iterations'] iterations = f.attrs['Iterations']
@@ -68,9 +69,22 @@ def plot_antenna_params(filename, tln=1, rxn=None, rx=None):
if rxn: if rxn:
if rx not in ['Ex', 'Ey', 'Ez']: if rx not in ['Ex', 'Ey', 'Ez']:
raise CmdInputError('The field component for the receiver antenna output must be Ex, Ey, or Ez') raise CmdInputError('The field component for the receiver antenna output must be Ex, Ey, or Ez')
rxpath = '/rxs/rx' + str(rxn) + '/' + rx
rxpath = '/rxs/rx' + str(rxn) + '/'
availableoutputs = list(f[rxpath].keys())
if rx not in availableoutputs:
raise CmdInputError('{} output requested, but the available output for receiver {} is {}'.format(rx, rxn, ', '.join(availableoutputs)))
rxpath += rx
# Received voltage # Received voltage
Vrec = f[rxpath][:] * -1 if rx == 'Ex':
Vrec = f[rxpath][:] * -1 * dxdydz[0]
elif rx == 'Ey':
Vrec = f[rxpath][:] * -1 * dxdydz[1]
elif rx == 'Ez':
Vrec = f[rxpath][:] * -1 * dxdydz[2]
f.close() f.close()
# Frequency bins # Frequency bins
@@ -336,11 +350,10 @@ def plot_antenna_params(filename, tln=1, rxn=None, rx=None):
#ax.grid() #ax.grid()
# Save a PDF/PNG of the figure # Save a PDF/PNG of the figure
#fig1.savefig(os.path.splitext(os.path.abspath(file))[0] + '_tl_params.png', dpi=150, format='png', bbox_inches='tight', pad_inches=0.1) #fig1.savefig(os.path.splitext(os.path.abspath(filename))[0] + '_tl_params.png', dpi=150, format='png', bbox_inches='tight', pad_inches=0.1)
#fig2.savefig(os.path.splitext(os.path.abspath(file))[0] + '_ant_params.png', dpi=150, format='png', bbox_inches='tight', pad_inches=0.1) #fig2.savefig(os.path.splitext(os.path.abspath(filename))[0] + '_ant_params.png', dpi=150, format='png', bbox_inches='tight', pad_inches=0.1)
#fig3.savefig(os.path.splitext(os.path.abspath(file))[0] + '_ant_params.png', dpi=150, format='png', bbox_inches='tight', pad_inches=0.1) #fig1.savefig(os.path.splitext(os.path.abspath(filename))[0] + '_tl_params.pdf', dpi=None, format='pdf', bbox_inches='tight', pad_inches=0.1)
#fig1.savefig(os.path.splitext(os.path.abspath(file))[0] + '_tl_params.pdf', dpi=None, format='pdf', bbox_inches='tight', pad_inches=0.1) #fig2.savefig(os.path.splitext(os.path.abspath(filename))[0] + '_ant_params.pdf', dpi=None, format='pdf', bbox_inches='tight', pad_inches=0.1)
#fig2.savefig(os.path.splitext(os.path.abspath(file))[0] + '_ant_params.pdf', dpi=None, format='pdf', bbox_inches='tight', pad_inches=0.1)
plt.show() plt.show()