KickJava   Java API By Example, From Geeks To Geeks.

Java > Open Source Codes > soot > jimple > toolkits > annotation > profiling > ProfilingGenerator


1 /* Soot - a J*va Optimization Framework
2  * Copyright (C) 2000 Feng Qian
3  *
4  * This library is free software; you can redistribute it and/or
5  * modify it under the terms of the GNU Lesser General Public
6  * License as published by the Free Software Foundation; either
7  * version 2.1 of the License, or (at your option) any later version.
8  *
9  * 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.annotation.profiling;
27
28 import soot.*;
29 import soot.jimple.*;
30 import soot.toolkits.graph.*;
31 import soot.util.*;
32
33 import soot.tagkit.*;
34
35 import java.util.*;
36 import soot.options.ProfilingOptions;
37
38 public class ProfilingGenerator extends BodyTransformer
39 {
40     public ProfilingGenerator( Singletons.Global g ) {}
41     public static ProfilingGenerator v() { return G.v().soot_jimple_toolkits_annotation_profiling_ProfilingGenerator(); }
42
43     public String JavaDoc mainSignature = "void main(java.lang.String[])";
44
45     // private String mainSignature = "long runBenchmark(java.lang.String[])";
46
47     protected void internalTransform(Body body, String JavaDoc phaseName, Map opts)
48     {
49         ProfilingOptions options = new ProfilingOptions( opts );
50     if (options.notmainentry())
51         mainSignature = "long runBenchmark(java.lang.String[])";
52
53     {
54         SootMethod m = body.getMethod();
55
56         SootClass counterClass = Scene.v().loadClassAndSupport("MultiCounter");
57         SootMethod reset = counterClass.getMethod("void reset()") ;
58         SootMethod report = counterClass.getMethod("void report()") ;
59         
60         boolean isMainMethod= m.getSubSignature().equals(mainSignature);
61         
62         Chain units = body.getUnits();
63
64         if (isMainMethod)
65         {
66             units.addFirst(Jimple.v().newInvokeStmt(
67                    Jimple.v().newStaticInvokeExpr(reset.makeRef())));
68         }
69
70         Iterator stmtIt = body.getUnits().snapshotIterator();
71         while (stmtIt.hasNext())
72         {
73             Stmt stmt = (Stmt)stmtIt.next();
74
75         if (stmt instanceof InvokeStmt)
76         {
77             InvokeExpr iexpr = (InvokeExpr)
78               ((InvokeStmt)stmt).getInvokeExpr() ;
79         
80             if (iexpr instanceof StaticInvokeExpr)
81             {
82                 SootMethod tempm = ((StaticInvokeExpr)iexpr).getMethod() ;
83             
84             if (tempm.getSignature().equals(
85                 "<java.lang.System: void exit(int)>"))
86             {
87                 units.insertBefore (Jimple.v().newInvokeStmt(
88                     Jimple.v().newStaticInvokeExpr(report.makeRef())), stmt) ;
89
90             }
91             }
92         }
93         else
94         if (isMainMethod
95             && ( stmt instanceof ReturnStmt
96              || stmt instanceof ReturnVoidStmt))
97         {
98             units.insertBefore(Jimple.v().newInvokeStmt(
99                 Jimple.v().newStaticInvokeExpr(report.makeRef())), stmt);
100         }
101         }
102     }
103     }
104 }
105
Popular Tags