From e0b312e95e8f1be81d03dd9016441578b4637fc8 Mon Sep 17 00:00:00 2001
From: Kevin Ngo <me@ngokevin.com>
Date: Fri, 12 Oct 2018 03:51:04 -0700
Subject: [PATCH] fix clearing walls on restart/pause

---
 src/components/beat-loader.js | 8 +++++++-
 src/components/wall.js        | 4 ++++
 2 files changed, 11 insertions(+), 1 deletion(-)

diff --git a/src/components/beat-loader.js b/src/components/beat-loader.js
index bc1da2c..48c1a3c 100644
--- a/src/components/beat-loader.js
+++ b/src/components/beat-loader.js
@@ -225,7 +225,13 @@ AFRAME.registerComponent('beat-loader', {
     this.beatsTime = 0;
     this.beatsTimeOffset = undefined;
     for (let i = 0; i < this.beatContainer.children.length; i++) {
-      this.beatContainer.children[i].components.beat.returnToPool(true);
+      let child = this.beatContainer.children[i];
+      if (child.components.beat) {
+        child.components.beat.returnToPool(true);
+      }
+      if (child.components.wall) {
+        child.components.wall.returnToPool(true);
+      }
     }
   }
 });
diff --git a/src/components/wall.js b/src/components/wall.js
index 4c593df..9a5394b 100644
--- a/src/components/wall.js
+++ b/src/components/wall.js
@@ -7,6 +7,10 @@ AFRAME.registerComponent('wall', {
     this.maxZ = 10;
   },
 
+  pause: function () {
+    this.el.object3D.visible = false;
+  },
+
   tock: function (time, delta) {
     this.el.object3D.position.z += this.data.speed * (delta / 1000);
     this.returnToPool();