zpc_ct/lib/StatesProg.py

import random
import time
import pygame
import sys
import pyaudio
import wave
import glob
import os
from itertools import cycle
from datetime import datetime
import ast
import random
from pathlib import Path
#====================================     

State = type("State", (object,), {})
#====================================     
class State(object):
	def __init__(self, FSM):
		self.FSM = FSM
		self.timer = 0
		self.startTime = 0
	def Enter(self):
		self.timer = 0
		self.startTime = 0
	def Execute(self):
		print('Executing')
	def Exit(self):
		print('Exiting')

#==================================== 

def play_seq(o, message):
	_id = 0
	o.clear_notes_on()
	for s in o.soundSlots:
		if s.notes[message[0]][message[1]][0] == 1:
			o.notes_on[_id] = 1
			o.soundSlots[_id].play(s.notes[message[0]][message[1]][1])
		_id += 1
	#print(o.notes_on)
def draw_header(o):
	o.draw.rectangle((0, 0, o.width, o.height), outline=0, fill=o.blue)
	o.center_text(o.header_text, o.h1, o.width, 25, o.light_grey)
	o.center_text("----------------------", o.h2, o.width, 55, o.light_grey)
			
def text_box1(o, a, b):
	w1 = 0
	w2 = (o.width / 2) - 50
	h1 = 45
	h2 = h1 + 10
	o.center_block(a, o.h2, [w1,h1,w2,h2], o.light_grey)
	o.center_block(b, o.h2, [w1,h1+10,w2,h2+25], o.light_grey)

def text_box2(o, a, b):
	w1 = 0
	w2 = (o.width / 2) - 50
	h1 = 90
	h2 = h1 + 10
	o.center_block(a, o.h2, [w1,h1,w2,h2], o.light_grey)
	o.center_block(b, o.h2, [w1,h1+10,w2,h2+25], o.light_grey)

def text_box3(o, a, b):
	w1 = (o.width / 2) + 40
	w2 = o.width
	h1 = 45
	h2 = h1 + 10
	o.center_block(a, o.h2, [w1,h1,w2,h2], o.light_grey)
	o.center_block(b, o.h2, [w1,h1+10,w2,h2+25], o.light_grey)

def text_box4(o, a, b):
	w1 = (o.width / 2) + 40
	w2 = o.width
	h1 = 90
	h2 = h1 + 10
	o.center_block(a, o.h2, [w1,h1,w2,h2], o.light_grey)
	o.center_block(b, o.h2, [w1,h1+10,w2,h2+25], o.light_grey)

def text_center(o, a, b):
	w1 = 0
	w2 = o.width
	h1 = 45
	h2 = h1 + 10
	o.draw.rectangle((0, 0, o.width, o.height), outline=0, fill=o.blue)
	o.center_block(a, o.h1, [w1,h1,w2,h2], o.light_grey)
	o.center_block(b, o.h1, [w1,h1+20,w2,h2+45], o.light_grey)


def draw_menu1(o):
	bpm_inc = 4
	x_size = o.width / 4
	y_size = o.height / 4
	og_x = 0
	o_x = og_x
	o_y = o.height
	text_padding = 6
	_id = 0
	while _id < 16:
	   
		o.draw.rectangle((o_x, o_y, o_x + x_size, o_y - y_size), outline=0, fill=o.blue)

		if _id == 0:
			o.center_block("Main", o.h2, [o_x, o_y, o_x + x_size, o_y - y_size], o.light_grey)
		if _id == 1:
			o.center_block("Note", o.h2, [o_x, o_y, o_x + x_size, o_y - y_size], o.light_grey)
		if _id == 2:
			o.center_block("Patt", o.h2, [o_x, o_y, o_x + x_size, o_y - y_size], o.light_grey)
		if _id == 3:
			o.center_block("Song", o.h2, [o_x, o_y, o_x + x_size, o_y - y_size], o.light_grey)
		if _id == 4:
			o.center_block("File", o.h2, [o_x, o_y, o_x + x_size, o_y - y_size], o.light_grey)
		if _id == 5:
			o.center_block("Util", o.h2, [o_x, o_y, o_x + x_size, o_y - y_size], o.light_grey)
		if _id == 6:
			o.center_block("Bank-", o.h2, [o_x, o_y, o_x + x_size, o_y - y_size], o.light_grey)
		if _id == 7:
			o.center_block("Bank+", o.h2, [o_x, o_y, o_x + x_size, o_y - y_size], o.light_grey)
		if _id == 8:
			o.center_block("Vol-", o.h2, [o_x, o_y, o_x + x_size, o_y - y_size], o.light_grey)
		if _id == 9:
			o.center_block("Vol+", o.h2, [o_x, o_y, o_x + x_size, o_y - y_size], o.light_grey)
		if _id == 11:
			o.center_block("", o.h2, [o_x, o_y, o_x + x_size, o_y - y_size], o.light_grey)
		if _id == 12:
			o.center_block("Play", o.h2, [o_x, o_y, o_x + x_size, o_y - y_size], o.light_grey)
		if _id == 13:
			o.center_block("Stop", o.h2, [o_x, o_y, o_x + x_size, o_y - y_size], o.light_grey)
		if _id == 14:
			o.center_block("BPM-", o.h2, [o_x, o_y, o_x + x_size, o_y - y_size], o.light_grey)
		if _id == 15:
			o.center_block("BPM+", o.h2, [o_x, o_y, o_x + x_size, o_y - y_size], o.light_grey)
			

		o_x = o_x + x_size
		_id += 1
		if _id % 4 == 0:
			o_y -= y_size
			o_x = og_x


def draw_menu2_sample(o):
	bpm_inc = 4
	x_size = o.width / 4
	y_size = o.height / 4
	og_x = 0
	o_x = og_x
	o_y = o.height
	text_padding = 6
	_id = 0
	while _id < 16:
	   
		o.draw.rectangle((o_x, o_y, o_x + x_size, o_y - y_size), outline=0, fill=o.dark_grey)

		
		if _id == 0:
			o.center_block("Copy", o.h2, [o_x, o_y, o_x + x_size, o_y - y_size], o.light_grey)
		if _id == 1:
			o.center_block("Paste", o.h2, [o_x, o_y, o_x + x_size, o_y - y_size], o.light_grey)
		if _id == 2:
			o.center_block("Del", o.h2, [o_x, o_y, o_x + x_size, o_y - y_size], o.light_grey)
		if _id == 3:
			o.center_block("Load", o.h2, [o_x, o_y, o_x + x_size, o_y - y_size], o.light_grey)

		if _id == 4:
			o.center_block("Vol-", o.h2, [o_x, o_y, o_x + x_size, o_y - y_size], o.light_grey)
		if _id == 5:
			o.center_block("Vol+", o.h2, [o_x, o_y, o_x + x_size, o_y - y_size], o.light_grey)
		if _id == 6:
			o.center_block("Pitch-", o.h2, [o_x, o_y, o_x + x_size, o_y - y_size], o.light_grey)
		if _id == 7:
			o.center_block("Pitch+", o.h2, [o_x, o_y, o_x + x_size, o_y - y_size], o.light_grey)
		o_x = o_x + x_size
		_id += 1
		if _id % 4 == 0:
			o_y -= y_size
			o_x = og_x

def draw_menu2_pattern(o):
	bpm_inc = 4
	x_size = o.width / 4
	y_size = o.height / 4
	og_x = 0
	o_x = og_x
	o_y = o.height
	text_padding = 6
	_id = 0
	while _id < 16:
	   
		o.draw.rectangle((o_x, o_y, o_x + x_size, o_y - y_size), outline=0, fill=o.dark_grey)

		if _id == 0:
			o.center_block("Copy", o.h2, [o_x, o_y, o_x + x_size, o_y - y_size], o.light_grey)
		if _id == 1:
			o.center_block("Paste", o.h2, [o_x, o_y, o_x + x_size, o_y - y_size], o.light_grey)
		if _id == 2:
			o.center_block("Clear", o.h2, [o_x, o_y, o_x + x_size, o_y - y_size], o.light_grey)
		if _id == 6:
			o.center_block("Bank -", o.h2, [o_x, o_y, o_x + x_size, o_y - y_size], o.light_grey)
		if _id == 7:
			o.center_block("Bank +", o.h2, [o_x, o_y, o_x + x_size, o_y - y_size], o.light_grey)

		o_x = o_x + x_size
		_id += 1
		if _id % 4 == 0:
			o_y -= y_size
			o_x = og_x

def draw_menu2_note(o):
	bpm_inc = 4
	x_size = o.width / 4
	y_size = o.height / 4
	og_x = 0
	o_x = og_x
	o_y = o.height
	text_padding = 6
	_id = 0
	while _id < 16:
	   
		o.draw.rectangle((o_x, o_y, o_x + x_size, o_y - y_size), outline=0, fill=o.dark_grey)

		if _id == 0:
			o.center_block("Vol -", o.h2, [o_x, o_y, o_x + x_size, o_y - y_size], o.light_grey)
		if _id == 1:
			o.center_block("Vol +", o.h2, [o_x, o_y, o_x + x_size, o_y - y_size], o.light_grey)
		
		o_x = o_x + x_size
		_id += 1
		if _id % 4 == 0:
			o_y -= y_size
			o_x = og_x

