class DummyWindow(QtWidgets.QWidget):
"""Wrap the HooksView with an additional button to trigger a read of the data + status change"""
def __init__(self):
super().__init__()
self._hooks_ctl = HooksController(None, DummyHooksCaller())
self._hooks_view = HooksView(self._hooks_ctl)
action_btn = QtWidgets.QPushButton("Refresh sessions")
action_btn.clicked.connect(self._refresh)
layout = QtWidgets.QVBoxLayout()
layout.addWidget(self._hooks_view)
layout.addWidget(action_btn)
self.setLayout(layout)
@QtCore.Slot()
def _refresh(self):
self._hooks_ctl.refresh_sessions()
@QtCore.Slot(QtGui.QCloseEvent)
def closeEvent(self, close_event):
self._hooks_ctl.stop_threads()
super().closeEvent(close_event)
class MainWindow(QtWidgets.QSplitter):
sceneLoaded = QtCore.Signal(dict)
error = QtCore.Signal(str)
def __init__(self, module_pkgname, hooks_scripts):
super().__init__(QtCore.Qt.Horizontal)
self._win_title_base = "Node.gl controller"
self.setWindowTitle(self._win_title_base)
self._module_pkgname = module_pkgname
self._scripts_mgr = ScriptsManager(module_pkgname)
self._hooks_caller = HooksCaller(hooks_scripts)
self._hooks_ctl = HooksController(self._get_scene, self._hooks_caller)
get_scene_func = self._get_scene
self._config = Config(module_pkgname)
# Apply previous geometry (position + dimensions)
rect = self._config.get("geometry")
if rect:
geometry = QtCore.QRect(*rect)
self.setGeometry(geometry)
graph_view = GraphView(get_scene_func, self._config)
export_view = ExportView(get_scene_func, self._config)
hooks_view = HooksView(self._hooks_ctl, self._config)
self._medias_view = MediasView(self._config)
serial_view = SerialView(get_scene_func)
self._tabs = [
("Hooks", hooks_view),
("Graph view", graph_view),
("Export", export_view),
("Medias", self._medias_view),
("Serialization", serial_view),
]
self._last_tab_index = -1
self._tab_widget = QtWidgets.QTabWidget()
for tab_name, tab_view in self._tabs:
self._tab_widget.addTab(tab_view, tab_name)
self._tab_widget.currentChanged.connect(self._currentTabChanged)
self._scene_toolbar = Toolbar(self._config)
self._scene_toolbar.sceneChanged.connect(self._scene_changed)
self._scene_toolbar.sceneChanged.connect(self._scene_changed_hook)
self._scene_toolbar.sceneChanged.connect(self._config.scene_changed)
self._scene_toolbar.aspectRatioChanged.connect(
self._config.set_aspect_ratio)
self._scene_toolbar.samplesChanged.connect(self._config.set_samples)
self._scene_toolbar.frameRateChanged.connect(
self._config.set_frame_rate)
self._scene_toolbar.logLevelChanged.connect(self._config.set_log_level)
self._scene_toolbar.clearColorChanged.connect(
self._config.set_clear_color)
self._scene_toolbar.backendChanged.connect(self._config.set_backend)
self._errbuf = QtWidgets.QPlainTextEdit()
self._errbuf.setFont(
QtGui.QFontDatabase.systemFont(QtGui.QFontDatabase.FixedFont))
self._errbuf.setReadOnly(True)
self._errbuf.hide()
tabs_and_errbuf = QtWidgets.QVBoxLayout()
tabs_and_errbuf.addWidget(self._tab_widget)
tabs_and_errbuf.addWidget(self._errbuf)
tabs_and_errbuf_widget = QtWidgets.QWidget()
tabs_and_errbuf_widget.setLayout(tabs_and_errbuf)
self.addWidget(self._scene_toolbar)
self.addWidget(tabs_and_errbuf_widget)
self.setStretchFactor(1, 1)
self._scene_toolbar.reload_btn.clicked.connect(
self._scripts_mgr.reload) # TODO: drop
self._scripts_mgr.error.connect(self._all_scripts_err)
self._scripts_mgr.scriptsChanged.connect(
self._scene_toolbar.on_scripts_changed)
self._scripts_mgr.start()
self.error.connect(self._scene_err)
self.sceneLoaded.connect(self._scene_loaded)
# Load the previous scene if the current and previously loaded
# module packages match
prev_pkgname = self._config.get("pkg")
prev_module = self._config.get("module")
prev_scene = self._config.get("scene")
if prev_pkgname == module_pkgname:
self._scene_toolbar.load_scene_from_name(prev_module, prev_scene)
@QtCore.Slot(str)
def _scene_err(self, err_str):
if err_str:
self._errbuf.setPlainText(err_str)
self._errbuf.show()
sys.stderr.write(err_str)
else:
self._errbuf.hide()
@QtCore.Slot(str)
def _all_scripts_err(self, err_str):
self._scene_toolbar.clear_scripts()
self._scene_err(err_str)
def _get_scene(self, **cfg_overrides):
cfg = self._scene_toolbar.get_cfg()
if cfg["scene"] is None:
return None
medias = self._medias_view.get_medias()
cfg["medias"] = medias if medias else None
cfg["files"] = []
cfg.update(cfg_overrides)
self._scripts_mgr.inc_query_count()
self._scripts_mgr.pause()
ret = query_scene(self._module_pkgname, **cfg)
self._scripts_mgr.update_filelist(ret["filelist"])
self._scripts_mgr.update_modulelist(ret["modulelist"])
self._scripts_mgr.resume()
self._scripts_mgr.dec_query_count()
if "error" in ret:
self.error.emit(ret["error"])
return None
self.error.emit(None)
self.sceneLoaded.emit(ret)
return ret
@QtCore.Slot(str, str)
def _scene_changed(self, module_name, scene_name):
self.setWindowTitle(
f"{self._win_title_base} - {module_name}.{scene_name}")
self._currentTabChanged(self._tab_widget.currentIndex())
@QtCore.Slot(str, str)
def _scene_changed_hook(self, module_name, scene_name):
self._hooks_ctl.process(module_name, scene_name)
@QtCore.Slot(dict)
def _scene_loaded(self, cfg):
self._scene_toolbar.set_cfg(cfg)
def _emit_geometry(self):
geometry = (self.x(), self.y(), self.width(), self.height())
self._config.geometry_changed(geometry)
@QtCore.Slot(QtGui.QResizeEvent)
def resizeEvent(self, resize_event):
super().resizeEvent(resize_event)
self._emit_geometry()
@QtCore.Slot(QtGui.QMoveEvent)
def moveEvent(self, move_event):
super().moveEvent(move_event)
self._emit_geometry()
@QtCore.Slot(QtGui.QCloseEvent)
def closeEvent(self, close_event):
self._hooks_ctl.stop_threads()
for name, widget in self._tabs:
widget.close()
super().closeEvent(close_event)
@QtCore.Slot(int)
def _currentTabChanged(self, index):
next_view = self._tabs[index][1]
prev_view = self._tabs[self._last_tab_index][1]
if index != self._last_tab_index and hasattr(prev_view, "leave"):
prev_view.leave()
if hasattr(next_view, "enter"):
next_view.enter()
self._last_tab_index = index