Added ability to remove function

02bwilson · Jul 16, 2023 · 707a4e1 · 707a4e1
1 parent 8e27640
commit 707a4e1
Showing 1 changed file with 16 additions and 3 deletions.
diff --git a/Widgets/SignalTable.py b/Widgets/SignalTable.py
@@ -1,9 +1,12 @@
+from PyQt6.QtCore import Qt, pyqtSignal
 from PyQt6.QtGui import QStandardItemModel, QStandardItem, QIcon, QPixmap
 from PyQt6.QtWidgets import QWidget, QGridLayout, QTableView, QHeaderView, QAbstractItemView, QPushButton
 
 
 class SignalTable(QWidget):
-    _PLUS_ICON_PATH = "../Content/plus.png"
+
+
+    signalDeleted = pyqtSignal(str)
 
     def __init__(self):
         super().__init__()
@@ -32,12 +35,22 @@ def setupTable(self):
         header.setSectionResizeMode(1, QHeaderView.ResizeMode.Stretch)
         self.mainGridLayout.addWidget(self.tableView, 0, 0)
 
-        self.tableModel.appendRow([QStandardItem("Example 1"), QStandardItem("sin(10t)")])
+        self.tableModel.appendRow([QStandardItem("Example 1"), QStandardItem("sin(5*t)")])
 
         self.addSignalButton.setStyleSheet("border:none; text-align:center; color: green; font-size: 24pt;")
         self.addSignalButton.setText("+")
         self.mainGridLayout.addWidget(self.addSignalButton, 1, 0)
 
     def signalAdded(self, data):
         fnStr = str(data["alpha"]) + "*" + data["function"] + "(" + data["beta"] + "t" + "+" + data["gamma"] + ")"
-        self.tableModel.appendRow([QStandardItem(data["name"]), QStandardItem(fnStr)])
+        nameItem = QStandardItem(data["name"])
+        self.tableModel.appendRow([nameItem, QStandardItem(fnStr)])
+
+    def keyPressEvent(self, a0):
+        super().keyPressEvent(a0)
+
+        if a0.key() == Qt.Key.Key_Delete:
+            selectedSignalsIndex = self.tableView.selectionModel().currentIndex().row()
+            signalName = self.tableModel.item(selectedSignalsIndex, 0).text()
+            self.tableModel.removeRow(selectedSignalsIndex)
+            self.signalDeleted.emit(signalName)