AG34972A.py 2.75 KB
from abstract_instrument import abstract_instrument
import socket

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

ALL_VAL_TYPE = "DCV, ACV, DCI, ACI, RES2W, RES4W, FREQ"

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

class AG34972A(abstract_instrument):
    def __init__(self, adress="192.168.0.72", vtype="DCV"):
        self.adress = adress
        self.port = 5025
        self.vtype = vtype

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

    def connect(self):
        try:
            print('Connecting to device @%s:%s...' %(self.adress, self.port))
            self.sock = socket.socket(socket.AF_INET,
                                 socket.SOCK_STREAM,
                                 socket.IPPROTO_TCP)
            self.sock.settimeout(2.0)    # Don't hang around forever
            self.sock.connect((self.adress, self.port))
            print('  --> Ok')

            print(self.model())

            if self.vtype == "DCV":
                self.send("CONF:VOLT:DC (@102:103)")
            elif self.vtype == "ACV":
                self.send("CONF:VOLT:AC (@102:103)")
            elif self.vtype == "DCI":
                self.send("CONF:CURR:DC (@102:103)")
            elif self.vtype == "ACI":
                self.send("CONF:CURR:AC (@102:103)")
            elif self.vtype == "RES2W":
                self.send("CONF:RES (@102:103)")
            elif self.vtype == "RES4W":
                self.send("CONF:FRES (@102:103)")
            elif self.vtype == "FREQ":
                self.send("CONF:FREQ (@102:103)")
            else:
                print("Wrong -v argument")
                raise

        except socket.timeout:
            print("Socket timeout error during connection.")
            raise
        except socket.error as er:
            print("Socket error during connection: " + str(er))
            raise
        except Exception as er:
            print("Unexpected error during connection: " + str(er))
            raise

    def getValue(self):
        self.send("MEAS? AUTO,DEF,(@102:103)")
        return self.read()

    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.sock.close()

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