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