123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195 |
- import FSM
- import utils as u
- import menus as m
- import pygame
- from itertools import cycle
- import glob
- import os
-
- #====================================
-
- State = type("State", (object,), {})
- #====================================
- class State(object):
- def __init__(self, FSM):
- self.FSM = FSM
- def Enter(self):
- pass
- def Execute(self):
- pass
- def Exit(self):
- pass
-
- 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()
- files = u.get_files(o.song_dir, '.sng')
-
- files_trimmed = [os.path.basename(x) for x in files]
- files_trimmed = [os.path.splitext(x)[0] for x in files_trimmed]
- if '/home/pi/zpc_ct/user/songs/_blank.sng' in self.globbed:
- self.globbed.remove('/home/pi/zpc_ct/user/songs/_blank.sng')
-
- 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)
-
- self.menu = self.names
- files_trimmed.insert(0, '..')
- self.menu = files_trimmed
- if '_blank' in self.menu:
- self.menu.remove('_blank')
- if o.keyState[16] > 0 or o.keyState[17] > 0:
- pass
- else:
- u.draw_header(o)
- self.draw_main(o)
- o.update_display(0)
-
- super(OpenSong, self).Enter()
-
- def Execute(self):
- o = self.FSM.owner
- m.menu1_actions(self, o)
- if o.keyState[16] == 1:
- m.draw_menu1(o)
- o.update_display(0)
- elif o.keyState[16] == 4:
- u.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:
- u.draw_header(o)
- self.draw_footer(o)
- self.draw_main(o)
- o.update_display(0)
-
-
- if o.keyState[16] > 0 or o.keyState[17] > 0:
- pass
- else:
- u.scroll_ud(o, self)
- if o.keyState[3] == 1:
- if self.menu[self.cur_el] == '..':
- if o.song_dir == o.base_song_dir:
- pass
- else:
- o.song_dir = os.path.dirname(os.path.dirname(o.song_dir))
- o.song_dir += '/'
- files = u.get_files(o.song_dir, '.sng')
- files_trimmed = [os.path.basename(x) for x in files]
- files_trimmed = [os.path.splitext(x)[0] for x in files_trimmed]
- files_trimmed.insert(0, '..')
- self.menu = files_trimmed
- if '_blank' in self.menu:
- self.menu.remove('_blank')
- u.draw_header(o)
- self.draw_footer(o)
- self.draw_main(o)
- o.update_display(0)
-
- elif os.path.isdir(o.song_dir + self.menu[self.cur_el]):
- o.song_dir += self.menu[self.cur_el] + '/'
- files = u.get_files(o.song_dir, '.sng')
- files_trimmed = [os.path.basename(x) for x in files]
- files_trimmed = [os.path.splitext(x)[0] for x in files_trimmed]
- self.names = [os.path.basename(x) for x in self.globbed]
- self.names = [os.path.splitext(x)[0] for x in self.names]
- self.menu = self.names
- files_trimmed.insert(0, '..')
- self.menu = files_trimmed
- if '_blank' in self.menu:
- self.menu.remove('_blank')
- u.draw_header(o)
- self.draw_footer(o)
- self.draw_main(o)
- o.update_display(0)
-
- else:
- o.mconf['default_song'] = self.menu[self.cur_el]
- o.save_config()
- o.load_song()
-
- def draw_main(self, o):
- al = []
- ypos = 30
- if self.cur_el > (len(self.menu) - 1):
- self.cur_el = 0
- if self.cur_el == 0:
- _iter = 0
- while _iter < 4:
- if _iter < (len(self.menu)):
- 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 os.path.isdir(o.song_dir + i):
- i += "/"
- 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:
- if os.path.isdir(o.song_dir + i):
- i += "/"
- 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
- u.play_seq(o, message)
-
- def Exit(self):
- self.FSM.owner.pub.unregister("beat", self)
|