/* global $ */ import _ from 'lodash' import $jit from '../patched/JIT' import Visualize from './Visualize' import JIT from './JIT' const Organize = { arrange: function (layout, centerNode) { // first option for layout to implement is 'grid', will do an evenly spaced grid with its center at the 0,0 origin if (layout == 'grid') { var numNodes = _.size(Visualize.mGraph.graph.nodes); // this will always be an integer, the # of nodes on your graph visualization var numColumns = Math.floor(Math.sqrt(numNodes)) // the number of columns to make an even grid var GRIDSPACE = 400 var row = 0 var column = 0 Visualize.mGraph.graph.eachNode(function (n) { if (column == numColumns) { column = 0 row += 1 } var newPos = new $jit.Complex() newPos.x = column * GRIDSPACE newPos.y = row * GRIDSPACE n.setPos(newPos, 'end') column += 1 }) Visualize.mGraph.animate(JIT.ForceDirected.animateSavedLayout) } else if (layout == 'grid_full') { // this will always be an integer, the # of nodes on your graph visualization var numNodes = _.size(Visualize.mGraph.graph.nodes) // var numColumns = Math.floor(Math.sqrt(numNodes)) // the number of columns to make an even grid // var GRIDSPACE = 400 var height = Visualize.mGraph.canvas.getSize(0).height var width = Visualize.mGraph.canvas.getSize(0).width var totalArea = height * width var cellArea = totalArea / numNodes var ratio = height / width var cellWidth = sqrt(cellArea / ratio) var cellHeight = cellArea / cellWidth var row = floor(height / cellHeight) var column = floor(width / cellWidth) var totalCells = row * column if (totalCells) Visualize.mGraph.graph.eachNode(function (n) { if (column == numColumns) { column = 0 row += 1 } var newPos = new $jit.Complex() newPos.x = column * GRIDSPACE newPos.y = row * GRIDSPACE n.setPos(newPos, 'end') column += 1 }) Visualize.mGraph.animate(JIT.ForceDirected.animateSavedLayout) } else if (layout == 'radial') { var centerX = centerNode.getPos().x var centerY = centerNode.getPos().y centerNode.setPos(centerNode.getPos(), 'end') console.log(centerNode.adjacencies) var lineLength = 200 var usedNodes = {} usedNodes[centerNode.id] = centerNode var radial = function (node, level, degree) { if (level == 1) { var numLinksTemp = _.size(node.adjacencies) var angleTemp = 2 * Math.PI / numLinksTemp } else { angleTemp = 2 * Math.PI / 20 } node.eachAdjacency(function (a) { var isSecondLevelNode = (centerNode.adjacencies[a.nodeTo.id] != undefined && level > 1) if (usedNodes[a.nodeTo.id] == undefined && !isSecondLevelNode) { var newPos = new $jit.Complex() newPos.x = level * lineLength * Math.sin(degree) + centerX newPos.y = level * lineLength * Math.cos(degree) + centerY a.nodeTo.setPos(newPos, 'end') usedNodes[a.nodeTo.id] = a.nodeTo radial(a.nodeTo, level + 1, degree) degree += angleTemp } }) } radial(centerNode, 1, 0) Visualize.mGraph.animate(JIT.ForceDirected.animateSavedLayout) } else if (layout == 'center_viewport') { var lowX = 0, lowY = 0, highX = 0, highY = 0 var oldOriginX = Visualize.mGraph.canvas.translateOffsetX var oldOriginY = Visualize.mGraph.canvas.translateOffsetY Visualize.mGraph.graph.eachNode(function (n) { if (n.id === 1) { lowX = n.getPos().x lowY = n.getPos().y highX = n.getPos().x highY = n.getPos().y } if (n.getPos().x < lowX) lowX = n.getPos().x if (n.getPos().y < lowY) lowY = n.getPos().y if (n.getPos().x > highX) highX = n.getPos().x if (n.getPos().y > highY) highY = n.getPos().y }) console.log(lowX, lowY, highX, highY) var newOriginX = (lowX + highX) / 2 var newOriginY = (lowY + highY) / 2 } else alert('please call function with a valid layout dammit!') } } export default Organize