test_gen_data.py
79 lines
| 2.1 KiB
| text/x-python
|
PythonLexer
pellion
|
r544 | import math as m | ||
import numpy as np | ||||
def int2hex(n,nbits): | ||||
if (nbits % 4) != 0 : | ||||
return 'ERROR 1!' | ||||
spec='0'+str(nbits/4)+'x' | ||||
if n >= (-2**(nbits-1)) and n <= (2**(nbits-1)-1) : | ||||
return format(n, spec) if n>=0 else format(2**nbits+n, spec) | ||||
else : | ||||
return 'ERROR 2!' | ||||
nb_point = 256 | ||||
t = np.arange(nb_point) | ||||
## f0 | ||||
ampl_f0_0 = pow(2,14) | ||||
freq_f0_0 = float(16)/256 | ||||
phi_f0_0 = 0 | ||||
ampl_f0_1 = pow(2,13) | ||||
freq_f0_1 = float(16)/256 | ||||
phi_f0_1 = 0 | ||||
ampl_f0_2 = pow(2,12) | ||||
freq_f0_2 = float(16)/256 | ||||
phi_f0_2 = 0 | ||||
ampl_f0_3 = pow(2,11) | ||||
freq_f0_3 = float(16)/256 | ||||
phi_f0_3 = 0 | ||||
ampl_f0_4 = pow(2,10) | ||||
freq_f0_4 = float(16)/256 | ||||
phi_f0_4 = 0 | ||||
x_f0 = [ampl_f0_0 * np.cos(2 * m.pi * freq_f0_0 * t + phi_f0_0 * m.pi / 180 ) , | ||||
ampl_f0_1 * np.cos(2 * m.pi * freq_f0_1 * t + phi_f0_1 * m.pi / 180 ) , | ||||
ampl_f0_2 * np.cos(2 * m.pi * freq_f0_2 * t + phi_f0_2 * m.pi / 180 ) , | ||||
ampl_f0_3 * np.cos(2 * m.pi * freq_f0_3 * t + phi_f0_3 * m.pi / 180 ) , | ||||
ampl_f0_4 * np.cos(2 * m.pi * freq_f0_4 * t + phi_f0_4 * m.pi / 180 ) ] | ||||
# x_f0 = [ampl_f0_0 * np.cos(2 * m.pi * freq_f0_0 * t + phi_f0_0 * m.pi / 180 ) , | ||||
# np.zeros(nb_point,dtype=np.int16) + 10 , | ||||
# np.zeros(nb_point,dtype=np.int16) - 10 , | ||||
# ampl_f0_3 * np.cos(2 * m.pi * freq_f0_3 * t + phi_f0_3 * m.pi / 180 ) , | ||||
# ampl_f0_4 * np.cos(2 * m.pi * freq_f0_4 * t + phi_f0_4 * m.pi / 180 ) ] | ||||
x_f0_int16 = [np.zeros(nb_point,dtype=np.int16), | ||||
np.zeros(nb_point,dtype=np.int16), | ||||
np.zeros(nb_point,dtype=np.int16), | ||||
np.zeros(nb_point,dtype=np.int16), | ||||
np.zeros(nb_point,dtype=np.int16)] | ||||
for j in xrange(5) : | ||||
for i in xrange(nb_point) : | ||||
x_f0_int16[j][i] = int(round(x_f0[j][i])) | ||||
f = open("data_f0.txt", 'w') | ||||
for i in xrange(nb_point) : | ||||
for j in xrange(5) : | ||||
f.write(int2hex(x_f0_int16[j][i],16)) | ||||
f.write('\n') | ||||
f.close | ||||
f = open("data_f1.txt", 'w') | ||||
for i in xrange(1) : | ||||
for j in xrange(5) : | ||||
f.write(int2hex(0,16)) | ||||
f.close | ||||
f = open("data_f2.txt", 'w') | ||||
for i in xrange(1) : | ||||
for j in xrange(5) : | ||||
f.write(int2hex(0,16)) | ||||
f.close | ||||