waxmorph.simulator.mech_step_sticky_implicit

waxmorph.simulator.mech_step_sticky_implicit#

waxmorph.simulator.mech_step_sticky_implicit(X, R, P, CT, particle_count, dt, X_next, P_next, device='cuda', grid=None, *, wnt=None)[source]#

Advance mechanics with local autodiff for polarity and thickness potentials.

Parameters:
  • X (array(ndim=1, dtype=vec3f))

  • R (array(ndim=1, dtype=float32))

  • P (array(ndim=1, dtype=vec3f))

  • CT (array(ndim=1, dtype=uint32))

  • particle_count (int32)

  • dt (float)

  • X_next (array(ndim=1, dtype=vec3f))

  • P_next (array(ndim=1, dtype=vec3f))

  • device (str)

  • grid (HashGrid | None)

  • wnt (array | None)