def __init__(self, model, parent=None):
super().__init__(parent)
self.model = model
self.default_filename = "peaktable.csv"
self.ui = Ui_PeakTableDialog()
self.ui.setupUi(self)
# set validator
valid = QtGui.QDoubleValidator()
self.ui.lineEdit_min.setValidator(valid)
self.ui.lineEdit_max.setValidator(valid)
# set table header to stretch mode
self.ui.tableWidget.horizontalHeader().setStretchLastSection(True)
# signal-slot connection
self.ui.okButton.clicked.connect(self.accept)
self.ui.saveCSVButton.clicked.connect(self.saveCSV)
self.ui.updateButton.clicked.connect(self.update)
self.ui.lineEdit_min.textChanged.connect(self.update)
self.ui.lineEdit_max.textChanged.connect(self.update)
class PeakTableDialog(QtWidgets.QDialog):
def __init__(self, model, parent=None):
super().__init__(parent)
self.model = model
self.default_filename = "peaktable.csv"
self.ui = Ui_PeakTableDialog()
self.ui.setupUi(self)
# set validator
valid = QtGui.QDoubleValidator()
self.ui.lineEdit_min.setValidator(valid)
self.ui.lineEdit_max.setValidator(valid)
# set table header to stretch mode
self.ui.tableWidget.horizontalHeader().setStretchLastSection(True)
# signal-slot connection
self.ui.okButton.clicked.connect(self.accept)
self.ui.saveCSVButton.clicked.connect(self.saveCSV)
self.ui.updateButton.clicked.connect(self.update)
self.ui.lineEdit_min.textChanged.connect(self.update)
self.ui.lineEdit_max.textChanged.connect(self.update)
# TODO: implement it
def saveCSV(self):
csvfile = QtWidgets.QFileDialog.getSaveFileName(
self, "Save CSV",
os.path.expanduser("~") + "/" + self.default_filename,
filter="Text files (*.csv)")
if not csvfile[0]:
return
text = "Filenames, Volume, Intensity\n"
for i in range(self.ui.tableWidget.rowCount()):
text += self.ui.tableWidget.item(i, 0).text()
text += ", "
text += self.ui.tableWidget.item(i, 1).text()
text += ", "
text += self.ui.tableWidget.item(i, 2).text()
text += "\n"
with open(csvfile[0], "w+") as f:
f.write(text)
def update(self):
# update table only when both textbox is not empty
if not self.ui.lineEdit_min.text():
return
if not self.ui.lineEdit_max.text():
return
# and min < max
min_vol = float(self.ui.lineEdit_min.text())
max_vol = float(self.ui.lineEdit_max.text())
if min_vol > max_vol:
return
self.__update_table(min_vol, max_vol)
# private methods
def __update_table(self, min_vol, max_vol):
data = self.model.get_current_data()
# if no data, exit
if len(data['filenames']) == 0:
return
flags = data['enable_flags']
f_ary = [f for f, fl in zip(data['filenames'], flags) if fl]
d_ary = [d for d, fl in zip(data['data'], flags) if fl]
n_row = sum(flags)
self.ui.tableWidget.setRowCount(n_row)
row = 0
ary = []
for f, d in zip(f_ary, d_ary):
minidx = np.argmin(np.abs(d[:, 0] - min_vol))
maxidx = np.argmin(np.abs(d[:, 0] - max_vol))
# get max volume and intensity
max_value = max(d[minidx:maxidx, 1])
max_volume_idx = np.argmax(d[minidx:maxidx, 1])
max_volume = d[max_volume_idx, 0] + min_vol
# alternative implementation
# max_volume = d[max_volume_idx+minidx, 0]
ary.append(max_value)
# set item to table
item = QtWidgets.QTableWidgetItem(f)
self.ui.tableWidget.setItem(row, 0, item)
item = QtWidgets.QTableWidgetItem(str(max_volume))
self.ui.tableWidget.setItem(row, 1, item)
item = QtWidgets.QTableWidgetItem(str(max_value))
self.ui.tableWidget.setItem(row, 2, item)
row += 1
if self.ui.normalizeCheckBox.checkState() == 2:
max_ary = np.array(ary)
max_all = np.max(max_ary)
max_ary /= max_all
for i in range(self.ui.tableWidget.rowCount()):
item = QtWidgets.QTableWidgetItem(str(max_ary[i]))
self.ui.tableWidget.setItem(i, 2, item)
