# Brassica rapa extraterrestrialus, the common space turnip
+import pymunk
+import pymunk.pygame_util
+import pygame.display
+
from .loader import loader
+from .constants import TURNIP_CATEGORY, LIGHT_CATEGORY
import random
+TURNIP_FILTER = pymunk.ShapeFilter(
+ mask=pymunk.ShapeFilter.ALL_MASKS ^ LIGHT_CATEGORY,
+ categories=TURNIP_CATEGORY)
+
+
+class TurnipInvalidPosition(Exception):
+ pass
+
+
class Turnip(object):
def __init__(self, **kwargs):
self._age = kwargs.get('age', 0)
self._pos = kwargs.get('pos', (0, 0))
+ space = kwargs.get('space', None)
self._update_image()
+ self._body = pymunk.Body(0, 0, pymunk.Body.STATIC)
+ self._shape = pymunk.Circle(self._body, 16)
+ self._shape.filter = TURNIP_FILTER
+ self._body.position = pymunk.pygame_util.from_pygame(
+ self._pos, pygame.display.get_surface())
+ if space.shape_query(self._shape):
+ raise TurnipInvalidPosition()
+ space.add(self._body, self._shape)
def _update_image(self):
self._image = loader.load_image('32', 'turnip%d.png' % (self._age + 1))
def serialize(self):
return {'age': self._age, 'pos': self._pos}
+ def remove(self):
+ # FIXME: Remove body from the space
+ pass
+
def grow(self):
self._age += 1
if self._age >= 3: