Spin–orbit interaction with nonlinear wave functions