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


1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70

package org.jetbrains.java.decompiler.main.collectors;

import java.util.HashMap;
import java.util.Map.Entry;
import java.util.Set;

public class BytecodeMappingTracer {

  private int current_sourceline;

  // bytecode offset, source line
  private HashMap<Integer, Integer> mapping = new HashMap<Integer, Integer>();

  public BytecodeMappingTracer() {}

  public BytecodeMappingTracer(int initial_source_line) {
    current_sourceline = initial_source_line;
  }

  public void incrementCurrentSourceLine() {
    current_sourceline++;
  }

  public void incrementCurrentSourceLine(int number_lines) {
    current_sourceline += number_lines;
  }

  public void shiftSourceLines(int shift) {
    for(Entry<Integer, Integer> entry : mapping.entrySet()) {
      entry.setValue(entry.getValue() + shift);
    }
  }

  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 void addTracer(BytecodeMappingTracer tracer) {
    if(tracer != null) {
      for(Entry<Integer, Integer> entry : tracer.mapping.entrySet()) {
        if(!mapping.containsKey(entry.getKey())) {
          mapping.put(entry.getKey(), entry.getValue());
        }
      }
    }
  }

  public HashMap<Integer, Integer> getMapping() {
    return mapping;
  }

  public int getCurrentSourceLine() {
    return current_sourceline;
  }

  public void setCurrentSourceLine(int current_sourceline) {
    this.current_sourceline = current_sourceline;
  }

}