TKilgore.py

# TEST version doesn't do GPIO
# Kilgore Trout

#-- Tools --
# Location
#/home/pi/trout
# Terminal
#H:\DROPBOX\Dropbox\DEV\RaspPi\putty 
#username: pi
#password: raspberry
# Transfer files (same login)
#D:\Program Files (x86)\FileZilla FTP Client\filezilla.exe
# Test audio
#aplay file.wav

#-- Hardware GPIO --
# 7 -> LED 
# 8,9 Mouth
# 10,11 Turn
# 12,13 Tail

# Power
# Need 3Amp power hooked up to micro USB + motors + Audio amp / speakers

import signal
import sys
import os
#import RPi.GPIO as GPIO
import time
import wave
import getopt
#import alsaaudio
import random


#-- Defines --

TAIL = 0
HEAD0 = 1
HEAD1 = 2
MOUTH = 3

#-- Data --
pins = [11, 15, 16, 18]

# array of fortune wav sounds

fortunes = ['03','05','06','07','10','12','14','16','17','18','19','21','22','26','27','30','32']

#-- Code --

#classes

class ProfileTimer:
	#
	function = time.clock

	# is providing a start time
	@staticmethod
	def value():
		return Timer.function()

	# calculating delta between current time and start time.
	@staticmethod
	def delta(start):
		return Timer.function() - start

		
class Timer:
	#
	function = time.time

	# is providing a start time
	@staticmethod
	def value():
		return Timer.function()

	# calculating delta between current time and start time.
	@staticmethod
	def delta(start):
		return Timer.function() - start		
		
#subroutines

#def setSampWidth(f):
	# 8bit is unsigned in wav files
#	if f.getsampwidth() == 1:
#		device.setformat(alsaaudio.PCM_FORMAT_U8)
	# Otherwise we assume signed data, little endian
#	elif f.getsampwidth() == 2:
#		device.setformat(alsaaudio.PCM_FORMAT_S16_LE)
#	elif f.getsampwidth() == 3:
#		device.setformat(alsaaudio.PCM_FORMAT_S24_LE)
#	elif f.getsampwidth() == 4:
#		device.setformat(alsaaudio.PCM_FORMAT_S32_LE)
#	else:
#		raise ValueError('Unsupported format')

		
def headUp():
	print "headUp"
	#GPIO.output(pins[HEAD0], True)
	#GPIO.output(pins[HEAD1], False)	
		
def headDown():
	print "headDown"
	#GPIO.output(pins[HEAD0], False)
	#GPIO.output(pins[HEAD1], True)

def tailStart():
	print "tailStart"
	#GPIO.output(pins[TAIL], True)
	
def tailStop():
	print "tailStop"
	#GPIO.output(pins[TAIL], False)

def mouthStart():
	print "mouthStart"
	#GPIO.output(pins[MOUTH], True)
	
def mouthStop():
	print "mouthStop"
	#GPIO.output(pins[MOUTH], False)
	
	
def headMove():
	a = 1
	while (a < 10000):
		headUp()

		s1 = Timer.value()
		while Timer.delta(s1) < 2.9:
			yield a
		
		headDown()

		s2 = Timer.value()
		while Timer.delta(s2) < 3.0:
			#print "foooo" + str(Timer.delta(s2))
			yield a
			
		yield a


def tailFlap():
	#print Timer.delta(start)
	a = 1
	while (a < 10000):
		tailStart()

		s1 = Timer.value()
		while Timer.delta(s1) < 0.4:
			yield a
		
		tailStop()

		s2 = Timer.value()
		while Timer.delta(s2) < 0.4:
			yield a
			
		yield a
		
def mouthFlap():
	#print Timer.delta(start)
	a = 1
	while (a < 10000):
		mouthStart()

		s1 = Timer.value()
		while Timer.delta(s1) < 0.2:
			yield a
		
		mouthStop()

		s2 = Timer.value()
		while Timer.delta(s2) < 0.3:
			yield a
			
		yield a
		
		
def resetPins():
	for n in pins:
		#GPIO.output(n, False)
		pass
		
		
def signal_handler(signal, frame):
	print 'You pressed Ctrl+C!'
	resetPins()
	sys.exit(0)

signal.signal(signal.SIGINT, signal_handler)

#

#more efficient way to get the size of array? ( I forgot how in Python )
maxfortune = 0
for file in fortunes:
	maxfortune = maxfortune + 1
	
# setup the pins as output 
for n in pins:
	#GPIO.setup(n, GPIO.OUT)
	#GPIO.output(n, False)
	pass
	
#device = alsaaudio.PCM()
#card = 'default'

while 1:

	print "START -------"
	t0 = Timer.value()
	while Timer.delta(t0) < 8.0:
		pass
		
	head = headMove()
	tail = tailFlap()
	mouth = mouthFlap()	
		  
	fortune_pick = random.randrange(0,maxfortune)
	fortuneFile = "Recording" + fortunes[fortune_pick]+".wav"
	print fortuneFile
	   
	#f = wave.open('wave/Recording32.wav', 'rb')

	# Set attributes
	#device.setchannels(f.getnchannels())
	#device.setrate(f.getframerate())
	#setSampWidth(f)
	#device.setperiodsize(320)   
	#data = f.readframes(320)

	# tail flap starts it
	t0 = Timer.value()
	while Timer.delta(t0) < 4.0:
		b = tail.next()

	# head forward still tail flapping
	headUp()
	t0 = Timer.value()
	while Timer.delta(t0) < 1.0:
		b = tail.next()

	# play audio with lip flap	
	#while data:
		# Read data from stdin
		#a = 1
		#device.write(data)
		#data = f.readframes(320)
		#a = head.next()
		#b = tail.next()
		#c = mouth.next()

	#temp version without audio	
	t0 = Timer.value()
	while Timer.delta(t0) < 2.0:
		a = head.next()
		b = tail.next()
		c = mouth.next()
		
		
	#stop mouth and put head down
	mouthStop()
	headDown()
		
	#but continue to flap tail for a few seconds	
	t0 = Timer.value()
	while Timer.delta(t0) < 2.0:
		b = tail.next()

	#stop tail	
	tailStop()
		
	#t0 = Timer.value()
	#while Timer.delta(t0) < 0.5:
	#	pass
		
	resetPins()

	#f.close()	

#Wait for button press

#Pick random audio sound from array

# load audio into memory 48K wave file

# Create Generator for Tail flap
# Create Generator for Turn control
# Create Generator for mouth lip sync - pass in audio raw data

# start audio playing

#  update tail flap
#  update turn control
#  update lip sync

# until audio finished

# return fish to neutral

# To top of program