CocoRoboDesktop/kityminder-editor/src/module/layout.js

/**
 * @fileOverview
 *
 * 布局模块
 *
 * @author: techird
 * @copyright: Baidu FEX, 2014
 */

define(function(require, exports, module) {
    var kity = require('../core/kity');
    var Command = require('../core/command');
    var Module = require('../core/module');

    /**
     * @command Layout
     * @description 设置选中节点的布局
     *     允许使用的布局可以使用 `kityminder.Minder.getLayoutList()` 查询
     * @param {string} name 布局的名称，设置为 null 则使用继承或默认的布局
     * @state
     *   0: 当前有选中的节点
     *  -1: 当前没有选中的节点
     * @return 返回首个选中节点的布局名称
     */
    var LayoutCommand = kity.createClass('LayoutCommand', {
        base: Command,

        execute: function(minder, name) {
            var nodes = minder.getSelectedNodes();
            nodes.forEach(function(node) {
                node.layout(name);
            });
        },

        queryValue: function(minder) {
            var node = minder.getSelectedNode();
            if (node) {
                return node.getData('layout');
            }
        },

        queryState: function(minder) {
            return minder.getSelectedNode() ? 0 : -1;
        }
    });

    /**
     * @command ResetLayout
     * @description 重设选中节点的布局，如果当前没有选中的节点，重设整个脑图的布局
     * @state
     *   0: 始终可用
     * @return 返回首个选中节点的布局名称
     */
    var ResetLayoutCommand = kity.createClass('ResetLayoutCommand', {
        base: Command,

        execute: function(minder) {
            var nodes = minder.getSelectedNodes();

            if (!nodes.length) nodes = [minder.getRoot()];

            nodes.forEach(function(node) {
                node.traverse(function(child) {
                    child.resetLayoutOffset();
                    if (!child.isRoot()) {
                        child.setData('layout', null);
                    }
                });
            });
            minder.layout(300);
        },

        enableReadOnly: true
    });

    Module.register('LayoutModule', {
        commands: {
            'layout': LayoutCommand,
            'resetlayout': ResetLayoutCommand
        },
        contextmenu: [{
            command: 'resetlayout'
        }, {
            divider: true
        }],

        commandShortcutKeys: {
            'resetlayout': 'Ctrl+Shift+L'
        }
    });

});