test_apply_textures working

Author: JulienChampagnol

schemas/apply_textures.json

@@ -1,5 +1,5 @@
 {
-  "rpc": "toggle_object_visibility",
+  "rpc": "apply_textures",
   "type": "object",
   "properties": {
     "id": {

tests/test_protocol.py

@@ -3,62 +3,70 @@
 images_dir_path = os.path.abspath("./tests/data/images")
 
 
-# def test_create_visualization(server):
-#     server.call("create_visualization")
-#     abs_path = os.path.join(images_dir_path, "create_visualization.jpeg")
-#     assert server.compare_image(3, abs_path) == True
+def test_create_visualization(server):
+    server.call("create_visualization")
+    abs_path = os.path.join(images_dir_path, "create_visualization.jpeg")
+    assert server.compare_image(3, abs_path) == True
 
 
-# def test_reset_camera(server):
-#     server.call("reset_camera")
-#     abs_path = os.path.join(images_dir_path, "reset_camera.jpeg")
-#     assert server.compare_image(3, abs_path) == True
+def test_reset_camera(server):
+    server.call("reset_camera")
+    abs_path = os.path.join(images_dir_path, "reset_camera.jpeg")
+    assert server.compare_image(3, abs_path) == True
 
 
-# def test_create_object_pipeline(server):
-#     server.call("create_object_pipeline", [{"id": "123456", "file_name": "hat.vtp"}])
-#     abs_path = os.path.join(images_dir_path, "create_object_pipeline.jpeg")
-#     assert server.compare_image(3, abs_path) == True
+def test_create_object_pipeline(server):
+    server.call("create_object_pipeline", [{"id": "123456", "file_name": "hat.vtp"}])
+    abs_path = os.path.join(images_dir_path, "create_object_pipeline.jpeg")
+    assert server.compare_image(3, abs_path) == True
 
 
-# def test_delete_object_pipeline(server):
-#     server.call("create_object_pipeline", [{"id": "123456789", "file_name": "hat.vtp"}])
-#     abs_path = os.path.join(images_dir_path, "create_object_pipeline.jpeg")
-#     assert server.compare_image(3, abs_path) == True
+def test_delete_object_pipeline(server):
+    server.call("create_object_pipeline", [{"id": "123456789", "file_name": "hat.vtp"}])
+    abs_path = os.path.join(images_dir_path, "create_object_pipeline.jpeg")
+    assert server.compare_image(3, abs_path) == True
 
-#     server.call("delete_object_pipeline", [{"id": "123456789"}])
-#     abs_path = os.path.join(images_dir_path, "delete_object_pipeline.jpeg")
-#     assert server.compare_image(3, abs_path) == True
+    server.call("delete_object_pipeline", [{"id": "123456789"}])
+    abs_path = os.path.join(images_dir_path, "delete_object_pipeline.jpeg")
+    assert server.compare_image(3, abs_path) == True
 
 
-# def test_toggle_object_visibility(server):
-#     server.call("create_object_pipeline", [{"id": "123456789", "file_name": "hat.vtp"}])
-#     abs_path = os.path.join(images_dir_path, "create_object_pipeline.jpeg")
-#     assert server.compare_image(3, abs_path) == True
+def test_toggle_object_visibility(server):
+    server.call("create_object_pipeline", [{"id": "123456789", "file_name": "hat.vtp"}])
+    abs_path = os.path.join(images_dir_path, "create_object_pipeline.jpeg")
+    assert server.compare_image(3, abs_path) == True
 
-#     server.call("toggle_object_visibility", [{"id": "123456789", "is_visible": False}])
-#     abs_path = os.path.join(images_dir_path, "toggle_object_visibility_1.jpeg")
-#     assert server.compare_image(3, abs_path) == True
+    server.call("toggle_object_visibility", [{"id": "123456789", "is_visible": False}])
+    abs_path = os.path.join(images_dir_path, "toggle_object_visibility_1.jpeg")
+    assert server.compare_image(3, abs_path) == True
 
-#     server.call("toggle_object_visibility", [{"id": "123456789", "is_visible": True}])
-#     abs_path = os.path.join(images_dir_path, "toggle_object_visibility_2.jpeg")
-#     assert server.compare_image(3, abs_path) == True
+    server.call("toggle_object_visibility", [{"id": "123456789", "is_visible": True}])
+    abs_path = os.path.join(images_dir_path, "toggle_object_visibility_2.jpeg")
+    assert server.compare_image(3, abs_path) == True
 
 
 def test_apply_textures(server):
 
-    server.call("create_object_pipeline", [{"id": "123456789", "file_name": "hat.vtp"}])
-    abs_path = os.path.join(images_dir_path, "create_object_pipeline.jpeg")
+    server.call(
+        "create_object_pipeline",
+        [{"id": "123456789", "file_name": "emperor_penguin.vtp"}],
+    )
+    abs_path = os.path.join(images_dir_path, "apply_textures_1.jpeg")
     assert server.compare_image(3, abs_path) == True
 
     server.call(
         "apply_textures",
         [
             {
                 "id": "123456789",
-                "textures": {"texture_name": "lambert2SG", "texture_file_name": ""},
+                "textures": [
+                    {
+                        "texture_name": "PenguinMat",
+                        "texture_file_name": "penguin_colors.vti",
+                    }
+                ],
             }
         ],
     )
-    abs_path = os.path.join(images_dir_path, "apply_textures.jpeg")
-    assert server.compare_image(1, abs_path) == True
+    abs_path = os.path.join(images_dir_path, "apply_textures_2.jpeg")
+    assert server.compare_image(3, abs_path) == True

vtk_protocol.py

@@ -122,6 +122,7 @@ def toggle_object_visibility(self, params):
 
     @exportRpc(apply_textures_json["rpc"])
     def apply_textures(self, params):
+        print(f"{params=}", flush=True)
         id = params["id"]
         textures = params["textures"]
         textures_array = []
@@ -138,10 +139,13 @@ def apply_textures(self, params):
         for index, value in enumerate(textures):
             texture_name = value["texture_name"]
             texture_file_name = value["texture_file_name"]
+            print(f"{texture_name=} {texture_file_name=}", flush=True)
 
             new_texture = vtk.vtkTexture()
             image_reader = vtk.vtkXMLImageDataReader()
-            image_reader.SetFileName(os.path.join(DATA_FOLDER_PATH, texture_file_name))
+            image_reader.SetFileName(
+                os.path.join(self.DATA_FOLDER_PATH, texture_file_name)
+            )
 
             shader_texture_name = f"VTK_TEXTURE_UNIT_{index}"
             polydata_mapper.MapDataArrayToMultiTextureAttribute(