2014-10-01 01:00:11 -04:00
|
|
|
package org.jetbrains.java.decompiler.main.collectors;
|
|
|
|
|
2014-10-07 13:03:09 -04:00
|
|
|
import java.util.HashMap;
|
|
|
|
import java.util.Map.Entry;
|
2014-10-01 01:00:11 -04:00
|
|
|
|
2014-10-15 00:26:49 -04:00
|
|
|
import org.jetbrains.java.decompiler.main.DecompilerContext;
|
|
|
|
import org.jetbrains.java.decompiler.main.TextBuffer;
|
|
|
|
|
2014-10-01 01:00:11 -04:00
|
|
|
public class BytecodeSourceMapper {
|
|
|
|
|
|
|
|
private int offset_total;
|
2014-10-05 22:27:26 -04:00
|
|
|
|
2014-10-01 01:00:11 -04:00
|
|
|
// class, method, bytecode offset, source line
|
2014-10-05 22:27:26 -04:00
|
|
|
private HashMap<String, HashMap<String, HashMap<Integer, Integer>>> mapping = new HashMap<String, HashMap<String, HashMap<Integer, Integer>>>();
|
|
|
|
|
2014-10-01 01:00:11 -04:00
|
|
|
public void addMapping(String classname, String methodname, int bytecode_offset, int source_line) {
|
2014-10-05 22:27:26 -04:00
|
|
|
|
2014-10-01 01:00:11 -04:00
|
|
|
HashMap<String, HashMap<Integer, Integer>> class_mapping = mapping.get(classname);
|
|
|
|
if(class_mapping == null) {
|
|
|
|
mapping.put(classname, class_mapping = new HashMap<String, HashMap<Integer, Integer>>());
|
|
|
|
}
|
2014-10-05 22:27:26 -04:00
|
|
|
|
2014-10-01 01:00:11 -04:00
|
|
|
HashMap<Integer, Integer> method_mapping = class_mapping.get(methodname);
|
|
|
|
if(method_mapping == null) {
|
|
|
|
class_mapping.put(methodname, method_mapping = new HashMap<Integer, Integer>());
|
|
|
|
}
|
2014-10-05 22:27:26 -04:00
|
|
|
|
2014-10-01 01:00:11 -04:00
|
|
|
// don't overwrite
|
|
|
|
if(!method_mapping.containsKey(bytecode_offset)) {
|
|
|
|
method_mapping.put(bytecode_offset, source_line);
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2014-10-05 22:27:26 -04:00
|
|
|
public void addTracer(String classname, String methodname, BytecodeMappingTracer tracer) {
|
|
|
|
for(Entry<Integer, Integer> entry : tracer.getMapping().entrySet()) {
|
|
|
|
addMapping(classname, methodname, entry.getKey(), entry.getValue());
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2014-10-15 00:26:49 -04:00
|
|
|
public void dumpMapping(TextBuffer buffer, boolean offsetsToHex) {
|
2014-10-05 22:27:26 -04:00
|
|
|
|
2014-10-01 01:00:11 -04:00
|
|
|
String lineSeparator = DecompilerContext.getNewLineSeparator();
|
2014-10-05 22:27:26 -04:00
|
|
|
|
2014-10-01 01:00:11 -04:00
|
|
|
for(Entry<String, HashMap<String, HashMap<Integer, Integer>>> class_entry : mapping.entrySet()) {
|
|
|
|
HashMap<String, HashMap<Integer, Integer>> class_mapping = class_entry.getValue();
|
|
|
|
buffer.append("class " + class_entry.getKey() + "{" + lineSeparator);
|
2014-10-05 22:27:26 -04:00
|
|
|
|
2014-10-01 01:00:11 -04:00
|
|
|
boolean is_first_method = true;
|
2014-10-05 22:27:26 -04:00
|
|
|
|
2014-10-01 01:00:11 -04:00
|
|
|
for(Entry<String, HashMap<Integer, Integer>> method_entry : class_mapping.entrySet()) {
|
|
|
|
HashMap<Integer, Integer> method_mapping = method_entry.getValue();
|
2014-10-05 22:27:26 -04:00
|
|
|
|
2014-10-01 01:00:11 -04:00
|
|
|
if(!is_first_method) {
|
2014-10-07 13:03:09 -04:00
|
|
|
buffer.appendLineSeparator();
|
2014-10-01 01:00:11 -04:00
|
|
|
}
|
2014-10-07 13:03:09 -04:00
|
|
|
buffer.appendIndent(1).append("method " + method_entry.getKey() + "{" + lineSeparator);
|
2014-10-05 22:27:26 -04:00
|
|
|
|
2014-10-01 01:00:11 -04:00
|
|
|
for(Entry<Integer, Integer> line : method_mapping.entrySet()) {
|
2014-10-15 00:26:49 -04:00
|
|
|
String strOffset = offsetsToHex ? Integer.toHexString(line.getKey()): line.getKey().toString();
|
|
|
|
buffer.appendIndent(2).append(strOffset).appendIndent(2).append((line.getValue() + offset_total) + lineSeparator);
|
2014-10-01 01:00:11 -04:00
|
|
|
}
|
2014-10-07 13:03:09 -04:00
|
|
|
buffer.appendIndent(1).append("}").appendLineSeparator();
|
2014-10-01 01:00:11 -04:00
|
|
|
is_first_method = false;
|
|
|
|
}
|
2014-10-07 13:03:09 -04:00
|
|
|
buffer.append("}").appendLineSeparator();
|
2014-10-01 01:00:11 -04:00
|
|
|
}
|
|
|
|
}
|
2014-10-05 22:27:26 -04:00
|
|
|
|
2014-10-01 01:00:11 -04:00
|
|
|
public int getTotalOffset() {
|
|
|
|
return offset_total;
|
|
|
|
}
|
|
|
|
|
|
|
|
public void setTotalOffset(int offset_total) {
|
|
|
|
this.offset_total = offset_total;
|
|
|
|
}
|
2014-10-05 22:27:26 -04:00
|
|
|
|
|
|
|
public void addTotalOffset(int offset_total) {
|
|
|
|
this.offset_total += offset_total;
|
|
|
|
}
|
|
|
|
|
|
|
|
|
2014-10-01 01:00:11 -04:00
|
|
|
}
|