ericteunis/m-thesis-introduction
1
0
Fork 0
mirror of https://gitlab.science.ru.nl/mthesis-edeboone/m-thesis-introduction.git synced 2024-11-13 01:53:31 +01:00
Code Issues Projects Releases Packages Wiki Activity

Continuous DTFT: only use one peak for fitting

Browse source
This commit is contained in:
Eric Teunis de Boone 2022-11-14 15:38:32 +01:00
parent 25fb4444d0
commit feccf64293
1 changed files with 27 additions and 6 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

31
fourier/06_direct_fourier.py
View file

@ -1,4 +1,12 @@
#!/usr/bin/env python3 #!/usr/bin/env python3
# vim: fdm=indent ts=4
__doc__ = \
"""
Show how the fourier transform can be calculated
in a continuous fashion
"""
if __name__ == "__main__": if __name__ == "__main__":
import numpy as np import numpy as np
import matplotlib.pyplot as plt import matplotlib.pyplot as plt
@ -83,13 +91,15 @@ if __name__ == "__main__":
from numpy.polynomial import Polynomial as P from numpy.polynomial import Polynomial as P
freq_out = np.zeros(len(phi_in)) freq_out = np.zeros(len(phi_in))
amp_cut = 0.8 amp_cut = 0.5
fig, ax = plt.subplots() fig, ax = plt.subplots()
ax.set_title("Frequency estimation by parabola fitting.\nStars are used for the parabola fit, vertical line is where $\\partial_f = 0 $") ax.set_title("Frequency estimation by parabola fitting.\nStars are used for the parabola fit, vertical line is where $\\partial_f = 0 $")
ax.set_xlabel("Frequency") ax.set_xlabel("Frequency")
ax.set_ylabel("Amplitude") ax.set_ylabel("Amplitude")
ax.axvline(f_beacon, lw=5, ls=(0,(5,5)))
for j, amp in enumerate(amp_out): for j, amp in enumerate(amp_out):
if j > 2: if j > 2:
@ -98,9 +108,20 @@ if __name__ == "__main__":
max_amp_idx = np.argmax(amp) max_amp_idx = np.argmax(amp)
max_amp = amp[max_amp_idx] max_amp = amp[max_amp_idx]
# filter amplitudes below amp_cut*max_amp # filter amplitudes below amp_cut*max_amp
valid_idx = amp >= amp_cut*max_amp valid_mask = amp >= amp_cut*max_amp
p_fit = P.fit(test_freqs[valid_idx], amp[valid_idx], 2) if True:
# make sure not to use other peaks
lower_mask = valid_mask[0:max_amp_idx]
upper_mask = valid_mask[max_amp_idx:]
lower_end = np.argmin(lower_mask[::-1])
upper_end = np.argmin(upper_mask)
valid_mask[0:(max_amp_idx - lower_end)] = False
valid_mask[(max_amp_idx + upper_end):] = False
p_fit = P.fit(test_freqs[valid_mask], amp[valid_mask], 2)
func = p_fit.convert() func = p_fit.convert()
# Find frequency of derivative == 0 # Find frequency of derivative == 0
@ -110,7 +131,7 @@ if __name__ == "__main__":
l = ax.plot(test_freqs, amp, marker='.') l = ax.plot(test_freqs, amp, marker='.')
ax.plot(test_freqs[valid_idx], amp[valid_idx], marker='*', color=l[0].get_color()) ax.plot(test_freqs[valid_mask], amp[valid_mask], marker='*', color=l[0].get_color())
ax.axvline(freq_out[j], color=l[0].get_color()) ax.axvline(freq_out[j], color=l[0].get_color())
if True: # plot the fit if True: # plot the fit
@ -126,7 +147,7 @@ if __name__ == "__main__":
# Amplitudes figure # Amplitudes figure
if not True: if not True:
fig, ax = plt.subplots() fig, ax = plt.subplots()
ax.set_ylabels("Amplitude") ax.set_ylabel("Amplitude")
ax.set_xlabel("Frequency") ax.set_xlabel("Frequency")
if True: if True:
for j, amp in enumerate(amp_out.T): for j, amp in enumerate(amp_out.T):