PLUGINS/STRUCTURE: Difference between revisions

From VASP Wiki
Line 16: Line 16:
The <code>constants</code> dataclass consists of the following inputs, listed here with their associated [https://numpy.org/doc/stable/user/basics.types.html datatypes]
The <code>constants</code> dataclass consists of the following inputs, listed here with their associated [https://numpy.org/doc/stable/user/basics.types.html datatypes]
<syntaxhighlight lang="python" line>
<syntaxhighlight lang="python" line>
    number_ions: int
number_ions: int
    number_ion_types: int
number_ion_types: int
    ion_types: NDArray[np.int32]
ion_types: NDArray[np.int32]
    atomic_numbers: NDArray[np.int32]
atomic_numbers: NDArray[np.int32]
    lattice_vectors: NDArray[np.float64]
lattice_vectors: NDArray[np.float64]
    positions: NDArray[np.float64]
positions: NDArray[np.float64]
    {{TAGBL|POMASS}}: NDArray[np.float64]
{{TAGBL|POMASS}}: NDArray[np.float64]
    total_energy: float
total_energy: float
    forces: NDArray[np.float64]
forces: NDArray[np.float64]
    stress: NDArray[np.float64]
stress: NDArray[np.float64]
    shape_grid: NDArray[np.float64]
shape_grid: NDArray[np.float64]
    charge_density: NDArray[np.float64]
charge_density: NDArray[np.float64]
</syntaxhighlight>
</syntaxhighlight>
Note that the {{FILE|INCAR}} tags are capitalized.
Note that the {{FILE|INCAR}} tags are capitalized.
Line 37: Line 37:
<code>forces</code> and <code>stress</code> are the computed forces and stress tensor and <code>charge_density</code> contains the charge density on the real space grid. <code>shape_grid</code> is a three dimensional integer array which stores the shape of the real space grid, {{TAG|NGXF}}, {{TAG|NGYF}} and {{TAG|NGZF}} and <code>charge_density</code> is the charge density on this real space grid.
<code>forces</code> and <code>stress</code> are the computed forces and stress tensor and <code>charge_density</code> contains the charge density on the real space grid. <code>shape_grid</code> is a three dimensional integer array which stores the shape of the real space grid, {{TAG|NGXF}}, {{TAG|NGYF}} and {{TAG|NGZF}} and <code>charge_density</code> is the charge density on this real space grid.
The <code>additions</code> dataclass consists of the following modifiable outputs
The <code>additions</code> dataclass consists of the following modifiable outputs
    lattice_vectors: NDArray[np.float64]
<syntaxhighlight lang="python" line>
    positions: NDArray[np.float64]
lattice_vectors: NDArray[np.float64]
positions: NDArray[np.float64]
</syntaxhighlight>


==Modifying quantities==
==Modifying quantities==

Revision as of 10:00, 14 October 2024

PLUGINS/STRUCTURE = .True. | .False.
Default: PLUGINS/STRUCTURE = .False. 

Description: PLUGINS/STRUCTURE calls the Python plugin for the structure interface for each ionic relaxation step


When PLUGINS/STRUCTURE=.TRUE., VASP calls the structure Python function at the end of each ionic relaxation step. The primary use-case of this tag is to modify the structure based on the computed energy, force and stress tensor.

Expected inputs

The structure Python function expects the following inputs,

    def structure(constants, additions):

where constants and additions and Python dataclasses. The constants dataclass consists of the following inputs, listed here with their associated datatypes

number_ions: int
number_ion_types: int
ion_types: NDArray[np.int32]
atomic_numbers: NDArray[np.int32]
lattice_vectors: NDArray[np.float64]
positions: NDArray[np.float64]
{{TAGBL|POMASS}}: NDArray[np.float64]
total_energy: float
forces: NDArray[np.float64]
stress: NDArray[np.float64]
shape_grid: NDArray[np.float64]
charge_density: NDArray[np.float64]

Note that the INCAR tags are capitalized. number_ions is the total number of ions listed in the POSCAR file, number_ion_types is the number of ion corresponding to each ion type in the convention of the POSCAR file, ion_types stores the total number of ion types, atomic_numbers contains the atomic number for each atom type, lattice_vectors and positions contain the lattice vectors and positions of the current SCF step forces and stress are the computed forces and stress tensor and charge_density contains the charge density on the real space grid. shape_grid is a three dimensional integer array which stores the shape of the real space grid, NGXF, NGYF and NGZF and charge_density is the charge density on this real space grid. The additions dataclass consists of the following modifiable outputs

lattice_vectors: NDArray[np.float64]
positions: NDArray[np.float64]

Modifying quantities

Modify the quantities listed in additions by adding to them.

def structure(constants, additions)
    additions.positions += np.ones((constants.number_ions,3))
Mind: You may not make modifications to quantities in constants