notwa/dsp
1
0
Fork 0
Code Issues Pull requests Releases Wiki Activity

update 36

Browse source
This commit is contained in:
Connor Olding 2017-09-30 04:04:27 -07:00
parent 03663da1a1
commit 413b156e8f
1 changed files with 6 additions and 6 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

12
lib/sweeps.py
View file

@ -37,19 +37,19 @@ def tsp(N, m=0.5):
if N < 0: if N < 0:
raise Exception("The number of length must be the positive number") raise Exception("The number of length must be the positive number")
NN = 2**np.floor(np.log2(N)) # nearest NN = int(2**np.floor(np.log2(N))) # nearest
NN2 = NN//2 NN2 = NN // 2
M = np.round(NN2*m) M = int(np.round(NN2 * m))
nn2 = np.arange(NN2 + 1)**2 nn2 = np.square(np.arange(NN2 + 1))
j = np.complex(0, 1) j = np.complex(0, 1)
H = np.exp(j*4*M*np.pi*nn2/NN**2) H = np.exp(j * 4 * M * np.pi * nn2 / np.square(NN))
H2 = np.r_[H, np.conj(H[1:NN2][::-1])] H2 = np.r_[H, np.conj(H[1:NN2][::-1])]
x = np.fft.ifft(H2) x = np.fft.ifft(H2)
x = np.r_[x[NN2 - M:NN + 1], x[0:NN2 - M + 1]] x = np.r_[x[NN2 - M:NN + 1], x[0:NN2 - M + 1]]
x = np.r_[x.real, np.zeros(1, N - NN)] x = np.r_[x.real, np.zeros(N - NN)]
return x return x