def menu1_actions(self, o):
	if o.keyState[16] == 2:
		if o.keyState[0] == 1:
			self.FSM.ToTransition('toMain')
		elif o.keyState[1] == 1:
			self.FSM.ToTransition('toEditSoundSequence')
		elif o.keyState[2] == 1:
			self.FSM.ToTransition('toSelectPattern')
		elif o.keyState[3] == 1:
			self.FSM.ToTransition('toEditSong')
		elif o.keyState[4] == 1:
			self.FSM.ToTransition('toFile')
		elif o.keyState[5] == 1:
			self.FSM.ToTransition('toUtil')
		if o.keyState[12] == 1:
			o.start_playback()
		if o.keyState[13] == 1:
			o.stop_playback()
		if o.keyState[6] == 1:
			if o.note_bank > 0:
				o.note_bank -= 1
		if o.keyState[7] == 1:
			if o.note_bank < 4:
				o.note_bank += 1
		if o.keyState[8] == 1:
			if o.volume > 0:
				o.volume -= 1
				text_center(o, "Master Volume", str(o.volume))
				o.update_display(0)
		if o.keyState[9] == 1:
			if o.volume < 16:
				o.volume += 1
				text_center(o, "Master Volume", str(o.volume))
				o.update_display(0)
		if o.keyState[11] == 1:
			#o.save_song()
			pass
		if o.keyState[14] == 1:
			if o.sconf.as_int('bpm') > 60:
				o.sconf['bpm'] = o.sconf.as_int('bpm') - o.bpm_inc
				o.update_bpm()
				text_center(o, "Master BPM", str(o.sconf['bpm']))
				o.update_display(0)
		if o.keyState[15] == 1:
			if o.sconf.as_int('bpm') < 240:
				o.sconf['bpm'] = o.sconf.as_int('bpm') + o.bpm_inc
				o.update_bpm()
				text_center(o, "Master BPM", str(o.sconf['bpm']))
				o.update_display(0)


def menu2_actions_sample(self, o):
	if o.keyState[17] == 2:
		if o.keyState[0] == 1:
			if o.soundSlots[o.eSound].volume > 0:
				print('copy')
				o.soundClipboard = [o.soundSlots[o.eSound].file, o.soundSlots[o.eSound].volume, o.soundSlots[o.eSound].pitch]
				print(o.soundClipboard)
				#text_center(o, "Sample Volume", str(o.soundSlots[o.eSound].volume))
				#o.update_display(0)
		elif o.keyState[1] == 1:
			
			print('pasting ', o.soundClipboard)
			
			o.soundSlots[o.eSound].file = o.soundClipboard[0]
			o.soundSlots[o.eSound].volume = o.soundClipboard[1]
			o.soundSlots[o.eSound].pitch = o.soundClipboard[2]
			o.soundSlots[o.eSound].create_sound()
			o.soundSlots[o.eSound].set_pitch()
			
			#s = o.SoundSlot(o.soundClipboard[0], o.eSound, o)
			#s.volume = o.soundClipboard[1]
			#s.pitch = o.soundClipboard[2]
			#s.notes = s.init_notes()
			#o.soundSlots[o.eSound] = s
			
		elif o.keyState[2] == 1:
			if o.soundSlots[o.eSound].pitch > -10:
				o.soundSlots[o.eSound].pitch -= 1
				text_center(o, "Sample Pitch", str(o.soundSlots[o.eSound].pitch))
				o.soundSlots[o.eSound].set_pitch()
				o.update_display(0)
		elif o.keyState[3] == 1:
			print('load')
			o.FSM.FSM.ToTransition("toLoadSound")

		elif o.keyState[4] == 1:
			if o.soundSlots[o.eSound].volume > 0:
				o.soundSlots[o.eSound].volume -= 1
				text_center(o, "Sample Volume", str(o.soundSlots[o.eSound].volume))
				o.update_display(0)
		elif o.keyState[5] == 1:
			if o.soundSlots[o.eSound].volume < 16:
				o.soundSlots[o.eSound].volume += 1
				text_center(o, "Sample Volume", str(o.soundSlots[o.eSound].volume))
				o.update_display(0)
		elif o.keyState[6] == 1:
			if o.soundSlots[o.eSound].pitch > -10:
				o.soundSlots[o.eSound].pitch -= 1
				text_center(o, "Sample Pitch", str(o.soundSlots[o.eSound].pitch))
				o.soundSlots[o.eSound].set_pitch()
				o.update_display(0)
		elif o.keyState[7] == 1:
			if o.soundSlots[o.eSound].pitch < 10:
				o.soundSlots[o.eSound].pitch += 1
				text_center(o, "Sample Pitch", str(o.soundSlots[o.eSound].pitch))
				o.soundSlots[o.eSound].set_pitch()
				o.update_display(0)

def menu2_actions_pattern(self, o):
	if o.keyState[17] == 2:
		if o.keyState[0] == 1:
			print('copy')
			o.patternClipboard = []
			for s in o.soundSlots:
				notes = []
				for i in s.notes[o.ePattern]:
					notes.append([i[0], i[1]])
					#i[0] = 0
					#i[1] = 0
				o.patternClipboard.append(notes)
				print('adding to clip ', notes)
			# if o.soundSlots[o.eSound].volume > 0:
			# 	o.soundSlots[o.eSound].volume -= 1
			# 	text_center(o, "Sample Volume", str(o.soundSlots[o.eSound].volume))
			# 	o.update_display(0)
		elif o.keyState[1] == 1:
			print('paste')
			if o.patternClipboard != []:
				p = 0
				for s in o.soundSlots:
					n = 0
					for i in s.notes[o.ePattern]:
						i[0] = o.patternClipboard[p][n][0]
						i[1] = o.patternClipboard[p][n][1]
						n += 1
					print('added ', o.patternClipboard[p])
					p += 1
					
				
			else:
				print('nothing to paste')
			# if o.soundSlots[o.eSound].volume < 16:
			# 	o.soundSlots[o.eSound].volume += 1
			# 	text_center(o, "Sample Volume", str(o.soundSlots[o.eSound].volume))
			# 	o.update_display(0)
		elif o.keyState[2] == 1:
			print('clear')
			print(o.soundSlots[o.eSound].notes[o.ePattern])
			for s in o.soundSlots:
				for i in s.notes[o.ePattern]:
					i[0] = 0
					i[1] = 0
			#print(o.soundSlots[o.eSound].notes[o.ePattern])
			# if o.soundSlots[o.eSound].pitch > -10:
			# 	o.soundSlots[o.eSound].pitch -= 1
			# 	text_center(o, "Sample Pitch", str(o.soundSlots[o.eSound].pitch))
			# 	o.soundSlots[o.eSound].set_pitch()
			# 	o.update_display(0)
		elif o.keyState[6] == 1:
			if o.pat_bank > 0:
				o.pat_bank -= 1
				text_center(o, "Pattern Bank", str(o.pat_bank))
				o.update_display(0)		

		elif o.keyState[7] == 1:
			if o.pat_bank < 4:
				o.pat_bank += 1
				text_center(o, "Pattern Bank", str(o.pat_bank))
				o.update_display(0)		
				
				

def menu2_actions_note(self, o):
	if o.keyState[17] == 2:
		if o.keyState[0] == 1:
			if o.note_vol > 0:
				o.note_vol -= 1
				text_center(o, "Note Volume", str(o.note_vol))
				o.update_display(0)		


		elif o.keyState[1] == 1:
			if o.note_vol < 16:
				o.note_vol += 1
				text_center(o, "Note Volume", str(o.note_vol))
				o.update_display(0)
			
		
					

def util_command(self, o, command):
	if command == "Clock":
		self.FSM.ToTransition('toClock')		
	elif command == "Ball":
		self.FSM.ToTransition('toBall')		
	elif command == "Exit":
		self.FSM.ToTransition('toExit')		
	elif command == "Theme":
		self.FSM.ToTransition('toSelTheme')		
	elif command == "Reboot":
		#self.FSM.ToTransition('toSelTheme')		
		os.system("sudo reboot")


def file_command(self, o, command):
	if command == "Save":
		o.save_song()
	
	elif command == "Open":
		self.FSM.ToTransition('toOpenSong')	
	elif command == "Save As":
		self.FSM.ToTransition('toEnterText')	
#==================================== 

class Example(State):
	def __init__(self,FSM):
		super(Example, self).__init__(FSM)    
		
	def Enter(self):
		self.FSM.stateLife = 1
		super(Example, self).Enter()        
		
	def Execute(self):
		self.FSM.stateLife += 1
		
	def Exit(self):
		pass

#==================================== 

class Startup(State):
	def __init__(self,FSM):
		super(Startup, self).__init__(FSM)    
		
	def Enter(self):
		self.FSM.stateLife = 1
		self.birth = time.perf_counter()
		super(Startup, self).Enter()        
		
	def Execute(self):
		self.FSM.stateLife += 1
		print('startup state')

		if hasattr(self, 'birth'):
			pass
		else:
			self.birth = time.perf_counter()

		self.FSM.ToTransition('toIntro')
		
	def Exit(self):
		pass

#==================================== 
			
class Intro(State):
	def __init__(self,FSM):
		super(Intro, self).__init__(FSM)    
		
	def Enter(self):
		self.FSM.stateLife = 1
		o = self.FSM.owner
		o.header_text = "Starting up..."
		o.draw.rectangle((0, 0, self.FSM.owner.width, self.FSM.owner.height), outline=0, fill=o.blue)
		#draw_header(o)
		#o.draw.rectangle((0, 0, o.width, o.height), outline=0, fill=o.blue)			
		o.center_block("Starting up...", o.h2, [0, 0, o.width, o.height], o.light_grey)
		#o.display_thread.start()
		o.update_display(0)
		#o.update_display(0)
		super(Intro, self).Enter()        
		
	def Execute(self):
		self.FSM.stateLife += 1
		o = self.FSM.owner
		self.FSM.ToTransition('toLoadDefault')
		
	def Exit(self):
		pass

#==================================== 
			
