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)
def __init__(self, module_pkgname, assets_dir, hooksdir): super(MainWindow, self).__init__(QtCore.Qt.Horizontal) self._win_title_base = 'Node.gl viewer' self.setWindowTitle(self._win_title_base) self._module_pkgname = module_pkgname self._scripts_mgr = ScriptsManager(module_pkgname) self._hooksdir = hooksdir medias = None if assets_dir: medias = [] for f in sorted(os.listdir(assets_dir)): ext = f.rsplit('.', 1)[-1].lower() path = op.join(assets_dir, f) if op.isfile(path) and ext in ('mp4', 'mkv', 'avi', 'webm', 'mov', 'lrv'): try: media = Media(path) except: pass else: medias.append(media) self._medias = medias get_scene_func = self._get_scene self._hooks = None self._config = Config(module_pkgname) # Apply previous geometry (position + dimensions) rect = self._config.get('geometry') if rect: geometry = QtCore.QRect(*rect) self.setGeometry(geometry) gl_view = GLView(get_scene_func, self._config) graph_view = GraphView(get_scene_func, self._config) export_view = ExportView(get_scene_func, self._config) serial_view = SerialView(get_scene_func) self._tabs = [ ('Player view', gl_view), ('Graph view', graph_view), ('Export', export_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(gl_view.set_aspect_ratio) self._scene_toolbar.aspectRatioChanged.connect(export_view.set_aspect_ratio) self._scene_toolbar.aspectRatioChanged.connect(self._config.set_aspect_ratio) self._scene_toolbar.samplesChanged.connect(gl_view.set_samples) self._scene_toolbar.samplesChanged.connect(self._config.set_samples) self._scene_toolbar.frameRateChanged.connect(gl_view.set_frame_rate) self._scene_toolbar.frameRateChanged.connect(graph_view.set_frame_rate) self._scene_toolbar.frameRateChanged.connect(export_view.set_frame_rate) 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(gl_view.set_clear_color) self._scene_toolbar.clearColorChanged.connect(self._config.set_clear_color) self._scene_toolbar.backendChanged.connect(gl_view.set_backend) self._scene_toolbar.backendChanged.connect(self._config.set_backend) self._scene_toolbar.hudChanged.connect(self._config.set_hud) self._errbuf = QtWidgets.QPlainTextEdit() self._errbuf.setFont(QtGui.QFontDatabase.systemFont(QtGui.QFontDatabase.FixedFont)) self._errbuf.setReadOnly(True) self._errbuf.hide() self._hooks_lbl = QtWidgets.QLabel() self._hooks_layout = QtWidgets.QHBoxLayout() self._hooks_layout.addWidget(self._hooks_lbl) self._hooks_widget = QtWidgets.QWidget() self._hooks_widget.setLayout(self._hooks_layout) self._hooks_widget.hide() tabs_and_errbuf = QtWidgets.QVBoxLayout() tabs_and_errbuf.addWidget(self._tab_widget) tabs_and_errbuf.addWidget(self._errbuf) tabs_and_errbuf.addWidget(self._hooks_widget) 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) # 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)