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