Rewrite the CPU version of the DFT+U routine for getting coefficients, to remove the explicit call to batch_get_states.

This should not be seen as a fully optimized version, as one can certainly do better, especially for the packed case.