Spin–orbit DFT with analytic gradients and applications to heavy element compounds