17.1.1.2. pysisyphus.calculators package
17.1.1.2.1. Submodules
17.1.1.2.2. pysisyphus.calculators.AFIR module
- class pysisyphus.calculators.AFIR.AFIR(calculator, fragment_indices, gamma, rho=1, p=6, ignore_hydrogen=False, zero_hydrogen=True, complete_fragments=True, dump=True, h5_fn='afir.h5', h5_group_name='afir', **kwargs)[source]
Bases:
Calculator- __init__(calculator, fragment_indices, gamma, rho=1, p=6, ignore_hydrogen=False, zero_hydrogen=True, complete_fragments=True, dump=True, h5_fn='afir.h5', h5_group_name='afir', **kwargs)[source]
Artifical Force Induced Reaction calculator.
Currently, there are no automated drivers to run large-scale AFIR calculations with many different initial orientations and/or increasing collision energy parameter γ. Nontheless, selected AFIR calculations can be carried out by hand. After convergence, artificial potential & forces, as well as real energies and forces can be plotted with 'pysisplot --afir'. The highest energy point along the AFIR path can then be selected for a subsequent TS-optimization, e.g. via 'pysistrj --get [index] optimzation.trj'.
Future versions of pysisyphus may provide drivers for more automatted AFIR calculations.
- Parameters:
calculator (
Calculator) -- Actual QC calculator that provides energies and its derivatives, that are modified by the AFIR calculator, e.g., ORCA or Psi4.fragment_indices (
List[List[int]]) -- List of lists of integers, specifying the separate fragments. If the indices in theses lists don't comprise all atoms in the molecule, the reamining indices will be added as a separate fragment. If a AFIR calculation is carried out with 2 fragments and 'complete_fragments' is True (see below) it is enough to specify only the indices of one fragment, e.g., for a system of 10 atoms 'fragment_indices=[[0,1,2,3]]' is enough. The second system will be set up automatically with indices [4,5,6,7,8,9].gamma (
Union[_SupportsArray[dtype[Any]],_NestedSequence[_SupportsArray[dtype[Any]]],bool,int,float,complex,str,bytes,_NestedSequence[Union[bool,int,float,complex,str,bytes]]]) -- Collision energy parameter γ in au. For 2 fragments it can be a single integer, while for > 2 fragments a list of gammas must be given, specifying the pair-wise collision energy parameters. For 3 fragments 3 gammas must be given [γ_01, γ_02, γ_12], for 4 fragments 6 gammas would be required [γ_01, γ_02, γ_03, γ_12, γ_13, γ_23] and so on.rho (
Union[_SupportsArray[dtype[Any]],_NestedSequence[_SupportsArray[dtype[Any]]],bool,int,float,complex,str,bytes,_NestedSequence[Union[bool,int,float,complex,str,bytes]]], default:1) -- Direction of the artificial force, either 1 or -1. The same comments as for gamma apply. For 2 fragments a single integer is enough, for > 2 fragments a list of rhos must be given (see above). For rho=1 fragments are pushed together, for rho=-1 fragments are pulled apart.p (
int, default:6) -- Exponent p used in the calculation of the weight function ω. Defaults to 6 and probably does not have to be changed.ignore_hydrogen (
bool, default:False) -- Whether hydrogens are ignored in the calculation of the artificial force. All weights between atom pairs containing hydrogen will be set to 0.zero_hydrogen (
bool, default:True) -- Whether to use 0.0 as covalent radius for hydrogen in the weight function. Compared to 'ignore_hydrogen', which results in zero weights for all atom pairs involving hydrogen, 'zero_hydrogen' may be non-zero, depending on the covalent radius of the second atom in the pair.complete_fragments (
bool, default:True) -- Whether an incomplete specification in 'fragment_indices' is automatically completed.dump (
bool, default:True) -- Whether an HDF5 file is created.h5_fn (
str, default:'afir.h5') -- Filename of the HDF5 file used for dumping.h5_group_name (
str, default:'afir') -- HDF5 group name used for dumping.**kwargs -- Keyword arguments passed to the Calculator baseclass.
- property charge
- get_hessian(atoms, coords, **prepare_kwargs)[source]
Get Hessian matrix. Fall back to numerical Hessian, if not overriden.
Preferrably, this method should provide an analytical Hessian.
- property mult
- set_atoms_and_funcs(atoms, coords)[source]
Initially atoms was also an argument to the constructor of AFIR. I removed it so creation becomes easier. The first time a calculation is requested with a proper atom set everything is set up (cov. radii, afir function and corresponding gradient). Afterwards there is only a check if atoms != None and it is expected that all functions are properly set.
fragment_indices can also be incomplete w.r.t. to the number of atoms. If the sum of the specified fragment atoms is less than the number of atoms present then all remaining unspecified atoms will be gathered in one fragment.
17.1.1.2.3. pysisyphus.calculators.AnaPot module
17.1.1.2.4. pysisyphus.calculators.AnaPot2 module
17.1.1.2.5. pysisyphus.calculators.AnaPot3 module
17.1.1.2.6. pysisyphus.calculators.AnaPot4 module
17.1.1.2.7. pysisyphus.calculators.AnaPotBase module
17.1.1.2.8. pysisyphus.calculators.AnaPotCBM module
17.1.1.2.9. pysisyphus.calculators.AtomAtomTransTorque module
17.1.1.2.10. pysisyphus.calculators.Calculator module
- class pysisyphus.calculators.Calculator.Calculator(calc_number=0, charge=0, mult=1, base_name='calculator', pal=1, mem=1000, keep_kind='all', check_mem=True, retry_calc=0, last_calc_cycle=None, clean_after=True, out_dir='qm_calcs', force_num_hess=False, num_hess_kwargs=None)[source]
Bases:
object- __init__(calc_number=0, charge=0, mult=1, base_name='calculator', pal=1, mem=1000, keep_kind='all', check_mem=True, retry_calc=0, last_calc_cycle=None, clean_after=True, out_dir='qm_calcs', force_num_hess=False, num_hess_kwargs=None)[source]
Base-class of all calculators.
Meant to be extended.
- Parameters:
calc_number (int, default=0) -- Identifier of the Calculator. Used in distinguishing it from other Calculators, e.g. in ChainOfStates calculations. Also used in the creation of filenames.
charge (int, default=0) -- Molecular charge.
mult (int, default=1) -- Molecular multiplicity (1 = singlet, 2 = doublet, ...)
base_name (str, default=calculator) -- Generated filenames will start with this string.
pal (int, default=1) -- Positive integer that gives the number of physical cores to use on 1 node.
mem (int, default=1000) -- Mememory per core in MB. The total amount of memory is given as mem*pal.
check_mem (bool, default=True) -- Whether to adjust the requested memory if too much is requested.
retry_calc (int, default=0) -- Number of additional retries when calculation failed.
last_calc_cycle (int) -- Internal variable used in restarts.
clean_after (bool) -- Delete temporary directory were calculations were executed after a calculation.
out_dir (str) -- Path that is prepended to generated filenames.
force_hess_kwargs (bool, default False) -- Force numerical Hessians.
num_hess_kwargs (dict) -- Keyword arguments for finite difference Hessian calculation.
- clean(path)[source]
Delete the temporary directory.
- Parameters:
path (Path) -- Directory to delete.
- conf_key = None
- get_hessian(atoms, coords, **prepare_kwargs)[source]
Get Hessian matrix. Fall back to numerical Hessian, if not overriden.
Preferrably, this method should provide an analytical Hessian.
- get_restart_info()[source]
Return a dict containing chkfiles.
- Returns:
restart_info -- Dictionary holding the calculator state. Used for restoring calculaters in restarted calculations.
- Return type:
dict
- keep(path)[source]
Backup calculation results.
- Parameters:
path (Path) -- Temporary directory of the calculation.
- Returns:
kept_fns -- Dictonary holding the filenames that were backed up. The keys correspond to the type of file.
- Return type:
dict
- log(message='')[source]
Write a log message.
Wraps the logger variable.
- Parameters:
message (str) -- Message to be logged.
- make_fn(name, counter=None, return_str=False)[source]
Make a full filename.
Return a full filename including the calculator name and the current counter given a suffix.
- Parameters:
name (str) -- Suffix of the filename.
counter (int, optional) -- If not given use the current calc_counter.
return_str (int, optional) -- Return a string instead of a Path when True.
- Returns:
fn -- Filename.
- Return type:
str
- property name
- prepare(inp)[source]
Prepare a temporary directory and write input.
Similar to prepare_path, but the input is also written into the prepared directory.
17.1.1.2. Paramters
- inpstr
Input to be written into the file
self.inp_fnin the prepared directory.
- returns:
- path: Path
Prepared directory.
- prepare_coords(atoms, coords)[source]
Get 3d coords in Angstrom.
Reshape internal 1d coords to 3d and convert to Angstrom.
- Parameters:
atoms (iterable) -- Atom descriptors (element symbols).
coords (np.array, 1d) -- 1D-array holding coordinates in Bohr.
- Returns:
coords -- 3D-array holding coordinates in Angstrom.
- Return type:
np.array, 3d
- prepare_path(use_in_run=False)[source]
Get a temporary directory handle.
Create a temporary directory that can later be used in a calculation.
- Parameters:
use_in_run (bool, option) -- Sets the internal variable
self.path_already_preparedthat is later read byself.run(). No new temporary directory will be created inself.run().- Returns:
- path: Path
Prepared directory.
- prepare_pattern(raw_pat)[source]
Prepare globs.
Transforms an entry of
self.to_keepinto a glob and a key suitable for the use inself.keep().- Parameters:
raw_pat (str) -- Entry of
self.to_keep- Returns:
pattern (str) -- Glob that can be used in Path.glob()
multi (bool) -- Flag if glob may match multiple files.
key (str) -- A key to be used in the
kept_fnsdict.
- prepare_turbo_coords(atoms, coords)[source]
Get a Turbomole coords string.
- Parameters:
atoms (iterable) -- Atom descriptors (element symbols).
coords (np.array, 1d) -- 1D-array holding coordinates in Bohr.
- Returns:
coords -- String holding coordinates in Turbomole coords format.
- Return type:
str
- prepare_xyz_string(atoms, coords)[source]
Returns a xyz string in Angstrom.
- Parameters:
atoms (iterable) -- Atom descriptors (element symbols).
coords (np.array, 1d) -- 1D-array holding coordinates in Bohr.
- Returns:
xyz_str -- Coordinates in .xyz format.
- Return type:
string
- print_out_fn(path)[source]
Print calculation output.
Prints the output of a calculator after a calculation.
- Parameters:
path (Path) -- Temporary directory of the calculation.
- restore_org_hessian()[source]
Restore original 'get_hessian' method, which may also fallback to numerical Hessians, if not implemented.
- run(inp, calc, add_args=None, env=None, shell=False, hold=False, keep=True, cmd=None, inc_counter=True, run_after=True, parser_kwargs=None, symlink=True)[source]
Run a calculation.
The bread-and-butter method to actually run an external quantum chemistry code.
- Parameters:
inp (str) -- Input for the external program that is written to the temp-dir.
calc (str, hashable) -- Key (and more or less type of calculation) to select the right parsing function from
self.parser_funcs.add_args (iterable, optional) -- Additional arguments that will be appended to the program call.
env (Environment, optional) -- A potentially modified environment for the subprocess call.
shell (bool, optional) -- Use a shell to execute the program call. Need for Turbomole were we chain program calls like dscf; escf.
hold (bool, optional) -- Wether to remove the temporary directory after the calculation.
keep (bool, optional) -- Wether to backup files as specified in
self.to_keep(). Usually you want this.cmd (str or iterable, optional) -- Overwrites
self.base_cmd.inc_counter (bool, optional) -- Wether to increment the counter after a calculation.
- Returns:
results -- Dictionary holding all applicable results of the calculations like the energy, a forces vector and/or excited state energies from TDDFT.
- Return type:
dict
- run_after(path)[source]
Meant to be extended.
This method is called after a calculation was done, but before entering
self.keep()andself.clean(). Can be used to call tools like formchk or ricctools.
- set_restart_info(restart_info)[source]
Sets restart information (chkfiles etc.) on the calculator.
- Parameters:
restart_info (dict) -- Dictionary holding the calculator state. Used for restoring calculaters in restarted calculations.
- verify_chkfiles(chkfiles)[source]
Checks if given chkfiles exist and return them as Paths
- Parameters:
chkfiles (dict) -- Dictionary holding the chkfiles. The keys correspond to the attribute names, the values are strs holding the (potentially full) filename (path).
- Returns:
paths -- Dictionary of Paths.
- Return type:
dict
- class pysisyphus.calculators.Calculator.HessKind(value)
Bases:
EnumAn enumeration.
- NUMERICAL = 2
- ORG = 1