src/app/widgets/cursors.py

   1 # -*- coding: utf-8 -*-
   2 """
   3 Curseurs winwii
   4
   5 $Id$
   6 $URL$
   7 """
   8
   9 import pygame
  10 import os
  11 from eventutils import EventHandlerMixin, event_handler
  12 from events import TIMEOUT
  13 from itertools import cycle
  14
  15 class WarpingCursor(pygame.sprite.DirtySprite, EventHandlerMixin):
  16     '''
  17     The class for animating the warping cursor
  18     '''
  19
  20     @staticmethod
  21     def _get_theme_images(name) :
  22         basePath = os.path.abspath(__file__).split(os.path.sep)[:-1]
  23         basePath.append('data')
  24         basePath.append(name)
  25         basePath = os.path.sep.join(basePath)
  26         images = [f for f in os.listdir(basePath) if os.path.splitext(f)[1] == '.png']
  27         return basePath, images
  28
  29
  30     def __init__(self, theme='black', duration=50, blinkMode=True):
  31         pygame.sprite.DirtySprite.__init__(self)
  32         imagesPath, images = WarpingCursor._get_theme_images(theme)
  33         flashImage = images.pop(images.index('flash.png'))
  34         flashImagePath = os.path.sep.join([imagesPath, flashImage])
  35         self.flashImage = pygame.image.load(flashImagePath).convert_alpha()
  36         images.sort(lambda a, b : cmp(*[int(os.path.splitext(f)[0]) for f in [a, b]]))
  37
  38         self.images = []
  39         for img in images :
  40             imagePath = os.path.sep.join([imagesPath, img])
  41             img = pygame.image.load(imagePath).convert_alpha()
  42             self.images.append(img)
  43
  44         # assumes that all images have same dimensions
  45         self.width = self.images[0].get_width()
  46         self.height = self.images[0].get_height()
  47         self.duration = duration
  48
  49         self.image = self.images[0]
  50         # workarround cursor alignement problem
  51         pygame.event.set_blocked(pygame.MOUSEMOTION)
  52         pygame.mouse.set_pos(pygame.mouse.get_pos())
  53         pygame.event.set_allowed(pygame.MOUSEMOTION)
  54         # ---
  55         x, y = pygame.mouse.get_pos()
  56         print 'mouse pos :', x, y
  57         left = x - self.width / 2
  58         top = y - self.height / 2
  59         self.rect = pygame.Rect((left, top), (self.width, self.height))
  60
  61         self.blinkMode = blinkMode
  62         self._startBlink()
  63
  64     def _stopBlink(self) :
  65         if self.blinkMode :
  66             pygame.time.set_timer(TIMEOUT, 0)
  67
  68     def _startBlink(self) :
  69         if self.blinkMode :
  70             self._blinking = True
  71             pygame.time.set_timer(TIMEOUT, self.duration)
  72             self.iterator = self.iterImages()
  73
  74     def iterImages(self) :
  75         for img in cycle(self.images) :
  76             yield img
  77
  78     @event_handler(TIMEOUT)
  79     def loadNext(self, event) :
  80         if self._blinking :
  81             self.dirty = 1
  82             self.image = self.iterator.next()
  83
  84     @event_handler(pygame.MOUSEBUTTONDOWN)
  85     def flashOn(self, event) :
  86         self.dirty = 1
  87         self._blinking = False
  88         self.image = self.flashImage
  89
  90     @event_handler(pygame.MOUSEBUTTONUP)
  91     def flashOff(self, event) :
  92         self.dirty = 1
  93         if self.blinkMode :
  94             self._blinking = True
  95             self.loadNext(event)
  96         else :
  97             self.image = self.images[0]
  98
  99     @event_handler(pygame.MOUSEMOTION)
 100     def move(self, event) :
 101         self.dirty = 1
 102         self.rect.move_ip(event.rel)