fix(GeodeClasses): literal viewer_elements_type

Author: JulienChampagnol

src/opengeodeweb_back/geode_objects/geode_graph.py

@@ -5,7 +5,7 @@
 import opengeode as og
 
 # Local application imports
-from .types import GeodeMeshType
+from .types import GeodeMeshType, ViewerElementsType
 from .geode_vertex_set import GeodeVertexSet
 
 
@@ -19,6 +19,10 @@ def __init__(self, graph: og.Graph | None = None) -> None:
     @classmethod
     def geode_object_type(cls) -> GeodeMeshType:
         return "Graph"
+    
+    @classmethod
+    def viewer_elements_type(cls) -> ViewerElementsType:
+        return "edges"
 
     def native_extension(self) -> str:
         return self.graph.native_extension()

src/opengeodeweb_back/geode_objects/geode_grid2d.py

@@ -7,6 +7,7 @@
 
 # Local application imports
 from .geode_mesh import GeodeMesh
+from .types import ViewerElementsType
 
 
 class GeodeGrid2D(GeodeMesh):
@@ -18,6 +19,10 @@ def is_3D(cls) -> bool:
     def is_viewable(cls) -> bool:
         return True
 
+    @classmethod
+    def viewer_elements_type(cls) -> ViewerElementsType:
+        return "polygons"
+
     def builder(self) -> object:
         return None
 

src/opengeodeweb_back/geode_objects/geode_grid3d.py

@@ -7,6 +7,7 @@
 
 # Local application imports
 from .geode_mesh import GeodeMesh
+from .types import ViewerElementsType
 
 
 class GeodeGrid3D(GeodeMesh):
@@ -18,6 +19,10 @@ def is_3D(cls) -> bool:
     def is_viewable(cls) -> bool:
         return True
 
+    @classmethod
+    def viewer_elements_type(cls) -> ViewerElementsType:
+        return "polyhedra"
+
     def builder(self) -> object:
         return None
 

src/opengeodeweb_back/geode_objects/geode_model.py

@@ -6,7 +6,7 @@
 import opengeode as og
 
 # Local application imports
-from .types import ViewerType
+from .types import ViewerType, ViewerElementsType
 from .geode_object import GeodeObject
 
 ComponentRegistry = dict[og.ComponentType, list[og.uuid]]
@@ -17,5 +17,9 @@ class GeodeModel(GeodeObject):
     def viewer_type(cls) -> ViewerType:
         return "model"
 
+    @classmethod
+    def viewer_elements_type(cls) -> ViewerElementsType:
+        return "default"
+
     @abstractmethod
     def mesh_components(self) -> ComponentRegistry: ...

src/opengeodeweb_back/geode_objects/geode_object.py

@@ -7,7 +7,7 @@
 import opengeode as og
 
 # Local application imports
-from .types import GeodeObjectType, ViewerType
+from .types import GeodeObjectType, ViewerType, ViewerElementsType
 
 
 class GeodeObject(ABC):
@@ -24,6 +24,10 @@ def geode_object_type(cls) -> GeodeObjectType: ...
     @abstractmethod
     def viewer_type(cls) -> ViewerType: ...
 
+    @classmethod
+    @abstractmethod
+    def viewer_elements_type(cls) -> ViewerElementsType: ...
+
     @classmethod
     @abstractmethod
     def is_3D(cls) -> bool: ...

src/opengeodeweb_back/geode_objects/geode_solid_mesh3d.py

@@ -8,6 +8,7 @@
 
 # Local application imports
 from .geode_vertex_set import GeodeVertexSet
+from .types import ViewerElementsType
 
 
 class GeodeSolidMesh3D(GeodeVertexSet):
@@ -25,6 +26,10 @@ def is_3D(cls) -> bool:
     def is_viewable(cls) -> bool:
         return True
 
+    @classmethod
+    def viewer_elements_type(cls) -> ViewerElementsType:
+        return "polyhedra"
+
     def builder(self) -> og.SolidMeshBuilder3D:
         return og.SolidMeshBuilder3D.create(self.solid_mesh)
 

src/opengeodeweb_back/geode_objects/geode_surface_mesh2d.py

@@ -8,6 +8,7 @@
 
 # Local application imports
 from .geode_vertex_set import GeodeVertexSet
+from .types import ViewerElementsType
 
 
 class GeodeSurfaceMesh2D(GeodeVertexSet):
@@ -26,6 +27,10 @@ def is_3D(cls) -> bool:
     @classmethod
     def is_viewable(cls) -> bool:
         return True
+    
+    @classmethod
+    def viewer_elements_type(cls) -> ViewerElementsType:
+        return "polygons"
 
     def builder(self) -> og.SurfaceMeshBuilder2D:
         return og.SurfaceMeshBuilder2D.create(self.surface_mesh)

src/opengeodeweb_back/geode_objects/geode_surface_mesh3d.py

@@ -8,6 +8,7 @@
 
 # Local application imports
 from .geode_vertex_set import GeodeVertexSet
+from .types import ViewerElementsType
 
 
 class GeodeSurfaceMesh3D(GeodeVertexSet):
@@ -27,6 +28,10 @@ def is_3D(cls) -> bool:
     def is_viewable(cls) -> bool:
         return True
 
+    @classmethod
+    def viewer_elements_type(cls) -> ViewerElementsType:
+        return "polygons"
+
     def builder(self) -> og.SurfaceMeshBuilder3D:
         return og.SurfaceMeshBuilder3D.create(self.surface_mesh)
 

src/opengeodeweb_back/geode_objects/geode_vertex_set.py

@@ -5,7 +5,7 @@
 import opengeode as og
 
 # Local application imports
-from .types import GeodeMeshType
+from .types import GeodeMeshType, ViewerElementsType
 from .geode_mesh import GeodeMesh
 
 
@@ -21,6 +21,10 @@ def __init__(self, vertex_set: og.VertexSet | None = None) -> None:
     @classmethod
     def geode_object_type(cls) -> GeodeMeshType:
         return "VertexSet"
+    
+    @classmethod
+    def viewer_elements_type(cls) -> ViewerElementsType:
+        return "points"
 
     def native_extension(self) -> str:
         return self.vertex_set.native_extension()

src/opengeodeweb_back/geode_objects/types.py

@@ -73,3 +73,5 @@ def geode_object_type(value: str) -> GeodeObjectType:
 
 
 ViewerType = Literal["mesh", "model"]
+
+ViewerElementsType = Literal["points", "edges", "polygons", "polyhedra", "default"]