A comparative study of ion diffusion during water imbibition in shale, sandstone and volcanic rock