2 * Copyright 2008 the original author or authors.
4 * Licensed under the Apache License, Version 2.0 (the "License");
5 * you may not use this file except in compliance with the License.
6 * You may obtain a copy of the License at
8 * http://www.apache.org/licenses/LICENSE-2.0
10 * Unless required by applicable law or agreed to in writing, software
11 * distributed under the License is distributed on an "AS IS" BASIS,
12 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
13 * See the License for the specific language governing permissions and
14 * limitations under the License.
16 package org.wamblee.system.graph.component;
18 import java.util.List;
20 import org.wamblee.system.container.InterfaceRestriction;
21 import org.wamblee.system.core.Component;
22 import org.wamblee.system.graph.Edge;
23 import org.wamblee.system.graph.Graph;
24 import org.wamblee.system.graph.Node;
25 import org.wamblee.system.graph.Visitor;
28 * Applies restrictions on a graph by removing all edges that violate the restriction.
30 * @author Erik Brakkee.
33 public class ApplyRestrictionsVisitor implements Visitor {
36 private InterfaceRestriction _restriction;
39 * Constructs the visitor.
40 * @param aGraph Graph.
41 * @param aRestriction Restriction.
43 public ApplyRestrictionsVisitor(Graph aGraph, InterfaceRestriction aRestriction) {
45 _restriction = aRestriction;
49 public void visitEdge(Edge aEdge) {
50 if ( aEdge.getFrom() instanceof RequiredInterfaceNode &&
51 aEdge.getTo() instanceof ProvidedInterfaceNode) {
52 RequiredInterfaceNode from = (RequiredInterfaceNode)aEdge.getFrom();
53 ProvidedInterfaceNode to = (ProvidedInterfaceNode) aEdge.getTo();
55 if ( _restriction.isViolated(from.getComponent(), from.getRequired(), to.getComponent(), to.getProvided())) {
56 _graph.removeEdge(aEdge);
62 public void visitNode(Node aNode) {