def __init__(self, model, parent=None):
super().__init__(parent)
self.model = model
self.ui = Ui_TmCalcDialog()
self.ui.setupUi(self)
valid = QtGui.QDoubleValidator()
self.ui.lineEdit.setValidator(valid)
self.ui.lineEdit_2.setValidator(valid)
self.ui.lineEdit_temp.setValidator(valid)
self.ui.treeWidget.setHeaderLabels(["Filename", "Temperature"])
self.filenames = get_enabled_filename(self.model)
for f in self.filenames:
# self.ui.comboBox.addItem(f)
temp = self.__guess_temp(f)
item = QtWidgets.QTreeWidgetItem([f, temp], 0)
self.ui.treeWidget.addTopLevelItem(item)
self.ui.treeWidget.setColumnWidth(0, 150)
self.ui.treeWidget.setColumnWidth(1, 100)
self.ui.set_temp_button.clicked.connect(self.set_temperature)
class TmCalcDialog(QtWidgets.QDialog):
def __init__(self, model, parent=None):
super().__init__(parent)
self.model = model
self.ui = Ui_TmCalcDialog()
self.ui.setupUi(self)
valid = QtGui.QDoubleValidator()
self.ui.lineEdit.setValidator(valid)
self.ui.lineEdit_2.setValidator(valid)
self.ui.lineEdit_temp.setValidator(valid)
self.ui.treeWidget.setHeaderLabels(["Filename", "Temperature"])
self.filenames = get_enabled_filename(self.model)
for f in self.filenames:
# self.ui.comboBox.addItem(f)
temp = self.__guess_temp(f)
item = QtWidgets.QTreeWidgetItem([f, temp], 0)
self.ui.treeWidget.addTopLevelItem(item)
self.ui.treeWidget.setColumnWidth(0, 150)
self.ui.treeWidget.setColumnWidth(1, 100)
self.ui.set_temp_button.clicked.connect(self.set_temperature)
def set_temperature(self):
item = self.ui.treeWidget.selectedItems()[0]
temp = self.ui.lineEdit_temp.text()
item.setData(1, 0, temp)
def get_temperature(self):
n = self.ui.treeWidget.topLevelItemCount()
list_ = []
for i in range(n):
temp = self.ui.treeWidget.topLevelItem(i).data(1, 0)
list_.append(float(temp))
return list_
def accept(self):
if not self.ui.lineEdit.text():
return
min_vol = float(self.ui.lineEdit.text())
max_vol = float(self.ui.lineEdit_2.text())
if min_vol > max_vol:
return
temp_list = np.array(self.get_temperature())
scale_factor = calc_yscale_factor(self.model, min_vol, max_vol)
tmplot_dialog = TmFitDialog(self)
try:
tmplot_dialog.fit(temp_list, scale_factor)
except RuntimeError as e:
return
tmplot_dialog.exec_()
super().accept()
# private
def __guess_temp(self, f):
matched = re.findall("\d+\.?\d*", f)
if not matched:
return ""
selected = [temp for temp in matched if len(temp) <= 2]
return selected[-1]
