KickJava   Java API By Example, From Geeks To Geeks.

Java > Open Source Codes > edu > umd > cs > findbugs > CallGraph


1 /*
2  * FindBugs - Find bugs in Java programs
3  * Copyright (C) 2003,2004 University of Maryland
4  *
5  * This library is free software; you can redistribute it and/or
6  * modify it under the terms of the GNU Lesser General Public
7  * License as published by the Free Software Foundation; either
8  * version 2.1 of the License, or (at your option) any later version.
9  *
10  * This library is distributed in the hope that it will be useful,
11  * but WITHOUT ANY WARRANTY; without even the implied warranty of
12  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
13  * Lesser General Public License for more details.
14  *
15  * You should have received a copy of the GNU Lesser General Public
16  * License along with this library; if not, write to the Free Software
17  * Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
18  */
19
20 package edu.umd.cs.findbugs;
21
22 import java.util.IdentityHashMap JavaDoc;
23
24 import org.apache.bcel.classfile.Method;
25
26 import edu.umd.cs.findbugs.graph.AbstractGraph;
27
28 public class CallGraph extends AbstractGraph<CallGraphEdge, CallGraphNode> {
29     private IdentityHashMap JavaDoc<Method, CallGraphNode> methodToNodeMap;
30
31     public CallGraph() {
32         this.methodToNodeMap = new IdentityHashMap JavaDoc<Method, CallGraphNode>();
33     }
34
35     public CallGraphEdge createEdge(CallGraphNode source, CallGraphNode target, CallSite callSite) {
36         CallGraphEdge edge = createEdge(source, target);
37         edge.setCallSite(callSite);
38         return edge;
39     }
40
41     public CallGraphNode addNode(Method method) {
42         CallGraphNode node = new CallGraphNode();
43         addVertex(node);
44         node.setMethod(method);
45         methodToNodeMap.put(method, node);
46         return node;
47     }
48
49     public CallGraphNode getNodeForMethod(Method method) {
50         return methodToNodeMap.get(method);
51     }
52
53     @Override JavaDoc
54     protected CallGraphEdge allocateEdge(CallGraphNode source, CallGraphNode target) {
55         return new CallGraphEdge(source, target);
56     }
57 }
58
59 // vim:ts=4
60
Popular Tags