KickJava   Java API By Example, From Geeks To Geeks.

Java > Open Source Codes > proguard > optimize > info > MethodInvocationMarker


1 /*
2  * ProGuard -- shrinking, optimization, obfuscation, and preverification
3  * of Java bytecode.
4  *
5  * Copyright (c) 2002-2007 Eric Lafortune (eric@graphics.cornell.edu)
6  *
7  * This library is free software; you can redistribute it and/or modify it
8  * under the terms of the GNU General Public License as published by the Free
9  * Software Foundation; either version 2 of the License, or (at your option)
10  * any later version.
11  *
12  * This library is distributed in the hope that it will be useful, but WITHOUT
13  * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
14  * FITNESS FOR A PARTICULAR PURPOSE. See the GNU Lesser General Public License
15  * for more details.
16  *
17  * You should have received a copy of the GNU Lesser General Public License
18  * along with this library; if not, write to the Free Software Foundation,
19  * Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
20  */
21 package proguard.optimize.info;
22
23 import proguard.classfile.util.SimplifiedVisitor;
24 import proguard.classfile.instruction.visitor.InstructionVisitor;
25 import proguard.classfile.instruction.*;
26 import proguard.classfile.constant.visitor.ConstantVisitor;
27 import proguard.classfile.constant.*;
28 import proguard.classfile.visitor.MemberVisitor;
29 import proguard.classfile.*;
30 import proguard.classfile.attribute.CodeAttribute;
31 import proguard.optimize.info.MethodOptimizationInfo;
32
33 /**
34  * This InstructionVisitor marks all methods that are invoked.
35  *
36  * @author Eric Lafortune
37  */
38 public class MethodInvocationMarker
39 extends SimplifiedVisitor
40 implements InstructionVisitor,
41              ConstantVisitor,
42              MemberVisitor
43 {
44     // Implementations for InstructionVisitor.
45
46     public void visitAnyInstruction(Clazz clazz, Method method, CodeAttribute codeAttribute, int offset, Instruction instruction) {}
47
48
49     public void visitConstantInstruction(Clazz clazz, Method method, CodeAttribute codeAttribute, int offset, ConstantInstruction constantInstruction)
50     {
51         clazz.constantPoolEntryAccept(constantInstruction.constantIndex, this);
52     }
53
54
55     // Implementations for ConstantVisitor.
56
57     public void visitAnyConstant(Clazz clazz, Constant constant) {}
58
59
60     public void visitStringConstant(Clazz clazz, StringConstant stringConstant)
61     {
62         // Mark the referenced method, if any.
63 stringConstant.referencedMemberAccept(this);
64     }
65
66
67     public void visitAnyMethodrefConstant(Clazz clazz, RefConstant refConstant)
68     {
69         // Mark the referenced method.
70 refConstant.referencedMemberAccept(this);
71     }
72
73
74     // Implementations for MemberVisitor.
75
76     public void visitAnyMember(Clazz Clazz, Member member) {}
77
78
79     public void visitProgramMethod(ProgramClass programClass, ProgramMethod programMethod)
80     {
81         incrementInvocationCount(programMethod);
82     }
83
84
85     // Small utility methods.
86
87     private static void incrementInvocationCount(Method method)
88     {
89         MethodOptimizationInfo info = MethodOptimizationInfo.getMethodOptimizationInfo(method);
90         if (info != null)
91         {
92             info.incrementInvocationCount();
93         }
94     }
95
96
97     public static int getInvocationCount(Method method)
98     {
99         MethodOptimizationInfo info = MethodOptimizationInfo.getMethodOptimizationInfo(method);
100         return info != null ? info.getInvocationCount() :
101                               Integer.MAX_VALUE;
102     }
103 }
104
Popular Tags