fix(TimeFloatImage): Enable init state for viewer

Kitware · Sep 30, 2017 · 2b85fac · 2b85fac
1 parent 8dd6329
commit 2b85fac
Show file tree

Hide file tree

Showing 4 changed files with 130 additions and 137 deletions.
diff --git a/dist/viewer.js b/dist/viewer.js
diff --git a/lib/types/TimeFloatImage.js b/lib/types/TimeFloatImage.js
@@ -28,6 +28,8 @@ export default function build({ viewer, dataType }) {
     return false;
   }
 
+  (viewer.queryDataModel.originalData.arguments_active || []).forEach(name => viewer.queryDataModel.setAnimationFlag(name, true));
+
   // No Magic Lens for us
   viewer.allowMagicLens = false;
 
@@ -146,6 +148,7 @@ export default function build({ viewer, dataType }) {
       chartBuilder: React.PropTypes.object,
       imageBuilder: React.PropTypes.object,
       queryDataModel: React.PropTypes.object,
+      extendedProps: React.PropTypes.object,
     },
 
     render() {
@@ -159,18 +162,24 @@ export default function build({ viewer, dataType }) {
           >
             <PlotlyRenderer chartBuilder={this.props.chartBuilder} ref={c => (this.plotly = c)} />
           </OverlayWindow>
-          <GenericViewer imageBuilder={this.props.imageBuilder} queryDataModel={this.props.queryDataModel} />
+          <GenericViewer
+            imageBuilder={this.props.imageBuilder}
+            queryDataModel={this.props.queryDataModel}
+            {...this.props.extendedProps}
+          />
         </div>);
     },
   });
 
   viewer.ui = 'ReactComponent';
+  const extendedProps = Object.assign({}, viewer.queryDataModel.originalData.initialState);
   viewer.component = (
     <Component
       ref={c => (viewer.instance = c)}
       imageBuilder={imageBuilder}
       queryDataModel={viewer.queryDataModel}
       chartBuilder={fakePainterAndChartBuilder}
+      extendedProps={extendedProps}
     />);
 
   // Make sure we trigger a render when the UI is mounted