コードを計測するためにマクロ注釈を使用しています。いくつかの式の範囲位置を取得するにはどうすればよいですか?
@ScalaKata object SHA {
val foo = "foo"
val bar = "bar"
foo; bar
// ...
}
// result: Map((75, 78) -> "foo", (80, 83) -> "bar")
計測マクロ:
package com.scalakata.eval
import scala.reflect.macros.blackbox.Context
import scala.language.experimental.macros
import scala.annotation.StaticAnnotation
object ScalaKataMacro {
def impl(c: Context)(annottees: c.Expr[Any]*): c.Expr[Any] = {
import c.universe._
val result: Tree = {
val eval = newTermName("eval$")
annottees.map(_.tree).toList match {
case q"object $name { ..$body }" :: Nil => {
val instr = newTermName("instr$")
implicit def lift = Liftable[c.universe.Position] { p =>
q"(${p.start}, ${p.end})"
}
def instrument(rhs: Tree): Tree = {
q"""
{
val t = $rhs
${instr}(${rhs.pos}) = t
t
}
"""
}
val bodyI = body.map {
case ident: Ident => instrument(ident)
case otherwise => otherwise
}
q"""
object $name {
val $instr = scala.collection.mutable.Map.empty[(Int, Int), Any]
def $eval() = {
..$bodyI
$instr
}
}
"""
}
}
}
c.Expr[Any](result)
}
}
class ScalaKata extends StaticAnnotation {
def macroTransform(annottees: Any*) = macro ScalaKataMacro.impl
}
範囲オプションを有効にしていますscalacOptions += "-Yrangepos"
私は現在、開始位置のみを取得しています:result: Map((75, 75) -> "foo", (80, 80) -> "bar")