camrasdevices.py

import hpib
from hpib import HpibDevice

_port = hpib.PrologixGpibUsb()

class CamrasHpibDevice(HpibDevice):
    """Wrapper around HPIB commands"""
    def __init__(self, address):
        super(CamrasHpibDevice, self).__init__(address, _port)
        self.__frequency = 0

    @property
    def frequency(self):
        """Returns the frequency (in Hz) of the device. Throws a RuntimeError if no response"""
        freq_str = self.query("freq?")
        if len(freq_str)==0:
            raise RuntimeError("Camras device at address {} is not responding".format(self._hpib_address))
        return int(float(self.query("freq?")))

    @frequency.setter
    def frequency(self, freq):
        """Set the device to the specified frequency. Throws a RuntimeError if failed"""
        print("setting frequency to", freq)
        self.command("freq {:d} Hz".format(int(freq)))
        new_freq = self.frequency
        if new_freq != freq:
            raise RuntimeError("Setting frequency failed: tried to set to {}, it is now {}".format(
                               freq, new_freq))

class Receiver(CamrasHpibDevice):
    """Wrapper around HPIB commands for the Rohde & Schwartz receiver"""

    def __init__(self):
        super(Receiver, self).__init__(1)

class LocalOscillator(CamrasHpibDevice):
    """Wrapper around HPIB commands for the local oscillator"""

    def __init__(self):
        super(LocalOscillator, self).__init__(28)

class SignalGenerator(CamrasHpibDevice):
    """Wrapper around HPIB commands for the signal generator"""

    def __init__(self):
        super(SignalGenerator, self).__init__(14)