def __init__(self, main_gui, active_data, parent=None):
self.main_gui = main_gui
self.data = active_data
QDialog.__init__(self, parent=parent)
self.setWindowModality(False)
self._open_instances.append(self)
self.ui = UiPlot()
self.ui.setupUi(self)
self.setWindowTitle('Detector and Pixel vs TOF views')
self.init_gui()
self.populate_widgets()
self.init_home_settings()
self.update_detector_tab_plot()
self.update_pixel_vs_tof_tab_plot()
self.ui.y_pixel_vs_tof_plot.leaveFigure.connect(self.leave_figure_plot)
self.ui.y_pixel_vs_tof_plot.toolbar.homeClicked.connect(self.home_clicked_plot)
self.ui.y_pixel_vs_tof_plot.toolbar.exportClicked.connect(self.export_yt)
self.ui.detector_plot.leaveFigure.connect(self.leave_figure_detector_plot)
self.ui.detector_plot.toolbar.homeClicked.connect(self.home_clicked_detector_plot)
self.ui.detector_plot.toolbar.exportClicked.connect(self.export_detector_view)
# hide low_res widgets
self.ui.frame_20.setVisible(False)
def __init__(self, parent = None,
data_type = 'data',
data = None,
row = 0):
self.parent = parent
self.data = data
self.data_type = data_type
self.row = row
self.col = 0 if data_type == 'data' else 1
self.is_data = True if data_type == 'data' else False
self.is_row_with_highest_q = self.is_row_with_higest_q()
QDialog.__init__(self, parent=parent)
self.setWindowModality(False)
self._open_instances.append(self)
self.ui = UiPlot()
self.ui.setupUi(self)
self.setWindowTitle('Detector and Pixel vs TOF views')
self.init_gui()
self.populate_widgets()
self.update_detector_tab_plot()
self.update_pixel_vs_tof_tab_plot()
self.ui.y_pixel_vs_tof_plot.leaveFigure.connect(self.leave_figure_plot)
self.ui.y_pixel_vs_tof_plot.toolbar.homeClicked.connect(self.home_clicked_plot)
self.ui.y_pixel_vs_tof_plot.toolbar.exportClicked.connect(self.export_yt)
self.ui.detector_plot.leaveFigure.connect(self.leave_figure_detector_plot)
self.ui.detector_plot.toolbar.homeClicked.connect(self.home_clicked_detector_plot)
self.ui.detector_plot.toolbar.exportClicked.connect(self.export_detector_view)
def __init__(self, parent=None, data_type='data', data=None, row=0):
self.parent = parent
self.data = data
self.data_type = data_type
self.row = row
self.col = 0 if data_type == 'data' else 1
self.is_data = True if data_type == 'data' else False
self.is_row_with_highest_q = self.is_row_with_higest_q()
QDialog.__init__(self, parent=parent)
self.setWindowModality(False)
self._open_instances.append(self)
self.ui = UiPlot()
self.ui.setupUi(self)
self.setWindowTitle('Detector and Pixel vs TOF views')
self.init_gui()
self.populate_widgets()
self.update_detector_tab_plot()
self.update_pixel_vs_tof_tab_plot()
self.ui.y_pixel_vs_tof_plot.leaveFigure.connect(self.leave_figure_plot)
self.ui.y_pixel_vs_tof_plot.toolbar.homeClicked.connect(
self.home_clicked_plot)
self.ui.y_pixel_vs_tof_plot.toolbar.exportClicked.connect(
self.export_yt)
self.ui.detector_plot.leaveFigure.connect(
self.leave_figure_detector_plot)
self.ui.detector_plot.toolbar.homeClicked.connect(
self.home_clicked_detector_plot)
self.ui.detector_plot.toolbar.exportClicked.connect(
self.export_detector_view)
self.widgets_to_show()
class Plot2dSFDialogREFL(QDialog):
main_gui = None
_open_instances = []
data = None
type = 'data'
manual_min_tof = None
manual_max_tof = None
auto_min_tof = None
auto_max_tof = None
def __init__(self, main_gui, active_data, parent=None):
self.main_gui = main_gui
self.data = active_data
QDialog.__init__(self, parent=parent)
self.setWindowModality(False)
self._open_instances.append(self)
self.ui = UiPlot()
self.ui.setupUi(self)
self.setWindowTitle('Detector and Pixel vs TOF views')
self.init_gui()
self.populate_widgets()
self.init_home_settings()
self.update_detector_tab_plot()
self.update_pixel_vs_tof_tab_plot()
self.ui.y_pixel_vs_tof_plot.leaveFigure.connect(self.leave_figure_plot)
self.ui.y_pixel_vs_tof_plot.toolbar.homeClicked.connect(self.home_clicked_plot)
self.ui.y_pixel_vs_tof_plot.toolbar.exportClicked.connect(self.export_yt)
self.ui.detector_plot.leaveFigure.connect(self.leave_figure_detector_plot)
self.ui.detector_plot.toolbar.homeClicked.connect(self.home_clicked_detector_plot)
self.ui.detector_plot.toolbar.exportClicked.connect(self.export_detector_view)
# hide low_res widgets
self.ui.frame_20.setVisible(False)
def export_yt(self):
_active_data = self.data
run_number = _active_data.run_number
default_filename = 'REFL_' + run_number + '_2dPxVsTof.txt'
path = self.main_gui.path_ascii
default_filename = path + '/' + default_filename
filename = QFileDialog.getSaveFileName(self, 'Create 2D Pixel VS TOF', default_filename)
if str(filename).strip() == '':
info('User Canceled Outpout ASCII')
return
self.main_gui.path_ascii = os.path.dirname(filename)
image = _active_data.ytofdata
RefRed.utilities.output_2d_ascii_file(filename, image)
def export_detector_view(self):
_active_data = self.data
run_number = _active_data.run_number
default_filename = 'REFL_' + run_number + '_2dDetectorView.txt'
path = self.main_gui.path_ascii
default_filename = path + '/' + default_filename
filename = QFileDialog.getSaveFileName(self, 'Create 2D Y Pixel VS X Pixel (Detector View)', default_filename)
if str(filename).strip() == '':
info('User Canceled Outpout ASCII')
return
self.main_gui.path_ascii = os.path.dirname(filename)
image = _active_data.xydata
RefRed.utilities.output_2d_ascii_file(filename, image)
def init_home_settings(self):
home_settings_1 = self.main_gui.yt_plot.toolbar.home_settings
self.ui.y_pixel_vs_tof_plot.toolbar.home_settings = home_settings_1
def leave_figure_plot(self):
[xmin, xmax]= self.ui.y_pixel_vs_tof_plot.canvas.ax.xaxis.get_view_interval()
[ymin, ymax]= self.ui.y_pixel_vs_tof_plot.canvas.ax.yaxis.get_view_interval()
self.ui.y_pixel_vs_tof_plot.canvas.ax.xaxis.set_data_interval(xmin, xmax)
self.ui.y_pixel_vs_tof_plot.canvas.ax.yaxis.set_data_interval(ymin, ymax)
self.ui.y_pixel_vs_tof_plot.draw()
self.data.all_plot_axis.yt_view_interval = [xmin, xmax, ymin, ymax]
def home_clicked_plot(self):
# [xmin,xmax,ymin,ymax] = self.data.all_plot_axis.yt_data_interval
[xmin,xmax,ymin,ymax] = self.ui.y_pixel_vs_tof_plot.toolbar.home_settings
self.ui.y_pixel_vs_tof_plot.canvas.ax.set_xlim([xmin,xmax])
self.ui.y_pixel_vs_tof_plot.canvas.ax.set_ylim([ymin,ymax])
self.ui.y_pixel_vs_tof_plot.draw()
def leave_figure_detector_plot(self):
[xmin, xmax]= self.ui.detector_plot.canvas.ax.xaxis.get_view_interval()
[ymin, ymax]= self.ui.detector_plot.canvas.ax.yaxis.get_view_interval()
self.ui.detector_plot.canvas.ax.xaxis.set_data_interval(xmin, xmax)
self.ui.detector_plot.canvas.ax.yaxis.set_data_interval(ymin, ymax)
self.ui.detector_plot.draw()
self.data.all_plot_axis.detector_view_interval = [xmin, xmax, ymin, ymax]
def home_clicked_detector_plot(self):
[xmin,xmax,ymin,ymax] = self.data.all_plot_axis.detector_data_interval
self.ui.detector_plot.canvas.ax.set_xlim([xmin,xmax])
self.ui.detector_plot.canvas.ax.set_ylim([ymin,ymax])
self.ui.detector_plot.draw()
def update_detector_tab_plot(self):
self.ui.detector_plot.clear()
xydata = self.data.xydata
self.ui.detector_plot.draw()
[ymax,xmax] = xydata.shape
self.ui.detector_plot.imshow(xydata, log=True, aspect='auto',
origin='lower', extent=[0,xmax,0,ymax])
self.ui.detector_plot.set_xlabel(u'x (pixel)')
self.ui.detector_plot.set_ylabel(u'y (pixel)')
[peak1, peak2, back1, back2, backFlag] = self.retrievePeakBack()
p1 = self.ui.detector_plot.canvas.ax.axhline(peak1,
color = RefRed.colors.PEAK_SELECTION_COLOR)
p2 = self.ui.detector_plot.canvas.ax.axhline(peak2,
color = RefRed.colors.PEAK_SELECTION_COLOR)
if backFlag:
b1 = self.ui.detector_plot.canvas.ax.axhline(back1,
color = RefRed.colors.BACK_SELECTION_COLOR)
b2 = self.ui.detector_plot.canvas.ax.axhline(back2,
color = RefRed.colors.BACK_SELECTION_COLOR)
if self.data.all_plot_axis.detector_data_interval is None:
self.ui.detector_plot.draw()
[xmin,xmax] = self.ui.detector_plot.canvas.ax.xaxis.get_view_interval()
[ymin,ymax] = self.ui.detector_plot.canvas.ax.yaxis.get_view_interval()
self.data.all_plot_axis.detector_data_interval = [xmin, xmax, ymin, ymax]
self.data.all_plot_axis.detector_view_interval = [xmin, xmax, ymin, ymax]
self.ui.detector_plot.toolbar.home_settings = [xmin, xmax, ymin, ymax]
else:
[xmin, xmax, ymin, ymax] = self.data.all_plot_axis.detector_view_interval
self.ui.detector_plot.canvas.ax.set_xlim([xmin, xmax])
self.ui.detector_plot.canvas.ax.set_ylim([ymin, ymax])
self.ui.detector_plot.draw()
def update_pixel_vs_tof_tab_plot(self):
ytof = self.data.ytofdata
tof_axis = self.data.tof_axis_auto_with_margin
tof_from = tof_axis[0]
tof_to = tof_axis[-1]
if tof_from > 1000: # stay in ms
tof_from /= 1000.
tof_to /= 1000.
pixel_from = 0
pixel_to = self.data.y.shape[0]-1
self.ui.y_pixel_vs_tof_plot.clear()
self.ui.y_pixel_vs_tof_plot.imshow(ytof, log=True, aspect='auto',
origin='lower', extent=[tof_from, tof_to, pixel_from, pixel_to])
self.ui.y_pixel_vs_tof_plot.set_xlabel(u't (ms)')
self.ui.y_pixel_vs_tof_plot.set_ylabel(u'y (pixel)')
[tmin,tmax,peak1,peak2,back1,back2,backFlag] = self.retrieveTofPeakBack()
if tmin > 1000: # stay in ms
tmin /= 1000
tmax /= 1000
t1 = self.ui.y_pixel_vs_tof_plot.canvas.ax.axvline(tmin,
color = RefRed.colors.TOF_SELECTION_COLOR)
t2 = self.ui.y_pixel_vs_tof_plot.canvas.ax.axvline(tmax,
color = RefRed.colors.TOF_SELECTION_COLOR)
p1 = self.ui.y_pixel_vs_tof_plot.canvas.ax.axhline(peak1,
color = RefRed.colors.PEAK_SELECTION_COLOR)
p2 = self.ui.y_pixel_vs_tof_plot.canvas.ax.axhline(peak2,
color = RefRed.colors.PEAK_SELECTION_COLOR)
if backFlag:
b1 = self.ui.y_pixel_vs_tof_plot.canvas.ax.axhline(back1,
color = RefRed.colors.BACK_SELECTION_COLOR)
b2 = self.ui.y_pixel_vs_tof_plot.canvas.ax.axhline(back2,
color = RefRed.colors.BACK_SELECTION_COLOR)
if self.data.all_plot_axis.yt_data_interval is None:
self.ui.y_pixel_vs_tof_plot.canvas.ax.set_ylim(0,pixel_to)
self.ui.y_pixel_vs_tof_plot.canvas.draw()
[xmin,xmax] = self.ui.y_pixel_vs_tof_plot.canvas.ax.xaxis.get_view_interval()
[ymin,ymax] = self.ui.y_pixel_vs_tof_plot.canvas.ax.yaxis.get_view_interval()
self.data.all_plot_axis.yt_data_interval = [xmin, xmax, ymin, ymax]
self.data.all_plot_axis.yt_view_interval = [xmin, xmax, ymin, ymax]
self.ui.y_pixel_vs_tof_plot.toolbar.home_settings = [xmin, xmax, ymin, ymax]
else:
[xmin,xmax,ymin,ymax] = self.data.all_plot_axis.yt_view_interval
self.ui.y_pixel_vs_tof_plot.canvas.ax.set_xlim([xmin,xmax])
self.ui.y_pixel_vs_tof_plot.canvas.ax.set_ylim([ymin,ymax])
self.ui.y_pixel_vs_tof_plot.draw()
def retrieveTofPeakBack(self):
tmin = float(self.ui.tof_from.text())
tmax = float(self.ui.tof_to.text())
[peak1,peak2,back1,back2, backFlag] = self.retrievePeakBack()
return [tmin, tmax, peak1, peak2, back1, back2, backFlag]
def retrievePeakBack(self):
peak1 = self.ui.peak1.value()
peak2 = self.ui.peak2.value()
back1 = self.ui.back1.value()
back2 = self.ui.back2.value()
backFlag = self.ui.back_flag.isChecked()
return [peak1,peak2,back1,back2,backFlag]
def init_gui(self):
palette = QPalette()
palette.setColor(QPalette.Foreground, Qt.red)
if self.data.new_detector_geometry_flag:
yrange = [0,303]
xrange = [0,255]
else:
yrange = [0,255]
xrange = [0,303]
self.ui.error_label.setVisible(False)
self.ui.error_label.setPalette(palette)
self.ui.peak1_label.setVisible(False)
self.ui.peak1_label.setPalette(palette)
self.ui.peak2_label.setVisible(False)
self.ui.peak2_label.setPalette(palette)
self.ui.back1_label.setVisible(False)
self.ui.back1_label.setPalette(palette)
self.ui.back2_label.setVisible(False)
self.ui.back2_label.setPalette(palette)
# peak
self.ui.peak1.setMinimum(yrange[0])
self.ui.peak1.setMaximum(yrange[1])
self.ui.peak2.setMinimum(yrange[0])
self.ui.peak2.setMaximum(yrange[1])
# back
self.ui.back1.setMinimum(yrange[0])
self.ui.back1.setMaximum(yrange[1])
self.ui.back2.setMinimum(yrange[0])
self.ui.back2.setMaximum(yrange[1])
# low res
self.ui.low_res1.setMinimum(xrange[0])
self.ui.low_res1.setMaximum(xrange[1])
self.ui.low_res2.setMinimum(xrange[0])
self.ui.low_res2.setMaximum(xrange[1])
def populate_widgets(self):
_data = self.data
peak = _data.peak
back = _data.back
back_flag = _data.back_flag
tof_auto_flag = _data.tof_auto_flag
tof_range_auto = _data.tof_range_auto
tof_range = _data.tof_range
# make sure we are in ms
tof_range_auto_min = float(tof_range_auto[0])
tof_range_auto_max = float(tof_range_auto[1])
if tof_range_auto_min > 1000:
tof_range_auto_min /= 1000.
tof_range_auto_max /= 1000.
self.auto_max_tof = tof_range_auto_max
self.auto_min_tof = tof_range_auto_min
tof_range_manual_min = float(tof_range[0])
tof_range_manual_max = float(tof_range[1])
if tof_range_manual_min > 1000:
tof_range_manual_min /= 1000.
tof_range_manual_max /= 1000.
self.manual_max_tof = tof_range_manual_max
self.manual_min_tof = tof_range_manual_min
self.manual_auto_tof_clicked(plot_refresh=False)
self.ui.peak1.setValue(int(peak[0]))
self.ui.peak2.setValue(int(peak[1]))
self.ui.back1.setValue(int(back[0]))
self.ui.back2.setValue(int(back[1]))
self.activate_or_not_back_widgets(back_flag, plot_refresh=False)
def activate_or_not_back_widgets(self, back_flag, plot_refresh=True):
self.ui.back_flag.setChecked(back_flag)
self.ui.back1.setEnabled(back_flag)
self.ui.back2.setEnabled(back_flag)
self.check_peak_back_input_validity()
if plot_refresh:
self.update_plots()
def sort_peak_back_input(self):
peak1 = self.ui.peak1.value()
peak2 = self.ui.peak2.value()
peak_min = min([peak1, peak2])
peak_max = max([peak1, peak2])
if peak_min != peak1:
self.ui.peak1.setValue(peak2)
self.ui.peak2.setValue(peak1)
back1 = self.ui.back1.value()
back2 = self.ui.back2.value()
back_min = min([back1, back2])
back_max = max([back1, back2])
if back_min != back1:
self.ui.back1.setValue(back2)
self.ui.back2.setValue(back1)
def update_plots(self):
self.update_pixel_vs_tof_tab_plot()
self.update_detector_tab_plot()
def manual_input_peak1(self):
self.sort_and_check_widgets()
self.update_plots()
def manual_input_peak2(self):
self.sort_and_check_widgets()
self.update_plots()
def manual_input_back1(self):
self.sort_and_check_widgets()
self.update_plots()
def manual_input_back2(self):
self.sort_and_check_widgets()
self.update_plots()
def sort_and_check_widgets(self):
self.sort_peak_back_input()
self.check_peak_back_input_validity()
def check_peak_back_input_validity(self):
peak1 = self.ui.peak1.value()
peak2 = self.ui.peak2.value()
back1 = self.ui.back1.value()
back2 = self.ui.back2.value()
_show_widgets_1 = False
_show_widgets_2 = False
if self.ui.back_flag.isChecked():
if back1 > peak1:
_show_widgets_1 = True
if back2 < peak2:
_show_widgets_2 = True
self.ui.back1_label.setVisible(_show_widgets_1)
self.ui.peak1_label.setVisible(_show_widgets_1)
self.ui.back2_label.setVisible(_show_widgets_2)
self.ui.peak2_label.setVisible(_show_widgets_2)
self.ui.error_label.setVisible(_show_widgets_1 or _show_widgets_2)
def manual_input_of_low_res_field(self):
value1 = self.ui.low_res1.value()
value2 = self.ui.low_res2.value()
value_min = min([value1,value2])
value_max = max([value1,value2])
self.ui.detector_plot()
def manual_input_of_tof_field(self):
tof1 = float(self.ui.tof_from.text())
tof2 = float(self.ui.tof_to.text())
tof_min = min([tof1, tof2])
tof_max = max([tof1, tof2])
str_tof_min = ("%.2f"%tof_min)
str_tof_max = ("%.2f"%tof_max)
self.ui.tof_from.setText(str_tof_min)
self.ui.tof_to.setText(str_tof_max)
self.manual_min_tof = tof_min
self.manual_max_tof = tof_max
self.update_pixel_vs_tof_tab_plot()
def manual_auto_tof_clicked(self, plot_refresh=True):
isManualChecked = self.ui.tof_manual_flag.isChecked()
self.activate_tof_widgets(isManualChecked)
if isManualChecked:
_from_value = "%.2f"%self.manual_min_tof
_to_value = "%.2f"%self.manual_max_tof
else:
_from_value = "%.2f"%self.auto_min_tof
_to_value = "%.2f"%self.auto_max_tof
self.ui.tof_from.setText(_from_value)
self.ui.tof_to.setText(_to_value)
if plot_refresh:
self.update_pixel_vs_tof_tab_plot()
def activate_tof_widgets(self, status):
self.ui.tof_from.setEnabled(status)
self.ui.tof_to.setEnabled(status)
self.ui.tof_from_label.setEnabled(status)
self.ui.tof_to_label.setEnabled(status)
self.ui.tof_from_units.setEnabled(status)
self.ui.tof_to_units.setEnabled(status)
# self.update_pixel_vs_tof_tab_plot()
def closeEvent(self, event=None):
[tof1, tof2, peak1, peak2, back1, back2, backFlag]= self.retrieveTofPeakBack()
self.main_gui.dataPeakFromValue.setValue(peak1)
self.main_gui.dataPeakToValue.setValue(peak2)
self.main_gui.dataBackFromValue.setValue(back1)
self.main_gui.dataBackToValue.setValue(back2)
self.main_gui.dataBackgroundFlag.setChecked(backFlag)
QApplication.processEvents()
self.main_gui.peakBackSpinBoxValueChanged(['peak','back'], with_plot_update=False)
self.main_gui.testPeakBackErrorWidgets()
self.main_gui.dataBackFromLabel.setEnabled(backFlag)
self.main_gui.dataBackFromValue.setEnabled(backFlag)
self.main_gui.dataBackToLabel.setEnabled(backFlag)
self.main_gui.dataBackToValue.setEnabled(backFlag)
tof_auto_switch = self.ui.tof_auto_flag.isChecked()
self.main_gui.tof_validation(tof_auto_switch, tof1, tof2, with_plot_update=False)
self.main_gui.displayPlot()
def activate_or_not_low_res_widgets(self, low_res_flag):
pass
def manual_input_of_low_res_field(self):
pass
class PopupPlot2d(QDialog):
parent = None
_open_instances = []
data = None
data_type = 'data'
row = 0
manual_min_tof = None
manual_max_tof = None
auto_min_tof = None
auto_max_tof = None
def __init__(self, parent=None, data_type='data', data=None, row=0):
self.parent = parent
self.data = data
self.data_type = data_type
self.row = row
self.col = 0 if data_type == 'data' else 1
self.is_data = True if data_type == 'data' else False
self.is_row_with_highest_q = self.is_row_with_higest_q()
QDialog.__init__(self, parent=parent)
self.setWindowModality(False)
self._open_instances.append(self)
self.ui = UiPlot()
self.ui.setupUi(self)
self.setWindowTitle('Detector and Pixel vs TOF views')
self.init_gui()
self.populate_widgets()
self.update_detector_tab_plot()
self.update_pixel_vs_tof_tab_plot()
self.ui.y_pixel_vs_tof_plot.leaveFigure.connect(self.leave_figure_plot)
self.ui.y_pixel_vs_tof_plot.toolbar.homeClicked.connect(
self.home_clicked_plot)
self.ui.y_pixel_vs_tof_plot.toolbar.exportClicked.connect(
self.export_yt)
self.ui.detector_plot.leaveFigure.connect(
self.leave_figure_detector_plot)
self.ui.detector_plot.toolbar.homeClicked.connect(
self.home_clicked_detector_plot)
self.ui.detector_plot.toolbar.exportClicked.connect(
self.export_detector_view)
self.widgets_to_show()
def widgets_to_show(self):
if self.is_row_with_highest_q:
enable_status = True
else:
enable_status = False
if not self.is_data:
return
self.ui.clocking_box.setEnabled(enable_status)
def is_row_with_higest_q(self):
o_gui_utility = GuiUtility(parent=self.parent)
return o_gui_utility.is_row_with_highest_q()
def export_yt(self):
_active_data = self.data
run_number = _active_data.run_number
default_filename = 'REFL_' + run_number + '_2dPxVsTof.txt'
path = self.parent.path_ascii
default_filename = path + '/' + default_filename
filename = QFileDialog.getSaveFileName(self, 'Create 2D Pixel VS TOF',
default_filename)
if str(filename).strip() == '':
# info('User Canceled Outpout ASCII')
return
self.parent.path_ascii = os.path.dirname(filename)
image = _active_data.ytofdata
RefRed.utilities.output_2d_ascii_file(filename, image)
def export_detector_view(self):
_active_data = self.data
run_number = _active_data.run_number
default_filename = 'REFL_' + run_number + '_2dDetectorView.txt'
path = self.parent.path_ascii
default_filename = path + '/' + default_filename
filename = QFileDialog.getSaveFileName(
self, 'Create 2D Y Pixel VS X Pixel (Detector View)',
default_filename)
if str(filename).strip() == '':
# info('User Canceled Outpout ASCII')
return
self.parent.path_ascii = os.path.dirname(filename)
image = _active_data.xydata
RefRed.utilities.output_2d_ascii_file(filename, image)
def leave_figure_plot(self):
[xmin, xmax
] = self.ui.y_pixel_vs_tof_plot.canvas.ax.xaxis.get_view_interval()
[ymin, ymax
] = self.ui.y_pixel_vs_tof_plot.canvas.ax.yaxis.get_view_interval()
self.ui.y_pixel_vs_tof_plot.canvas.ax.xaxis.set_data_interval(
xmin, xmax)
self.ui.y_pixel_vs_tof_plot.canvas.ax.yaxis.set_data_interval(
ymin, ymax)
self.ui.y_pixel_vs_tof_plot.draw()
self.data.all_plot_axis.yt_view_interval = [xmin, xmax, ymin, ymax]
def home_clicked_plot(self):
[xmin, xmax, ymin, ymax] = self.data.all_plot_axis.yt_data_interval
self.ui.y_pixel_vs_tof_plot.canvas.ax.set_xlim([xmin, xmax])
self.ui.y_pixel_vs_tof_plot.canvas.ax.set_ylim([ymin, ymax])
self.ui.y_pixel_vs_tof_plot.draw()
def leave_figure_detector_plot(self):
[xmin,
xmax] = self.ui.detector_plot.canvas.ax.xaxis.get_view_interval()
[ymin,
ymax] = self.ui.detector_plot.canvas.ax.yaxis.get_view_interval()
self.ui.detector_plot.canvas.ax.xaxis.set_data_interval(xmin, xmax)
self.ui.detector_plot.canvas.ax.yaxis.set_data_interval(ymin, ymax)
self.ui.detector_plot.draw()
self.data.all_plot_axis.detector_view_interval = [
xmin, xmax, ymin, ymax
]
def home_clicked_detector_plot(self):
[xmin, xmax, ymin,
ymax] = self.data.all_plot_axis.detector_data_interval
self.ui.detector_plot.canvas.ax.set_xlim([xmin, xmax])
self.ui.detector_plot.canvas.ax.set_ylim([ymin, ymax])
self.ui.detector_plot.draw()
def get_clocking_values(self):
clock1 = self.ui.clock1.value()
clock2 = self.ui.clock2.value()
return [clock1, clock2]
def update_detector_tab_plot(self):
self.ui.detector_plot.clear()
xydata = self.data.xydata
self.ui.detector_plot.draw()
[ymax, xmax] = xydata.shape
self.ui.detector_plot.imshow(xydata,
log=True,
aspect='auto',
origin='lower',
extent=[0, xmax, 0, ymax])
self.ui.detector_plot.set_xlabel(u'x (pixel)')
self.ui.detector_plot.set_ylabel(u'y (pixel)')
[lowres1, lowres2, lowresFlag, peak1, peak2, back1, back2,
backFlag] = self.retrieveLowResPeakBack()
[clock1, clock2] = self.get_clocking_values()
if lowresFlag:
lr1 = self.ui.detector_plot.canvas.ax.axvline(
lowres1, color=RefRed.colors.LOWRESOLUTION_SELECTION_COLOR)
lr2 = self.ui.detector_plot.canvas.ax.axvline(
lowres2, color=RefRed.colors.LOWRESOLUTION_SELECTION_COLOR)
p1 = self.ui.detector_plot.canvas.ax.axhline(
peak1, color=RefRed.colors.PEAK_SELECTION_COLOR)
p2 = self.ui.detector_plot.canvas.ax.axhline(
peak2, color=RefRed.colors.PEAK_SELECTION_COLOR)
if self.is_data:
c1 = self.ui.detector_plot.canvas.ax.axhline(
clock1, color=RefRed.colors.CLOCKING_SELECTION_COLOR)
c1 = self.ui.detector_plot.canvas.ax.axhline(
clock2, color=RefRed.colors.CLOCKING_SELECTION_COLOR)
if backFlag:
b1 = self.ui.detector_plot.canvas.ax.axhline(
back1, color=RefRed.colors.BACK_SELECTION_COLOR)
b2 = self.ui.detector_plot.canvas.ax.axhline(
back2, color=RefRed.colors.BACK_SELECTION_COLOR)
if self.data.all_plot_axis.detector_data_interval is None:
self.ui.detector_plot.draw()
[xmin,
xmax] = self.ui.detector_plot.canvas.ax.xaxis.get_view_interval()
[ymin,
ymax] = self.ui.detector_plot.canvas.ax.yaxis.get_view_interval()
self.data.all_plot_axis.detector_data_interval = [
xmin, xmax, ymin, ymax
]
self.data.all_plot_axis.detector_view_interval = [
xmin, xmax, ymin, ymax
]
self.ui.detector_plot.toolbar.home_settings = [
xmin, xmax, ymin, ymax
]
else:
[xmin, xmax, ymin,
ymax] = self.data.all_plot_axis.detector_view_interval
self.ui.detector_plot.canvas.ax.set_xlim([xmin, xmax])
self.ui.detector_plot.canvas.ax.set_ylim([ymin, ymax])
self.ui.detector_plot.draw()
def update_pixel_vs_tof_tab_plot(self):
ytof = self.data.ytofdata
tof_axis = self.data.tof_axis_auto_with_margin
tof_from = tof_axis[0]
tof_to = tof_axis[-1]
if tof_from > 1000: # stay in ms
tof_from /= 1000.
tof_to /= 1000.
pixel_from = 0
pixel_to = self.data.y.shape[0] - 1
self.ui.y_pixel_vs_tof_plot.clear()
self.ui.y_pixel_vs_tof_plot.imshow(
ytof,
log=True,
aspect='auto',
origin='lower',
extent=[tof_from, tof_to, pixel_from, pixel_to])
self.ui.y_pixel_vs_tof_plot.set_xlabel(u't (ms)')
self.ui.y_pixel_vs_tof_plot.set_ylabel(u'y (pixel)')
[tmin, tmax, peak1, peak2, back1, back2,
backFlag] = self.retrieveTofPeakBack()
[clock1, clock2] = self.get_clocking_values()
t1 = self.ui.y_pixel_vs_tof_plot.canvas.ax.axvline(
tmin, color=RefRed.colors.TOF_SELECTION_COLOR)
t2 = self.ui.y_pixel_vs_tof_plot.canvas.ax.axvline(
tmax, color=RefRed.colors.TOF_SELECTION_COLOR)
p1 = self.ui.y_pixel_vs_tof_plot.canvas.ax.axhline(
peak1, color=RefRed.colors.PEAK_SELECTION_COLOR)
p2 = self.ui.y_pixel_vs_tof_plot.canvas.ax.axhline(
peak2, color=RefRed.colors.PEAK_SELECTION_COLOR)
if self.is_data:
c1 = self.ui.y_pixel_vs_tof_plot.canvas.ax.axhline(
clock1, color=RefRed.colors.CLOCKING_SELECTION_COLOR)
c1 = self.ui.y_pixel_vs_tof_plot.canvas.ax.axhline(
clock2, color=RefRed.colors.CLOCKING_SELECTION_COLOR)
if backFlag:
b1 = self.ui.y_pixel_vs_tof_plot.canvas.ax.axhline(
back1, color=RefRed.colors.BACK_SELECTION_COLOR)
b2 = self.ui.y_pixel_vs_tof_plot.canvas.ax.axhline(
back2, color=RefRed.colors.BACK_SELECTION_COLOR)
if self.data.all_plot_axis.yt_data_interval is None:
self.ui.y_pixel_vs_tof_plot.canvas.ax.set_ylim(0, pixel_to)
self.ui.y_pixel_vs_tof_plot.canvas.draw()
[xmin, xmax
] = self.ui.y_pixel_vs_tof_plot.canvas.ax.xaxis.get_view_interval(
)
[ymin, ymax
] = self.ui.y_pixel_vs_tof_plot.canvas.ax.yaxis.get_view_interval(
)
self.data.all_plot_axis.yt_data_interval = [xmin, xmax, ymin, ymax]
self.data.all_plot_axis.yt_view_interval = [xmin, xmax, ymin, ymax]
self.ui.y_pixel_vs_tof_plot.toolbar.home_settings = [
xmin, xmax, ymin, ymax
]
else:
[xmin, xmax, ymin, ymax] = self.data.all_plot_axis.yt_view_interval
self.ui.y_pixel_vs_tof_plot.canvas.ax.set_xlim([xmin, xmax])
self.ui.y_pixel_vs_tof_plot.canvas.ax.set_ylim([ymin, ymax])
self.ui.y_pixel_vs_tof_plot.draw()
def retrieveTofPeakBack(self):
tmin = float(self.ui.tof_from.text())
tmax = float(self.ui.tof_to.text())
[peak1, peak2, back1, back2, backFlag] = self.retrievePeakBack()
return [tmin, tmax, peak1, peak2, back1, back2, backFlag]
def retrieveLowResPeakBack(self):
lowres1 = self.ui.low_res1.value()
lowres2 = self.ui.low_res2.value()
lowresFlag = self.ui.low_res_flag.isChecked()
[peak1, peak2, back1, back2, backFlag] = self.retrievePeakBack()
return [
lowres1, lowres2, lowresFlag, peak1, peak2, back1, back2, backFlag
]
def retrieveLowRes(self):
lowres1 = self.ui.low_res1.value()
lowres2 = self.ui.low_res2.value()
lowresFlag = self.ui.low_res_flag.isChecked()
return [lowres1, lowres2, lowresFlag]
def retrievePeakBack(self):
peak1 = self.ui.peak1.value()
peak2 = self.ui.peak2.value()
back1 = self.ui.back1.value()
back2 = self.ui.back2.value()
backFlag = self.ui.back_flag.isChecked()
return [peak1, peak2, back1, back2, backFlag]
def init_gui(self):
palette = QPalette()
palette.setColor(QPalette.Foreground, Qt.red)
if self.data.new_detector_geometry_flag:
yrange = [0, 303]
xrange = [0, 255]
else:
yrange = [0, 255]
xrange = [0, 303]
self.ui.error_label.setVisible(False)
self.ui.error_label.setPalette(palette)
self.ui.peak1_label.setVisible(False)
self.ui.peak1_label.setPalette(palette)
self.ui.peak2_label.setVisible(False)
self.ui.peak2_label.setPalette(palette)
self.ui.back1_label.setVisible(False)
self.ui.back1_label.setPalette(palette)
self.ui.back2_label.setVisible(False)
self.ui.back2_label.setPalette(palette)
def populate_widgets(self):
_data = self.data
peak = _data.peak
back = _data.back
back_flag = RefRed.utilities.str2bool(_data.back_flag)
low_res = _data.low_res
low_res_flag = RefRed.utilities.str2bool(_data.low_res_flag)
tof_auto_flag = RefRed.utilities.str2bool(_data.tof_auto_flag)
clock = _data.clocking
tof_range_auto = _data.tof_range_auto
tof_range = _data.tof_range
# make sure we are in ms
tof_range_auto_min = float(tof_range_auto[0])
tof_range_auto_max = float(tof_range_auto[1])
if tof_range_auto_min > 1000:
tof_range_auto_min /= 1000.
tof_range_auto_max /= 1000.
self.auto_max_tof = tof_range_auto_max
self.auto_min_tof = tof_range_auto_min
tof_range_manual_min = float(tof_range[0])
tof_range_manual_max = float(tof_range[1])
if tof_range_manual_min > 1000:
tof_range_manual_min /= 1000.
tof_range_manual_max /= 1000.
self.manual_max_tof = tof_range_manual_max
self.manual_min_tof = tof_range_manual_min
if not tof_auto_flag:
self.ui.tof_auto_flag.setChecked(False)
self.ui.tof_manual_flag.setChecked(True)
self.manual_auto_tof_clicked()
self.ui.peak1.setValue(int(peak[0]))
self.ui.peak2.setValue(int(peak[1]))
self.ui.back1.setValue(int(back[0]))
self.ui.back2.setValue(int(back[1]))
self.activate_or_not_back_widgets(back_flag)
self.ui.low_res1.setValue(int(low_res[0]))
self.ui.low_res2.setValue(int(low_res[1]))
self.activate_or_not_low_res_widgets(low_res_flag)
if not self.is_data:
self.ui.clocking_box.setVisible(False)
self.ui.clock1.setValue(int(clock[0]))
self.ui.clock2.setValue(int(clock[1]))
def activate_or_not_back_widgets(self, back_flag):
self.ui.back_flag.setChecked(back_flag)
self.ui.back1.setEnabled(back_flag)
self.ui.back2.setEnabled(back_flag)
self.check_peak_back_input_validity()
self.update_plots()
def activate_or_not_low_res_widgets(self, low_res_flag):
self.ui.low_res1.setEnabled(low_res_flag)
self.ui.low_res2.setEnabled(low_res_flag)
self.ui.low_res1_label.setEnabled(low_res_flag)
self.ui.low_res2_label.setEnabled(low_res_flag)
self.update_detector_tab_plot()
def sort_peak_back_input(self):
peak1 = self.ui.peak1.value()
peak2 = self.ui.peak2.value()
peak_min = min([peak1, peak2])
peak_max = max([peak1, peak2])
if peak_min != peak1:
self.ui.peak1.setValue(peak2)
self.ui.peak2.setValue(peak1)
back1 = self.ui.back1.value()
back2 = self.ui.back2.value()
back_min = min([back1, back2])
back_max = max([back1, back2])
if back_min != back1:
self.ui.back1.setValue(back2)
self.ui.back2.setValue(back1)
def update_plots(self):
self.update_pixel_vs_tof_tab_plot()
self.update_detector_tab_plot()
def manual_input_peak1(self):
self.sort_and_check_widgets()
self.update_plots()
def manual_input_peak2(self):
self.sort_and_check_widgets()
self.update_plots()
def manual_input_back1(self):
self.sort_and_check_widgets()
self.update_plots()
def manual_input_back2(self):
self.sort_and_check_widgets()
self.update_plots()
def sort_and_check_widgets(self):
self.sort_peak_back_input()
self.check_peak_back_input_validity()
def clock_spinbox(self):
self.update_plots()
def check_peak_back_input_validity(self):
peak1 = self.ui.peak1.value()
peak2 = self.ui.peak2.value()
back1 = self.ui.back1.value()
back2 = self.ui.back2.value()
_show_widgets_1 = False
_show_widgets_2 = False
if self.ui.back_flag.isChecked():
if back1 > peak1:
_show_widgets_1 = True
if back2 < peak2:
_show_widgets_2 = True
self.ui.back1_label.setVisible(_show_widgets_1)
self.ui.peak1_label.setVisible(_show_widgets_1)
self.ui.back2_label.setVisible(_show_widgets_2)
self.ui.peak2_label.setVisible(_show_widgets_2)
self.ui.error_label.setVisible(_show_widgets_1 or _show_widgets_2)
def manual_input_of_low_res_field(self):
value1 = self.ui.low_res1.value()
value2 = self.ui.low_res2.value()
value_min = min([value1, value2])
value_max = max([value1, value2])
self.ui.low_res1.setValue(value_min)
self.ui.low_res2.setValue(value_max)
self.ui.detector_plot()
def manual_input_of_tof_field(self):
tof1 = float(self.ui.tof_from.text())
tof2 = float(self.ui.tof_to.text())
tof_min = min([tof1, tof2])
tof_max = max([tof1, tof2])
str_tof_min = ("%.2f" % tof_min)
str_tof_max = ("%.2f" % tof_max)
self.ui.tof_from.setText(str_tof_min)
self.ui.tof_to.setText(str_tof_max)
self.manual_min_tof = tof_min
self.manual_max_tof = tof_max
self.update_pixel_vs_tof_tab_plot()
def manual_auto_tof_clicked(self):
isManualChecked = self.ui.tof_manual_flag.isChecked()
self.activate_tof_widgets(isManualChecked)
if isManualChecked:
_from_value = "%.2f" % self.manual_min_tof
_to_value = "%.2f" % self.manual_max_tof
else:
_from_value = "%.2f" % self.auto_min_tof
_to_value = "%.2f" % self.auto_max_tof
self.ui.tof_from.setText(_from_value)
self.ui.tof_to.setText(_to_value)
def activate_tof_widgets(self, status):
self.ui.tof_from.setEnabled(status)
self.ui.tof_to.setEnabled(status)
self.ui.tof_from_label.setEnabled(status)
self.ui.tof_to_label.setEnabled(status)
self.ui.tof_from_units.setEnabled(status)
self.ui.tof_to_units.setEnabled(status)
self.update_pixel_vs_tof_tab_plot()
def closeEvent(self, event=None):
[lowres1, lowres2, lowresFlag] = self.retrieveLowRes()
[tof1, tof2, peak1, peak2, back1, back2,
backFlag] = self.retrieveTofPeakBack()
tof_auto_switch = self.ui.tof_auto_flag.isChecked()
big_table_data = self.parent.big_table_data
_data = big_table_data[self.row, self.col]
_data.peak = [str(peak1), str(peak2)]
_data.back = [str(back1), str(back2)]
_data.back_flag = backFlag
_data.tof_range_auto = [
self.auto_min_tof * 1000, self.auto_max_tof * 1000
]
_data.tof_range = [
self.manual_min_tof * 1000, self.manual_max_tof * 1000
]
_data.tof_range_manual = _data.tof_range
_data.tof_range_auto_flag = tof_auto_switch
_data.tof_auto_flag = tof_auto_switch
_data.low_res = [str(lowres1), str(lowres2)]
_data.low_res_flag = lowresFlag
big_table_data[self.row, self.col] = _data
if self.is_row_with_highest_q:
_clock1 = self.ui.clock1.value()
_clock2 = self.ui.clock2.value()
_clocking = [str(_clock1), str(_clock2)]
_data = big_table_data[0, 0]
index = 0
while (_data is not None):
_data.clocking = _clocking
big_table_data[index, 0] = _data
_data = big_table_data[index + 1, 0]
index += 1
self.parent.ui.dataPrimFromValue.setValue(_clock1)
self.parent.ui.dataPrimToValue.setValue(_clock2)
self.parent.big_table_data = big_table_data
# tof
o_gui_utility = GuiUtility(parent=self.parent)
o_gui_utility.set_auto_tof_range_radio_button(status=tof_auto_switch)
self.parent.ui.TOFmanualFromValue.setText("%.2f" % tof1)
self.parent.ui.TOFmanualToValue.setText("%.2f" % tof2)
if self.data_type == 'data':
self.parent.ui.dataPeakFromValue.setValue(peak1)
self.parent.ui.dataPeakToValue.setValue(peak2)
self.parent.ui.dataBackFromValue.setValue(back1)
self.parent.ui.dataBackToValue.setValue(back2)
self.parent.ui.dataBackgroundFlag.setChecked(backFlag)
#self.parent.data_peak_and_back_validation(False)
self.parent.ui.dataBackFromLabel.setEnabled(backFlag)
self.parent.ui.dataBackFromValue.setEnabled(backFlag)
self.parent.ui.dataBackToLabel.setEnabled(backFlag)
self.parent.ui.dataBackToValue.setEnabled(backFlag)
self.parent.ui.dataLowResFromValue.setValue(lowres1)
self.parent.ui.dataLowResToValue.setValue(lowres2)
self.parent.ui.dataLowResFromLabel.setEnabled(lowresFlag)
self.parent.ui.dataLowResFromValue.setEnabled(lowresFlag)
self.parent.ui.dataLowResToLabel.setEnabled(lowresFlag)
self.parent.ui.dataLowResToValue.setEnabled(lowresFlag)
else:
self.parent.ui.normPeakFromValue.setValue(peak1)
self.parent.ui.normPeakToValue.setValue(peak2)
self.parent.ui.normBackFromValue.setValue(back1)
self.parent.ui.normBackToValue.setValue(back2)
self.parent.ui.normBackgroundFlag.setChecked(backFlag)
#self.parent.norm_peak_and_back_validation(False)
self.parent.ui.normBackFromLabel.setEnabled(backFlag)
self.parent.ui.normBackFromValue.setEnabled(backFlag)
self.parent.ui.normBackToLabel.setEnabled(backFlag)
self.parent.ui.normBackToValue.setEnabled(backFlag)
self.parent.ui.normLowResFromValue.setValue(lowres1)
self.parent.ui.normLowResToValue.setValue(lowres2)
self.parent.ui.normLowResFromLabel.setEnabled(lowresFlag)
self.parent.ui.normLowResFromValue.setEnabled(lowresFlag)
self.parent.ui.normLowResToLabel.setEnabled(lowresFlag)
self.parent.ui.normLowResToValue.setEnabled(lowresFlag)
DisplayPlots(parent=self.parent,
row=self.row,
is_data=self.is_data,
plot_yt=True,
plot_yi=True,
plot_it=True,
plot_ix=True,
refresh_reduction_table=False)
if not tof_auto_switch:
o_auto_tof_range = AutoTofRangeRadioButtonHandler(
parent=self.parent)
o_auto_tof_range.setup()
o_auto_tof_range.line_edit_validation()
class PopupPlot2d(QDialog):
parent = None
_open_instances = []
data = None
data_type = 'data'
row = 0
manual_min_tof = None
manual_max_tof = None
auto_min_tof = None
auto_max_tof = None
def __init__(self, parent = None,
data_type = 'data',
data = None,
row = 0):
self.parent = parent
self.data = data
self.data_type = data_type
self.row = row
self.col = 0 if data_type == 'data' else 1
self.is_data = True if data_type == 'data' else False
self.is_row_with_highest_q = self.is_row_with_higest_q()
QDialog.__init__(self, parent=parent)
self.setWindowModality(False)
self._open_instances.append(self)
self.ui = UiPlot()
self.ui.setupUi(self)
self.setWindowTitle('Detector and Pixel vs TOF views')
self.init_gui()
self.populate_widgets()
self.update_detector_tab_plot()
self.update_pixel_vs_tof_tab_plot()
self.ui.y_pixel_vs_tof_plot.leaveFigure.connect(self.leave_figure_plot)
self.ui.y_pixel_vs_tof_plot.toolbar.homeClicked.connect(self.home_clicked_plot)
self.ui.y_pixel_vs_tof_plot.toolbar.exportClicked.connect(self.export_yt)
self.ui.detector_plot.leaveFigure.connect(self.leave_figure_detector_plot)
self.ui.detector_plot.toolbar.homeClicked.connect(self.home_clicked_detector_plot)
self.ui.detector_plot.toolbar.exportClicked.connect(self.export_detector_view)
def is_row_with_higest_q(self):
o_gui_utility = GuiUtility(parent = self.parent)
return o_gui_utility.is_row_with_highest_q()
def export_yt(self):
_active_data = self.data
run_number = _active_data.run_number
default_filename = 'REFL_' + run_number + '_2dPxVsTof.txt'
path = self.parent.path_ascii
default_filename = path + '/' + default_filename
filename = QFileDialog.getSaveFileName(self, 'Create 2D Pixel VS TOF', default_filename)
if str(filename).strip() == '':
# info('User Canceled Outpout ASCII')
return
self.parent.path_ascii = os.path.dirname(filename)
image = _active_data.ytofdata
RefRed.utilities.output_2d_ascii_file(filename, image)
def export_detector_view(self):
_active_data = self.data
run_number = _active_data.run_number
default_filename = 'REFL_' + run_number + '_2dDetectorView.txt'
path = self.parent.path_ascii
default_filename = path + '/' + default_filename
filename = QFileDialog.getSaveFileName(self, 'Create 2D Y Pixel VS X Pixel (Detector View)', default_filename)
if str(filename).strip() == '':
# info('User Canceled Outpout ASCII')
return
self.parent.path_ascii = os.path.dirname(filename)
image = _active_data.xydata
RefRed.utilities.output_2d_ascii_file(filename, image)
def leave_figure_plot(self):
[xmin, xmax]= self.ui.y_pixel_vs_tof_plot.canvas.ax.xaxis.get_view_interval()
[ymin, ymax]= self.ui.y_pixel_vs_tof_plot.canvas.ax.yaxis.get_view_interval()
self.ui.y_pixel_vs_tof_plot.canvas.ax.xaxis.set_data_interval(xmin, xmax)
self.ui.y_pixel_vs_tof_plot.canvas.ax.yaxis.set_data_interval(ymin, ymax)
self.ui.y_pixel_vs_tof_plot.draw()
self.data.all_plot_axis.yt_view_interval = [xmin, xmax, ymin, ymax]
def home_clicked_plot(self):
[xmin,xmax,ymin,ymax] = self.data.all_plot_axis.yt_data_interval
self.ui.y_pixel_vs_tof_plot.canvas.ax.set_xlim([xmin,xmax])
self.ui.y_pixel_vs_tof_plot.canvas.ax.set_ylim([ymin,ymax])
self.ui.y_pixel_vs_tof_plot.draw()
def leave_figure_detector_plot(self):
[xmin, xmax]= self.ui.detector_plot.canvas.ax.xaxis.get_view_interval()
[ymin, ymax]= self.ui.detector_plot.canvas.ax.yaxis.get_view_interval()
self.ui.detector_plot.canvas.ax.xaxis.set_data_interval(xmin, xmax)
self.ui.detector_plot.canvas.ax.yaxis.set_data_interval(ymin, ymax)
self.ui.detector_plot.draw()
self.data.all_plot_axis.detector_view_interval = [xmin, xmax, ymin, ymax]
def home_clicked_detector_plot(self):
[xmin,xmax,ymin,ymax] = self.data.all_plot_axis.detector_data_interval
self.ui.detector_plot.canvas.ax.set_xlim([xmin,xmax])
self.ui.detector_plot.canvas.ax.set_ylim([ymin,ymax])
self.ui.detector_plot.draw()
def get_clocking_values(self):
clock1 = self.ui.clock1.value()
clock2 = self.ui.clock2.value()
return [clock1, clock2]
def update_detector_tab_plot(self):
self.ui.detector_plot.clear()
xydata = self.data.xydata
self.ui.detector_plot.draw()
[ymax,xmax] = xydata.shape
self.ui.detector_plot.imshow(xydata, log=True, aspect='auto',
origin='lower', extent=[0,xmax,0,ymax])
self.ui.detector_plot.set_xlabel(u'x (pixel)')
self.ui.detector_plot.set_ylabel(u'y (pixel)')
[lowres1, lowres2, lowresFlag, peak1, peak2, back1, back2, backFlag] = self.retrieveLowResPeakBack()
[clock1, clock2] = self.get_clocking_values()
if lowresFlag:
lr1 = self.ui.detector_plot.canvas.ax.axvline(lowres1,
color = RefRed.colors.LOWRESOLUTION_SELECTION_COLOR)
lr2 = self.ui.detector_plot.canvas.ax.axvline(lowres2,
color = RefRed.colors.LOWRESOLUTION_SELECTION_COLOR)
p1 = self.ui.detector_plot.canvas.ax.axhline(peak1,
color = RefRed.colors.PEAK_SELECTION_COLOR)
p2 = self.ui.detector_plot.canvas.ax.axhline(peak2,
color = RefRed.colors.PEAK_SELECTION_COLOR)
c1 = self.ui.detector_plot.canvas.ax.axhline(clock1,
color = RefRed.colors.CLOCKING_SELECTION_COLOR)
c1 = self.ui.detector_plot.canvas.ax.axhline(clock2,
color = RefRed.colors.CLOCKING_SELECTION_COLOR)
if backFlag:
b1 = self.ui.detector_plot.canvas.ax.axhline(back1,
color = RefRed.colors.BACK_SELECTION_COLOR)
b2 = self.ui.detector_plot.canvas.ax.axhline(back2,
color = RefRed.colors.BACK_SELECTION_COLOR)
if self.data.all_plot_axis.detector_data_interval is None:
self.ui.detector_plot.draw()
[xmin,xmax] = self.ui.detector_plot.canvas.ax.xaxis.get_view_interval()
[ymin,ymax] = self.ui.detector_plot.canvas.ax.yaxis.get_view_interval()
self.data.all_plot_axis.detector_data_interval = [xmin, xmax, ymin, ymax]
self.data.all_plot_axis.detector_view_interval = [xmin, xmax, ymin, ymax]
self.ui.detector_plot.toolbar.home_settings = [xmin, xmax, ymin, ymax]
else:
[xmin, xmax, ymin, ymax] = self.data.all_plot_axis.detector_view_interval
self.ui.detector_plot.canvas.ax.set_xlim([xmin, xmax])
self.ui.detector_plot.canvas.ax.set_ylim([ymin, ymax])
self.ui.detector_plot.draw()
def update_pixel_vs_tof_tab_plot(self):
ytof = self.data.ytofdata
tof_axis = self.data.tof_axis_auto_with_margin
tof_from = tof_axis[0]
tof_to = tof_axis[-1]
if tof_from > 1000: # stay in ms
tof_from /= 1000.
tof_to /= 1000.
pixel_from = 0
pixel_to = self.data.y.shape[0]-1
self.ui.y_pixel_vs_tof_plot.clear()
self.ui.y_pixel_vs_tof_plot.imshow(ytof, log=True, aspect='auto',
origin='lower', extent=[tof_from, tof_to, pixel_from, pixel_to])
self.ui.y_pixel_vs_tof_plot.set_xlabel(u't (ms)')
self.ui.y_pixel_vs_tof_plot.set_ylabel(u'y (pixel)')
[tmin,tmax,peak1,peak2,back1,back2,backFlag] = self.retrieveTofPeakBack()
[clock1, clock2] = self.get_clocking_values()
t1 = self.ui.y_pixel_vs_tof_plot.canvas.ax.axvline(tmin,
color = RefRed.colors.TOF_SELECTION_COLOR)
t2 = self.ui.y_pixel_vs_tof_plot.canvas.ax.axvline(tmax,
color = RefRed.colors.TOF_SELECTION_COLOR)
p1 = self.ui.y_pixel_vs_tof_plot.canvas.ax.axhline(peak1,
color = RefRed.colors.PEAK_SELECTION_COLOR)
p2 = self.ui.y_pixel_vs_tof_plot.canvas.ax.axhline(peak2,
color = RefRed.colors.PEAK_SELECTION_COLOR)
c1 = self.ui.y_pixel_vs_tof_plot.canvas.ax.axhline(clock1,
color = RefRed.colors.CLOCKING_SELECTION_COLOR)
c1 = self.ui.y_pixel_vs_tof_plot.canvas.ax.axhline(clock2,
color = RefRed.colors.CLOCKING_SELECTION_COLOR)
if backFlag:
b1 = self.ui.y_pixel_vs_tof_plot.canvas.ax.axhline(back1,
color = RefRed.colors.BACK_SELECTION_COLOR)
b2 = self.ui.y_pixel_vs_tof_plot.canvas.ax.axhline(back2,
color = RefRed.colors.BACK_SELECTION_COLOR)
if self.data.all_plot_axis.yt_data_interval is None:
self.ui.y_pixel_vs_tof_plot.canvas.ax.set_ylim(0,pixel_to)
self.ui.y_pixel_vs_tof_plot.canvas.draw()
[xmin,xmax] = self.ui.y_pixel_vs_tof_plot.canvas.ax.xaxis.get_view_interval()
[ymin,ymax] = self.ui.y_pixel_vs_tof_plot.canvas.ax.yaxis.get_view_interval()
self.data.all_plot_axis.yt_data_interval = [xmin, xmax, ymin, ymax]
self.data.all_plot_axis.yt_view_interval = [xmin, xmax, ymin, ymax]
self.ui.y_pixel_vs_tof_plot.toolbar.home_settings = [xmin, xmax, ymin, ymax]
else:
[xmin,xmax,ymin,ymax] = self.data.all_plot_axis.yt_view_interval
self.ui.y_pixel_vs_tof_plot.canvas.ax.set_xlim([xmin,xmax])
self.ui.y_pixel_vs_tof_plot.canvas.ax.set_ylim([ymin,ymax])
self.ui.y_pixel_vs_tof_plot.draw()
def retrieveTofPeakBack(self):
tmin = float(self.ui.tof_from.text())
tmax = float(self.ui.tof_to.text())
[peak1,peak2,back1,back2, backFlag] = self.retrievePeakBack()
return [tmin, tmax, peak1, peak2, back1, back2, backFlag]
def retrieveLowResPeakBack(self):
lowres1 = self.ui.low_res1.value()
lowres2 = self.ui.low_res2.value()
lowresFlag = self.ui.low_res_flag.isChecked()
[peak1,peak2,back1,back2, backFlag] = self.retrievePeakBack()
return [lowres1,lowres2,lowresFlag,peak1,peak2,back1,back2,backFlag]
def retrieveLowRes(self):
lowres1 = self.ui.low_res1.value()
lowres2 = self.ui.low_res2.value()
lowresFlag = self.ui.low_res_flag.isChecked()
return [lowres1, lowres2, lowresFlag]
def retrievePeakBack(self):
peak1 = self.ui.peak1.value()
peak2 = self.ui.peak2.value()
back1 = self.ui.back1.value()
back2 = self.ui.back2.value()
backFlag = self.ui.back_flag.isChecked()
return [peak1, peak2, back1, back2, backFlag]
def init_gui(self):
palette = QPalette()
palette.setColor(QPalette.Foreground, Qt.red)
if self.data.new_detector_geometry_flag:
yrange = [0,303]
xrange = [0,255]
else:
yrange = [0,255]
xrange = [0,303]
self.ui.error_label.setVisible(False)
self.ui.error_label.setPalette(palette)
self.ui.peak1_label.setVisible(False)
self.ui.peak1_label.setPalette(palette)
self.ui.peak2_label.setVisible(False)
self.ui.peak2_label.setPalette(palette)
self.ui.back1_label.setVisible(False)
self.ui.back1_label.setPalette(palette)
self.ui.back2_label.setVisible(False)
self.ui.back2_label.setPalette(palette)
def populate_widgets(self):
_data = self.data
peak = _data.peak
back = _data.back
back_flag = RefRed.utilities.str2bool(_data.back_flag)
low_res = _data.low_res
low_res_flag = RefRed.utilities.str2bool(_data.low_res_flag)
tof_auto_flag = RefRed.utilities.str2bool(_data.tof_auto_flag)
clock = _data.clocking
tof_range_auto = _data.tof_range_auto
tof_range = _data.tof_range
# make sure we are in ms
tof_range_auto_min = float(tof_range_auto[0])
tof_range_auto_max = float(tof_range_auto[1])
if tof_range_auto_min > 1000:
tof_range_auto_min /= 1000.
tof_range_auto_max /= 1000.
self.auto_max_tof = tof_range_auto_max
self.auto_min_tof = tof_range_auto_min
tof_range_manual_min = float(tof_range[0])
tof_range_manual_max = float(tof_range[1])
if tof_range_manual_min > 1000:
tof_range_manual_min /= 1000.
tof_range_manual_max /= 1000.
self.manual_max_tof = tof_range_manual_max
self.manual_min_tof = tof_range_manual_min
if not tof_auto_flag:
self.ui.tof_auto_flag.setChecked(False)
self.ui.tof_manual_flag.setChecked(True)
self.manual_auto_tof_clicked()
self.ui.peak1.setValue(int(peak[0]))
self.ui.peak2.setValue(int(peak[1]))
self.ui.back1.setValue(int(back[0]))
self.ui.back2.setValue(int(back[1]))
self.activate_or_not_back_widgets(back_flag)
self.ui.low_res1.setValue(int(low_res[0]))
self.ui.low_res2.setValue(int(low_res[1]))
self.activate_or_not_low_res_widgets(low_res_flag)
self.ui.clock1.setValue(int(clock[0]))
self.ui.clock2.setValue(int(clock[1]))
def activate_or_not_back_widgets(self, back_flag):
self.ui.back_flag.setChecked(back_flag)
self.ui.back1.setEnabled(back_flag)
self.ui.back2.setEnabled(back_flag)
self.check_peak_back_input_validity()
self.update_plots()
def activate_or_not_low_res_widgets(self, low_res_flag):
self.ui.low_res1.setEnabled(low_res_flag)
self.ui.low_res2.setEnabled(low_res_flag)
self.ui.low_res1_label.setEnabled(low_res_flag)
self.ui.low_res2_label.setEnabled(low_res_flag)
self.update_detector_tab_plot()
def sort_peak_back_input(self):
peak1 = self.ui.peak1.value()
peak2 = self.ui.peak2.value()
peak_min = min([peak1, peak2])
peak_max = max([peak1, peak2])
if peak_min != peak1:
self.ui.peak1.setValue(peak2)
self.ui.peak2.setValue(peak1)
back1 = self.ui.back1.value()
back2 = self.ui.back2.value()
back_min = min([back1, back2])
back_max = max([back1, back2])
if back_min != back1:
self.ui.back1.setValue(back2)
self.ui.back2.setValue(back1)
def update_plots(self):
self.update_pixel_vs_tof_tab_plot()
self.update_detector_tab_plot()
def manual_input_peak1(self):
self.sort_and_check_widgets()
self.update_plots()
def manual_input_peak2(self):
self.sort_and_check_widgets()
self.update_plots()
def manual_input_back1(self):
self.sort_and_check_widgets()
self.update_plots()
def manual_input_back2(self):
self.sort_and_check_widgets()
self.update_plots()
def sort_and_check_widgets(self):
self.sort_peak_back_input()
self.check_peak_back_input_validity()
def clock_spinbox(self):
self.update_plots()
def check_peak_back_input_validity(self):
peak1 = self.ui.peak1.value()
peak2 = self.ui.peak2.value()
back1 = self.ui.back1.value()
back2 = self.ui.back2.value()
_show_widgets_1 = False
_show_widgets_2 = False
if self.ui.back_flag.isChecked():
if back1 > peak1:
_show_widgets_1 = True
if back2 < peak2:
_show_widgets_2 = True
self.ui.back1_label.setVisible(_show_widgets_1)
self.ui.peak1_label.setVisible(_show_widgets_1)
self.ui.back2_label.setVisible(_show_widgets_2)
self.ui.peak2_label.setVisible(_show_widgets_2)
self.ui.error_label.setVisible(_show_widgets_1 or _show_widgets_2)
def manual_input_of_low_res_field(self):
value1 = self.ui.low_res1.value()
value2 = self.ui.low_res2.value()
value_min = min([value1,value2])
value_max = max([value1,value2])
self.ui.low_res1.setValue(value_min)
self.ui.low_res2.setValue(value_max)
self.ui.detector_plot()
def manual_input_of_tof_field(self):
tof1 = float(self.ui.tof_from.text())
tof2 = float(self.ui.tof_to.text())
tof_min = min([tof1, tof2])
tof_max = max([tof1, tof2])
str_tof_min = ("%.2f"%tof_min)
str_tof_max = ("%.2f"%tof_max)
self.ui.tof_from.setText(str_tof_min)
self.ui.tof_to.setText(str_tof_max)
self.manual_min_tof = tof_min
self.manual_max_tof = tof_max
self.update_pixel_vs_tof_tab_plot()
def manual_auto_tof_clicked(self):
isManualChecked = self.ui.tof_manual_flag.isChecked()
self.activate_tof_widgets(isManualChecked)
if isManualChecked:
_from_value = "%.2f"%self.manual_min_tof
_to_value = "%.2f"%self.manual_max_tof
else:
_from_value = "%.2f"%self.auto_min_tof
_to_value = "%.2f"%self.auto_max_tof
self.ui.tof_from.setText(_from_value)
self.ui.tof_to.setText(_to_value)
def activate_tof_widgets(self, status):
self.ui.tof_from.setEnabled(status)
self.ui.tof_to.setEnabled(status)
self.ui.tof_from_label.setEnabled(status)
self.ui.tof_to_label.setEnabled(status)
self.ui.tof_from_units.setEnabled(status)
self.ui.tof_to_units.setEnabled(status)
self.update_pixel_vs_tof_tab_plot()
def closeEvent(self, event = None):
[lowres1, lowres2, lowresFlag] = self.retrieveLowRes()
[tof1, tof2, peak1, peak2, back1, back2, backFlag]= self.retrieveTofPeakBack()
tof_auto_switch = self.ui.tof_auto_flag.isChecked()
big_table_data = self.parent.big_table_data
_data = big_table_data[self.row, self.col]
_data.peak = [str(peak1), str(peak2)]
_data.back = [str(back1), str(back2)]
_data.back_flag = backFlag
_data.tof_range_auto = [self.auto_min_tof * 1000, self.auto_max_tof * 1000]
_data.tof_range = [self.manual_min_tof * 1000, self.manual_max_tof * 1000]
_data.tof_range_manual = _data.tof_range
_data.tof_range_auto_flag = tof_auto_switch
_data.tof_auto_flag = tof_auto_switch
_data.low_res = [str(lowres1), str(lowres2)]
_data.low_res_flag = lowresFlag
big_table_data[self.row, self.col] = _data
if self.is_row_with_highest_q:
_clock1 = self.ui.clock1.value()
_clock2 = self.ui.clock2.value()
_clocking = [str(_clock1), str(_clock2)]
_data = big_table_data[0, 0]
index = 0
while (_data is not None):
_data.clocking = _clocking
big_table_data[index, 0] = _data
_data = big_table_data[index+1, 0]
index += 1
self.parent.ui.dataPrimFromValue.setValue(_clock1)
self.parent.ui.dataPrimToValue.setValue(_clock2)
self.parent.big_table_data = big_table_data
# tof
o_gui_utility = GuiUtility(parent = self.parent)
o_gui_utility.set_auto_tof_range_radio_button(status = tof_auto_switch)
self.parent.ui.TOFmanualFromValue.setText("%.2f" %tof1)
self.parent.ui.TOFmanualToValue.setText("%.2f" %tof2)
if self.data_type == 'data':
self.parent.ui.dataPeakFromValue.setValue(peak1)
self.parent.ui.dataPeakToValue.setValue(peak2)
self.parent.ui.dataBackFromValue.setValue(back1)
self.parent.ui.dataBackToValue.setValue(back2)
self.parent.ui.dataBackgroundFlag.setChecked(backFlag)
#self.parent.data_peak_and_back_validation(False)
self.parent.ui.dataBackFromLabel.setEnabled(backFlag)
self.parent.ui.dataBackFromValue.setEnabled(backFlag)
self.parent.ui.dataBackToLabel.setEnabled(backFlag)
self.parent.ui.dataBackToValue.setEnabled(backFlag)
self.parent.ui.dataLowResFromValue.setValue(lowres1)
self.parent.ui.dataLowResToValue.setValue(lowres2)
self.parent.ui.dataLowResFromLabel.setEnabled(lowresFlag)
self.parent.ui.dataLowResFromValue.setEnabled(lowresFlag)
self.parent.ui.dataLowResToLabel.setEnabled(lowresFlag)
self.parent.ui.dataLowResToValue.setEnabled(lowresFlag)
else:
self.parent.ui.normPeakFromValue.setValue(peak1)
self.parent.ui.normPeakToValue.setValue(peak2)
self.parent.ui.normBackFromValue.setValue(back1)
self.parent.ui.normBackToValue.setValue(back2)
self.parent.ui.normBackgroundFlag.setChecked(backFlag)
#self.parent.norm_peak_and_back_validation(False)
self.parent.ui.normBackFromLabel.setEnabled(backFlag)
self.parent.ui.normBackFromValue.setEnabled(backFlag)
self.parent.ui.normBackToLabel.setEnabled(backFlag)
self.parent.ui.normBackToValue.setEnabled(backFlag)
self.parent.ui.normLowResFromValue.setValue(lowres1)
self.parent.ui.normLowResToValue.setValue(lowres2)
self.parent.ui.normLowResFromLabel.setEnabled(lowresFlag)
self.parent.ui.normLowResFromValue.setEnabled(lowresFlag)
self.parent.ui.normLowResToLabel.setEnabled(lowresFlag)
self.parent.ui.normLowResToValue.setEnabled(lowresFlag)
DisplayPlots(parent = self.parent,
row = self.row,
is_data = self.is_data,
plot_yt = True,
plot_yi = True,
plot_it = True,
plot_ix = True,
refresh_reduction_table = False)
if not tof_auto_switch:
o_auto_tof_range = AutoTofRangeRadioButtonHandler(parent = self.parent)
o_auto_tof_range.setup()
o_auto_tof_range.line_edit_validation()
