From 5df1159bec104dc6541a186bca0770a85a13bc68 Mon Sep 17 00:00:00 2001
From: group1 <f16_group1@pbridge.adm.cs.cmu.edu>
Date: Sun, 16 Oct 2016 22:19:49 -0400
Subject: Go faster

---
 fish_show.py | 43 ++++++++++++++++++++++++++++++-------------
 1 file changed, 30 insertions(+), 13 deletions(-)

diff --git a/fish_show.py b/fish_show.py
index e410895..2f4716b 100644
--- a/fish_show.py
+++ b/fish_show.py
@@ -11,17 +11,18 @@ from fish_types import Fish
 
 class Light(object):
     """Represent a bridge light and its state."""
-    __slots__=('light', 'color', 'transitioning_to', 'x', 'y', 'step')
+    __slots__=('light', 'color', 'transitioning_to', 'x', 'y', 'step', 'result_color')
     def __init__(self, x, y, light):
         """Store all facts about a light."""
         self.light = light
         self.color = color.black
+        self.result_color = self.color
         self.transitioning_to = self.color
         self.step = 0
         self.x = x
         self.y = y
 
-    def update(self, depth, fishes):
+    def update_base(self, depth):
         """Change color depending on where the fish is."""
         if abs(self.transitioning_to - self.color) <= self.step:
             self.transitioning_to = pick_HSV(depth).to_RGB()
@@ -32,16 +33,20 @@ class Light(object):
         )
 
         self.color = new_background
-        new_color = new_background
-        for fish in fishes:
-            coverage = compute_coverage(fish, self.x, self.y)
-            if coverage > 0.01:
-                fish_color = fish.color_at(self.x, self.y)
-                new_color = coverage * fish_color + (1 - coverage) * new_color
+        self.result_color = new_background
+
+    def update_fish(self, fish):
+        coverage = compute_coverage(fish, self.x, self.y)
+        fish_color = fish.color_at(self.x, self.y)
+        self.result_color = coverage * fish_color + (1 - coverage) * self.result_color
+
+    def blit(self):
         self.light.setRGBRaw(
-            *map(lambda x: clamp(0, x, 1), new_color.components)
+            *map(lambda x: clamp(0, x, 1), self.result_color.components)
         )
 
+        
+
 def compute_coverage(fish, x, y):
     """Calculate the coverage of a fish over a box."""
     middle_of_fish = fish.x + fish.width / 2
@@ -93,8 +98,8 @@ class FishShow(Show):
     MAX_DEPTH = 6
 
     def init(self):
-        self.lights = [Light(x, y, self.bridge.get_light(x, y))
-                       for y in xrange(self.bridge.HEIGHT)
+        self.lights = [[Light(x, y, self.bridge.get_light(x, y))
+                       for y in xrange(self.bridge.HEIGHT)]
                        for x in xrange(self.bridge.WIDTH)]
         self.fishes = set()
         self.tick_count = 0
@@ -106,11 +111,23 @@ class FishShow(Show):
             new_fish = Fish.by_type(event[1]['type'])
             self.fishes.add(new_fish)
 
+        for col in self.lights:
+            for light in col:
+                light.update_base(self.depth)
+
         for fish in list(self.fishes):
             if not fish.update():
                 self.fishes.remove(fish)
-        for light in self.lights:
-            light.update(self.depth, self.fishes)
+
+            for x in xrange(int(fish.x), int(fish.x+fish.width+1)):
+                if x < len(self.lights):
+                    for light in self.lights[x]:
+                        light.update_fish(fish)
+
+        for col in self.lights:
+            for light in col:
+                light.blit()
+
 
         self.tick_count += 1
         if self.tick_count % self.TICKS_PER_DEPTH == 0:
-- 
cgit v1.3.1