# by Carl Monk (@ForToffee)
# github.com/ForToffee/Robosapien
# based on work from http://playground.arduino.cc/Main/RoboSapienIR

import time
import pigpio # http://abyz.co.uk/rpi/pigpio/python.html

CODE_RSTurnRight       = 0x80
CODE_RSRightArmUp      = 0x81
CODE_RSRightArmOut     = 0x82
CODE_RSTiltBodyRight   = 0x83
CODE_RSRightArmDown    = 0x84
CODE_RSRightArmIn      = 0x85
CODE_RSWalkForward     = 0x86
CODE_RSWalkBackward    = 0x87
CODE_RSTurnLeft        = 0x88
CODE_RSLeftArmUp       = 0x89
CODE_RSLeftArmOut      = 0x8A
CODE_RSTiltBodyLeft    = 0x8B
CODE_RSLeftArmDown     = 0x8C
CODE_RSLeftArmIn       = 0x8D
CODE_RSStop            = 0x8E
CODE_RSWakeUp          = 0xB1
CODE_RSBurp            = 0xC2
CODE_RSRightHandStrike = 0xC0
CODE_RSNoOp            = 0xEF

CYCLE = 833

class Robo(object):

	def __init__(self, pin):
		self.pin = pin
		self.pi = pigpio.pi() # Connect to Pi.
		self.pi.set_mode(pin, pigpio.OUTPUT) # IR TX connected to this GPIO.
		self.pi.write(self.pin, 1)
		self.pi.wave_clear()
		
		self.wf_head = self.add_wave(8, 8)
		self.wf_hi = self.add_wave(4, 1)
		self.wf_lo = self.add_wave(1, 1)
		self.wf_tail = self.add_wave(8, 8)
		
		self.keep_alive = self.create_code(CODE_RSNoOp)


	def add_wave(self, hi, lo):
		self.pi.wave_add_generic([pigpio.pulse(1<<self.pin, 0, hi * CYCLE), pigpio.pulse(0, 1<<self.pin, lo * CYCLE)])
		return self.pi.wave_create()
	
	def create_code(self, code):
		data = code
		print data
		wave = []
		wave.append(self.wf_head)

		for x in range(8):
			if (data & 128 != 0):
				wave.append(self.wf_hi)
				print 1
			else:
				wave.append(self.wf_lo)
				print 0
			data <<= 1

		wave.append(self.wf_tail)
		print wave
		print "end"
		return wave

	def send_wave(self, wave):
		#print wave
		self.pi.wave_chain(wave)
		while self.pi.wave_tx_busy():
			time.sleep(0.002)

		self.pi.write(self.pin, 1)
	
	def send_code(self, code):
		self.send_wave(self.create_code(code))
		time.sleep(0.5)

	def clean_up(self):
		pi.wave_clear()
		pi.stop() # Disconnect from Pi.
	

Create robo.py