AG34461A.py 2.1 KB
from abstract_instrument import abstract_instrument
import socket

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

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

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

class AG34461A(abstract_instrument):
	def __init__(self, adress="192.168.0.61", 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(10.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")
			elif self.vtype == "ACV":
				self.send("CONF:VOLT:AC")
			elif self.vtype == "DCI":
				self.send("CONF:CURR:DC")
			elif self.vtype == "ACI":
				self.send("CONF:CURR:AC")
			elif self.vtype == "RES2W":
				self.send("CONF:RES")
			elif self.vtype == "RES4W":
				self.send("CONF:FRES")
			elif self.vtype == "FREQ":
				self.send("CONF:FREQ")
			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("READ?")
		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)