fix eye icon on load instance

VirtualFlyBrain · Dec 6, 2019 · a8bbba4 · a8bbba4
1 parent fdc722e
commit a8bbba4
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/components/interface/TreeWidget.js b/components/interface/TreeWidget.js
@@ -421,7 +421,7 @@ export default class TreeWidget extends React.Component {
               <div> {rowInfo.node.description} </div>
             </div>)}>
           <div
-            className={(rowInfo.node.instanceId === this.state.nodeSelected.instanceId)
+            className={(this.state.nodeSelected !== undefined && rowInfo.node.instanceId === this.state.nodeSelected.instanceId)
               ? "nodeFound nodeSelected"
               : "nodeSelected"}
             onClick={ e => {
@@ -462,6 +462,7 @@ export default class TreeWidget extends React.Component {
   componentDidMount () {
     var that = this;
     document.addEventListener('mousedown', this.monitorMouseClick, false);
+
     GEPPETTO.on(GEPPETTO.Events.Select, function (instance) {
       that.updateTree(instance);
     });
@@ -471,6 +472,10 @@ export default class TreeWidget extends React.Component {
         that.setState({ nodeSelected: undefined });
       }
     });
+
+    GEPPETTO.on(GEPPETTO.Events.Instances_created, function () {
+      that.setState({ displayColorPicker: false });
+    });
   }
 
   render () {