cirq-bb84-network

import cirq


class Network:
    def send(self, encode_function):
        qubits = [cirq.LineQubit(i) for i in range(3)]
        circuit = cirq.Circuit(
            encode_function(qubits[0]),
            cirq.H(qubits[1]),
            cirq.CNOT(qubits[1], qubits[2]),
            cirq.CNOT(qubits[0], qubits[1]),
            cirq.H(qubits[0])
        )
        return circuit, qubits

    def reflect_measure_results(self, circuit, qubits):
        # 測定結果が1ならボブのqubitにXゲートを追加
        circuit.append([
            cirq.CX(qubits[1], qubits[2]),
        ])
        # 測定結果が1ならボブのqubitにZゲートを追加
        circuit.append([
            cirq.CZ(qubits[0], qubits[2]),
        ])