00001 package edu.ksu.cis.bandera.abstraction.gui;
00002
00003
00004
00005
00006
00007
00008
00009
00010
00011
00012
00013
00014
00015
00016
00017
00018
00019
00020
00021
00022
00023
00024
00025
00026
00027
00028
00029
00030
00031
00032
00033
00034
00035
00036
00037
00038 import java.util.*;
00039 import javax.swing.*;
00040 import javax.swing.table.*;
00041 import ca.mcgill.sable.soot.*;
00042 import javax.swing.tree.TreeNode;
00043 import ca.mcgill.sable.soot.jimple.*;
00044 import edu.ksu.cis.bandera.abstraction.*;
00045 import edu.ksu.cis.bandera.abstraction.util.*;
00046 public class MethodTreeNode extends TypeTreeNode {
00047 static Hashtable methodsTable;
00048 protected SootMethod method;
00049
00050
00051
00052 public MethodTreeNode(SootMethod method) {
00053 super(method.getName());
00054 this.method = method;
00055 methodsTable.put(method, this);
00056 if (!method.isBodyStored(Jimple.v()))
00057 new BuildAndStoreBody(Jimple.v(), new StoredBody(ClassFile.v()), 0).resolveFor(method);
00058 JimpleBody body = (JimpleBody) method.getBody(Jimple.v());
00059 for (ca.mcgill.sable.util.Iterator i = body.getLocals().iterator(); i.hasNext();) {
00060 Local l = (Local) i.next();
00061 if (!l.getName().startsWith("JJJCTEMP$") && !"$ret".equals(l.getName()))
00062 addChild(new LocalTreeNode(l));
00063 }
00064 }
00065
00066
00067
00068
00069
00070
00071
00072 public TableCellEditor getCellEditor(int column) {
00073
00074
00075
00076
00077
00078
00079
00080 return null;
00081 }
00082
00083
00084
00085
00086
00087
00088
00089 public TableCellRenderer getCellRenderer(int column) {
00090 if (column == 1) {
00091 DefaultTableCellRenderer renderer = new DefaultTableCellRenderer();
00092 renderer.setToolTipText(method.getParameterTypes().toString());
00093 return renderer;
00094 } else
00095
00096
00097
00098
00099
00100
00101 return null;
00102 }
00103
00104
00105
00106
00107
00108 public SootMethod getMethod() {
00109 return method;
00110 }
00111
00112
00113
00114
00115
00116 public void getOption(Hashtable result) {
00117 HashSet locals = new HashSet();
00118 for (int i = 0; i < getChildCount(); i++) {
00119 LocalTreeNode local = (LocalTreeNode) getChildAt(i);
00120 if (local.getAbstraction() != null) {
00121 Local lcl = (Local) local.getVar();
00122 locals.add(lcl);
00123 LocalMethod lm = new LocalMethod(method, lcl);
00124 result.put(lm, local.getAbstraction());
00125 }
00126 }
00127 result.put(method, locals);
00128 ((HashSet) result.get(method.getDeclaringClass())).add(method);
00129 }
00130
00131
00132
00133
00134
00135 public String getParams() {
00136 return method.getParameterTypes().toString().replace('{', '(').replace('}', ')');
00137 }
00138
00139
00140
00141
00142
00143 public String toString() {
00144 String name = getName().toString();
00145 if ("<init>".equals(name)) {
00146 name = method.getDeclaringClass().getName();
00147 int i = name.lastIndexOf(".");
00148 if (i != -1) name = name.substring(i + 1);
00149 }
00150 return name;
00151 }
00152 }