feat(fast): add a fast option to choose rendering pipeline

Author: jourdain

src/e3sm_quickview/app.py

@@ -19,11 +19,21 @@
 from e3sm_quickview.components import css, dialogs, doc, drawers, file_browser, toolbars
 from e3sm_quickview.pipeline import EAMVisSource
 from e3sm_quickview.utils import cli, compute
-from e3sm_quickview.view_manager2 import ViewManager
 
 v3.enable_lab()
 
 
+def create_view_manager(single_view, server, source):
+    if single_view:
+        from e3sm_quickview.view_manager2 import ViewManager
+
+        return ViewManager(server, source)
+
+    from e3sm_quickview.view_manager import ViewManager
+
+    return ViewManager(server, source)
+
+
 class EAMApp(TrameApp):
     def __init__(self, server=None):
         super().__init__(server)
@@ -63,7 +73,7 @@ def __init__(self, server=None):
         self.source = EAMVisSource()
 
         # Helpers
-        self.view_manager = ViewManager(self.server, self.source)
+        self.view_manager = create_view_manager(args.fast, self.server, self.source)
         self.file_browser = file_browser.ParaViewFileBrowser(
             self.server,
             prefix="pv_files",
@@ -210,13 +220,19 @@ def _build_ui(self, **_):
                                 toolbars.Animation()
 
                             # View of all the variables
-                            with rca.ImageStream(
-                                self.view_manager._render_window,
-                                encoder="turbo-jpeg",
-                                ctx_name="view",
-                            ):
-                                # To debug vtkRenderWindow content
-                                # html.Img(src=["image?.src"], height="300px")
+                            if self.view_manager.use_image_stream:
+                                with rca.ImageStream(
+                                    self.view_manager._render_window,
+                                    encoder="turbo-jpeg",
+                                    ctx_name="view",
+                                ):
+                                    # To debug vtkRenderWindow content
+                                    # html.Img(src=["image?.src"], height="300px")
+                                    client.ServerTemplate(
+                                        name=("active_layout", "auto_layout"),
+                                        v_if="variables_selected.length",
+                                    )
+                            else:
                                 client.ServerTemplate(
                                     name=("active_layout", "auto_layout"),
                                     v_if="variables_selected.length",

src/e3sm_quickview/utils/cli.py

@@ -31,5 +31,11 @@ def configure_and_parse(parser):
         action="store_true",
         help="Use user home as home for file browsing",
     )
+    parser.add_argument(
+        "--fast",
+        dest="fast",
+        action="store_true",
+        help="Use a single vtkRenderWindow to share GPU memory",
+    )
 
     return parser.parse_known_args()[0]

src/e3sm_quickview/view_manager.py

@@ -310,6 +310,7 @@ def _build_ui(self):
 class ViewManager(TrameComponent):
     def __init__(self, server, source):
         super().__init__(server)
+        self.use_image_stream = False
         self.source = source
         self._var2view = {}
         self._camera_sync_in_progress = False

src/e3sm_quickview/view_manager2.py

@@ -319,6 +319,7 @@ def _build_ui(self):
 class ViewManager(TrameComponent):
     def __init__(self, server, source):
         super().__init__(server)
+        self.use_image_stream = True
         self._camera = vtkCamera(parallel_projection=1)
         self._render_window = vtkRenderWindow()
         self._render_window.OffScreenRenderingOn()