#!/usr/bin/env python
  
  # -*- coding: utf-8 -*-
  
  import argparse, time, os, instruments, inspect
  
  #==============================================================================
  
  # Path
  PATH = os.getcwd()
  # Sampling time acquisition
  SAMPLING_TIME = 1
  # File duration
  FILE_DURATION = 3600*24
  # Default instrument
  INSTRUMENT = None
  # Default instrument adress
  ADRESS = None
  #Default val type
  VAL_TYPE = None
  
  #==============================================================================
  
  def parse():

  	"""
  	Specific parsing procedure for transfering data from any abstract instrument.
  	:returns: populated namespace (parser)
  	"""
  
  	parser = argparse.ArgumentParser(description = 'Acquire data from INSTRUMENT',
  									 epilog = 'Example: \'./datalogger.py -i myInstrument -st 10\' logs myInstrument every 10 seconds to output file YYYYMMDD-HHMMSS-INSTRUMENT.dat')
  
  	parser.add_argument('-l',
  						action='store_true',
  						dest='list',
  						default=False,
  						help='List all available instruments')
  
  	parser.add_argument('-I',
  						action='store',
  						dest='instLog',
  						default=INSTRUMENT,
  						help='Instrument to log (default '+str(INSTRUMENT)+')')
  
  	parser.add_argument('-ip',
  						action='store',
  						dest='adress',
  						default=ADRESS,
  						help='Adress of instrument (IP, USB...) (default '+str(ADRESS)+')')
  
  	parser.add_argument('-v',
  						action='store',
  						dest='vtype',
  						default=VAL_TYPE,
  						help='Value type to measure (default '+str(VAL_TYPE)+')')
  
  	parser.add_argument('-st',
  						action='store',
  						dest='samplingtime',
  						default=SAMPLING_TIME,
  						help='Sampling time acquistion (default '+str(SAMPLING_TIME)+' second)')
  
  	parser.add_argument('-fd',
  						action='store',
  						dest='fileduration',
  						default=FILE_DURATION,
  						help='File duration (infinite : \'-fd -1\') (default '+str(FILE_DURATION)+' second)')
  
  	parser.add_argument('-p',
  						action='store',
  						dest='path',
  						default=PATH,
  						help='Absolute path (default '+PATH+')')
  
  	args = parser.parse_args()
  	return args

  
  #==============================================================================
  
  def acq_routine(instrument, path, samplingtime, fileduration):

  	instrument.connect()
  	t0 = time.time()
  	filename = time.strftime("%Y%m%d-%H%M%S", time.gmtime(t0)) + '-' + instrument.model() + '.dat'
  	print('Opening %s' %filename)
  	try:
  		year = time.strftime("%Y", time.gmtime(t0))
  		month = time.strftime("%Y-%m", time.gmtime(t0))
  		os.chdir(path + '/' + year + '/' + month)
  	except:
  		try:
  			os.chdir(path + '/' + year)
  			os.mkdir(month)
  			os.chdir(path + '/' + year + '/' + month)
  		except:
  			os.chdir(path)
  			os.mkdir(year)
  			os.chdir(path + '/' + year)
  			os.mkdir(month)
  			os.chdir(path + '/' + year + '/' + month)
  
  	data_file = open(filename, 'wr', 0)
  
  	# Infinite loop
  	while True:
  		# tic
  		tic = time.time()
  
  		if time.time() - t0 >= fileduration:
  			t0 = time.time()
  			print('Closing %s
  ' %filename)
  			data_file.close()
  
  			try:
  				year = time.strftime("%Y", time.gmtime(t0))
  				month = time.strftime("%Y-%m", time.gmtime(t0))
  				os.chdir(path + '/' + year + '/' + month)
  			except:
  				try:
  					os.chdir(path + '/' + year)
  					os.mkdir(month)
  					os.chdir(path + '/' + year + '/' + month)
  				except:
  					os.chdir(path)
  					os.mkdir(year)
  					os.chdir(path + '/' + year)
  					os.mkdir(month)
  					os.chdir(path + '/' + year + '/' + month)
  
  			filename = time.strftime("%Y%m%d-%H%M%S", time.gmtime(t0)) + '-' + instrument.model() + '.dat'
  			print('Opening %s
  ' %filename)
  			data_file = open(filename, 'wr', 0)
  
  		try:
  			try:
  				#epoch time
  				epoch = time.time()
  				#MJD time
  				mjd = epoch / 86400.0 + 40587
  				# Meas values
  				meas = instrument.getValue()
  				meas = meas.replace(",", "\t")
  				meas = meas.replace(";", "\t")
  				meas = meas.replace("+", "")
  
  				string = "%f\t%f\t%s" % (epoch, mjd, meas)
  				data_file.write(string) # Write in a file
  				print(string)
  
  				# Sleep until sampletime
  				time.sleep(samplingtime - (time.time() - tic))
  
  			except Exception as ex:
  				print 'Exception during controler data reading: ' + str(ex)
  
  		except KeyboardInterrupt:
  			print '
    --> Disconnected'
  			instrument.disconnect()
  			data_file.close()
  
  			# To stop the loop in a clean way
  			break

  
  #==============================================================================
  
  def main():

  	"""
  	Main script
  	"""
  	# Parse command line
  	args = parse()
  	# path
  	path = args.path
  	# Sampling time
  	samplingtime=float(args.samplingtime)
  	# File duration
  	fileduration=int(args.fileduration)
  	# Instrument to log
  	instLog = args.instLog
  	# instrument adress
  	adress = args.adress
  	# val type
  	vtype = args.vtype
  
  	try:
  		if args.list:
  			print('
  Instruments:')
  			for name, obj in inspect.getmembers(instruments):
  				if inspect.ismodule(obj) and name.startswith('__') == False and name.startswith('abstract') == False:
  					print('
  ' + name)
  					exec('print("\t" + instruments.%s.ALL_VAL_TYPE)'%name)
  
  		else:
  			if instLog == None:
  				raise Exception('No instrument selected !')
  
  			if adress == None and vtype == None:
  				exec('myInstrument = instruments.%s.%s()'%(instLog, instLog))
  			elif adress == None and vtype != None:
  				exec('myInstrument = instruments.%s.%s(vtype="%s")'%(instLog, instLog, vtype))
  			elif adress != None and vtype != None:
  				exec('myInstrument = instruments.%s.%s(adress="%s", vtype="%s")'%(instLog, instLog, adress, vtype))
  			acq_routine(myInstrument, path, samplingtime, fileduration)
  
  	except Exception as ex:
  			print 'Oops: '+str(ex)

  
  #==============================================================================
  
  if __name__ == "__main__":

  	main()