wip test protocols

Author: JulienChampagnol

src/opengeodeweb_viewer/object/object_methods.py

@@ -141,11 +141,25 @@ def SetPolygonsVisibility(self, id, visibility):
         actor.SetVisibility(visibility)
         self.render()
 
+
     def SetPolygonsColor(self, id, color):
         actor = self.get_object(id)["actor"]
         actor.GetProperty().SetColor(color)
         self.render()
 
+    def SetPolyhedronsVisibility(self, id, visibility):
+        actor = self.get_object(id)["actor"]
+        actor.SetVisibility(visibility)
+        self.render()
+    
+    def SetPolyhedronsColor(self, id, color):
+        reader = self.get_object(id)["reader"]
+        cells = reader.GetOutput().GetCellData()
+        mapper = self.get_object(id)["mapper"]
+        mapper.ScalarVisibilityOff()
+        cells.SetColor(color)
+        
+
     def clearColors(self, id):
         db = self.get_object(id)
         mapper = db["mapper"]

src/opengeodeweb_viewer/rpc/mesh/polyhedrons/polyhedrons_protocols.py

@@ -0,0 +1,50 @@
+# Standard library imports
+import os
+
+# Third party imports
+from wslink import register as exportRpc
+
+# Local application imports
+from opengeodeweb_viewer.utils_functions import get_schemas_dict, validate_schema
+from opengeodeweb_viewer.rpc.mesh.mesh_protocols import VtkMeshView
+
+class VtkMeshPolyhedronsView(VtkMeshView):
+    mesh_polyhedrons_prefix = "opengeodeweb_viewer.mesh.polyhedrons."
+    mesh_polyhedrons_schemas_dict = get_schemas_dict(os.path.join(os.path.dirname(__file__), "schemas"))
+
+    def __init__(self):
+        super().__init__()
+
+    # @exportRpc(mesh_polyhedrons_prefix + mesh_polyhedrons_schemas_dict["visibility"]["rpc"])
+    # def setMeshPolyhedronsVisibility(self, params):
+    #     print(self.mesh_polyhedrons_prefix + self.mesh_polyhedrons_schemas_dict["visibility"]["rpc"], f"{params=}", flush=True)
+    #     validate_schema(params, self.mesh_polyhedrons_schemas_dict["visibility"])
+    #     id = params["id"]
+    #     visibility = bool(params["visibility"])
+    #     self.SetPolyhedronsVisibility(id, visibility)
+
+    @exportRpc(mesh_polyhedrons_prefix + mesh_polyhedrons_schemas_dict["color"]["rpc"])
+    def setMeshPolyhedronsColor(self, params):
+        print(self.mesh_polyhedrons_prefix + self.mesh_polyhedrons_schemas_dict["color"]["rpc"], f"{params=}", flush=True)
+        validate_schema(params, self.mesh_polyhedrons_schemas_dict["color"])
+        print("id", params["id"], flush=True)
+        id = params["id"]
+        print("color", params["color"], flush=True)
+        red, green, blue = params["color"]["r"]/255, params["color"]["g"]/255, params["color"]["b"]/255
+        self.SetPolyhedronsColor(id, [red, green, blue])
+
+    # @exportRpc(mesh_polyhedrons_prefix + mesh_polyhedrons_schemas_dict["vertex_attribute"]["rpc"])
+    # def setMeshPolyhedronsVertexAttribute(self, params):
+    #     print(self.mesh_polyhedrons_prefix + self.mesh_polyhedrons_schemas_dict["vertex_attribute"]["rpc"], f"{params=}", flush=True)
+    #     validate_schema(params, self.mesh_polyhedrons_schemas_dict["vertex_attribute"])
+    #     id = params["id"]
+    #     name = str(params["name"])
+    #     self.setMeshVertexAttribute(id, name)
+
+    # @exportRpc(mesh_polyhedrons_prefix + mesh_polyhedrons_schemas_dict["polygon_attribute"]["rpc"])
+    # def setMeshPolyhedronsPolygonAttribute(self, params):
+    #     print(self.mesh_polyhedrons_prefix + self.mesh_polyhedrons_schemas_dict["polygon_attribute"]["rpc"], f"{params=}", flush=True)
+    #     validate_schema(params, self.mesh_polyhedrons_schemas_dict["polygon_attribute"])
+    #     id = params["id"]
+    #     name = str(params["name"])
+    #     self.setMeshPolygonAttribute(id, name)

