From f098be0253742c3a78f03af0626e4f29e93e3971 Mon Sep 17 00:00:00 2001 From: Craig Warren Date: Wed, 27 Jan 2016 10:20:16 +0000 Subject: [PATCH] Better calculation of time array, using linspace instead of arange. --- tools/plot_Ascan.py | 3 ++- tools/plot_antenna_params.py | 4 ++-- tools/plot_builtin_wave.py | 3 ++- 3 files changed, 6 insertions(+), 4 deletions(-) diff --git a/tools/plot_Ascan.py b/tools/plot_Ascan.py index 17a424ea..5310f522 100644 --- a/tools/plot_Ascan.py +++ b/tools/plot_Ascan.py @@ -40,7 +40,8 @@ f = h5py.File(file, 'r') nrx = f.attrs['nrx'] dt = f.attrs['dt'] iterations = f.attrs['Iterations'] -time = np.arange(0, dt * iterations, dt) +time = np.linspace(0, 1, iterations) +time *= (iterations * dt) # Check for single output component when doing a FFT if args.fft: diff --git a/tools/plot_antenna_params.py b/tools/plot_antenna_params.py index 5b2f93f7..71ad92be 100644 --- a/tools/plot_antenna_params.py +++ b/tools/plot_antenna_params.py @@ -43,8 +43,8 @@ iterations = f.attrs['Iterations'] #iterations = int((1 / df) / dt) # Calculate time array and frequency bin spacing -time = np.arange(0, dt * iterations, dt) -time = time[0:iterations] +time = np.linspace(0, 1, iterations) +time *= (iterations * dt) df = 1 / np.amax(time) print('Time window: {:g} s ({} iterations)'.format(np.amax(time), iterations)) diff --git a/tools/plot_builtin_wave.py b/tools/plot_builtin_wave.py index c8cac493..8ec95172 100644 --- a/tools/plot_builtin_wave.py +++ b/tools/plot_builtin_wave.py @@ -62,7 +62,8 @@ else: timewindow = (int(args.timewindow) - 1) * dt iterations = int(args.timewindow) -time = np.arange(0, timewindow, dt) +time = np.linspace(0, 1, iterations) +time *= (iterations * dt) waveform = np.zeros(len(time)) timeiter = np.nditer(time, flags=['c_index'])