学步园

上一篇文章介绍了一些背景知识以及如何创建一个ASTParser，下面给出一些使用AST的例子。

创建ASTParser并将Java源文件解析成AST:

// Initialize ASTParser<br /> ASTParser parser = ASTParser.newParser(AST.JLS3); //initialize<br /> parser.setKind(ASTParser.K_COMPILATION_UNIT); //to parse compilation unit<br /> parser.setSource(content.toCharArray()); //content is a string which stores the java source<br /> parser.setResolveBindings(true);<br /> CompilationUnit result = (CompilationUnit) parser.createAST(null);

调用imports()方法得到源文件的一系列import声明：

//show import declarations in order<br /> List importList = result.imports();<br /> System.out.println("import:");<br /> for(Object obj : importList) {<br /> ImportDeclaration importDec = (ImportDeclaration)obj;<br /> System.out.println(importDec.getName());<br /> }

调用types()方法得到类名称：

//show class name<br /> List types = result.types();<br /> TypeDeclaration typeDec = (TypeDeclaration) types.get(0);<br /> System.out.println("className:"+typeDec.getName());

调用TypeDeclaration的getField方法得到类里定义的field：

//show fields<br /> FieldDeclaration fieldDec[]=typeDec.getFields();<br /> System.out.println("Fields:");<br /> for(FieldDeclaration field: fieldDec)<br /> {<br /> System.out.println("Field fragment:"+field.fragments());<br /> System.out.println("Field type:"+field.getType());<br /> }

调用TypeDeclaration的getMethods方法得到类里定义的list of methods；同理，可以调用MethodDeclaration里的各种方法得到method的方法名，参数，返回类型等等：

//show methods<br /> MethodDeclaration methodDec[] = typeDec.getMethods();<br /> System.out.println("Method:");<br /> for (MethodDeclaration method : methodDec)<br /> {<br /> //get method name<br /> SimpleName methodName=method.getName();<br /> System.out.println("method name:"+methodName);</p> <p> //get method parameters<br /> List param=method.parameters();<br /> System.out.println("method parameters:"+param);</p> <p> //get method return type<br /> Type returnType=method.getReturnType2();<br /> System.out.println("method return type:"+returnType);

一个方法的内容对应一个block，可以用getBody()得到；一个block又可以被分解成一系列statements，可以用statements()方法得到：

//get method body<br /> Block body=method.getBody();<br /> List statements=body.statements(); //get the statements of the method body<br /> Iterator iter=statements.iterator();<br /> while(iter.hasNext())<br /> {<br /> //get each statement<br /> Statement stmt=(Statement)iter.next();

接下来需要根据每个statement的类型来对源码进行相应的解析。Statement有很多子类，这里只给出其中的几类介绍（具体可以参考我上一篇文章：ASTParser介绍；最好直接去Eclipse documentation官网查看相关文档）。基本流程就是先判断statement是哪种类型的实例，做出相应的ClassCast，再调用相应的方法。

1. ExpressionStatement （又包括Assignement， MethodInvocation等子类）。

if(stmt instanceof ExpressionStatement)<br /> {<br /> ExpressionStatement expressStmt=(ExpressionStatement) stmt;<br /> Expression express=expressStmt.getExpression();</p> <p> if(express instanceof Assignment)<br /> {<br /> Assignment assign=(Assignment)express;<br /> System.out.println("LHS:"+assign.getLeftHandSide()+"; ");<br /> System.out.println("Op:"+assign.getOperator()+"; ");<br /> System.out.println("RHS:"+assign.getRightHandSide());</p> <p> }<br /> else if(express instanceof MethodInvocation)<br /> {<br /> MethodInvocation mi=(MethodInvocation) express;<br /> System.out.println("invocation name:"+mi.getName());<br /> System.out.println("invocation exp:"+mi.getExpression());<br /> System.out.println("invocation arg:"+mi.arguments());</p> <p> }<br /> System.out.println();</p> <p> }

2. IfStatement

else if(stmt instanceof IfStatement)<br /> {<br /> IfStatement ifstmt=(IfStatement) stmt;<br /> InfixExpression wex=(InfixExpression) ifstmt.getExpression();<br /> System.out.println("if-LHS:"+wex.getLeftOperand()+"; ");<br /> System.out.println("if-op:"+wex.getOperator()+"; ");<br /> System.out.println("if-RHS:"+wex.getRightOperand());<br /> System.out.println();<br /> }

3. VariableDeclarationStatement

else if(stmt instanceof VariableDeclarationStatement)<br /> {<br /> VariableDeclarationStatement var=(VariableDeclarationStatement) stmt;<br /> System.out.println("Type of variable:"+var.getType());<br /> System.out.println("Name of variable:"+var.fragments());<br /> System.out.println();</p> <p> }

4. ReturnStatement

else if(stmt instanceof ReturnStatement)<br /> {<br /> ReturnStatement rtstmt=(ReturnStatement) stmt;<br /> System.out.println("return:"+rtstmt.getExpression());<br /> System.out.println();<br /> }

ASTParser 作为programming language编译器的一部分，功能机制可以说是相当强大。以上的例子只是冰山一角。使用ASTParser可以对java的源文件进行各种解析，用户只需要指明源文件，调用一下ASTParser的createAST方法就能得到完整的AST了，剩下的就是用户根据实际需要对AST进行各种manipulate了。

Eclipse documentation详细记录了AST相关的各类描述，继承关系等等。另外，如果对AST机制或者在Eclipse里使用ASTParser有什么疑问的话，可以参考我的另一篇文章：Eclipse JDT--AST入门，里面整理了很多相关资料。

估计接下来一段时间就是继续研究AST了，还请各位高手多多指教~~~

最后一句碎碎念：尽请转载，不过请注明出处哦，必须的~~~:)