# 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 . import os, argparse import h5py import numpy as np import matplotlib.pyplot as plt """Plots B-scan.""" # Parse command line arguments parser = argparse.ArgumentParser(description='Plots B-scan.', usage='cd gprMax; python -m tools.plot_hdf5_Bscan outputfile field') parser.add_argument('outputfile', help='name of output file including path') parser.add_argument('field', help='name of field to be plotted, i.e. Ex, Ey, Ez') args = parser.parse_args() file = args.outputfile field = args.field path = '/rxs/rx1' f = h5py.File(file, 'r') data = f[path + '/' + field] # Plot B-scan image fig = plt.figure(num=file, figsize=(20, 10), facecolor='w', edgecolor='w') plt.imshow(data, extent=[0, data.shape[1], data.shape[0]*f.attrs['dt'], 0], interpolation='nearest', aspect='auto', cmap='seismic', vmin=-np.amax(np.abs(data)), vmax=np.amax(np.abs(data))) plt.xlabel('Trace number') plt.ylabel('Time [s]') plt.grid() cb = plt.colorbar() cb.set_label('Field strength [V/m]') plt.show() #fig.savefig(os.path.splitext(os.path.abspath(file))[0] + '.pdf', dpi=None, format='pdf', bbox_inches='tight', pad_inches=0.1) f.close()