acquire_and_plot.py

"""
Use SixChannelReader.py to acquire some data from the Arduino
and make a simple plot using matplotlib

Author: James Keaveney
"""
import SixChannelReader as SCR
import matplotlib.pyplot as plt
import numpy as np
import time

def main():
	""" Get data logger data and plot it. Convert raw 12-bit ADC data to voltage """
	st = time.clock()
	
	SR = SCR.SerialDataLogger(92)
	t, C1, C2, C3, C4, C5, C6 = SR.get_data()
	#save data for later
	filename = 'TestData'
	SR.save_data(filename)	

	et = time.clock()
	print 'Total time elapsed: ', et-st
	
	
	# plot it
	fig = plt.figure("Test acquire plot",figsize=(6,11))
	plt.clf()
	
	ax1 = fig.add_subplot(611)
	ax2 = fig.add_subplot(612,sharex=ax1)
	ax3 = fig.add_subplot(613,sharex=ax1)
	ax4 = fig.add_subplot(614,sharex=ax1)
	ax5 = fig.add_subplot(615,sharex=ax1)
	ax6 = fig.add_subplot(616,sharex=ax1)
	
	t /= 1e3
	ax1.plot(t,C1*3.3/4095)
	ax2.plot(t,C2*3.3/4095)
	ax3.plot(t,C3*3.3/4095)
	ax4.plot(t,C4*3.3/4095)
	ax5.plot(t,C5*3.3/4095)
	ax6.plot(t,C6*3.3/4095)
	
	ax1.set_xlim(t[0],t[-1])
	
	for ax in [ax1,ax2,ax3,ax4,ax5,ax6]:
		ax.set_ylim(0,3.3)
		
	
	ax6.set_xlabel('Time (ms)')
	
	ax1.set_ylabel('A0 (V)')
	ax2.set_ylabel('A1 (V)')
	ax3.set_ylabel('A2 (V)')
	ax4.set_ylabel('A3 (V)')
	ax5.set_ylabel('A4 (V)')
	ax6.set_ylabel('A5 (V)')
	
	fig.tight_layout()
	
	fig.savefig('./LatestTestData.pdf')
	fig.savefig('./LatestTestData.png')

def test_reading():
	""" 
	routine to test stability/repeatability of read/sync process
	Loop over reading and make sure that the size of the arrays 
	returned are the same length
	"""
	SR = SCR.SerialDataLogger(92)
	for i in range(10):
		st = time.clock()
		
		t, C1, C2, C3, C4, C5, C6 = SR.get_data()
		#save data for later
		filename = 'TestData'
		SR.save_data(filename)	

		et = time.clock()
		print 'Total time elapsed: ', et-st
		print 'Length of data points on each axis: ', len(t)

if __name__ == '__main__':
	main()