X-Git-Url: https://git.ctpug.org.za/?a=blobdiff_plain;f=naja%2Fwidgets%2Fselector.py;h=dcbc74498b930b3e4d2b31aa6318eae7d59d6b57;hb=d66e2b06ae1d22ec3908177fad044c494d458642;hp=c0c91b96e8ec584fbc76e437c5605d70dd11a568;hpb=2a9f5c76556f00f631ea10a6cba44a92bdd27512;p=naja.git diff --git a/naja/widgets/selector.py b/naja/widgets/selector.py index c0c91b9..dcbc744 100644 --- a/naja/widgets/selector.py +++ b/naja/widgets/selector.py @@ -1,8 +1,10 @@ import pygame.locals as pgl +from naja.constants import KEYS from naja.widgets.base import Container from naja.resources import resources -from naja.resources.mutators import R270 +from naja.resources.mutators import EIGHT_BIT, R270 +from naja.sound import sound class SelectorWidget(Container): @@ -10,26 +12,35 @@ class SelectorWidget(Container): super(SelectorWidget, self).__init__(*args, **kwargs) self.position = 0 self.selector = resources.get_image('bits', 'arrow_on.png', - transforms=(R270,)) + transforms=(R270, EIGHT_BIT)) def render(self, surface): super(SelectorWidget, self).render(surface) pos = self.selector.get_rect() selected = self.widgets[self.position] + if not selected.is_selectable(): + selected = self.change_pos(1) + pos = pos.move(selected.pos) pos = pos.move(-pos.width * 1.5, (selected.size[1] - pos.height) / 2) surface.blit(self.selector, pos) + def change_pos(self, change): + self.position = (self.position + change) % len(self.widgets) + while not self.widgets[self.position].is_selectable(): + self.position = (self.position + change) % len(self.widgets) + return self.widgets[self.position] + def handle_event(self, ev): if ev.type == pgl.KEYDOWN: - if ev.key in (pgl.K_DOWN, pgl.K_UP): - if ev.key == pgl.K_DOWN: - self.position += 1 + if ev.key in KEYS.UP + KEYS.DOWN: + if ev.key in KEYS.DOWN: + self.change_pos(1) else: - self.position -= 1 - self.position %= len(self.widgets) + self.change_pos(-1) + sound.play_sound('zoop.ogg', volume=0.05) return True - elif ev.key == pgl.K_RETURN: + elif ev.key in KEYS.SELECT: return self.widgets[self.position].callback('click') return super(SelectorWidget, self).handle_event(ev)