Correct the assignment of wavefunction stride 2 for corner source
Add backward compatibility for Dirac field sources.