fix(serializer): add encoding option

Kitware · Feb 13, 2024 · 9c20a89 · 9c20a89
1 parent abf0bd1
commit 9c20a89
Show file tree

Hide file tree

Showing 2 changed files with 18 additions and 2 deletions.
diff --git a/trame_vtk/modules/vtk/serializers/__init__.py b/trame_vtk/modules/vtk/serializers/__init__.py
@@ -7,7 +7,7 @@
 from .export import extract_array_hash
 from .synchronization_context import SynchronizationContext
 from .utils import reference_id
-from .initialize import encode_lut
+from .initialize import encode_lut, skip_light
 
 logger = logging.getLogger(__name__)
 # By default, only show critical messages for serializers
@@ -17,8 +17,16 @@
     # If this environment variable is set, print out all messages
     logger.setLevel(logging.DEBUG)
 
+
+def configure_serializer(**options):
+    skip_light(options.get("skip_light", True))
+    encode_lut(options.get("encode_lut", True))
+
+
 __all__ = [
+    "configure_serializer",
     "encode_lut",
+    "skip_light",
     "reference_id",
     "initialize_serializers",
     "mesh",

diff --git a/trame_vtk/modules/vtk/serializers/initialize.py b/trame_vtk/modules/vtk/serializers/initialize.py
@@ -33,13 +33,19 @@
 logger = logging.getLogger(__name__)
 
 CONVERT_LUT = False
+SKIP_LIGHT = False
 
 
 def encode_lut(value=True):
     global CONVERT_LUT
     CONVERT_LUT = value
 
 
+def skip_light(value=True):
+    global SKIP_LIGHT
+    SKIP_LIGHT = value
+
+
 def lookup_table_serializer_selector(*args, **kwargs):
     global CONVERT_LUT
     if CONVERT_LUT:
@@ -125,7 +131,9 @@ def initialize_serializers():
             "vtkOpenGLRenderer",
         ],
         # Lights
-        light_serializer: [
+        light_serializer: []
+        if SKIP_LIGHT
+        else [
             "vtkLight",
             "vtkPVLight",
             "vtkOpenGLLight",