Reorganised and changed some of the naming of the tools scripts.

tools/plot_Bscan.py

@@ -0,0 +1,55 @@
+# 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 <http://www.gnu.org/licenses/>.
+
+import os, argparse
+import h5py
+import numpy as np
+import matplotlib.pyplot as plt
+
+from gprMax.exceptions import CmdInputError
+
+"""Plots B-scan."""
+
+# Parse command line arguments
+parser = argparse.ArgumentParser(description='Plots B-scan.', usage='cd gprMax; python -m tools.plot_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]
+
+# Check that there is more than one A-scan present
+if data.shape[1] == 1:
+    raise CmdInputError('{} contains only a single A-scan.'.format(file))
+
+# 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()