from abstract_instrument import abstract_instrument
  import os
  
  #==============================================================================
  
  ALL_VAL_TYPE = "PWR"
  
  #==============================================================================
  
  class PM100D(abstract_instrument):

  	def __init__(self, adress="/dev/usbtmc0", vtype="PWR"):
  		self.adress = adress
  		self.vtype = vtype

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

  	def connect(self):
  		try:
  			print('Connecting to device @%s...' %(self.adress))
  			self.FILE = os.open(self.adress, os.O_RDWR)
  			print('  --> Ok')

  			print(self.model())

  			if self.vtype == "PWR":
  				1
  			else:
  				print("Wrong -v argument")
  				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):
  		return os.read(self.FILE, 300)

  	def disconnect(self):
  		self.send('*RST')

  	def send(self, command):
  		os.write(self.FILE, command)