interactive integration; new user-defined fit functions reloads model list; fixed requirements.txt

src/gui_qt/graphs/drawings.py

@@ -0,0 +1,161 @@
+from ..Qt import QtWidgets, QtCore
+from .._py.guidelinewidget import Ui_Form
+
+
+class DrawingsWidget(QtWidgets.QWidget, Ui_Form):
+
+    def __init__(self, parent=None):
+        super().__init__(parent=parent)
+
+        self.connected_figure=None
+
+        self.setupUi(self)
+
+    def __call__(self, graphs):
+        for gid, name in graphs:
+            self.graph_comboBox.addItem(name, userData=gid)
+
+    def clear(self):
+        self.graph_comboBox.clear()
+
+    @QtCore.pyqtSlot(int, name='on_mode_comboBox_currentIndexChanged')
+    def change_draw_type(self, idx: int):
+        self.stackedWidget.setCurrentIndex(idx)
+
+
+
+"""
+
+        self.lines = {}
+        self.comments = {}
+
+        self.vh_pos_lineEdit.setValidator(QtGui.QDoubleValidator())
+
+        self.tableWidget.installEventFilter(self)
+
+    @QtCore.pyqtSlot(name='on_pushButton_clicked')
+    def make_line(self):
+        invalid = True
+
+        idx = self.mode_comboBox.currentIndex()
+        try:
+            pos = float(self.vh_pos_lineEdit.text())
+            # Vertical: idx=0; horizontal: idx = 1
+            angle = 90*abs(1-idx)
+            invalid = False
+        except ValueError:
+            pos = None
+            angle = None
+            pass
+
+        if invalid:
+            QtWidgets.QMessageBox().information(self, 'Invalid input', 'Input is not a valid number')
+            return
+
+        qcolor = QtGui.QColor.fromRgb(*self.color_comboBox.value.rgb())
+        comment = self.comment_lineEdit.text()
+        line = LogInfiniteLine(pos=pos, angle=angle, movable=self.drag_checkBox.isChecked(), pen=qcolor)
+        line.sigPositionChanged.connect(self.move_line)
+
+        self.make_table_row(pos, angle, qcolor, comment)
+
+        graph_id = self.graph_comboBox.currentData()
+        try:
+            self.lines[graph_id].append(line)
+            self.comments[graph_id].append(comment)
+        except KeyError:
+            self.lines[graph_id] = [line]
+            self.comments[graph_id] = [comment]
+
+        self.line_created.emit(line, graph_id)
+
+    def set_graphs(self, graphs: list):
+        for graph_id, name in graphs:
+            self.graph_comboBox.addItem(name, userData=graph_id)
+
+    def remove_graph(self, graph_id: str):
+        idx = self.graph_comboBox.findData(graph_id)
+        if idx != -1:
+            self.graph_comboBox.removeItem(idx)
+
+        if graph_id in self.lines:
+            del self.lines[graph_id]
+
+    @QtCore.pyqtSlot(int, name='on_graph_comboBox_currentIndexChanged')
+    def change_graph(self, idx: int):
+        self.tableWidget.clear()
+        self.tableWidget.setRowCount(0)
+
+        graph_id = self.graph_comboBox.itemData(idx)
+        if graph_id in self.lines:
+            lines = self.lines[graph_id]
+            comments = self.comments[graph_id]
+            for i, line in enumerate(lines):
+                self.make_table_row(line.pos(), line.angle, line.pen.color(), comments[i])
+
+    def make_table_row(self, position, angle, color, comment):
+        if angle == 0:
+            try:
+                pos_label = 'x = ' + str(position.y())
+            except AttributeError:
+                pos_label = 'x = {position}'
+
+        elif angle == 90:
+            try:
+                pos_label = f'y = {position.x()}'
+            except AttributeError:
+                pos_label = f'y = {position}'
+
+        else:
+            raise ValueError('Only horizontal or vertical lines are supported')
+
+        item = QtWidgets.QTableWidgetItem(pos_label)
+        item.setFlags(QtCore.Qt.ItemIsSelectable)
+        item.setForeground(QtGui.QBrush(QtGui.QColor('black')))
+
+        row_count = self.tableWidget.rowCount()
+        self.tableWidget.setRowCount(row_count+1)
+        self.tableWidget.setItem(row_count, 0, item)
+
+        item2 = QtWidgets.QTableWidgetItem(comment)
+        self.tableWidget.setItem(row_count, 1, item2)
+
+        colitem = QtWidgets.QTableWidgetItem('  ')
+        colitem.setBackground(QtGui.QBrush(color))
+        colitem.setFlags(QtCore.Qt.ItemIsSelectable)
+        self.tableWidget.setVerticalHeaderItem(row_count, colitem)
+
+    def eventFilter(self, src: QtCore.QObject, evt: QtCore.QEvent) -> bool:
+        if evt.type() == QtCore.QEvent.KeyPress:
+            if evt.key() == QtCore.Qt.Key_Delete:
+                self.delete_line()
+                return True
+
+        return super().eventFilter(src, evt)
+
+    def delete_line(self):
+        remove_rows = sorted([item.row() for item in self.tableWidget.selectedItems()])
+        graph_id = self.graph_comboBox.currentData()
+        current_lines = self.lines[graph_id]
+
+        print(remove_rows)
+        for i in reversed(remove_rows):
+            print(i)
+            self.tableWidget.removeRow(i)
+            self.line_deleted.emit(current_lines[i], graph_id)
+
+            current_lines.pop(i)
+            self.comments[graph_id].pop(i)
+
+    @QtCore.pyqtSlot(object)
+    def move_line(self, line: InfiniteLine):
+        current_idx = self.graph_comboBox.currentData()
+        graphs = self.lines[current_idx]
+        i = -1
+        for i, line_i in enumerate(graphs):
+            if line == line_i:
+                break
+        pos = line.value()
+        text_item = self.tableWidget.item(i, 0)
+        text_item.setText(text_item.text()[:4]+f'{pos:.4g}')
+"""