1 files changed, 21 insertions, 6 deletions
diff --git a/public/assets/javascripts/rectangles/engine/rooms/_walls.js b/public/assets/javascripts/rectangles/engine/rooms/_walls.js
index f0cd558..7438cd4 100644
--- a/public/assets/javascripts/rectangles/engine/rooms/_walls.js
+++ b/public/assets/javascripts/rectangles/engine/rooms/_walls.js
@@ -35,6 +35,7 @@
 		var base = this
 		
 		base.list = []
+		base.floors = []
 		base.lookup = {}
 		base.colors = {}
 
@@ -57,11 +58,21 @@
 		    base.lookup[wall.id] = wall
 		  })
 		}
+
+		base.assignFloors = function(floors){
+		  base.floors = floors
+		  floors.forEach(function(floor){
+		    base.lookup[floor.id] = floor
+		  })
+		}
 		
 		base.bind = function(){
 			base.list.forEach(function(wall){
 				wall.bind()
 			})
+			base.floors.forEach(function(floor){
+			  floor.bind()
+			})
 		}
 
 		base.count = function(){
@@ -69,16 +80,20 @@
 		}
 	
 		base.forEach = function(f){
-			return base.list.forEach(f)
+			return base.values().forEach(f)
 		}
 	
 		base.map = function(f){
-			return base.list.map(f)
+			return base.values().map(f)
+		}
+	
+		base.values = function(){
+			return _.values(base.lookup)
 		}
 		
 		base.serialize = function(){
 		  var data = []
-			base.list.forEach(function(wall){
+			base.forEach(function(wall){
 			  data.push(wall.serialize())
 			})
 			return data
@@ -97,7 +112,7 @@
       var outlineColor = rgb_string(Walls.colors.outline)
       var floorColor = rgb_string(Walls.colors.floor)
       var ceilingColor = rgb_string(Walls.colors.ceiling)
-      Walls.forEach(function(wall){
+      Walls.list.forEach(function(wall){
         wall.outline(wallColor, outlineColor)
       })
       Rooms.forEach(function(room){
@@ -111,7 +126,7 @@
       wall: function(rgb){
         var rgbaColor = rgba_string(rgb, app.defaults.wallOpacity)
         Walls.colors.wall = rgb
-        Walls.forEach(function(wall){
+        Walls.list.forEach(function(wall){
           wall.outline(rgbaColor, null)
         })
       },
@@ -119,7 +134,7 @@
       outline: function(rgb){
         var rgbColor = rgb_string(rgb)
         Walls.colors.outline = rgb
-        Walls.forEach(function(wall){
+        Walls.list.forEach(function(wall){
           wall.outline(null, rgbColor)
         })
       },