changes, new rpc : picked_ids formerly get_mouse

Author: SpliiT

src/opengeodeweb_viewer/rpc/viewer/schemas/get_mouse.json renamed to src/opengeodeweb_viewer/rpc/viewer/schemas/picked_ids.json

@@ -1,5 +1,5 @@
 {
-    "rpc": "get_mouse",
+    "rpc": "picked_ids",
     "schema": {
         "type": "object",
         "properties": {
@@ -8,11 +8,18 @@
             },
             "y": {
                 "type": "number"
+            },
+            "ids": {
+                "type": "array",
+                "items": {
+                    "type": "string"
+                }
             }
         },
         "required": [
             "x",
-            "y"
+            "y",
+            "ids"
         ]
     }
 }

src/opengeodeweb_viewer/rpc/viewer/viewer_protocols.py

@@ -1,5 +1,6 @@
 # Standard library imports
 import json
+import math
 import os
 from pathlib import Path
 
@@ -147,12 +148,43 @@ def reset(self, params):
         renderWindow = self.getView("-1")
         renderWindow.GetRenderers().GetFirstRenderer().RemoveAllViewProps()
 
-    @exportRpc(prefix + schemas_dict["get_mouse"]["rpc"])
-    def getMouse(self, params):
-        print(self.schemas_dict["get_mouse"]["rpc"], f"{params=}", flush=True)
-        validate_schema(params, self.schemas_dict["get_mouse"])
+    def computeEpsilon(self, renderer, z):
+        renderer.SetDisplayPoint(0, 0, z)
+        renderer.DisplayToWorld()
+        windowLowerLeft = renderer.GetWorldPoint()
+        size = renderer.GetRenderWindow().GetSize()
+        renderer.SetDisplayPoint(size[0], size[1], z)
+        renderer.DisplayToWorld()
+        windowUpperRight = renderer.GetWorldPoint()
+        epsilon = 0
+        for i in range(3):
+            epsilon += (windowUpperRight[i] - windowLowerLeft[i]) * (
+                windowUpperRight[i] - windowLowerLeft[i]
+            )
+        return math.sqrt(epsilon) * 0.0125
+
+    @exportRpc(prefix + schemas_dict["picked_ids"]["rpc"])
+    def pickedIds(self, params):
+        print(self.schemas_dict["picked_ids"]["rpc"], f"{params=}", flush=True)
+        validate_schema(params, self.schemas_dict["picked_ids"])
         x = params["x"]
         y = params["y"]
-        mouse_ids = ["id1", "id2", "id3"]
+        ids = params["ids"]
 
-        return {"mouse_ids": mouse_ids}
+        renderWindow = self.getView("-1")
+        renderer = renderWindow.GetRenderers().GetFirstRenderer()
+        picker = vtk.vtkWorldPointPicker()
+        picker.Pick([x, y, 0], renderer)
+        point = picker.GetPickPosition()
+        epsilon = self.computeEpsilon(renderer, point[2])
+        bbox = vtk.vtkBoundingBox()
+        bbox.AddPoint(point[0] + epsilon, point[1] + epsilon, point[2] + epsilon)
+        bbox.AddPoint(point[0] - epsilon, point[1] - epsilon, point[2] - epsilon)
+
+        array_ids = []
+        for id in ids:
+            bounds = self.get_object(id)["actor"].GetBounds()
+            if bbox.Intersects(bounds):
+                array_ids.append(id)
+
+        return {"array_ids": array_ids}

src/tests/test_viewer_protocols.py

@@ -137,22 +137,25 @@ def test_take_screenshot(server):
     assert server.images_diff(first_image_path, second_image_path) == 0.0
 
 
-def test_get_mouse(server):
+def test_picked_ids(server):
+
+    test_register_mesh(server)
+
     server.call(
-        VtkViewerView.prefix + VtkViewerView.schemas_dict["get_mouse"]["rpc"],
-        [{"x": 100, "y": 200}],
+        VtkViewerView.prefix + VtkViewerView.schemas_dict["picked_ids"]["rpc"],
+        [{"x": 100, "y": 200, "ids": ["123456789"]}],
     )
     response = server.get_response()
 
-    print(f"Response: {response}")
+    print(f"Response: {response}", flush=True)
 
     assert "result" in response, f"Key 'result' not found in response: {response}"
 
     assert (
-        "mouse_ids" in response["result"]
-    ), f"Key 'mouse_ids' not found in response['result']: {response['result']}"
+        "array_ids" in response["result"]
+    ), f"Key 'array_ids' not found in response['result']: {response['result']}"
 
-    mouse_ids = response["result"]["mouse_ids"]
-    assert isinstance(mouse_ids, list), f"Expected a list, but got {type(mouse_ids)}"
-    assert all(isinstance(id, str) for id in mouse_ids), "All IDs should be strings"
-    assert len(mouse_ids) > 0, "The list of mouse_ids should not be empty"
+    array_ids = response["result"]["array_ids"]
+    assert isinstance(array_ids, list), f"Expected a list, but got {type(array_ids)}"
+    assert all(isinstance(id, str) for id in array_ids), "All IDs should be strings"
+    assert len(array_ids) > 0, "The list of array_ids should not be empty"