Examples with Edge org.kie.workbench.common.stunner.core.graph.Edge used on opensource projects

Example 71 with Edge

use of org.kie.workbench.common.stunner.core.graph.Edge in project kie-wb-common by kiegroup.

the class BPMN2JsonParser method initialize.

/*
        ****************************************************************
        *               Custom Stunner logic
        ****************************************************************
     */
@SuppressWarnings("unchecked")
private void initialize(final ContextualParser.Context parsingContext) {
    Graph graph = diagram.getGraph();
    final Map<String, EdgeParser> edgeParsers = new HashMap<>();
    new ChildrenTraverseProcessorImpl(new TreeWalkTraverseProcessorImpl()).traverse(graph, new AbstractChildrenTraverseCallback<Node<View, Edge>, Edge<Child, Node>>() {

        final Stack<NodeParser> parsers = new Stack<NodeParser>();

        NodeParser currentParser = null;

        @Override
        public void startGraphTraversal(Graph<DefinitionSet, Node<View, Edge>> graph) {
            super.startGraphTraversal(graph);
        }

        @Override
        public boolean startNodeTraversal(final List<Node<View, Edge>> parents, final Node<View, Edge> node) {
            super.startNodeTraversal(parents, node);
            onNodeTraversal(node);
            return true;
        }

        @Override
        public void startNodeTraversal(final Node<View, Edge> node) {
            super.startNodeTraversal(node);
            onNodeTraversal(node);
        }

        private void onNodeTraversal(final Node node) {
            NodeParser p = new NodeParser("", node);
            if (null != currentParser) {
                parsers.peek().addChild(p);
            } else {
                BPMN2JsonParser.this.rootParser = p;
            }
            currentParser = p;
            List<Edge> outEdges = node.getOutEdges();
            if (null != outEdges && !outEdges.isEmpty()) {
                for (Edge edge : outEdges) {
                    // Only add the edges with view connector types into the resulting structure to generate the bpmn definition.
                    if (edge.getContent() instanceof ViewConnector && !edgeParsers.containsKey(edge.getUUID())) {
                        edgeParsers.put(edge.getUUID(), new EdgeParser("", (Edge) edge));
                    }
                }
            }
        }

        @Override
        public void startEdgeTraversal(Edge<Child, Node> edge) {
            super.startEdgeTraversal(edge);
            parsers.push(currentParser);
        }

        @Override
        public void endEdgeTraversal(Edge<Child, Node> edge) {
            super.endEdgeTraversal(edge);
            currentParser = parsers.pop();
        }

        @Override
        public void endGraphTraversal() {
            super.endGraphTraversal();
        }
    });
    // In oryx format, all edges are added into the main BPMNDiagram node.
    if (null != rootParser && !edgeParsers.isEmpty()) {
        for (EdgeParser edgeParser : edgeParsers.values()) {
            rootParser.addChild(edgeParser);
        }
    }
    // Initialize all the element parsers added in the tree.
    BPMN2JsonParser.this.rootParser.initialize(parsingContext);
    System.out.println("End of children and view traverse");
}

Example 72 with Edge

use of org.kie.workbench.common.stunner.core.graph.Edge in project kie-wb-common by kiegroup.

the class CaseManagementSetChildNodeGraphCommand method getEdgeForTarget.

@SuppressWarnings("unchecked")
private Optional<Edge> getEdgeForTarget(final Optional<Node> parent, final Node child) {
    return parent.flatMap((p) -> {
        Edge edge = null;
        final List<Edge> outEdges = parent.get().getOutEdges();
        if (!(outEdges == null || outEdges.isEmpty())) {
            for (Edge outEdge : outEdges) {
                if (outEdge.getContent() instanceof Child) {
                    final Node targetNode = outEdge.getTargetNode();
                    if (child.equals(targetNode)) {
                        edge = outEdge;
                        break;
                    }
                }
            }
        }
        return Optional.ofNullable(edge);
    });
}

Example 73 with Edge

use of org.kie.workbench.common.stunner.core.graph.Edge in project kie-wb-common by kiegroup.

the class CaseManagementCanvasHandlerTest method checkRemoveChildRenderableShapes.

@Test
@SuppressWarnings("unchecked")
public void checkRemoveChildRenderableShapes() {
    final ActivityShape parentShape = makeShape();
    final ActivityShape childShape = makeShape();
    final Node<View<BPMNViewDefinition>, Edge> parentNode = makeNode("parent", parentShape);
    final Node<View<BPMNViewDefinition>, Edge> childNode = makeNode("child", childShape);
    handler.register(parentShape, parentNode, true);
    handler.register(childShape, childNode, true);
    handler.addChild(parentNode, childNode);
    handler.removeChild(parentNode, childNode);
    verify(canvas, times(1)).deleteChildShape(parentShape, childShape);
    verify(layer, never()).removeShape(childShape);
}

Example 74 with Edge

use of org.kie.workbench.common.stunner.core.graph.Edge in project kie-wb-common by kiegroup.

the class CaseManagementCanvasHandlerTest method checkRegisterRenderableShapes.

@Test
@SuppressWarnings("unchecked")
public void checkRegisterRenderableShapes() {
    final ActivityShape shape = makeShape();
    final Node<View<BPMNViewDefinition>, Edge> node = makeNode("uuid", shape);
    handler.register(shape, node, true);
    verify(canvas, times(1)).addShape(eq(shape));
    verify(canvas, times(1)).draw();
}

Example 75 with Edge

use of org.kie.workbench.common.stunner.core.graph.Edge in project kie-wb-common by kiegroup.

the class CaseManagementCanvasHandlerTest method checkDeregisterRenderableShapes.

@Test
public void checkDeregisterRenderableShapes() {
    final ActivityShape shape = makeShape();
    final Node<View<BPMNViewDefinition>, Edge> node = makeNode("uuid", shape);
    handler.deregister(shape, node, true);
    verify(canvas, times(1)).deleteShape(eq(shape));
    verify(canvas, times(1)).draw();
}