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