Fixing graph reset (#377)

Author: antoninklopp

src/components/graph/Graph.jsx

@@ -16,6 +16,7 @@ import {
   checkForGraphElementsChanges,
   getCenterAndZoomTransformation,
   initializeGraphState,
+  initializeNodes,
 } from "./graph.helper";
 import { renderGraph } from "./graph.renderer";
 import { merge, debounce, throwErr } from "../../utils";
@@ -478,13 +479,20 @@ export default class Graph extends React.Component {
    */
   resetNodesPositions = () => {
     if (!this.state.config.staticGraph) {
+      let initialNodesState = initializeNodes(this.props.data.nodes);
       for (let nodeId in this.state.nodes) {
         let node = this.state.nodes[nodeId];
 
         if (node.fx && node.fy) {
           Reflect.deleteProperty(node, "fx");
           Reflect.deleteProperty(node, "fy");
         }
+
+        if (nodeId in initialNodesState) {
+          let initialNode = initialNodesState[nodeId];
+          node.x = initialNode.x;
+          node.y = initialNode.y;
+        }
       }
 
       this.state.simulation.alphaTarget(this.state.config.d3.alphaTarget).restart();

src/components/graph/graph.helper.js

@@ -102,7 +102,7 @@ function _initializeLinks(graphLinks, config) {
  * and highlighted values.
  * @memberof Graph/helper
  */
-function _initializeNodes(graphNodes) {
+function initializeNodes(graphNodes) {
   let nodes = {};
   const n = graphNodes.length;
 
@@ -394,7 +394,7 @@ function initializeGraphState({ data, id, config }, state) {
 
   let newConfig = { ...merge(DEFAULT_CONFIG, config || {}) },
     links = _initializeLinks(graph.links, newConfig), // matrix of graph connections
-    nodes = _tagOrphanNodes(_initializeNodes(graph.nodes), links);
+    nodes = _tagOrphanNodes(initializeNodes(graph.nodes), links);
   const { nodes: d3Nodes, links: d3Links } = graph;
   const formatedId = id.replace(/ /g, "_");
   const simulation = _createForceSimulation(newConfig.width, newConfig.height, newConfig.d3 && newConfig.d3.gravity);
@@ -561,4 +561,5 @@ export {
   initializeGraphState,
   updateNodeHighlightedValue,
   getNormalizedNodeCoordinates,
+  initializeNodes,
 };