import numpy as np
from qilisdk.analog import Schedule, X, Y, Z
from qilisdk.backends import QutipBackend
from qilisdk.core import ket, tensor_prod
from qilisdk.functionals import TimeEvolution
T = 10 # Total evolution time
dt = 0.1 # Time-step
steps = np.linspace(0, T, int(T / dt))
nqubits = 1
# Define two Hamiltonians for the simulation
H1 = sum(X(i) for i in range(nqubits))
H2 = sum(Z(i) for i in range(nqubits))
# Create a schedule for the time evolution
schedule = Schedule(
T,
dt,
hamiltonians={"h1": H1, "h2": H2},
schedule={t: {"h1": 1 - steps[t] / T, "h2": steps[t] / T} for t in range(len(steps))},
)
# Prepare an initial state (equal superposition)
state = tensor_prod([(ket(0) + ket(1)).unit() for _ in range(nqubits)]).unit()
# Perform time evolution on the CUDA backend with observables to monitor
time_evolution = TimeEvolution(
schedule=schedule, initial_state=state, observables=[Z(0), X(0), Y(0)], store_intermediate_results=True
)
backend = QutipBackend()
results = backend.execute(time_evolution)
print("Time Evolution Results:")
print("Final Expectation Values:", results.final_expected_values)
print("Final State:", results.final_state)
