From 6a03e08a25988ae0205804bc28a620bf3961b5db Mon Sep 17 00:00:00 2001
From: Craig Warren <craig_warren@me.com>
Date: Wed, 13 Jan 2016 15:08:54 +0000
Subject: [PATCH] Corrected bug with reading when using a single custom
 excitation.

---
 gprMax/input_cmds_singleuse.py | 5 ++++-
 1 file changed, 4 insertions(+), 1 deletion(-)

diff --git a/gprMax/input_cmds_singleuse.py b/gprMax/input_cmds_singleuse.py
index 5a388fe0..952e9167 100644
--- a/gprMax/input_cmds_singleuse.py
+++ b/gprMax/input_cmds_singleuse.py
@@ -248,7 +248,10 @@ def process_singlecmds(singlecmds, multicmds, G):
             w = Waveform()
             w.ID = waveformIDs[waveform]
             w.type = 'user'
-            w.uservalues = waveformvalues[:,waveform]
+            if len(waveformvalues.shape) == 1:
+                w.uservalues = waveformvalues[:]
+            else:
+                w.uservalues = waveformvalues[:,waveform]
         
             if G.messages:
                 print('User waveform {} created.'.format(w.ID))