import qutip as qt
import numpy as np
import logging
from src.nv_quantum_setup import NVCenter

# Configure logger
logger = logging.getLogger(__name__)
class SimulationEngine: #Manages the time-evolution of the quantum system

  def __init__(self, system: NVCenter):
    self.system = system
    self.cfg = system.cfg
  def run(self):
    """Executes the simulation using QuTiP mesolve/sesolve."""
    logger.info(f"Starting Simulation: {self.cfg.n_steps} steps, T_final={self.cfg.t_final}s")
    # 1. Construct Hamiltonian: H = H0 + [H_int, h(t)]
    H0 = self.system.get_static_hamiltonian()
    H_int = self.system.get_interaction_operator()
    # QuTiP format for time-dependent H
    H = [H0, [H_int, self._strain_func]]
    args = {'h_max': self.cfg.h_max, 'omega_gw': self.cfg.omega_gw}