AST轉換對象如何工作？

Question

有人可以幫助我理解允許AST在scala中進行修改的轉換對象嗎？

網絡上有很多例子，但我很難理解遞歸是如何發生的。

E.g.如果我有這樣的AST：

def foo = { 
true 
} 

或

def foo = true 

並希望將其轉換爲

def foo = { 
println("foo") 
true 
} 

如何轉換功能應該像。在這一點上我有這樣的事情：

override def transform(tree: Tree) = tree match { 
case defdef @ DefDef(_,_,_,_,_,rhs) => rhs match{ 
    case b: Block => treeCopy.Block(b, createPrintln :: b.stats, b.expr)   
    case _ => //Manage functions without block 
} 
case t => super.transform(t) 
} 

private def createPrintln = Apply(Select(Ident("System.out"), newTermName("println")), List(Literal(Constant("foo")))) 

但不工作，誠實我只是將我在例子中看到，但對樹正在建設想不通。對「傻瓜」的解釋將不勝感激。

[編輯]

塞尼亞例子是好的，但有一個稍微複雜一點的樹一樣，當我仍然停留：

imports bla.bla 
class MyObject{ 

    val x = 0 
    def foo = true 
    def foo2 = { 1 } 
} 
object MyObject 

預期結果：

import bla.bla 
class MyObject{ 

    val x = 0 
    def foo = { println("foo"); true } 
    def foo2 = { println("foo"); 1 } 
} 
object MyObject 

Answer 1

你的原Tree是這樣的：

scala> import scala.reflect.runtime.universe._ 
import scala.reflect.runtime.universe._ 

scala> showRaw{ reify { def foo = { true } }.tree } 
res0: String = Block(List(DefDef(Modifiers(), newTermName("foo"), List(), List(), TypeTree(), Literal(Constant(true)))), Literal(Constant(()))) 

無外Block：

scala> showRaw{ reify { def foo = { true } }.tree match { case Block(List(defdef), _) => defdef } } 
res1: String = DefDef(Modifiers(), newTermName("foo"), List(), List(), TypeTree(), Literal(Constant(true))) 

所以你rhs變量不是Block。

所以，你應該用Block(createPrintln, t)

更換//Manage functions without block我猜你transform方法應該返回DefDef，不Block。

def addPrintln(t: Tree): Block = t match { 
    case b :Block => treeCopy.Block(b, createPrintln :: b.stats, b.expr) 
    case t => Block(createPrintln, t) 
} 

override def transform(tree: Tree) = tree match { 
    case defdef @ DefDef(mods, name, tparams, vparamss, tpt, rhs) => 
    treeCopy.DefDef(defdef, mods, name, tparams, vparamss, tpt, addPrintln(rhs)) 
    case t => super.transform(t) 
} 

測試：

scala> def createPrintln = Apply(Select(Ident("System.out"), newTermName("println")), List(Literal(Constant("foo")))) 
createPrintln: reflect.runtime.universe.Apply 

scala> def addPrintln(t: Tree): Block = t match { 
    | case b :Block => treeCopy.Block(b, createPrintln :: b.stats, b.expr) 
    | case t => Block(createPrintln, t) 
    | } 
addPrintln: (t: reflect.runtime.universe.Tree)reflect.runtime.universe.Block 

scala> def transform(tree: Tree) = tree match { 
    | case defdef @ DefDef(mods, name, tparams, vparamss, tpt, rhs) => 
    |  treeCopy.DefDef(defdef, mods, name, tparams, vparamss, tpt, addPrintln(rhs)) 
    | } 
transform: (tree: reflect.runtime.universe.Tree)reflect.runtime.universe.DefDef 

scala> val defdef = reify { def foo = { true } }.tree match { case Block(List(defdef), _) => defdef } 
defdef: reflect.runtime.universe.Tree = def foo = true 

scala> transform(defdef) 
res0: reflect.runtime.universe.DefDef = 
def foo = { 
    System.out.println("foo"); 
    true 
} 

UPD：

super.transform調用父類的實現。有關代碼，請參見internal implementation。