mirror of
https://github.com/TREX-CoE/qmc-lttc.git
synced 2024-12-22 04:15:01 +01:00
14 lines
265 B
Python
14 lines
265 B
Python
|
import numpy as np
|
||
|
|
|
||
|
|
def potential(r):
|
||
|
|
return -1. / np.sqrt(np.dot(r,r))
|
||
|
|
|
||
|
|
def psi(a, r):
|
||
|
|
return np.exp(-a*np.sqrt(np.dot(r,r)))
|
||
|
|
|
||
|
|
def kinetic(a,r):
|
||
|
|
return -0.5 * (a**2 - (2.*a)/np.sqrt(np.dot(r,r)))
|
||
|
|
|
||
|
|
def e_loc(a,r):
|
||
|
|
return kinetic(a,r) + potential(r)
|