第一布里渊区3D可视化工具vbz-GUI.py

北大-陶豫

分享一个我编写的Python脚本，用于可视化晶体的第一布里渊区（Wigner-Seitz原胞）。这个脚本在vaspkit的vbz.py基础上修改，基于VASP计算和vaspkit后处理中常见的POSCAR和KPATH.in文件，提供了以下功能：
3D交互式可视化：使用matplotlib的3D绘图功能，可以显示布里渊区、高对称点、k点路径以及倒格矢和笛卡尔坐标轴。
无死角鼠标拖动旋转：采用基于屏幕投影的旋转算法，确保在拖动过程中不会出现旋转死区，实现平滑的视角转换。不像Multiwfn拖动时会有死角
点击高对称点显示坐标：在图形界面中点击任意高对称点，会在终端显示该点的分数坐标和笛卡尔坐标，方便用户记录和分析。
状态保存与加载：当前的视角、缩放以及显示设置可以保存到文件中，下次启动时自动加载，方便用户继续分析。
自动文件生成：如果当前目录下没有HIGH_SYMMETRY_POINTS文件，脚本会自动从KPATH.in文件中提取所有高对称点（去重）并生成该文件，省去了手动准备的步骤。
灵活的可视化控制：用户可以通过左侧控制面板选择显示或隐藏布里渊区、高对称点、k点路径、倒格矢、笛卡尔坐标轴等元素，并且可以分别控制笛卡尔坐标轴和倒格矢在布里渊区内的部分是否显示。
导出高质量图片：支持将当前视图导出为PNG或PDF格式，便于在论文或报告中使用。

效果图：




附件是脚本和示例的输入文件。