diff --git a/Widgets/AddSignal.py b/Widgets/AddSignal.py index c49c283..169aa72 100644 --- a/Widgets/AddSignal.py +++ b/Widgets/AddSignal.py @@ -26,6 +26,8 @@ def __init__(self): self.betaLineEdit = QLineEdit() self.gammaLineEdit = QLineEdit() + self.signalNameList = list() + self.functionComboBox = QComboBox() self.functionComboBox.addItems(["sin", "cos", "square", "sawtooth", "constant"]) @@ -65,7 +67,11 @@ def __init__(self): self.mainGridLayout.addWidget(self.addButton, 99, 0, 1, 2) self.setLayout(self.mainGridLayout) + def removeSignal(self, signalName): + if signalName in self.signalNameList: + self.signalNameList.remove(signalName) def addSignal(self): + data = {"name": self.nameLineEdit.text(), "function": self.functionComboBox.currentText(), "operator": self.operatorComboBox.currentText(), @@ -73,6 +79,14 @@ def addSignal(self): "beta": self.betaLineEdit.text(), "gamma": self.gammaLineEdit.text()} + if data["name"] in self.signalNameList: + self.nameLineEdit.setStyleSheet("border: 1px solid red") + + return + + self.addSignalSignal.emit(data) + self.signalNameList.append(data["name"]) + self.close()