class LoadDefault(State):
	def __init__(self,FSM):
		super(LoadDefault, self).__init__(FSM)    
		
	def Enter(self):
		self.FSM.stateLife = 1


		o = self.FSM.owner
		o.init_slots()
		o.load_song()



		self.bank = 0
		self.sounds = []
		self.globbed = glob.glob('/home/pi/skull/*.wav') + glob.glob('/home/pi/skull/*.mp3')
		self.globbed.sort()
		obj_id = 0
		
		# for s in o.sconf['sounds']:
		# 	p1 = self.FSM.owner.SoundSlot(s, obj_id, o)
		# 	self.FSM.owner.soundSlots.append(p1)
		# 	obj_id += 1

		# _iter = 0

		# lst = ast.literal_eval(o.sconf['notes'][0])
		# print('length ', len(lst))

		# for n in o.soundSlots:
		# 	m = []
		# 	if _iter < (len(o.sconf.as_list('notes')) - 1):
		# 		lst = ast.literal_eval(o.sconf.as_list('notes')[_iter])
		# 		n.notes = lst
		# 	_iter += 1

		super(LoadDefault, self).Enter()        
		
	def Execute(self):
		self.FSM.stateLife += 1
		self.FSM.ToTransition('toMain')
		
	def Exit(self):
		pass

#==================================== 

#==================================== 
			
class Main(State):
	def __init__(self,FSM):
		super(Main, self).__init__(FSM)    
		
	def Enter(self):
		print('hello main')
		self.FSM.owner.pub.register("beat", self)
		self.FSM.stateLife = 1
		o = self.FSM.owner 
		o.header_text = "Sound Select"
		if o.keyState[16] > 0 or o.keyState[17] > 0:
			pass
		else:
			draw_header(o)
			self.draw_square()

			text_box1(o, "Vol", str(o.soundSlots[o.eSound].volume))
			text_box2(o, "BPM", str(o.sconf['bpm']))
			text_box4(o, "Bank", str(o.note_bank))
			text_box3(o, "Pitch", str(o.soundSlots[o.eSound].pitch))
			o.update_display(0)

		#o.soundSlots[2].volume = 2
		super(Main, self).Enter()        
		
	def Execute(self):
		o = self.FSM.owner 
		menu1_actions(self, o)
		menu2_actions_sample(self, o)
		
		if o.keyState[16] == 1:
			draw_menu1(o)
			o.update_display(0)
		elif o.keyState[16] == 4:
			draw_header(o)
			self.draw_square()
			#print('esound ', o.eSound)
			#print('oss ', o.soundSlots[o.eSound].volume)
			text_box1(o, "SVol", str(o.soundSlots[o.eSound].volume))
			text_box2(o, "BPM", str(o.sconf['bpm']))
			text_box4(o, "Bank", str(o.note_bank))
			text_box3(o, "Pitch", str(o.soundSlots[o.eSound].pitch))
			o.update_display(0)


		if o.keyState[17] == 1:
			draw_menu2_sample(o)
			o.update_display(0)
		elif o.keyState[17] == 4:
			draw_header(o)
			self.draw_square()
			text_box1(o, "SVol", str(o.soundSlots[o.eSound].volume))
			text_box2(o, "BPM", str(o.sconf['bpm']))
			text_box4(o, "Bank", str(o.note_bank))
			text_box3(o, "Pitch", str(o.soundSlots[o.eSound].pitch))
			o.update_display(0)

		_id = 0
		for k in o.keyState:
			if _id > 15:
				pass
			else:
				if o.keyState[16] > 0 or o.keyState[17] > 0:
					pass
				else:
					if k == 1:
						note = _id + o.note_bank * 16
						#if len(o.soundSlots) > note:
						if 12 == 12:
							
							#o.soundSlots[note].play(o.note_vol)
							o.soundSlots[note].play(o.note_vol)
							o.eSound = note
							print('now editing sound ', o.eSound) 
							draw_header(o)
							#text_box1(o, "SVol", str(o.soundSlots[o.eSound].volume))
							text_box1(o, "SVol", str(o.slots[o.eSound].volume))
							text_box2(o, "BPM", str(o.sconf['bpm']))
							text_box4(o, "Bank", str(o.note_bank))
							#text_box3(o, "Pitch", str(o.soundSlots[o.eSound].pitch))
							text_box3(o, "Pitch", str(o.soundSlots[o.eSound].pitch))
							self.draw_square()
							o.update_display(0)
			_id += 1

	def ReceiveMessage(self, message):
		o = self.FSM.owner 
		play_seq(o, message)
		if o.keyState[16] > 0 or o.keyState[17] > 0:
			pass
		else:
			draw_header(o)            
			self.draw_square()   
			text_box1(o, "SVol", str(o.soundSlots[o.eSound].volume))
			text_box2(o, "BPM", str(o.sconf['bpm']))
			text_box4(o, "Bank", str(o.note_bank))
			text_box3(o, "Pitch", str(o.soundSlots[o.eSound].pitch))			         
			o.update_display(0)
	
	def draw_square(self):
		o = self.FSM.owner 
		size = 22
		og_x = (o.width / 2) - (size * 2)
		o_x = og_x
		o_y = 127
		_id = o.note_bank * 16 
	
		while _id < ((o.note_bank * 16) + 16):
			if _id == o.eSound:
				o.draw.rectangle((o_x, o_y, o_x + size, o_y - size), outline=o.light_grey, fill=o.light_grey, width=1)
			elif o.notes_on[_id] == 1:
				o.draw.rectangle((o_x, o_y, o_x + size, o_y - size), outline=o.light_grey, fill=o.grey, width=1)
			#elif _id > (len(o.soundSlots) - 1):
			elif self.has_sample(o, _id) == False:
				o.draw.rectangle((o_x, o_y, o_x + size, o_y - size), outline=o.light_grey, fill=o.dark_blue, width=1)


			else:
				o.draw.rectangle((o_x, o_y, o_x + size, o_y - size), outline=o.light_grey, fill=o.blue, width=1)
			o_x = o_x + size
			_id += 1
			if _id % 4 == 0:
				o_y -= size
				o_x = og_x

	def has_sample(self, o, slot):
		if o.soundSlots[slot].file != None:
			return True
		else:
			return False

	def Exit(self):
		self.FSM.owner.pub.unregister("beat", self)

#==================================== 
#==================================== 
			
class PadPlayer(State):
	def __init__(self,FSM):
		super(PadPlayer, self).__init__(FSM)    
		
	def Enter(self):
		self.FSM.stateLife = 1
		print('entering pad player')
		super(PadPlayer, self).Enter()        
		
	def Execute(self):
		self.FSM.stateLife += 1


		_id = 0
		for k in self.FSM.owner.keyState:
			if _id < 16 and k == 1:
				self.FSM.owner.soundSlots[_id].play()
			_id += 1
		if self.FSM.owner.keys:
			
			print("Pressed: ", self.FSM.owner.keys)
		#self.FSM.ToTransition('toLand')
		
	def Exit(self):
		pass

#==================================== 
			
class SeqPlayer(State):
	def __init__(self,FSM):
		super(SeqPlayer, self).__init__(FSM)    
		
	def Enter(self):
		self.FSM.stateLife = 1
		self.beat = 0
		self.FSM.owner.pub.register("beat", self)
		print('entering seq player')
		o = self.FSM.owner
		o.draw.rectangle((0, 0, o.width, o.height), outline=0, fill=o.blue)
		o.draw.text((20, 0), 'SeqPlayer', font=o.h1, fill="#FFFFFF")
		o.update_display(0)
		super(SeqPlayer, self).Enter()        
		
	def Execute(self):
		self.FSM.stateLife += 1
		o = self.FSM.owner

		if o.keyState[12] == 1:
			o.start_playback()
		if o.keyState[13] == 1:
			o.stop_playback()        
		#self.beat = o.playhead
		#print('key0 ', o.keyState[16])
		if o.keyState[16] == 2:
			#print('key0 ', o.keyState[16])
			if o.keyState[0] == 1:
				print('to main')
				self.FSM.ToTransition('toMain')

	def ReceiveMessage(self, message):
		#print(self.name, 'got message', message)
		o = self.FSM.owner
		o.draw.rectangle((0, 0, o.width, o.height), outline=0, fill=o.blue)
		o.draw.text((20, 0), str(message[0]) + ' - ' + str(message[1]), font=o.h1, fill="#FFFFFF")
		o.update_display(0)
		play_seq(o, message)

	def Exit(self):
		self.FSM.owner.pub.unregister("beat", self)

#==================================== 
#==================================== 
			
class SelectSound(State):
	def __init__(self,FSM):
		super(SelectSound, self).__init__(FSM)    
		
	def Enter(self):
		self.FSM.stateLife = 1
		o = self.FSM.owner 
		o.header_text = "SelSou"
		o.pub.register("beat", self)
		draw_header(o)
		self.draw_square()
		o.update_display(0)
		super(SelectSound, self).Enter()        
		
	def Execute(self):
		o = self.FSM.owner 
		if o.keyState[16] == 1:
			draw_menu1(o)
			o.update_display(0)
		elif o.keyState[16] == 4:
			draw_header(o)
			self.draw_square()
			o.update_display(0)

		if o.keyState[16] == 2:
			if o.keyState[0] == 1:
				self.FSM.ToTransition('toMain')
		else:
			_id = 0
			for k in o.keyState:
				if _id > 15:
					pass
				else:
					if k == 1:
						o.eSound = _id
						o.draw.rectangle((0, 0, o.width, o.height), outline=0, fill=o.blue)
						o.draw.text((0, 0), "SelSou", font=o.h1, fill=o.color_a)
						o.draw.text((20, 60), str(o.eSound), font=o.h2, fill=o.color_b)
						self.draw_square()
						o.update_display(0)
						self.FSM.owner.soundSlots[_id].play()
						#print('now editing sound ', _id)    
				_id += 1        

	def ReceiveMessage(self, message):
		_id = 0
		o = self.FSM.owner 
		o.draw.rectangle((0, 0, o.width, o.height), outline=0, fill=(0, 0, 0))
		o.draw.text((0, 0), "SelSou", font=o.h1, fill="#FFFFFF")
		o.draw.text((0, 20), str(o.eSound), font=o.h2, fill="#FFFFFF")
		o.update_display(0)
		play_seq(o, message)

	def draw_square(self):
		o = self.FSM.owner 
		size = 32
		og_x = 110
		o_x = og_x
		o_y = 0
		_id = 0
	
		for n in o.soundSlots[o.eSound].notes[o.ePattern]:
			if _id == o.eSound:
				o.draw.rectangle((o_x, o_y, o_x + size, o_y + size), outline=0, fill=o.color_b)
			else:
				o.draw.rectangle((o_x, o_y, o_x + size, o_y + size), outline=0, fill=o.dark_grey)
			o_x = o_x + size
			_id += 1
			if _id % 4 == 0:
				o_y += size
				o_x = og_x

		
	def Exit(self):
		self.FSM.owner.pub.unregister("beat", self)

