# Connect to instruments using configuration profiles
# These will use simulation mode if physical instruments aren't available

osc = AutoInstrument.from_config("keysight/DSOX1204G")
awg = AutoInstrument.from_config("keysight/EDU33212A")

print("✅ Instruments connected successfully!")
print(f"📊 Oscilloscope: {type(osc).__name__}")
print(f"🌊 Waveform Generator: {type(awg).__name__}")

# Query oscilloscope identification
osc_id = osc.id()
print(f"📊 Oscilloscope ID: {osc_id}")

# Query waveform generator identification
awg_id = awg.id()
print(f"🌊 AWG ID: {awg_id}")

# Reset AWG to known state and configure basic waveform
awg.reset()
print("🔄 AWG reset to default state")

# Enable channel 1 output
awg.output(1, True)
print("🔌 Channel 1 output enabled")

# Set waveform type to sine wave
awg.set_waveform(1, "SIN")
print("🌊 Waveform set to sine wave")

# Set frequency to 25 kHz
frequency = 25e3  # 25 kHz
awg.set_frequency(1, frequency)
print(f"📡 Frequency set to {frequency / 1000:.1f} kHz")

# Set output amplitude to 1.8V peak-to-peak
amplitude = 1.8  # Volts peak-to-peak
awg.set_voltage(1, amplitude)
print(f"⚡ Amplitude set to {amplitude}V peak-to-peak")

# Turn off channel 1 display (not used in this example)
osc.display_channel(1, False)
print("📺 Channel 1 display disabled")

# Enable channel 2 display (where we'll connect the AWG output)
osc.display_channel(2, True)
print("📺 Channel 2 display enabled")

# Set trigger source to channel 2
osc.set_trigger_source(2)
print("⚡ Trigger source set to channel 2")

# Set timebase to 10 μs/div (appropriate for 25 kHz signal)
timebase = 1e-5  # 10 μs/div
osc.set_timebase_scale(timebase)
print(f"⏱️ Timebase set to {timebase * 1e6:.0f} μs/div")

# Set channel 2 vertical scale and offset
v_scale = 10  # V/div (adjust based on expected signal amplitude)
v_offset = 0  # No vertical offset
osc.set_channel_axis(2, v_scale, v_offset)
print(f"📈 Channel 2: {v_scale}V/div, offset: {v_offset}V")

# Display measurement configuration summary
print("📋 MEASUREMENT CONFIGURATION SUMMARY")
print("=" * 40)
print("🌊 AWG Channel 1:")
print("   • Waveform: Sine wave")
print(f"   • Frequency: {frequency / 1000:.1f} kHz")
print(f"   • Amplitude: {amplitude}V p-p")
print("   • Output: Enabled")
print()
print("📊 Oscilloscope Channel 2:")
print("   • Display: Enabled")
print(f"   • Timebase: {timebase * 1e6:.0f} μs/div")
print(f"   • Vertical: {v_scale}V/div")
print("   • Trigger: Channel 2")
print()
print("✅ Configuration complete! Connect AWG Ch1 output to OSC Ch2 input.")

# Properly close instrument connections
try:
    osc.close()
    print("📊 Oscilloscope connection closed")
except Exception as e:
    print(f"⚠️ Error closing oscilloscope: {e}")

try:
    awg.close()
    print("🌊 AWG connection closed")
except Exception as e:
    print(f"⚠️ Error closing AWG: {e}")

print("\n✅ All instruments disconnected successfully!")