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

Question

如何访问Scala宏中的代码文件名和行号？我看了SIP-19，它说使用宏可以很容易地实现。。。

编辑：为了澄清，我想要调用方的代码文件和行号。我已经有了一个调试宏，我想修改它以打印调用debug 的人的行号和文件名

Answer 1

您需要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
  }

Answer 2

如果你的意思是源代码中当前位置的文件名和行号，对于2.10，我对SO问题的回答是你想要的：

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的方式似乎不受欢迎。

你也可以看看我的项目Scart的摘录；这就是我如何使用这种技术来发出用于调试目的的跟踪。

Answer 3

"编写Scala编译器插件"中的例子展示了如何访问当前位置的行号和当前编号，正如其他答案所提到的那样。

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")
}