M-Labs
/
artiq
mirror of https://github.com/m-labs/artiq.git
2
0
Fork 2
Code Issues Projects Releases Wiki Activity

flopping_f_simulation: use ufuncs in model()

This commit is contained in:
Robert Jördens 2016-05-12 18:30:52 +02:00
parent 5d58258bf2
commit 2a5eaea411
1 changed files with 4 additions and 10 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

14
artiq/examples/master/repository/flopping_f_simulation.py
View File

@ -13,14 +13,9 @@ def model(x, F0):
tpi = 0.03 tpi = 0.03
A = 80 A = 80
B = 40 B = 40
return A+(B-A)/2/(4*tpi**2*(x-F0)**2+1)*(1-cos(pi*t/tpi*sqrt(4*tpi**2*(x-F0)**2+1))) return A + (B - A)/2/(4*tpi**2*(x - F0)**2+1)*(
1 - np.cos(np.pi*t/tpi*np.sqrt(4*tpi**2*(x - F0)**2 + 1))
)
def model_numpy(xdata, F0):
r = np.zeros(len(xdata))
for i, x in enumerate(xdata):
r[i] = model(x, F0)
return r
class FloppingF(EnvExperiment): class FloppingF(EnvExperiment):
@ -68,8 +63,7 @@ class FloppingF(EnvExperiment):
assert frequency.shape == brightness.shape assert frequency.shape == brightness.shape
self.set_dataset("flopping_f_frequency", frequency, self.set_dataset("flopping_f_frequency", frequency,
broadcast=True, save=False) broadcast=True, save=False)
popt, pcov = curve_fit(model_numpy, popt, pcov = curve_fit(model, frequency, brightness,
frequency, brightness,
p0=[self.get_dataset("flopping_freq", 1500.0)]) p0=[self.get_dataset("flopping_freq", 1500.0)])
perr = np.sqrt(np.diag(pcov)) perr = np.sqrt(np.diag(pcov))
if perr < 0.1: if perr < 0.1: