Propagated bytecode-to-source tracer

src/org/jetbrains/java/decompiler/code/cfg/BasicBlock.java

@@ -93,6 +93,14 @@ public class BasicBlock implements IGraphNode {
     }
   }
 
+  public Integer getOldOffset(int index) {
+    if(index < instrOldOffsets.size()) {
+      return instrOldOffsets.get(index);
+    } else {
+      return -1;
+    }
+  }
+
   public int size() {
     return seq.length();
   }

src/org/jetbrains/java/decompiler/main/ClassWriter.java

@@ -164,7 +164,8 @@ public class ClassWriter {
     ClassNode outerNode = (ClassNode)DecompilerContext.getProperty(DecompilerContext.CURRENT_CLASS_NODE);
     DecompilerContext.setProperty(DecompilerContext.CURRENT_CLASS_NODE, node);
 
-    BytecodeMappingTracer tracer = new BytecodeMappingTracer();
+    int total_offset_lines = 0;
+    BytecodeMappingTracer dummy_tracer = new BytecodeMappingTracer();
 
     try {
       // last minute processing
@@ -177,8 +178,13 @@ public class ClassWriter {
 
       String lineSeparator = DecompilerContext.getNewLineSeparator();
 
+      // write class definition
+      int start_class_def = buffer.length();
       writeClassDefinition(node, buffer, indent);
 
+      // count lines in class definition the easiest way
+      total_offset_lines = buffer.substring(start_class_def).toString().split(lineSeparator, -1).length - 1;
+
       boolean hasContent = false;
 
       // fields
@@ -204,7 +210,7 @@ public class ClassWriter {
           enumFields = false;
         }
 
-        fieldToJava(wrapper, cl, fd, buffer, indent + 1, tracer);
+        fieldToJava(wrapper, cl, fd, buffer, indent + 1, dummy_tracer); // FIXME: insert real tracer
 
         hasContent = true;
       }
@@ -225,9 +231,13 @@ public class ClassWriter {
         if (hasContent) {
           buffer.append(lineSeparator);
         }
-        boolean methodSkipped = !methodToJava(node, mt, buffer, indent + 1, tracer);
+        BytecodeMappingTracer method_tracer = new BytecodeMappingTracer(total_offset_lines);
+        boolean methodSkipped = !methodToJava(node, mt, buffer, indent + 1, method_tracer);
         if (!methodSkipped) {
           hasContent = true;
+          DecompilerContext.getBytecodeSourceMapper().addTracer(cl.qualifiedName,
+                                  InterpreterUtil.makeUniqueKey(mt.getName(), mt.getDescriptor()), method_tracer);
+          total_offset_lines = method_tracer.getCurrentSourceline();
         }
         else {
           buffer.setLength(position);
@@ -558,6 +568,10 @@ public class ClassWriter {
     MethodWrapper methodWrapper = wrapper.getMethodWrapper(mt.getName(), mt.getDescriptor());
 
     boolean hideMethod = false;
+    int start_index_method = buffer.length();
+
+    String indentString = InterpreterUtil.getIndentString(indent);
+    String lineSeparator = DecompilerContext.getNewLineSeparator();
 
     MethodWrapper outerWrapper = (MethodWrapper)DecompilerContext.getProperty(DecompilerContext.CURRENT_METHOD_WRAPPER);
     DecompilerContext.setProperty(DecompilerContext.CURRENT_METHOD_WRAPPER, methodWrapper);
@@ -569,9 +583,6 @@ public class ClassWriter {
       boolean isDeprecated = mt.getAttributes().containsKey("Deprecated");
       boolean clinit = false, init = false, dinit = false;
 
-      String indentString = InterpreterUtil.getIndentString(indent);
-      String lineSeparator = DecompilerContext.getNewLineSeparator();
-
       MethodDescriptor md = MethodDescriptor.parseDescriptor(mt.getDescriptor());
 
       int flags = mt.getAccessFlags();
@@ -793,6 +804,9 @@ public class ClassWriter {
 
         if (root != null && !methodWrapper.decompiledWithErrors) { // check for existence
           try {
+
+            tracer.setCurrentSourceline(buffer.substring(start_index_method).split(lineSeparator, -1).length - 1);
+
             String code = root.toJava(indent + 1, tracer);
 
             hideMethod = (clinit || dinit || hideConstructor(wrapper, init, throwsExceptions, paramCount)) && code.length() == 0;
@@ -820,6 +834,10 @@ public class ClassWriter {
       DecompilerContext.setProperty(DecompilerContext.CURRENT_METHOD_WRAPPER, outerWrapper);
     }
 
+    // save total lines
+    // TODO: optimize
+    tracer.setCurrentSourceline(buffer.substring(start_index_method).split(lineSeparator, -1).length - 1);
+
     return !hideMethod;
   }
 

src/org/jetbrains/java/decompiler/main/ClassesProcessor.java

@@ -262,10 +262,13 @@ public class ClassesProcessor {
       new ClassWriter().classToJava(root, classBuffer, 0);
 
       String lineSeparator = DecompilerContext.getNewLineSeparator();
+      int total_offset_lines = 0;
 
       int index = cl.qualifiedName.lastIndexOf("/");
       if (index >= 0) {
+        total_offset_lines++;
         String packageName = cl.qualifiedName.substring(0, index).replace('/', '.');
+
         buffer.append("package ");
         buffer.append(packageName);
         buffer.append(";");
@@ -273,15 +276,20 @@ public class ClassesProcessor {
         buffer.append(lineSeparator);
       }
 
-      if (importCollector.writeImports(buffer)) {
+      int import_lines_written = importCollector.writeImports(buffer);
+      if (import_lines_written > 0) {
         buffer.append(lineSeparator);
+        total_offset_lines += import_lines_written + 1;
       }
 
       buffer.append(classBuffer);
 
       if(DecompilerContext.getOption(IFernflowerPreferences.BYTECODE_SOURCE_MAPPING)) {
+        BytecodeSourceMapper mapper = DecompilerContext.getBytecodeSourceMapper();
+        mapper.addTotalOffset(total_offset_lines);
+
         buffer.append(lineSeparator);
-        DecompilerContext.getBytecodeSourceMapper().dumpMapping(classBuffer);
+        mapper.dumpMapping(buffer);
       }
     }
     finally {

src/org/jetbrains/java/decompiler/main/collectors/BytecodeMappingTracer.java

@@ -1,26 +1,53 @@
 package org.jetbrains.java.decompiler.main.collectors;
 
 import java.util.HashMap;
+import java.util.Set;
 
 public class BytecodeMappingTracer {
 
   private int current_sourceline;
 
   // bytecode offset, source line
-  private HashMap<Integer, Integer> mapping;
+  private HashMap<Integer, Integer> mapping = new HashMap<Integer, Integer>();
+
+  public BytecodeMappingTracer() {}
+
+  public BytecodeMappingTracer(int initial_source_line) {
+    current_sourceline = initial_source_line;
+  }
 
   public void incrementSourceLine() {
     current_sourceline++;
   }
 
+  public void incrementSourceLine(int number_lines) {
+    current_sourceline += number_lines;
+  }
+
   public void addMapping(int bytecode_offset) {
     if(!mapping.containsKey(bytecode_offset)) {
       mapping.put(bytecode_offset, current_sourceline);
     }
   }
 
+  public void addMapping(Set<Integer> bytecode_offsets) {
+    if(bytecode_offsets != null) {
+      for(Integer bytecode_offset : bytecode_offsets) {
+        addMapping(bytecode_offset);
+      }
+    }
+  }
+
   public HashMap<Integer, Integer> getMapping() {
     return mapping;
   }
 
+  public int getCurrentSourceline() {
+    return current_sourceline;
+  }
+
+  public void setCurrentSourceline(int current_sourceline) {
+    this.current_sourceline = current_sourceline;
+  }
+
 }

src/org/jetbrains/java/decompiler/main/collectors/BytecodeSourceMapper.java

@@ -11,7 +11,7 @@ public class BytecodeSourceMapper {
   private int offset_total;
 
   // class, method, bytecode offset, source line
-  private HashMap<String, HashMap<String, HashMap<Integer, Integer>>> mapping;
+  private HashMap<String, HashMap<String, HashMap<Integer, Integer>>> mapping = new HashMap<String, HashMap<String, HashMap<Integer, Integer>>>();
 
   public void addMapping(String classname, String methodname, int bytecode_offset, int source_line) {
 
@@ -31,6 +31,12 @@ public class BytecodeSourceMapper {
     }
   }
 
+  public void addTracer(String classname, String methodname, BytecodeMappingTracer tracer) {
+    for(Entry<Integer, Integer> entry : tracer.getMapping().entrySet()) {
+      addMapping(classname, methodname, entry.getKey(), entry.getValue());
+    }
+  }
+
   public void dumpMapping(StringBuilder buffer) {
 
     String lineSeparator = DecompilerContext.getNewLineSeparator();
@@ -70,6 +76,9 @@ public class BytecodeSourceMapper {
     this.offset_total = offset_total;
   }
 
+  public void addTotalOffset(int offset_total) {
+    this.offset_total += offset_total;
+  }
 
 
 }

src/org/jetbrains/java/decompiler/main/collectors/ImportCollector.java

@@ -107,17 +107,23 @@ public class ImportCollector {
     return retname == null ? nshort : retname;
   }
 
-  public boolean writeImports(StringBuilder buffer) {
+  public int writeImports(StringBuilder buffer) {
+
+    int importlines_written = 0;
+    String new_line_separator = DecompilerContext.getNewLineSeparator();
+
     List<String> imports = packImports();
 
     for (String s : imports) {
       buffer.append("import ");
       buffer.append(s);
       buffer.append(";");
-      buffer.append(DecompilerContext.getNewLineSeparator());
+      buffer.append(new_line_separator);
+
+      importlines_written++;
     }
 
-    return imports.size() > 0;
+    return importlines_written;
   }
 
   private List<String> packImports() {

src/org/jetbrains/java/decompiler/modules/decompiler/ExprProcessor.java

@@ -331,6 +331,7 @@ public class ExprProcessor implements CodeConstants {
     for (int i = 0; i < seq.length(); i++) {
 
       Instruction instr = seq.getInstr(i);
+      Integer bytecode_offset = block.getOldOffset(i);
 
       switch (instr.opcode) {
         case opc_aconst_null:

src/org/jetbrains/java/decompiler/modules/decompiler/exps/ArrayExprent.java

@@ -96,6 +96,8 @@ public class ArrayExprent extends Exprent {
       res = "((" + ExprProcessor.getCastTypeName(objarr) + ")" + res + ")";
     }
 
+    tracer.addMapping(bytecode);
+
     return res + "[" + index.toJava(indent, tracer) + "]";
   }
 

src/org/jetbrains/java/decompiler/modules/decompiler/exps/AssertExprent.java

@@ -38,12 +38,15 @@ public class AssertExprent extends Exprent {
 
     buffer.append("assert ");
 
+    tracer.addMapping(bytecode);
+
     if (parameters.get(0) == null) {
       buffer.append("false");
     }
     else {
       buffer.append(parameters.get(0).toJava(indent, tracer));
     }
+
     if (parameters.size() > 1) {
       buffer.append(" : ");
       buffer.append(parameters.get(1).toJava(indent, tracer));

src/org/jetbrains/java/decompiler/modules/decompiler/exps/AssignmentExprent.java

@@ -153,6 +153,8 @@ public class AssignmentExprent extends Exprent {
 
     buffer.append(condtype == CONDITION_NONE ? " = " : funceq[condtype]).append(res);
 
+    tracer.addMapping(bytecode);
+
     return buffer.toString();
   }
 

src/org/jetbrains/java/decompiler/modules/decompiler/exps/ConstExprent.java

@@ -112,6 +112,8 @@ public class ConstExprent extends Exprent {
     boolean literal = DecompilerContext.getOption(IFernflowerPreferences.LITERALS_AS_IS);
     boolean ascii = DecompilerContext.getOption(IFernflowerPreferences.ASCII_STRING_CHARACTERS);
 
+    tracer.addMapping(bytecode);
+
     if (consttype.type != CodeConstants.TYPE_NULL && value == null) {
       return ExprProcessor.getCastTypeName(consttype);
     }

src/org/jetbrains/java/decompiler/modules/decompiler/exps/ExitExprent.java

@@ -77,6 +77,9 @@ public class ExitExprent extends Exprent {
 
   @Override
   public String toJava(int indent, BytecodeMappingTracer tracer) {
+
+    tracer.addMapping(bytecode);
+
     if (exittype == EXIT_RETURN) {
       StringBuilder buffer = new StringBuilder();
 

src/org/jetbrains/java/decompiler/modules/decompiler/exps/Exprent.java

@@ -54,6 +54,9 @@ public class Exprent {
 
   public int id;
 
+  //offsets of bytecode instructions decompiled to this exprent
+  public Set<Integer> bytecode = new HashSet<Integer>();
+
   {
     // set exprent id
     id = DecompilerContext.getCounterContainer().getCounterAndIncrement(CounterContainer.EXPRENT_COUNTER);

src/org/jetbrains/java/decompiler/modules/decompiler/exps/FieldExprent.java

@@ -159,6 +159,8 @@ public class FieldExprent extends Exprent {
 
     buf.append(name);
 
+    tracer.addMapping(bytecode);
+
     return buf.toString();
   }
 

src/org/jetbrains/java/decompiler/modules/decompiler/exps/FunctionExprent.java

@@ -450,6 +450,8 @@ public class FunctionExprent extends Exprent {
   @Override
   public String toJava(int indent, BytecodeMappingTracer tracer) {
 
+    tracer.addMapping(bytecode);
+
     if (functype <= FUNCTION_USHR) {
       return wrapOperandString(lstOperands.get(0), false, indent, tracer) + operators[functype] +
              wrapOperandString(lstOperands.get(1), true, indent, tracer);

src/org/jetbrains/java/decompiler/modules/decompiler/exps/IfExprent.java

@@ -114,6 +114,7 @@ public class IfExprent extends Exprent {
 
   @Override
   public String toJava(int indent, BytecodeMappingTracer tracer) {
+    tracer.addMapping(bytecode);
     return "if(" + condition.toJava(indent, tracer) + ")";
   }
 

src/org/jetbrains/java/decompiler/modules/decompiler/exps/InvocationExprent.java

@@ -194,6 +194,8 @@ public class InvocationExprent extends Exprent {
     String super_qualifier = null;
     boolean isInstanceThis = false;
 
+    tracer.addMapping(bytecode);
+
     if (invocationTyp == INVOKE_DYNAMIC) {
       //			ClassNode node = (ClassNode)DecompilerContext.getProperty(DecompilerContext.CURRENT_CLASSNODE);
       //

src/org/jetbrains/java/decompiler/modules/decompiler/exps/MonitorExprent.java

@@ -52,6 +52,9 @@ public class MonitorExprent extends Exprent {
 
   @Override
   public String toJava(int indent, BytecodeMappingTracer tracer) {
+
+    tracer.addMapping(bytecode);
+
     if (montype == MONITOR_ENTER) {
       return "synchronized(" + value.toJava(indent, tracer) + ")";
     }

src/org/jetbrains/java/decompiler/modules/decompiler/exps/SwitchExprent.java

@@ -84,6 +84,7 @@ public class SwitchExprent extends Exprent {
 
   @Override
   public String toJava(int indent, BytecodeMappingTracer tracer) {
+    tracer.addMapping(bytecode);
     return "switch(" + value.toJava(indent, tracer) + ")";
   }
 

src/org/jetbrains/java/decompiler/modules/decompiler/exps/VarExprent.java

@@ -85,6 +85,8 @@ public class VarExprent extends Exprent {
   public String toJava(int indent, BytecodeMappingTracer tracer) {
     StringBuilder buffer = new StringBuilder();
 
+    tracer.addMapping(bytecode);
+
     if (classdef) {
       ClassNode child = DecompilerContext.getClassProcessor().getMapRootClasses().get(vartype.value);
       new ClassWriter().classToJava(child, buffer, indent);

src/org/jetbrains/java/decompiler/modules/decompiler/stats/Statement.java

@@ -489,9 +489,9 @@ public class Statement {
   }
 
   // to be overwritten
-  //public String toJava() {
+  public String toJava() {
-  //  return toJava(0);
+    return toJava(0, new BytecodeMappingTracer());
-  //}
+  }
 
   public String toJava(int indent, BytecodeMappingTracer tracer) {
     throw new RuntimeException("not implemented");