001/**
002 * www.jcoverage.com
003 * Copyright (C)2003 jcoverage ltd.
004 *
005 * This file is part of jcoverage.
006 *
007 * jcoverage is free software; you can redistribute it and/or modify
008 * it under the terms of the GNU General Public License as published
009 * by the Free Software Foundation; either version 2 of the License,
010 * or (at your option) any later version.
011 *
012 * jcoverage is distributed in the hope that it will be useful, but
013 * WITHOUT ANY WARRANTY; without even the implied warranty of
014 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
015 * General Public License for more details.
016 *
017 * You should have received a copy of the GNU General Public License
018 * along with jcoverage; if not, write to the Free Software
019 * Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307
020 * USA
021 *
022 */
023package com.jcoverage.util;
024
025import java.lang.reflect.Method;
026
027import org.apache.bcel.Constants;
028
029import org.apache.bcel.classfile.ConstantClass;
030import org.apache.bcel.classfile.ConstantUtf8;
031
032import org.apache.bcel.generic.ANEWARRAY;
033import org.apache.bcel.generic.ArrayType;
034import org.apache.bcel.generic.BasicType;
035import org.apache.bcel.generic.BranchHandle;
036import org.apache.bcel.generic.InstructionConstants;
037import org.apache.bcel.generic.GOTO;
038import org.apache.bcel.generic.ClassGen;
039import org.apache.bcel.generic.ConstantPoolGen;
040import org.apache.bcel.generic.FieldGen;
041import org.apache.bcel.generic.IFNONNULL;
042import org.apache.bcel.generic.InstructionFactory;
043import org.apache.bcel.generic.InstructionList;
044import org.apache.bcel.generic.InstructionHandle;
045import org.apache.bcel.generic.InvokeInstruction;
046import org.apache.bcel.generic.LDC;
047import org.apache.bcel.generic.LocalVariableGen;
048import org.apache.bcel.generic.LocalVariableInstruction;
049import org.apache.bcel.generic.MethodGen;
050import org.apache.bcel.generic.ObjectType;
051import org.apache.bcel.generic.NEW;
052
053import org.apache.bcel.generic.Type;
054
055
056public class ClassGenHelper {
057  final ClassGen cg;
058  final InstructionFactory factory;
059
060  private ClassGenHelper(ClassGen cg) {
061    this.cg=cg;
062    this.factory=new InstructionFactory(cg);
063  }
064
065  public static ClassGenHelper newInstance(ClassGen cg) {
066    return new ClassGenHelper(cg);
067  }
068
069  public ClassGen getClassGen() {
070    return cg;
071  }
072  public ConstantPoolGen getConstantPool() {
073    return cg.getConstantPool();
074  }
075
076  private InstructionList createInvoke(Class cl,String methodName,Class returnType,Type[] parameterTypes,short kind) {
077    InstructionList il=new InstructionList();
078    il.append(factory.createInvoke(cl.getName(),methodName,TypeHelper.getType(returnType),parameterTypes,kind));
079    return il;
080  }
081
082  public InstructionList createInvokeVirtual(Class cl,String methodName,Class returnType) {
083    return createInvoke(cl,methodName,returnType,Type.NO_ARGS,Constants.INVOKEVIRTUAL);
084  }
085
086  public InstructionList createInvokeVirtual(Class cl,String methodName,Class returnType,Class signature) {
087    return createInvokeVirtual(cl,methodName,returnType,new Class[]{signature});
088  }
089  public InstructionList createInvokeVirtual(Class cl,String methodName,Class returnType,Class[] signature) {
090    return createInvoke(cl,methodName,returnType,TypeHelper.getTypes(signature),Constants.INVOKEVIRTUAL);
091  }
092
093  public InstructionList createInvokeStatic(Class cl,String methodName,Class returnType) {
094    return createInvoke(cl,methodName,returnType,Type.NO_ARGS,Constants.INVOKESTATIC);
095  }
096
097  public InstructionList createInvokeStatic(Class cl,String methodName,Class returnType,Class[] signature) {
098    return createInvoke(cl,methodName,returnType,TypeHelper.getTypes(signature),Constants.INVOKESTATIC);
099  }
100
101  public InstructionList createInvokeInterface(Class cl,String methodName,Class returnType) {
102    return createInvoke(cl,methodName,returnType,Type.NO_ARGS,Constants.INVOKEINTERFACE);
103  }
104
105  public InstructionList createInvokeInterface(Class cl,String methodName,Class returnType,Class signature) {
106    return createInvokeInterface(cl,methodName,returnType,new Class[]{signature});
107  }
108
109  public InstructionList createInvokeInterface(Class cl,String methodName,Class returnType,Class[] signature) {
110    return createInvoke(cl,methodName,returnType,TypeHelper.getTypes(signature),Constants.INVOKEINTERFACE);
111  }
112
113  public InstructionList createInvokeSpecial(Class cl,String methodName,Class returnType) {
114    return createInvoke(cl,methodName,returnType,Type.NO_ARGS,Constants.INVOKESPECIAL);
115  }
116
117  public InstructionList createInvokeSpecial(Class cl,String methodName,Class returnType,Class[] signature) {
118    return createInvoke(cl,methodName,returnType,TypeHelper.getTypes(signature),Constants.INVOKESPECIAL);
119  }
120
121  public InstructionList createInvokeSpecial(Method m) {
122    return createInvokeSpecial(m.getDeclaringClass(),m.getName(),m.getReturnType(),m.getParameterTypes());
123  }
124
125  public String getClassName(InstructionHandle handleToInvokeInstruction) {
126    return getClassName((InvokeInstruction)handleToInvokeInstruction.getInstruction());
127  }
128
129  public String getClassName(InvokeInstruction invoke) {
130    return invoke.getClassName(getConstantPool());
131  }
132}