state.omega in vtk export

python/scontact.py

@@ -319,10 +319,7 @@ class ParticleProblem :
 
         state = self.state()
         v = state.v
-        if self._rotation_enabled:
-            omega = self.omega()
-        else :
-            omega = np.zeros((v.shape[0],1))
+        omega = state.omega
         x = state.x
         material = self._get_idx("ParticleMaterial").reshape(-1,1)
         forced = self._get_idx("ForcedFlag").reshape(-1,1)