From 921aad9e2ff824afca0e410ab0aaffd8a8f60460 Mon Sep 17 00:00:00 2001
From: Victor Woeltjen <victor.woeltjen@nasa.gov>
Date: Fri, 12 Dec 2014 15:14:20 -0800
Subject: [PATCH] [Plot] Update plot immediately on mode change

Make sure current telemetry is drawn immediately upon
switching between Overlaid and Stacked plot modes.
WTD-625.
---
 platform/features/plot/src/PlotController.js | 17 ++++++++++-------
 1 file changed, 10 insertions(+), 7 deletions(-)

diff --git a/platform/features/plot/src/PlotController.js b/platform/features/plot/src/PlotController.js
index cca3a15f30..2cc77ccb8f 100644
--- a/platform/features/plot/src/PlotController.js
+++ b/platform/features/plot/src/PlotController.js
@@ -87,6 +87,13 @@ define(
                 modeOptions = new PlotModeOptions(telemetryObjects || []);
             }
 
+            // Update all sub-plots
+            function update() {
+                modeOptions.getModeHandler()
+                    .getSubPlots()
+                    .forEach(updateSubplot);
+            }
+
             $scope.$watch("telemetry.getTelemetryObjects()", setupModes);
             $scope.$watch("telemetry.getMetadata()", setupAxes);
             $scope.$on("telemetryUpdate", plotTelemetry);
@@ -143,7 +150,8 @@ define(
                  *        getModeOptions()
                  */
                 setMode: function (mode) {
-                    return modeOptions.setMode(mode);
+                    modeOptions.setMode(mode);
+                    plotTelemetry();
                 },
                 /**
                  * Get all individual plots contained within this Plot view.
@@ -156,12 +164,7 @@ define(
                 /**
                  * Explicitly update all plots.
                  */
-                update: function () {
-                    modeOptions.getModeHandler()
-                        .getSubPlots()
-                        .forEach(updateSubplot);
-                }
-
+                update: update
             };
         }