def __init__(self,
blacs,
globals_files,
connection_table_labscript,
output_path,
close_notification_func=None):
QDialog.__init__(self, blacs['ui'])
self.setAttribute(
Qt.WA_DeleteOnClose,
True) # make sure the dialog is deleted when the window is closed
self.globals_files = globals_files
self.labscript_file = connection_table_labscript
self.output_path = output_path
self.tempfilename = self.output_path.strip('.h5') + '.temp.h5'
self.blacs = blacs
self.close_notification_func = close_notification_func
self.ui = UiLoader().load(
os.path.join(os.path.dirname(os.path.realpath(__file__)),
'compile_and_restart.ui'))
self.output_box = OutputBox(self.ui.verticalLayout)
self.ui.restart.setEnabled(False)
# Connect buttons
self.ui.restart.clicked.connect(self.restart)
self.ui.compile.clicked.connect(self.compile)
self.ui.cancel.clicked.connect(self.reject)
self.ui.setParent(self)
self.ui.show()
self.show()
self.compile()
def main():
app = QtWidgets.QApplication(sys.argv)
window = QtWidgets.QWidget()
layout = QtWidgets.QVBoxLayout(window)
outputbox = OutputBox(layout)
funcs = [
regular_print_stdout,
regular_print_stderr,
libc_printf,
echo_hello,
multiprocessing_Process,
]
for f in funcs:
button = QtWidgets.QPushButton(f.__name__)
button.clicked.connect(f)
layout.addWidget(button)
redirect_stdout = OutputInterceptor('localhost', outputbox.port)
redirect_sterr = OutputInterceptor('localhost', outputbox.port, streamname='stderr')
redirect_stdout.connect()
redirect_sterr.connect()
window.resize(800, 500)
window.show()
app.exec_()
redirect_stdout.disconnect()
redirect_sterr.disconnect()
outputbox.shutdown()
def __init__(self, blacs, globals_files,connection_table_labscript, output_path, sequence_id_format, close_notification_func=None):
QDialog.__init__(self,blacs['ui'])
self.setAttribute(Qt.WA_DeleteOnClose, True) # make sure the dialog is deleted when the window is closed
self.globals_files = globals_files
self.sequence_id_format = sequence_id_format
self.labscript_file = connection_table_labscript
self.output_path = output_path
self.tempfilename = self.output_path.strip('.h5')+'.temp.h5'
self.blacs = blacs
self.close_notification_func = close_notification_func
self.ui = UiLoader().load(os.path.join(os.path.dirname(os.path.realpath(__file__)),'compile_and_restart.ui'))
self.output_box = OutputBox(self.ui.verticalLayout)
self.ui.restart.setEnabled(False)
# Connect buttons
self.ui.restart.clicked.connect(self.restart)
self.ui.compile.clicked.connect(self.compile)
self.ui.cancel.clicked.connect(self.reject)
self.ui.setParent(self)
self.ui.show()
self.show()
self.compile()
class CompileAndRestart(QDialog):
def __init__(self,
blacs,
globals_files,
connection_table_labscript,
output_path,
close_notification_func=None):
QDialog.__init__(self, blacs['ui'])
self.setAttribute(
Qt.WA_DeleteOnClose,
True) # make sure the dialog is deleted when the window is closed
self.globals_files = globals_files
self.labscript_file = connection_table_labscript
self.output_path = output_path
self.tempfilename = self.output_path.strip('.h5') + '.temp.h5'
self.blacs = blacs
self.close_notification_func = close_notification_func
self.ui = UiLoader().load(
os.path.join(os.path.dirname(os.path.realpath(__file__)),
'compile_and_restart.ui'))
self.output_box = OutputBox(self.ui.verticalLayout)
self.ui.restart.setEnabled(False)
# Connect buttons
self.ui.restart.clicked.connect(self.restart)
self.ui.compile.clicked.connect(self.compile)
self.ui.cancel.clicked.connect(self.reject)
self.ui.setParent(self)
self.ui.show()
self.show()
self.compile()
def closeEvent(self, event):
if not self.ui.cancel.isEnabled():
event.ignore()
else:
event.accept()
def on_activate_default(self, window):
if self.button_restart.get_sensitive():
self.restart()
elif self.button_compile.get_sensitive():
self.compile()
def compile(self):
self.ui.compile.setEnabled(False)
self.ui.cancel.setEnabled(False)
self.ui.restart.setEnabled(False)
msg = 'Recompiling connection table'
self.ui.label.setText(msg)
self.output_box.output(msg + '\n')
runmanager.compile_labscript_with_globals_files_async(
self.labscript_file, self.globals_files, self.tempfilename,
self.output_box.port, self.finished_compiling)
@inmain_decorator(True)
def finished_compiling(self, success):
self.ui.compile.setEnabled(True)
self.ui.cancel.setEnabled(True)
if success:
try:
os.remove(self.output_path)
except OSError:
# File doesn't exist, no need to delete then:
pass
try:
os.rename(self.tempfilename, self.output_path)
except OSError:
self.output_box.output(
'Couldn\'t replace existing connection table h5 file. Is it open in another process?\n',
red=True)
self.ui.label.setText('Compilation failed.')
self.ui.restart.setEnabled(False)
os.remove(self.tempfilename)
else:
self.ui.restart.setEnabled(True)
self.ui.cancel.setEnabled(False)
msg = 'Compilation succeeded, restart when ready'
self.ui.label.setText(msg)
self.output_box.output(msg + '\n')
else:
self.ui.restart.setEnabled(False)
msg = 'Compilation failed. Please fix the errors in the connection table (python file) and try again'
self.ui.label.setText(msg)
self.output_box.output(msg + '\n')
try:
os.remove(self.tempfilename)
except Exception:
pass
def restart(self):
#gobject.timeout_add(100, self.blacs.destroy)
if self.close_notification_func:
self.close_notification_func()
QTimer.singleShot(100, self.blacs['ui'].close)
self.accept()
self.blacs['set_relaunch'](True)
class CompileAndRestart(QDialog):
def __init__(self, blacs, globals_files,connection_table_labscript, output_path, sequence_id_format, close_notification_func=None):
QDialog.__init__(self,blacs['ui'])
self.setAttribute(Qt.WA_DeleteOnClose, True) # make sure the dialog is deleted when the window is closed
self.globals_files = globals_files
self.sequence_id_format = sequence_id_format
self.labscript_file = connection_table_labscript
self.output_path = output_path
self.tempfilename = self.output_path.strip('.h5')+'.temp.h5'
self.blacs = blacs
self.close_notification_func = close_notification_func
self.ui = UiLoader().load(os.path.join(os.path.dirname(os.path.realpath(__file__)),'compile_and_restart.ui'))
self.output_box = OutputBox(self.ui.verticalLayout)
self.ui.restart.setEnabled(False)
# Connect buttons
self.ui.restart.clicked.connect(self.restart)
self.ui.compile.clicked.connect(self.compile)
self.ui.cancel.clicked.connect(self.reject)
self.ui.setParent(self)
self.ui.show()
self.show()
self.compile()
def closeEvent(self,event):
if not self.ui.cancel.isEnabled():
event.ignore()
else:
event.accept()
def on_activate_default(self,window):
if self.button_restart.get_sensitive():
self.restart()
elif self.button_compile.get_sensitive():
self.compile()
def compile(self):
self.ui.compile.setEnabled(False)
self.ui.cancel.setEnabled(False)
self.ui.restart.setEnabled(False)
self.ui.label.setText('Recompiling connection table')
# TODO: why are we calling runmanager here rather than using the info
# in the file to do it?
runmanager.compile_labscript_with_globals_files_async(self.labscript_file,
self.globals_files, self.tempfilename, self.sequence_id_format, "",
self.output_box.port, self.finished_compiling)
@inmain_decorator(True)
def finished_compiling(self, success):
self.ui.compile.setEnabled(True)
self.ui.cancel.setEnabled(True)
if success:
self.ui.restart.setEnabled(True)
self.ui.cancel.setEnabled(False)
self.ui.label.setText('Compilation succeeded, restart when ready')
try:
os.remove(self.output_path)
except OSError:
# File doesn't exist, no need to delete then:
pass
try:
os.rename(self.tempfilename,self.output_path)
except OSError:
self.output_box.output('Couldn\'t replace existing connection table h5 file. Is it open in another process?', red=True)
self.ui.label.setText('Compilation failed.')
self.ui.restart.setEnabled(False)
os.remove(self.tempfilename)
else:
self.ui.restart.setEnabled(False)
self.ui.label.setText('Compilation failed. Please fix the errors in the connection table (python file) and try again')
try:
os.remove(self.tempfilename)
except Exception:
pass
def restart(self):
#gobject.timeout_add(100, self.blacs.destroy)
if self.close_notification_func:
self.close_notification_func()
QTimer.singleShot(100, self.blacs['ui'].close)
self.accept()
self.blacs['set_relaunch'](True)