123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167 |
- 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 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/user/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:
- u.draw_header(o)
- self.draw_main(o)
-
- o.update_display(0)
- super(SelTheme, 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)
- 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:
- m.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
- u.play_seq(o, message)
-
- def Exit(self):
- self.FSM.owner.pub.unregister("beat", self)
|