import time
from qunetsim.components import Host
from qunetsim.components import Network
import random
from qunetsim.objects import Qubit
def distribute(host, nodes):
    # distribute=True => sender doesn't keep part of the GHZ
    host.send_ghz(nodes, distribute=True)
def sender(host, distributor, r, epr_id):
    q = host.get_ghz(distributor, wait=10)
    b = random.choice(['0', '1'])
    host.send_broadcast(b)
    if b == '1':
        q.Z()
    host.add_epr(r, q, q_id=epr_id)
    sending_qubit = Qubit(host)
    sending_qubit.X()
    print('Sending %s' % sending_qubit.id)
    # Generate EPR if none shouldn't change anything, but if there is
    # no shared entanglement between s and r, then there should
    # be a mistake in the protocol
    host.send_teleport(r, sending_qubit, generate_epr_if_none=False, await_ack=False)
    host.empty_classical()
def receiver(host, distributor, s, epr_id):
    q = host.get_ghz(distributor, wait=10)
    b = random.choice(['0', '1'])
    host.send_broadcast(b)
    messages = []
    # Await broadcast messages from all parties
    while len(messages) < 3:
        messages = host.classical
        messages = [x for x in messages if x.content != 'ACK']
        time.sleep(0.5)
    print([m.content for m in messages])
    parity = int(b)
    for m in messages:
        if m.sender != s:
            parity = parity ^ int(m.content)
    if parity == 1:
        q.Z()

# -*- coding: utf-8 -*-
import numpy as np
import binascii
from random import getrandbits
from qunetsim.components.host import Host
from qunetsim.components.network import Network
from qunetsim.objects import Qubit
from qunetsim.objects import Logger
Logger.DISABLED = True
class BB84:
    def __init__(self):
        print("Welcome to BB84 demo")
    def select_encoding(self, length):
        alice_bitstring = ""
        alice_bases = ""
        for i in range(length):
            alice_bitstring += (str(getrandbits(1)))
            alice_bases += (str(getrandbits(1)))
            # 0 means (0,1) basis and 1 means(+,-)basis
        return alice_bitstring, alice_bases
    def select_measurement(self, length):   # Bob bases
        bases = ""
        for i in range(length):
            bases += (str(getrandbits(1)))
        return bases
    def encode(self, host_sender, bitstring, bases):
        encoded_qubits = []
        for i in range(len(bitstring)):
            q = Qubit(host_sender)
            if bases[i] == "0":
                if bitstring[i] == "0":
                    pass  # initialized in 0
                elif bitstring[i] == "1":
                    q.X()
            elif bases[i] == "1":
                if bitstring[i] == "0":
                    q.H()
                elif bitstring[i] == "1":
                    q.X()
                    q.H()

import time
import numpy as np
from qunetsim.components import Host
from qunetsim.components import Network
from qunetsim.objects import Logger
from qunetsim.objects import Qubit
WAIT_TIME = 10
def checksum_sender(host, q_size, receiver_id, checksum_size_per_qubit):
    bit_arr = np.random.randint(2, size=q_size)
    Logger.get_instance().log('Bit array to be sent: ' + str(bit_arr))
    qubits = []
    for i in range(q_size):
        q_tmp = Qubit(host)
        if bit_arr[i] == 1:
            q_tmp.X()
        qubits.append(q_tmp)
    check_qubits = host.add_checksum(qubits, checksum_size_per_qubit)
    checksum_size = int(q_size / checksum_size_per_qubit)
    qubits.append(check_qubits)
    checksum_cnt = 0
    for i in range(q_size + checksum_size):
        if i < q_size:
            q = qubits[i]
        else:
            q = qubits[q_size][checksum_cnt]
            checksum_cnt = checksum_cnt + 1
        host.send_qubit(receiver_id, q, await_ack=True)
def checksum_receiver(host, q_size, sender_id, checksum_size_per_qubit):
    qubits = []
    checksum_size = int(q_size / checksum_size_per_qubit)
    while len(qubits) < (q_size + checksum_size):
        q = host.get_qubit(sender_id, wait=WAIT_TIME)
        qubits.append(q)
        Logger.get_instance().log(str(host.host_id) + ': received qubit')
    checksum_qubits = []
    checksum_cnt = 0
    for i in range(len(qubits)):
        if checksum_cnt < checksum_size:
            checksum_qubits.append(qubits[q_size + i]['q'])
            checksum_cnt = checksum_cnt + 1