import sys
import time
import nidaqmx
from PyQt6 import QtCore, QtGui, QtWidgets
from PyQt6.QtCore import QTimer
from PyQt6.QtGui import QPixmap
# Local module and scripts
from pyccapt.control.core import runtime
from pyccapt.control.gui import tooltips
[docs]
class Ui_Gates(object):
"""
Load the GUI based on the configuration file.
Args:
None
Returns:
None
"""
def __init__(self, variables, conf, parent=None):
"""
Load the GUI based on the configuration file.
Args:
variables (object): Global variables
conf (dict): Configuration file
parent (object): Parent object
Returns:
None
"""
self.flag_super_user = False
self.variables = variables
self.conf = conf
self.parent = parent
[docs]
def setupUi(self, Gates):
"""
Load the GUI based on the configuration file.
Args:
Gates (object): Parent object
Returns:
None
"""
Gates.setObjectName("Gates")
Gates.resize(434, 426)
self.gridLayout_3 = QtWidgets.QGridLayout(Gates)
self.gridLayout_3.setObjectName("gridLayout_3")
self.verticalLayout_4 = QtWidgets.QVBoxLayout()
self.verticalLayout_4.setObjectName("verticalLayout_4")
self.diagram = QtWidgets.QLabel(parent=Gates)
self.diagram.setMinimumSize(QtCore.QSize(378, 246))
self.diagram.setMaximumSize(QtCore.QSize(16777215, 16777215))
self.diagram.setStyleSheet(
"QWidget{\n"
" border: 2px solid gray;\n"
" background: rgb(255, 255, 255)\n"
" }\n"
" "
)
self.diagram.setText("")
self.diagram.setObjectName("diagram")
self.verticalLayout_4.addWidget(self.diagram)
self.gridLayout = QtWidgets.QGridLayout()
self.gridLayout.setObjectName("gridLayout")
self.verticalLayout_3 = QtWidgets.QVBoxLayout()
self.verticalLayout_3.setObjectName("verticalLayout_3")
self.led_cryo = QtWidgets.QLabel(parent=Gates)
self.led_cryo.setMinimumSize(QtCore.QSize(50, 50))
self.led_cryo.setMaximumSize(QtCore.QSize(50, 50))
self.led_cryo.setAlignment(QtCore.Qt.AlignmentFlag.AlignCenter)
self.led_cryo.setObjectName("led_cryo")
self.verticalLayout_3.addWidget(self.led_cryo, 0, QtCore.Qt.AlignmentFlag.AlignHCenter)
self.cryo_switch = QtWidgets.QPushButton(parent=Gates)
self.cryo_switch.setMinimumSize(QtCore.QSize(0, 25))
self.cryo_switch.setStyleSheet(
"QPushButton{\n"
" background: rgb(193, 193, 193)\n"
" }\n"
" "
)
self.cryo_switch.setObjectName("cryo_switch")
self.verticalLayout_3.addWidget(self.cryo_switch)
self.gridLayout.addLayout(self.verticalLayout_3, 0, 2, 1, 1)
self.verticalLayout = QtWidgets.QVBoxLayout()
self.verticalLayout.setObjectName("verticalLayout")
self.led_main_chamber = QtWidgets.QLabel(parent=Gates)
self.led_main_chamber.setMinimumSize(QtCore.QSize(50, 50))
self.led_main_chamber.setMaximumSize(QtCore.QSize(50, 50))
self.led_main_chamber.setAlignment(QtCore.Qt.AlignmentFlag.AlignCenter)
self.led_main_chamber.setObjectName("led_main_chamber")
self.verticalLayout.addWidget(self.led_main_chamber, 0, QtCore.Qt.AlignmentFlag.AlignHCenter)
self.main_chamber_switch = QtWidgets.QPushButton(parent=Gates)
self.main_chamber_switch.setMinimumSize(QtCore.QSize(0, 25))
self.main_chamber_switch.setStyleSheet(
"QPushButton{\n"
" background: rgb(193, 193, 193)\n"
" }\n"
" "
)
self.main_chamber_switch.setObjectName("main_chamber_switch")
self.verticalLayout.addWidget(self.main_chamber_switch)
self.gridLayout.addLayout(self.verticalLayout, 0, 0, 1, 1)
self.verticalLayout_2 = QtWidgets.QVBoxLayout()
self.verticalLayout_2.setObjectName("verticalLayout_2")
self.led_load_lock = QtWidgets.QLabel(parent=Gates)
self.led_load_lock.setMinimumSize(QtCore.QSize(50, 50))
self.led_load_lock.setMaximumSize(QtCore.QSize(50, 50))
self.led_load_lock.setAlignment(QtCore.Qt.AlignmentFlag.AlignCenter)
self.led_load_lock.setObjectName("led_load_lock")
self.verticalLayout_2.addWidget(self.led_load_lock, 0, QtCore.Qt.AlignmentFlag.AlignHCenter)
self.load_lock_switch = QtWidgets.QPushButton(parent=Gates)
self.load_lock_switch.setMinimumSize(QtCore.QSize(0, 25))
self.load_lock_switch.setStyleSheet(
"QPushButton{\n"
" background: rgb(193, 193, 193)\n"
" }\n"
" "
)
self.load_lock_switch.setObjectName("load_lock_switch")
self.verticalLayout_2.addWidget(self.load_lock_switch)
self.gridLayout.addLayout(self.verticalLayout_2, 0, 1, 1, 1)
self.verticalLayout_4.addLayout(self.gridLayout)
self.gridLayout_2 = QtWidgets.QGridLayout()
self.gridLayout_2.setObjectName("gridLayout_2")
self.superuser = QtWidgets.QPushButton(parent=Gates)
sizePolicy = QtWidgets.QSizePolicy(QtWidgets.QSizePolicy.Policy.Fixed, QtWidgets.QSizePolicy.Policy.Fixed)
sizePolicy.setHorizontalStretch(0)
sizePolicy.setVerticalStretch(0)
sizePolicy.setHeightForWidth(self.superuser.sizePolicy().hasHeightForWidth())
self.superuser.setSizePolicy(sizePolicy)
self.superuser.setMinimumSize(QtCore.QSize(400, 25))
self.superuser.setStyleSheet(
"QPushButton{\n"
" background: rgb(193, 193, 193)\n"
" }\n"
" "
)
self.superuser.setObjectName("superuser")
self.gridLayout_2.addWidget(self.superuser, 0, 1, 1, 1)
self.Error = QtWidgets.QLabel(parent=Gates)
self.Error.setMinimumSize(QtCore.QSize(400, 30))
font = QtGui.QFont()
font.setPointSize(10)
font.setBold(True)
font.setStrikeOut(False)
self.Error.setFont(font)
self.Error.setAlignment(QtCore.Qt.AlignmentFlag.AlignCenter)
self.Error.setWordWrap(True)
self.Error.setTextInteractionFlags(QtCore.Qt.TextInteractionFlag.LinksAccessibleByMouse)
self.Error.setObjectName("Error")
self.gridLayout_2.addWidget(self.Error, 1, 1, 1, 1)
spacerItem = QtWidgets.QSpacerItem(
40, 20, QtWidgets.QSizePolicy.Policy.Expanding, QtWidgets.QSizePolicy.Policy.Minimum
)
self.gridLayout_2.addItem(spacerItem, 0, 2, 1, 1)
spacerItem1 = QtWidgets.QSpacerItem(
40, 20, QtWidgets.QSizePolicy.Policy.Expanding, QtWidgets.QSizePolicy.Policy.Minimum
)
self.gridLayout_2.addItem(spacerItem1, 0, 0, 1, 1)
self.verticalLayout_4.addLayout(self.gridLayout_2)
self.gridLayout_3.addLayout(self.verticalLayout_4, 0, 0, 1, 1)
self.retranslateUi(Gates)
QtCore.QMetaObject.connectSlotsByName(Gates)
tooltips.apply_tooltips(self, tooltips.GATES_TOOLTIPS)
# Diagram and LEDs ##############
self.diagram_close_all = QPixmap('./files/close_all.png')
self.diagram_main_open = QPixmap('./files/main_open.png')
self.diagram_load_open = QPixmap('./files/load_open.png')
self.diagram_cryo_open = QPixmap('./files/cryo_open.png')
self.diagram_load_main_open = QPixmap('./files/load_main_open.png')
self.diagram_cryo_main_open = QPixmap('./files/cryo_main_open.png')
self.diagram_cryo_load_open = QPixmap('./files/cryo_load_open.png')
self.diagram_all_open = QPixmap('./files/cryo_load_main_open.png')
self.led_red = QPixmap('./files/led-red-on.png')
self.led_green = QPixmap('./files/green-led-on.png')
self.diagram.setPixmap(self.diagram_close_all)
self.led_main_chamber.setPixmap(self.led_red)
self.led_load_lock.setPixmap(self.led_red)
self.led_cryo.setPixmap(self.led_red)
###
self.main_chamber_switch.clicked.connect(lambda: self.gates(1))
self.load_lock_switch.clicked.connect(lambda: self.gates(2))
self.cryo_switch.clicked.connect(lambda: self.gates(3))
self.superuser.clicked.connect(self.super_user_access)
# Create a QTimer to hide the warning message after 8 seconds
self.timer = QTimer(self.parent)
self.timer.timeout.connect(self.hideMessage)
self.original_button_style = self.superuser.styleSheet()
[docs]
def retranslateUi(self, Gates):
"""
Args:
Gates: The main window
Returns:
None
"""
_translate = QtCore.QCoreApplication.translate
###
Gates.setWindowTitle(_translate("Ui_Gates", "PyCCAPT Gates Control"))
Gates.setWindowIcon(QtGui.QIcon('./files/logo.png'))
###
self.led_cryo.setText(_translate("Gates", "Cryo"))
self.cryo_switch.setText(_translate("Gates", "Cryo"))
self.led_main_chamber.setText(_translate("Gates", "Main"))
self.main_chamber_switch.setText(_translate("Gates", "Main Chamber"))
self.led_load_lock.setText(_translate("Gates", "Load"))
self.load_lock_switch.setText(_translate("Gates", "Load Lock"))
self.superuser.setText(_translate("Gates", "Override Access"))
self.Error.setText(_translate("Gates", "<html><head/><body><p><br/></p></body></html>"))
[docs]
def super_user_access(self):
"""
The function for super user access
Args:
None
Returns:
None
"""
if not self.flag_super_user:
warning = QtWidgets.QMessageBox(parent=self.superuser)
warning.setIcon(QtWidgets.QMessageBox.Icon.Warning)
warning.setWindowTitle("Confirm Access Override")
warning.setText("Gate override can bypass interlocks and may be dangerous.")
warning.setInformativeText("Only continue if you really want to override gate access.")
warning.setStandardButtons(QtWidgets.QMessageBox.StandardButton.Yes | QtWidgets.QMessageBox.StandardButton.No)
warning.setDefaultButton(QtWidgets.QMessageBox.StandardButton.No)
if warning.exec() != QtWidgets.QMessageBox.StandardButton.Yes:
self.error_message("Override Access canceled.")
self.timer.start(8000)
return
self.flag_super_user = True
self.superuser.setStyleSheet("QPushButton{\nbackground: rgb(0, 255, 26)\n}")
self.error_message("!!! Override Access Granted !!!")
elif self.flag_super_user:
self.flag_super_user = False
self.superuser.setStyleSheet(self.original_button_style)
self.error_message("!!! Override Access deactivated !!!")
self.timer.start(8000)
[docs]
def gates(self, gate_num):
"""
The function for applying the command of closing or opening gate
Args:
gate_num: The number of the gate to be opened or closed
Returns:
None
"""
def switch_gate(num):
"""
The function for opening or closing the gate
Args:
num: The number of the gate to be opened or closed
Returns:
None
"""
with nidaqmx.Task() as task:
if self.conf["gates"] != "off":
task.do_channels.add_do_chan(self.conf["COM_PORT_gates"] + 'line%s' % num)
task.start()
task.write([True])
time.sleep(0.5)
task.write([False])
else:
print('The gates control is off')
def error_gate():
"""
The function for showing the error message in the GUI
Args:
None
Returns:
None
"""
if self.variables.start_flag:
self.error_message("!!! An experiment is running !!!")
else:
self.error_message("!!! Close the previous opened gate first !!!")
self.timer.start(8000)
# Main gate
if gate_num == 1:
if (
not self.variables.start_flag
and (
not self.variables.flag_load_gate
and not self.variables.flag_cryo_gate
and self.variables.flag_pump_load_lock
)
) or self.flag_super_user:
if not self.variables.flag_main_gate: # Open the main gate
if self.conf["gates"] == "on":
switch_gate(0)
self.led_main_chamber.setPixmap(self.led_green)
self.variables.flag_main_gate = True
elif self.variables.flag_main_gate: # Close the main gate
if self.conf["gates"] == "on":
switch_gate(1)
self.led_main_chamber.setPixmap(self.led_red)
self.variables.flag_main_gate = False
else:
error_gate()
# Buffer gate
elif gate_num == 2:
if (
not self.variables.start_flag
and (
not self.variables.flag_main_gate
and not self.variables.flag_cryo_gate
and self.variables.flag_pump_load_lock
)
) or self.flag_super_user:
if not self.variables.flag_load_gate: # Open the main gate
if self.conf["gates"] == "on":
switch_gate(2)
self.led_load_lock.setPixmap(self.led_green)
self.variables.flag_load_gate = True
elif self.variables.flag_load_gate: # Close the main gate
if self.conf["gates"] == "on":
switch_gate(3)
self.led_load_lock.setPixmap(self.led_red)
self.variables.flag_load_gate = False
else:
error_gate()
# Cryo gate
elif gate_num == 3:
if (
not self.variables.start_flag
and (
not self.variables.flag_main_gate
and not self.variables.flag_load_gate
and self.variables.flag_pump_load_lock
)
) or self.flag_super_user:
if not self.variables.flag_cryo_gate: # Open the main gate
if self.conf["gates"] == "on":
switch_gate(4)
self.led_cryo.setPixmap(self.led_green)
self.variables.flag_cryo_gate = True
elif self.variables.flag_cryo_gate: # Close the main gate
if self.conf["gates"] == "on":
switch_gate(5)
self.led_cryo.setPixmap(self.led_red)
self.variables.flag_cryo_gate = False
else:
error_gate()
else:
print('The gate number is not correct')
# change the diagram and the LEDs
if self.variables.flag_main_gate and self.variables.flag_load_gate and self.variables.flag_cryo_gate:
self.diagram.setPixmap(self.diagram_all_open)
elif self.variables.flag_main_gate and self.variables.flag_load_gate and not self.variables.flag_cryo_gate:
self.diagram.setPixmap(self.diagram_load_main_open)
elif self.variables.flag_main_gate and not self.variables.flag_load_gate and self.variables.flag_cryo_gate:
self.diagram.setPixmap(self.diagram_cryo_main_open)
elif not self.variables.flag_main_gate and self.variables.flag_load_gate and self.variables.flag_cryo_gate:
self.diagram.setPixmap(self.diagram_cryo_load_open)
elif not self.variables.flag_main_gate and not self.variables.flag_load_gate and self.variables.flag_cryo_gate:
self.diagram.setPixmap(self.diagram_cryo_open)
elif not self.variables.flag_main_gate and self.variables.flag_load_gate and not self.variables.flag_cryo_gate:
self.diagram.setPixmap(self.diagram_load_open)
elif self.variables.flag_main_gate and not self.variables.flag_load_gate and not self.variables.flag_cryo_gate:
self.diagram.setPixmap(self.diagram_main_open)
else:
self.diagram.setPixmap(self.diagram_close_all)
[docs]
def error_message(self, message):
"""
The function for showing the error message in the GUI
Args:
message: The message to be shown in the GUI
Returns:
None
"""
_translate = QtCore.QCoreApplication.translate
self.Error.setText(
_translate(
"OXCART", "<html><head/><body><p><span style=\" color:#ff0000;\">" + message + "</span></p></body></html>"
)
)
[docs]
def hideMessage(self):
"""
The function for hiding the error message in the GUI
Args:
None
Returns:
None
"""
# Hide the message and stop the timer
_translate = QtCore.QCoreApplication.translate
self.Error.setText(
_translate("OXCART", "<html><head/><body><p><span style=\" color:#ff0000;\"></span></p></body></html>")
)
self.timer.stop()
[docs]
def stop(self):
"""
The function for stopping the background activity
Args:
None
Returns:
None
"""
# Add any additional cleanup code here
pass
[docs]
class GatesWindow(QtWidgets.QWidget):
closed = QtCore.pyqtSignal() # Define a custom closed signal
def __init__(self, gui_gates, *args, **kwargs):
"""
Initialize the GatesWindow class.
Args:
gui_gates: GUI for gates.
*args, **kwargs: Additional arguments for QWidget initialization.
"""
super().__init__(*args, **kwargs)
self.gui_gates = gui_gates
[docs]
def closeEvent(self, event):
"""
Handle the close event of the GatesWindow.
Args:
event: Close event.
"""
if getattr(self, "force_close", False):
event.accept()
return
event.ignore()
self.hide()
self.closed.emit()
[docs]
def setWindowStyleFusion(self):
# Set the Fusion style
QtWidgets.QApplication.setStyle("Fusion")
if __name__ == "__main__":
try:
conf, _ = runtime.load_project_config()
except Exception as exc:
print('Can not load the configuration file')
print(exc)
sys.exit()
shared = runtime.create_shared_context(conf)
app = QtWidgets.QApplication(sys.argv)
app.setStyle('Fusion')
Gates = QtWidgets.QWidget()
ui = Ui_Gates(shared.variables, conf)
ui.setupUi(Gates)
Gates.show()
sys.exit(app.exec())