/
import_data.py
41 lines (35 loc) · 1.34 KB
/
import_data.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
#!/usr/bin/env python3
## vi: tabstop=4 shiftwidth=4 softtabstop=4 expandtab
import numpy as np
from data import coeff_data, dip_data, eri_data, orben_data
def import_data():
n_bas = 7
data = {
"n_alpha": 5,
"n_beta": 5,
"n_orbs_alpha": 7,
"n_orbs_beta": 7,
"n_bas": n_bas,
"energy_scf": -7.4959319286025718e+01,
"energy_nuclear_repulsion": 9.251479269240862,
"restricted": True,
"threshold": 1e-12,
"spin_multiplicity": 1,
"multipoles": {
"elec_0": -10,
"nuclear_0": 10,
},
}
n_orbs = data["n_orbs_alpha"] + data["n_orbs_beta"]
data["multipoles"]["nuclear_1"] = np.array([1.693194615993441, 0.,
1.196196642772152])
data["multipoles"]["elec_1"] = np.array(dip_data).reshape(3, n_bas, n_bas)
data["occupation_f"] = np.array(5 * [1] + [0, 0] + 5 * [1] + [0, 0.])
data["orbcoeff_fb"] = np.array(coeff_data).reshape((n_orbs, n_bas))
data["orben_f"] = np.array(orben_data).reshape((n_orbs))
data["eri_ffff"] = np.array(eri_data).reshape((n_orbs, n_orbs,
n_orbs, n_orbs))
data["fock_ff"] = np.zeros((n_orbs, n_orbs))
for i in range(n_orbs):
data["fock_ff"][i, i] = orben_data[i]
return data