1
0
mirror of https://github.com/TREX-CoE/qmc-lttc.git synced 2024-06-30 00:44:56 +02:00
qmc-lttc/hydrogen.py

22 lines
438 B
Python
Raw Normal View History

2021-01-03 18:45:58 +01:00
import numpy as np
def potential(r):
2021-01-26 00:22:37 +01:00
distance = np.sqrt(np.dot(r,r))
assert (distance > 0)
return -1. / distance
2021-01-03 18:45:58 +01:00
def psi(a, r):
return np.exp(-a*np.sqrt(np.dot(r,r)))
def kinetic(a,r):
2021-01-26 00:22:37 +01:00
distance = np.sqrt(np.dot(r,r))
assert (distance > 0.)
return -0.5 * (a**2 - (2.*a)/distance)
2021-01-03 18:45:58 +01:00
def e_loc(a,r):
return kinetic(a,r) + potential(r)
2021-01-12 01:01:52 +01:00
def drift(a,r):
ar_inv = -a/np.sqrt(np.dot(r,r))
return r * ar_inv