Merge pull request #75 from pariterre/master

Fixed a bug if no meshes are present
pyomeca · Jun 29, 2022 · 4f06df5 · 4f06df5
2 parents 2613b45 + 8cd8f74
commit 4f06df5
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/bioviz/__init__.py b/bioviz/__init__.py
@@ -449,6 +449,7 @@ def __init__(
         self.nQ = self.model.nbQ()
         self.Q = np.zeros(self.nQ)
         self.idx_markers_to_remove = []
+        self.show_segment_is_on = [False] * self.model.nbSegment()
         if self.show_markers:
             self.Markers = InterfacesCollections.Markers(self.model)
             self.markers = Markers(np.ndarray((3, self.model.nbMarkers(), 1)))
@@ -467,7 +468,6 @@ def __init__(
             self.CoMbySegment = InterfacesCollections.CoMbySegment(self.model)
             self.segments_center_of_mass = Markers(np.ndarray((3, self.model.nbSegment(), 1)))
         if self.show_meshes:
-            self.show_segment_is_on = []
             self.mesh = []
             self.meshPointsInMatrix = InterfacesCollections.MeshPointsInMatrix(self.model)
             for i, vertices in enumerate(self.meshPointsInMatrix.get_data(Q=self.Q, compute_kin=False)):
@@ -477,7 +477,7 @@ def __init__(
                     else np.ndarray((0, 3), dtype="int32")
                 )
                 self.mesh.append(Mesh(vertex=vertices, triangles=triangles.T))
-                self.show_segment_is_on.append(True)
+                self.show_segment_is_on[i] = True
         if self.show_muscles:
             self.model.updateMuscles(self.Q, True)
             self.muscles = []