X-Git-Url: https://git.ctpug.org.za/?a=blobdiff_plain;f=tabakrolletjie%2Fobstacles.py;h=81dd9cec8d937c64a649f8067391ad095f9c92ff;hb=aeed77cd9edcf4a13c391a919a033862ff31cbdf;hp=d187a599294de659c54a70a3e08a27ba5692230f;hpb=2e20539ebb1efedfdc675962961c10b6985c09a7;p=tabakrolletjie.git

diff --git a/tabakrolletjie/obstacles.py b/tabakrolletjie/obstacles.py
index d187a59..81dd9ce 100644
--- a/tabakrolletjie/obstacles.py
+++ b/tabakrolletjie/obstacles.py
@@ -1,8 +1,31 @@
 """ Obstacles for light and space mould. """
 
+import pygame.locals as pgl
+
 import pymunk
 import pymunk.pygame_util
 import pygame.draw
+import pygame.surface
+
+from .constants import (SCREEN_SIZE, OBSTACLE_CATEGORY)
+from .loader import loader
+
+OBSTACLE_FILTER = pymunk.ShapeFilter(categories=OBSTACLE_CATEGORY)
+
+
+class ObstacleManager(object):
+    """ Manages a set of obstacles. """
+
+    def __init__(self, space, gamestate):
+        self._space = space
+        self._obstacles = [
+            BaseObstacle.load(cfg) for cfg in gamestate.station["obstacles"]]
+        for obs in self._obstacles:
+            obs.add(self._space)
+
+    def render(self, surface):
+        for obs in self._obstacles:
+            obs.render(surface)
 
 
 class BaseObstacle(object):
@@ -13,6 +36,8 @@ class BaseObstacle(object):
     def add(self, space):
         if self.body.space is not None:
             space.remove(self.body, *self.body.shapes)
+        for shape in self.shapes:
+            shape.filter = OBSTACLE_FILTER
         space.add(self.body, *self.shapes)
 
     def render(self, surface):
@@ -33,10 +58,23 @@ class Wall(BaseObstacle):
     def __init__(self, vertices):
         super(Wall, self).__init__()
         self.shapes.append(pymunk.Poly(self.body, vertices))
+        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:
+                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()
+            self._image.blit(wall_texture, (0, 0), None, pgl.BLEND_RGBA_MULT)
+            
+        return self._image
 
     def render(self, surface):
-        for shape in self.shapes:
-            pygame_poly = [
-                pymunk.pygame_util.to_pygame(v, surface) for v in
-                shape.get_vertices()]
-            pygame.draw.polygon(surface, (0, 0, 0), pygame_poly)
+        surface.blit(self.get_image(), (0, 0), None, 0)