def fit_multi_peaks(self):
QApplication.setOverrideCursor(Qt.WaitCursor)
_peak_range_list = [
tuple(_range) for _range in
self.parent._ui_graphicsView_fitSetup.list_peak_ranges
]
_peak_center_list = [
np.mean([left, right]) for (left, right) in _peak_range_list
]
_peak_tag_list = [
"peak{}".format(_index)
for _index, _ in enumerate(_peak_center_list)
]
_peak_function_name = str(
self.parent.ui.comboBox_peakType.currentText())
_peak_xmin_list = [left for (left, _) in _peak_range_list]
_peak_xmax_list = [right for (_, right) in _peak_range_list]
# Fit peak
hd_ws = self.parent.hidra_workspace
_wavelength = hd_ws.get_wavelength(True, True)
fit_engine = PeakFitEngineFactory.getInstance(hd_ws,
_peak_function_name,
'Linear',
wavelength=_wavelength)
fit_result = fit_engine.fit_multiple_peaks(_peak_tag_list,
_peak_xmin_list,
_peak_xmax_list)
self.parent.fit_result = fit_result
self.parent.populate_fit_result_table(fit_result=fit_result)
# self.parent.update_list_of_2d_plots_axis()
o_gui = GuiUtilities(parent=self.parent)
o_gui.set_1D_2D_axis_comboboxes(with_clear=True,
fill_raw=True,
fill_fit=True)
o_gui.initialize_combobox()
o_gui.enabled_export_csv_widgets(enabled=True)
o_gui.enabled_2dplot_widgets(enabled=True)
o_plot = Plot(parent=self.parent)
o_plot.plot_2d()
QApplication.restoreOverrideCursor()
def setup_ui(self):
"""define the layout, widgets and signals"""
# promote
self.ui.graphicsView_fitResult = qt_util.promote_widget(self, self.ui.graphicsView_fitResult_frame,
GeneralDiffDataView)
self.ui.graphicsView_fitResult.setEnabled(False)
self.ui.graphicsView_fitResult.set_subplots(1, 1)
self.ui.graphicsView_plot2D = qt_util.promote_widget(self, self.ui.graphicsView_2dPlot_frame,
MplGraphicsViewContourPlot)
self.ui.tableView_fitSummary = qt_util.promote_widget(self, self.ui.tableView_fitSummary_frame,
FitResultTable)
self._promote_peak_fit_setup()
self._init_widgets()
# set up handling
self.ui.pushButton_browseHDF.clicked.connect(self.browse_hdf)
self.ui.lineEdit_listSubRuns.returnPressed.connect(self.plot_diff_data)
self.ui.pushButton_FitPeaks.clicked.connect(self.fit_peaks)
self.ui.horizontalScrollBar_SubRuns.valueChanged.connect(self.plot_scan)
self.ui.radioButton_individualSubRuns.clicked.connect(self.individual_sub_runs)
self.ui.radioButton_listSubRuns.clicked.connect(self.list_sub_runs)
self.ui.actionQuit.triggered.connect(self.do_quit)
self.ui.actionSave.triggered.connect(self.save)
self.ui.actionSaveAs.triggered.connect(self.save_as)
self.ui.actionAdvanced_Peak_Fit_Settings.triggered.connect(self.do_launch_adv_fit)
self.ui.pushButton_exportCSV.clicked.connect(self.export_csv)
self.ui.actionQuick_Fit_Result_Check.triggered.connect(self.do_make_movie)
self.ui.lineEdit_subruns_2dplot.returnPressed.connect(self.list_subruns_2dplot_returned)
self.ui.lineEdit_subruns_2dplot.textChanged.connect(self.list_subruns_2dplot_changed)
self.ui.pushButton_save_peak_range.clicked.connect(self.clicked_save_peak_range)
self.ui.pushButton_load_peak_range.clicked.connect(self.clicked_load_peak_range)
self.ui.tableView_fitSummary.itemSelectionChanged.connect(self.fit_result_table_selection_changed)
self.ui.radioButton_fit_value.clicked.connect(self.fit_table_radio_buttons)
self.ui.radioButton_fit_error.clicked.connect(self.fit_table_radio_buttons)
self.ui.spinBox_peak_index.valueChanged.connect(self.fit_table_radio_buttons)
self.ui.comboBox_xaxisNames.currentIndexChanged.connect(self.axis_1d_changed)
self.ui.comboBox_yaxisNames.currentIndexChanged.connect(self.axis_1d_changed)
self.ui.plot1d_xaxis_peak_label_comboBox.currentIndexChanged.connect(self.axis_1d_changed)
self.ui.plot1d_yaxis_peak_label_comboBox.currentIndexChanged.connect(self.axis_1d_changed)
self.ui.comboBox_xaxisNames_2dplot.currentIndexChanged.connect(self.axis_2d_changed)
self.ui.comboBox_yaxisNames_2dplot.currentIndexChanged.connect(self.axis_2d_changed)
self.ui.comboBox_zaxisNames_2dplot.currentIndexChanged.connect(self.axis_2d_changed)
self.ui.plot2d_xaxis_peak_label_comboBox.currentIndexChanged.connect(self.axis_2d_changed)
self.ui.plot2d_yaxis_peak_label_comboBox.currentIndexChanged.connect(self.axis_2d_changed)
self.ui.plot2d_zaxis_peak_label_comboBox.currentIndexChanged.connect(self.axis_2d_changed)
self.ui.radioButton_contour.clicked.connect(self.axis_2d_changed)
self.ui.radioButton_3dline.clicked.connect(self.axis_2d_changed)
self.ui.radioButton_3dscatter.clicked.connect(self.axis_2d_changed)
self.ui.peak_range_table.cellChanged.connect(self.peak_range_table_changed)
# tracker for sample log names and peak parameter names
self._sample_log_name_set = set()
self._function_param_name_set = set()
# mutexes
self._sample_log_names_mutex = False
# TODO - 20181124 - New GUI parameters (After FitPeaks)
# checkBox_showFitError
# checkBox_showFitValue
# others
# TODO - 20181124 - Make this table's column flexible!
self.ui.tableView_fitSummary.setup(peak_param_names=list())
o_gui = GuiUtilities(parent=self)
o_gui.enabled_fitting_widgets(False)
o_gui.enabled_1dplot_widgets(False)
o_gui.check_axis1d_status()
o_gui.enabled_2dplot_widgets(False)
o_gui.check_axis2d_status()
o_gui.make_visible_listsubruns_warning(False)
o_gui.enabled_export_csv_widgets(False)
o_gui.enabled_peak_ranges_widgets(False)
o_gui.enabled_save_peak_range_widget(False)
o_gui.enabled_sub_runs_interation_widgets(False)
# for debugging only
self.ui.radioButton_contour.setEnabled(False)
self.ui.radioButton_3dline.setEnabled(False)