First commit

tools/plot_hdf5_Ascan.py

@@ -0,0 +1,49 @@
+# 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
+from .plot_fields import plot_Ascan
+
+"""Plots electric and magnetic fields from all receiver points in the given output file. Each receiver point is plotted in a new figure window."""
+
+# Parse command line arguments
+parser = argparse.ArgumentParser(description='Plots electric and magnetic fields from all receiver points in the given output file. Each receiver point is plotted in a new figure window.', usage='cd gprMax; python -m tools.plot_hdf5_Ascan outputfile')
+parser.add_argument('outputfile', help='name of output file including path')
+args = parser.parse_args()
+
+file = args.outputfile
+f = h5py.File(file, 'r')
+nrx = f.attrs['nrx']
+time = np.arange(0, f.attrs['dt'] * f.attrs['Iterations'], f.attrs['dt'])
+time = time / 1E-9
+
+for rx in range(1, nrx + 1):
+    path = '/rxs/rx' + str(rx) + '/'
+    Ex = f[path + 'Ex'][:]
+    Ey = f[path + 'Ey'][:]
+    Ez = f[path + 'Ez'][:]
+    Hx = f[path + 'Hx'][:]
+    Hy = f[path + 'Hy'][:]
+    Hz = f[path + 'Hz'][:]
+    fig, plt = plot_Ascan('rx' + str(rx), time, Ex, Ey, Ez, Hx, Hy, Hz)
+    #fig.savefig(os.path.splitext(os.path.abspath(file))[0] + '.pdf', dpi=None, format='pdf', bbox_inches='tight', pad_inches=0.1)
+    plt.show()
+
+f.close()