Structure of example directory; sender and receiver applications for NetQASM qubit teleportation example

Author: CardamaS99

examples/netqasm/teleport/app_receiver.py

@@ -0,0 +1,68 @@
+from netqasm.runtime.settings import Simulator, get_simulator
+from netqasm.sdk import EPRSocket
+from netqasm.sdk.external import NetQASMConnection, Socket, get_qubit_state
+from netqasm.sdk.toolbox.sim_states import get_fidelity, qubit_from, to_dm
+
+
+def main(app_config=None):
+    log_config = app_config.log_config
+
+    # Create a socket to recv classical information
+    socket = Socket("receiver", "sender", log_config=log_config)
+
+    # Create a EPR socket for entanglement generation
+    epr_socket = EPRSocket("sender")
+
+    # Initialize the connection
+    receiver = NetQASMConnection(
+        app_name=app_config.app_name, log_config=log_config, epr_sockets=[epr_socket]
+    )
+    with receiver:
+        epr = epr_socket.recv_keep()[0]
+        receiver.flush()
+
+        # Get the corrections
+        m1, m2 = socket.recv_structured().payload
+        print(f"`receiver` got corrections: {m1}, {m2}")
+        if m2 == 1:
+            print("`receiver` will perform X correction")
+            epr.X()
+        if m1 == 1:
+            print("`receiver` will perform Z correction")
+            epr.Z()
+
+        receiver.flush()
+
+        if get_simulator() == Simulator.NETSQUID:
+            # Get the qubit state
+            # NOTE only possible in simulation, not part of actual application
+            dm = get_qubit_state(epr)
+            print(f"`receiver` recieved the teleported state {dm}")
+
+            # Reconstruct the original qubit to compare with the received one.
+            # NOTE only to check simulation results, normally the Sender does not
+            # need to send the phi and theta values!
+            msg = socket.recv_silent()  # don't log this
+            print(f"received silent message: {msg}")
+            phi, theta = eval(msg)
+
+            original = qubit_from(phi, theta)
+            original_dm = to_dm(original)
+            fidelity = get_fidelity(original, dm)
+
+            return {
+                "original_state": original_dm.tolist(),
+                "correction1": "Z" if m1 == 1 else "None",
+                "correction2": "X" if m2 == 1 else "None",
+                "received_state": dm.tolist(),
+                "fidelity": fidelity,
+            }
+        else:
+            return {
+                "correction1": "Z" if m1 == 1 else "None",
+                "correction2": "X" if m2 == 1 else "None",
+            }
+
+
+if __name__ == "__main__":
+    main()

examples/netqasm/teleport/app_sender.py

@@ -0,0 +1,58 @@
+from netqasm.logging.output import get_new_app_logger
+from netqasm.runtime.settings import Simulator, get_simulator
+from netqasm.sdk import EPRSocket, Qubit
+from netqasm.sdk.classical_communication.message import StructuredMessage
+from netqasm.sdk.external import NetQASMConnection, Socket
+from netqasm.sdk.toolbox import set_qubit_state
+
+
+def main(app_config=None, phi=0.0, theta=0.0):
+    log_config = app_config.log_config
+    app_logger = get_new_app_logger(app_name="sender", log_config=log_config)
+
+    # Create a socket to send classical information
+    socket = Socket("sender", "receiver", log_config=log_config)
+
+    # Create a EPR socket for entanglement generation
+    epr_socket = EPRSocket("receiver")
+
+    print("`sender` will start to teleport a qubit to `receiver`")
+
+    # Initialize the connection to the backend
+    sender = NetQASMConnection(
+        app_name=app_config.app_name, log_config=log_config, epr_sockets=[epr_socket]
+    )
+    with sender:
+        # Create a qubit to teleport
+        q = Qubit(sender)
+        set_qubit_state(q, phi, theta)
+
+        # Create EPR pairs
+        epr = epr_socket.create_keep()[0]
+
+        # Teleport
+        q.cnot(epr)
+        q.H()
+        m1 = q.measure()
+        m2 = epr.measure()
+
+    # Send the correction information
+    m1, m2 = int(m1), int(m2)
+
+    app_logger.log(f"m1 = {m1}")
+    app_logger.log(f"m2 = {m2}")
+    print(
+        f"`sender` measured the following teleportation corrections: m1 = {m1}, m2 = {m2}"
+    )
+    print("`sender` will send the corrections to `receiver`")
+
+    socket.send_structured(StructuredMessage("Corrections", (m1, m2)))
+
+    if get_simulator() == Simulator.NETSQUID:
+        socket.send_silent(str((phi, theta)))
+
+    return {"m1": m1, "m2": m2}
+
+
+if __name__ == "__main__":
+    main()