forked from nanovna-v2/NanoVNA2-firmware
-
Notifications
You must be signed in to change notification settings - Fork 3
/
gen_sin.py
45 lines (34 loc) · 927 Bytes
/
gen_sin.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
#!/usr/bin/python
from math import *
from scipy import signal
# sine period in samples
N = 100
# integration period
accumPeriod = 200
# window decay length
windowN = 100
scale = (2**15 - 1)
if windowN > 0:
window = signal.gaussian(windowN, std=7)
#window = signal.bartlett(N)
rectWindow = signal.boxcar(windowN)
finalWindow = signal.convolve(window, rectWindow)
finalWindow.resize(len(finalWindow) + 1)
print len(finalWindow)
assert len(finalWindow) == accumPeriod
windowMax = 0
for x in finalWindow:
if abs(x) > windowMax:
windowMax = abs(x)
finalWindow /= windowMax
print 'const int16_t sinTable[%d] = {' % (accumPeriod*2)
for i in range(accumPeriod):
arg = float(i) / N * 2*pi
re = cos(arg)
im = -sin(arg)
vals = [re, im]
if windowN > 0:
vals = [x*finalWindow[i] for x in vals]
vals = [int(round(x*scale)) for x in vals]
vals = [str(x) for x in vals]
print vals[0] + ', ' + vals[1] + ','