scala AST Select 节点找不到从父节点继承的成员

Question

我正在编写一个名为assign的宏，其任务是将一个实例的成员的值分配给另一个实例，并在成员的名称中添加特定的前缀。例如，我有一个具有名为my_prefix_data, my_prefix_rden, ...的成员的实例，并且我希望从另一个具有相应成员data, rden, ...的实例中为这些成员赋值。我已经制作了这个宏的原型版本，它只处理my_prefix_data <- data赋值。赋值将使用特殊的方法:=进行，因为该宏最终将应用于Chisel代码。

不幸的是，它不起作用。宏实现看起来像这个

package my_macros
import scala.language.experimental.macros
import scala.reflect.macros.blackbox

object Macro {
def assign(foo: Any, prefix: String, bundle: Any): Unit = macro Impl.assign
}
class Impl(val c: blackbox.Context) {
def assign(
foo: c.Tree, // Foo
prefix: c.Tree, // string
bundle: c.Tree // bundle
): c.Tree = {
import c.universe._
val Literal(Constant(p: String)) = prefix
Apply(
Select(
Select(
foo,
TermName(s"${p}_data")
),
TermName(":=")
),
List(
Select(
bundle,
TermName("data")
)
)
)
}
}

我设置了一个失败的测试用例，如

package my_macros
import org.scalatest._
class MacroSpec extends FlatSpec {
behavior of "Macro"
it should "access parent methods" in {
class Data(initial: Int) {
var value: Int = initial
def :=(other: Data): Unit = {
value += other.value
}
}
class UInt(initial: Int) extends Data(initial) {}
class FooBundle {
val my_prefix_data = new UInt(1)
}
class BarBundle {
val data = new UInt(2)
}
val foo = new FooBundle
val bar = new BarBundle
Macro.assign(foo, "my_prefix", bar)
assert(foo.my_prefix_data.value == bar.data.value)
}
}

它失败了，出现错误

[error] value := is not a member of UInt
[error]   Expression does not convert to assignment because receiver is not assignable.
[error]     Macro.assign(foo, "my_prefix", bar)
[error]                 ^

为什么SelectAST节点无法从UInt的父类Data中找到继承的:=？

Answer 1

继承是不相关的。对于

class UInt(initial: Int) {
var value: Int = initial
def :=(other: UInt): Unit = {
value += other.value
}
}

行为是一样的。

用TermName("$colon$eq")或TermName(":=").encodedName替换TermName(":=")或替换手动构建的树

val Literal(Constant(p: String)) = prefix
Apply(
Select(
Select(
foo,
TermName(s"${p}_data")
),
TermName(":=")
),
List(
Select(
bundle,
TermName("data")
)
)
)

带有准报价

val q"${p: String}" = prefix
val pdata = TermName(s"${p}_data")
q"$foo.$pdata.:=($bundle.data)"

然后代码编译。