def show_create_project_dialog(self) -> None:
"""Dialog to create new project."""
file = os.path.join(os.environ['BCISTREAM_ROOT'], 'framework', 'qtgui',
'new_project.ui')
project = QUiLoader().load(file, self.parent_frame)
project.buttonBox.button(QDialogButtonBox.Ok).clicked.connect(
lambda evt: self.create_project(
project.lineEdit_project_name.text(),
project.radioButton_visualization.isChecked(),
project.radioButton_stimulus_delivery.isChecked(),
project.radioButton_data_analysis.isChecked()))
project.buttonBox.button(
QDialogButtonBox.Ok).clicked.connect(lambda evt: project.destroy())
project.buttonBox.button(QDialogButtonBox.Cancel).clicked.connect(
lambda evt: project.destroy())
project.lineEdit_project_name.textChanged.connect(
lambda str_: project.buttonBox.button(QDialogButtonBox.Ok).
setDisabled(os.path.isdir(os.path.join(self.projects_dir, str_))))
center = QDesktopWidget().availableGeometry().center()
geometry = project.frameGeometry()
geometry.moveCenter(center)
project.move(geometry.topLeft())
project.show()
class ConfigurationFrame(QMainWindow):
"""Auxiliar window with configuration options."""
# ----------------------------------------------------------------------
def __init__(self, *args, **kwargs):
super().__init__(*args, **kwargs)
frame = os.path.join(
os.environ['BCISTREAM_ROOT'], 'framework', 'qtgui', 'configurations.ui')
self.main = QUiLoader().load(frame)
self.main.widget_restart.hide()
self.config = ConfigManager()
self.original_config = ConfigManager(os.path.join(
os.environ['BCISTREAM_ROOT'], 'assets', 'bciframework.default'))
theme = self.config.get('framework', 'theme', 'light')
self.main.radioButton_light.setChecked(theme == 'light')
self.main.radioButton_dark.setChecked(theme == 'dark')
self.connect()
# ----------------------------------------------------------------------
def show(self) -> None:
"""Show frame."""
center = QDesktopWidget().availableGeometry().center()
geometry = self.main.frameGeometry()
geometry.moveCenter(center)
self.main.move(geometry.topLeft())
self.main.show()
# ----------------------------------------------------------------------
def connect(self) -> None:
"""Connect events."""
self.main.lineEdit_user_directory.setText(os.environ['BCISTREAM_HOME'])
self.main.lineEdit_projects_directory.setText(
os.path.join(os.environ['BCISTREAM_HOME'], 'default_extensions'))
self.main.lineEdit_records_directory.setText(
os.path.join(os.environ['BCISTREAM_HOME'], 'records'))
self.main.radioButton_light.clicked.connect(
lambda: self.config.set('framework', 'theme', 'light', save=True))
self.main.radioButton_dark.clicked.connect(
lambda: self.config.set('framework', 'theme', 'dark', save=True))
self.main.pushButton_restore_projects.clicked.connect(
self.restore_projects)
self.main.pushButton_reset_projects.clicked.connect(self.reset_projects)
self.main.pushButton_remove_records.clicked.connect(self.remove_records)
self.main.pushButton_reset_montages.clicked.connect(self.reset_montages)
self.main.pushButton_reset_connections.clicked.connect(
self.reset_connections)
# ----------------------------------------------------------------------
def restore_projects(self, *args, **kwargs) -> None:
"""Copy defautl project into the user projects directory."""
shutil.copytree(os.path.join(os.environ['BCISTREAM_ROOT'], 'default_extensions'),
os.path.join(os.environ['BCISTREAM_HOME'], 'default_extensions'), dirs_exist_ok=True)
shutil.copytree(os.path.join(os.environ['BCISTREAM_ROOT'], 'kafka_scripts'),
os.path.join(os.environ['BCISTREAM_HOME'], 'kafka_scripts'), dirs_exist_ok=True)
# ----------------------------------------------------------------------
def reset_projects(self, *args, **kwargs) -> None:
"""Remove user projects nad copy the default projects."""
shutil.rmtree(os.path.join(
os.environ['BCISTREAM_HOME'], 'default_extensions'))
shutil.copytree(os.path.join(os.environ['BCISTREAM_ROOT'], 'default_extensions'),
os.path.join(os.environ['BCISTREAM_HOME'], 'default_extensions'))
shutil.rmtree(os.path.join(
os.environ['BCISTREAM_HOME'], 'kafka_scripts'))
shutil.copytree(os.path.join(os.environ['BCISTREAM_ROOT'], 'kafka_scripts'),
os.path.join(os.environ['BCISTREAM_HOME'], 'kafka_scripts'))
# ----------------------------------------------------------------------
def remove_records(self, *args, **kwargs) -> None:
"""Clear the directory with records."""
shutil.rmtree(os.path.join(os.environ['BCISTREAM_HOME'], 'records'))
# ----------------------------------------------------------------------
def reset_montages(self, *args, **kwargs) -> None:
"""Remove user defined montages."""
self.reset_section('montages')
# ----------------------------------------------------------------------
def reset_connections(self) -> None:
"""Remove user modified connection configurations."""
self.reset_section('connection')
# ----------------------------------------------------------------------
def reset_section(self, section: str) -> None:
"""Overwrite user section with default section."""
self.config.remove_section(section)
for option in self.original_config.options(section):
self.config.set(section, option,
self.original_config.get(section, option))
self.config.save()