#==================================== 
			
class SelectPattern(State):
	def __init__(self,FSM):
		super(SelectPattern, self).__init__(FSM)    
		
	def Enter(self):
		self.FSM.stateLife = 1
		o = self.FSM.owner 
		o.header_text = "Select Pattern"
		o.pub.register("beat", self)
		self.cur_playing = 0
		self.active_patterns = []
		self.get_active_patterns(o)
		if o.keyState[16] > 0 or o.keyState[17] > 0:
			pass
		else:
			draw_header(o)
			self.draw_square()
			text_box1(o, "Pat", str(o.ePattern))
			text_box4(o, "Bank", str(o.pat_bank))
			o.update_display(0)
		
		super(SelectPattern, self).Enter()        
		
	def Execute(self):
		o = self.FSM.owner 
		menu1_actions(self, o)
		menu2_actions_pattern(self, o)
		if o.keyState[16] == 1:
			draw_menu1(o)
			o.update_display(0)
		elif o.keyState[16] == 4:
			draw_header(o)
			self.draw_square()
			text_box1(o, "Pat", str(o.ePattern))
			text_box4(o, "Bank", str(o.pat_bank))
			o.update_display(0)

		if o.keyState[17] == 1:
			draw_menu2_pattern(o)
			o.update_display(0)
		elif o.keyState[17] == 4:
			draw_header(o)
			self.draw_square()
			#text_box1(o, "SVol", str(o.soundSlots[o.eSound].volume))
			#text_box2(o, "BPM", str(o.sconf['bpm']))
			text_box1(o, "Pat", str(o.ePattern))
			text_box4(o, "Bank", str(o.pat_bank))
			#text_box3(o, "Pitch", str(o.soundSlots[o.eSound].pitch))
			self.get_active_patterns(o)
			o.update_display(0)


		if o.keyState[16] > 0 or o.keyState[17] > 0:
			pass
		else:
			_id = 0
			for k in o.keyState:
				if _id > 15:
					pass
				else:
					if k == 1:
						o.ePattern = (o.pat_bank * 16) +_id 

						draw_header(o)
						self.draw_square()
						
						text_box1(o, "Pat", str(o.ePattern))
						text_box4(o, "Bank", str(o.pat_bank))
						o.update_display(0)
						

				_id += 1        

	def ReceiveMessage(self, message):
		o = self.FSM.owner
		play_seq(o, message) 
		
		if message[0] != self.cur_playing:
			#print('playing ', message)
			self.cur_playing = message[0]
			if o.keyState[16] > 0 or o.keyState[17] > 0:
				pass
			else:
				draw_header(o)
				self.draw_square()
				text_box1(o, "Pat", str(o.ePattern))
				text_box4(o, "PBank", str(o.pat_bank))
				o.update_display(0)

	def draw_square(self):
		o = self.FSM.owner 
		size = 22
		og_x = (o.width / 2) - (size * 2)
		o_x = og_x
		o_y = 127
		_id = 0
		for n in o.soundSlots[o.eSound].notes[o.ePattern]:
			if (_id + (o.pat_bank * 16)) == o.ePattern:
				o.draw.rectangle((o_x, o_y, o_x + size, o_y - size), outline=o.olive, fill=o.olive, width=1)
			elif (_id + (o.pat_bank * 16)) == self.cur_playing:
				o.draw.rectangle((o_x, o_y, o_x + size, o_y - size), outline=o.olive, fill=o.light_grey, width=1)
			elif self.active_patterns[_id] == 1:
				o.draw.rectangle((o_x, o_y, o_x + size, o_y - size), outline=o.light_grey, fill=o.blue, width=1)
			else:
				o.draw.rectangle((o_x, o_y, o_x + size, o_y - size), outline=o.light_grey, fill=o.dark_blue, width=1)
			o_x = o_x + size
			_id += 1
			if _id % 4 == 0:
				o_y -= size
				o_x = og_x          

	def get_active_patterns(self, o):
		#for n in o.soundSlots[o.eSound]
		self.active_patterns = [0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0]
		pat_iter = o.pat_bank * 16
		_iter = 0
		print('is this note on? ', o.soundSlots[0].notes[0][0][0])


		while pat_iter < ((o.pat_bank * 16) + 16):
			
			for s in o.soundSlots:
				print('checking soundSlot ', s.id)
				pat_used = 0
				for p in s.notes[pat_iter]:
					if self.active_patterns[_iter] == 1:
						break
					if p[0] == 1:
						#pat_used = 1
						self.active_patterns[_iter] = 1
						print('found ', pat_iter, ' -- ', p[0])
						break
				#self.active_patterns.append(pat_used)
			pat_iter += 1
			_iter += 1
	
		print('active patterns ', self.active_patterns)

	def Exit(self):
		self.FSM.owner.pub.unregister("beat", self)

#==================================== 
			
class EditSoundSequence(State):
	def __init__(self,FSM):
		super(EditSoundSequence, self).__init__(FSM)    
		
	def Enter(self):
		self.FSM.stateLife = 1
		o = self.FSM.owner 
		o.header_text = "Note Edit"
		o.pub.register("beat", self)
		if o.keyState[16] > 0 or o.keyState[17] > 0:
			pass
		else:
			draw_header(o)
			self.draw_square()
			text_box1(o, "Note Vol", str(o.note_vol))
			o.update_display(0)
		super(EditSoundSequence, self).Enter()        
		
	def Execute(self):
		o = self.FSM.owner 
		menu1_actions(self, o)
		menu2_actions_note(self, o)
		if o.keyState[16] == 1:
			draw_menu1(o)
			o.update_display(0)
		elif o.keyState[16] == 4:
			draw_header(o)
			self.draw_square()
			text_box1(o, "Note Vol", str(o.note_vol))
			o.update_display(0)

		if o.keyState[17] == 1:
			draw_menu2_note(o)
			o.update_display(0)
		elif o.keyState[17] == 4:
			draw_header(o)
			self.draw_square()
			text_box1(o, "Note Vol", str(o.note_vol))
			
			o.update_display(0)

		if o.keyState[16] == 2:
			pass

		else:
			_id = 0
			for k in self.FSM.owner.keyState:
				if _id < 16 and k == 1:
					if o.soundSlots[o.eSound].notes[o.ePattern][_id][0] == 1:
						print('turn note off')
						o.soundSlots[o.eSound].notes[o.ePattern][_id][0] = 0
						o.soundSlots[o.eSound].notes[o.ePattern][_id][1] = 0
					else:
						print('turn note on')
						o.soundSlots[o.eSound].notes[o.ePattern][_id][0] = 1
						o.soundSlots[o.eSound].notes[o.ePattern][_id][1] = o.note_vol
				_id += 1

		
	def draw_square(self):
		o = self.FSM.owner 
		size = 22
		og_x = (o.width / 2) - (size * 2)
		o_x = og_x
		o_y = 127
		_id = 0
		for n in o.soundSlots[o.eSound].notes[o.ePattern]:
			if _id == o.playhead:
				if o.curPattern == o.ePattern:
					o.draw.rectangle((o_x, o_y, o_x + size, o_y - size), outline=o.olive, fill=o.olive, width=1)
				else:
					o.draw.rectangle((o_x, o_y, o_x + size, o_y - size), outline=o.olive, fill=o.pink, width=1)
			elif n[0] == 1:
				o.draw.rectangle((o_x, o_y, o_x + size, o_y - size), outline=o.light_grey, fill=o.light_grey, width=1)
			else:
				o.draw.rectangle((o_x, o_y, o_x + size, o_y - size), outline=o.light_grey, fill=o.blue, width=1)
			o_x = o_x + size
			_id += 1
			if _id % 4 == 0:
				o_y -= size
				o_x = og_x

	def ReceiveMessage(self, message):
		_id = 0
		o = self.FSM.owner 
		play_seq(o, message)
		if o.keyState[16] > 0 or o.keyState[17] > 0:
			pass
		else:
			draw_header(o)            
			self.draw_square()  
			text_box1(o, "Note Vol", str(o.note_vol))
		o.update_display(0)
		
	def Exit(self):
		self.FSM.owner.pub.unregister("beat", self)

#==================================== 
			
class EditSong(State):
	def __init__(self,FSM):
		super(EditSong, self).__init__(FSM)    
		
	def Enter(self):
		o = self.FSM.owner 
		o.header_text = "Edit Song"
		o.pub.register("beat", self)
		if o.keyState[16] > 0 or o.keyState[17] > 0:
			pass
		else:
			draw_header(o)
			o.draw.text((20, 30), "Current Song", font=o.h2, fill=o.light_grey)
			o.draw.text((20, 50), str(o.song), font=o.h2, fill=o.light_grey)
			o.update_display(0)
		self.new_song = []
		super(EditSong, self).Enter()        
		
	def Execute(self):
		o = self.FSM.owner 

		menu1_actions(self, o)
		if o.keyState[16] == 1:
			draw_menu1(o)
			o.update_display(0)
		elif o.keyState[16] == 4:
			draw_header(o)
			if self.new_song == []:
				o.draw.text((20, 30), "Current Song", font=o.h2, fill=o.light_grey)
				o.draw.text((20, 50), str(o.song), font=o.h2, fill=o.light_grey)
			else:
				o.draw.text((20, 30), "New Song", font=o.h2, fill=o.pink)
				o.draw.text((20, 50), str(self.new_song), font=o.h2, fill=o.pink)
			o.update_display(0)

		if o.keyState[16] > 0:
			pass
		else:
			_id = 0
			for k in self.FSM.owner.keyState:
				if k == 1 and _id < 16:
					self.new_song.append((_id + (o.pat_bank * 16)))
					draw_header(o)
					o.draw.text((20, 30), "New Song", font=o.h2, fill=o.pink)
					o.draw.text((20, 50), str(self.new_song), font=o.h2, fill=o.pink)
					o.update_display(0)
				_id += 1        

	def ReceiveMessage(self, message):
		_id = 0
		o = self.FSM.owner 
		play_seq(o, message)

	def Exit(self):
		o = self.FSM.owner 
		self.FSM.owner.pub.unregister("beat", self)
		if self.new_song != []:
			o.song = self.new_song
			o.songCycle = cycle(self.new_song)
			self.curPattern = next(o.songCycle)
			o.songStart = o.curPattern

