Use 'diagk0' when appropriate in pdos and writewave

This might save time when computing the COOP-related wavefunctions,
and when carrying out a pDOS calculation with the 'gamma' sampling.