dagre-layout/lib/coordinate-system.js

Graph layout for JavaScript

import _ from 'lodash'

function adjust (g) {
  const rankDir = g.graph().rankdir.toLowerCase()
  if (rankDir === 'lr' || rankDir === 'rl') {
    swapWidthHeight(g)
  }
}

function undo (g) {
  const rankDir = g.graph().rankdir.toLowerCase()
  if (rankDir === 'bt' || rankDir === 'rl') {
    reverseY(g)
  }

  if (rankDir === 'lr' || rankDir === 'rl') {
    swapXY(g)
    swapWidthHeight(g)
  }
}

function swapWidthHeight (g) {
  _.forEach(g.nodes(), function (v) { swapWidthHeightOne(g.node(v)) })
  _.forEach(g.edges(), function (e) { swapWidthHeightOne(g.edge(e)) })
}

function swapWidthHeightOne (attrs) {
  const w = attrs.width
  attrs.width = attrs.height
  attrs.height = w
}

function reverseY (g) {
  _.forEach(g.nodes(), function (v) { reverseYOne(g.node(v)) })

  _.forEach(g.edges(), function (e) {
    const edge = g.edge(e)
    _.forEach(edge.points, reverseYOne)
    if (_.has(edge, 'y')) {
      reverseYOne(edge)
    }
  })
}

function reverseYOne (attrs) {
  attrs.y = -attrs.y
}

function swapXY (g) {
  _.forEach(g.nodes(), function (v) { swapXYOne(g.node(v)) })

  _.forEach(g.edges(), function (e) {
    const edge = g.edge(e)
    _.forEach(edge.points, swapXYOne)
    if (_.has(edge, 'x')) {
      swapXYOne(edge)
    }
  })
}

function swapXYOne (attrs) {
  const x = attrs.x
  attrs.x = attrs.y
  attrs.y = x
}

export default {
  adjust,
  undo
}