This commit is contained in:
david rice
2026-01-20 08:24:36 +00:00
parent 95332f3788
commit 41803786aa
5 changed files with 112 additions and 25 deletions

98
main.py
View File

@@ -69,6 +69,10 @@ class MainWindow(QMainWindow):
self.ui.connButton.setCheckable(True)
self.ui.connButton.clicked.connect(self.conn_button_press)
# Add global power button action
self.ui.powerButton.setCheckable(True)
self.ui.powerButton.clicked.connect(self.power_button_press)
# Add radio button actions
self.ui.radiolocal.toggled.connect(self.handle_all_radios)
self.ui.radioscript.toggled.connect(self.handle_all_radios)
@@ -87,7 +91,11 @@ class MainWindow(QMainWindow):
self.ui.setv.setHidden(True)
self.ui.actvLabel.setHidden(True)
self.ui.actv.setHidden(True)
self.ui.scriptLabel.setHidden(True)
self.ui.script.setHidden(True)
self.ui.openButton.setHidden(True)
self.ui.runButton.setHidden(True)
# Set initial size
initial_width = 1000
initial_height = 600
@@ -101,8 +109,8 @@ class MainWindow(QMainWindow):
self.setFixedSize(1000, 600)
# Setup serial port
#self.serial = QSerialPort(self)
#self.serial.readyRead.connect(self.on_data_received)
self.serial = QSerialPort(self)
self.serial.readyRead.connect(self.on_data_received)
# Create serial rx flag
self.serial_rx_flag = False
@@ -137,6 +145,10 @@ class MainWindow(QMainWindow):
self.ui.setv.setHidden(False)
self.ui.actvLabel.setHidden(False)
self.ui.actv.setHidden(False)
self.ui.scriptLabel.setHidden(True)
self.ui.script.setHidden(True)
self.ui.openButton.setHidden(True)
self.ui.runButton.setHidden(True)
elif button.text() == "SCRIPT CONTROL":
self.ui.powerButton.setHidden(True)
@@ -144,20 +156,56 @@ class MainWindow(QMainWindow):
self.ui.setv.setHidden(True)
self.ui.actvLabel.setHidden(True)
self.ui.actv.setHidden(True)
self.ui.scriptLabel.setHidden(False)
self.ui.script.setHidden(False)
self.ui.openButton.setHidden(False)
self.ui.runButton.setHidden(False)
elif button.text() == "PULSE TRIGGERED":
self.ui.powerButton.setHidden(True)
self.ui.setvLabel.setHidden(True)
self.ui.setv.setHidden(True)
self.ui.actvLabel.setHidden(True)
self.ui.actv.setHidden(True)
self.ui.actv.setHidden(True)
self.ui.scriptLabel.setHidden(True)
self.ui.script.setHidden(True)
self.ui.openButton.setHidden(True)
self.ui.runButton.setHidden(True)
else:
self.ui.powerButton.setHidden(True)
self.ui.setvLabel.setHidden(True)
self.ui.setv.setHidden(True)
self.ui.actvLabel.setHidden(True)
self.ui.actv.setHidden(True)
self.ui.actv.setHidden(True)
self.ui.scriptLabel.setHidden(True)
self.ui.script.setHidden(True)
self.ui.openButton.setHidden(True)
self.ui.runButton.setHidden(True)
def power_button_press (self):
text_pow_state = self.ui.powerButton.text()
if text_pow_state == "POWER OFF":
self.ui.powerButton.setText("POWER ON")
command = 83 #0x53 'S' ASCII
state = 1
data = (command, state)
byte_data = self.pack_integers_to_bytes(*data)
self.serial.write(QByteArray(byte_data))
else:
self.ui.powerButton.setText("POWER OFF")
command = 83 #0x53 'S' ASCII
state = 0
data = (command, state)
byte_data = self.pack_integers_to_bytes(*data)
self.serial.write(QByteArray(byte_data))
def conn_button_press (self):
text_conn = self.ui.connButton.text()
@@ -171,23 +219,21 @@ class MainWindow(QMainWindow):
QApplication.processEvents()
# Specify serial port and baud
#self.serial.setPortName("/dev/ttyACM0")
#self.serial.setBaudRate(115200)
self.serial.setPortName("/dev/ttyACM0")
self.serial.setBaudRate(115200)
#if not self.serial.open(QIODevice.ReadWrite):
#print(f"Failed to open port: {self.serial.errorString()}")
if not self.serial.open(QIODevice.ReadWrite):
print(f"Failed to open port: {self.serial.errorString()}")
command = 73 #0x49 'I' ASCII
state = 0
data = (command, state)
byte_data = self.pack_integers_to_bytes(*data)
#self.serial.write(QByteArray(byte_data))
self.serial.write(QByteArray(byte_data))
self.timeouttimer.start()
self.serial_rx_flag = True
while ((self.serial_rx_flag == False) and (self.timeoutflag == False)):
QApplication.processEvents()
@@ -195,7 +241,7 @@ class MainWindow(QMainWindow):
if self.timeoutflag == True:
self.windowresponse = NoDeviceWindow.get_response(self)
#self.serial.close()
self.serial.close()
self.ui.connButton.setText("CONNECT")
self.ui.sn.setHidden(True)
self.ui.snLabel.setHidden(True)
@@ -206,6 +252,15 @@ class MainWindow(QMainWindow):
self.ui.radiolocal.setHidden(True)
self.ui.radioscript.setHidden(True)
self.ui.radiopulse.setHidden(True)
self.ui.powerButton.setHidden(True)
self.ui.setvLabel.setHidden(True)
self.ui.setv.setHidden(True)
self.ui.actvLabel.setHidden(True)
self.ui.actv.setHidden(True)
self.ui.scriptLabel.setHidden(True)
self.ui.script.setHidden(True)
self.ui.openButton.setHidden(True)
self.ui.runButton.setHidden(True)
else:
self.process_serial_number()
@@ -215,7 +270,7 @@ class MainWindow(QMainWindow):
self.timeoutflag = False
else:
#self.serial.close()
self.serial.close()
self.ui.connButton.setText("CONNECT")
self.ui.sn.setHidden(True)
self.ui.snLabel.setHidden(True)
@@ -225,7 +280,16 @@ class MainWindow(QMainWindow):
self.ui.fw.setText("---")
self.ui.radiolocal.setHidden(True)
self.ui.radioscript.setHidden(True)
self.ui.radiopulse.setHidden(True)
self.ui.radiopulse.setHidden(True)
self.ui.powerButton.setHidden(True)
self.ui.setvLabel.setHidden(True)
self.ui.setv.setHidden(True)
self.ui.actvLabel.setHidden(True)
self.ui.actv.setHidden(True)
self.ui.scriptLabel.setHidden(True)
self.ui.script.setHidden(True)
self.ui.openButton.setHidden(True)
self.ui.runButton.setHidden(True)
QApplication.processEvents()
@@ -283,7 +347,6 @@ class MainWindow(QMainWindow):
sub_sn = self.store_buffer[0:19]
sn_chars = [chr(c) for c in sub_sn if 0 <= c <= 127]
sn_str = "".join(sn_chars)
sn_str = "ARRIVE-POWERSIM-001"
self.ui.sn.setText(sn_str)
# Get and update FW Rev
@@ -294,8 +357,7 @@ class MainWindow(QMainWindow):
minor = sub_fw[2:3]
full = major + sep + minor
fw_chars = [chr(c) for c in full if 0 <= c <= 127]
fw_str = "".join(fw_chars)
fw_str = "0:1"
fw_str = "".join(fw_chars)
self.ui.fw.setText(fw_str)
self.ui.sn.setHidden(False)