test and adds a visibility_component rpc

Author: MaxNumerique

src/opengeodeweb_viewer/rpc/model/model_protocols.py

@@ -90,3 +90,13 @@ def setModelComponentsColor(self, rpc_params: RpcParams) -> None:
         params = schemas.ComponentsColor.from_dict(rpc_params)
         color = params.color
         self.SetBlocksColor(params.id, params.block_ids, color.r, color.g, color.b)
+
+    @exportRpc(model_prefix + model_schemas_dict["components_visibility"]["rpc"])
+    def setModelComponentsVisibility(self, rpc_params: RpcParams) -> None:
+        validate_schema(
+            rpc_params,
+            self.model_schemas_dict["components_visibility"],
+            self.model_prefix,
+        )
+        params = schemas.ComponentsVisibility.from_dict(rpc_params)
+        self.SetBlocksVisibility(params.id, params.block_ids, params.visibility)

src/opengeodeweb_viewer/rpc/model/schemas/__init__.py

@@ -2,3 +2,4 @@
 from .register import *
 from .deregister import *
 from .components_color import *
+from .components_visibility import *

src/opengeodeweb_viewer/rpc/model/schemas/components_color.py

@@ -1,17 +1,18 @@
+from dataclasses_json import DataClassJsonMixin
 from dataclasses import dataclass
-from opengeodeweb_microservice.schemas import Color
+from typing import Optional, List
 
 
 @dataclass
-class ComponentsColor:
-    id: str
-    block_ids: list[int]
-    color: Color
+class ColorClass(DataClassJsonMixin):
+    b: int
+    g: int
+    r: int
+    a: Optional[float] = None
+
 
-    @classmethod
-    def from_dict(cls, data: dict) -> "ComponentsColor":
-        return cls(
-            id=data["id"],
-            block_ids=data["block_ids"],
-            color=Color(**data["color"]),
-        )
+@dataclass
+class ComponentsColor(DataClassJsonMixin):
+    block_ids: List[int]
+    color: ColorClass
+    id: str

src/opengeodeweb_viewer/rpc/model/schemas/components_visibility.json

@@ -0,0 +1,26 @@
+{
+  "rpc": "components_visibility",
+  "type": "object",
+  "properties": {
+    "id": {
+      "type": "string",
+      "minLength": 1
+    },
+    "block_ids": {
+      "type": "array",
+      "items": {
+        "type": "integer"
+      },
+      "minItems": 1
+    },
+    "visibility": {
+      "type": "boolean"
+    }
+  },
+  "required": [
+    "id",
+    "block_ids",
+    "visibility"
+  ],
+  "additionalProperties": false
+}

src/opengeodeweb_viewer/rpc/model/schemas/components_visibility.py

@@ -0,0 +1,10 @@
+from dataclasses_json import DataClassJsonMixin
+from dataclasses import dataclass
+from typing import List
+
+
+@dataclass
+class ComponentsVisibility(DataClassJsonMixin):
+    block_ids: List[int]
+    id: str
+    visibility: bool

tests/data/images/model/components_color.jpeg

@@ -53,3 +53,34 @@ def test_deregister_model(
         [{"id": "123456789"}],
     )
     assert server.compare_image("model/deregister.jpeg") == True
+
+
+def test_components_color_model(
+    server: ServerMonitor, dataset_factory: Callable[..., str]
+) -> None:
+
+    test_register_model_cube(server, dataset_factory)
+
+    server.call(
+        VtkModelView.model_prefix
+        + VtkModelView.model_schemas_dict["components_color"]["rpc"],
+        [{"id": "123456789", "block_ids": [48, 49], "color": {"r": 255, "g": 0, "b": 0}}],
+    )
+    assert server.compare_image("model/components_color.jpeg") == True
+
+
+def test_components_visibility_color_model(
+    server: ServerMonitor, dataset_factory: Callable[..., str]
+) -> None:
+    test_register_model_cube(server, dataset_factory)
+
+    server.call(
+        VtkModelView.model_prefix + "components_visibility",
+        [{"id": "123456789", "block_ids": [48, 49], "visibility": False}],
+    )
+    
+    server.call(
+        VtkModelView.model_prefix + "components_color",
+        [{"id": "123456789", "block_ids": list(range(36, 47)), "color": {"r": 0, "g": 255, "b": 0}}],
+    )
+    assert server.compare_image("model/components_visibility_color.jpeg") == True