Skip to content

Commit 2235c5d

Browse files
committed
fixes
1 parent dab4e97 commit 2235c5d

File tree

8 files changed

+18
-29
lines changed

8 files changed

+18
-29
lines changed

src/opengeodeweb_viewer/rpc/mesh/edges/mesh_edges_protocols.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -47,5 +47,5 @@ def setMeshEdgesWidth(self, rpc_params: RpcParams) -> None:
4747
validate_schema(
4848
rpc_params, self.mesh_edges_schemas_dict["width"], self.mesh_edges_prefix
4949
)
50-
params = schemas.Color.from_dict(rpc_params)
50+
params = schemas.Width.from_dict(rpc_params)
5151
self.SetEdgesWidth(params.id, params.width)

src/opengeodeweb_viewer/rpc/mesh/mesh_protocols.py

Lines changed: 1 addition & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -15,7 +15,6 @@
1515
get_schemas_dict,
1616
validate_schema,
1717
RpcParams,
18-
RpcParamsWithColor,
1918
)
2019
from opengeodeweb_viewer.object.object_methods import VtkObjectView
2120
from opengeodeweb_viewer.vtk_protocol import vtkData
@@ -92,7 +91,7 @@ def setMeshOpacity(self, rpc_params: RpcParams) -> None:
9291
self.SetOpacity(params.id, params.opacity)
9392

9493
@exportRpc(mesh_prefix + mesh_schemas_dict["color"]["rpc"])
95-
def setMeshColor(self, rpc_params: RpcParamsWithColor) -> None:
94+
def setMeshColor(self, rpc_params: RpcParams) -> None:
9695
validate_schema(rpc_params, self.mesh_schemas_dict["color"], self.mesh_prefix)
9796
params = schemas.Color.from_dict(rpc_params)
9897
color = params.color

src/opengeodeweb_viewer/rpc/mesh/points/mesh_points_protocols.py

Lines changed: 2 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -9,7 +9,6 @@
99
get_schemas_dict,
1010
validate_schema,
1111
RpcParams,
12-
RpcParamsWithColor,
1312
)
1413
from opengeodeweb_viewer.rpc.mesh.mesh_protocols import VtkMeshView
1514
from . import schemas
@@ -35,7 +34,7 @@ def setMeshPointsVisibility(self, rpc_params: RpcParams) -> None:
3534
self.SetPointsVisibility(params.id, params.visibility)
3635

3736
@exportRpc(mesh_points_prefix + mesh_points_schemas_dict["color"]["rpc"])
38-
def setMeshPointsColor(self, rpc_params: RpcParamsWithColor) -> None:
37+
def setMeshPointsColor(self, rpc_params: RpcParams) -> None:
3938
validate_schema(
4039
rpc_params, self.mesh_points_schemas_dict["color"], self.mesh_points_prefix
4140
)
@@ -59,4 +58,4 @@ def setMeshPointsVertexAttribute(self, rpc_params: RpcParams) -> None:
5958
self.mesh_points_prefix,
6059
)
6160
params = schemas.VertexAttribute.from_dict(rpc_params)
62-
self.displayAttributeOnVertices(params.id, pramas.name)
61+
self.displayAttributeOnVertices(params.id, params.name)

src/opengeodeweb_viewer/rpc/mesh/polygons/polygons_protocols.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -65,5 +65,5 @@ def setMeshPolygonsPolygonAttribute(self, rpc_params: RpcParams) -> None:
6565
self.mesh_polygons_schemas_dict["polygon_attribute"],
6666
self.mesh_polygons_prefix,
6767
)
68-
params = schemas.Color.from_dict(rpc_params)
68+
params = schemas.PolygonAttribute.from_dict(rpc_params)
6969
self.displayAttributeOnCells(params.id, params.name)

src/opengeodeweb_viewer/rpc/viewer/viewer_protocols.py

Lines changed: 10 additions & 12 deletions
Original file line numberDiff line numberDiff line change
@@ -23,8 +23,6 @@
2323
get_schemas_dict,
2424
validate_schema,
2525
RpcParams,
26-
RpcParamsWithColor,
27-
RpcParamsWithList,
2826
)
2927
from opengeodeweb_viewer.vtk_protocol import VtkView
3028
from . import schemas
@@ -97,7 +95,7 @@ def resetVisualization(self, rpc_params: RpcParams) -> None:
9795
self.render()
9896

