squito · January 28, 2016 07:41 · Oct 25, 2013 · Oct 22, 2013 · Oct 22, 2013 · Oct 22, 2013
diff --git a/dynamicTraitGoal.scala b/dynamicTraitGoal.scala
@@ -0,0 +1,13 @@
+trait SomeData {
+  def x: Int
+  def y: Float
+  def z: Float
+}
+
+trait OtherData {
+  def a: Float
+  def b: Int
+}
+
+@ByteBufferBacked(classOf[SomeData]) class SomeDataImpl
+@ByteBufferBacked(classOf[OtherData]) class OtherDataImpl
diff --git a/quasiParentClass.scala b/quasiParentClass.scala
@@ -1 +1,2 @@
- val q"class $ignore extends $addedType" = q"class Foo extends com.imranrashid.oleander.macros.SimpleTrait"
+val q"class $ignore extends $addedType" = 
+  q"class Foo extends com.imranrashid.oleander.macros.SimpleTrait"
diff --git a/addedTrait → addedTrait.scala b/addedTrait → addedTrait.scala
diff --git a/annotations_goal → annotationsGoal.scala b/annotations_goal → annotationsGoal.scala
diff --git a/full_macro → fullMacro.scala b/full_macro → fullMacro.scala
diff --git a/goal_class → goalClass.scala b/goal_class → goalClass.scala
diff --git a/quasiClassDef → quasiClassDef.scala b/quasiClassDef → quasiClassDef.scala
diff --git a/quasiClassDefUnapply → quasiClassDefUnapply.scala b/quasiClassDefUnapply → quasiClassDefUnapply.scala
diff --git a/quasiDefDef → quasiDefDef.scala b/quasiDefDef → quasiDefDef.scala
diff --git a/quasiParentClass → quasiParentClass.scala b/quasiParentClass → quasiParentClass.scala
diff --git a/quasiReplImports → quasiReplImports.scala b/quasiReplImports → quasiReplImports.scala
diff --git a/reify → reify.scala b/reify → reify.scala
diff --git a/reifyDefDefs → reifyDefDefs.scala b/reifyDefDefs → reifyDefDefs.scala
diff --git a/repl_imports → replImports.scala b/repl_imports → replImports.scala
diff --git a/showRaw → showRaw.scala b/showRaw → showRaw.scala
diff --git a/trait → trait.scala b/trait → trait.scala
diff --git a/quasiClassDef b/quasiClassDef
@@ -0,0 +1 @@
+val classdef = q"""class Foo extends A with B with C {}"""
diff --git a/quasiClassDefUnapply b/quasiClassDefUnapply
@@ -0,0 +1 @@
+val q"class $cname extends $parent with ..$traits { ..$body }" = classdef
diff --git a/quasiParentClass b/quasiParentClass
@@ -0,0 +1 @@
+ val q"class $ignore extends $addedType" = q"class Foo extends com.imranrashid.oleander.macros.SimpleTrait"
diff --git a/reifyDefDefs b/reifyDefDefs
@@ -0,0 +1,4 @@
+val newDefDefs = reify {
+  def x = 5
+  def y = 7.0f
+}.tree match { case Block(defs, _) => defs}
diff --git a/addedTrait b/addedTrait
@@ -0,0 +1,4 @@
+ val addedTrait = Select(Select(Select(
+   Select(Ident(newTermName("com")), newTermName("imranrashid")),
+   newTermName("oleander")),newTermName("macros")),
+   newTypeName("SimpleTrait"))
diff --git a/annotations_goal b/annotations_goal
@@ -0,0 +1,6 @@
+trait SimpleTrait {
+  def x: Int
+  def y: Float
+}
+
+@FillTraitDefs class Foo extends SimpleTrait {}
diff --git a/full_macro b/full_macro
@@ -0,0 +1,29 @@
+class FillTraitDefs extends StaticAnnotation {
+  def macroTransform(annottees: Any*) = macro SimpleTraitImpl.addDefs
+}
+
+object SimpleTraitImpl {
+
+  def addDefs(c: Context)(annottees: c.Expr[Any]*): c.Expr[Any] = {
+    import c.universe._
+    val inputs = annottees.map(_.tree).toList
+    val newDefDefs = List(
+      DefDef(Modifiers(), newTermName("x"), List(), List(), TypeTree(), Literal(Constant(5))),
+      DefDef(Modifiers(), newTermName("y"), List(), List(), TypeTree(), Literal(Constant(7.0f)))
+    )
+    val modDefs = inputs map {tree => tree match {
+      case ClassDef(mods, name, something, template) =>
+        val q = template match {
+          case Template(superMaybe, emptyValDef, defs) =>
+            Template(superMaybe, emptyValDef, defs ++ newDefDefs)
+          case y =>
+            y
+        }
+        ClassDef(mods, name, something, q)
+      case x =>
+        x
+    }}
+    val result = c.Expr(Block(modDefs, Literal(Constant())))
+    result
+  }
+}
diff --git a/quasiDefDef b/quasiDefDef
@@ -0,0 +1,2 @@
+scala> q"def x = 5"
+res3: reflect.runtime.universe.DefDef = def x = 5
diff --git a/quasiReplImports b/quasiReplImports
@@ -0,0 +1,5 @@
+import language.experimental.macros
+import reflect.macros.Context
+import scala.annotation.StaticAnnotation
+import scala.reflect.runtime.{universe => ru}
+import ru._
diff --git a/reify b/reify
@@ -0,0 +1,2 @@
+scala> ru.showRaw{ru.reify{def x = 5}}
+res11: String = Expr(Block(List(DefDef(Modifiers(), newTermName("x"), List(), List(), TypeTree(), Literal(Constant(5)))), Literal(Constant(()))))
diff --git a/repl_imports b/repl_imports
@@ -0,0 +1,4 @@
+import language.experimental.macros
+import reflect.macros.Context
+import scala.annotation.StaticAnnotation
+import scala.reflect.runtime.{universe => ru}
diff --git a/showRaw b/showRaw
@@ -0,0 +1,12 @@
+// AST of defining a val
+scala> ru.showRaw{ru.reify{val x = 5}}
+res1: String = Expr(Block(List(ValDef(Modifiers(), newTermName("x"), TypeTree(), Literal(Constant(5)))), Literal(Constant(()))))
+
+// AST for defining a class
+scala> ru.showRaw{ru.reify{class B}}
+res2: String = Expr(Block(List(ClassDef(Modifiers(), newTypeName("B"), List(), Template(List(Ident(newTypeName("AnyRef"))), emptyValDef, List(DefDef(Modifiers(), nme.CONSTRUCTOR, List(), List(List()), TypeTree(), Block(List(Apply(Select(Super(This(tpnme.EMPTY), tpnme.EMPTY), nme.CONSTRUCTOR), List())), Literal(Constant(())))))))), Literal(Constant(()))))
+
+// AST for defining a class with a templated parent -- but showRaw is wrong here!
+scala> ru.showRaw{ru.reify{class B extends collection.mutable.Seq[String]}}
+res7: String = Expr(Block(List(ClassDef(Modifiers(), newTypeName("B"), List(), Template(List(AppliedTypeTree(Ident(scala.collection.mutable.Seq), List(Select(Ident(scala.Predef), newTypeName("String"))))), emptyValDef, List(DefDef(Modifiers(), nme.CONSTRUCTOR, List(), List(List()), TypeTree(), Block(List(Apply(Select(Super(This(tpnme.EMPTY), tpnme.EMPTY), nme.CONSTRUCTOR), List())), Literal(Constant(())))))))), Literal(Constant(()))))
+
diff --git a/goal_class b/goal_class
@@ -0,0 +1,4 @@
+class B(bb: ByteBuffer) extends A {
+  def x = bb.getInt(0)
+  def y = bb.getFloat(4)
+}
diff --git a/trait b/trait
@@ -0,0 +1,4 @@
+trait A {
+  def x: Int
+  def y: Float
+}
Original file line number	Diff line number	Diff line change
		@@ -0,0 +1 @@
		val classdef = q"""class Foo extends A with B with C {}"""
Original file line number	Diff line number	Diff line change
		@@ -0,0 +1 @@
		val q"class $cname extends $parent with ..$traits { ..$body }" = classdef
Original file line number	Diff line number	Diff line change
		@@ -0,0 +1 @@
		val q"class $ignore extends $addedType" = q"class Foo extends com.imranrashid.oleander.macros.SimpleTrait"
Original file line number	Diff line number	Diff line change
		@@ -0,0 +1,2 @@
		scala> q"def x = 5"
		res3: reflect.runtime.universe.DefDef = def x = 5
Original file line number	Diff line number	Diff line change
		@@ -0,0 +1,2 @@
		scala> ru.showRaw{ru.reify{def x = 5}}
		res11: String = Expr(Block(List(DefDef(Modifiers(), newTermName("x"), List(), List(), TypeTree(), Literal(Constant(5)))), Literal(Constant(()))))