def test_series_measurement_one_in_queue(self, mock_waveform_thread,
log_checker):
# setup
handler = MeasurementHandler()
print(handler._queue)
handler.add_to_queue(self.lp.create_waveform(), self.settings)
side_effect_array = [
np.array([[0, 2.5, 4.5, 6.5], [1, 3.5, 5.5, 7.5]])
]
with patch.object(handler, 'single_measurement',
side_effect=side_effect_array) as method:
# perform
observed_data_list = handler.series_measurement()
# assert
for observed_data in observed_data_list:
np.testing.assert_array_equal(
observed_data,
np.array([[0, 2.5, 4.5, 6.5], [1, 3.5, 5.5, 7.5]])
)
log_checker.check(
('root', 'INFO', 'Measurement #1 complete'),
('root', 'INFO', 'Total: 1 measurements performed'),
)
def __init__(self, app, view1):
"""
:params app: a wx App object
:params view1: a wx Frame object
"""
super(Controller, self).__init__()
self.view1 = view1
self.app = app
# make directory for storing data files
self.app_dir = os.getcwd()
data_dir = os.path.join(self.app_dir, "data_directory")
if not os.path.exists(data_dir):
os.makedirs(data_dir)
self.data_dir = data_dir
self._set_event_bindings()
self.view1.Show()
# app state
self.uploaded = False
# the hardware interface
self.measurement_handler = MeasurementHandler()
# settings
self.preamp_settings = None
self.temperature_settings = None
self.wafer_settings = None
# data sets
self.pc_calibration_data = PCCalibrationData()
def test__run_thread(self, mock_waveform_thread):
mock_waveform_thread.return_value.time = 3
mock_waveform_thread.return_value.Read_Data = np.array([])
handler = MeasurementHandler()
handler.add_to_queue(self.lp.create_waveform(), self.settings)
def test_series_measurement_empty_queue(self, mock_waveform_thread,
log_checker):
handler = MeasurementHandler()
observed_data_list = handler.series_measurement()
self.assertEqual(
len(observed_data_list), 0)
log_checker.check(
('root', 'INFO', 'Total: 0 measurements performed'),
)
def test_single_measurement_no_averaging(self, mock_waveform_thread):
handler = MeasurementHandler()
handler.add_to_queue(self.lp.create_waveform(), self.settings)
return_tuple = (np.array([1, 2, 3, 4, 5, 6]), np.array([0, 1]))
with patch.object(handler, '_run_thread',
return_value=return_tuple) as method:
test_dataset = handler.single_measurement()
self.assertEqual(1, method.call_count)
np.testing.assert_array_equal(
test_dataset,
np.array([[0, 1, 3, 5], [1, 2, 4, 6]])
)
self.assertEqual(len(handler._queue), 0)
def __init__(self, app, view1):
"""
:params app: a wx App object
:params view1: a wx Frame object
"""
super(Controller, self).__init__()
self.view1 = view1
self.app = app
# make directory for storing data files
self.app_dir = os.getcwd()
data_dir = os.path.join(self.app_dir, "data_directory")
if not os.path.exists(data_dir):
os.makedirs(data_dir)
self.data_dir = data_dir
self._set_event_bindings()
self.view1.Show()
# app state
self.uploaded = False
# the hardware interface
self.measurement_handler = MeasurementHandler()
# settings
self.temperature_settings = None
self.wafer_settings = None
# data sets
self.pc_calibration_data = PCCalibrationData()
def onPCCalibration(self, event):
print('onPCCalibration')
self.view.ShowCalibrationModal()
null_metadata = copy.deepcopy(self.view.metadata)
null_metadata.waveform = "NullWave"
null_metadata.averaging = 1
null_pulse = LightPulse(null_metadata)
# Using that instance we then run the lights,
# and measure the outputs
measurement_handler = MeasurementHandler()
measurement_handler.add_to_queue(
null_pulse.complete_waveform,
null_metadata
)
calibration_data = measurement_handler.single_measurement()
self.view.metadata.pc_calibration.update_data(calibration_data)
# We then plot the datas, this has to be changed if the plots want
# to be updated on the fly.
pub.sendMessage('calibration.pc')
def test_single_measurement_with_averaging(self, mock_waveform_thread):
# setup
settings = ExperimentSettings()
settings.averaging = 2
handler = MeasurementHandler()
return_tuple = [
(np.array([2, 3, 4, 5, 6, 7]), np.array([0, 1])),
(np.array([3, 4, 5, 6, 7, 8]), np.array([0, 1])),
]
handler.add_to_queue(self.lp.create_waveform(), settings)
with patch.object(handler, '_run_thread', side_effect=return_tuple) as method:
# perform
test_dataset = handler.single_measurement()
# assert
self.assertEqual(2, method.call_count)
np.testing.assert_array_equal(
test_dataset,
np.array([[0, 2.5, 4.5, 6.5], [1, 3.5, 5.5, 7.5]])
)
self.assertEqual(len(handler._queue), 0)
def Perform_Measurement(self, event):
print("GUIController: Perform_Measurement")
# Using that instance we then run the lights,
# and measure the outputs
self.measurement_handler = MeasurementHandler()
measurement_handler.add_to_queue(
self.light_pulse.complete_waveform,
self.metadata
)
raw_data = self.measurement_handler.single_measurement()
self.Data.updateRawData(raw_data)
self.Data.Data = utils.bin_data(raw_data, self.metadata.binning)
# We then plot the datas, this has to be changed if the plots want
# to be updated on the fly.
pub.sendMessage('update.plot')
class Controller(object):
"""A Controller to coordinate the UI, data, and hardware"""
def __init__(self, app, view1):
"""
:params app: a wx App object
:params view1: a wx Frame object
"""
super(Controller, self).__init__()
self.view1 = view1
self.app = app
# make directory for storing data files
self.app_dir = os.getcwd()
data_dir = os.path.join(self.app_dir, "data_directory")
if not os.path.exists(data_dir):
os.makedirs(data_dir)
self.data_dir = data_dir
self._set_event_bindings()
self.view1.Show()
# app state
self.uploaded = False
# the hardware interface
self.measurement_handler = MeasurementHandler()
# settings
self.preamp_settings = None
self.temperature_settings = None
self.wafer_settings = None
# data sets
self.pc_calibration_data = PCCalibrationData()
def data_output_dir(self, event):
"""
Select a different directory to save experimental results
"""
self.data_dir = self.view1.ask_user_for_dir(
message="Choose a directory",
defaultPath=self.data_dir,
)
def save_settings(self, event):
"""
Saves settings as displayed in the forms
"""
settings = {}
settings["temperature_settings"] = self.view1.get_temperature_form(
allow_incomplete=True)
settings["experiment_settings"] = self.view1.get_experiment_form(
allow_incomplete=True)
file_dir, file_name = self.view1.ask_user_for_filename(
defaultDir=self.app_dir,
message='Save your file',
wildcard='*.*',
style=wx.SAVE)
if file_dir is not None:
save_metadata(settings, file_name, file_dir)
def load_settings(self, event):
"""
Loads settings from the forms
"""
# TODO: change so just fills form out
# and then reads into data structure when press perform measurement
self.view1.clear_experiment_form()
self.measurement_handler.clear_queue()
file_dir, file_name = self.view1.ask_user_for_filename(
defaultDir=self.app_dir,
message='Choose a file',
wildcard='*.*',
style=wx.OPEN)
if file_dir is not None:
# try:
config_dict = load_metadata(file_name, file_dir)
settings = self._parse_config(config_dict)
for setting in settings["experiment_settings"]:
self.measurement_handler.add_to_queue(
LightPulse(setting).create_waveform(), setting)
self.temperature_settings = settings["temp_settings"]
# except Exception as e:
# print("An Exception occurred: {0}".format(e))
self.view1.set_experiment_form(self.measurement_handler.as_list())
self.view1.set_temperature_form(self.temperature_settings.as_dict())
self.measurement_handler.clear_queue()
def upload(self, event):
"""
Bulk upload settings from a text file
"""
self.measurement_handler.clear_queue()
file_dir, file_name = self.view1.ask_user_for_filename(
defaultDir=self.app_dir,
style=wx.OPEN,
message='Choose a file',
wildcard='*.*')
if file_dir is not None:
config_dict = load_metadata(file_name, file_dir)
settings = self._parse_config(config_dict)
for setting in settings["experiment_settings"]:
self.measurement_handler.add_to_queue(
LightPulse(setting).create_waveform(), setting)
self.temperature_settings = settings["temp_settings"]
self.view1.set_temperature_form(
self.temperature_settings.as_dict())
self.view1.disable_all_settings_inputs()
self.view1.show_info_modal(
"{0} experiment settings uploaded successfully!".format(
len(self.measurement_handler._queue)))
self.uploaded = True
def display(self, event):
"""
Plot waveforms in modal
"""
self.PlotModal = PlotModal(self.app)
waveform_list = []
settings_dict = {}
try:
settings_dict[
"experiment_settings"] = self.view1.get_experiment_form()
experiment_settings = self._parse_experiment_settings(
settings_dict)
for setting in experiment_settings:
waveform_list.append(-LightPulse(setting).create_waveform())
self.PlotModal.Show()
self.PlotModal.plot_data(waveform_list)
except PVInputError as e:
self.view1.show_error_modal(str(e))
def perform_measurement(self, event):
"""
Perform the queued measurements
"""
try:
if not self.uploaded:
# try to retrieve settings from the various forms
config_dict = {}
config_dict["temperature_settings"] = (
self.view1.get_temperature_form())
config_dict["wafer_settings"] = self.view1.get_wafer_form()
# gets the measurement settings
config_dict["experiment_settings"] = (
self.view1.get_experiment_form())
try:
settings = self._parse_config(config_dict)
except KeyError as e:
self.view1.show_error_modal(str(e))
# create the wave forms for each run
for setting in settings["experiment_settings"]:
self.measurement_handler.add_to_queue(setting)
self.wafer_settings = settings["wafer_settings"]
self.temperature_settings = settings["temp_settings"]
if self.data_dir is not None:
save_metadata(self.wafer_settings.as_dict(), self.data_dir,
self.wafer_settings.id)
if self.measurement_handler.is_queue_empty():
raise (PVInputError("No measurements loaded."))
# Do the actual measurements
# TODO: refactor in separate method
dataset_list = []
total_measurements = 0
self.PlotModal = PlotModal(self.app)
self.PlotModal.Show()
# while there things to measure
while not self.measurement_handler.is_queue_empty():
single_dataset = self.measurement_handler.single_measurement()
dataset_list.append(single_dataset)
total_measurements = total_measurements + 1
ts = int(time.time())
dataset_name = (str(total_measurements) +
self.wafer_settings.id + str(ts))
save_data(single_dataset, dataset_name, self.data_dir)
# print single_dataset.shape
self.PlotModal.plot_data(single_dataset[:, 0],
[x for x in single_dataset[:, 1:].T])
except PVInputError as e:
self.view1.show_error_modal(str(e))
def calibrate_pc(self, event):
""""
Perform and calibrate the PC measurement
"""
self.view1.show_calibration_modal()
null_metadata = ExperimentSettings()
null_metadata.waveform = "NullWave"
pc_data = self.measurement_handler.pc_calibration_measurement(
null_metadata)
self.pc_calibration.update_data(pc_data)
def show_calibration_const(self, event):
"""
Display the calibration constants
"""
message = str(self.pc_calibration_data.as_dict())
self.view1.show_info_modal(message)
def _set_event_bindings(self):
"""
Setups all program behavior which isn't strictly related to the UI
"""
self.view1.m_dataOutputDir.Bind(wx.EVT_BUTTON, self.data_output_dir)
self.view1.m_save.Bind(wx.EVT_BUTTON, self.save_settings)
self.view1.m_load.Bind(wx.EVT_BUTTON, self.load_settings)
self.view1.m_upload.Bind(wx.EVT_BUTTON, self.upload)
self.view1.m_display.Bind(wx.EVT_BUTTON, self.display)
self.view1.m_performMeasurement.Bind(wx.EVT_BUTTON,
self.perform_measurement)
self.view1.m_calibratePC.Bind(wx.EVT_BUTTON, self.calibrate_pc)
self.view1.m_showCalibrationConst.Bind(wx.EVT_BUTTON,
self.show_calibration_const)
def _parse_experiment_settings(self, config):
measurement_list = []
experiments_list = config["experiment_settings"]
for item in experiments_list:
try:
measurement_list.append(
ExperimentSettings(
waveform=item['waveform'],
duration=item['duration'],
amplitude=item['amplitude'],
offset_before=item['offset_before'],
offset_after=item['offset_after'],
sample_rate=item['sample_rate'],
channel=item['channel'],
binning=item['binning'],
averaging=item['averaging'],
ref_gain=item['ref_gain'],
pl_gain=item['pl_gain'],
))
except KeyError as e:
raise ("Missing value in Experiment Settings: {0}".format(e))
return measurement_list
def _parse_config(self, config):
"""
Takes a dictionary of settings and constructs appropriate objects
"""
settings = {}
settings["experiment_settings"] = self._parse_experiment_settings(
config)
try:
settings["temp_settings"] = TemperatureSettings(
start_temp=config["temperature_settings"]["start_temp"],
end_temp=config["temperature_settings"]["end_temp"],
step_temp=config["temperature_settings"]["step_temp"],
step_wait=config["temperature_settings"]["step_wait"],
temperature_scale=config["temperature_settings"]
["temperature_scale"])
except KeyError as e:
raise KeyError(
"Missing value in Temperature Settings: {0}".format(e))
try:
settings["wafer_settings"] = Wafer(
wafer_id=config["wafer_settings"]["wafer_id"],
thickness=config["wafer_settings"]["wafer_thickness"],
na=config["wafer_settings"]["wafer_na"],
nd=config["wafer_settings"]["wafer_nd"],
diffused=config["wafer_settings"]["wafer_diffused"],
num_sides=config["wafer_settings"]["wafer_num_sides"])
except KeyError as e:
raise KeyError("Missing value in Wafer settings: {0}".format(e))
return settings
class GUIController(FrameSkeleton):
"""
Controller to handle interface with wx UI
Attributes
----------
MyFrame1:
Fig1
Data
LoadPath
Path
measurement_type
"""
def __init__(self, parent):
# models
self.metadata = ExperimentSettings()
self.Data = ExperimentData(metadata=self.metadata)
self.light_pulse = LightPulse(self.metadata)
# setup file data
self.dirname = os.getcwd()
self.data_file = "untitled.dat"
self.metadata_file = "untitled.inf"
self.__InitUI(parent)
self.__InitValidators()
def __InitUI(self, parent):
# initialize parent class
FrameSkeleton.__init__(self, parent)
# setup Matplotlib Canvas panel
self.Fig1 = CanvasPanel(self.Figure1_Panel)
# make status bars
m_statusBar = wx.StatusBar(self)
self.SetStatusBar(m_statusBar)
# self.m_statusBar.SetStatusText("Ready to go!")
self.data_panel = DataProcessingPanel(self.m_notebook1)
self.m_notebook1.AddPage(
self.data_panel,
u"Data Processing",
# True
)
# Setup the Menu
menu_bar = wx.MenuBar()
# File Menu
filem = wx.Menu()
filem.Append(wx.ID_OPEN, "Open\tCtrl+O")
filem.Append(wx.ID_ANY, "Run\tCtrl+R")
filem.Append(wx.ID_ABOUT, "About")
filem.Append(wx.ID_EXIT, "Exit\tCtrl+X")
# TODO: add and bind event handlers
menu_bar.Append(filem, "&File")
self.SetMenuBar(menu_bar)
# initialise view with model parameters
self.m_voltageRange.AppendItems(INPUT_VOLTAGE_RANGE_STR)
self.m_Waveform.AppendItems(WAVEFORMS)
self.m_Output.AppendItems(OUTPUTS)
self.setWaveformParameters()
self.setCollectionParameters()
def __InitValidators(self):
# Waveform validators
self.m_Intensity.SetValidator(NumRangeValidator(numeric_type='int'))
self.m_Threshold.SetValidator(NumRangeValidator(numeric_type='int'))
self.m_Period.SetValidator(NumRangeValidator(numeric_type='float'))
self.m_Offset_Before.SetValidator(NumRangeValidator(numeric_type='float'))
self.m_Offset_After.SetValidator(NumRangeValidator(numeric_type='float'))
# Data collection validators
self.m_samplingFreq.SetValidator(NumRangeValidator(numeric_type='float'))
self.m_Averaging.SetValidator(NumRangeValidator(numeric_type='int'))
self.m_Binning.SetValidator(NumRangeValidator(numeric_type='int'))
def setPCCalibrationMean(self):
self.m_pcCalibrationMean.SetValue('{0:3.3f}'.format(
self.metadata.pc_calibration_mean
))
def setPCCalibrationStd(self):
self.m_pcCalibrationStd.SetValue('{0:3.3f}'.format(
self.metadata.pc_calibration_std
))
def setSampleDataPoints(self, sample_data_points):
self.m_DataPoint.SetValue('{0:.2e}'.format(sample_data_points))
def setFrequency(self, frequence_val):
self.m_Frequency.SetValue('{0:3.3f}'.format(frequence_val))
def setWaveformParameters(self):
self.m_Intensity.SetValue(str(self.metadata.amplitude))
self.m_Period.SetValue(str(self.metadata.duration))
self.m_Offset_Before.SetValue(str(self.metadata.offset_before))
self.m_Offset_After.SetValue(str(self.metadata.offset_after))
self.setFrequency(self.metadata.get_frequency())
def setCollectionParameters(self):
self.m_Binning.SetValue(str(self.metadata.binning))
self.m_Averaging.SetValue(str(self.metadata.averaging))
self.m_Threshold.SetValue(str(self.metadata.threshold))
self.m_samplingFreq.SetValue(str(self.metadata.sample_rate))
self.setSampleDataPoints(self.metadata.get_total_data_points())
#################################
# Event Handlers for Measurements
def Perform_Measurement(self, event):
print("GUIController: Perform_Measurement")
# Using that instance we then run the lights,
# and measure the outputs
self.measurement_handler = MeasurementHandler()
measurement_handler.add_to_queue(
self.light_pulse.complete_waveform,
self.metadata
)
raw_data = self.measurement_handler.single_measurement()
self.Data.updateRawData(raw_data)
self.Data.Data = utils.bin_data(raw_data, self.metadata.binning)
# We then plot the datas, this has to be changed if the plots want
# to be updated on the fly.
pub.sendMessage('update.plot')
def fftHandler(self):
channel = self.data_panel.m_fftChoice.GetStringSelection()
freq_data = self.Data.fftOperator(channel, self.metadata.get_total_time())
self.PlotData(freq_data, title=['FFT of Raw data', 'Frequency (hz)', 'Voltage (V)'])
def onWaveformParameters(self, event):
self.metadata.A = float(self.m_Intensity.GetValue())
self.metadata.Duration = float(self.m_Period.GetValue())
self.metadata.Offset_Before = float(self.m_Offset_Before.GetValue())
self.metadata.Offset_After = float(self.m_Offset_After.GetValue())
self.metadata.Waveform = self.m_Waveform.GetStringSelection()
pub.sendMessage('waveform.change')
def onCollectionParameters(self, event):
# TODO: refactor so it obeys DRY
self.metadata.binning = int(self.m_Binning.GetValue())
self.metadata.averaging = int(self.m_Averaging.GetValue())
self.metadata.channel = self.m_Output.GetStringSelection()
self.metadata.threshold = float(self.m_Threshold.GetValue())
self.metadata.sample_rate = float(self.m_samplingFreq.GetValue())
self.metadata.sample_data_points = self.metadata.get_total_data_points()
pub.sendMessage('collection.change')
##########################
# Plot Handlers
#
def plotHandler(self):
self.PlotData(self.Data.Data)
def PlotData(self, data, data_labels=['Reference', 'PC', 'PL'],
title=['Raw Data', 'Time (s)', 'Voltage (V)'], e=None):
self.Fig1.clear()
labels = data_labels
colours = ['b', 'r', 'g']
# this is done not to clog up the plot with many points
if data.shape[0] > 1000:
num = data.shape[0] // 1000
else:
num = 1
# This plots the figure
for i, label, colour in zip(data[:, 1:].T, labels, colours):
self.Fig1.draw_points(
data[::num, 0],
i[::num],
'.',
Color=colour,
Label=label
)
self.Fig1.legend()
self.Fig1.labels(title[0], title[1], title[2])
self.Fig1.update()
if e is not None:
e.skip()
#################
# Helper methods:
def defaultFileDialogOptions(self):
"""
Return a dictionary with file dialog options that can be
used in both the save file dialog as well as in the open
file dialog.
"""
return dict(message='Choose a file', defaultDir=self.dirname, wildcard='*.*')
def askUserForFilename(self, **dialogOptions):
dialog = wx.FileDialog(self, **dialogOptions)
if dialog.ShowModal() == wx.ID_OK:
userProvidedFilename = True
self.data_file = dialog.GetFilename()
self.dirname = dialog.GetDirectory()
else:
userProvidedFilename = False
dialog.Destroy()
return userProvidedFilename
def ShowCalibrationModal(self):
msg_text = (
"Please remove sample from measurement area\n"
"Only one PC calibration is necessary per experimental session"
)
wx.MessageBox(msg_text, 'Info', wx.OK | wx.ICON_INFORMATION)
##########################
# App state Event Handlers
#
def onSave(self, event):
"""
Method to handle dialogue window and saving data to file
"""
dialog = wx.FileDialog(
None,
'Save measurement data and metadata',
self.dirname,
'',
r'DAT and INF files (*.dat;*.inf)|*.inf;*.dat',
wx.FD_SAVE
)
if dialog.ShowModal() == wx.ID_OK:
dialog_path = dialog.GetPath()
self.dirname = os.path.dirname(dialog_path)
self.SaveName = os.path.splitext(os.path.basename(dialog_path))[0]
experiment_settings = self.metadata.get_settings_as_dict()
waveform_settings = self.light_pulse.get_settings_as_dict()
metadata_dict = experiment_settings.copy()
utils.save_data(self.Data.Data, self.SaveName, self.dirname)
utils.save_metadata(
metadata_dict,
self.SaveName,
self.dirname
)
else:
print('Canceled save')
dialog.Destroy()
event.Skip()
def onSaveData(self, event):
print("onSaveData")
if self.askUserForFilename(style=wx.SAVE,
**self.defaultFileDialogOptions()):
utils.save_data(self.Data.Data, self.data_file, self.dirname)
# print(filename)
# fullpath = os.path.join(self.dirname, self.data_file)
# self.Data.updateRawData(utils.load_data(fullpath))
def onLoad(self, event):
"""
Method to handle load metadata dialog window and update metadata state
"""
dialog = wx.FileDialog(
None,
'Select a metadata file',
self.dirname,
'',
r'*.inf',
wx.FD_OPEN
)
if dialog.ShowModal() == wx.ID_OK:
metadata_dict = utils.load_metadata(dialog.GetPath())
metadata_stringified = dict(
[a, str(x)] for a, x in metadata_dict.iteritems()
)
print(metadata_stringified)
# experimental data
self.m_Output.SetStringSelection(metadata_stringified[u'Channel'])
self.m_Averaging.SetValue(metadata_stringified[u'Averaging'])
try:
self.m_Binning.SetValue(metadata_stringified[u'Measurement_Binning'])
except:
self.m_Binning.SetValue(metadata_stringified[u'Binning'])
self.m_Threshold.SetValue(metadata_stringified[u'Threshold_mA'])
# waveform data
self.m_Intensity.SetValue(metadata_stringified[u'Intensity_v'])
self.m_Waveform.SetStringSelection(metadata_stringified[u'Waveform'])
self.m_Offset_Before.SetValue(metadata_stringified[u'Offset_Before_ms'])
self.m_Offset_After.SetValue(metadata_stringified[u'Offset_After_ms'])
self.m_Period.SetValue(metadata_stringified[u'Peroid_s'])
dialog.Destroy()
event.Skip()
def onLoadData(self, event):
"""
Handlers loading of new data set into Frame
"""
print("start data loading")
if self.askUserForFilename(style=wx.OPEN,
**self.defaultFileDialogOptions()):
fullpath = os.path.join(self.dirname, self.data_file)
self.Data.updateRawData(utils.load_data(fullpath))
print(self.Data.Data)
pub.sendMessage('update.plot')
self.onWaveformParameters(self, event)
self.onCollectionParameters(self, event)
def onExit(self, event):
self.Close()
def OnAbout(self, event):
dialog = wx.MessageDialog(
self,
'An PV experimental assistant in wxPython',
'About PVapp',
wx.OK
)
dialog.ShowModal()
dialog.Destroy()
def onStatusUpdate(self, status, is_error=False):
self.SetStatusText(status, 0)
if is_error:
self.m_statusBar.SetBackgroundColour('RED')
self.m_statusBar.SetStatusText(status)
else:
self.m_statusBar.SetStatusText(status)
def test_add_to_queue(self, mock_waveform_thread):
handler = MeasurementHandler()
handler.add_to_queue(self.lp.create_waveform(), self.settings)
self.assertEqual(len(handler._queue), 1)
class Controller(object):
"""A Controller to coordinate the UI, data, and hardware"""
def __init__(self, app, view1):
"""
:params app: a wx App object
:params view1: a wx Frame object
"""
super(Controller, self).__init__()
self.view1 = view1
self.app = app
# make directory for storing data files
self.app_dir = os.getcwd()
data_dir = os.path.join(self.app_dir, "data_directory")
if not os.path.exists(data_dir):
os.makedirs(data_dir)
self.data_dir = data_dir
self._set_event_bindings()
self.view1.Show()
# app state
self.uploaded = False
# the hardware interface
self.measurement_handler = MeasurementHandler()
# settings
self.temperature_settings = None
self.wafer_settings = None
# data sets
self.pc_calibration_data = PCCalibrationData()
def data_output_dir(self, event):
"""
Select a different directory to save experimental results
"""
self.data_dir = self.view1.ask_user_for_dir(
message="Choose a directory",
defaultPath=self.data_dir,
)
def save_settings(self, event):
"""
Saves settings as displayed in the forms
"""
settings = {}
settings["temperature_settings"] = self.view1.get_temperature_form(
allow_incomplete=True
)
settings["experiment_settings"] = self.view1.get_experiment_form(
allow_incomplete=True
)
file_dir, file_name = self.view1.ask_user_for_filename(
defaultDir=self.app_dir,
message='Save your file',
wildcard='*.*',
style=wx.SAVE
)
if file_dir is not None:
save_metadata(settings, file_name, file_dir)
def load_settings(self, event):
"""
Loads settings from the forms
"""
# TODO: change so just fills form out
# and then reads into data structure when press perform measurement
self.view1.clear_experiment_form()
self.measurement_handler.clear_queue()
file_dir, file_name = self.view1.ask_user_for_filename(
defaultDir=self.app_dir,
message='Choose a file',
wildcard='*.*',
style=wx.OPEN
)
if file_dir is not None:
# try:
config_dict = load_metadata(file_name, file_dir)
settings = self._parse_config(config_dict)
for setting in settings["experiment_settings"]:
self.measurement_handler.add_to_queue(
LightPulse(setting).create_waveform(),
setting
)
self.temperature_settings = settings["temp_settings"]
# except Exception as e:
# print("An Exception occurred: {0}".format(e))
self.view1.set_experiment_form(
self.measurement_handler.as_list()
)
self.view1.set_temperature_form(
self.temperature_settings.as_dict()
)
self.measurement_handler.clear_queue()
def upload(self, event):
"""
Bulk upload settings from a text file
"""
self.measurement_handler.clear_queue()
file_dir, file_name = self.view1.ask_user_for_filename(
defaultDir=self.app_dir,
style=wx.OPEN,
message='Choose a file',
wildcard='*.*'
)
if file_dir is not None:
config_dict = load_metadata(file_name, file_dir)
settings = self._parse_config(config_dict)
for setting in settings["experiment_settings"]:
self.measurement_handler.add_to_queue(
LightPulse(setting).create_waveform(),
setting
)
self.temperature_settings = settings["temp_settings"]
self.view1.set_temperature_form(
self.temperature_settings.as_dict()
)
self.view1.disable_all_settings_inputs()
self.view1.show_info_modal(
"{0} experiment settings uploaded successfully!".format(
len(self.measurement_handler._queue)
)
)
self.uploaded = True
def display(self, event):
"""
Plot waveforms in modal
"""
self.PlotModal = PlotModal(self.app)
waveform_list = []
settings_dict = {}
try:
settings_dict["experiment_settings"] = self.view1.get_experiment_form()
experiment_settings = self._parse_experiment_settings(settings_dict)
for setting in experiment_settings:
waveform_list.append(-LightPulse(setting).create_waveform())
self.PlotModal.Show()
self.PlotModal.plot_data(waveform_list)
except PVInputError as e:
self.view1.show_error_modal(str(e))
def perform_measurement(self, event):
"""
Perform the queued measurements
"""
try:
if not self.uploaded:
# try to retrieve settings from the various forms
config_dict = {}
config_dict["temperature_settings"] = (
self.view1.get_temperature_form()
)
config_dict["wafer_settings"] = self.view1.get_wafer_form()
config_dict["experiment_settings"] = (
self.view1.get_experiment_form()
)
try:
settings = self._parse_config(config_dict)
except KeyError as e:
self.view1.show_error_modal(str(e))
for setting in settings["experiment_settings"]:
self.measurement_handler.add_to_queue(
LightPulse(setting).create_waveform(),
setting
)
self.wafer_settings = settings["wafer_settings"]
self.temperature_settings = settings["temp_settings"]
if self.data_dir is not None:
save_metadata(
self.wafer_settings.as_dict(),
self.data_dir,
self.wafer_settings.id
)
if self.measurement_handler.is_queue_empty():
raise(PVInputError("No measurements loaded."))
# Do the actual measurements
# TODO: refactor in separate method
dataset_list = []
total_measurements = 0
self.PlotModal = PlotModal(self.app)
self.PlotModal.Show()
while not self.measurement_handler.is_queue_empty():
single_dataset = self.measurement_handler.single_measurement()
dataset_list.append(single_dataset)
total_measurements = total_measurements + 1
ts = int(time.time())
dataset_name = (
str(total_measurements) +
self.wafer_settings.id +
str(ts)
)
save_data(single_dataset, dataset_name, self.data_dir)
# print single_dataset.shape
self.PlotModal.plot_data([x for x in single_dataset[:,1:].T])
except PVInputError as e:
self.view1.show_error_modal(str(e))
def calibrate_pc(self, event):
""""
Perform and calibrate the PC measurement
"""
self.view1.show_calibration_modal()
null_metadata = ExperimentSettings()
null_metadata.waveform = "NullWave"
pc_data = self.measurement_handler.pc_calibration_measurement(
null_metadata
)
self.pc_calibration.update_data(pc_data)
def show_calibration_const(self, event):
"""
Display the calibration constants
"""
message = str(self.pc_calibration_data.as_dict())
self.view1.show_info_modal(message)
def _set_event_bindings(self):
"""
Setups all program behavior which isn't strictly related to the UI
"""
self.view1.m_dataOutputDir.Bind(wx.EVT_BUTTON, self.data_output_dir)
self.view1.m_save.Bind(wx.EVT_BUTTON, self.save_settings)
self.view1.m_load.Bind(wx.EVT_BUTTON, self.load_settings)
self.view1.m_upload.Bind(wx.EVT_BUTTON, self.upload)
self.view1.m_display.Bind(wx.EVT_BUTTON, self.display)
self.view1.m_performMeasurement.Bind(
wx.EVT_BUTTON, self.perform_measurement
)
self.view1.m_calibratePC.Bind(
wx.EVT_BUTTON, self.calibrate_pc
)
self.view1.m_showCalibrationConst.Bind(
wx.EVT_BUTTON, self.show_calibration_const
)
def _parse_experiment_settings(self, config):
measurement_list = []
experiments_list = config["experiment_settings"]
for item in experiments_list:
try:
measurement_list.append(
ExperimentSettings(
waveform=item['waveform'],
duration=item['duration'],
amplitude=item['amplitude'],
offset_before=item['offset_before'],
offset_after=item['offset_after'],
sample_rate=item['sample_rate'],
channel=item['channel'],
binning=item['binning'],
averaging=item['averaging']
)
)
except KeyError as e:
raise("Missing value in Experiment Settings: {0}".format(e))
return measurement_list
def _parse_config(self, config):
"""
Takes a dictionary of settings and constructs appropriate objects
"""
settings = {}
settings["experiment_settings"] = self._parse_experiment_settings(
config
)
try:
settings["temp_settings"] = TemperatureSettings(
start_temp=config["temperature_settings"]["start_temp"],
end_temp=config["temperature_settings"]["end_temp"],
step_temp=config["temperature_settings"]["step_temp"],
step_wait=config["temperature_settings"]["step_wait"],
temperature_scale=config["temperature_settings"][
"temperature_scale"
]
)
except KeyError as e:
raise KeyError(
"Missing value in Temperature Settings: {0}".format(e)
)
try:
settings["wafer_settings"] = Wafer(
wafer_id=config["wafer_settings"]["wafer_id"],
thickness=config["wafer_settings"]["wafer_thickness"],
na=config["wafer_settings"]["wafer_na"],
nd=config["wafer_settings"]["wafer_nd"],
diffused=config["wafer_settings"]["wafer_diffused"],
num_sides=config["wafer_settings"]["wafer_num_sides"]
)
except KeyError as e:
raise KeyError("Missing value in Wafer settings: {0}".format(e))
return settings