Scala宏:在类定义

时间:2016-12-04 23:55:33

标签: scala scala-macros

有点时髦的场景,最好概述如下。

假设我有sealed trait Painting[T],然后是一些更专业的变体,例如trait ImpressionistPainting[T] extends Painting[T]等,构建一个简单的密封类型系列。

然后我有一个带有f bound的多态类型的类,如下所示:

class Gallery[T <: Gallery[T]]()(implicit helper: GalleryHelper[T])

并且a:

trait GalleryHelper[T <: Gallery[T]] {
  def paintings: Set[Painting[_]]
}
object GalleryHelper {
  implicit def materialize[T <: Gallery[T]]: GalleyHelper[T] = {
    macro MyMacro.materialize[T]
  }

  def apply[T <: Gallery[T]]()(
   implicit ev: GalleryHelper[T]
  ): GalleryHelper[T] = ev 
}

到目前为止,这是非常基本且简单的foo条形码。

此设置的整个目标是宏实现一个帮助器列表,我需要用它来描述库,或者在这种情况下任意键入&#34;用户内容&#34;,如下:

class MyGallery extends Gallery[MyGallery] {
  object `le-reve` extends CubistPainting[Picasso]
  object starry_night extends PostImpressionistPainting[VanGogh]
  // ..
}

现在有了一些宏的compat乐趣,我想宏实现有问题的助手,并过滤T <: Gallery[T]的成员来提取那些MemberType <:< Painting[_]

所有非常简单,下面比简单type.decls.filter(_.typeSignature <:< typeOf[Filter]更复杂,因为它需要按照它们编写的顺序返回所有继承成员的列表,例如,给定画廊扩展其他库。

  def findMembersWithBound[T : WeakTypeTag, Filter : TypeTag](
    exclusions: Symbol => Option[Symbol] = { s: Symbol => Some(s) }
  ): Set[Symbol] = {
    val tpe = weakTypeOf[T].typeSymbol.typeSignature

    (
      for {
        baseClass <- tpe.baseClasses.reverse.flatMap(exclusions(_))
        symbol <- baseClass.typeSignature.members.sorted
        if symbol.typeSignature <:< typeOf[Filter]
      } yield symbol
    )(collection.breakOut)
  }

因此,在隐式宏中,类型T的模块成员的基本遍历需要按子类型进行过滤,在本例中为Painting[_],然后查看用户在何时提供的特定类型参数扩展Painting的变体。类型系列是密封的,因此用户可以使用Painting扩展object子类,而不是直接Painting[_],如果这是相关的话。

@macrocompat.bundle
class MyMacro(val c: blackbox.Context) {
  import c.universe._
  def materialize[T <: Gallery[T]]: Tree = {
    val galleryTpe = weakTypeOf[T]
    val fields = findMembersWithBound[T, Painting[_]](exclusions)

    val colMembers = sourceMembers.map { member =>
      val memberType = member.typeSignatureIn(galleryTpe)

      memberType.baseClasses.find(colSymbol ==) match {
        case Some(root) => {
          // Herein lies the problem, typeArgs is Nil.
          root.typeSignatureIn(memberType).typeArgs.headOption match {
            case Some(colSignature) => colSignature
            case None => c.abort(
              c.enclosingPosition,
              s"Could not find the artist for ${member.asModule.name}"
          )
        }
      }
      case None => c.abort(c.enclosingPosition, s"Could not find root painting type for ${member.asModule.name}")
    }
  }
}

问题是,传递给Painting的所有原始类型参数都不再可见,即使typeSignature要在范围内进行评估等等,我也是如此只是想确保梵高不会成为众所周知的沃尔多。

dealias的正确API是什么,或者是否会再次显示这些typeArgs?目前是一个空列表。

1 个答案:

答案 0 :(得分:2)

好的,那就是&#34;隐藏得很好&#34;使用asSeenFrom执行此操作的方法。相关部分是:

root.typeSignature.typeParams match {
  case head :: Nil => head.asType.toType.asSeenFrom(memberType, colSymbol)
  case _ => c.abort(
    c.enclosingPosition,
   "Expected exactly one type parameter provided for painting type"
  )
}