1 files changed, 265 insertions, 0 deletions
diff --git a/src/main/javassist/compiler/JvstCodeGenWitlLineNumber.java b/src/main/javassist/compiler/JvstCodeGenWitlLineNumber.java
new file mode 100644
index 00000000..969948c5
--- /dev/null
+++ b/src/main/javassist/compiler/JvstCodeGenWitlLineNumber.java
@@ -0,0 +1,265 @@
+package javassist.compiler;
+
+import javassist.ClassPool;
+import javassist.CtClass;
+import javassist.bytecode.Bytecode;
+import javassist.bytecode.LineNumberAttribute;
+import javassist.bytecode.LineNumberAttributeBuilder;
+import javassist.compiler.ast.*;
+
+public class JvstCodeGenWitlLineNumber extends JvstCodeGen {
+    private final LineNumberAttributeBuilder lineNumberAttributeBuilder = new LineNumberAttributeBuilder();
+    public JvstCodeGenWitlLineNumber(Bytecode b, CtClass cc, ClassPool cp) {
+        super(b, cc, cp);
+    }
+
+    public LineNumberAttribute toLineNumberAttribute() {
+        return lineNumberAttributeBuilder.build(bytecode.getConstPool());
+    }
+
+    @Override
+    public void atASTList(ASTList n) throws CompileError {
+        lineNumberAttributeBuilder.put(bytecode.currentPc(), n);
+        super.atASTList(n);
+    }
+
+    @Override
+    public void atPair(Pair n) throws CompileError {
+        lineNumberAttributeBuilder.put(bytecode.currentPc(), n);
+        super.atPair(n);
+    }
+
+    @Override
+    public void atSymbol(Symbol n) throws CompileError {
+        lineNumberAttributeBuilder.put(bytecode.currentPc(), n);
+        super.atSymbol(n);
+    }
+
+    @Override
+    public void atFieldDecl(FieldDecl field) throws CompileError {
+        lineNumberAttributeBuilder.put(bytecode.currentPc(), field);
+        super.atFieldDecl(field);
+    }
+
+    @Override
+    public void atMethodDecl(MethodDecl method) throws CompileError {
+        lineNumberAttributeBuilder.put(bytecode.currentPc(), method);
+        super.atMethodDecl(method);
+    }
+
+    @Override
+    public void atMethodBody(Stmnt s, boolean isCons, boolean isVoid) throws CompileError {
+        lineNumberAttributeBuilder.put(bytecode.currentPc(), s);
+        super.atMethodBody(s, isCons, isVoid);
+    }
+
+    @Override
+    public void atStmnt(Stmnt st) throws CompileError {
+        lineNumberAttributeBuilder.put(bytecode.currentPc(), st);
+        super.atStmnt(st);
+    }
+
+    @Override
+    public void atDeclarator(Declarator d) throws CompileError {
+        lineNumberAttributeBuilder.put(bytecode.currentPc(), d);
+        super.atDeclarator(d);
+    }
+
+    @Override
+    public void atAssignExpr(AssignExpr expr) throws CompileError {
+        lineNumberAttributeBuilder.put(bytecode.currentPc(), expr);
+        super.atAssignExpr(expr);
+    }
+
+    @Override
+    protected void atAssignExpr(AssignExpr expr, boolean doDup) throws CompileError {
+        lineNumberAttributeBuilder.put(bytecode.currentPc(), expr);
+        super.atAssignExpr(expr, doDup);
+    }
+
+    @Override
+    protected void atAssignCore(Expr expr, int op, ASTree right, int type, int dim, String cname) throws CompileError {
+        lineNumberAttributeBuilder.put(bytecode.currentPc(), expr);
+        super.atAssignCore(expr, op, right, type, dim, cname);
+    }
+
+    @Override
+    public void atCondExpr(CondExpr expr) throws CompileError {
+        lineNumberAttributeBuilder.put(bytecode.currentPc(), expr);
+        super.atCondExpr(expr);
+    }
+
+    @Override
+    public void atBinExpr(BinExpr expr) throws CompileError {
+        lineNumberAttributeBuilder.put(bytecode.currentPc(), expr);
+        super.atBinExpr(expr);
+    }
+
+    @Override
+    public void atInstanceOfExpr(InstanceOfExpr expr) throws CompileError {
+        lineNumberAttributeBuilder.put(bytecode.currentPc(), expr);
+        super.atInstanceOfExpr(expr);
+    }
+
+    @Override
+    public void atExpr(Expr expr) throws CompileError {
+        lineNumberAttributeBuilder.put(bytecode.currentPc(), expr);
+        super.atExpr(expr);
+    }
+
+    @Override
+    public void atClassObject(Expr expr) throws CompileError {
+        lineNumberAttributeBuilder.put(bytecode.currentPc(), expr);
+        super.atClassObject(expr);
+    }
+
+    @Override
+    public void atArrayRead(ASTree array, ASTree index) throws CompileError {
+        lineNumberAttributeBuilder.put(bytecode.currentPc(), array);
+        super.atArrayRead(array, index);
+    }
+
+    @Override
+    protected void arrayAccess(ASTree array, ASTree index) throws CompileError {
+        lineNumberAttributeBuilder.put(bytecode.currentPc(), array);
+        super.arrayAccess(array, index);
+    }
+
+    @Override
+    public void atArrayPlusPlus(int token, boolean isPost, Expr expr, boolean doDup) throws CompileError {
+        lineNumberAttributeBuilder.put(bytecode.currentPc(), expr);
+        super.atArrayPlusPlus(token, isPost, expr, doDup);
+    }
+
+    @Override
+    protected void atPlusPlusCore(int dup_code, boolean doDup, int token, boolean isPost, Expr expr) throws CompileError {
+        lineNumberAttributeBuilder.put(bytecode.currentPc(), expr);
+        super.atPlusPlusCore(dup_code, doDup, token, isPost, expr);
+    }
+
+    @Override
+    public void atVariable(Variable v) throws CompileError {
+        lineNumberAttributeBuilder.put(bytecode.currentPc(), v);
+        super.atVariable(v);
+    }
+
+    @Override
+    public void atKeyword(Keyword k) throws CompileError {
+        lineNumberAttributeBuilder.put(bytecode.currentPc(), k);
+        super.atKeyword(k);
+    }
+
+    @Override
+    public void atStringL(javassist.compiler.ast.StringL s) throws CompileError {
+        lineNumberAttributeBuilder.put(bytecode.currentPc(), s);
+        super.atStringL(s);
+    }
+
+    @Override
+    public void atIntConst(IntConst i) throws CompileError {
+        lineNumberAttributeBuilder.put(bytecode.currentPc(), i);
+        super.atIntConst(i);
+    }
+
+    @Override
+    public void atDoubleConst(DoubleConst d) throws CompileError {
+        lineNumberAttributeBuilder.put(bytecode.currentPc(), d);
+        super.atDoubleConst(d);
+    }
+
+    @Override
+    public void atMember(Member mem) throws CompileError {
+        lineNumberAttributeBuilder.put(bytecode.currentPc(), mem);
+        super.atMember(mem);
+    }
+
+    @Override
+    protected void atFieldAssign(Expr expr, int op, ASTree left, ASTree right, boolean doDup) throws CompileError {
+        lineNumberAttributeBuilder.put(bytecode.currentPc(), expr);
+        super.atFieldAssign(expr, op, left, right, doDup);
+    }
+
+    @Override
+    public void atCastExpr(CastExpr expr) throws CompileError {
+        lineNumberAttributeBuilder.put(bytecode.currentPc(), expr);
+        super.atCastExpr(expr);
+    }
+
+    @Override
+    protected void atCastToRtype(CastExpr expr) throws CompileError {
+        lineNumberAttributeBuilder.put(bytecode.currentPc(), expr);
+        super.atCastToRtype(expr);
+    }
+
+    @Override
+    protected void atCastToWrapper(CastExpr expr) throws CompileError {
+        lineNumberAttributeBuilder.put(bytecode.currentPc(), expr);
+        super.atCastToWrapper(expr);
+    }
+
+    @Override
+    public void atCallExpr(CallExpr expr) throws CompileError {
+        lineNumberAttributeBuilder.put(bytecode.currentPc(), expr);
+        super.atCallExpr(expr);
+    }
+
+    @Override
+    protected void atCflow(ASTList cname, int lineNumber) throws CompileError {
+        lineNumberAttributeBuilder.put(bytecode.currentPc(), cname);
+        super.atCflow(cname, lineNumber);
+    }
+
+    @Override
+    protected void atTryStmnt(Stmnt st) throws CompileError {
+        lineNumberAttributeBuilder.put(bytecode.currentPc(), st);
+        super.atTryStmnt(st);
+    }
+
+    @Override
+    public void atNewExpr(NewExpr expr) throws CompileError {
+        lineNumberAttributeBuilder.put(bytecode.currentPc(), expr);
+        super.atNewExpr(expr);
+    }
+
+    @Override
+    public void atNewArrayExpr(NewExpr expr) throws CompileError {
+        lineNumberAttributeBuilder.put(bytecode.currentPc(), expr);
+        super.atNewArrayExpr(expr);
+    }
+
+    @Override
+    protected void atArrayVariableAssign(ArrayInit init, int varType, int varArray, String varClass) throws CompileError {
+        lineNumberAttributeBuilder.put(bytecode.currentPc(), init);
+        super.atArrayVariableAssign(init, varType, varArray, varClass);
+    }
+
+    @Override
+    public void atArrayInit(ArrayInit init) throws CompileError {
+        lineNumberAttributeBuilder.put(bytecode.currentPc(), init);
+        super.atArrayInit(init);
+    }
+
+    @Override
+    protected void atMultiNewArray(int type, ASTList classname, ASTList size) throws CompileError {
+        lineNumberAttributeBuilder.put(bytecode.currentPc(), classname);
+        super.atMultiNewArray(type, classname, size);
+    }
+
+    @Override
+    public void atMethodCallCore(CtClass targetClass, String mname, ASTList args, boolean isStatic, boolean isSpecial, int aload0pos, MemberResolver.Method found) throws CompileError {
+        lineNumberAttributeBuilder.put(bytecode.currentPc(), args);
+        super.atMethodCallCore(targetClass, mname, args, isStatic, isSpecial, aload0pos, found);
+    }
+
+    @Override
+    protected void atFieldRead(ASTree expr) throws CompileError {
+        lineNumberAttributeBuilder.put(bytecode.currentPc(), expr);
+        super.atFieldRead(expr);
+    }
+
+    @Override
+    protected void atFieldPlusPlus(int token, boolean isPost, ASTree oprand, Expr expr, boolean doDup) throws CompileError {
+        lineNumberAttributeBuilder.put(bytecode.currentPc(), expr);
+        super.atFieldPlusPlus(token, isPost, oprand, expr, doDup);
+    }
+}