wisemapping-open-source/mindplot/src/main/javascript/ConnectionLine.js

/*
 *    Copyright [2011] [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.
 */

mindplot.ConnectionLine = new Class({
    initialize:function(sourceNode, targetNode, lineType) {
        $assert(targetNode, 'parentNode node can not be null');
        $assert(sourceNode, 'childNode node can not be null');
        $assert(sourceNode != targetNode, 'Cilcular connection');

        this._targetTopic = targetNode;
        this._sourceTopic = sourceNode;

        var strokeColor = mindplot.ConnectionLine.getStrokeColor();
        var line;
        var ctrlPoints = this._getCtrlPoints(sourceNode, targetNode);
        if (targetNode.getType() == mindplot.model.INodeModel.CENTRAL_TOPIC_TYPE) {
            line = this._createLine(lineType, mindplot.ConnectionLine.CURVED);
            if (line.getType() == "CurvedLine") {
                line.setSrcControlPoint(ctrlPoints[0]);
                line.setDestControlPoint(ctrlPoints[1]);
            }
            line.setStroke(1, 'solid', strokeColor);
        } else {
            line = this._createLine(lineType, mindplot.ConnectionLine.SIMPLE_CURVED);
            if (line.getType() == "CurvedLine") {
                line.setSrcControlPoint(ctrlPoints[0]);
                line.setDestControlPoint(ctrlPoints[1]);
            }
            line.setStroke(1, 'solid', strokeColor);
        }
        line.setFill(mindplot.ConnectionLine.getStrokeColor());
        this._line2d = line;
    },

    _getCtrlPoints : function(sourceNode, targetNode) {
        var srcPos = sourceNode.workoutOutgoingConnectionPoint(targetNode.getPosition());
        var destPos = targetNode.workoutIncomingConnectionPoint(sourceNode.getPosition());
        var deltaX = (srcPos.x - destPos.x) / 3;
        return [new core.Point(deltaX, 0), new core.Point(-deltaX, 0)];
    },

    _createLine : function(lineType, defaultStyle) {
        if (!$defined(lineType)) {
            lineType = defaultStyle;
        }
        lineType = parseInt(lineType);
        this._lineType = lineType;
        var line = null;
        switch (lineType) {
            case mindplot.ConnectionLine.POLYLINE:
                line = new web2d.PolyLine();
                break;
            case mindplot.ConnectionLine.CURVED:
                line = new web2d.CurvedLine();
                break;
            case mindplot.ConnectionLine.SIMPLE_CURVED:
                line = new web2d.CurvedLine();
                line.setStyle(web2d.CurvedLine.SIMPLE_LINE);
                break;
            default:
                line = new web2d.Line();
                break;
        }
        return line;
    },

    setVisibility : function(value) {
        this._line2d.setVisibility(value);
    },

    isVisible : function() {
        return this._line2d.isVisible();
    },

    setOpacity : function(opacity) {
        this._line2d.setOpacity(opacity);
    },

    redraw : function() {
        var line2d = this._line2d;
        var sourceTopic = this._sourceTopic;
        var sourcePosition = sourceTopic.getPosition();

        var targetTopic = this._targetTopic;
        var targetPosition = targetTopic.getPosition();

        var sPos,tPos;
        sPos = sourceTopic.workoutOutgoingConnectionPoint(targetPosition, false);
        tPos = targetTopic.workoutIncomingConnectionPoint(sourcePosition, false);

        line2d.setFrom(tPos.x, tPos.y);
        line2d.setTo(sPos.x, sPos.y);

        if (line2d.getType() == "CurvedLine") {
            var ctrlPoints = this._getCtrlPoints(this._sourceTopic, this._targetTopic);
            line2d.setSrcControlPoint(ctrlPoints[0]);
            line2d.setDestControlPoint(ctrlPoints[1]);
        }

        // Add connector ...
        this._positionateConnector(targetTopic);

    },

    _positionateConnector : function(targetTopic) {
        var targetPosition = targetTopic.getPosition();
        var offset = mindplot.Topic.CONNECTOR_WIDTH / 2;
        var targetTopicSize = targetTopic.getSize();
        var y;
        if (targetTopic.getShapeType() == mindplot.model.INodeModel.SHAPE_TYPE_LINE) {
            y = targetTopicSize.height;
        } else {
            y = targetTopicSize.height / 2;
        }
        y = y - offset;

        var connector = targetTopic.getShrinkConnector();
        if (Math.sign(targetPosition.x) > 0) {
            var x = targetTopicSize.width;
            connector.setPosition(x, y);
        }
        else {
            var x = -mindplot.Topic.CONNECTOR_WIDTH;
            connector.setPosition(x, y);
        }
    },

    setStroke : function(color, style, opacity) {
        var line2d = this._line2d;
        this._line2d.setStroke(null, null, color, opacity);
    },

    addToWorkspace : function(workspace) {
        workspace.appendChild(this._line2d);
        this._line2d.moveToBack();
    },

    removeFromWorkspace : function(workspace) {
        workspace.removeChild(this._line2d);
    },

    getTargetTopic : function() {
        return this._targetTopic;
    },

    getSourceTopic : function() {
        return this._sourceTopic;
    },

    getLineType : function() {
        return this._lineType;
    },

    getLine : function() {
        return this._line2d;
    },

    getModel : function() {
        return this._model;
    },

    setModel : function(model) {
        this._model = model;
    },

    getType : function() {
        return "ConnectionLine";
    },

    getId : function() {
        return this._model.getId();
    },

    moveToBack : function() {
        this._line2d.moveToBack();
    },

    moveToFront : function() {
        this._line2d.moveToFront();
    }
});

mindplot.ConnectionLine.getStrokeColor = function() {
    return '#495879';
};

mindplot.ConnectionLine.SIMPLE = 0;
mindplot.ConnectionLine.POLYLINE = 1;
mindplot.ConnectionLine.CURVED = 2;
mindplot.ConnectionLine.SIMPLE_CURVED = 3;