from qolab.hardware.scpi import SCPIinstr
from qolab.hardware.basic import BasicInstrument

class RFGenerator(BasicInstrument):
    # Minimal set of methods to be implemented by a RFGenerator
    def __init__(self):
        BasicInstrument.__init__(self)
        self.config['Device type']='RFGenerator'
        self.config['Device model'] = 'Generic RF generator Without Hardware interface'
        self.config['FnamePrefix'] = 'rfgen'
        self.deviceProperties = ['FreqFixed', ]

class RFGeneratorSCPI(SCPIinstr, RFGenerator):
    """     
    Do not instantiate directly, use
    rm = pyvisa.ResourceManager()
    RFGeneratorSCPI(rm.open_resource('TCPIP::192.168.0.2::INSTR'))
    """
    def __init__(self, resource):
        SCPIinstr.__init__(self, resource)
        RFGenerator.__init__(self)
        self.config['DeviceId'] = str.strip(self.idn)

from .agilent_e8257d import AgilentE8257D