#!/us'>r/bin/python
# Copyright Nigel Brooke 12/23/2013 
import os
import time
import threading
import smbus


Sensors = [ 'LM75 Raspberry Pi', 'Thermocouple Raspberry pi','EOF']


Period = 1
Displaylines = 10
Lines = 0

def Readsensor():
	Sensortype ={ 0:ReadLM75,
			    1:ReadThermocouple,
				2:ReadUSBLM75,
				3:ReadADC,
			     }
	Sensortype[Selection]()
	

def ReadLM75():
	#Note this is set up for a Rev A raspberry pi the SMBus address will need to be changed for a Rev B Device
	global Data
	global Units
	Units = "Deg C"
	bus = smbus.SMBus(0)
	busdata = bus.read_i2c_block_data(0x48, 0)
	TempMSB = busdata[0]
	TempLSB = busdata[1]
	Data = (((TempMSB << 8) | TempLSB) >>7) * 0.5

def ReadThermocouple():
	global Data
	Data = 10
	
def ReadUSBLM75():
	global Data
	Data = 15
	
def ReadADC():
	global Data
	Data = 20

def cls():
	os.system('clear')


def Listsensors():
	
	global Selection
	i=0
	print "************"
	print "**PiLogger**"
	print "************"
	print ""
			
	
	while(Sensors[i] != 'EOF'):
		print i,' ',Sensors[i]
		i=i+1
	done = True
	while (done):
		print ""
		Selection = raw_input("Please select sensor by typing number:- ")		
		try:
			Selection = int(Selection)
			if (0<= Selection <=i-1):
				print "Sensor selected:- ",Sensors[Selection]
				done = False						
			else:
				print "Enter valid value"				
		except:
			continue
	
				
def Tick(Secs):
		Starttime = time.time()
		while((time.time()-Starttime)<Secs):
			i=0
			
def Writetofile(SampleNum,Data,Unit):
		global Lines
		global Displaylines
		if (SampleNum == 0) :
			cls()
			print "************"
			print "**PiLogger**"
			print "************"
			if (logtofile == True):
				print ""
				print "Loging to File ", filename
			print ""
			print "Press Ctrl-C to exit"
			print ""
			print "Sample Number    Time        Reading   Units"
			if (logtofile == True):
				fo = open(filename,"a")
				fo.write("Sample Number,Time,Reading,Units\n")
				fo.close
				
		now = time.localtime(time.time())
		print "     ",SampleNum,"        ",time.strftime("%T",now),"  ",Data,"    ",Unit
		Lines = Lines + 1
		if (Lines == Displaylines):
			cls()
			print "************"
			print "**PiLogger**"
			print "************"
			if (logtofile == True):
				print ""
				print "Loging to File ", filename
			print ""
			print "Press Ctrl-C to exit"
			print ""
			print "Sample Number    Time        Reading   Units"
			Lines = 0
			
		if (logtofile == True):
				fo = open(filename,"a")
				SampleNum = str(SampleNum)
				fo.write(SampleNum)
				fo.write(",")
				fo.write(time.strftime("%T",now))
				fo.write(",")
				Data = str(Data)
				fo.write(Data)
				fo.write(",")
				fo.write(Unit)
				Unit = str(Unit)
				fo.write("\n")
				fo.close
				
def fileinfo():
	cls()
	global filename
	global logtofile
	logtofile = False
	log =raw_input("Do you want to log to file ?(Y/N)")
	if (log.upper() == 'Y'):
		filename =raw_input("Please enter filename: ")
		logtofile = True
		time.sleep(1)
	else:
		print "Log to file not selected"
		time.sleep(1)
			
def LogInterval():
	done = False
	global Period
	cls()
	print "************"
	print "**PiLogger**"
	print "************"
	print ""
			
	while not (done):
		Period = raw_input('Enter logging interval in seconds:- ')
		try:
			Period = int(Period)
			if (Period < 1):
				print "Enter an integer number higher than 1 "
			else:
				done = True
			
		except:
			print "Please enter a number"
			continue
			
				
count = 0
cls()
Listsensors()
LogInterval()
fileinfo()
try:
	while True :
		SampleNum = count
		Readsensor()
		Writetofile(SampleNum,Data,Units)
		count = count + 1
		Tick(Period)
except KeyboardInterrupt:
	pass