Newer
Older
import logging
import numpy as np
from time import sleep
from pymeasure.experiment import Procedure, IntegerParameter, FloatParameter
from ..classes.stage import Stage
from ..classes.dac import DAC
log = logging.getLogger(__name__)
log.addHandler(logging.NullHandler())
class Y_Sweep(Procedure):
name = "Y-Sweep"
acq_time = FloatParameter('Aquisition time', units='s', default=1)
freq = FloatParameter('Aquisition Frequency', units='Hz', default=50)
y_min = FloatParameter('From y', units='mm', default=0)
y_max = FloatParameter('To y', units='mm', default=5)
y_step = FloatParameter('Step', units='mm', default=0.1)
x = FloatParameter('Position x', units='mm', default=0)
field = FloatParameter('Magnetic field (max 6A)', units='V', default=0)
DATA_COLUMNS = ['Iteration', 'X Position', 'Y Position', 'Voltage', 'Magnetic Field [A]', 'Magnetic Field [mT]']
def startup(self):
log.info("Setting up the data aquisition")
def execute(self):
log.info("Starting the voltage acquisition")
voltage = DAC("ai0")
field = DAC("ao1")
y_positions = np.linspace(self.y_min, self.y_max, int((self.y_max-self.y_min)//self.y_step+1), endpoint=True)
field.write(self.field)
sleep(1)
try:
stage = Stage()
stage.enable_connections()
except:
log.info("Close connection with motors before running an experiment, motors can not be used")
stage.move_x_to(self.x)
stage.move_y_to(self.y_min)
stage.wait_stable()
# Start sweeping loop
for i in range(len(y_positions)):
stage.move_y_to(y_positions[i])
average_voltage = np.arange(0, int(round(self.acq_time * self.freq)), dtype='float64')
for step in range(len(average_voltage)):
average_voltage[step] = voltage.read()
sleep(1/self.freq)
data = {
'Y Position': y_positions[i],
'Voltage': np.mean(average_voltage)
}
log.debug("Produced numbers: %s" % data)
self.emit('results', data)
self.emit('progress', 100 * i / len(y_positions))
if self.should_stop():
break
field.write(0.)
stage.close_connections()
def shutdown(self):
log.info("Finished acquisition")