@antv/layout 0.3.11 → 1.0.0-alpha.0

package/src/layout/forceAtlas2/quad.ts

@@ -1,115 +0,0 @@
-import { PointTuple } from "../types";
-
-/**
- * @fileOverview quad
- * @author shiwu.wyy@antfin.com
- */
-
-type QuadProps = {
-  xmid: number;
-  ymid: number;
-  length: number;
-  massCenter?: PointTuple;
-  mass?: number;
-};
-
-export default class Quad {
-  public xmid: number;
-  public ymid: number;
-  public length: number;
-  public massCenter: PointTuple;
-  public mass: number;
-  constructor(params: QuadProps) {
-    /**
-     * the center position of this quad
-     * @type  {number}
-     */
-    this.xmid = params.xmid;
-    /**
-     * the center position of this quad
-     * @type  {number}
-     */
-    this.ymid = params.ymid;
-    /**
-     * the length of this quad
-     * @type  {number}
-     */
-    this.length = params.length;
-    /**
-     * the mass center of this quad
-     * @type  {number}
-     */
-    this.massCenter = params.massCenter || [0, 0];
-    /**
-     * the mass of this quad
-     * @type  {number}
-     */
-    this.mass = params.mass || 1;
-  }
-  getLength() {
-    return this.length;
-  }
-  contains(x: number, y: number) {
-    const halfLen = this.length / 2;
-    return (x <= this.xmid + halfLen &&
-      x >= this.xmid - halfLen &&
-      y <= this.ymid + halfLen &&
-      y >= this.ymid - halfLen);
-  }
-  // northwest quadrant
-  // tslint:disable-next-line
-  NW() {
-    const x = this.xmid - this.length / 4;
-    const y = this.ymid + this.length / 4;
-    const len = this.length / 2;
-    const params: QuadProps = {
-      xmid: x,
-      ymid: y,
-      length: len
-    };
-    const NW = new Quad(params);
-    return NW;
-  }
-  // northeast
-  // tslint:disable-next-line
-  NE() {
-    const x = this.xmid + this.length / 4;
-    const y = this.ymid + this.length / 4;
-    const len = this.length / 2;
-    const params = {
-      xmid: x,
-      ymid: y,
-      length: len
-    };
-    const NE = new Quad(params);
-    return NE;
-  }
-  // southwest
-  // tslint:disable-next-line
-  SW() {
-    const x = this.xmid - this.length / 4;
-    const y = this.ymid - this.length / 4;
-    const len = this.length / 2;
-    const params = {
-      xmid: x,
-      ymid: y,
-      length: len
-    };
-    const SW = new Quad(params);
-    return SW;
-  }
-  // southeast
-  // tslint:disable-next-line
-  SE() {
-    const x = this.xmid + this.length / 4;
-    const y = this.ymid - this.length / 4;
-    const len = this.length / 2;
-    const params = {
-      xmid: x,
-      ymid: y,
-      length: len
-    };
-    const SE = new Quad(params);
-    return SE;
-  }
-}

package/src/layout/forceAtlas2/quadTree.ts

