add reflection

src/YaoCompiler.jl

@@ -23,7 +23,10 @@ export @device,
     AnyReg,
     # reexport YaoLocations
     Locations,
-    CtrlLocations
+    CtrlLocations,
+    # reflection
+    @yao_code_lowered,
+    @yao_code_typed
 
 using MLStyle
 using YaoAPI
@@ -61,6 +64,7 @@ include("compiler/printing.jl")
 include("compiler/intrinsics.jl")
 include("compiler/syntax.jl")
 include("compiler/interp.jl")
+include("compiler/reflection.jl")
 
 include("codegen/llvmopt.jl")
 include("codegen/native.jl")

src/compiler/reflection.jl

@@ -0,0 +1,78 @@
+macro yao_code_lowered(ex)
+    esc(yao_code_lowered_m(ex))
+end
+
+macro yao_code_lowered(ctrl, ex)
+    esc(yao_code_lowered_m(ctrl, ex))
+end
+
+macro yao_code_typed(ex)
+    esc(yao_code_typed_m(ex))
+end
+
+macro yao_code_typed(ctrl, ex)
+    esc(yao_code_typed_m(ctrl, ex))
+end
+
+function yao_code_lowered_m(ex)
+    @match ex begin
+        # apply(reg, gate, locs)
+        :($locs => $gate) => quote
+            Base.code_lowered(
+                $Intrinsics.apply,
+                ($YaoCompiler.AnyReg, typeof($gate), typeof($Locations($(locs))))
+            )
+        end
+        # apply(reg, gate)
+        _ => quote
+            Base.code_lowered($Intrinsics.apply, ($YaoCompiler.AnyReg, typeof($ex)))
+        end
+    end
+end
+
+function yao_code_lowered_m(ctrl, ex)
+    @match ex begin
+        # apply(reg, gate, locs, ctrl)
+        :($locs => $gate) => quote
+            Base.code_lowered(
+                $Intrinsics.apply,
+                ($YaoCompiler.AnyReg, typeof($gate), typeof($Locations($(locs))), typeof($CtrlLocations($ctrl)))
+            )
+        end
+        _ => error("expect a locs => gate expression")
+    end
+end
+
+function yao_code_typed_m(ex)
+    @match ex begin
+        # apply(reg, gate, locs)
+        :($locs => $gate) => quote
+            $Base.code_typed(
+                $Intrinsics.apply,
+                ($YaoCompiler.AnyReg, typeof($gate), typeof($Locations($(locs))));
+                interp=$(YaoInterpreter())
+            )
+        end
+        # apply(reg, gate)
+        _ => quote
+            $Base.code_typed(
+                $Intrinsics.apply, ($YaoCompiler.AnyReg, typeof($ex));
+                interp=$(YaoInterpreter()),
+            )
+        end
+    end
+end
+
+function yao_code_typed_m(ctrl, ex)
+    @match ex begin
+        # apply(reg, gate, locs, ctrl)
+        :($locs => $gate) => quote
+            Base.code_typed(
+                $Intrinsics.apply,
+                ($YaoCompiler.AnyReg, typeof($gate), typeof($Locations($(locs))), typeof($CtrlLocations($ctrl)));
+                interp=$(YaoInterpreter()),
+            )
+        end
+        _ => error("expect a locs => gate expression")
+    end
+end

test/interp.jl

@@ -28,9 +28,11 @@ end
     return ret
 end
 
+ci, type = @yao_code_typed(test_pure_quantum())[1]
+
 op = test_pure_quantum()
 interp = YaoInterpreter()
-ci, type = code_typed(Intrinsics.apply, (AnyReg, typeof(op)); interp)[1]
+ci, type = @yao_code_typed(op)[1]
 
 @test_codeinfo ci begin
     Expr(