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 import pygame.transform
   7
   8 from .loader import loader
   9 from .constants import TURNIP_CATEGORY, LIGHT_CATEGORY
  10 import random
  11
  12 TURNIP_FILTER = pymunk.ShapeFilter(
  13     mask=pymunk.ShapeFilter.ALL_MASKS ^ LIGHT_CATEGORY,
  14     categories=TURNIP_CATEGORY)
  15
  16
  17 class TurnipInvalidPosition(Exception):
  18     pass
  19
  20
  21 class Turnip(object):
  22     TURNIP_RADIUS = 12
  23
  24     def __init__(self, **kwargs):
  25         self._age = kwargs.get('age', 0)
  26         self._pos = kwargs.get('pos', (0, 0))
  27         space = kwargs.get('space', None)
  28         self._rotation = 90 * (hash(tuple(self._pos)) % 4)
  29         self._update_image()
  30         self.eaten = False  # Flag for boyd
  31         self._body = pymunk.Body(0, 0, pymunk.Body.STATIC)
  32         self._shape = pymunk.Circle(self._body, self.TURNIP_RADIUS)
  33         self._shape.filter = TURNIP_FILTER
  34         self._body.position = pymunk.pygame_util.from_pygame(
  35             self._pos, pygame.display.get_surface())
  36         # Add a reference so space query can get the object
  37         self._body.turnip = self
  38         if space.shape_query(self._shape):
  39             raise TurnipInvalidPosition()
  40         space.add(self._body, self._shape)
  41
  42     def _update_image(self):
  43         self._image = loader.load_image('48', 'turnip%d.png' % (self._age + 1))
  44         self._image = pygame.transform.rotate(self._image, self._rotation)
  45
  46     def render(self, surface):
  47         rx, ry = self._pos
  48         surface.blit(self._image, (rx - self.TURNIP_RADIUS, ry - self.TURNIP_RADIUS), None)
  49
  50     def serialize(self):
  51         return {'age': self._age, 'pos': self._pos}
  52
  53     def remove(self):
  54         self._body.turnip = None
  55         space = self._body.space
  56         space.remove(self._body, self._shape)
  57
  58     def grow(self):
  59         self._age += 1
  60         if self._age >= 3:
  61             # Mature, so harvest it for seeds
  62             self.remove()
  63             return random.randint(1, 3)
  64         else:
  65             self._update_image()
  66         return None