tabakrolletjie/turnip.py

   1 # Brassica rapa extraterrestrialus, the common space turnip
   2
   3 import pymunk
   4 import pymunk.pygame_util
   5 import pygame.display
   6
   7 from .loader import loader
   8 from .constants import TURNIP_CATEGORY, LIGHT_CATEGORY
   9 import random
  10
  11 TURNIP_FILTER = pymunk.ShapeFilter(
  12     mask=pymunk.ShapeFilter.ALL_MASKS ^ LIGHT_CATEGORY,
  13     categories=TURNIP_CATEGORY)
  14
  15
  16 class TurnipInvalidPosition(Exception):
  17     pass
  18
  19
  20 class Turnip(object):
  21
  22     def __init__(self, **kwargs):
  23         self._age = kwargs.get('age', 0)
  24         self._pos = kwargs.get('pos', (0, 0))
  25         space = kwargs.get('space', None)
  26         self._update_image()
  27         self._body = pymunk.Body(0, 0, pymunk.Body.STATIC)
  28         self._shape = pymunk.Circle(self._body, 16)
  29         self._shape.filter = TURNIP_FILTER
  30         self._body.position = pymunk.pygame_util.from_pygame(
  31             self._pos, pygame.display.get_surface())
  32         if space.shape_query(self._shape):
  33             raise TurnipInvalidPosition()
  34         space.add(self._body, self._shape)
  35
  36     def _update_image(self):
  37         self._image = loader.load_image('32', 'turnip%d.png' % (self._age + 1))
  38
  39     def render(self, surface):
  40         surface.blit(self._image, self._pos, None)
  41
  42     def serialize(self):
  43         return {'age': self._age, 'pos': self._pos}
  44
  45     def remove(self):
  46         # FIXME: Remove body from the space
  47         pass
  48
  49     def grow(self):
  50         self._age += 1
  51         if self._age >= 3:
  52             # Mature, so harvest it for seeds
  53             return random.randint(1, 3)
  54         else:
  55             self._update_image()
  56         return None