Display power usage per hour.

[tabakrolletjie.git] / tabakrolletjie / turnip.py

diff --git a/tabakrolletjie/turnip.py b/tabakrolletjie/turnip.py
index 5f4e8c52f2273651f349281acbae1b42f6e8284e..72705fccdf7aa18e3548357ffc14954e0d9a41a8 100644 (file)
--- a/tabakrolletjie/turnip.py
+++ b/tabakrolletjie/turnip.py
@@ -3,6 +3,7 @@
 import pymunk
 import pymunk.pygame_util
 import pygame.display
+import pygame.transform
 
 from .loader import loader
 from .constants import TURNIP_CATEGORY, LIGHT_CATEGORY
@@ -13,6 +14,14 @@ TURNIP_FILTER = pymunk.ShapeFilter(
     categories=TURNIP_CATEGORY)
 
 
+def check_turnips(space, pos, max_distance):
+    point_info = space.point_query_nearest(
+        pos, max_distance, pymunk.ShapeFilter(mask=TURNIP_CATEGORY))
+    if point_info is not None:
+        return True
+    return False
+
+
 class TurnipInvalidPosition(Exception):
     pass
 
@@ -23,10 +32,11 @@ class Turnip(object):
         self._age = kwargs.get('age', 0)
         self._pos = kwargs.get('pos', (0, 0))
         space = kwargs.get('space', None)
+        self._rotation = 90 * (hash(self._pos) % 4)
         self._update_image()
         self.eaten = False  # Flag for boyd
         self._body = pymunk.Body(0, 0, pymunk.Body.STATIC)
-        self._shape = pymunk.Circle(self._body, 16)
+        self._shape = pymunk.Circle(self._body, 24)
         self._shape.filter = TURNIP_FILTER
         self._body.position = pymunk.pygame_util.from_pygame(
             self._pos, pygame.display.get_surface())
@@ -37,7 +47,8 @@ class Turnip(object):
         space.add(self._body, self._shape)
 
     def _update_image(self):
-        self._image = loader.load_image('32', 'turnip%d.png' % (self._age + 1))
+        self._image = loader.load_image('48', 'turnip%d.png' % (self._age + 1))
+        self._image = pygame.transform.rotate(self._image, self._rotation)
 
     def render(self, surface):
         surface.blit(self._image, self._pos, None)