@@ -1,107 +0,0 @@
-import Body from './body';
-import Quad from './quad';
-
-/**
- * @fileOverview quadTree
- * @author shiwu.wyy@antfin.com
- */
-
-export default class QuadTree {
-  public body: Body | null;
-  public quad: Quad | null;
-  public theta: number;
-  public NW: QuadTree | null;
-  public NE: QuadTree | null;
-  public SW: QuadTree | null;
-  public SE: QuadTree | null;
-
-  // each quadtree represents a quadrant and an aggregate body
-  // that represents all bodies inside the quadrant
-  constructor(param: Quad | null) {
-    /**
-     * (aggregated) body in this quad
-     * @type  {object}
-     */
-    this.body = null;
-    /**
-     * tree representing the northwest quadrant
-     * @type  {object}
-     */
-    this.quad = null;
-    this.NW = null;
-    this.NE = null;
-    this.SW = null;
-    this.SE = null;
-    /**
-     * threshold
-     * @type  {number}
-     */
-    this.theta = 0.5;
-    if (param != null) this.quad = param;
-  }
-  // insert a body(node) into the tree
-  insert(bo: Body) {
-    // if this node does not contain a body, put the new body bo here
-    if (this.body == null) {
-      this.body = bo;
-      return;
-    }
-    // internal node
-    if (!this._isExternal()) {
-      // update mass info
-      this.body = this.body.add(bo);
-      // insert body into quadrant
-      this._putBody(bo);
-    } else { // external node
-      // divide this region into four children
-      if (this.quad) {
-        this.NW = new QuadTree(this.quad.NW());
-        this.NE = new QuadTree(this.quad.NE());
-        this.SW = new QuadTree(this.quad.SW());
-        this.SE = new QuadTree(this.quad.SE());
-      }
-
-      // insert this body and bo
-      this._putBody(this.body);
-      this._putBody(bo);
-      // update the mass info
-      this.body = this.body.add(bo);
-
-    }
-  }
-  // inserts bo into a quad
-  // tslint:disable-next-line
-  _putBody(bo: Body) {
-    if (!this.quad) return;
-    if (bo.in(this.quad.NW()) && this.NW) this.NW.insert(bo);
-    else if (bo.in(this.quad.NE()) && this.NE) this.NE.insert(bo);
-    else if (bo.in(this.quad.SW()) && this.SW )this.SW.insert(bo);
-    else if (bo.in(this.quad.SE()) && this.SE) this.SE.insert(bo);
-  }
-  // tslint:disable-next-line
-  _isExternal() {
-    // four children are null
-    return (this.NW == null && this.NE == null && this.SW == null && this.SE == null);
-  }
-  // update the forces
-  updateForce(bo: Body) {
-    if (this.body == null || bo === this.body) {
-      return;
-    }
-    // if the current node is external
-    if (this._isExternal()) bo.addForce(this.body);
-    // internal nodes
-    else {
-      const s = this.quad ? this.quad.getLength() : 0;
-      const d = this.body.distanceTo(bo);
-      // b is far enough
-      if ((s / d) < this.theta) bo.addForce(this.body);
-      else {
-        this.NW && this.NW.updateForce(bo);
-        this.NE && this.NE.updateForce(bo);
-        this.SW && this.SW.updateForce(bo);
-        this.SE && this.SE.updateForce(bo);
-      }
-    }
-  }
-}

package/src/layout/fruchterman.ts

