Hardware Drivers

# Auto-detection settings
auto_detect:
  enabled: true
  usb: true
  serial: true
  gpib: true
  tcp: false  # Requires explicit config

# Explicit device configurations
devices:
  # USB TMC device (auto-detect friendly)
  - name: main_dmm
    driver: keysight_34461a
    connection:
      type: usb_tmc
      serial: MY54321  # Optional: specific unit
    config:
      default_timeout: 5.0
      
  # TCP/IP device (must be explicit)
  - name: network_psu
    driver: rigol_dp832
    connection:
      type: tcp
      address: "192.168.1.100"
      port: 5555
      
  # Serial device
  - name: test_fixture
    driver: custom_fixture
    connection:
      type: serial
      port: "/dev/ttyUSB0"
      baudrate: 115200
      parity: none
      stopbits: 1
      
  # GPIB device
  - name: legacy_counter
    driver: hp_53131a
    connection:
      type: gpib
      address: 12
      adapter: prologix

from seqmaster.drivers.base import BaseDriver, DriverCapability
from seqmaster.drivers.registry import register_driver


@register_driver(
    name="my_custom_dmm",
    vendor="MyCompany",
    models=["DMM-1000", "DMM-2000"],
    usb_ids=[("1234", "5678")]  # VID, PID for auto-detect
)
class MyCustomDMM(BaseDriver):
    """Driver for MyCompany DMM-1000/2000 series."""
    
    # Declare capabilities this driver provides
    capabilities = [
        DriverCapability.MEASURE_DC_VOLTAGE,
        DriverCapability.MEASURE_DC_CURRENT,
        DriverCapability.MEASURE_RESISTANCE,
    ]
    
    def initialize(self):
        """Initialize the device."""
        self.write("*RST")  # Reset to defaults
        self.write("*CLS")  # Clear status
        
        # Verify identity
        idn = self.query("*IDN?")
        self.logger.info(f"Connected to: {idn}")
        
    def measure_dc_voltage(self, range="AUTO") -> float:
        """Measure DC voltage.
        
        Args:
            range: Measurement range (AUTO, 0.1, 1, 10, 100, 1000)
            
        Returns:
            Measured voltage in volts
        """
        self.write(f"CONF:VOLT:DC {range}")
        value = self.query("READ?")
        return float(value)
        
    def measure_dc_current(self, range="AUTO") -> float:
        """Measure DC current."""
        self.write(f"CONF:CURR:DC {range}")
        value = self.query("READ?")
        return float(value)
        
    def measure_resistance(self, range="AUTO", four_wire=False) -> float:
        """Measure resistance."""
        mode = "FRES" if four_wire else "RES"
        self.write(f"CONF:{mode} {range}")
        value = self.query("READ?")
        return float(value)
        
    def close(self):
        """Cleanup when disconnecting."""
        self.write("*RST")  # Reset before closing
        super().close()

measurement:
  voltage:
    dc:
      - keysight_34461a
      - fluke_8845a
      - rigol_dm3058
    ac:
      - keysight_34461a
      - fluke_8845a
  current:
    dc:
      - keysight_34461a
      - keithley_2000
    ac:
      - keysight_34461a
  resistance:
    two_wire:
      - keysight_34461a
      - fluke_8845a
    four_wire:
      - keysight_34461a
      - keithley_2000

source:
  dc_power:
    - rigol_dp832
    - keysight_e36xx
    - siglent_spd3303x
  waveform:
    - rigol_dg1022z
    - siglent_sdg1032x

digital_io:
  gpio:
    - raspberry_pi_gpio
    - arduino_firmata
    - labjack_u3
  relay:
    - usb_relay_hid
    - modbus_relay