#==================================== 

class File(State):
	def __init__(self,FSM):
		super(File, self).__init__(FSM)    
		
	def Enter(self):
		self.FSM.stateLife = 1
		o = self.FSM.owner 
		o.header_text = "File"
		o.pub.register("beat", self)
		self.cur_el = 0
		if o.keyState[16] > 0 or o.keyState[17] > 0:
			pass
		else:
			draw_header(o)
			self.draw_footer(o)
			self.draw_main(o)
			o.update_display(0)
		self.menu = ["Save", "Save As", "Open", "New"]
		
		super(File, self).Enter()        
		
	def Execute(self):
		o = self.FSM.owner 
		menu1_actions(self, o)
		if o.keyState[16] == 1:
			draw_menu1(o)
			#self.draw_footer(o)
			o.update_display(0)
		elif o.keyState[16] == 4:
			draw_header(o)
			self.draw_footer(o)
			self.draw_main(o)
			o.update_display(0)

		if o.keyState[0] == 1 or o.keyState[0] == 3 or o.keyState[1] == 1 or o.keyState[1] == 3 or o.keyState[2] == 1 or o.keyState[2] == 3 or o.keyState[3] == 1 or o.keyState[3] == 3:
			draw_header(o)
			self.draw_footer(o)
			self.draw_main(o)
			o.update_display(0)	
			print('draw butt')				


		if o.keyState[16] > 0 or o.keyState[17] > 0:
			pass
		else:
			if o.keyState[0] == 1:
				if self.cur_el > 0:
					self.cur_el -= 1
					print('cur el down is ', self.cur_el)
			if o.keyState[1] == 1:
				if self.cur_el < (len(self.menu) - 1):
					self.cur_el += 1
					print('cur el up is ', self.cur_el)
			
			if o.keyState[3] == 1:
				file_command(self, o, self.menu[self.cur_el])
				pass

	def draw_footer(self, o):
		_iter = 0
		width = o.width / 4
		xpos = 0
		yposa = 135 - 25
		yposb = 135
		opts = ["Up", "Down", "Esc", "OK"]
		while _iter < 4:
			o.draw.rectangle((xpos, yposa, xpos + width, yposb), outline=0, fill=o.blue)
			
			if o.keyState[_iter] == 1:
				o.draw.rectangle((xpos, yposa, xpos + width, yposb), outline=0, fill=o.light_grey)
				o.center_block(opts[_iter], o.h2, [xpos,yposa,xpos + width,yposb], o.blue)
			else:
				o.center_block(opts[_iter], o.h2, [xpos,yposa,xpos + width,yposb], o.light_grey)
			
			xpos += width
			_iter += 1


	def draw_main(self, o):
		al = []
		ypos = 30
		if self.cur_el == 0:
			_iter = 0
			while len(al) < 4:
				al.append(self.menu[_iter])
				_iter += 1
		elif self.cur_el == 1:
			al.append(self.menu[self.cur_el - 1])
			al.append(self.menu[self.cur_el])
			al.append(self.menu[self.cur_el + 1])
			al.append(self.menu[self.cur_el + 2])
		elif self.cur_el == (len(self.menu) - 2):
			al.append(self.menu[self.cur_el - 2])
			al.append(self.menu[self.cur_el - 1])
			al.append(self.menu[self.cur_el])
			al.append(self.menu[self.cur_el + 1])
		elif self.cur_el == (len(self.menu) - 1):
			al.append(self.menu[self.cur_el - 3])
			al.append(self.menu[self.cur_el - 2])
			al.append(self.menu[self.cur_el - 1])
			al.append(self.menu[self.cur_el])
		else:
			al.append(self.menu[self.cur_el - 1])
			al.append(self.menu[self.cur_el])
			al.append(self.menu[self.cur_el + 1])
			al.append(self.menu[self.cur_el + 2])
		_iter = 0 
		for i in al:
			if i == self.menu[self.cur_el]:
			#if _iter == self.cur_el:
				o.draw.rectangle((0, ypos, o.width, ypos + 20), outline=0, fill=o.light_grey)
				o.center_block(i, o.h2, [0, ypos, o.width,ypos + 20], o.blue)
			else:
				o.draw.rectangle((0, ypos, o.width, ypos + 20), outline=0, fill=o.blue)
				o.center_block(i, o.h2, [0, ypos, o.width,ypos + 20], o.light_grey)
			ypos += 20
			_iter += 1

	def ReceiveMessage(self, message):
		o = self.FSM.owner 
		play_seq(o, message)
		
	def Exit(self):
		self.FSM.owner.pub.unregister("beat", self)

class Util(State):
	def __init__(self,FSM):
		super(Util, self).__init__(FSM)    
		
	def Enter(self):
		o = self.FSM.owner 
		o.header_text = "Utilities"
		o.pub.register("beat", self)
		self.menu = ["Clock", "Ball", "Exit", "Shutdown", "Reboot", "Theme", "Update"]
		self.cur_el = 0

		if o.keyState[16] > 0 or o.keyState[17] > 0:
			pass
		else:
			draw_header(o)
			self.draw_footer(o)
			self.draw_main(o)
			o.update_display(0)

		
		super(Util, self).Enter()        
		
	def Execute(self):
		o = self.FSM.owner 
		menu1_actions(self, o)
		if o.keyState[16] == 1:
			draw_menu1(o)
			o.update_display(0)
		elif o.keyState[16] == 4:
			draw_header(o)
			self.draw_footer(o)
			self.draw_main(o)
			o.update_display(0)

		if o.keyState[0] == 1 or o.keyState[0] == 3 or o.keyState[1] == 1 or o.keyState[1] == 3 or o.keyState[2] == 1 or o.keyState[2] == 3 or o.keyState[3] == 1 or o.keyState[3] == 3:
			draw_header(o)
			self.draw_footer(o)
			self.draw_main(o)
			o.update_display(0)	
			print('draw butt')				

		if o.keyState[16] > 0 or o.keyState[17] > 0:
			pass
		else:
			if o.keyState[0] == 1:
				if self.cur_el > 0:
					self.cur_el -= 1
					print('cur el down is ', self.cur_el)
			if o.keyState[1] == 1:
				if self.cur_el < (len(self.menu) - 1):
					self.cur_el += 1
					print('cur el up is ', self.cur_el)
			
			if o.keyState[3] == 1:
				util_command(self, o, self.menu[self.cur_el])
		
	def ReceiveMessage(self, message):
		_id = 0
		o = self.FSM.owner 
		play_seq(o, message)
		# if o.keyState[16] > 0 or o.keyState[17] > 0:
		# 	pass
		# else:
		# 	draw_header(o)           		
		# o.update_display(0)

	def draw_main(self, o):
		al = []
		ypos = 30
		if self.cur_el == 0:
			_iter = 0
			while len(al) < 4:
				al.append(self.menu[_iter])
				_iter += 1
		elif self.cur_el == 1:
			al.append(self.menu[self.cur_el - 1])
			al.append(self.menu[self.cur_el])
			al.append(self.menu[self.cur_el + 1])
			al.append(self.menu[self.cur_el + 2])
		elif self.cur_el == (len(self.menu) - 2):
			al.append(self.menu[self.cur_el - 2])
			al.append(self.menu[self.cur_el - 1])
			al.append(self.menu[self.cur_el])
			al.append(self.menu[self.cur_el + 1])
		elif self.cur_el == (len(self.menu) - 1):
			al.append(self.menu[self.cur_el - 3])
			al.append(self.menu[self.cur_el - 2])
			al.append(self.menu[self.cur_el - 1])
			al.append(self.menu[self.cur_el])
		else:
			al.append(self.menu[self.cur_el - 1])
			al.append(self.menu[self.cur_el])
			al.append(self.menu[self.cur_el + 1])
			al.append(self.menu[self.cur_el + 2])
		_iter = 0 
		for i in al:
			if i == self.menu[self.cur_el]:
			#if _iter == self.cur_el:
				o.draw.rectangle((0, ypos, o.width, ypos + 20), outline=0, fill=o.light_grey)
				o.center_block(i, o.h2, [0, ypos, o.width,ypos + 20], o.blue)
			else:
				o.draw.rectangle((0, ypos, o.width, ypos + 20), outline=0, fill=o.blue)
				o.center_block(i, o.h2, [0, ypos, o.width,ypos + 20], o.light_grey)
			ypos += 20
			_iter += 1

	def draw_footer(self, o):
		_iter = 0
		width = o.width / 4
		xpos = 0
		yposa = 135 - 25
		yposb = 135
		opts = ["Up", "Down", "Esc", "OK"]
		while _iter < 4:
			o.draw.rectangle((xpos, yposa, xpos + width, yposb), outline=0, fill=o.blue)
			
			if o.keyState[_iter] == 1:
				o.draw.rectangle((xpos, yposa, xpos + width, yposb), outline=0, fill=o.light_grey)
				o.center_block(opts[_iter], o.h2, [xpos,yposa,xpos + width,yposb], o.blue)
			else:
				o.center_block(opts[_iter], o.h2, [xpos,yposa,xpos + width,yposb], o.light_grey)
			
			xpos += width
			_iter += 1

	def Exit(self):
		self.FSM.owner.pub.unregister("beat", self)        



