let util = require("../../util"); let DataSet = require('../../DataSet'); let DataView = require('../../DataView'); import Node from "./components/Node"; import Label from "./components/shared/Label"; class NodesHandler { constructor(body, images, groups, layoutEngine) { this.body = body; this.images = images; this.groups = groups; this.layoutEngine = layoutEngine; // create the node API in the body container this.body.functions.createNode = this.create.bind(this); this.nodesListeners = { add: (event, params) => {this.add(params.items);}, update: (event, params) => {this.update(params.items, params.data);}, remove: (event, params) => {this.remove(params.items);} }; this.options = {}; this.defaultOptions = { borderWidth: 1, borderWidthSelected: undefined, brokenImage:undefined, color: { border: '#2B7CE9', background: '#97C2FC', highlight: { border: '#2B7CE9', background: '#D2E5FF' }, hover: { border: '#2B7CE9', background: '#D2E5FF' } }, fixed: { x:false, y:false }, font: { color: '#343434', size: 14, // px face: 'arial', background: 'none', stroke: 0, // px strokeColor: '#ffffff', align: 'horizontal' }, group: undefined, hidden: false, icon: { face: 'FontAwesome', //'FontAwesome', code: undefined, //'\uf007', size: 50, //50, color:'#2B7CE9' //'#aa00ff' }, image: undefined, // --> URL label: undefined, level: undefined, mass: 1, physics: true, scaling: { min: 10, max: 30, label: { enabled: false, min: 14, max: 30, maxVisible: 30, drawThreshold: 3 }, customScalingFunction: function (min,max,total,value) { if (max === min) { return 0.5; } else { let scale = 1 / (max - min); return Math.max(0,(value - min)*scale); } } }, shadow:{ enabled: false, size:10, x:5, y:5 }, shape: 'ellipse', size: 25, title: undefined, value: undefined, x: undefined, y: undefined }; util.extend(this.options, this.defaultOptions); this.bindEventListeners(); } bindEventListeners() { // refresh the nodes. Used when reverting from hierarchical layout this.body.emitter.on('refreshNodes', this.refresh.bind(this)); this.body.emitter.on('refresh', this.refresh.bind(this)); this.body.emitter.on("destroy", () => { delete this.body.functions.createNode; delete this.nodesListeners.add; delete this.nodesListeners.update; delete this.nodesListeners.remove; delete this.nodesListeners; }); } setOptions(options) { if (options !== undefined) { Node.parseOptions(this.options, options); // update the shape in all nodes if (options.shape !== undefined) { for (let nodeId in this.body.nodes) { if (this.body.nodes.hasOwnProperty(nodeId)) { this.body.nodes[nodeId].updateShape(); } } } // update the shape size in all nodes if (options.font !== undefined) { Label.parseOptions(this.options.font,options); for (let nodeId in this.body.nodes) { if (this.body.nodes.hasOwnProperty(nodeId)) { this.body.nodes[nodeId].updateLabelModule(); this.body.nodes[nodeId]._reset(); } } } // update the shape size in all nodes if (options.size !== undefined) { for (let nodeId in this.body.nodes) { if (this.body.nodes.hasOwnProperty(nodeId)) { this.body.nodes[nodeId]._reset(); } } } // update the state of the letiables if needed if (options.hidden !== undefined || options.physics !== undefined) { this.body.emitter.emit('_dataChanged'); } } } /** * Set a data set with nodes for the network * @param {Array | DataSet | DataView} nodes The data containing the nodes. * @private */ setData(nodes, doNotEmit = false) { let oldNodesData = this.body.data.nodes; if (nodes instanceof DataSet || nodes instanceof DataView) { this.body.data.nodes = nodes; } else if (Array.isArray(nodes)) { this.body.data.nodes = new DataSet(); this.body.data.nodes.add(nodes); } else if (!nodes) { this.body.data.nodes = new DataSet(); } else { throw new TypeError('Array or DataSet expected'); } if (oldNodesData) { // unsubscribe from old dataset util.forEach(this.nodesListeners, function (callback, event) { oldNodesData.off(event, callback); }); } // remove drawn nodes this.body.nodes = {}; if (this.body.data.nodes) { // subscribe to new dataset let me = this; util.forEach(this.nodesListeners, function (callback, event) { me.body.data.nodes.on(event, callback); }); // draw all new nodes let ids = this.body.data.nodes.getIds(); this.add(ids, true); } if (doNotEmit === false) { this.body.emitter.emit("_dataChanged"); } } /** * Add nodes * @param {Number[] | String[]} ids * @private */ add(ids, doNotEmit = false) { let id; let newNodes = []; for (let i = 0; i < ids.length; i++) { id = ids[i]; let properties = this.body.data.nodes.get(id); let node = this.create(properties);; newNodes.push(node); this.body.nodes[id] = node; // note: this may replace an existing node } this.layoutEngine.positionInitially(newNodes); if (doNotEmit === false) { this.body.emitter.emit("_dataChanged"); } } /** * Update existing nodes, or create them when not yet existing * @param {Number[] | String[]} ids * @private */ update(ids, changedData) { let nodes = this.body.nodes; let dataChanged = false; for (let i = 0; i < ids.length; i++) { let id = ids[i]; let node = nodes[id]; let data = changedData[i]; if (node !== undefined) { // update node node.setOptions(data, this.constants); } else { dataChanged = true; // create node node = this.create(properties); nodes[id] = node; } } if (dataChanged === true) { this.body.emitter.emit("_dataChanged"); } else { this.body.emitter.emit("_dataUpdated"); } } /** * Remove existing nodes. If nodes do not exist, the method will just ignore it. * @param {Number[] | String[]} ids * @private */ remove(ids) { let nodes = this.body.nodes; for (let i = 0; i < ids.length; i++) { let id = ids[i]; delete nodes[id]; } this.body.emitter.emit("_dataChanged"); } /** * create a node * @param properties * @param constructorClass */ create(properties, constructorClass = Node) { return new constructorClass(properties, this.body, this.images, this.groups, this.options) } refresh() { let nodes = this.body.nodes; for (let nodeId in nodes) { let node = undefined; if (nodes.hasOwnProperty(nodeId)) { node = nodes[nodeId]; } let data = this.body.data.nodes._data[nodeId]; if (node !== undefined && data !== undefined) { node.setOptions({fixed:false}); node.setOptions(data); } } } /** * Returns the positions of the nodes. * @param ids --> optional, can be array of nodeIds, can be string * @returns {{}} */ getPositions(ids) { let dataArray = {}; if (ids !== undefined) { if (Array.isArray(ids) === true) { for (let i = 0; i < ids.length; i++) { if (this.body.nodes[ids[i]] !== undefined) { let node = this.body.nodes[ids[i]]; dataArray[ids[i]] = {x: Math.round(node.x), y: Math.round(node.y)}; } } } else { if (this.body.nodes[ids] !== undefined) { let node = this.body.nodes[ids]; dataArray[ids] = {x: Math.round(node.x), y: Math.round(node.y)}; } } } else { for (let nodeId in this.body.nodes) { if (this.body.nodes.hasOwnProperty(nodeId)) { let node = this.body.nodes[nodeId]; dataArray[nodeId] = {x: Math.round(node.x), y: Math.round(node.y)}; } } } return dataArray; } /** * Load the XY positions of the nodes into the dataset. */ storePositions() { // todo: add support for clusters and hierarchical. let dataArray = []; for (let nodeId in this.body.nodes) { if (this.body.nodes.hasOwnProperty(nodeId)) { let node = this.body.nodes[nodeId]; if (this.body.data.nodes._data[nodeId].x != Math.round(node.x) || this.body.data.nodes._data[nodeId].y != Math.round(node.y)) { dataArray.push({id:nodeId,x:Math.round(node.x),y:Math.round(node.y)}); } } } this.body.data.nodes.update(dataArray); } /** * get the bounding box of a node. * @param nodeId * @returns {j|*} */ getBoundingBox(nodeId) { if (this.body.nodes[nodeId] !== undefined) { return this.body.nodes[nodeId].shape.boundingBox; } } /** * Get the Ids of nodes connected to this node. * @param nodeId * @returns {Array} */ getConnectedNodes(nodeId) { let nodeList = []; if (this.body.nodes[nodeId] !== undefined) { let node = this.body.nodes[nodeId]; let nodeObj = {}; // used to quickly check if node already exists for (let i = 0; i < node.edges.length; i++) { let edge = node.edges[i]; if (edge.toId === nodeId) { if (nodeObj[edge.fromId] === undefined) { nodeList.push(edge.fromId); nodeObj[edge.fromId] = true; } } else if (edge.fromId === nodeId) { if (nodeObj[edge.toId] === undefined) { nodeList.push(edge.toId) nodeObj[edge.toId] = true; } } } } return nodeList; } /** * Get the ids of the edges connected to this node. * @param nodeId * @returns {*} */ getEdges(nodeId) { let edgeList = []; if (this.body.nodes[nodeId] !== undefined) { let node = this.body.nodes[nodeId]; for (let i = 0; i < node.edges.length; i++) { edgeList.push(node.edges[i].id) } } return nodeList; } } export default NodesHandler;