feat(factory from biorbd.Model)

examples/biorbd/from_biorbd_model.py

@@ -0,0 +1,31 @@
+import biorbd
+import numpy as np
+
+from pyorerun import BiorbdModel, PhaseRerun
+
+
+def main():
+    # building some time components
+    nb_frames = 50
+    nb_seconds = 1
+    t_span = np.linspace(0, nb_seconds, nb_frames)
+
+    biorbd_model = biorbd.Model("models/Wu_Shoulder_Model_kinova_scaled_adjusted_2.bioMod")
+    prr_model = BiorbdModel.from_biorbd_object(biorbd_model)
+
+    # building some generalized coordinates
+    q = np.zeros((biorbd_model.nbQ(), nb_frames))
+    q[10, :] = np.linspace(0, np.pi / 8, nb_frames)
+    q[12, :] = np.linspace(0, np.pi / 3, nb_frames)
+    q[11, :] = np.linspace(0, np.pi / 4, nb_frames)
+    q[13, :] = np.linspace(0, np.pi / 8, nb_frames)
+    q[14, :] = np.linspace(0, np.pi / 8, nb_frames)
+    q[15, :] = np.linspace(0, np.pi / 8, nb_frames)
+
+    viz = PhaseRerun(t_span)
+    viz.add_animated_model(prr_model, q)
+    viz.rerun("msk_model")
+
+
+if __name__ == "__main__":
+    main()

pyorerun/biorbd_components/model_interface.py

@@ -36,10 +36,14 @@ class BiorbdModelNoMesh:
     A class to handle a biorbd model and its transformations
     """
 
-    def __init__(self, path, options=None):
+    def __init__(self, path: str, options=None):
         self.path = path
         self.model = biorbd.Model(path)
-        self.options = options if options is not None else DisplayModelOptions()
+        self.options: DisplayModelOptions = options if options is not None else DisplayModelOptions()
+
+    @classmethod
+    def from_biorbd_object(cls, model: biorbd.Model, options=None):
+        return cls(model.path().absolutePath().to_string(), options)
 
     @property
     def name(self):

pyorerun/biorbd_components/model_updapter.py

@@ -76,8 +76,8 @@ def create_muscles_updater(self):
             self.name,
             properties=LineStripProperties(
                 strip_names=self.model.muscle_names,
-                color=np.array(self.model.options.muscle_color),
-                radius=self.model.options.muscle_radius,
+                color=np.array(self.model.options.muscles_color),
+                radius=self.model.options.muscles_radius,
             ),
             update_callable=self.model.muscle_strips,
         )