metamaps--metamaps/frontend/src/Metamaps/Organize.js

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 window.alert('please call function with a valid layout dammit!')
  }
}

export default Organize