class Kilroy(QtGui.QMainWindow):
def __init__(self, parameters):
super(Kilroy, self).__init__()
# Parse parameters into internal attributes
self.verbose = parameters.get("verbose")
self.valve_com_port = parameters.get("valves_com_port")
self.tcp_port = parameters.get("tcp_port")
self.pump_com_port = parameters.get("pump_com_port")
self.pump_ID = parameters.get("pump_ID")
if not hasattr(parameters, "num_simulated_valves"):
self.num_simulated_valves = 0
else:
self.num_simulated_valves = parameters.get("num_simulated_valves")
if not hasattr(parameters, "protocols_file"):
self.protocols_file = "default_config.xml"
else:
self.protocols_file = parameters.get("protocols_file")
if not hasattr(parameters, "commands_file"):
self.commands_file = "default_config.xml"
else:
self.commands_file = parameters.get("commands_file")
if not hasattr(parameters, "simulate_pump"):
self.simulate_pump = False
else:
self.simulate_pump = parameters.get("simulate_pump")
# Define additional internal attributes
self.received_message = None
# Create ValveChain instance
self.valveChain = ValveChain(com_port = self.valve_com_port,
num_simulated_valves = self.num_simulated_valves,
verbose = self.verbose)
# Create PumpControl instance
self.pumpControl = PumpControl(parameters = parameters)
# Create KilroyProtocols instance and connect signals
self.kilroyProtocols = KilroyProtocols(protocol_xml_path = self.protocols_file,
command_xml_path = self.commands_file,
verbose = self.verbose)
self.kilroyProtocols.command_ready_signal.connect(self.sendCommand)
self.kilroyProtocols.status_change_signal.connect(self.handleProtocolStatusChange)
self.kilroyProtocols.completed_protocol_signal.connect(self.handleProtocolComplete)
# Create Kilroy TCP Server and connect signals
self.tcpServer = TCPServer(port = self.tcp_port,
server_name = "Kilroy",
verbose = self.verbose)
self.tcpServer.messageReceived.connect(self.handleTCPData)
# Create GUI
self.createGUI()
# ----------------------------------------------------------------------------------------
# Close
# ----------------------------------------------------------------------------------------
def close(self):
self.kilroyProtocols.close()
self.tcpServer.close()
self.valveChain.close()
self.pumpControl.close()
print "\nKilroy was here!"
# ----------------------------------------------------------------------------------------
# Create master GUI
# ----------------------------------------------------------------------------------------
def createGUI(self):
self.mainLayout = QtGui.QGridLayout()
self.mainLayout.addWidget(self.kilroyProtocols.mainWidget, 0, 0, 2, 2)
self.mainLayout.addWidget(self.kilroyProtocols.valveCommands.mainWidget, 2, 0, 1, 1)
self.mainLayout.addWidget(self.kilroyProtocols.pumpCommands.mainWidget, 2, 1, 1, 1)
self.mainLayout.addWidget(self.valveChain.mainWidget, 0, 2, 2, 2)
self.mainLayout.addWidget(self.pumpControl.mainWidget, 0, 4, 2, 1)
#self.mainLayout.addWidget(self.tcpServer.mainWidget, 2, 2, 1, 4)
# ----------------------------------------------------------------------------------------
# Redirect protocol status change from kilroyProtocols to valveChain
# ----------------------------------------------------------------------------------------
def handleProtocolStatusChange(self):
status = self.kilroyProtocols.getStatus()
if status[0] >= 0: # Protocol is running
self.valveChain.setEnabled(False)
self.pumpControl.setEnabled(False)
else:
self.valveChain.setEnabled(True)
self.pumpControl.setEnabled(True)
# ----------------------------------------------------------------------------------------
# Handle a protocol complete signal from the valve protocols
# ----------------------------------------------------------------------------------------
def handleProtocolComplete(self, message):
# If the protocol was sent by TCP pass on the complete signal
if (self.received_message is not None) and self.received_message.getID() == message.getID():
self.tcpServer.sendMessage(message)
self.received_message = None # Reset the received_message
# ----------------------------------------------------------------------------------------
# Handle protocol request sent via TCP server
# ----------------------------------------------------------------------------------------
def handleTCPData(self, message):
# Confirm that message is a protocol message
if not message.getType() == "Kilroy Protocol":
message.setError(True, "Wrong message type sent to Kilroy: " + message.getType())
self.tcpServer.sendMessage(message)
elif not self.kilroyProtocols.isValidProtocol(message.getData("name")):
message.setError(True, "Invalid Kilroy Protocol")
self.tcpServer.sendMessage(message)
elif message.isTest():
required_time = self.kilroyProtocols.requiredTime(message.getData("name"))
message.addResponse("duration", required_time)
self.tcpServer.sendMessage(message)
else: # Valid, non-test message
# Keep track of valid messages issued via TCP
self.received_message = message
# Start the protocol
self.kilroyProtocols.startProtocolRemotely(message)
# ----------------------------------------------------------------------------------------
# Redirect commands from kilroy protocol class to valves or pump
# ----------------------------------------------------------------------------------------
def sendCommand(self):
command_data = self.kilroyProtocols.getCurrentCommand()
if command_data[0] == "valve":
self.valveChain.receiveCommand(command_data[1])
elif command_data[0] == "pump":
self.pumpControl.receiveCommand(command_data[1])
else:
print "Received command of unknown type: " + str(command_data[0])
