Update stage to handle switching between documents.

Signed-off-by: Nivesh Rajbhandari <mqg734@motorola.com>

1 files changed, 13 insertions, 0 deletions

diff --git a/js/stage/stage.reel/stage.js b/js/stage/stage.reel/stage.js
index a3903ac7..ad05f1cc 100755
--- a/js/stage/stage.reel/stage.js
+++ b/js/stage/stage.reel/stage.js
@@ -248,6 +248,7 @@ exports.Stage = Montage.create(Component, {
 
 
             this.eventManager.addEventListener( "openDocument", this, false);
+            this.eventManager.addEventListener( "switchDocument", this, false);
             this.eventManager.addEventListener( "enableStageMove", this, false);
             this.eventManager.addEventListener( "disableStageMove", this, false);
 
@@ -261,6 +262,18 @@ exports.Stage = Montage.create(Component, {
     // Event details will contain the active document prior to opening a new one
     handleOpenDocument: {
         value: function(evt) {
+            this._initFromDocument(evt);
+        }
+    },
+
+    handleSwitchDocument: {
+        value: function(evt) {
+            this._initFromDocument(evt);
+        }
+    },
+
+    _initFromDocument: {
+        value: function(evt) {
             var designView = this.application.ninja.currentDocument.model.views.design;
 
             this.hideCanvas(false);