9997
@exportRpc(viewer_prefix + viewer_schemas_dict["set_background_color"]["rpc"])
100-
def setBackgroundColor(self, rpc_params: RpcParamsWithColor) -> None:
98+
def setBackgroundColor(self, rpc_params: RpcParams) -> None:
10199
validate_schema(
102100
rpc_params,
103101
self.viewer_schemas_dict["set_background_color"],
@@ -147,16 +145,16 @@ def takeScreenshot(self, rpc_params: RpcParams) -> dict[str, str | bytes]:
147145
w2if.ReadFrontBufferOff()
148146
w2if.Update()
149147
output_extension = params.output_extension
150-
if output_extension == "png":
148+
if output_extension == schemas.OutputExtension.PNG:
151149
writer = vtkPNGWriter()
152-
elif output_extension in ["jpg", "jpeg"]:
150+
elif output_extension == schemas.OutputExtension.JPG:
153151
if not include_background:
154152
raise Exception("output_extension not supported with background")
155153
writer = vtkJPEGWriter()
156154
else:
157155
raise Exception("output_extension not supported")
158156

159-
new_filename = params.filename + "." + output_extension
157+
new_filename = params.filename + "." + output_extension.value
160158
file_path = os.path.join(self.DATA_FOLDER_PATH, new_filename)
161159
writer.SetFileName(file_path)
162160
writer.SetInputConnection(w2if.GetOutputPort())
@@ -219,7 +217,7 @@ def computeEpsilon(self, renderer: vtkRenderer, z: float) -> float:
219217
return math.sqrt(epsilon) * 0.0125
220218

221219
@exportRpc(viewer_prefix + viewer_schemas_dict["picked_ids"]["rpc"])
222-
def pickedIds(self, rpc_params: RpcParamsWithList) -> dict[str, list[str]]:
220+
def pickedIds(self, rpc_params: RpcParams) -> dict[str, list[str]]:
223221
validate_schema(
224222
rpc_params, self.viewer_schemas_dict["picked_ids"], self.viewer_prefix
225223
)
@@ -275,11 +273,11 @@ def updateCamera(self, rpc_params: RpcParams) -> None:
275273
renderWindow = self.getView("-1")
276274
camera = renderWindow.GetRenderers().GetFirstRenderer().GetActiveCamera()
277275

278-
camera.SetFocalPoint(*camara_options.focal_point)
279-
camera.SetViewUp(*camara_options.view_up)
280-
camera.SetPosition(*camara_options.position)
281-
camera.SetViewAngle(camara_options.view_angle)
282-
camera.SetClippingRange(*camara_options.clipping_range)
276+
camera.SetFocalPoint(camera_options.focal_point)
277+
camera.SetViewUp(camera_options.view_up)
278+
camera.SetPosition(camera_options.position)
279+
camera.SetViewAngle(camera_options.view_angle)
280+
camera.SetClippingRange(camera_options.clipping_range)
283281
self.render()
284282

285283
@exportRpc(viewer_prefix + viewer_schemas_dict["render_now"]["rpc"])

src/opengeodeweb_viewer/utils_functions.py

Lines changed: 0 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -6,14 +6,7 @@
66
# Third party imports
77
import fastjsonschema # type: ignore
88

9-
type JsonPrimitive = str | int | float | bool
10-
type JsonValue = JsonPrimitive | dict[str, JsonValue] | list[JsonValue]
119
type RpcParams = dict[str, str]
12-
13-
type ColorDict = dict[str, int]
14-
type RpcParamsWithColor = dict[str, JsonPrimitive | ColorDict]
15-
type RpcParamsWithList = dict[str, JsonPrimitive | list[str]]
16-
1710
type SchemaDict = dict[str, str]
1811

1912

src/opengeodeweb_viewer/vtk_protocol.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1,7 +1,7 @@
11
# Standard library imports
22
import os
33
from typing import cast, Any, Literal
4-
from dataclasses import dataclass
4+
from dataclasses import dataclass, field
55

66
# Third party imports
77
from vtk.web import protocols as vtk_protocols # type: ignore
@@ -25,7 +25,7 @@ class vtkData:
2525
reader: vtkAlgorithm
2626
mapper: vtkMapper
2727
filter: vtkAlgorithm | None = None
28-
actor: vtkActor = vtkActor()
28+
actor: vtkActor = field(default_factory=vtkActor)
2929
max_dimension: Literal["points", "edges", "polygons", "polyhedra", "default"] = (
3030
"default"
3131
)

src/opengeodeweb_viewer/vtkw_server.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -4,7 +4,7 @@
44

55
# Third party imports
66
from vtk.web import wslink as vtk_wslink # type: ignore
7-
from vtk.web import protocols as vtk_protocols # type: ignore
7+
from vtk.web import protocols as vtk_protocols
88
from wslink import server # type: ignore
99
from vtkmodules.vtkRenderingCore import vtkRenderer, vtkRenderWindow
1010
from opengeodeweb_microservice.database import connection

0 commit comments

Comments
 (0)