Use of np.isclose rather than equality when finding frequency of maximum power.

gprMax/grid.py

@@ -255,7 +255,7 @@ def dispersion_analysis(G):
                     # FFT
                     freqs, power = fft_power(waveformvalues, G.dt)
                     # Get frequency for max power
-                    freqmaxpower = np.where(power == 0)[0][0]
+                    freqmaxpower = np.where(np.isclose(power, 0))[0][0]
 
                     # Set maximum frequency to a threshold drop from maximum power, ignoring DC value
                     try:

tools/plot_Ascan.py

@@ -86,7 +86,7 @@ def mpl_plot(filename, outputs=Rx.defaultoutputs, fft=False):
             if fft:
                 # FFT
                 freqs, power = fft_power(outputdata, dt)
-                freqmaxpower = np.where(power == 0)[0][0]
+                freqmaxpower = np.where(np.isclose(power, 0))[0][0]
 
                 # Set plotting range to -60dB from maximum power or 4 times
                 # frequency at maximum power

tools/plot_source_wave.py

@@ -105,7 +105,7 @@ def mpl_plot(w, timewindow, dt, iterations, fft=False):
 
         # Set plotting range to 4 times frequency at max power of waveform or
         # 4 times the centre frequency
-        freqmaxpower = np.where(power == 0)[0][0]
+        freqmaxpower = np.where(np.isclose(power, 0))[0][0]
         if freqs[freqmaxpower] > w.freq:
             pltrange = np.where(freqs > 4 * freqs[freqmaxpower])[0][0]
         else: