#!/usr/bin/python

########################################################################
#
#	Matrix3:  test of Pi Matrix board
#
# 	This application will test out your Pi LED 8x8 Matrix board
# 	by selectively turning on and off each row, column, and pixel.
# 
# 	Use the concepts from these simple test routines
# 	to make more interesting displays.
#
# 	Author:  Bruce E. Hall
# 	Date  :  03 Feb 2013
#
########################################################################

import smbus			#gives us a connection to the I2C bus
import time				#for timing delays

#Definitions for the MCP23017 chip
ADDR   = 0x20			#The I2C address of our chip
DIRA   = 0x00			#PortA I/O direction, by pin. 0=output, 1=input
DIRB   = 0x01			#PortB I/O direction, by pin. 0=output, 1=input
PORTA  = 0x12			#Register address for PortA
PORTB  = 0x13			#Register address for PortB
OUTPUT = 0				
INPUT  = 1

#
#	Lower Level LED Display routines
#   Write data directly to the Pi Matrix board
#

def Init23017 ():
	#Set up the 23017 for 16 output pins
	bus.write_byte_data(ADDR,DIRA,0x00);  #all zeros = all outputs on PortA
	bus.write_byte_data(ADDR,DIRB,0x00);  #all zeros = all outputs on PortB

def TurnOffLEDS ():
	bus.write_byte_data(ADDR,PORTA,0x00)  #set all columns low
	bus.write_byte_data(ADDR,PORTB,0x00)  #set all rows low
	
def TurnOnLEDS ():
	bus.write_byte_data(ADDR,PORTA,0xFF)  #set all columns low
	bus.write_byte_data(ADDR,PORTB,0x00)  #set all rows low
	
def SetLED (row,col):
	#turn on an individual LED at (row,col).  All other LEDS off.
	bus.write_byte_data(ADDR,PORTA,0x80>>col)
	bus.write_byte_data(ADDR,PORTB,~(1<<row))
	
def SetColumn (col):
	#turn on all LEDs in the specified column. expects input of 0-7
	bus.write_byte_data(ADDR,PORTB,0x00)
	bus.write_byte_data(ADDR,PORTA,0x80>>col)
	
def SetRow (row):
	#turn on all LEDs in the specified row.  expects input of 0-7
	bus.write_byte_data(ADDR,PORTA,0xFF)
	bus.write_byte_data(ADDR,PORTB,~(1<<row))

	
#
#	Routines for testing our Pi Matrix board
#	Each one calls on one or more of the low-level routines above
#	

	
def Pause():
	#Turn off LEDS and wait a while between cases
	TurnOffLEDS();
	time.sleep(0.5);
	
def FlashLEDS (delay):
	#Flash all of the LEDS on/off for the specified time
	TurnOnLEDS()
	time.sleep(delay)
	TurnOffLEDS()
	time.sleep(delay)

def LEDtest (delay):
	#Turn on all 64 LEDS for the specified time delay, in seconds
	print "\nLighting all LEDS for %d seconds" % delay
	TurnOnLEDS()				#turn them all on
	time.sleep(delay)			#wait a while
	Pause()						#then turn them off
	
def FlashTest (numCycles,delay):
	print "\nFlash all of the LEDS"
	for count in range(0,numCycles):
		FlashLEDS(delay)
	Pause()
	
def ColumnTest (numCycles):
	#Turn on each Column.  
	#Keep PortB (rows) low & set each bit in PortA (columns) high
	#This will actually light LEDS in reverse order, col 7 to col0,
	#because Port A bit 0 is wired to Col7, A1 to Col6,..., A7 to Col0
	print "\nTurn on Columns 0 to 7"
	for count in range(0,numCycles):
		print "...cycle",count+1
		for col in range(0,8):
			SetColumn(col)
			time.sleep(0.5)
		time.sleep(1)
	Pause()
	
def RowTest (numCycles):
	#Turn on each row, from row 0 to row 7
	#Keep PortA (columns) high & selectively turn a bit in PortB (rows) low
	print "\nTurn on Rows 0 to 7"		
	for count in range(0,numCycles):
		print "...cycle",count+1
		for row in range(0,8):
			SetRow(row)
			time.sleep(0.5)
		time.sleep(1)
	Pause()
	
def PixelTest (numCycles):
	#Flash each LED according to its (row,col) coordinate
	print "\nFlash each LED in (row,col) order"
	for count in range(0,numCycles):
		for row in range(0,8):
			for col in range(0,8):
				SetLED(row,col)
				time.sleep(0.1)
	Pause()

	
#
#	Here is the program body: call each of the text routines in turn.
#	Global variable 'bus' is our connection to the I2C bus
#

print "\nPi Matrix test program starting"
bus = smbus.SMBus(1);           #Use '1' for newer Pi boards; 0 for oldies
Init23017()						#Set all 16 I/O pins as output
LEDtest(5)						#Turn on all LEDS to verify connectivity
FlashTest(25,0.1)				#Flash them all for a little fun
ColumnTest(3)					#Sequentially turn on Columns 0 to 7
RowTest(3)						#Sequentially turn on Rows 0 to 7
PixelTest(1)					#Flash each pixel in row,col order
FlashLEDS(0.1)					#Visual end of test
print "\nDone."