#!/usr/bin/python

import RPi.GPIO as GPIO
from time import sleep

GPIO.setmode(GPIO.BOARD)

# Define Pins
# These segments define the LED that will be on or off (High/Low)
segmentA = 16
segmentB = 18
segmentC = 38
segmentD = 36
segmentE = 15
segmentF = 12
segmentG = 22
segmentH = 40
# These digits define which of the seven segment displays will be used (in this case we are using a 6 digit multi-display)
digit1 = 7
digit2 = 11
digit3 = 13
digit4 = 29
digit5 = 31
digit6 = 33

# Set up Pins
GPIO.setup(segmentA,GPIO.OUT)
GPIO.setup(segmentB,GPIO.OUT)
GPIO.setup(segmentC,GPIO.OUT)
GPIO.setup(segmentD,GPIO.OUT)
GPIO.setup(segmentE,GPIO.OUT)
GPIO.setup(segmentF,GPIO.OUT)
GPIO.setup(segmentG,GPIO.OUT)
GPIO.setup(segmentH,GPIO.OUT)
GPIO.setup(digit1,GPIO.OUT)
GPIO.setup(digit2,GPIO.OUT)
GPIO.setup(digit3,GPIO.OUT)
GPIO.setup(digit4,GPIO.OUT)
GPIO.setup(digit5,GPIO.OUT)
GPIO.setup(digit6,GPIO.OUT)

# Define functions

# Turn on an individual LED Segment
# Called by display()
# Param:  	segmentList - A list of individual LED segments to turn on
# 			digitList - A list of displays in the range 1-6 to display the value on
def turnOn(segmentList, digitList) :
	# Provide common Annode to the display chosen (1 through 6)
	for x in digitList :
		whichDigit = 'digit' + str(x)
		GPIO.output(eval(whichDigit),GPIO.LOW)
	# Light up the individual segments that will produce the value passed
	for x in segmentList :
		whichSegment = 'segment' + x
		GPIO.output(eval(whichSegment),GPIO.HIGH)
	return

# Turns off the individual LED Segments
# Called by display()
# Param:  	segmentList - A list of individual LED segments to turn off
# 			digitList - A list of displays in the range 1-6 to turn off
def turnOff(segmentList, digitList) :
	# Turn off the display by bringing the common Annode to high
	for x in digitList :
		whichDigit = 'digit' + str(x)
		GPIO.output(eval(whichDigit),GPIO.HIGH)
	# Turn off the individual LED segment by bringing the LED to low
	for x in segmentList :
		whichSegment = 'segment' + x
		GPIO.output(eval(whichSegment),GPIO.LOW)
	return

# Calls turnOff() to reset the display, then turnOn() to light up the correct data
# Called by multiplex() to turn on an particular value for one or multiple of the displays
# Param:  	whichNumber - The value to be displayed
# 			whichDigit - A list of segments in the range 1-6 to display the value
def display(whichNumber,whichDigit) :
	# Turn off all segments on all digits (Start from a clean slate)
	turnOff(['A','B','C','D','E','F','G','H'],[1,2,3,4,5,6])
	# Turn on the appropriate segments for the value passed (whichNumber) on each of the digits defined in whichDigit
	if (whichNumber == 0) :
		turnOn(['A','B','C','D','E','F'], whichDigit)
	elif (whichNumber == 1) :
		turnOn(['B','C'], whichDigit)
	elif (whichNumber == 2) :
		turnOn(['A','B','D','E','G'], whichDigit)
	elif (whichNumber == 3) :
		turnOn(['A','B','C','D','G'], whichDigit)
	elif (whichNumber == 4) :
		turnOn(['B','C','F','G'], whichDigit)
	elif (whichNumber == 5) :
		turnOn(['A','C','D','F','G'], whichDigit)
	elif (whichNumber == 6) :
		turnOn(['A','C','D','E','F','G'], whichDigit)
	elif (whichNumber == 7) :
		turnOn(['A','B','C'], whichDigit)
	elif (whichNumber == 8) :
		turnOn(['A','B','C','D','E','F','G'], whichDigit)
	elif (whichNumber == 9) :
		turnOn(['A','B','C','D','F','G'], whichDigit)
	elif (whichNumber == 10) :
		turnOn(['H'], whichDigit)
	return

# Calls display() to turn on a sequence of values.  This is a wrapper function to loop through multiple different values
# Param: 	whichValue - A list of (list of (value and (list of digits)))
def multiplex(whichValue) :
	# 500 times through the loop gives a long enough amount of time to read the display
	for x in range(1,500) :
		# Loop through the various values to be displayed
		for value in whichValue :
			# value[0] is the actual value to be displayed and value[1] is the list of digits to display the value on
			display(value[0],value[1])
			# We need a fast refresh so that the change/swap is not visible to the human eye
			sleep(0.001)
	return


# Run the sucker
# Dots are represented by the number 10
multiplex([
	[1,[1]],
	[9,[2]],
	[2,[3]],
	[10,[3]],
	[1,[4]],
	[6,[5]],
	[8,[6]],
	[10,[6]]
	])


GPIO.cleanup()