src/opengeodeweb_viewer/rpc/mesh/polyhedrons/schemas/color.json

@@ -0,0 +1,46 @@
+{
+  "rpc": "color",
+  "type": "object",
+  "properties": {
+    "id": {
+      "type": "string"
+    },
+    "color": {
+      "type": "object",
+      "properties": {
+        "r": {
+          "type": "integer",
+          "minimum": 0,
+          "maximum": 255
+        },
+        "g": {
+          "type": "integer",
+          "minimum": 0,
+          "maximum": 255
+        },
+        "b": {
+          "type": "integer",
+          "minimum": 0,
+          "maximum": 255
+        },
+        "a": {
+          "type": "number",
+          "minimum": 0,
+          "maximum": 1,
+          "default": 1
+        }
+      },
+      "required": [
+        "r",
+        "g",
+        "b"
+      ],
+      "additionalProperties": false
+    }
+  },
+  "required": [
+    "id",
+    "color"
+  ],
+  "additionalProperties": false
+}

src/opengeodeweb_viewer/rpc/mesh/polyhedrons/schemas/polygon_attribute.json

@@ -0,0 +1,17 @@
+{
+  "rpc": "polygon_attribute",
+  "type": "object",
+  "properties": {
+    "id": {
+      "type": "string"
+    },
+    "name": {
+      "type": "string"
+    }
+  },
+  "required": [
+    "id",
+    "name"
+  ],
+  "additionalProperties": false
+}

src/opengeodeweb_viewer/rpc/mesh/polyhedrons/schemas/vertex_attribute.json

@@ -0,0 +1,17 @@
+{
+  "rpc": "vertex_attribute",
+  "type": "object",
+  "properties": {
+    "id": {
+      "type": "string"
+    },
+    "name": {
+      "type": "string"
+    }
+  },
+  "required": [
+    "id",
+    "name"
+  ],
+  "additionalProperties": false
+}

src/opengeodeweb_viewer/rpc/mesh/polyhedrons/schemas/visibility.json

@@ -0,0 +1,17 @@
+{
+  "rpc": "visibility",
+  "type": "object",
+  "properties": {
+    "id": {
+      "type": "string"
+    },
+    "visibility": {
+      "type": "boolean"
+    }
+  },
+  "required": [
+    "id",
+    "visibility"
+  ],
+  "additionalProperties": false
+}

src/tests/data/images/mesh/polyhedrons/register.jpeg

@@ -0,0 +1,20 @@
+# Standard library imports
+
+# Third party imports
+from opengeodeweb_viewer.rpc.mesh.mesh_protocols import VtkMeshView
+from opengeodeweb_viewer.rpc.mesh.polyhedrons.polyhedrons_protocols import VtkMeshPolyhedronsView
+
+# Local application imports
+
+
+def test_register_mesh(server):
+
+    server.call(VtkMeshView.mesh_prefix + VtkMeshView.mesh_schemas_dict["register"]["rpc"], [{"id": "123456789", "file_name": "hybrid_solid.vtu"}])
+    assert server.compare_image(3, "mesh/polyhedrons/register.jpeg") == True
+
+def test_polyhedrons_color(server):
+
+    test_register_mesh(server)
+
+    server.call(VtkMeshPolyhedronsView.mesh_polyhedrons_prefix + VtkMeshPolyhedronsView.mesh_polyhedrons_schemas_dict["color"]["rpc"], [{"id": "123456789", "color": {"r": 255, "g": 0, "b": 0}}])
+    assert server.compare_image(3, "mesh/polyhedrons/color.jpeg") == True