Merge branch 'master' of ctpug.org.za:tabakrolletjie

[tabakrolletjie.git] / tabakrolletjie / lights.py

diff --git a/tabakrolletjie/lights.py b/tabakrolletjie/lights.py
index 2590e310bc3247680ce13eef5f2bea7b1fdc7964..ff7035abb0cfe09ceafa09ef344907b29114e5d5 100644 (file)
--- a/tabakrolletjie/lights.py
+++ b/tabakrolletjie/lights.py
@@ -3,6 +3,7 @@
 
 import pymunk
 import pymunk.pygame_util
+import pygame.display
 import pygame.draw
 
 from .constants import (
@@ -63,6 +64,40 @@ def calculate_ray_polys(space, body, position):
     return ray_polys
 
 
+class LightManager(object):
+    """ Manages a set of lights. """
+
+    def __init__(self, space, gamestate):
+        self._space = space
+        self._lights = [
+            BaseLight.load(cfg) for cfg in gamestate.station["lights"]]
+        for light in self._lights:
+            light.add(self._space)
+
+    def toggle_nearest(self, *args, **kw):
+        light = self.nearest(*args, **kw)
+        if light:
+            light.toggle()
+
+    def nearest(self, pos, surfpos=True, max_distance=1.0):
+        if surfpos:
+            surface = pygame.display.get_surface()
+            pos = pymunk.pygame_util.from_pygame(pos, surface)
+        point_info = self._space.point_query_nearest(
+            pos, 1.0, pymunk.ShapeFilter(mask=FITTINGS_CATEGORY))
+        if point_info is not None:
+            return point_info.shape.body.light
+        return None
+
+    def render_light(self, surface):
+        for light in self._lights:
+            light.render_light(surface)
+
+    def render_fittings(self, surface):
+        for light in self._lights:
+            light.render_fitting(surface)
+
+
 class BaseLight(object):
     """ Common light functionality. """
 
@@ -125,7 +160,7 @@ class BaseLight(object):
         subsurface.set_alpha(50)
         surface.blit(subsurface, (0, 0), None)
 
-    def render_fittings(self, surface):
+    def render_fitting(self, surface):
         pygame.draw.circle(
             surface, (255, 255, 0),
             pymunk.pygame_util.to_pygame(self.fitting.offset, surface),