Examples with EdgeStatement org.eclipse.elk.alg.graphviz.dot.dot.EdgeStatement used on opensource projects

Example 6 with EdgeStatement

use of org.eclipse.elk.alg.graphviz.dot.dot.EdgeStatement in project elk by eclipse.

the class DotImporter method transformEdge.

/**
 * Transforms an edge.
 *
 * @param statement an edge statement
 * @param parent the parent node
 * @param transData the transformation data instance
 * @param defaultProps default values for edge options
 */
private void transformEdge(final EdgeStatement statement, final ElkNode parent, final IDotTransformationData<GraphvizModel, ElkNode> transData, final IPropertyHolder defaultProps) {
    String sourceName = statement.getSourceNode().getName();
    ElkNode source = transformNode(sourceName, parent, transData);
    ElkPort sourcePort = null;
    if (statement.getSourceNode().getPort() != null) {
        String portName = statement.getSourceNode().getPort().getCompass_pt();
        if (portName == null) {
            portName = statement.getSourceNode().getPort().getName();
        }
        sourcePort = transformPort(portName, source, transData);
    }
    ListIterator<EdgeTarget> targetIter = statement.getEdgeTargets().listIterator();
    while (targetIter.hasNext()) {
        EdgeTarget edgeTarget = targetIter.next();
        ElkEdge elkedge = ElkGraphUtil.createEdge(null);
        if (sourcePort != null) {
            elkedge.getSources().add(sourcePort);
        } else {
            elkedge.getSources().add(source);
        }
        ElkNode target;
        ElkPort targetPort = null;
        Node edgeTargetNode = edgeTarget.getTargetnode();
        String targetName;
        if (edgeTargetNode == null) {
            targetName = edgeTarget.getTargetSubgraph().getName();
            target = transformNode(targetName, parent, transData);
        } else {
            targetName = edgeTargetNode.getName();
            target = transformNode(targetName, parent, transData);
            if (edgeTargetNode.getPort() != null) {
                String portName = edgeTargetNode.getPort().getCompass_pt();
                if (portName == null) {
                    portName = edgeTargetNode.getPort().getName();
                }
                targetPort = transformPort(portName, target, transData);
            }
        }
        if (targetPort != null) {
            elkedge.getTargets().add(targetPort);
        } else {
            elkedge.getTargets().add(target);
        }
        elkedge.copyProperties(defaultProps);
        if (targetIter.previousIndex() == 0) {
            // this is the first target - just store the edge statement
            elkedge.setProperty(PROP_STATEMENT, statement);
        } else {
            // the edge statement has more that one target - create a copy
            EdgeStatement newStatement = DotFactory.eINSTANCE.createEdgeStatement();
            Node sourceNode = DotFactory.eINSTANCE.createNode();
            sourceNode.setName(sourceName);
            newStatement.setSourceNode(sourceNode);
            targetIter.remove();
            newStatement.getEdgeTargets().add(edgeTarget);
            for (Attribute attr : statement.getAttributes()) {
                newStatement.getAttributes().add(EcoreUtil.copy(attr));
            }
            elkedge.setProperty(PROP_STATEMENT, newStatement);
        }
        // evaluate attributes for the new edge
        for (Attribute attr : statement.getAttributes()) {
            String value = trimValue(attr);
            if (Attributes.LABEL.equals(attr.getName())) {
                ElkLabel label = ElkGraphUtil.createLabel(value, elkedge);
                label.setProperty(CoreOptions.EDGE_LABELS_PLACEMENT, EdgeLabelPlacement.CENTER);
            } else if (Attributes.HEADLABEL.equals(attr.getName())) {
                ElkLabel label = ElkGraphUtil.createLabel(value, elkedge);
                label.setProperty(CoreOptions.EDGE_LABELS_PLACEMENT, EdgeLabelPlacement.HEAD);
            } else if (Attributes.TAILLABEL.equals(attr.getName())) {
                ElkLabel label = ElkGraphUtil.createLabel(value, elkedge);
                label.setProperty(CoreOptions.EDGE_LABELS_PLACEMENT, EdgeLabelPlacement.TAIL);
            } else {
                transformAttribute(elkedge, attr, transData);
            }
        }
        // the edge target is the source for the next edge target
        source = target;
        sourceName = targetName;
        sourcePort = targetPort;
    }
}