Skip to content

Commit 6d1d220

Browse files
committed
Add support for reading NUMERICAL_DESCRIPTOR from STRU in pyabacus.
1 parent 030ac36 commit 6d1d220

File tree

1 file changed

+5
-0
lines changed
  • python/pyabacus/src/pyabacus/io

1 file changed

+5
-0
lines changed

python/pyabacus/src/pyabacus/io/stru.py

Lines changed: 5 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -107,6 +107,7 @@ def read_stru(fpath: str) -> Dict[str, Any]:
107107
"""Read an ABACUS STRU file and return its content as a dictionary."""
108108
block_title = ['ATOMIC_SPECIES',
109109
'NUMERICAL_ORBITAL',
110+
'NUMERICAL_DESCRIPTOR',
110111
'LATTICE_CONSTANT',
111112
'LATTICE_PARAMETER',
112113
'LATTICE_VECTORS',
@@ -144,6 +145,10 @@ def _trim(line):
144145
for i, s in enumerate(stru['species']):
145146
s['orb_file'] = blocks['NUMERICAL_ORBITAL'][i].strip()
146147

148+
#============ NUMERICAL_DESCRIPTOR ============
149+
if 'NUMERICAL_DESCRIPTOR' in blocks:
150+
stru['desc'] = blocks['NUMERICAL_DESCRIPTOR'][0].strip()
151+
147152
#============ ATOMIC_POSITIONS ============
148153
stru['coord_type'] = blocks['ATOMIC_POSITIONS'][0]
149154
index = {s['symbol']: i for i, s in enumerate(stru['species'])}

0 commit comments

Comments
 (0)