resetting omega in predictor corrector

python/time_integration.py

@@ -85,6 +85,7 @@ def predictor_corrector_iterate(fluid, particles, dt, min_nsub=1, contact_tol=1e
     vp0 = np.copy(particles.velocity())
     pp0 = np.copy(particles.position())
     cp0 = np.copy(particles.contact_forces())
+    op0 = np.copy(particles.omega())
 
     # Predictor
     #
@@ -113,6 +114,7 @@ def predictor_corrector_iterate(fluid, particles, dt, min_nsub=1, contact_tol=1e
         fluid.advance_concentration(dt)
     # Reset solid velocities
     particles.velocity()[:,:] = vp0 
+    particles.omega()[:,:] = op0
     _advance_particles(particles,(alpha*f0+(1-alpha)*f1),dt,min_nsub,contact_tol,after_sub_iter=after_sub_iter,max_split=max_split)
     # Give to the fluid the solid information
     fluid.set_particles(particles.mass(), particles.volume(), particles.position(), particles.velocity(), particles.contact_forces())