shrub achieved

[tabakrolletjie.git] / tabakrolletjie / obstacles.py

diff --git a/tabakrolletjie/obstacles.py b/tabakrolletjie/obstacles.py
index 81dd9cec8d937c64a649f8067391ad095f9c92ff..9adb80308180f40a76accc528492d7fe9026c35c 100644 (file)
--- a/tabakrolletjie/obstacles.py
+++ b/tabakrolletjie/obstacles.py
@@ -63,16 +63,43 @@ class Wall(BaseObstacle):
     def get_image(self):
         if self._image is None:
             self._image = pygame.surface.Surface(SCREEN_SIZE).convert_alpha()
-            self._image.fill((0,0,0,0))
-                
+            self._image.fill((0, 0, 0, 0))
+
             for shape in self.shapes:
                 pygame_poly = [
                     pymunk.pygame_util.to_pygame(v, self._image) for v in
                     shape.get_vertices()]
                 pygame.draw.polygon(self._image, (255, 255, 255), pygame_poly)
-                
-            wall_texture = loader.load_image("textures", "stone.png").convert_alpha()
+
+            wall_texture = loader.load_image(
+                "textures", "stone.png").convert_alpha()
             self._image.blit(wall_texture, (0, 0), None, pgl.BLEND_RGBA_MULT)
+
+        return self._image
+
+    def render(self, surface):
+        surface.blit(self.get_image(), (0, 0), None, 0)
+
+
+class Shrub(BaseObstacle):
+
+    def __init__(self, shrublets):
+        super(Shrub, self).__init__()
+        for [x, y, r] in shrublets:
+            self.shapes.append(pymunk.Circle(self.body, r, offset=(x, y)))
+        self._image = None
+
+    def get_image(self):
+        if self._image is None:
+            self._image = pygame.surface.Surface(SCREEN_SIZE).convert_alpha()
+            self._image.fill((0,0,0,0))
+                
+            for shape in self.shapes:
+                centre = pymunk.pygame_util.to_pygame(shape.offset, self._image)
+                pygame.draw.circle(self._image, (255, 255, 255), centre, int(shape.radius))
+                
+            shrub_texture = loader.load_image("textures", "shrub.png").convert_alpha()
+            self._image.blit(shrub_texture, (0, 0), None, pgl.BLEND_RGBA_MULT)
             
         return self._image