Check out the demo app code that is linked in the NIP. It should make it more clear about when each ratchet is turned and what data goes into each calculation.

It also shows you how skipped/out of order messages are handled.