interim save

This commit is contained in:
Dominik Demuth 2023-07-05 17:35:31 +02:00
parent bc946e1027
commit 8de4a0cbd3
3 changed files with 49 additions and 27 deletions

View File

@ -15,14 +15,15 @@ from nmreval.data import DSC, Points
class TgCalculator(QtWidgets.QWizard, Ui_Wizard): class TgCalculator(QtWidgets.QWizard, Ui_Wizard):
newTg = QtCore.pyqtSignal(dict) newData = QtCore.pyqtSignal(dict, str)
def __init__(self, management, parent=None): def __init__(self, management, parent=None):
super().__init__(parent=parent) super().__init__(parent=parent)
self.setupUi(self) self.setupUi(self)
self.listWidget = QListWidgetSelect(parent=self) self.listWidget = QListWidgetSelect(parent=self)
self.listWidget.setObjectName('listWidget') self.listWidget.setSelectionMode(self.listWidget.ExtendedSelection)
self.listWidget.itemChanged.connect(self.change_visibility)
self.setSideWidget(self.listWidget) self.setSideWidget(self.listWidget)
self.listWidget.setSizePolicy(QtWidgets.QSizePolicy(QtWidgets.QSizePolicy.MinimumExpanding, QtWidgets.QSizePolicy.MinimumExpanding)) self.listWidget.setSizePolicy(QtWidgets.QSizePolicy(QtWidgets.QSizePolicy.MinimumExpanding, QtWidgets.QSizePolicy.MinimumExpanding))
@ -241,9 +242,9 @@ class TgCalculator(QtWidgets.QWizard, Ui_Wizard):
ret_dic[key] = (tg_pts, line) ret_dic[key] = (tg_pts, line)
self.newTg.emit(ret_dic) self.newData.emit(ret_dic, 'tg')
@QtCore.pyqtSlot(QtWidgets.QListWidgetItem, name='on_listWidget_itemChanged') @QtCore.pyqtSlot(QtWidgets.QListWidgetItem)
def change_visibility(self, item: QtWidgets.QListWidgetItem): def change_visibility(self, item: QtWidgets.QListWidgetItem):
is_checked = bool(item.checkState()) is_checked = bool(item.checkState())
plot = self._plots[item.data(QtCore.Qt.UserRole)] plot = self._plots[item.data(QtCore.Qt.UserRole)]
@ -303,6 +304,8 @@ class TgCalculator(QtWidgets.QWizard, Ui_Wizard):
m = [] m = []
for idx in range(self.listWidget.count()): for idx in range(self.listWidget.count()):
item = self.listWidget.item(idx) item = self.listWidget.item(idx)
if item.checkState() == QtCore.Qt.Unchecked:
continue
key = item.data(QtCore.Qt.UserRole) key = item.data(QtCore.Qt.UserRole)
data, _ = self._dsc[key] data, _ = self._dsc[key]
@ -324,6 +327,25 @@ class TgCalculator(QtWidgets.QWizard, Ui_Wizard):
self._hodge[tg_type] = (plot, data, fitplots, fit) self._hodge[tg_type] = (plot, data, fitplots, fit)
@QtCore.pyqtSlot(name='on_pushButton_4_clicked')
def export_hodge(self):
ret_dic2 = {}
for cb in (self.checkBox, self.checkBox_4, self.checkBox_6, self.checkBox_5, self.checkBox_7):
if cb.isChecked():
item = cb.text().lower()
v = self._hodge.get(item)
ret_dic2[item] = v[1]
if self.new_graph_tau_check.isChecked():
ret_dic2['graph'] = self.new_graph_tau_combo.currentData()
else:
ret_dic2['graph'] = ''
print(ret_dic2)
self.newData.emit(ret_dic2, 'hodge')
def close(self) -> bool: def close(self) -> bool:
self.clear() self.clear()
return super().close() return super().close()
@ -350,13 +372,5 @@ class TgCalculator(QtWidgets.QWizard, Ui_Wizard):
ret_dic[key] = (tg_pts, line) ret_dic[key] = (tg_pts, line)
ret_dic2 = []
for i in range(self.hodge_selection.count()):
if self.hodge_selection.isChecked(i):
item = self.hodge_selection.itemText(i).lower()
v = self._hodge.get(item)
ret_dic2.append(v[1])
self.newTg.emit(ret_dic, ret_dic2, graph_id) self.newTg.emit(ret_dic, ret_dic2, graph_id)
self.close() self.close()

View File

@ -1143,7 +1143,7 @@ class NMRMainWindow(QtWidgets.QMainWindow, Ui_BaseWindow):
def show_tg_dialog(self): def show_tg_dialog(self):
if self._tg_dialog is None: if self._tg_dialog is None:
self._tg_dialog = TgCalculator(self.management, parent=self) self._tg_dialog = TgCalculator(self.management, parent=self)
self._tg_dialog.newTg.connect(self.management.addTg) self._tg_dialog.newData.connect(self.management.addTg)
else: else:
self._tg_dialog() self._tg_dialog()
self._tg_dialog.show() self._tg_dialog.show()

View File

@ -773,24 +773,32 @@ class UpperManagement(QtCore.QObject):
self.newData.emit(new_data, self.current_graph) self.newData.emit(new_data, self.current_graph)
@QtCore.pyqtSlot(dict) @QtCore.pyqtSlot(dict, str)
def addTg(self, dic1: dict): def addTg(self, dic1: dict, dtype: str):
for k, (tg, lines) in dic1.items(): if dtype == 'tg':
p: ExperimentContainer = self[k] for k, (tg, lines) in dic1.items():
col = p.plot_real.linecolor p: ExperimentContainer = self[k]
col = p.plot_real.linecolor
tg_data_id = [] tg_data_id = []
if tg is not None: if tg is not None:
tg_data_id.append(self.add(tg, color=col)) tg_data_id.append(self.add(tg, color=col))
for line in lines: for line in lines:
set_id = self.add(line, color=col) set_id = self.add(line, color=col)
self[set_id].setLine(style=LineStyle.Dashed) self[set_id].setLine(style=LineStyle.Dashed)
self[set_id].setSymbol(symbol=SymbolStyle.No) self[set_id].setSymbol(symbol=SymbolStyle.No)
tg_data_id.append(set_id) tg_data_id.append(set_id)
self.newData.emit(tg_data_id, self.current_graph) self.newData.emit(tg_data_id, self.current_graph)
elif dtype == 'hodge':
set_id_list = []
graph_id = dic1.pop('graph')
for v in dic1.values():
set_id_list.append(self.add(v))
self.newData.emit(set_id_list, graph_id)
@QtCore.pyqtSlot(int, dict) @QtCore.pyqtSlot(int, dict)
def smooth_data(self, npoints, param_kwargs): def smooth_data(self, npoints, param_kwargs):