从Scala宏访问代码文件和行号?

时间:2014-04-24 06:03:35

标签: scala reflection scala-2.10 scala-macros scala-2.11

如何在Scala宏中访问代码文件的名称和行号?我查看了SIP-19,它说可以使用宏轻松实现...

编辑: 为了澄清,我想要调用者的代码文件和行号。我已经有debug macro,我想修改它以打印调用debug的人的行号和文件名

3 个答案:

答案 0 :(得分:4)

您需要c.macroApplication.pos,其中c适用于Context

c.enclosingPosition找到堆栈中具有位置的最近的宏。 (参见其他答案。)例如,如果您的断言宏为F"%p: $msg"生成了一个树但未指定位置,则F宏将无位置。

来自字符串插值器宏的示例F"%p"

  /* Convert enhanced conversions to something format likes.
   * %Q for quotes, %p for position, %Pf for file, %Pn line number,
   * %Pc column %Po offset.
   */
  private def downConvert(parts: List[Tree]): List[Tree] = {
    def fixup(t: Tree): Tree = {
      val Literal(Constant(s: String)) = t
      val r = "(?<!%)%(p|Q|Pf|Po|Pn|Pc)".r
      def p = c.macroApplication.pos
      def f(m: Match): String = m group 1 match {
        case "p"  => p.toString
        case "Pf" => p.source.file.name
        case "Po" => p.point.toString
        case "Pn" => p.line.toString
        case "Pc" => p.column.toString
        case "Q"  => "\""
      }
      val z = r.replaceAllIn(s, f _)
      Literal(Constant(z)) //setPos t.pos
    }
    parts map fixup
  }

答案 1 :(得分:1)

如果你的意思是源代码中当前位置的文件名和行号,对于2.10,我对你的回答是that SO question

def $currentPosition:String = macro _currentPosition
def _currentPosition(c:Context):c.Expr[String]={ import c.universe._
  val pos = c.enclosingPosition
  c.Expr(Literal(Constant(s"${pos.source.path}: line ${pos.line}, column ${pos.column}")))
}

这也适用于2.11,虽然这种创建AST的方式似乎已被弃用。

您还可以查看at that excerpt of my project Scart;这就是我使用这种技术发出跟踪以进行调试的方法。

答案 2 :(得分:0)

编写Scala编译器插件的示例&#39;显示了如何访问当前位置的行名和当前号,正如其他答案所提到的那样。

http://www.scala-lang.org/old/node/140

除了上面的答案,您还可以从CompilationUnit返回的AST中获取位置。

例如:

def apply(unit: CompilationUnit) {
    // Get the AST
    val tree = unit.body  

    // Get the Position
    // Scala.util.parsing.input.Position
    val myPos = tree.pos 

    // Do something with the pos
    unit.warning(pos, "Hello world")
}