Add plane, still need to really implement function + started creating the scene

Author: bakpaul

examples/stlib/SofaScene.py

@@ -0,0 +1,41 @@
+from stlib.geometries.plane import PlaneParameters
+from stlib.collision import CollisionParameters
+from stlib.collision import Collision
+from splib.core.enum_types import CollisionPrimitive
+import dataclasses
+import numpy as np
+
+
+
+def createScene(root):
+    ##Solvers
+
+
+
+    ##Environement
+
+    planes_lengthNormal = np.array([])
+    planes_length = 1
+    planes_width = 1
+    planes_nbEdgeLength = 1
+    planes_nbEdgeWidth = 1
+
+    plane1_collisionParams = CollisionParameters()
+    plane1_collisionParams.name = "UP"
+    plane1_collisionParams.primitives = [CollisionPrimitive.TRIANGLES]
+    plane1_collisionParams.kwargs = {"TriangleCollision" : {"moving" : False, "simulated" : False}}
+    plane1_collisionParams.geometry = PlaneParameters(np.array([0,0,0]), np.array([0,0,-1]),
+                                                      planes_lengthNormal, planes_length, planes_width, planes_nbEdgeLength, planes_nbEdgeWidth)
+    root.add(Collision, plane1_collisionParams)
+
+
+    plane2_collisionParams = CollisionParameters()
+    plane2_collisionParams.name = "DOWN"
+    plane2_collisionParams.primitives = [CollisionPrimitive.TRIANGLES]
+    plane2_collisionParams.kwargs = {"TriangleCollision" : {"moving" : False, "simulated" : False}}
+    plane2_collisionParams.geometry = PlaneParameters(np.array([0,0,0]), np.array([0,0,1]),
+                                                      planes_lengthNormal, planes_length, planes_width, planes_nbEdgeLength, planes_nbEdgeWidth)
+    root.add(Collision, plane2_collisionParams)
+
+
+    ## Real models

stlib/geometries/__geometry__.py

@@ -4,7 +4,9 @@
 from splib.topology.static import addStaticTopology
 from splib.core.enum_types import ElementType
 from splib.core.utils import DEFAULT_VALUE
-from Sofa.Core import Object 
+from Sofa.Core import Object
+
+import numpy as np
 
 
 class Geometry(BasePrefab):...
@@ -33,6 +35,10 @@ class GeometryParameters(BaseParameters):
 
     dynamicTopology : bool = False
 
+    def Data(self):
+        return InternalDataProvider()
+
+
 
 class Geometry(BasePrefab):
     # container : Object # This should be more specialized into the right SOFA type

stlib/geometries/plane.py

@@ -0,0 +1,27 @@
+from stlib.geometries import GeometryParameters, InternalDataProvider
+import dataclasses
+import numpy as np
+
+@dataclasses.dataclass
+class PlaneDataProvider(InternalDataProvider):
+    center : np.ndarray[float]
+    normal : np.ndarray[float]
+    lengthNormal : np.ndarray[float]
+    nbEdgeLength : int
+    nbEdgeWidth : int
+    width : float
+    edgeSize : float
+
+    def __post_init__(self, **kwargs):
+        InternalDataProvider.__init__(self,**kwargs)
+
+    def generateAttribute(self):
+        #TODO
+        pass
+
+
+class PlaneParameters(GeometryParameters):
+
+    def __init__(self,  center, normal, lengthNormal, length, width, nbEdgeLength, nbEdgeWidth, dynamicTopology = False):
+        GeometryParameters.__init__(data = PlaneDataProvider(center=center, normal=normal, lengthNormal=lengthNormal, length=length, width=width, nbEdgeLength=nbEdgeLength, nbEdgeWidth=nbEdgeWidth),
+                                    dynamicTopology = dynamicTopology)