Question

我需要一些指导，试图解决我使用树语法遇到的问题。基本上，我希望能够做到的是可以在树中找到的替换/复制语句。最好通过一个例子来解释。

以下是输入示例：

int a = 10;
new function A;

function A {
    int x;
    int y;
    new function B;
}

function B {
    float b = 20;
}

通缉输出（稍后）：

int a = 10;
int x;
int y;
float b = 20;

这是一个简单的搜索和替换功能块内的语句。我的问题是ANTLR是否通过树语法提供了一种方法？

这是一个应解析上述输入的语法：

Test.g

grammar Test;
options {
  language = Java;
  output = AST;
}
tokens {
    VARDECL;
    FUNDEF;
    FUNCALL;
    BLOCK;
    ASSIGN;

    Assign = '=';
    EqT    = '==';  
    NEq    = '!=';
    LT     = '<';
    LTEq   = '<=';
    GT     = '>';
    GTEq   = '>=';
    NOT    = '!';
    PLUS   = '+';
    MINUS  = '-';
    MULT   = '*';
    DIV    = '/';
}

parse: statements+
     ;

statements : varDeclare
           | funcDefinition
           | funcCall
           ;

funcDefinition : 'function' id '{' funcBlock* '}' -> ^(FUNDEF id ^(BLOCK funcBlock*))
               ;

funcBlock : varDeclare
          | funcCall
          ;

funcCall : 'new' 'function' id ';' -> ^(FUNCALL id)
         ;

varDeclare : type id equalExp? ';' -> ^(VARDECL type id equalExp?)
           ;

equalExp : (Assign^ (expression | '...' )) 
         ;

expression : binaryExpression
           ;

binaryExpression : addingExpression ((EqT|NEq|LTEq|GTEq|LT|GT)^ addingExpression)*
                 ;

addingExpression : multiplyingExpression ((PLUS|MINUS)^ multiplyingExpression)*
                 ;

multiplyingExpression : unaryExpression 
                        ((MULT|DIV)^ unaryExpression)*
                      ;

unaryExpression: ((NOT|MINUS))^ primitiveElement
               | primitiveElement
               ;   

primitiveElement : literalExpression
                 | id
                 | '(' expression ')' -> expression
                 ;  

literalExpression : INT
                  ;              

id : IDENTIFIER
   ;

type : 'int'    
     | 'float'
     ; 

// L E X I C A L   R U L E S      

INT : DIGITS ;   

IDENTIFIER : LETTER (LETTER | DIGIT)*;

WS  :   ( ' '
        | '\t'
        | '\r'
        | '\n'
        ) {$channel=HIDDEN;}
    ;

fragment LETTER : ('a'..'z' | 'A'..'Z' | '_') ;
fragment DIGITS: DIGIT+;
fragment DIGIT : '0'..'9';

Test.java

import java.io.FileWriter;
import java.io.IOException;
import java.io.PrintWriter;

import org.antlr.runtime.ANTLRStringStream;
import org.antlr.runtime.CommonTokenStream;
import org.antlr.runtime.RuleReturnScope;
import org.antlr.runtime.tree.CommonTree;
import org.antlr.runtime.tree.DOTTreeGenerator;
import org.antlr.stringtemplate.StringTemplate;

public class Test {
    public static void main(String[] args) throws Exception {
        String src = "int a = 10;\r\n" + 
                "new function A;\r\n" + 
                "\r\n" + 
                "function A {\r\n" + 
                "   int x;\r\n" + 
                "   int y;\r\n" + 
                "   new function B;\r\n" + 
                "}\r\n" + 
                "\r\n" + 
                "function B{\r\n" + 
                "   float b = 20;\r\n" + 
                "}";
        TestLexer lexer = new TestLexer(new ANTLRStringStream(src));
        CommonTokenStream tokenStream = new CommonTokenStream(lexer);
        TestParser parser = new TestParser(tokenStream);
        RuleReturnScope r = parser.parse();
        System.out.println("Tree:" + ((CommonTree) r.getTree()).toStringTree() + "\n");


        CommonTree t = (CommonTree)r.getTree();
        generateGraph(t, "Tree.dot");   
    }

    private static void generateGraph(CommonTree t, String file) throws IOException {
        DOTTreeGenerator gen = new DOTTreeGenerator();
        StringTemplate st = gen.toDOT(t);
        String output = file;
        PrintWriter out = new PrintWriter(new FileWriter(output));
        out.println(st);
        out.close();
    }
}

Tree.dot AST graph

如何搜索每个FUNCALL并使用Tree Grammar替换为BLOCK的内容？

提前致谢！

Answer 1

在你的语法中，你将以ID作为关键字制作你的FUNDEF BLOCK的表blockMap。

然后在你的树语法中这样的东西，虽然它可能需要一些调整。你将为funcCall制定规则：

funcCall : ^(FUNCALL id) -> {input.getTreeAdaptor().dupTree(blockMap.get(id)}
     ;

这是从地图中复制BLOCK的原因： input.getTreeAdaptor（）。dupTree（...）

您需要制作副本，因为节点会跟踪其父级，因此您只能在树中的一个位置使用它们。

使用ANTLR树语法复制树的节点

1 个答案: