Révision e1c5f171 modules/screening.py
b/modules/screening.py | ||
---|---|---|
9 | 9 |
atoms.info[prop_name] = prop_val |
10 | 10 |
|
11 | 11 |
|
12 |
def select_confs(orig_conf_list: list, calc_dirs: list, magns: list, num_sel: int, code: str): |
|
12 |
def select_confs(orig_conf_list: list, calc_dirs: list, magns: list, |
|
13 |
num_sel: int, code: str): |
|
13 | 14 |
"""Takes a list ase.Atoms and selects the most different magnitude-wise. |
14 | 15 |
|
15 | 16 |
Given a list of ase.Atoms objects and a list of magnitudes, it selects a |
... | ... | |
29 | 30 |
from modules.formats import collect_energies |
30 | 31 |
|
31 | 32 |
conf_list = deepcopy(orig_conf_list) |
32 |
selected_ids = []
|
|
33 |
conf_enrgs, mois, selected_ids = [], [], []
|
|
33 | 34 |
if num_sel >= len(conf_list): |
34 | 35 |
logger.warning('Number of conformers per magnitude is equal or larger ' |
35 | 36 |
'than the total number of conformers. Using all ' |
... | ... | |
65 | 66 |
return [conf_list[idx] for idx in selected_ids] |
66 | 67 |
|
67 | 68 |
|
68 |
def get_vect_angle(v1, v2, degrees=False):
|
|
69 |
def get_vect_angle(v1: list, v2: list, degrees=False):
|
|
69 | 70 |
"""Computes the angle between two vectors. |
70 | 71 |
|
71 | 72 |
@param v1: The first vector. |
Formats disponibles : Unified diff