From 60d60e21590c58e4cc20f053eb04f61539ec5b72 Mon Sep 17 00:00:00 2001
From: Abhishek Bhatt <46929125+Abhishek-1Bhatt@users.noreply.github.com>
Date: Thu, 2 Nov 2023 08:29:21 -0400
Subject: [PATCH] update the superdense coding to work with RegRefs (#55)

---
 src/CircuitZoo/CircuitZoo.jl       | 22 +++++++++++-----------
 test/test_circuitzoo_superdense.jl |  4 ++--
 2 files changed, 13 insertions(+), 13 deletions(-)

diff --git a/src/CircuitZoo/CircuitZoo.jl b/src/CircuitZoo/CircuitZoo.jl
index 114fb96f..e470e663 100644
--- a/src/CircuitZoo/CircuitZoo.jl
+++ b/src/CircuitZoo/CircuitZoo.jl
@@ -870,7 +870,7 @@ julia> initialize!((regA[1], regB[1]), (L0⊗L0+L1⊗L1)/√2);
 
 julia> message = [1, 1];
 
-julia> SDEncode()(regA, message);
+julia> SDEncode()(regA[1], message);
 ```
 
 See also [`SDDecode`](@ref)
@@ -878,12 +878,12 @@ See also [`SDDecode`](@ref)
 struct SDEncode <: AbstractCircuit
 end
 
-function (circuit::SDEncode)(reg, message)
+function (circuit::SDEncode)(rref, message)
     if message[2] == 1
-        apply!(reg[1], X)
+        apply!(rref, X)
     end
     if message[1] == 1
-        apply!(reg[1], Z)
+        apply!(rref, Z)
     end
 end
 
@@ -900,7 +900,7 @@ using the entangled bell pair stored in the registers regA and regB after Alice'
 Returns a Tuple of the decoded message.
 
 ```jldoctest
-julia> regA = Register(1); regB = Register(2);
+julia> regA = Register(1); regB = Register(1);
 
 julia> initialize!((regA[1], regB[1]), (L0⊗L0+L1⊗L1)/√2);
 
@@ -908,7 +908,7 @@ julia> message = [1, 1];
 
 julia> SDEncode()(regA, message);
 
-julia> SDDecode()(regA, regB)
+julia> SDDecode()(regA[1], regB[1])
 (1, 1)
 ```
 
@@ -917,11 +917,11 @@ See also [`SDEncode`](@ref)
 struct SDDecode <: AbstractCircuit
 end
 
-function (circuit::SDDecode)(regA, regB)
-    apply!((regA[1], regB[1]), CNOT)
-    apply!(regA[1], H)
-    b1 = project_traceout!(regA, 1, Z)
-    b2 = project_traceout!(regB, 1, Z)
+function (circuit::SDDecode)(rrefA, rrefB)
+    apply!((rrefA, rrefB), CNOT)
+    apply!(rrefA, H)
+    b1 = project_traceout!(rrefA, Z)
+    b2 = project_traceout!(rrefB, Z)
     return b1-1, b2-1
 end
 
diff --git a/test/test_circuitzoo_superdense.jl b/test/test_circuitzoo_superdense.jl
index db3bb35a..7f79bcab 100644
--- a/test/test_circuitzoo_superdense.jl
+++ b/test/test_circuitzoo_superdense.jl
@@ -17,8 +17,8 @@ for i in 1:8
     message = Tuple(rand(0:1, 2))
 
     # Use the circuits to encode and decode the message
-    SDEncode()(ra, message)
-    rec = SDDecode()(ra, rb)
+    SDEncode()(ra[1], message)
+    rec = SDDecode()(ra[1], rb[1])
 
     @test message == rec
 end