update printing

Author: Andreas Dann

src/main/java/de/upb/soot/diff/printing/PrinterUtils.java

@@ -0,0 +1,72 @@
+package de.upb.soot.diff.printing;
+
+import com.google.common.base.Strings;
+import soot.ArrayType;
+import soot.PrimType;
+import soot.RefType;
+import soot.Scene;
+import soot.SootClass;
+import soot.SootMethod;
+import soot.Type;
+import soot.VoidType;
+
+import java.util.List;
+
+public class PrinterUtils {
+    public static String simplifyTypeName(Type type) {
+        String typeName = type.toQuotedString();
+        if (type instanceof RefType) {
+            return simplifyClassName(((RefType) type).getSootClass());
+        } else if (type instanceof ArrayType) {
+            return simplifyTypeName(((ArrayType) type).baseType)
+                    + Strings.repeat("[]", ((ArrayType) type).numDimensions);
+        } else if (type instanceof PrimType) {
+            return typeName;
+        } else if (type instanceof VoidType) {
+            return typeName;
+        }
+
+        throw new IllegalStateException("Unhandled Soot type " + type);
+    }
+
+    public static String simplifyClassName(SootClass cl) {
+        if (Scene.v().isExcluded(cl)) {
+            return (Scene.v().quotedNameOf(cl.getName()));
+        } else {
+            return (Scene.v().quotedNameOf(cl.getShortName()));
+        }
+    }
+
+    public static String getSignature(SootClass cl, String subSignature) {
+        StringBuilder buffer = new StringBuilder();
+        buffer.append("<");
+        buffer.append(simplifyClassName(cl));
+        buffer.append(": ");
+        buffer.append(subSignature);
+        buffer.append(">");
+        return buffer.toString();
+    }
+
+    public static String getSubSignature(String name, List<Type> params, Type returnType) {
+        StringBuilder buffer = new StringBuilder();
+        buffer.append(simplifyTypeName(returnType));
+        buffer.append(" ");
+        buffer.append(Scene.v().quotedNameOf(name));
+        buffer.append("(");
+        if (params != null) {
+            for (int i = 0; i < params.size(); ++i) {
+                buffer.append(simplifyTypeName((Type) params.get(i)));
+                if (i < params.size() - 1) {
+                    buffer.append(",");
+                }
+            }
+        }
+
+        buffer.append(")");
+        return buffer.toString();
+    }
+
+    public static String getMethodSignature(SootMethod method) {
+        return getSubSignature(method.getName(), method.getParameterTypes(), method.getReturnType());
+    }
+}

src/main/java/de/upb/soot/diff/printing/SimpleUnitPrinter.java

@@ -0,0 +1,67 @@
+package de.upb.soot.diff.printing;
+
+import soot.Body;
+import soot.NormalUnitPrinter;
+import soot.RefType;
+import soot.Scene;
+import soot.SootFieldRef;
+import soot.SootMethodRef;
+import soot.Type;
+import soot.jimple.IdentityRef;
+
+/**
+ * Omits the package name of classes that are not in the JDK. Useful for minhashing for example.
+ */
+public class SimpleUnitPrinter extends NormalUnitPrinter {
+    public SimpleUnitPrinter(Body body) {
+        super(body);
+    }
+
+    public static String getSimpleMethodRef(SootMethodRef methodRef) {
+        String subSignature =
+                PrinterUtils.getSubSignature(
+                        methodRef.getName(), methodRef.getParameterTypes(), methodRef.getReturnType());
+        return PrinterUtils.getSignature(methodRef.getDeclaringClass(), subSignature);
+    }
+
+    public static String getSimpleFieldRef(SootFieldRef f) {
+        return PrinterUtils.getSignature(f.declaringClass(), getSubSignature(f.name(), f.type()));
+    }
+
+    public static String getSubSignature(String name, Type type) {
+        StringBuilder buffer = new StringBuilder();
+        buffer.append(PrinterUtils.simplifyTypeName(type) + " " + Scene.v().quotedNameOf(name));
+        return buffer.toString();
+    }
+
+    @Override
+    public void type(Type t) {
+        handleIndent();
+        String s = t == null ? "<null>" : PrinterUtils.simplifyTypeName(t);
+        output.append(s);
+    }
+
+    @Override
+    public void methodRef(SootMethodRef m) {
+        handleIndent();
+        output.append(getSimpleMethodRef(m));
+    }
+
+    @Override
+    public void fieldRef(SootFieldRef f) {
+        handleIndent();
+        output.append(getSimpleFieldRef(f));
+    }
+
+    @Override
+    public void identityRef(IdentityRef r) {
+        Type type = r.getType();
+
+        if (type instanceof RefType) {
+            handleIndent();
+            output.append(PrinterUtils.simplifyClassName(((RefType) type).getSootClass()));
+        } else {
+            super.identityRef(r);
+        }
+    }
+}