diff --git a/tools/plot_diffs.py b/tools/plot_diffs.py new file mode 100644 index 00000000..21934c7e --- /dev/null +++ b/tools/plot_diffs.py @@ -0,0 +1,56 @@ +# Copyright (C) 2015, Craig Warren +# +# This module is licensed under the Creative Commons Attribution-ShareAlike 4.0 International License. +# To view a copy of this license, visit http://creativecommons.org/licenses/by-sa/4.0/. +# +# Please use the attribution at http://dx.doi.org/10.1190/1.3548506 + +import os, argparse +import h5py +import numpy as np +np.seterr(divide='ignore') +import matplotlib.pyplot as plt + +"""Plots the differences (in dB) between a response and a reference response.""" + +# Parse command line arguments +parser = argparse.ArgumentParser(description='Plots the differences (in dB) between a response and a reference response.', usage='cd gprMax; python -m tools.plot_diffs refoutputfile outputfile') +parser.add_argument('refoutputfile', help='name of output file including path containing reference response') +parser.add_argument('outputfile', help='name of output file including path') +args = parser.parse_args() + +# Load (from gprMax output file) the reference response +f = h5py.File(args.refoutputfile, 'r') +tmp = f['/rxs/rx1/'] +fieldname = list(tmp.keys())[0] +refresp = np.array(tmp[fieldname]) + +# Load (from gprMax output file) the response +f = h5py.File(args.outputfile, 'r') +tmp = f['/rxs/rx1/'] +fieldname = list(tmp.keys())[0] +modelresp = np.array(tmp[fieldname]) + +# Calculate differences +diffdB = np.abs(modelresp - refresp) / np.amax(np.abs(refresp)) +diffdB = 20 * np.log10(diffdB) +print(np.abs(np.sum(diffdB[-np.isneginf(diffdB)])) / len(diffdB[-np.isneginf(diffdB)])) + +# Plot differences +fig, ax = plt.subplots(subplot_kw=dict(xlabel='Iterations', ylabel='Error [dB]'), num=args.outputfile, figsize=(20, 10), facecolor='w', edgecolor='w') +ax.plot(diffdB, 'r', lw=2) +ax.grid() +plt.show() + + + + + + + + + + + + + \ No newline at end of file