model-navigator-standalone
Version:
Standalone MDF graph model visualizer
147 lines (143 loc) • 7.33 kB
JavaScript
function ownKeys(e, r) { var t = Object.keys(e); if (Object.getOwnPropertySymbols) { var o = Object.getOwnPropertySymbols(e); r && (o = o.filter(function (r) { return Object.getOwnPropertyDescriptor(e, r).enumerable; })), t.push.apply(t, o); } return t; }
function _objectSpread(e) { for (var r = 1; r < arguments.length; r++) { var t = null != arguments[r] ? arguments[r] : {}; r % 2 ? ownKeys(Object(t), !0).forEach(function (r) { _defineProperty(e, r, t[r]); }) : Object.getOwnPropertyDescriptors ? Object.defineProperties(e, Object.getOwnPropertyDescriptors(t)) : ownKeys(Object(t)).forEach(function (r) { Object.defineProperty(e, r, Object.getOwnPropertyDescriptor(t, r)); }); } return e; }
function _defineProperty(e, r, t) { return (r = _toPropertyKey(r)) in e ? Object.defineProperty(e, r, { value: t, enumerable: !0, configurable: !0, writable: !0 }) : e[r] = t, e; }
function _toPropertyKey(t) { var i = _toPrimitive(t, "string"); return "symbol" == typeof i ? i : i + ""; }
function _toPrimitive(t, r) { if ("object" != typeof t || !t) return t; var e = t[Symbol.toPrimitive]; if (void 0 !== e) { var i = e.call(t, r || "default"); if ("object" != typeof i) return i; throw new TypeError("@@toPrimitive must return a primitive value."); } return ("string" === r ? String : Number)(t); }
import "core-js/modules/es.array.includes.js";
import "core-js/modules/es.string.includes.js";
import "core-js/modules/esnext.iterator.constructor.js";
import "core-js/modules/esnext.iterator.filter.js";
import "core-js/modules/esnext.iterator.for-each.js";
import "core-js/modules/esnext.iterator.map.js";
import "core-js/modules/web.dom-collections.iterator.js";
/* eslint no-unused-vars: 0 */
/* eslint-disable react-hooks/rules-of-hooks */
import React, { useCallback, useEffect, useState, useContext } from 'react';
import { forceCollide, forceCenter, forceSimulation, forceLink, forceManyBody, forceX, forceY } from 'd3-force';
import { useSelector, useDispatch } from 'react-redux';
import { addEdge, useNodesState, useEdgesState, useReactFlow, useStore } from 'reactflow';
import CircularProgress from '@mui/material/CircularProgress';
import _ from 'lodash';
import { collide } from './collide.js';
import { ModelContext } from '../../../Model/ModelContext';
import { ConfigContext } from '../../../Config/ConfigContext';
import CanvasView from './CanvasView';
import { defaultStyleAttributes } from '../../../Config/nav.config';
import { createNodesAndEdges } from '../GraphUtils/MDFutils';
import { setMatchingNodeTitle } from './util';
import { reactFlowGraphInitialized, reactFlowNodeDragStarted, reactFlowPanelClicked, reactFlowGraphViewChanged, reactFlowGraphDataCalculated, selectModelID, selectGraphViewConfig } from '../../../../../features/graph/graphSlice';
import { selectIsSearchMode, selectCurrentSearchKeyword, selectSearchResult } from '../../../../../features/search/searchSlice';
import { selectHiddenNodes } from '../../../../../features/filter/filterSlice';
const defaultIcon = defaultStyleAttributes.graph.icon;
const numTicks = 3; // number of simulation ticks to get initial layout - put in config later
const createNodeCoordinatesSetter = (nodes, edges) => {
// must clone edges, since simulation replaces source and target with
// actual node objects, and we need the simple handles for rendering
// in flowgraph
let sim_nodes = _.cloneDeep(nodes);
let sim_edges = _.cloneDeep(edges);
let simulation = forceSimulation().force('charge', forceManyBody().strength(-1000))
// .force('x', forceX().x(0).strength(0.05))
// .force('y', forceY().y(0).strength(0.05))
// .force('collide', collide())
.force('center', forceCenter()).alphaTarget(0.05).stop();
simulation.nodes(sim_nodes).force('link', forceLink(sim_edges).id(d => d.id).strength(0.01).distance(50));
simulation.tick(numTicks);
const positionByID = _.zipObject(sim_nodes.map(node => node.id), sim_nodes.map(node => ({
x: node.x,
y: node.y
})));
// closure preserves original initted coordinates
return (nodes, edges) => {
nodes.forEach(node => {
node.position.x = positionByID[node.id].x;
node.position.y = positionByID[node.id].y;
});
};
};
let nodeCoordinatesSetter = null;
let flowData = null;
const getLayoutedElements = (nodes, edges, isSearchMode, defaultIcon, searchResult, currentSearchKeyword, hiddenNodes, coordinatesSetter) => {
const nodes_r = _.cloneDeep(nodes.filter(n => !hiddenNodes.includes(n.id)));
const edges_r = _.cloneDeep(edges.filter(ed => !(hiddenNodes.includes(ed.source) || hiddenNodes.includes(ed.target))));
nodes_r.forEach(node => {
if (!node.data.icon) {
node.data.icon = defaultIcon;
}
});
if (isSearchMode) {
const matchingNodeTitle = setMatchingNodeTitle(searchResult);
if (matchingNodeTitle) {
nodes_r.forEach(node => {
if (matchingNodeTitle[node.id]) {
node.data.matchedNodeNameQuery = currentSearchKeyword;
}
});
}
}
coordinatesSetter(nodes_r, edges_r);
return {
nodes_r,
edges_r
};
};
const CanvasController = _ref => {
let {
graphViewConfig,
tabViewWidth,
onClearSearchResult,
highlightedNodes
} = _ref;
const model = useContext(ModelContext);
const config = useContext(ConfigContext);
if (tabViewWidth === 0 || !model) {
// eslint-disable-line no-undef
return /*#__PURE__*/React.createElement(CircularProgress, null);
}
const dispatch = useDispatch();
const isSearchMode = useSelector(selectIsSearchMode);
const searchResult = useSelector(selectSearchResult);
const currentSearchKeyword = useSelector(selectCurrentSearchKeyword);
const hiddenNodes = useSelector(selectHiddenNodes);
// leave in React std form for flowGraph
const [nodes, setNodes, onNodesChange] = useNodesState([]);
const [edges, setEdges, onEdgesChange] = useEdgesState([]);
const modelID = useSelector(selectModelID);
// idea: if model changes, run initialization
let n = 0;
useEffect(() => {
let id = modelID;
flowData = createNodesAndEdges(model, config);
nodeCoordinatesSetter = createNodeCoordinatesSetter(flowData.nodes, flowData.edges);
dispatch(reactFlowGraphInitialized({
graphViewConfig,
model
}));
}, [modelID]);
useEffect(() => {
const {
nodes_r: layoutNodes,
edges_r: layoutEdges
} = getLayoutedElements(flowData.nodes, flowData.edges, isSearchMode, defaultIcon, searchResult, currentSearchKeyword, hiddenNodes, nodeCoordinatesSetter);
dispatch(reactFlowGraphDataCalculated({
flowData
}));
setNodes(layoutNodes);
setEdges(layoutEdges);
}, [isSearchMode, searchResult, currentSearchKeyword, hiddenNodes, dispatch, setEdges, setNodes]); // deps ensure updates on filter changes
const onConnect = useCallback(params => setEdges(eds => addEdge(_objectSpread(_objectSpread({}, params), {}, {
type: 'smoothstep',
animated: true
}), eds)), [setEdges]);
if (nodes.length === 0 && edges.length === 0) {
return /*#__PURE__*/React.createElement(CircularProgress, null);
}
return /*#__PURE__*/React.createElement(CanvasView, {
nodes: nodes,
edges: edges,
onNodesChange: onNodesChange,
onEdgesChange: onEdgesChange,
onConnect: onConnect
});
};
export default CanvasController;