class Ball(State):
	def __init__(self,FSM):
		super(Ball, self).__init__(FSM)    
		
	def Enter(self):
		self.ball_size = 10
		self.ballx = self.ball_size
		self.bally = self.ball_size
		
		self.ball_speed = 15
		self.ball_x_dir = 1
		self.ball_y_dir = 1
		
		self.ball2x = self.ball_size + 180
		self.ball2y = self.ball_size + 70
	
		self.ball2_x_dir = -1
		self.ball2_y_dir = -1

		self.xmult = 1.0

		o = self.FSM.owner 
		o.header_text = "Ball"
		o.pub.register("beat", self)
		if o.keyState[16] > 0 or o.keyState[17] > 0:
			pass
		else:
			draw_header(o)
			o.update_display(0)
		super(Ball, self).Enter()        
		
	def Execute(self):
		o = self.FSM.owner 
		menu1_actions(self, o)
		if o.keyState[16] == 1:
			draw_menu1(o)
			o.update_display(0)
		

		if o.keyState[16] > 0 or o.keyState[17] > 0:
			pass
		else:			
			now = datetime.now()
			current_time = time.strftime("%-I:%M %p")


			if self.ballx < (0 + self.ball_size):
				self.ball_x_dir = 1
			if self.ballx > (o.width - self.ball_size):
				self.ball_x_dir = -1

			if self.bally < (0 + self.ball_size):
				self.ball_y_dir = 1
			if self.bally > (o.height - self.ball_size):
				self.ball_y_dir = -1

			# if self.ballx > (o.width - self.ball_size) or self.ballx < (0 + self.ball_size):
			# 	self.ball_x_dir *= -1
			# if self.bally > (o.height - self.ball_size) or self.bally < (0 + self.ball_size):
			# 	self.ball_y_dir *= -1

			self.ballx += self.ball_speed * self.ball_x_dir
			self.bally += self.ball_speed * self.ball_y_dir


			if self.ball2x < (0 + self.ball_size):
				self.ball2_x_dir = 1
				self.xmult = random.uniform(.6, 1.2)
			if self.ball2x > (o.width - self.ball_size):
				self.ball2_x_dir = -1
				self.xmult = random.uniform(.6, 1.2)

			if self.ball2y < (0 + self.ball_size):
				self.ball2_y_dir = 1
			if self.ball2y > (o.height - self.ball_size):
				self.ball2_y_dir = -1			


			# if self.ball2x > (o.width - self.ball_size) or self.ball2x < (0 + self.ball_size):
			# 	self.ball2_x_dir *= -1
			# 	self.xmult = random.uniform(.8, 1.2)
			# if self.ball2y > (o.height - self.ball_size) or self.ball2y < (0 + self.ball_size):
			# 	self.ball2_y_dir *= -1

			self.ball2x += (self.ball_speed * self.ball2_x_dir) * self.xmult
			self.ball2y += (self.ball_speed * self.ball2_y_dir) 

			o.draw.rectangle((0, 0, o.width, o.height), outline=0, fill=o.blue)
			
			o.draw.ellipse((self.ballx - self.ball_size, self.bally - self.ball_size, self.ballx + self.ball_size, self.bally + self.ball_size), fill = o.light_grey, outline =o.light_grey)
			o.draw.ellipse((self.ball2x - self.ball_size, self.ball2y - self.ball_size, self.ball2x + self.ball_size, self.ball2y + self.ball_size), fill = o.light_grey, outline =o.light_grey)
			#o.center_block(current_time, o.h3, [0,0,o.width,o.height], o.pink)
			o.update_display(0)
		
	def ReceiveMessage(self, message):
		o = self.FSM.owner 
		play_seq(o, message)
		
	def Exit(self):
		self.FSM.owner.pub.unregister("beat", self)        

#==================================== 


class Clock(State):
	def __init__(self,FSM):
		super(Clock, self).__init__(FSM)    
		
	def Enter(self):
		o = self.FSM.owner 
		current_time = time.strftime("%-I:%M:%S")
		self.ball_sizex, self.ball_sizey = o.h3.getsize("18:88:88")
		self.ball_sizex /= 2
		self.ball_sizey /= 2
		self.ballx = o.width / 2 
		self.bally = o.height / 2
		self.ball_speed = 1
		self.ball_x_dir = 1
		self.ball_y_dir = 1
		
		o.header_text = "Clock"
		o.pub.register("beat", self)
		if o.keyState[16] > 0 or o.keyState[17] > 0:
			pass
		else:
			pass
		super(Clock, self).Enter()        
		
	def Execute(self):
		o = self.FSM.owner 
		menu1_actions(self, o)

		if o.keyState[16] == 1:
			draw_menu1(o)
			o.update_display(0)


		if o.keyState[16] > 0 or o.keyState[17] > 0:
			pass
		else:			
			current_time = time.strftime("%-I:%M:%S")

			if self.ballx > (o.width - self.ball_sizex) or self.ballx < (0 + self.ball_sizex):
				self.ball_x_dir *= -1
			if (self.bally > o.height - self.ball_sizey) or (self.bally < 0 + self.ball_sizey):
				self.ball_y_dir *= -1

			self.ballx += self.ball_speed * self.ball_x_dir
			self.bally += self.ball_speed * self.ball_y_dir
			o.draw.rectangle((0, 0, o.width, o.height), outline=0, fill=o.bg_color)			
			o.draw.text((self.ballx - self.ball_sizex, self.bally - self.ball_sizey), current_time, align='center', font=o.h3, fill=o.light_grey)
			o.update_display(0)
		
	def ReceiveMessage(self, message):
		o = self.FSM.owner 
		play_seq(o, message)
		
	def Exit(self):
		self.FSM.owner.pub.unregister("beat", self)        

#==================================== 

class Exit(State):
	def __init__(self,FSM):
		super(Exit, self).__init__(FSM)    
		
	def Enter(self):
		o = self.FSM.owner
		o.draw.rectangle((0, 0, o.width, o.height), outline=0, fill=o.blue)			
		o.center_block("goodbye", o.h2, [0, 0, o.width,o.height], o.light_grey)
		o.update_display(0)
		pygame.quit()
		sys.exit()
		super(Exit, self).Enter()        
		
	def Execute(self):
		self.FSM.stateLife += 1
		
	def Exit(self):
		pass

#==================================== 

class SelTheme(State):
	def __init__(self,FSM):
		super(SelTheme, self).__init__(FSM)    
		
	def Enter(self):
		o = self.FSM.owner
		o.header_text = "Select Theme"
		o.pub.register("beat", self)
		self.cur_el = 0
		self.globbed = glob.glob('/home/pi/zpc_ct/themes/*.thm')
		self.globbed.sort()
		self.names = [os.path.basename(x) for x in self.globbed]
		self.names = [os.path.splitext(x)[0] for x in self.names]
		if o.mconf['theme'] in self.names:
			_id = self.names.index(o.mconf['theme'])
			self.cur_el = _id
			#print('found theme ', _id)
		else:
			print('nope ', o.mconf['theme'], ' - ', self.globbed)
		#self.theme = ConfigObj("/home/pi/zpc_ct/themes/" + self.mconf['theme'])
		#print(os.path.splitext(x)[0])
		self.menu = self.names

		#self.cur_el = 0
		print('theme menu is ', self.menu)

		if o.keyState[16] > 0 or o.keyState[17] > 0:
			pass
		else:
			draw_header(o)
			self.draw_main(o)			
			
			o.update_display(0)
		super(SelTheme, self).Enter()    

	def Execute(self):
		o = self.FSM.owner 
		menu1_actions(self, o)
		if o.keyState[16] == 1:
			draw_menu1(o)
			o.update_display(0)
		elif o.keyState[16] == 4:
			draw_header(o)
			self.draw_footer(o)
			self.draw_main(o)
			o.update_display(0)

		if o.keyState[0] == 1 or o.keyState[0] == 3 or o.keyState[1] == 1 or o.keyState[1] == 3 or o.keyState[2] == 1 or o.keyState[2] == 3 or o.keyState[3] == 1 or o.keyState[3] == 3:
			draw_header(o)
			self.draw_footer(o)
			self.draw_main(o)
			o.update_display(0)	
			print('draw butt')				

		if o.keyState[16] > 0 or o.keyState[17] > 0:
			pass
		else:
			if o.keyState[0] == 1:
				if self.cur_el > 0:
					self.cur_el -= 1
					o.mconf['theme'] = self.menu[self.cur_el]					
					o.save_config()
					o.load_config()
					o.apply_theme()
					print('cur el down is ', self.cur_el)

					#o.applyTheme
			if o.keyState[1] == 1:
				if self.cur_el < (len(self.menu) - 1):
					self.cur_el += 1
					o.mconf['theme'] = self.menu[self.cur_el]					
					o.save_config()
					o.load_config()
					o.apply_theme()
					print('cur el up is ', self.cur_el)
					print('loading theme ', self.menu[self.cur_el], '-- ', o.mconf['theme'])
			if o.keyState[3] == 1:
				util_command(self, o, self.menu[self.cur_el])

	def draw_main(self, o):
		al = []
		ypos = 30
		if self.cur_el == 0:
			_iter = 0
			while len(al) < 4:
				al.append(self.menu[_iter])
				_iter += 1
		elif self.cur_el == 1:
			al.append(self.menu[self.cur_el - 1])
			al.append(self.menu[self.cur_el])
			al.append(self.menu[self.cur_el + 1])
			al.append(self.menu[self.cur_el + 2])
		elif self.cur_el == (len(self.menu) - 2):
			al.append(self.menu[self.cur_el - 2])
			al.append(self.menu[self.cur_el - 1])
			al.append(self.menu[self.cur_el])
			al.append(self.menu[self.cur_el + 1])
		elif self.cur_el == (len(self.menu) - 1):
			al.append(self.menu[self.cur_el - 3])
			al.append(self.menu[self.cur_el - 2])
			al.append(self.menu[self.cur_el - 1])
			al.append(self.menu[self.cur_el])
		else:
			al.append(self.menu[self.cur_el - 1])
			al.append(self.menu[self.cur_el])
			al.append(self.menu[self.cur_el + 1])
			al.append(self.menu[self.cur_el + 2])
		_iter = 0 
		for i in al:
			if i == self.menu[self.cur_el]:
			#if _iter == self.cur_el:
				o.draw.rectangle((0, ypos, o.width, ypos + 20), outline=0, fill=o.light_grey)
				o.center_block(i, o.h2, [0, ypos, o.width,ypos + 20], o.blue)
			else:
				o.draw.rectangle((0, ypos, o.width, ypos + 20), outline=0, fill=o.blue)
				o.center_block(i, o.h2, [0, ypos, o.width,ypos + 20], o.light_grey)
			ypos += 20
			_iter += 1		    

	def draw_footer(self, o):
		_iter = 0
		width = o.width / 4
		xpos = 0
		yposa = 135 - 25
		yposb = 135
		opts = ["Up", "Down", "", ""]
		while _iter < 4:
			o.draw.rectangle((xpos, yposa, xpos + width, yposb), outline=0, fill=o.blue)
			
			if o.keyState[_iter] == 1:
				o.draw.rectangle((xpos, yposa, xpos + width, yposb), outline=0, fill=o.light_grey)
				o.center_block(opts[_iter], o.h2, [xpos,yposa,xpos + width,yposb], o.blue)
			else:
				o.center_block(opts[_iter], o.h2, [xpos,yposa,xpos + width,yposb], o.light_grey)
			
			xpos += width
			_iter += 1
		

	def ReceiveMessage(self, message):
		o = self.FSM.owner 
		play_seq(o, message)
		
	def Exit(self):
		self.FSM.owner.pub.unregister("beat", self)