@@ -1,361 +0,0 @@
-/**
- * @fileOverview fruchterman layout
- * @author shiwu.wyy@antfin.com
- */
-
-import {
-  OutNode,
-  Edge,
-  PointTuple,
-  IndexMap,
-  Point,
-  FruchtermanLayoutOptions
-} from "./types";
-import { Base } from "./base";
-import { getEdgeTerminal, isNumber } from "../util";
-
-type NodeMap = {
-  [key: string]: INode;
-};
-
-type INode = OutNode & {
-  cluster: string;
-};
-
-const SPEED_DIVISOR = 800;
-
-/**
- * fruchterman 布局
- */
-export class FruchtermanLayout extends Base {
-  /** 布局中心 */
-  public center: PointTuple;
-
-  /** 停止迭代的最大迭代数 */
-  public maxIteration: number = 1000;
-
-  /** 是否启动 worker */
-  public workerEnabled: boolean = false;
-
-  /** 重力大小，影响图的紧凑程度 */
-  public gravity: number = 10;
-
-  /** 速度 */
-  public speed: number = 5;
-
-  /** 是否产生聚类力 */
-  public clustering: boolean = false;
-
-  /** 聚类力大小 */
-  public clusterGravity: number = 10;
-
-  public nodes: INode[] | null = [];
-
-  public edges: Edge[] | null = [];
-
-  public width: number = 300;
-
-  public height: number = 300;
-
-  public nodeMap: NodeMap = {};
-
-  public nodeIdxMap: IndexMap = {};
-
-  /** 迭代结束的回调函数 */
-  public onLayoutEnd: () => void = () => {};
-
-  /** 每次迭代结束的回调函数 */
-  public tick: (() => void) | null = () => {};
-
-  /** 是否使用 window.setInterval 运行迭代 */
-  public animate: boolean = true;
-
-  /** 迭代中的标识 */
-  private timeInterval: number;
-
-  constructor(options?: FruchtermanLayoutOptions) {
-    super();
-    this.updateCfg(options);
-  }
-
-  public getDefaultCfg() {
-    return {
-      maxIteration: 1000,
-      gravity: 10,
-      speed: 1,
-      clustering: false,
-      clusterGravity: 10,
-      animate: true
-    };
-  }
-
-  /**
-   * 执行布局
-   */
-  public execute() {
-    const self = this;
-    const nodes = self.nodes;
-
-    if (self.timeInterval !== undefined && typeof window !== "undefined") {
-      window.clearInterval(self.timeInterval);
-    }
-
-    if (!nodes || nodes.length === 0) {
-      self.onLayoutEnd?.();
-      return;
-    }
-
-    if (!self.width && typeof window !== "undefined") {
-      self.width = window.innerWidth;
-    }
-    if (!self.height && typeof window !== "undefined") {
-      self.height = window.innerHeight;
-    }
-    if (!self.center) {
-      self.center = [self.width / 2, self.height / 2];
-    }
-    const center = self.center;
-
-    if (nodes.length === 1) {
-      nodes[0].x = center[0];
-      nodes[0].y = center[1];
-      self.onLayoutEnd?.();
-      return;
-    }
-    const nodeMap: NodeMap = {};
-    const nodeIdxMap: IndexMap = {};
-    nodes.forEach((node, i) => {
-      if (!isNumber(node.x)) node.x = Math.random() * this.width;
-      if (!isNumber(node.y)) node.y = Math.random() * this.height;
-      nodeMap[node.id] = node;
-      nodeIdxMap[node.id] = i;
-    });
-    self.nodeMap = nodeMap;
-    self.nodeIdxMap = nodeIdxMap;
-    // layout
-    return self.run();
-  }
-
-  public run() {
-    const self = this;
-    const nodes = self.nodes;
-    if (!nodes) return;
-    const { edges, maxIteration, workerEnabled, clustering, animate } = self;
-    const clusterMap: {
-      [key: string]: {
-        name: string | number;
-        cx: number;
-        cy: number;
-        count: number;
-      };
-    } = {};
-    if (clustering) {
-      nodes.forEach((n) => {
-        if (clusterMap[n.cluster] === undefined) {
-          clusterMap[n.cluster] = {
-            name: n.cluster,
-            cx: 0,
-            cy: 0,
-            count: 0
-          };
-        }
-      });
-    }
-    if (workerEnabled || !animate) {
-      for (let i = 0; i < maxIteration; i++) {
-        self.runOneStep(clusterMap);
-      }
-      self.onLayoutEnd?.();
-    } else {
-      if (typeof window === "undefined") return;
-      let iter = 0;
-      // interval for render the result after each iteration
-      this.timeInterval = window.setInterval(() => {
-        self.runOneStep(clusterMap);
-        iter++;
-        if (iter >= maxIteration) {
-          self.onLayoutEnd?.();
-          window.clearInterval(self.timeInterval);
-        }
-      }, 0);
-    }
-    return {
-      nodes,
-      edges
-    };
-  }
-
-  private runOneStep(clusterMap: any) {
-    const self = this;
-    const nodes = self.nodes;
-    if (!nodes) return;
-    const { edges, center, gravity, speed, clustering } = self;
-    const area = self.height * self.width;
-    const maxDisplace = Math.sqrt(area) / 10;
-    const k2 = area / (nodes.length + 1);
-    const k = Math.sqrt(k2);
-    const displacements: Point[] = [];
-    nodes.forEach((_, j) => {
-      displacements[j] = { x: 0, y: 0 };
-    });
-    self.applyCalculate(nodes, edges, displacements, k, k2);
-
-    // gravity for clusters
-    if (clustering) {
-      // re-compute the clustering centers
-      for (const key in clusterMap) {
-        clusterMap[key].cx = 0;
-        clusterMap[key].cy = 0;
-        clusterMap[key].count = 0;
-      }
-      nodes.forEach((n) => {
-        const c = clusterMap[n.cluster];
-        if (isNumber(n.x)) {
-          c.cx += n.x;
-        }
-        if (isNumber(n.y)) {
-          c.cy += n.y;
-        }
-        c.count++;
-      });
-      for (const key in clusterMap) {
-        clusterMap[key].cx /= clusterMap[key].count;
-        clusterMap[key].cy /= clusterMap[key].count;
-      }
-
-      // compute the cluster gravity forces
-      const clusterGravity = self.clusterGravity || gravity;
-      nodes.forEach((n, j) => {
-        if (!isNumber(n.x) || !isNumber(n.y)) return;
-        const c = clusterMap[n.cluster];
-        const distLength = Math.sqrt(
-          (n.x - c.cx) * (n.x - c.cx) + (n.y - c.cy) * (n.y - c.cy)
-        );
-        const gravityForce = k * clusterGravity;
-        displacements[j].x -= (gravityForce * (n.x - c.cx)) / distLength;
-        displacements[j].y -= (gravityForce * (n.y - c.cy)) / distLength;
-      });
-    }
-
-    // gravity
-    nodes.forEach((n, j) => {
-      if (!isNumber(n.x) || !isNumber(n.y)) return;
-      const gravityForce = 0.01 * k * gravity;
-      displacements[j].x -= gravityForce * (n.x - center[0]);
-      displacements[j].y -= gravityForce * (n.y - center[1]);
-    });
-
-    // move
-    nodes.forEach((n: any, j) => {
-      if (isNumber(n.fx) && isNumber(n.fy)) {
-        n.x = n.fx;
-        n.y = n.fy;
-        return;
-      }
-      if (!isNumber(n.x) || !isNumber(n.y)) return;  
-      const distLength = Math.sqrt(
-        displacements[j].x * displacements[j].x +
-          displacements[j].y * displacements[j].y
-      );
-      if (distLength > 0) {
-        // && !n.isFixed()
-        const limitedDist = Math.min(
-          maxDisplace * (speed / SPEED_DIVISOR),
-          distLength
-        );
-        n.x += (displacements[j].x / distLength) * limitedDist;
-        n.y += (displacements[j].y / distLength) * limitedDist;
-      }
-    });
-
-    self.tick?.();
-  }
-
-  private applyCalculate(
-    nodes: INode[],
-    edges: Edge[] | null,
-    displacements: Point[],
-    k: number,
-    k2: number
-  ) {
-    const self = this;
-    self.calRepulsive(nodes, displacements, k2);
-    if (edges) self.calAttractive(edges, displacements, k);
-  }
-
-  private calRepulsive(nodes: INode[], displacements: Point[], k2: number) {
-    nodes.forEach((v, i) => {
-      displacements[i] = { x: 0, y: 0 };
-      nodes.forEach((u, j) => {
-        if (i === j) {
-          return;
-        }
-        if (
-          !isNumber(v.x) ||
-          !isNumber(u.x) ||
-          !isNumber(v.y) ||
-          !isNumber(u.y)
-        ) {
-          return;
-        }
-        let vecX = v.x - u.x;
-        let vecY = v.y - u.y;
-        let vecLengthSqr = vecX * vecX + vecY * vecY;
-        if (vecLengthSqr === 0) {
-          vecLengthSqr = 1;
-          const sign = i > j ? 1 : -1;
-          vecX = 0.01 * sign;
-          vecY = 0.01 * sign;
-        }
-        const common = k2 / vecLengthSqr;
-        displacements[i].x += vecX * common;
-        displacements[i].y += vecY * common;
-      });
-    });
-  }
-
-  private calAttractive(edges: Edge[], displacements: Point[], k: number) {
-    edges.forEach((e) => {
-      const source = getEdgeTerminal(e, 'source');
-      const target = getEdgeTerminal(e, 'target');
-      if (!source || !target) return;
-      const uIndex = this.nodeIdxMap[source];
-      const vIndex = this.nodeIdxMap[target];
-      if (uIndex === vIndex) {
-        return;
-      }
-      const u = this.nodeMap[source];
-      const v = this.nodeMap[target];
-      if (!isNumber(v.x) || !isNumber(u.x) || !isNumber(v.y) || !isNumber(u.y)) {
-        return;
-      }
-      const vecX = v.x - u.x;
-      const vecY = v.y - u.y;
-      const vecLength = Math.sqrt(vecX * vecX + vecY * vecY);
-      const common = (vecLength * vecLength) / k;
-      displacements[vIndex].x -= (vecX / vecLength) * common;
-      displacements[vIndex].y -= (vecY / vecLength) * common;
-      displacements[uIndex].x += (vecX / vecLength) * common;
-      displacements[uIndex].y += (vecY / vecLength) * common;
-    });
-  }
-
-  public stop() {
-    if (this.timeInterval && typeof window !== "undefined") {
-      window.clearInterval(this.timeInterval);
-    }
-  }
-
-  public destroy() {
-    const self = this;
-    self.stop();
-    self.tick = null;
-    self.nodes = null;
-    self.edges = null;
-    self.destroyed = true;
-  }
-
-  public getType() {
-    return "fruchterman";
-  }
-}