jono/selcal
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Fixed some broken scripts

Browse Source
This commit is contained in:
Jono Targett 2024-05-25 13:07:57 +09:30
parent 61b7b33a84
commit 6efc2a41eb
4 changed files with 13 additions and 13 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
scripts/selcald/analyze.py
View File

@ -51,8 +51,8 @@ def butter_bandpass_filter(data, lowcut, highcut, fs, order=5):
# tone synthesis # tone synthesis
def note(freq, cycles, amp=32767.0, rate=44100): def note(freq, cycles, amp=32767.0, rate=44100):
len = cycles * (1.0/rate) len = cycles * (1.0/rate)
t = np.linspace(0, len, len * rate) t = np.linspace(0, len, int(len * rate))
if freq is 0: if freq == 0:
data = np.zeros(int(len * rate)) data = np.zeros(int(len * rate))
else: else:
data = np.sin(2 * np.pi * freq * t) * amp data = np.sin(2 * np.pi * freq * t) * amp
@ -244,7 +244,7 @@ def analyze(file_name):
ax_S.axis('off') ax_S.axis('off')
fig.set_tight_layout(True) fig.set_tight_layout(True)
fig.show() plt.show()
if __name__ == "__main__": if __name__ == "__main__":

6
scripts/selcald/correlate.py
View File

@ -41,8 +41,8 @@ def voss(nrows, ncols=16):
# tone synthesis # tone synthesis
def note(freq, len, amp=1, rate=RATE): def note(freq, len, amp=1, rate=RATE):
t = np.linspace(0, len, len * rate) t = np.linspace(0, len, int(len * rate))
if freq is 0: if freq == 0:
data = np.zeros(int(len * rate)) data = np.zeros(int(len * rate))
else: else:
data = np.sin(2*np.pi*freq*t)*amp data = np.sin(2*np.pi*freq*t)*amp
@ -79,6 +79,6 @@ ax_corr.set_title('Cross-correlated with Reference')
ax_orig.margins(0, 0.1) ax_orig.margins(0, 0.1)
fig.set_tight_layout(True) fig.set_tight_layout(True)
fig.show() plt.show()
write('test.wav', 44100, sig_noise) write('test.wav', 44100, sig_noise)

10
scripts/selcald/match_frequency.py
View File

@ -12,8 +12,8 @@ RATE = 44100
# tone synthesis # tone synthesis
def tone(freq, cycles, amp=1, rate=RATE): def tone(freq, cycles, amp=1, rate=RATE):
len = cycles * (1.0/rate) len = cycles * (1.0/rate)
t = np.linspace(0, len, len * rate) t = np.linspace(0, len, int(len * rate))
if freq is 0: if freq == 0:
data = np.zeros(int(len * rate)) data = np.zeros(int(len * rate))
else: else:
data = np.sin(2 * np.pi * freq * t) * amp data = np.sin(2 * np.pi * freq * t) * amp
@ -37,11 +37,11 @@ for carrier in [312, 473, 716, 1084, 1479]:
ax.semilogy(freqs, response, label='tone = {}'.format(carrier)) ax.semilogy(freqs, response, label='tone = {}'.format(carrier))
ax.set_title('Matched filter response') ax.set_title('Matched filter response')
# ax.axvline(626.67, ls=':') # Guardband markers ax.axvline(626.67, ls=':') # Guardband markers
# ax.axvline(695.01, ls=':') ax.axvline(695.01, ls=':')
ax.legend(loc='best') ax.legend(loc='best')
ax.margins(0, 0.1) ax.margins(0, 0.1)
fig.set_tight_layout(True) fig.set_tight_layout(True)
fig.show() plt.show()

4
scripts/selcald/receiver.py
View File

@ -69,8 +69,8 @@ def butter_bandpass_filter(data, lowcut, highcut, fs, order=5):
# tone synthesis # tone synthesis
def note(freq, cycles, amp=32767.0, rate=44100): def note(freq, cycles, amp=32767.0, rate=44100):
len = cycles * (1.0/rate) len = cycles * (1.0/rate)
t = np.linspace(0, len, len * rate) t = np.linspace(0, len, int(len * rate))
if freq is 0: if freq == 0:
data = np.zeros(int(len * rate)) data = np.zeros(int(len * rate))
else: else:
data = np.sin(2 * np.pi * freq * t) * amp data = np.sin(2 * np.pi * freq * t) * amp