diff options
author | David Luevano <55825613+luevano@users.noreply.github.com> | 2019-12-12 04:26:29 -0700 |
---|---|---|
committer | David Luevano <55825613+luevano@users.noreply.github.com> | 2019-12-12 04:26:29 -0700 |
commit | 00fdf0166789286a83649485b864a2738aaa70d3 (patch) | |
tree | 582465cc2925413a548de9fd85320e7fdbe3af1b /lj_matrix.py | |
parent | 81f44c4fe753f7aecbda34a7873dd0228df8a335 (diff) |
Reformat matrices calculation
Diffstat (limited to 'lj_matrix.py')
-rw-r--r-- | lj_matrix.py | 8 |
1 files changed, 8 insertions, 0 deletions
diff --git a/lj_matrix.py b/lj_matrix.py index 353631633..6769bc0c3 100644 --- a/lj_matrix.py +++ b/lj_matrix.py @@ -20,6 +20,8 @@ LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE. """ +import time +from misc import printc import math import numpy as np from numpy.linalg import eig @@ -177,7 +179,13 @@ def lj_matrix_multiple(mol_data, as_eig: if data should be returned as matrix or array of eigenvalues. bohr_radius_units: if units should be in bohr's radius units. """ + printc('L-J Matrices calculation started.', 'CYAN') + tic = time.perf_counter() + ljm_data = np.array([lj_matrix(mol, nc, max_len, as_eig, bohr_radius_units) for mol, nc in zip(mol_data, nc_data)]) + toc = time.perf_counter() + printc('\tL-JM calculation took {:.4f} seconds.'.format(toc-tic), 'GREEN') + return ljm_data |