Code from Vanderbilt's print berry book!
Vanderbilt, D. (2018). Berry phases in electronic structure theory: Electric polarization, orbital magnetization and topological insulators. Cambridge University Press. https://doi.org/10.1017/9781316662205