Spaces:
Runtime error
Runtime error
File size: 934 Bytes
1c703f0 |
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 |
from .fast_math import op_ket, bj
def time_evolve_h(phi_in, ham, w0, dt, bj_list=None, eps=10e-5):
phi_0 = phi_in
phi_1 = op_ket(ham, phi_in)
if bj_list is None:
phi_out = bj(m=0, x=w0 * dt) * phi_0 + 2 * (-1j) * bj(m=1, x=w0 * dt) * phi_1
m = 2
bjm = bj(m, w0 * dt)
while abs(bjm) > eps:
phi_2 = 2 * op_ket(ham, phi_1) - phi_0
phi_out = phi_out + 2 * (-1j ** m) * bj(m, x=w0 * dt) * phi_2
phi_0 = phi_1
phi_1 = phi_2
m += 1
bjm = bj(m, w0 * dt)
return phi_out
else:
phi_out = bj_list[0] * phi_0 + 2 * (-1j) * bj_list[1] * phi_1
m = 2
while abs(bj_list[m]) > eps:
phi_2 = 2 * op_ket(ham, phi_1) - phi_0
phi_out = phi_out + 2 * (-1j ** m) * bj_list[m] * phi_2
phi_0 = phi_1
phi_1 = phi_2
m += 1
return phi_out
|