X-Git-Url: https://git.ctpug.org.za/?a=blobdiff_plain;f=tabakrolletjie%2Flights.py;h=ff7035abb0cfe09ceafa09ef344907b29114e5d5;hb=758bdeeae4acd227637b9bc269b159ba470babbd;hp=8abb1a6801249125b2cafd51a1b0cf63fccf7bbe;hpb=9a21993be81120b5dbe1f3d6caf4fb8f2cc27404;p=tabakrolletjie.git

diff --git a/tabakrolletjie/lights.py b/tabakrolletjie/lights.py
index 8abb1a6..ff7035a 100644
--- 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 (
@@ -73,6 +74,21 @@ class LightManager(object):
         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)