Note
Go to the end to download the full example code.
Flow for QHA calculations with 2 DOFs
Warning: This code is still under development.

Downloading repository from: https://github.com/PseudoDojo/ONCVPSP-PBEsol-PDv0.4/archive/refs/heads/master.zip ...
Installing ONCVPSP-PBEsol-SR-PDv0.4 in: /home/runner/.abinit/pseudos/ONCVPSP-PBEsol-SR-PDv0.4
0.00iB [00:00, ?iB/s]
776kiB [00:00, 6.88MiB/s]
1.49MiB [00:00, 6.66MiB/s]
2.25MiB [00:00, 6.29MiB/s]
3.16MiB [00:00, 7.02MiB/s]
3.92MiB [00:00, 7.09MiB/s]
4.64MiB [00:00, 7.09MiB/s]
5.39MiB [00:00, 7.11MiB/s]
6.10MiB [00:00, 6.81MiB/s]
6.82MiB [00:01, 6.60MiB/s]
7.58MiB [00:01, 6.63MiB/s]
8.44MiB [00:01, 6.68MiB/s]
9.25MiB [00:01, 6.64MiB/s]
9.98MiB [00:01, 6.76MiB/s]
10.7MiB [00:01, 6.66MiB/s]
11.5MiB [00:01, 6.68MiB/s]
12.3MiB [00:01, 6.65MiB/s]
13.2MiB [00:01, 6.85MiB/s]
14.0MiB [00:02, 6.85MiB/s]
14.9MiB [00:02, 7.01MiB/s]
15.7MiB [00:02, 7.28MiB/s]
16.5MiB [00:02, 7.15MiB/s]
17.2MiB [00:02, 6.85MiB/s]
18.0MiB [00:02, 6.76MiB/s]
18.7MiB [00:02, 6.66MiB/s]
19.6MiB [00:02, 6.77MiB/s]
20.4MiB [00:02, 6.98MiB/s]
21.2MiB [00:03, 7.20MiB/s]
21.9MiB [00:03, 6.77MiB/s]
22.7MiB [00:03, 6.96MiB/s]
23.4MiB [00:03, 7.00MiB/s]
24.1MiB [00:03, 6.90MiB/s]
24.8MiB [00:03, 6.69MiB/s]
25.6MiB [00:03, 6.61MiB/s]
26.5MiB [00:03, 6.71MiB/s]
27.3MiB [00:03, 7.05MiB/s]
28.0MiB [00:04, 6.87MiB/s]
28.7MiB [00:04, 6.60MiB/s]
29.5MiB [00:04, 6.71MiB/s]
30.3MiB [00:04, 6.72MiB/s]
31.1MiB [00:04, 6.77MiB/s]
31.8MiB [00:04, 6.77MiB/s]
32.6MiB [00:04, 7.09MiB/s]
33.4MiB [00:04, 6.93MiB/s]
34.1MiB [00:05, 6.69MiB/s]
34.9MiB [00:05, 7.11MiB/s]
35.7MiB [00:05, 7.00MiB/s]
36.4MiB [00:05, 6.60MiB/s]
37.2MiB [00:05, 6.58MiB/s]
38.0MiB [00:05, 6.72MiB/s]
38.8MiB [00:05, 6.13MiB/s]
39.8MiB [00:05, 7.01MiB/s]
40.6MiB [00:05, 7.34MiB/s]
41.4MiB [00:06, 7.45MiB/s]
41.6MiB [00:06, 6.87MiB/s]
Validating md5 checksums of ONCVPSP-PBEsol-SR-PDv0.4...
Checksum test: OK
Installation completed successfully in 7.01 [s]
import sys
import os
import numpy as np
import abipy.abilab as abilab
import abipy.data as abidata
from abipy import flowtk
from abipy.flowtk.qha_2d import Qha2dFlow
def build_flow(options):
"""
Create a `Qha2dFlow` for quasi-harmonic calculations with 2 DOFs
"""
# Working directory (default is the name of the script with '.py' removed and "run_" replaced by "flow_")
if not options.workdir:
__file__ = os.path.join(os.getcwd(), "run_qha_2d.py")
options.workdir = os.path.basename(__file__).replace(".py", "").replace("run_", "flow_")
# Initialize structure and pseudos for ZnO.
structure = abilab.Structure.from_abistring("""
natom 4
ntypat 2
typat
1 1 2
2
znucl 30 8
xred
0.0000000000 0.0000000000 -0.0025137620
0.3333333333 0.6666666667 0.4974862380
0.0000000000 0.0000000000 0.3835201241
0.3333333333 0.6666666667 0.8835201241
acell 1.0 1.0 1.0
rprim
6.3016720624 0.0000000000 0.0000000000
-3.1508360312 5.4574080923 0.0000000000
0.0000000000 0.0000000000 9.7234377918
""")
# Use NC PBEsol pseudos from pseudodojo v0.4
from abipy.flowtk.psrepos import get_oncvpsp_pseudos
pseudos = get_oncvpsp_pseudos(xc_name="PBEsol", version="0.4")
# Select k-mesh for electrons and q-mesh for phonons.
#ngkpt = [6, 6, 4]; ngqpt = [1, 1, 1]
ngkpt = [2, 2, 2]; ngqpt = [1, 1, 1]
with_becs = True
with_quad = False
#with_quad = not structure.has_zero_dynamical_quadrupoles
scf_input = abilab.AbinitInput(structure, pseudos)
# Set other important variables
scf_input.set_vars(
nband=scf_input.num_valence_electrons // 2,
nline=10,
nbdbuf=0,
nstep=100,
ecutsm=1.0,
#tolvrs=1.0e-18, # SCF stopping criterion (modify default)
tolvrs=1.0e-6, # SCF stopping criterion (modify default)
)
#scf_input.set_scf_nband_semicond()
scf_input.set_kmesh(ngkpt=ngkpt, shiftk=[0, 0, 0])
bo_strains_a = [-5, 0, 5, 10, 15]
bo_strains_c = [-5, 0, 5, 10, 15]
#bo_strains_a = [0, 5, 10, 15, 20]
#bo_strains_c = [0, 5, 10, 15, 20]
# This is just for testing purposes
#bo_strains_a = [0, 5]
#bo_strains_c = [0, 5]
bo_strains_a = [0, ]
bo_strains_c = [0, 5]
bo_strains_a = np.array(bo_strains_a) / 100
bo_strains_c = np.array(bo_strains_c) / 100
bo_strains_ac = [bo_strains_a, bo_strains_c]
phdos_strains_ac = bo_strains_ac
return Qha2dFlow.from_scf_input(options.workdir, scf_input, bo_strains_ac, phdos_strains_ac, ngqpt,
with_becs, with_quad, edos_ngkpt=None)
# This block generates the thumbnails in the Abipy gallery.
# You can safely REMOVE this part if you are using this script for production runs.
if os.getenv("READTHEDOCS", False):
__name__ = None
import tempfile
options = flowtk.build_flow_main_parser().parse_args(["-w", tempfile.mkdtemp()])
build_flow(options).graphviz_imshow()
@flowtk.flow_main
def main(options):
"""
This is our main function that will be invoked by the script.
flow_main is a decorator implementing the command line interface.
Command line args are stored in `options`.
"""
return build_flow(options)
if __name__ == "__main__":
sys.exit(main())
Total running time of the script: (0 minutes 7.214 seconds)