#==================================== 

class OpenSong(State):
	def __init__(self,FSM):
		super(OpenSong, self).__init__(FSM)    
		
	def Enter(self):
		o = self.FSM.owner
		o.header_text = "Open Song"
		o.pub.register("beat", self)
		self.cur_el = 0
		self.globbed = glob.glob('/home/pi/zpc_ct/user/songs/*.sng')
		self.globbed.sort()
		if '/home/pi/zpc_ct/user/songs/_blank.sng' in self.globbed:
			self.globbed.remove('/home/pi/zpc_ct/user/songs/_blank.sng')
		#del self.globbed[0]
		self.names = [os.path.basename(x) for x in self.globbed]
		self.names = [os.path.splitext(x)[0] for x in self.names]
		print('we got these songs ', self.names)
		print('in these paths ', self.globbed)
		if o.mconf['default_song'] in self.names:
			_id = self.names.index(o.mconf['default_song'])
			self.cur_el = _id
		else:
			print('nope ', o.mconf['default_song'], ' - ', self.globbed)
		self.menu = self.names

		#self.cur_el = 0
		#print('theme menu is ', self.menu)

		if o.keyState[16] > 0 or o.keyState[17] > 0:
			pass
		else:
			draw_header(o)
			self.draw_main(o)			
			
			o.update_display(0)
		super(OpenSong, self).Enter()    

	def Execute(self):
		o = self.FSM.owner 
		menu1_actions(self, o)
		if o.keyState[16] == 1:
			draw_menu1(o)
			o.update_display(0)
		elif o.keyState[16] == 4:
			draw_header(o)
			self.draw_footer(o)
			self.draw_main(o)
			o.update_display(0)

		if o.keyState[0] == 1 or o.keyState[0] == 3 or o.keyState[1] == 1 or o.keyState[1] == 3 or o.keyState[2] == 1 or o.keyState[2] == 3 or o.keyState[3] == 1 or o.keyState[3] == 3:
			draw_header(o)
			self.draw_footer(o)
			self.draw_main(o)
			o.update_display(0)	
			print('draw butt')				

		if o.keyState[16] > 0 or o.keyState[17] > 0:
			pass
		else:
			if o.keyState[0] == 1:
				if self.cur_el > 0:
					self.cur_el -= 1
					# o.mconf['theme'] = self.menu[self.cur_el]					
					# o.save_config()
					# o.load_config()
					# o.apply_theme()
					print('cur el down is ', self.cur_el)

					#o.applyTheme
			if o.keyState[1] == 1:
				if self.cur_el < (len(self.menu) - 1):
					self.cur_el += 1
					# o.mconf['theme'] = self.menu[self.cur_el]					
					# o.save_config()
					# o.load_config()
					# o.apply_theme()
					print('cur el up is ', self.cur_el)
					#print('loading theme ', self.menu[self.cur_el], '-- ', o.mconf['theme'])
			if o.keyState[3] == 1:
				#open_command(self, o, self.menu[self.cur_el])
				o.mconf['default_song'] = self.menu[self.cur_el]
				o.load_song()
				#pass

	def draw_main(self, o):
		al = []
		ypos = 30
		if self.cur_el == 0:
			_iter = 0
			while len(al) < 4:
				al.append(self.menu[_iter])
				_iter += 1
		elif self.cur_el == 1:
			al.append(self.menu[self.cur_el - 1])
			al.append(self.menu[self.cur_el])
			al.append(self.menu[self.cur_el + 1])
			al.append(self.menu[self.cur_el + 2])
		elif self.cur_el == (len(self.menu) - 2):
			al.append(self.menu[self.cur_el - 2])
			al.append(self.menu[self.cur_el - 1])
			al.append(self.menu[self.cur_el])
			al.append(self.menu[self.cur_el + 1])
		elif self.cur_el == (len(self.menu) - 1):
			al.append(self.menu[self.cur_el - 3])
			al.append(self.menu[self.cur_el - 2])
			al.append(self.menu[self.cur_el - 1])
			al.append(self.menu[self.cur_el])
		else:
			al.append(self.menu[self.cur_el - 1])
			al.append(self.menu[self.cur_el])
			al.append(self.menu[self.cur_el + 1])
			al.append(self.menu[self.cur_el + 2])
		_iter = 0 
		for i in al:
			if i == self.menu[self.cur_el]:
			#if _iter == self.cur_el:
				o.draw.rectangle((0, ypos, o.width, ypos + 20), outline=0, fill=o.light_grey)
				o.center_block(i, o.h2, [0, ypos, o.width,ypos + 20], o.blue)
			else:
				o.draw.rectangle((0, ypos, o.width, ypos + 20), outline=0, fill=o.blue)
				o.center_block(i, o.h2, [0, ypos, o.width,ypos + 20], o.light_grey)
			ypos += 20
			_iter += 1		    

	def draw_footer(self, o):
		_iter = 0
		width = o.width / 4
		xpos = 0
		yposa = 135 - 25
		yposb = 135
		opts = ["Up", "Down", "Esc", "Open"]
		while _iter < 4:
			o.draw.rectangle((xpos, yposa, xpos + width, yposb), outline=0, fill=o.blue)
			
			if o.keyState[_iter] == 1:
				o.draw.rectangle((xpos, yposa, xpos + width, yposb), outline=0, fill=o.light_grey)
				o.center_block(opts[_iter], o.h2, [xpos,yposa,xpos + width,yposb], o.blue)
			else:
				o.center_block(opts[_iter], o.h2, [xpos,yposa,xpos + width,yposb], o.light_grey)
			
			xpos += width
			_iter += 1
		

	def ReceiveMessage(self, message):
		o = self.FSM.owner 
		play_seq(o, message)
		
	def Exit(self):
		self.FSM.owner.pub.unregister("beat", self)


class EnterText(State):
	def __init__(self,FSM):
		super(EnterText, self).__init__(FSM)    
		
	def Enter(self):
		o = self.FSM.owner 
		
		self.input_string = ""
		#if 
		o.header_text = "EnterText"
		o.pub.register("beat", self)
		self.last_action = pygame.time.get_ticks()
		self.caps = False
		self.cur_but = 18
		self.cur_letter = ""
		if o.keyState[16] > 0 or o.keyState[17] > 0:
			pass
		else:
			pass

		self.skips = [7, 3, 11, 12, 13, 14, 15]
		
		self.lc_labels = [["p", "q", "r", "s"], ["t", "u", "v"], ["w", "x", "y", "z"], ["C", "A", "P", "S"], 
				["g", "h", "i"], ["j", "k", "l"], ["m", "n", "o"], ["D", "E", "L"],
				["_", "1", "2", "3", "4", "5", "6", "7", "8", "9", "0", "-"], ["a", "b", "c"], ["d", "e", "f"], ["O", "K"],
				[], [], [], []]

		self.uc_labels = [["P", "Q", "R", "S"], ["T", "U", "V"], ["W", "X", "Y", "Z"], ["C", "A", "P", "S"], 
				["G", "H", "I"], ["J", "K", "L"], ["M", "N", "O"], ["D", "E", "L"],
				["_", "1", "2", "3", "4", "5", "6", "7", "8", "9", "0", "-"], ["A", "B", "C"], ["D", "E", "F"], ["O", "K"],
				[], [], [], []]

		self.labels = self.lc_labels

		#self.songCycle = cycle(self.song)
		#self.curPattern = next(self.songCycle)

		self.cycles = []
		for l in self.labels:
			self.cycles.append(cycle(l))



		self.draw(o)
		o.update_display(0)
		super(EnterText, self).Enter()        
		
	def Execute(self):
		o = self.FSM.owner 
		y_size = o.height / 4
		if o.keyState[11] == 1:
			self.FSM.ToTransition('toFile')		
		if o.keyState[16] > 0 and o.keyState[17] > 0:
			pass
		else:
			_iter = 0
			for k in o.keyState:
				if k == 1:
					cur_time = pygame.time.get_ticks()
					print(cur_time - self.last_action)
					
					#print('cur letter ', next(self.cycles[_iter]))
					if _iter not in self.skips:
						self.last_action = cur_time
						self.cur_letter = next(self.cycles[_iter])
						self.cur_but = _iter
						self.draw(o)
						o.center_block(self.input_string + self.cur_letter, o.h2, [0, 0, o.width, y_size], o.light_grey)

					if _iter == 7:
						
						self.input_string = self.input_string[:-1]
						#self.input_string = self.input_string.rstrip(self.input_string[-1])
						print('deleting')
						self.last_action = 100000000
						self.draw(o)
						o.center_block(self.input_string, o.h2, [0, 0, o.width, y_size], o.light_grey)

					elif _iter == 3:
						print('caps')
						if self.caps:
							self.caps = False
							self.labels = self.lc_labels

						else:
							self.caps = True
							self.labels = self.uc_labels
						self.draw(o)
						o.center_block(self.input_string, o.h2, [0, 0, o.width, y_size], o.light_grey)
						self.cycles = []
						for l in self.labels:
							self.cycles.append(cycle(l))

					o.update_display(0)
				_iter += 1

		cur_time = pygame.time.get_ticks()
		delay = cur_time - self.last_action
		if delay > 1000 and delay < 100000:
			self.input_string += self.cur_letter
			self.last_action = 100000000
			self.reset_cycles()
			self.draw(o)

			o.center_block(self.input_string, o.h2, [0, 0, o.width, y_size], o.light_grey)
			o.update_display(0)
	
	def ReceiveMessage(self, message):
		o = self.FSM.owner 
		play_seq(o, message)

	def draw(self, o):
		x_size = o.width / 4
		y_size = o.height / 4
		og_x = 0
		o_x = og_x
		o_y = o.height
		text_padding = 6
		
		_id = 0
		while _id < 16:
			lab = ""
			for i in self.labels[_id]:
				lab += i

			if _id == 8:
				lab = "_12"

			o.draw.rectangle((o_x, o_y, o_x + x_size, o_y - y_size), outline=0, fill=o.blue)
			o.center_block(lab, o.h2, [o_x, o_y, o_x + x_size, o_y - y_size], o.light_grey)

			o_x = o_x + x_size
			_id += 1
			if _id % 4 == 0:
				o_y -= y_size
				o_x = og_x
		o.draw.rectangle((0, 0, o.width, y_size), outline=0, fill=o.blue)
		
	def reset_cycles(self):
		self.cycles = []
		for l in self.labels:
			self.cycles.append(cycle(l))

	def Exit(self):
		self.FSM.owner.pub.unregister("beat", self)        

