def _reset_acquisition_gui(self,
text=None,
level=None,
keep_filename=False):
"""
Set back every GUI elements to be ready for the next acquisition
text (None or str): a (error) message to display instead of the
estimated acquisition time
level (None or logging.*): logging level of the text, shown as an icon.
If None, no icon is shown.
keep_filename (bool): if True, will not update the filename
"""
self.btn_cancel.Hide()
self.btn_acquire.Enable()
self._tab_panel.Layout()
self._resume_streams()
if not keep_filename:
self.conf.fn_count = update_counter(self.conf.fn_count)
# Update filename even if keep_filename is True (but don't update counter). This
# ensures that the time is always up to date.
self.filename.value = create_filename(self.conf.last_path,
self.conf.fn_ptn,
self.conf.last_extension,
self.conf.fn_count)
if text is not None:
self.lbl_acqestimate.SetLabel(text)
self._show_status_icons(level)
else:
self.update_acquisition_time()
def _reset_acquisition_gui(self, text=None, level=None, keep_filename=False):
"""
Set back every GUI elements to be ready for the next acquisition
text (None or str): a (error) message to display instead of the
estimated acquisition time
level (None or logging.*): logging level of the text, shown as an icon.
If None, no icon is shown.
keep_filename (bool): if True, will not update the filename
"""
self.btn_cancel.Hide()
self.btn_acquire.Enable()
self._tab_panel.Layout()
self._resume_streams()
if not keep_filename:
self.conf.fn_count = update_counter(self.conf.fn_count)
# Update filename even if keep_filename is True (but don't update counter). This
# ensures that the time is always up to date.
self.filename.value = create_filename(self.conf.last_path, self.conf.fn_ptn,
self.conf.last_extension, self.conf.fn_count)
if text is not None:
self.lbl_acqestimate.SetLabel(text)
self._show_status_icons(level)
else:
self.update_acquisition_time()
def on_acquisition_done(self, future):
""" Callback called when the acquisition is finished (either successfully or cancelled) """
if self._main_data_model.opm:
self._main_data_model.opm.setAcqQuality(path.ACQ_QUALITY_FAST)
# bind button back to direct closure
self.btn_cancel.Bind(wx.EVT_BUTTON, self.on_close)
self._resume_settings()
self.acquiring = False
self.acq_future = None # To avoid holding the ref in memory
self._acq_future_connector = None
try:
data = future.result(1) # timeout is just for safety
self.conf.fn_count = update_counter(self.conf.fn_count)
except CancelledError:
# put back to original state:
# re-enable the acquire button
self.btn_secom_acquire.Enable()
# hide progress bar (+ put pack estimated time)
self.update_acquisition_time()
self.gauge_acq.Hide()
self.Layout()
return
except Exception:
# We cannot do much: just warn the user and pretend it was cancelled
logging.exception("Acquisition failed")
self.btn_secom_acquire.Enable()
self.lbl_acqestimate.SetLabel("Acquisition failed.")
self.lbl_acqestimate.Parent.Layout()
# leave the gauge, to give a hint on what went wrong.
return
# Now store the data (as pyramidal data), and open it again (but now it's
# backed with the persistent storage.
try:
exporter = dataio.find_fittest_converter(self.filename)
if exporter.CAN_SAVE_PYRAMID:
exporter.export(self.filename, data, pyramid=True)
else:
logging.warning(
"File format doesn't support saving image in pyramidal form"
)
exporter.export(self.filename)
self.data = exporter.open_data(self.filename).content
except Exception:
# We cannot do much: just warn the user and pretend it was cancelled
logging.exception("Storage failed")
self.btn_secom_acquire.Enable()
self.lbl_acqestimate.SetLabel("Storage failed.")
self.lbl_acqestimate.Parent.Layout()
return
self.terminate_listeners()
self.EndModal(wx.ID_OPEN)
def _update_filename(self):
"""
Set filename from pattern in conf file.
"""
fn = create_filename(self.conf.last_path, self.conf.fn_ptn, '.tiff', self.conf.fn_count)
self.conf.fn_count = update_counter(self.conf.fn_count)
# Update the widget, without updating the pattern and counter again
self.filename.unsubscribe(self._on_filename)
self.filename.value = fn
self.filename.subscribe(self._on_filename)
def _update_filename(self):
"""
Set filename from pattern in conf file
"""
fn = create_filename(self.conf.last_path, self.conf.fn_ptn, '.png', self.conf.fn_count)
self.conf.fn_count = update_counter(self.conf.fn_count)
# Update the widget, without updating the pattern and counter again
self.filename.unsubscribe(self._on_filename)
self.filename.value = fn
self.filename.subscribe(self._on_filename)
def on_acquisition_done(self, future):
""" Callback called when the acquisition is finished (either successfully or cancelled) """
if self._main_data_model.opm:
self._main_data_model.opm.setAcqQuality(path.ACQ_QUALITY_FAST)
# bind button back to direct closure
self.btn_cancel.Bind(wx.EVT_BUTTON, self.on_close)
self._resume_settings()
self.acquiring = False
# re-enable estimation time updates
self._view.lastUpdate.subscribe(self.on_streams_changed)
self.acq_future = None # To avoid holding the ref in memory
self._acq_future_connector = None
try:
# TODO: Add code for getting the data from the acquisition future
self.data = future.result(1) # timeout is just for safety
self.conf.fn_count = update_counter(self.conf.fn_count)
except CancelledError:
# put back to original state:
# re-enable the acquire button
self.btn_secom_acquire.Enable()
# hide progress bar (+ put pack estimated time)
self.update_acquisition_time()
self.gauge_acq.Hide()
self.Layout()
return
except Exception:
# We cannot do much: just warn the user and pretend it was cancelled
logging.exception("Acquisition failed")
self.btn_secom_acquire.Enable()
self.lbl_acqestimate.SetLabel("Acquisition failed.")
self.lbl_acqestimate.Parent.Layout()
# leave the gauge, to give a hint on what went wrong.
return
self.terminate_listeners()
self.EndModal(wx.ID_OPEN)
def test_update_counter(self):
self.assertEqual(update_counter('0'), '1')
self.assertEqual(update_counter('0005'), '0006')
self.assertEqual(update_counter('9'), '10')
self.assertEqual(update_counter('000'), '001')
self.assertRaises(AssertionError, update_counter, '-5')
def on_acquisition_done(self, future):
""" Callback called when the acquisition is finished (either successfully or cancelled) """
if self._main_data_model.opm:
self._main_data_model.opm.setAcqQuality(path.ACQ_QUALITY_FAST)
# bind button back to direct closure
self.btn_cancel.Bind(wx.EVT_BUTTON, self.on_close)
self._resume_settings()
self.acquiring = False
# re-enable estimation time updates
self._view.lastUpdate.subscribe(self.on_streams_changed)
self.acq_future = None # To avoid holding the ref in memory
self._acq_future_connector = None
try:
data, exp = future.result(1) # timeout is just for safety
self.conf.fn_count = update_counter(self.conf.fn_count)
except CancelledError:
# put back to original state:
# re-enable the acquire button
self.btn_secom_acquire.Enable()
# hide progress bar (+ put pack estimated time)
self.update_acquisition_time()
self.gauge_acq.Hide()
self.Layout()
return
except Exception:
# We cannot do much: just warn the user and pretend it was cancelled
logging.exception("Acquisition failed")
self.btn_secom_acquire.Enable()
self.lbl_acqestimate.SetLabel("Acquisition failed.")
self.lbl_acqestimate.Parent.Layout()
# leave the gauge, to give a hint on what went wrong.
return
# Handle the case acquisition failed "a bit"
if exp:
logging.warning("Acquisition failed (after %d streams): %s",
len(data), exp)
# save result to file
self.lbl_acqestimate.SetLabel("Saving file...")
self.lbl_acqestimate.Parent.Layout()
try:
thumb = acqmng.computeThumbnail(self._view.stream_tree, future)
filename = self.filename.value
exporter = dataio.get_converter(self.conf.last_format)
exporter.export(filename, data, thumb)
logging.info("Acquisition saved as file '%s'.", filename)
# Allow to see the acquisition
self.btn_secom_acquire.SetLabel("VIEW")
self.last_saved_file = filename
except Exception:
logging.exception("Saving acquisition failed")
self.btn_secom_acquire.Enable()
self.lbl_acqestimate.SetLabel("Saving acquisition file failed.")
self.lbl_acqestimate.Parent.Layout()
return
if exp:
self.lbl_acqestimate.SetLabel("Acquisition failed (partially).")
else:
self.lbl_acqestimate.SetLabel("Acquisition completed.")
# As the action is complete, rename "Cancel" to "Close"
self.btn_cancel.SetLabel("Close")
self.lbl_acqestimate.Parent.Layout()
# Make sure the file is not overridden
self.btn_secom_acquire.Enable()
def on_acquisition_done(self, future):
""" Callback called when the acquisition is finished (either successfully or cancelled) """
if self._main_data_model.opm:
self._main_data_model.opm.setAcqQuality(path.ACQ_QUALITY_FAST)
# bind button back to direct closure
self.btn_cancel.Bind(wx.EVT_BUTTON, self.on_close)
self._resume_settings()
self.acquiring = False
# re-enable estimation time updates
self._view.lastUpdate.subscribe(self.on_streams_changed)
self.acq_future = None # To avoid holding the ref in memory
self._acq_future_connector = None
try:
data, exp = future.result(1) # timeout is just for safety
self.conf.fn_count = update_counter(self.conf.fn_count)
except CancelledError:
# put back to original state:
# re-enable the acquire button
self.btn_secom_acquire.Enable()
# hide progress bar (+ put pack estimated time)
self.update_acquisition_time()
self.gauge_acq.Hide()
self.Layout()
return
except Exception:
# We cannot do much: just warn the user and pretend it was cancelled
logging.exception("Acquisition failed")
self.btn_secom_acquire.Enable()
self.lbl_acqestimate.SetLabel("Acquisition failed.")
self.lbl_acqestimate.Parent.Layout()
# leave the gauge, to give a hint on what went wrong.
return
# Handle the case acquisition failed "a bit"
if exp:
logging.warning("Acquisition failed (after %d streams): %s",
len(data), exp)
# save result to file
self.lbl_acqestimate.SetLabel("Saving file...")
self.lbl_acqestimate.Parent.Layout()
try:
thumb = acq.computeThumbnail(self._view.stream_tree, future)
filename = self.filename.value
exporter = dataio.get_converter(self.conf.last_format)
exporter.export(filename, data, thumb)
logging.info("Acquisition saved as file '%s'.", filename)
# Allow to see the acquisition
self.btn_secom_acquire.SetLabel("VIEW")
self.last_saved_file = filename
except Exception:
logging.exception("Saving acquisition failed")
self.btn_secom_acquire.Enable()
self.lbl_acqestimate.SetLabel("Saving acquisition file failed.")
self.lbl_acqestimate.Parent.Layout()
return
if exp:
self.lbl_acqestimate.SetLabel("Acquisition failed (partially).")
else:
self.lbl_acqestimate.SetLabel("Acquisition completed.")
# As the action is complete, rename "Cancel" to "Close"
self.btn_cancel.SetLabel("Close")
self.lbl_acqestimate.Parent.Layout()
# Make sure the file is not overridden
self.btn_secom_acquire.Enable()