from simulations.core import EnvironmentalFieldSimulator, CHSHExperimentSimulator
from simulations.core.multi_scale_simulation import EnvironmentalCorrelation, OpenQuantumSystem

# Option 1: Traditional Field Simulator
env_sim = EnvironmentalFieldSimulator(
    field_mass=1e-6,        # 1 μeV
    coupling_strength=1e-3,  # Weak coupling
    temperature=300.0        # Room temperature
)

# Set up Bell test experiment
chsh_sim = CHSHExperimentSimulator(env_sim)

# Run simulation
results = chsh_sim.simulate_bell_experiment(n_trials=10000)
print(f"CHSH parameter: {results['S_mean']:.4f} ± {results['S_std']:.4f}")

# Option 2: Advanced Multi-scale Simulator
env = EnvironmentalCorrelation(
    correlation_type='structured',
    correlation_time=2.0,
    coupling_strength=0.1,
    temperature=0.1
)

# Create demonstration of quantum correlation enhancement
amplifier = QuantumCorrelationAmplifier()
results = amplifier.demonstrate_eqfe()

# Find optimal conditions for enhancement
temperatures = np.linspace(50, 400, 20)
enhancements = []

for T in temperatures:
    env_sim = EnvironmentalFieldSimulator(
        field_mass=1e-6,
        coupling_strength=1e-3,
        temperature=T
    )
    A = env_sim.amplification_factor(measurement_time=1e-6)
    enhancements.append(A)

optimal_temp = temperatures[np.argmax(enhancements)]
print(f"Optimal temperature: {optimal_temp:.1f} K")