X-Git-Url: https://scm.cri.mines-paristech.fr/git/minwii.git/blobdiff_plain/e4e71a016e3c9457744817ed104b0b497aaa5ddd..baf4f2fe0e04d6b95c428813dd920b597031ec04:/src/app/widgets/cursors.py

diff --git a/src/app/widgets/cursors.py b/src/app/widgets/cursors.py
index 9315211..1e3fc4e 100755
--- a/src/app/widgets/cursors.py
+++ b/src/app/widgets/cursors.py
@@ -20,7 +20,7 @@ class WarpingCursor(pygame.sprite.DirtySprite, EventHandlerMixin):
     @staticmethod
     def _get_theme_images(name) :
         basePath = os.path.abspath(__file__).split(os.path.sep)[:-1]
-        basePath.append('data')
+        basePath.extend(['data', 'cursor'])
         basePath.append(name)
         basePath = os.path.sep.join(basePath)
         images = [f for f in os.listdir(basePath) if os.path.splitext(f)[1] == '.png']
@@ -29,7 +29,6 @@ class WarpingCursor(pygame.sprite.DirtySprite, EventHandlerMixin):
     
     def __init__(self, theme='black', duration=50, blinkMode=True):
         pygame.sprite.DirtySprite.__init__(self)
-        pygame.mouse.set_visible(False)
         imagesPath, images = WarpingCursor._get_theme_images(theme)
         flashImage = images.pop(images.index('flash.png'))
         flashImagePath = os.path.sep.join([imagesPath, flashImage]) 
@@ -48,17 +47,29 @@ class WarpingCursor(pygame.sprite.DirtySprite, EventHandlerMixin):
         self.duration = duration
         
         self.image = self.images[0]
-        self.rect = pygame.Rect((-self.width/2,-self.height/2), (self.width, self.height))
+        # workarround cursor alignement problem
+        pygame.event.set_blocked(pygame.MOUSEMOTION)
+        pygame.mouse.set_pos(pygame.mouse.get_pos())
+        pygame.event.set_allowed(pygame.MOUSEMOTION)
+        # ---
+        x, y = pygame.mouse.get_pos()
+        left = x - self.width / 2
+        top = y - self.height / 2
+        self.rect = pygame.Rect((left, top), (self.width, self.height))
         
         self.blinkMode = blinkMode
         self._startBlink()
-            
+    
     def _startBlink(self) :
         if self.blinkMode :
             self._blinking = True
             pygame.time.set_timer(TIMEOUT, self.duration)
             self.iterator = self.iterImages()
     
+    def _stopBlink(self) :
+        if self.blinkMode :
+            pygame.time.set_timer(TIMEOUT, 0)
+            
     def iterImages(self) :
         for img in cycle(self.images) :
             yield img
@@ -88,3 +99,9 @@ class WarpingCursor(pygame.sprite.DirtySprite, EventHandlerMixin):
     def move(self, event) :
         self.dirty = 1
         self.rect.move_ip(event.rel)
+    
+    def setPosition(self, pos) :
+        self.dirty = 1
+        x, y = pos
+        rx, ry = self.rect.centerx, self.rect.centery
+        self.rect.move_ip(x-rx, y-ry)