如何獲得一個scala宏以替換方法調用

Question

如何獲得一個scala宏以替換方法調用？

我的目標是創建一個名爲ToStringAdder的特徵。假設我有一個具有此特徵的對象x，那麼當我呼叫x.add(any)時，我希望宏實際調用x.add(any, string)，其中字符串是AST的字符串表示形式。 （當'any'是一個函數時，這樣我可以有很好的tostring）。

• 我寫的文檔 http://docs.scala-lang.org/overviews/macros/overview.html中給出的宏，
• 閱讀http://docs.scala-lang.org/sips/pending/self-cleaning-macros.html
• 閱讀PDF文件http://scalamacros.org/paperstalks/2013-04-22-LetOurPowersCombine.pdf。
• 我已經看過了代碼https://github.com/retronym/macrocosm/blob/master/src/main/scala/com/github/retronym/macrocosm/Macrocosm.scala
• 而在https://github.com/pniederw/expecty/tree/master/src/main/scala/org/expecty， 的代碼，但我很傷心地說，我不神交這一切
• 我有看scaladoc，但。 ..woo ..這很複雜...我懷疑我需要一些重要的時間來加快編譯器的工作速度。

除了Expecty，我所看到的所有這些例子都有效地使用了靜態方法調用：不使用調用宏的對象。 Expecty有以下方法，它給了我一個關於如何檢測'隱式this'的線索，但是我找不到在reify調用中引用它的方法。

private[this] def recordAllValues(expr: Tree): Tree = expr match { 
    case New(_) => expr // only record after ctor call 
    case Literal(_) => expr // don't record 
    // don't record value of implicit "this" added by compiler; couldn't find a better way to detect implicit "this" than via point 
    case Select(x@This(_), y) if getPosition(expr).point == getPosition(x).point => expr 
    case _ => recordValue(recordSubValues(expr), expr) 
    } 

那麼我該如何去取代調用宏的對象的調用。我此刻的代碼如下，它是需要進行排序

trait ToStringAdder { 
    def add(param: Any): Any = macro ToStringAdder.toStringAndValueImpl 
    def add(param: Any, toStringBasedOnAST: String): Any ; //This is the actual method I want the above method call to be replaced by 
} 

object ToStringAdder { 
    def toStringAndValueImpl(c: Context)(param: c.Expr[Any]): c.Expr[Unit] = { 
    import c.universe._ 
    val paramRep = show(param.tree) 
    val paramRepTree = Literal(Constant(paramRep)) 
    val paramRepExpr = c.Expr[String](paramRepTree) 
    //need to put something here 
    reify { c.someMethodCall("something to represent the method any", param.splice, paramRepExpr.splice) } 
    } 
} 

Answer 1

你可以得到Tree您ToStringAdder實例作爲c.prefix在物化調用的代碼。

試試這個：

reify { c.Expr[ToStringAdder](c.prefix.tree).splice.add(param.splice, c.literal(paramRep).splice) } 

證明它的工作原理：

scala> :paste 
// Entering paste mode (ctrl-D to finish) 

import scala.language.experimental.macros 
import reflect.macros.Context 

trait ToStringAdder { 
    def add(param: Any): Any = macro ToStringAdder.toStringAndValueImpl 
    def add(param: Any, toStringBasedOnAST: String): Any ; //This is the actual method I want the above method call to be replaced by 
} 

object ToStringAdder { 
    def toStringAndValueImpl(c: Context)(param: c.Expr[Any]): c.Expr[Any] = { 
    import c.universe._ 
    val paramRep = show(param.tree) 
    reify { (c.Expr[ToStringAdder](c.prefix.tree)).splice.add(param.splice, c.literal(paramRep).splice) } 
    } 
} 


// Exiting paste mode, now interpreting. 

import scala.language.experimental.macros 
import reflect.macros.Context 
defined trait ToStringAdder 
defined module ToStringAdder 

scala> class ToStringAdder1 extends ToStringAdder { 
    | def add(param: Any, toStringBasedOnAST: String): Any = s"param: $param \ntoStringBasedOnAST: $toStringBasedOnAST" 
    | } 
defined class ToStringAdder1 

scala> new ToStringAdder1().add((i: Int) => i*2) 
res0: Any = 
param: <function1> 
toStringBasedOnAST: ((i: Int) => i.*(2))