01: /* Soot - a J*va Optimization Framework
02: * Copyright (C) 2003 Ondrej Lhotak
03: *
04: * This library is free software; you can redistribute it and/or
05: * modify it under the terms of the GNU Lesser General Public
06: * License as published by the Free Software Foundation; either
07: * version 2.1 of the License, or (at your option) any later version.
08: *
09: * This library is distributed in the hope that it will be useful,
10: * but WITHOUT ANY WARRANTY; without even the implied warranty of
11: * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
12: * Lesser General Public License for more details.
13: *
14: * You should have received a copy of the GNU Lesser General Public
15: * License along with this library; if not, write to the
16: * Free Software Foundation, Inc., 59 Temple Place - Suite 330,
17: * Boston, MA 02111-1307, USA.
18: */
19:
20: /*
21: * Modified by the Sable Research Group and others 1997-1999.
22: * See the 'credits' file distributed with Soot for the complete list of
23: * contributors. (Soot is distributed at http://www.sable.mcgill.ca/soot)
24: */
25:
26: package soot.jimple.toolkits.callgraph;
27:
28: import soot.*;
29: import soot.util.*;
30: import java.util.*;
31:
32: public class TopologicalOrderer {
33: CallGraph cg;
34: List<SootMethod> order = new ArrayList<SootMethod>();
35: NumberedSet visited = new NumberedSet(Scene.v().getMethodNumberer());
36:
37: public TopologicalOrderer(CallGraph cg) {
38: this .cg = cg;
39: }
40:
41: public void go() {
42: Iterator methods = cg.sourceMethods();
43: while (methods.hasNext()) {
44: SootMethod m = (SootMethod) methods.next();
45: dfsVisit(m);
46: }
47: }
48:
49: private void dfsVisit(SootMethod m) {
50: if (visited.contains(m))
51: return;
52: visited.add(m);
53: Iterator targets = new Targets(cg.edgesOutOf(m));
54: while (targets.hasNext()) {
55: SootMethod target = (SootMethod) targets.next();
56: dfsVisit(target);
57: }
58: order.add(m);
59: }
60:
61: public List<SootMethod> order() {
62: return order;
63: }
64: }
|