from abstract_instrument import abstract_instrument
  from labjack import ljm
  import numpy
  
  #==============================================================================

  ALL_VAL_TYPE = ['RES', 'TEMP PT100 K', 'TEMP PT100 C', 'TEMP AIR K']
  ALL_CHANNELS = ['1', '2', '3', '4', 'TEMP AIR K']

  ADDRESS = "192.168.0.25"

  CONF_CHANNELS = [["AIN0", "AIN10"], ["AIN2", "AIN11"], ["AIN4", "AIN12"], ["AIN6", "AIN13"], ["TEMPERATURE_AIR_K"]]

  VISHAY_CHANNELS = [1000., 1000., 1079., 10000.]

  
  #==============================================================================
  
  class T7Pro(abstract_instrument):

  	def __init__(self, channels, vtypes, address):
  		self.address = address
  		self.channels = channels
  		self.vtypes = vtypes

  	def model(self):
  		return 'T7Pro'

  	def connect(self):
  		print('Connecting to device @%s...' %(self.address))
  		self.handle = ljm.openS("T7", "ETHERNET", self.address)
  		print('  --> Ok')
  		print(self.model())
  		self.configure()

  	def configure(self):
  		names = ["AIN0_NEGATIVE_CH", "AIN0_RANGE", "AIN0_RESOLUTION_INDEX",
  				"AIN1_NEGATIVE_CH", "AIN1_RANGE", "AIN1_RESOLUTION_INDEX",
  				"AIN2_NEGATIVE_CH", "AIN2_RANGE", "AIN2_RESOLUTION_INDEX",
  				"AIN3_NEGATIVE_CH", "AIN3_RANGE", "AIN3_RESOLUTION_INDEX",
  				"AIN4_NEGATIVE_CH", "AIN4_RANGE", "AIN4_RESOLUTION_INDEX",
  				"AIN5_NEGATIVE_CH", "AIN5_RANGE", "AIN5_RESOLUTION_INDEX",
  				"AIN6_NEGATIVE_CH", "AIN6_RANGE", "AIN6_RESOLUTION_INDEX",
  				"AIN7_NEGATIVE_CH", "AIN7_RANGE", "AIN7_RESOLUTION_INDEX",
  				#"AIN8_NEGATIVE_CH", "AIN8_RANGE", "AIN8_RESOLUTION_INDEX",
  				#"AIN9_NEGATIVE_CH", "AIN9_RANGE", "AIN9_RESOLUTION_INDEX",
  				"AIN10_NEGATIVE_CH", "AIN10_RANGE", "AIN10_RESOLUTION_INDEX",
  				"AIN11_NEGATIVE_CH", "AIN11_RANGE", "AIN11_RESOLUTION_INDEX",
  				"AIN12_NEGATIVE_CH", "AIN12_RANGE", "AIN12_RESOLUTION_INDEX",
  				"AIN13_NEGATIVE_CH", "AIN13_RANGE", "AIN13_RESOLUTION_INDEX"
  				]
  		l_names = len(names)
  		aValues = [1, 1, 12,#0
  				199, 1, 12,#1
  				3, 1, 12,#2
  				199, 1, 12,#3
  				5, 1, 12,#4
  				199, 1, 12,#5
  				7, 1, 12,#6
  				199, 1, 12,#7
  				#199, 1, 12,#8
  				#199, 1, 12,#9
  				199, 1, 12,#10
  				199, 1, 12,#11
  				199, 1, 12,#12
  				199, 1, 12#13
  				]

  		ljm.eWriteNames(self.handle, l_names, names, aValues)

  	def getValue(self):
  		strMes = ''
  		for ch in self.channels:
  			if self.vtypes[self.channels.index(ch)] == 'RES':
  				raw = self.read(CONF_CHANNELS[ALL_CHANNELS.index(ch)])
  				strMes = strMes + str(VISHAY_CHANNELS[ALL_CHANNELS.index(ch)]*raw[0]/raw[1]) + ';'
  			elif self.vtypes[self.channels.index(ch)] == 'TEMP PT100 K':
  				raw = self.read(CONF_CHANNELS[ALL_CHANNELS.index(ch)])
  				strMes = strMes + str(((VISHAY_CHANNELS[ALL_CHANNELS.index(ch)]*raw[0]/raw[1])/100.-1)/0.003850+273.15) + ';'
  			elif self.vtypes[self.channels.index(ch)] == 'TEMP PT100 C':
  				raw = self.read(CONF_CHANNELS[ALL_CHANNELS.index(ch)])
  				strMes = strMes + str(((VISHAY_CHANNELS[ALL_CHANNELS.index(ch)]*raw[0]/raw[1])/100.-1)/0.003850) + ';'
  			elif self.vtypes[self.channels.index(ch)] == 'TEMP AIR K':
  				raw = self.read(CONF_CHANNELS[ALL_CHANNELS.index(ch)])
  				strMes = strMes + str(raw[0]) + ';'

  		strMes = strMes[0:-1] + '
  '
  		return(strMes)

  	def read(self, names):
  		return ljm.eReadNames(self.handle, len(names), names)

  	def disconnect(self):
  		ljm.close(self.handle)

  	def send(self, command):
  		pass