master-thesis/python/energy_flow_proper/01_zero_temperature/config.py

70 lines
1.7 KiB
Python
Raw Normal View History

from hops.core.hierarchy_parameters import HIParams, HiP, IntP, SysP, ResultType
from hops.core.hierarchyLib import HI
from hops.util.bcf_fits import get_ohm_g_w
from hops.util.truncation_schemes import TruncationScheme_Power_multi
import hops.util.bcf
import numpy as np
import hops.util.matrixLib as ml
from stocproc import StocProc_FFT
wc = 5
s = 1
# The BCF fit
bcf_terms = 6
g, w = get_ohm_g_w(bcf_terms, s, wc)
integration = IntP(t_max=5, t_steps=1000)
system = SysP(
H_sys=0.5 * np.array([[1, 0], [0, -1]]),
L=0.5 * np.array([[0, 1], [1, 0]]),
psi0=np.array([1, 1]),
g=g,
w=w,
bcf_scale=0.8,
T=0,
)
params = HIParams(
SysP=system,
IntP=integration,
HiP=HiP(
nonlinear=True,
normalized_by_hand=True,
result_type=ResultType.ALL,
truncation_scheme=TruncationScheme_Power_multi.from_g_w(
g=g, w=w, p=1, q=0.5, kfac=1.4
),
),
Eta=StocProc_FFT(
spectral_density=hops.util.bcf.OhmicSD_zeroTemp(
s,
1,
wc,
),
alpha=hops.util.bcf.OhmicBCF_zeroTemp(
s,
1,
wc,
),
t_max=integration.t_max,
intgr_tol=1e-3,
intpl_tol=1e-3,
negative_frequencies=False,
),
EtaTherm=StocProc_FFT(
spectral_density=hops.util.bcf.Ohmic_StochasticPotentialDensity(
s, 1, wc, beta=1 / system.__non_key__["T"]
),
alpha=hops.util.bcf.Ohmic_StochasticPotentialCorrelations(
s, 1, wc, beta=1 / system.__non_key__["T"]
),
t_max=integration.t_max,
intgr_tol=1e-3,
intpl_tol=1e-3,
negative_frequencies=False,
)
if system.__non_key__["T"]
else None,
)