#==================================== 

class LoadSound(State):
	def __init__(self,FSM):
		super(LoadSound, self).__init__(FSM)    
		
	def Enter(self):
		o = self.FSM.owner
		o.header_text = "Load Sound"
		o.pub.register("beat", self)
		self.cur_el = 0
		

		self.menu = []

		self.menu.append("..")
		self.menu += self.get_file_list(o)
		self.last_preview = None

		if o.keyState[16] > 0 or o.keyState[17] > 0:
			pass
		else:
			draw_header(o)
			self.draw_main(o)			
			
			o.update_display(0)
		super(LoadSound, self).Enter()    

	def Execute(self):
		o = self.FSM.owner 
		menu1_actions(self, o)
		if o.keyState[16] == 1:
			draw_menu1(o)
			o.update_display(0)
		elif o.keyState[16] == 4:
			draw_header(o)
			self.draw_footer(o)
			self.draw_main(o)
			o.update_display(0)

		if o.keyState[0] == 1 or o.keyState[0] == 3 or o.keyState[1] == 1 or o.keyState[1] == 3 or o.keyState[2] == 1 or o.keyState[2] == 3 or o.keyState[3] == 1 or o.keyState[3] == 3:
			draw_header(o)
			self.draw_footer(o)
			self.draw_main(o)
			o.update_display(0)	
			#print('draw butt')				

		if o.keyState[16] > 0 or o.keyState[17] > 0:
			pass
		else:
			if o.keyState[0] == 1:
				if self.cur_el > 0:
					self.cur_el -= 1
					# o.mconf['theme'] = self.menu[self.cur_el]					
					# o.save_config()
					# o.load_config()
					# o.apply_theme()
					#print('cur el down is ', self.cur_el)

					#o.applyTheme
			if o.keyState[1] == 1:
				if self.cur_el < (len(self.menu) - 1):
					self.cur_el += 1
					# o.mconf['theme'] = self.menu[self.cur_el]					
					# o.save_config()
					# o.load_config()
					# o.apply_theme()
					#print('cur el up is ', self.cur_el)
					#print('loading theme ', self.menu[self.cur_el], '-- ', o.mconf['theme'])
			if o.keyState[2] == 1:
				if self.is_wav():
					if self.last_preview == self.menu[self.cur_el]:
						pass
					else:
						snd = pygame.mixer.Sound(o.load_song_dir + self.menu[self.cur_el])
						pygame.mixer.Sound.play(snd) 

			if o.keyState[3] == 1:
				if self.menu[self.cur_el] == '..':
					print('directory up')
					if o.load_song_dir == '/home/pi/zpc_ct/sounds/':
						print('root dir - cant go up')
					else:
						o.load_song_dir = os.path.dirname(os.path.dirname(o.load_song_dir))
						o.load_song_dir += '/'
						print('new up dir is ', o.load_song_dir)
						self.menu = []
						self.menu.append("..")
						self.menu += self.get_file_list(o)

						self.cur_el = 0

				elif os.path.isdir(o.load_song_dir + self.menu[self.cur_el]):
					print('this is a directory ', o.load_song_dir + self.menu[self.cur_el])
					o.load_song_dir = o.load_song_dir + self.menu[self.cur_el] + '/'
					self.menu = []
					self.menu.append("..")
					self.menu += self.get_file_list(o)
					self.cur_el = 0

				else:
					print('this is not a directory ', o.load_song_dir + self.menu[self.cur_el])
					o.soundSlots[o.eSound].file = o.load_song_dir + self.menu[self.cur_el]
					o.soundSlots[o.eSound].volume = 16
					o.soundSlots[o.eSound].pitch = 0
					o.soundSlots[o.eSound].create_sound()
					o.soundSlots[o.eSound].set_pitch()
				#pass

	def get_file_list(self, o):
		#root, dirs, files = os.walk(o.load_song_dir).next()
		dirs = os.listdir(o.load_song_dir)
		#print('got file list')
		#print(root)
		#print(dirs)
		dirs.sort()
		#print(files)

		return dirs

	def draw_main(self, o):
		al = []
		ypos = 30
		#print(self.menu)
		if self.cur_el == 0:
			_iter = 0
			while len(al) < 4:
				if _iter < len(self.menu):
					#print('appending to al', _iter, ' ', len(self.menu))
					al.append(self.menu[_iter])
				else:
					break
				_iter += 1
		elif self.cur_el == 1:
			al.append(self.menu[self.cur_el - 1])
			al.append(self.menu[self.cur_el])
			if (self.cur_el + 1) < len(self.menu):
				al.append(self.menu[self.cur_el + 1])
			if (self.cur_el + 2) < len(self.menu):
				al.append(self.menu[self.cur_el + 2])
		elif self.cur_el == (len(self.menu) - 2):
			al.append(self.menu[self.cur_el - 2])
			al.append(self.menu[self.cur_el - 1])
			al.append(self.menu[self.cur_el])
			al.append(self.menu[self.cur_el + 1])
		elif self.cur_el == (len(self.menu) - 1):
			al.append(self.menu[self.cur_el - 3])
			al.append(self.menu[self.cur_el - 2])
			al.append(self.menu[self.cur_el - 1])
			al.append(self.menu[self.cur_el])
		else:
			al.append(self.menu[self.cur_el - 1])
			al.append(self.menu[self.cur_el])
			al.append(self.menu[self.cur_el + 1])
			al.append(self.menu[self.cur_el + 2])
		_iter = 0 
		#print(al)
		for i in al:
			if i == self.menu[self.cur_el]:
			#if _iter == self.cur_el:
				o.draw.rectangle((0, ypos, o.width, ypos + 20), outline=0, fill=o.light_grey)
				o.center_block(i, o.h2, [0, ypos, o.width,ypos + 20], o.blue)
			else:
				o.draw.rectangle((0, ypos, o.width, ypos + 20), outline=0, fill=o.blue)
				o.center_block(i, o.h2, [0, ypos, o.width,ypos + 20], o.light_grey)
			ypos += 20
			_iter += 1		    

	def draw_footer(self, o):
		_iter = 0
		width = o.width / 4
		xpos = 0
		yposa = 135 - 25
		yposb = 135
		opts = ["Up", "Down", "Play", "Open"]
		while _iter < 4:
			o.draw.rectangle((xpos, yposa, xpos + width, yposb), outline=0, fill=o.blue)
			
			if o.keyState[_iter] == 1:
				o.draw.rectangle((xpos, yposa, xpos + width, yposb), outline=0, fill=o.light_grey)
				o.center_block(opts[_iter], o.h2, [xpos,yposa,xpos + width,yposb], o.blue)
			else:
				if _iter == 2:
					if self.is_wav():
						o.center_block(opts[_iter], o.h2, [xpos,yposa,xpos + width,yposb], o.light_grey)
					else:
						o.center_block(opts[_iter], o.h2, [xpos,yposa,xpos + width,yposb], o.dark_grey)	
				else:
					o.center_block(opts[_iter], o.h2, [xpos,yposa,xpos + width,yposb], o.light_grey)	
			
			xpos += width
			_iter += 1
		
	def is_wav(self):
		if len(self.menu[self.cur_el]) > 3 and (self.menu[self.cur_el][-1] == 'V' or self.menu[self.cur_el][-1] == 'v') and (self.menu[self.cur_el][-2] == 'A' or self.menu[self.cur_el][-2] == 'a'):
			return True
		else:
			return False

	def ReceiveMessage(self, message):
		o = self.FSM.owner 
		play_seq(o, message)
		
	def Exit(self):
		self.FSM.owner.pub.unregister("beat", self)