Cleanup (formatting; minor optimization)

master
Roman Shevchenko 9 years ago
parent bfddced4e2
commit c825dd68ec
  1. 58
      src/org/jetbrains/java/decompiler/struct/consts/ConstantPool.java

@ -1,5 +1,5 @@
/* /*
* Copyright 2000-2015 JetBrains s.r.o. * Copyright 2000-2016 JetBrains s.r.o.
* *
* Licensed under the Apache License, Version 2.0 (the "License"); * Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License. * you may not use this file except in compliance with the License.
@ -27,20 +27,21 @@ import org.jetbrains.java.decompiler.util.DataInputFullStream;
import java.io.DataInputStream; import java.io.DataInputStream;
import java.io.IOException; import java.io.IOException;
import java.util.ArrayList; import java.util.ArrayList;
import java.util.BitSet;
import java.util.List; import java.util.List;
@SuppressWarnings("AssignmentToForLoopParameter")
public class ConstantPool implements NewClassNameBuilder { public class ConstantPool implements NewClassNameBuilder {
public static final int FIELD = 1; public static final int FIELD = 1;
public static final int METHOD = 2; public static final int METHOD = 2;
private final List<PooledConstant> pool = new ArrayList<PooledConstant>(); private final List<PooledConstant> pool;
private final PoolInterceptor interceptor; private final PoolInterceptor interceptor;
public ConstantPool(DataInputStream in) throws IOException { public ConstantPool(DataInputStream in) throws IOException {
int size = in.readUnsignedShort(); int size = in.readUnsignedShort();
int[] pass = new int[size]; pool = new ArrayList<PooledConstant>(size);
BitSet[] nextPass = {new BitSet(size), new BitSet(size), new BitSet(size)};
// first dummy constant // first dummy constant
pool.add(null); pool.add(null);
@ -53,54 +54,59 @@ public class ConstantPool implements NewClassNameBuilder {
case CodeConstants.CONSTANT_Utf8: case CodeConstants.CONSTANT_Utf8:
pool.add(new PrimitiveConstant(CodeConstants.CONSTANT_Utf8, in.readUTF())); pool.add(new PrimitiveConstant(CodeConstants.CONSTANT_Utf8, in.readUTF()));
break; break;
case CodeConstants.CONSTANT_Integer: case CodeConstants.CONSTANT_Integer:
pool.add(new PrimitiveConstant(CodeConstants.CONSTANT_Integer, new Integer(in.readInt()))); pool.add(new PrimitiveConstant(CodeConstants.CONSTANT_Integer, new Integer(in.readInt())));
break; break;
case CodeConstants.CONSTANT_Float: case CodeConstants.CONSTANT_Float:
pool.add(new PrimitiveConstant(CodeConstants.CONSTANT_Float, new Float(in.readFloat()))); pool.add(new PrimitiveConstant(CodeConstants.CONSTANT_Float, new Float(in.readFloat())));
break; break;
case CodeConstants.CONSTANT_Long: case CodeConstants.CONSTANT_Long:
pool.add(new PrimitiveConstant(CodeConstants.CONSTANT_Long, new Long(in.readLong()))); pool.add(new PrimitiveConstant(CodeConstants.CONSTANT_Long, new Long(in.readLong())));
pool.add(null); pool.add(null);
i++; i++;
break; break;
case CodeConstants.CONSTANT_Double: case CodeConstants.CONSTANT_Double:
pool.add(new PrimitiveConstant(CodeConstants.CONSTANT_Double, new Double(in.readDouble()))); pool.add(new PrimitiveConstant(CodeConstants.CONSTANT_Double, new Double(in.readDouble())));
pool.add(null); pool.add(null);
i++; i++;
break; break;
case CodeConstants.CONSTANT_Class: case CodeConstants.CONSTANT_Class:
case CodeConstants.CONSTANT_String: case CodeConstants.CONSTANT_String:
case CodeConstants.CONSTANT_MethodType: case CodeConstants.CONSTANT_MethodType:
pool.add(new PrimitiveConstant(tag, in.readUnsignedShort())); pool.add(new PrimitiveConstant(tag, in.readUnsignedShort()));
pass[i] = 1; nextPass[0].set(i);
break; break;
case CodeConstants.CONSTANT_NameAndType:
pool.add(new LinkConstant(tag, in.readUnsignedShort(), in.readUnsignedShort()));
nextPass[0].set(i);
break;
case CodeConstants.CONSTANT_Fieldref: case CodeConstants.CONSTANT_Fieldref:
case CodeConstants.CONSTANT_Methodref: case CodeConstants.CONSTANT_Methodref:
case CodeConstants.CONSTANT_InterfaceMethodref: case CodeConstants.CONSTANT_InterfaceMethodref:
case CodeConstants.CONSTANT_NameAndType:
case CodeConstants.CONSTANT_InvokeDynamic: case CodeConstants.CONSTANT_InvokeDynamic:
pool.add(new LinkConstant(tag, in.readUnsignedShort(), in.readUnsignedShort())); pool.add(new LinkConstant(tag, in.readUnsignedShort(), in.readUnsignedShort()));
if (tag == CodeConstants.CONSTANT_NameAndType) { nextPass[1].set(i);
pass[i] = 1;
}
else {
pass[i] = 2;
}
break; break;
case CodeConstants.CONSTANT_MethodHandle: case CodeConstants.CONSTANT_MethodHandle:
pool.add(new LinkConstant(tag, in.readUnsignedByte(), in.readUnsignedShort())); pool.add(new LinkConstant(tag, in.readUnsignedByte(), in.readUnsignedShort()));
pass[i] = 3; nextPass[2].set(i);
break; break;
} }
} }
// resolving complex pool elements // resolving complex pool elements
for (int passIndex = 1; passIndex <= 3; passIndex++) { for (BitSet pass : nextPass) {
for (int i = 1; i < size; i++) { int idx = 0;
if (pass[i] == passIndex) { while ((idx = pass.nextSetBit(idx + 1)) > 0) {
pool.get(i).resolveConstant(this); pool.get(idx).resolveConstant(this);
}
} }
} }
@ -116,6 +122,7 @@ public class ConstantPool implements NewClassNameBuilder {
case CodeConstants.CONSTANT_Utf8: case CodeConstants.CONSTANT_Utf8:
in.readUTF(); in.readUTF();
break; break;
case CodeConstants.CONSTANT_Integer: case CodeConstants.CONSTANT_Integer:
case CodeConstants.CONSTANT_Float: case CodeConstants.CONSTANT_Float:
case CodeConstants.CONSTANT_Fieldref: case CodeConstants.CONSTANT_Fieldref:
@ -125,16 +132,19 @@ public class ConstantPool implements NewClassNameBuilder {
case CodeConstants.CONSTANT_InvokeDynamic: case CodeConstants.CONSTANT_InvokeDynamic:
in.discard(4); in.discard(4);
break; break;
case CodeConstants.CONSTANT_Long: case CodeConstants.CONSTANT_Long:
case CodeConstants.CONSTANT_Double: case CodeConstants.CONSTANT_Double:
in.discard(8); in.discard(8);
i++; i++;
break; break;
case CodeConstants.CONSTANT_Class: case CodeConstants.CONSTANT_Class:
case CodeConstants.CONSTANT_String: case CodeConstants.CONSTANT_String:
case CodeConstants.CONSTANT_MethodType: case CodeConstants.CONSTANT_MethodType:
in.discard(2); in.discard(2);
break; break;
case CodeConstants.CONSTANT_MethodHandle: case CodeConstants.CONSTANT_MethodHandle:
in.discard(3); in.discard(3);
} }
@ -155,7 +165,7 @@ public class ConstantPool implements NewClassNameBuilder {
className = oldClassName; className = oldClassName;
} }
String newElement = interceptor.getName(className + " " + elementName + " " + descriptor); String newElement = interceptor.getName(className + ' ' + elementName + ' ' + descriptor);
if (newElement != null) { if (newElement != null) {
elementName = newElement.split(" ")[1]; elementName = newElement.split(" ")[1];
} }
@ -196,7 +206,7 @@ public class ConstantPool implements NewClassNameBuilder {
ln.type == CodeConstants.CONSTANT_Methodref || ln.type == CodeConstants.CONSTANT_Methodref ||
ln.type == CodeConstants.CONSTANT_InterfaceMethodref)) { ln.type == CodeConstants.CONSTANT_InterfaceMethodref)) {
String newClassName = buildNewClassname(ln.classname); String newClassName = buildNewClassname(ln.classname);
String newElement = interceptor.getName(ln.classname + " " + ln.elementname + " " + ln.descriptor); String newElement = interceptor.getName(ln.classname + ' ' + ln.elementname + ' ' + ln.descriptor);
String newDescriptor = buildNewDescriptor(ln.type == CodeConstants.CONSTANT_Fieldref, ln.descriptor); String newDescriptor = buildNewDescriptor(ln.type == CodeConstants.CONSTANT_Fieldref, ln.descriptor);
if (newClassName != null || newElement != null || newDescriptor != null) { if (newClassName != null || newElement != null || newDescriptor != null) {
@ -220,10 +230,10 @@ public class ConstantPool implements NewClassNameBuilder {
if (vt.arrayDim > 0) { if (vt.arrayDim > 0) {
for (int i = 0; i < vt.arrayDim; i++) { for (int i = 0; i < vt.arrayDim; i++) {
buffer.append("["); buffer.append('[');
} }
buffer.append("L").append(newName).append(";"); buffer.append('L').append(newName).append(';');
} }
else { else {
buffer.append(newName); buffer.append(newName);
@ -243,4 +253,4 @@ public class ConstantPool implements NewClassNameBuilder {
return MethodDescriptor.parseDescriptor(descriptor).buildNewDescriptor(this); return MethodDescriptor.parseDescriptor(descriptor).buildNewDescriptor(this);
} }
} }
} }
Loading…
Cancel
Save