atomate.vasp package¶
Subpackages¶
- atomate.vasp.analysis package
- atomate.vasp.builders package
- Submodules
- atomate.vasp.builders.bandgap_estimation module
- atomate.vasp.builders.base module
- atomate.vasp.builders.boltztrap_materials module
- atomate.vasp.builders.dielectric module
- atomate.vasp.builders.file_materials module
- atomate.vasp.builders.fix_tasks module
- atomate.vasp.builders.materials_descriptor module
- atomate.vasp.builders.materials_ehull module
- atomate.vasp.builders.tags module
- atomate.vasp.builders.tasks_materials module
- Module contents
- atomate.vasp.examples package
- atomate.vasp.firetasks package
- atomate.vasp.fireworks package
- atomate.vasp.tests package
- atomate.vasp.workflows package
- Subpackages
- atomate.vasp.workflows.base package
- Submodules
- atomate.vasp.workflows.base.adsorption module
- atomate.vasp.workflows.base.bulk_modulus module
- atomate.vasp.workflows.base.core module
- atomate.vasp.workflows.base.deformations module
- atomate.vasp.workflows.base.elastic module
- atomate.vasp.workflows.base.gibbs module
- atomate.vasp.workflows.base.neb module
- atomate.vasp.workflows.base.raman module
- atomate.vasp.workflows.base.thermal_expansion module
- Module contents
- atomate.vasp.workflows.presets package
- atomate.vasp.workflows.tests package
- Submodules
- atomate.vasp.workflows.tests.test_adsorbate_workflow module
- atomate.vasp.workflows.tests.test_bulk_modulus_workflow module
- atomate.vasp.workflows.tests.test_elastic_workflow module
- atomate.vasp.workflows.tests.test_neb_workflow module
- atomate.vasp.workflows.tests.test_raman_workflow module
- atomate.vasp.workflows.tests.test_vasp_workflows module
- Module contents
- atomate.vasp.workflows.base package
- Module contents
- Subpackages
Submodules¶
atomate.vasp.config module¶
atomate.vasp.database module¶
-
class
atomate.vasp.database.
VaspCalcDb
(host=u'localhost', port=27017, database=u'vasp', collection=u'tasks', user=None, password=None)¶ Bases:
atomate.utils.database.CalcDb
Class to help manage database insertions of Vasp drones
-
__init__
(host=u'localhost', port=27017, database=u'vasp', collection=u'tasks', user=None, password=None)¶
-
build_indexes
(indexes=None, background=True)¶ Build the indexes.
- Args:
- indexes (list): list of single field indexes to be built. background (bool): Run in the background or not.
- TODO: make sure that the index building is sensible and check for
- existing indexes.
-
get_band_structure
(task_id)¶
-
get_dos
(task_id)¶
-
insert_gridfs
(d, collection=u'fs', compress=True, oid=None)¶ Insert the given document into GridFS.
- Args:
- d (dict): the document collection (string): the GridFS collection name compress (bool): Whether to compress the data or not oid (ObjectId()): the _id of the file; if specified, it must not already exist in GridFS
- Returns:
- file id, the type of compression used.
-
insert_task
(task_doc, parse_dos=False, parse_bs=False)¶ Inserts a task document (e.g., as returned by Drone.assimilate()) into the database. Handles putting DOS and band structure into GridFS as needed.
- Args:
- task_doc: (dict) the task document parse_dos: (bool) attempt to parse dos in task_doc and insert into Gridfs parse_bs: (bool) attempt to parse bandstructure in task_doc and insert into Gridfs
- Returns:
- (int) - task_id of inserted document
-
reset
()¶
-
atomate.vasp.drones module¶
-
class
atomate.vasp.drones.
VaspDrone
(runs=None, parse_dos=False, compress_dos=False, bandstructure_mode=False, compress_bs=False, additional_fields=None, use_full_uri=True)¶ Bases:
pymatgen.apps.borg.hive.AbstractDrone
pymatgen-db VaspToDbTaskDrone with updated schema and documents processing methods. Please refer to matgendb.creator.VaspToDbTaskDrone documentation.
-
__init__
(runs=None, parse_dos=False, compress_dos=False, bandstructure_mode=False, compress_bs=False, additional_fields=None, use_full_uri=True)¶
-
as_dict
()¶
-
assimilate
(path)¶ Adapted from matgendb.creator Parses vasp runs(vasprun.xml file) and insert the result into the db. Get the entire task doc from the vasprum.xml and the OUTCAR files in the path. Also adds some post-processed info.
- Args:
- path (str): Path to the directory containing vasprun.xml and OUTCAR files
- Returns:
- (dict): a task dictionary
-
filter_files
(path, file_pattern=u'vasprun.xml')¶ Find the files that match the pattern in the given path and return them in an ordered dictionary. The searched for files are filtered by the run types defined in self.runs. e.g. [“relax1”, “relax2”, ...]. Only 2 schemes of the file filtering is enabled: searching for run types in the list of files and in the filenames. Modify this method if more sophisticated filtering scheme is needed.
- Args:
- path (string): path to the folder file_pattern (string): files to be searched for
- Returns:
- OrderedDict of the names of the files to be processed further. The key is set from list of run types: self.runs
-
classmethod
from_dict
(d)¶
-
generate_doc
(dir_name, vasprun_files, outcar_files)¶ Adapted from matgendb.creator.generate_doc
-
get_valid_paths
(path)¶ There are some restrictions on the valid directory structures:
- There can be only one vasp run in each directory. Nested directories are fine.
- Directories designated “relax1”...”relax9” are considered to be parts of a multiple-optimization run.
- Directories containing vasp output with ”.relax1”...”.relax9” are also considered as parts of a multiple-optimization run.
-
post_process
(dir_name, d)¶ Post-processing for various files other than the vasprun.xml and OUTCAR. Looks for files: transformations.json and custodian.json. Modify this if other output files need to be processed.
- Args:
- dir_name:
- The dir_name.
- d:
- Current doc generated.
-
process_vasprun
(dir_name, taskname, filename)¶ Adapted from matgendb.creator
Process a vasprun.xml file.
-
schema
= {u'input': set([u'pseudo_potential', u'hubbards', u'xc_override', u'potcar_spec', u'is_hubbard', u'structure', u'is_lasph']), u'calcs_reversed': set([u'dir_name', u'run_type', u'task', u'input', u'formula_anonymous', u'nelements', u'vasp_version', u'nsites', u'composition_reduced', u'completed_at', u'formula_reduced_abc', u'composition_unit_cell', u'output', u'formula_pretty', u'has_vasp_completed', u'elements']), u'root': set([u'calcs_reversed', u'elements', u'input', u'composition_reduced', u'formula_pretty', u'nelements', u'nsites', u'chemsys', u'analysis', u'completed_at', u'state', u'formula_reduced_abc', u'run_stats', u'composition_unit_cell', u'dir_name', u'output', u'formula_anonymous', u'schema']), u'analysis': set([u'delta_volume', u'errors', u'max_force', u'delta_volume_percent', u'warnings']), u'output': set([u'is_gap_direct', u'density', u'is_metal', u'energy', u'bandgap', u'vbm', u'energy_per_atom', u'spacegroup', u'cbm', u'structure'])}¶
-
static
set_analysis
(d, max_force_threshold=0.5, volume_change_threshold=0.2)¶ Adapted from matgendb.creator
set the ‘analysis’ key
-
validate_doc
(d)¶ Sanity check. Make sure all the important keys are set
-
atomate.vasp.powerups module¶
atomate.vasp.submission_filter module¶
-
class
atomate.vasp.submission_filter.
SubmissionFilter
(is_valid=True, potcar_exists=True, max_natoms=200, is_ordered=True, not_in_MP=True, MAPI_KEY=None, require_bandstructure=False)¶ Bases:
pymatgen.alchemy.filters.AbstractStructureFilter
-
NO_POTCARS
= [u'Po', u'At', u'Rn', u'Fr', u'Ra', u'Am', u'Cm', u'Bk', u'Cf', u'Es', u'Fm', u'Md', u'No', u'Lr']¶
-
__init__
(is_valid=True, potcar_exists=True, max_natoms=200, is_ordered=True, not_in_MP=True, MAPI_KEY=None, require_bandstructure=False)¶ Initialize a submission filter for checking that structures are valid for calculations.
- Args:
- is_valid (bool): If true, checks structure validity potcar_exists (bool): If true, ensures all elements have VASP PAW_PBE POTCAR max_natoms (int): If not None, ensures structure has <=max_natoms atoms is_ordered (bool): If true, ensures structure is ordered not_in_MP (bool): If true, ensures structure not in MP MAPI_KEY (str): For MP checks, your MAPI key if not previously set as config var require_bandstructure (bool): For MP checks, require a band structure calc
-
as_dict
()¶
-
classmethod
from_dict
(d)¶
-
test
(structure)¶
-