Package include path fix and add optimization for computing coverage

1 files changed, 3 insertions, 1 deletions

diff --git a/fish_show.py b/fish_show.py
index 6677ddd..84b0516 100644
--- a/fish_show.py
+++ b/fish_show.py
@@ -89,6 +89,7 @@ class Fish(object):
 
 class Light(object):
     """Represent a bridge light and its state."""
+    __slots__=('light', 'color', 'transitioning_to', 'x', 'y', 'step')
     def __init__(self, x, y, light):
         """Store all facts about a light."""
         self.light = light
@@ -112,7 +113,8 @@ class Light(object):
         new_color = new_background
         for fish in fishes:
             coverage = compute_coverage(fish, self.x, self.y)
-            new_color = coverage * fish.color + (1 - coverage) * new_color
+            if coverage > 0.01:
+                new_color = coverage * fish.color + (1 - coverage) * new_color
         self.light.setRGBRaw(
             *map(lambda x: clamp(0, x, 1), new_color.components)
         )