ffrgb / meshviewer

define(['d3-selection', 'd3-force', 'd3-zoom', 'd3-drag', 'd3-timer', 'd3-ease', 'd3-interpolate', 'utils/math', 'forcegraph/draw'],

  function (d3Selection, d3Force, d3Zoom, d3Drag, d3Timer, d3Ease, d3Interpolate, math, draw) {

    'use strict';

    return function (linkScale, sidebar) {

      var self = this;

      var el;

      var canvas;

      var ctx;

      var force;

      var forceLink;

      var transform = d3Zoom.zoomIdentity;

      var intNodes = [];

      var dictNodes = {};

      var intLinks = [];

      var movetoTimer;

      var initial = 1.8;

      var NODE_RADIUS_DRAG = 10;

      var NODE_RADIUS_SELECT = 15;

      var LINK_RADIUS_SELECT = 12;

      var ZOOM_ANIMATE_DURATION = 350;

      var ZOOM_MIN = 1 / 8;

      var ZOOM_MAX = 3;

      var FORCE_ALPHA = 0.01;

      draw.setTransform(transform);

      function resizeCanvas() {

        canvas.width = el.offsetWidth;

        canvas.height = el.offsetHeight;

        draw.setMaxArea(canvas.width, canvas.height);

      }

      function transformPosition(p) {

        transform.x = p.x;

        transform.y = p.y;

        transform.k = p.k;

      }

      function moveTo(callback, forceMove) {

        clearTimeout(movetoTimer);

        if (!forceMove && force.alpha() > 0.3) {

          movetoTimer = setTimeout(function timerOfMoveTo() {

            moveTo(callback);

          }, 300);

          return;

        }

        var result = callback();

        var x = result[0];

        var y = result[1];

        var k = result[2];

        var end = { k: k };

        end.x = (canvas.width + sidebar.getWidth()) / 2 - x * k;

        end.y = canvas.height / 2 - y * k;

        var start = { x: transform.x, y: transform.y, k: transform.k };

        var interpolate = d3Interpolate.interpolateObject(start, end);

        var timer = d3Timer.timer(function (t) {

          if (t >= ZOOM_ANIMATE_DURATION) {

            timer.stop();

            return;

          }

          var v = interpolate(d3Ease.easeQuadInOut(t / ZOOM_ANIMATE_DURATION));

          transformPosition(v);

          window.requestAnimationFrame(redraw);

        });

      }

      function onClick() {

        if (d3Selection.event.defaultPrevented) {

          return;

        }

        var e = transform.invert([d3Selection.event.clientX, d3Selection.event.clientY]);

        var n = force.find(e[0], e[1], NODE_RADIUS_SELECT);

        if (n !== undefined) {

          router.fullUrl({ node: n.o.node_id });

          return;

        }

        e = { x: e[0], y: e[1] };

        var closedLink;

        var radius = LINK_RADIUS_SELECT;

        intLinks

          .forEach(function (d) {

            var distance = math.distanceLink(e, d.source, d.target);

            if (distance < radius) {

              closedLink = d;

              radius = distance;

            }

          });

        if (closedLink !== undefined) {

          router.fullUrl({ link: closedLink.o.id });

        }

      }

      function redraw() {

        ctx.save();

        ctx.clearRect(0, 0, canvas.width, canvas.height);

        ctx.translate(transform.x, transform.y);

        ctx.scale(transform.k, transform.k);

        intLinks.forEach(draw.drawLink);

        intNodes.forEach(draw.drawNode);

        ctx.restore();

      }

      el = document.createElement('div');

      el.classList.add('graph');

      forceLink = d3Force.forceLink()

        .distance(function (d) {

          if (d.o.type.indexOf('vpn') === 0) {

            return 0;

          }

          return 75;

        })

        .strength(function (d) {

          if (d.o.type.indexOf('vpn') === 0) {

            return 0.02;

          }

          return Math.max(0.5, d.o.source_tq);

        });

      var zoom = d3Zoom.zoom()

        .scaleExtent([ZOOM_MIN, ZOOM_MAX])

        .on('zoom', function () {

          transform = d3Selection.event.transform;

          draw.setTransform(transform);

          redraw();

        });

      force = d3Force.forceSimulation()

        .force('link', forceLink)

        .force('charge', d3Force.forceManyBody())

        .force('x', d3Force.forceX().strength(0.02))

        .force('y', d3Force.forceY().strength(0.02))

        .force('collide', d3Force.forceCollide())

        .on('tick', redraw)

        .alphaDecay(0.025);

      var drag = d3Drag.drag()

        .subject(function () {

          var e = transform.invert([d3Selection.event.x, d3Selection.event.y]);

          var n = force.find(e[0], e[1], NODE_RADIUS_DRAG);

          if (n !== undefined) {

            n.x = d3Selection.event.x;

            n.y = d3Selection.event.y;

            return n;

          }

          return undefined;

        })

        .on('start', function () {

          if (!d3Selection.event.active) {

            force.alphaTarget(FORCE_ALPHA).restart();

          }

          d3Selection.event.subject.fx = transform.invertX(d3Selection.event.subject.x);

          d3Selection.event.subject.fy = transform.invertY(d3Selection.event.subject.y);

        })

        .on('drag', function () {

          d3Selection.event.subject.fx = transform.invertX(d3Selection.event.x);

          d3Selection.event.subject.fy = transform.invertY(d3Selection.event.y);

        })

        .on('end', function () {

          if (!d3Selection.event.active) {

            force.alphaTarget(0);

          }

          d3Selection.event.subject.fx = null;

          d3Selection.event.subject.fy = null;

        });

      canvas = d3Selection.select(el)

        .append('canvas')

        .on('click', onClick)

        .call(drag)

        .call(zoom)

        .node();

      ctx = canvas.getContext('2d');

      draw.setCTX(ctx);

      window.addEventListener('resize', function () {

        resizeCanvas();

        redraw();

      });

      self.setData = function setData(data) {

        intNodes = data.nodes.all.map(function (d) {

          var e = dictNodes[d.node_id];

          if (!e) {

            e = {};

            dictNodes[d.node_id] = e;

          }

          e.o = d;

          return e;

        });

        intLinks = data.links.filter(function (d) {

          return data.nodeDict[d.source.node_id].is_online && data.nodeDict[d.target.node_id].is_online;

        }).map(function (d) {

          return {

            o: d,

            source: dictNodes[d.source.node_id],

            target: dictNodes[d.target.node_id],

            color: linkScale(d.source_tq),

            color_to: linkScale(d.target_tq)

          };

        });

        force.nodes(intNodes);

        forceLink.links(intLinks);

        force.alpha(initial).velocityDecay(0.15).restart();

        if (initial === 1.8) {

          initial = 0.5;

        }

        resizeCanvas();

      };

      self.resetView = function resetView() {

        moveTo(function calcToReset() {

          draw.setHighlight(null);

          return [0, 0, (ZOOM_MIN + config.forceGraph.zoomModifier) / 2];

        }, true);

      };

      self.gotoNode = function gotoNode(d) {

        moveTo(function calcToNode() {

          draw.setHighlight({ type: 'node', id: d.node_id });

          var n = dictNodes[d.node_id];

          if (n) {

            return [n.x, n.y, (ZOOM_MAX + 1) / 2];

          }

          return self.resetView();

        });

      };

      self.gotoLink = function gotoLink(d) {

        moveTo(function calcToLink() {

          draw.setHighlight({ type: 'link', id: d[0].id });

          var l = intLinks.find(function (link) {

            return link.o.id === d[0].id;

          });

          if (l) {

            return [(l.source.x + l.target.x) / 2, (l.source.y + l.target.y) / 2, (ZOOM_MAX / 2) + ZOOM_MIN];

          }

          return self.resetView();

        });

      };

      self.gotoLocation = function gotoLocation() {

        // ignore

      };

      self.destroy = function destroy() {

        force.stop();

        canvas.parentNode.removeChild(canvas);

        force = null;

        if (el.parentNode) {

          el.parentNode.removeChild(el);

        }

      };

      self.render = function render(d) {

        d.appendChild(el);

        resizeCanvas();

      };

      return self;

    };

  });