wisemapping-frontend/packages/mindplot/src/components/layout/GridSorter.js

/*
 *    Copyright [2015] [wisemapping]
 *
 *   Licensed under WiseMapping Public License, Version 1.0 (the "License").
 *   It is basically the Apache License, Version 2.0 (the "License") plus the
 *   "powered by wisemapping" text requirement on every single page;
 *   you may not use this file except in compliance with the License.
 *   You may obtain a copy of the license at
 *
 *       http://www.wisemapping.org/license
 *
 *   Unless required by applicable law or agreed to in writing, software
 *   distributed under the License is distributed on an "AS IS" BASIS,
 *   WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 *   See the License for the specific language governing permissions and
 *   limitations under the License.
 */
import { $assert } from '@wisemapping/core-js';
import AbstractBasicSorter from './AbstractBasicSorter';

/**
 * @class
 * @extends mindplot.layout.AbstractBasicSorter
 */
const GridSorter = new Class(/** @lends GridSorter */{
  Extends: AbstractBasicSorter,

  /**
     * @param {} treeSet
     * @param {} node
     * @return offsets
     */
  computeOffsets(treeSet, node) {
    $assert(treeSet, 'treeSet can no be null.');
    $assert(node, 'node can no be null.');
    $assert('order can no be null.');

    const children = this._getSortedChildren(treeSet, node);

    // Compute heights ...
    const me = this;
    const heights = children.map((child) => ({
      id: child.getId(),
      height: me._computeChildrenHeight(treeSet, child),
    }));

    // Calculate the offsets ...
    const result = {};
    for (let i = 0; i < heights.length; i++) {
      const even = i % 2 == 0 ? 1 : -1;

      const zeroHeight = i == 0 ? 0 : heights[0].height / 2 * even;
      let middleHeight = 0;
      for (let j = i - 2; j > 0; j -= 2) {
        middleHeight += heights[j].height * even;
      }
      const finalHeight = i == 0 ? 0 : heights[i].height / 2 * even;

      const yOffset = zeroHeight + middleHeight + finalHeight;
      const xOffset = node.getSize().width + GridSorter.GRID_HORIZONTAR_SIZE;

      $assert(!isNaN(xOffset), 'xOffset can not be null');
      $assert(!isNaN(yOffset), 'yOffset can not be null');

      result[heights[i].id] = { x: xOffset, y: yOffset };
    }
    return result;
  },

  /**
     * @return {String} the print name of this class
     */
  toString() {
    return 'Grid Sorter';
  },

});

/**
 * @constant
 * @type {Number}
 * @default
 */
GridSorter.GRID_HORIZONTAR_SIZE = 20;
/**
 * @constant
 * @type {Number}
 * @default
 */
GridSorter.INTER_NODE_VERTICAL_DISTANCE = 50;

export default GridSorter;