@7polo/kityminder-core/src/template/fish-bone.js

KityMinder Core Implement

/**
 * @fileOverview
 *
 * 默认模板 - 鱼骨头模板
 *
 * @author: techird
 * @copyright: Baidu FEX, 2014
 */
define(function(require, exports, module) {
    var template = require('../core/template');

    template.register('fish-bone', {

        getLayout: function(node) {

            if (node.getData('layout')) return node.getData('layout');

            var level = node.getLevel();

            // 根节点
            if (level === 0) {
                return 'fish-bone-master';
            }

            // 一级节点
            if (level === 1) {
                return 'fish-bone-slave';
            }

            return node.getLayoutPointPreview().y > 0 ? 'filetree-up': 'filetree-down';
        },

        getConnect: function(node) {
            switch (node.getLevel()) {
                case 1: return 'fish-bone-master';
                case 2: return 'line';
                default: return 'l';
            }
        }
    });
});