artiq/examples/master/repository/flopping_f_simulation.py

63 lines
1.6 KiB
Python
Raw Normal View History

2015-01-13 19:12:35 +08:00
from math import sqrt, cos, pi
import time
import random
import numpy as np
from scipy.optimize import curve_fit
2015-01-13 19:12:35 +08:00
from artiq import *
def model(x, F0):
t = 0.02
tpi = 0.03
A = 80
B = 40
2015-01-13 19:12:35 +08:00
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)))
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(Experiment, AutoDB):
"""Flopping F simulation"""
2015-01-13 19:12:35 +08:00
class DBKeys:
npoints = Argument(100)
min_freq = Argument(1000)
max_freq = Argument(2000)
F0 = Argument(1500)
noise_amplitude = Argument(0.1)
2015-01-13 19:12:35 +08:00
frequency = Result()
brightness = Result()
flopping_freq = Parameter()
realtime_results = {
("frequency", "brightness"): "xy"
}
2015-01-13 19:12:35 +08:00
def run(self):
for i in range(self.npoints):
frequency = (self.max_freq-self.min_freq)*i/(self.npoints - 1) + self.min_freq
brightness = model(frequency, self.F0) + self.noise_amplitude*random.random()
2015-01-13 19:12:35 +08:00
self.frequency.append(frequency)
self.brightness.append(brightness)
time.sleep(0.1)
2015-05-17 16:11:00 +08:00
self.scheduler.submit(self.scheduler.pipeline_name, self.scheduler.expid,
self.scheduler.priority, time.time() + 20)
2015-01-13 19:12:35 +08:00
def analyze(self):
popt, pcov = curve_fit(model_numpy,
self.frequency.read, self.brightness.read,
p0=[self.flopping_freq])
perr = np.sqrt(np.diag(pcov))
if perr < 0.1:
self.flopping_freq = float(popt)