AG34461A_avg.py 2.59 KB
from abstract_instrument import abstract_instrument
import socket
import time

#==============================================================================

ALL_VAL_TYPE = ['DCV', 'ACV', 'DCI', 'ACI', 'RES2W', 'RES4W', 'FREQ']
ALL_CHANNELS = ['1']

ADDRESS = "192.168.0.61"
CONF_VAL_TYPE = ['CONF:VOLT:DC', 'CONF:VOLT:AC', 'CONF:CURR:DC', 'CONF:CURR:AC', 'CONF:RES', 'CONF:FRES', 'CONF:FREQ']

#==============================================================================

class AG34461A_avg(abstract_instrument):
    def __init__(self, channels, vtypes, address):
        self.address = address
        self.port = 5025
        self.channels = channels
        self.vtypes = vtypes

    def model(self):
        #self.send("*IDN?")
        #return self.read()
        return "AG34461A"

    def connect(self):
        print('Connecting to device @%s:%s...' %(self.address, self.port))
        self.sock = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
        self.sock.settimeout(10.0)    # Don't hang around forever
        self.sock.connect((self.address, self.port))
        self.send("SYST:BEEP")
        print('  --> Ok')
        print(self.model())
        self.configure()

    def configure(self):
        self.send("*RST")
        for ch in self.channels:
            self.send(CONF_VAL_TYPE[ALL_VAL_TYPE.index(self.vtypes[self.channels.index(ch)])])
        self.send("CONF:VOLT:DC 10")
        #self.send("VOLT:DC:NPLC 10")
        self.send("SAMP:COUN 1")
        self.send("TRIG:COUN 5")
        self.send("TRIG:DEL 0")
        self.send("SENS:ZERO:AUTO OFF")
        self.send("TRIG:SOUR IMM")
        self.send("INIT")

    def getValue(self):
        mes = ''
        for ch in self.channels:
            self.send("FETC?")
            mesTemp = self.read()
            mesTemp = map(float, mesTemp.split(','))
            mes = mes + '\t' + str(sum(mesTemp)/len(mesTemp))
            self.send("INIT")
        mes = mes + '\n'
        return mes

    def read(self):
        ans = ''
        nb_data_list = []
        nb_data = ''
        try:
            while ans != '\n':
                ans = self.sock.recv(1)
                nb_data_list.append(ans) # Return the number of data
            list_size = len(nb_data_list)
            for j in range (0, list_size):
                nb_data = nb_data+nb_data_list[j]
            return nb_data
        except socket.timeout:
            print "Socket timeout error when reading."
            raise

    def disconnect(self):
        self.send('*RST')
        self.send("SYST:BEEP")
        self.sock.close()

    def send(self, command):
        self.sock.send("%s\n"%command)