Source code for twod_materials.intercalation.startup
from __future__ import print_function, division, unicode_literals
import os
from pymatgen.core.structure import Structure
from pymatgen.core.periodic_table import Element
from pymatgen.analysis.defects.point_defects import (
Interstitial, ValenceIonicRadiusEvaluator
)
import operator
from monty.dev import requires
import twod_materials
from monty.serialization import loadfn
PACKAGE_PATH = twod_materials.__file__.replace('__init__.pyc', '')
PACKAGE_PATH = PACKAGE_PATH.replace('__init__.py', '')
PACKAGE_PATH = '/'.join(PACKAGE_PATH.split('/')[:-2])
try:
import zeo
zeo_found = True
except ImportError:
zeo_found = False
try:
config_vars = loadfn(os.path.join(os.path.expanduser('~'), 'config.yaml'))
except:
print('WARNING: No config.yaml file was found. please configure the '\
'config.yaml and put it in your home directory.')
# Still set them for testing purposes.
config_vars = loadfn(os.path.join(PACKAGE_PATH, 'config.yaml'))
if 'queue_system' in config_vars:
QUEUE = config_vars['queue_system'].lower()
elif '/ufrc/' in os.getcwd():
QUEUE = 'slurm'
elif '/scratch/' in os.getcwd():
QUEUE = 'pbs'
@requires(zeo_found, 'get_voronoi_nodes requires Zeo++ cython extension to be '
'installed. Please contact developers of Zeo++ to obtain it.')
[docs]def inject_ions(structure, ion, atomic_fraction):
"""
Adds ions to a percentage of interstitial sites into a structure
that results in an at% less than or equal to the specified
atomic_fraction. Starts by filling interstitial sites with the
largest voronoi radius, and then works downward.
Args:
structure (Structure): Pymatgen Structure object to
intercalate into.
ion (str): name of atom to intercalate, e.g. 'Li', or 'Mg'.
atomic_fraction (int): This fraction of the final intercalated
structure will be intercalated atoms. Must be < 1.0.
Returns:
structure. Includes intercalated atoms.
"""
specie = Element(ion)
# If the structure isn't big enough to accomodate such a small
# atomic fraction, multiply it in the x direction.
n_ions = 1.
while not n_ions / structure.num_sites <= atomic_fraction:
structure.make_supercell([2, 1, 1])
evaluator = ValenceIonicRadiusEvaluator(structure)
interstitial = Interstitial(structure, radii=evaluator.radii,
valences=evaluator.valences)
interstitial_sites = [
(site._fcoords, site.properties.get('voronoi_radius', None))
for site in interstitial._defect_sites
]
# Sort the interstitial sites by their voronoi radii.
interstitial_sites.sort(key=operator.itemgetter(1))
interstitial_sites.reverse()
i = 0
while n_ions / (structure.num_sites + 1) <= atomic_fraction:
try:
structure.append(species=specie,
coords=interstitial_sites[i][0],
validate_proximity=True)
n_ions += 1
i += 1
evaluator = ValenceIonicRadiusEvaluator(structure)
interstitial = Interstitial(structure, radii=evaluator.radii,
valences=evaluator.valences)
interstitial_sites = [
(site._fcoords, site.properties.get('voronoi_radius', None))
for site in interstitial._defect_sites
]
# Sort the interstitial sites by their voronoi radii.
interstitial_sites.sort(key=operator.itemgetter(1))
interstitial_sites.reverse()
except ValueError:
i += 1
except IndexError:
raise ValueError('The atomic_fraction specified exceeds the '
'number of available interstitial sites in this '
'structure. Please choose a smaller '
'atomic_fraction.')
return structure