You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
31 lines
1.3 KiB
31 lines
1.3 KiB
from dataclasses import dataclass, field |
|
from channel import Channel, ChannelSym |
|
import random as r |
|
import typing |
|
from utils import bin_encode, bin_decode |
|
|
|
|
|
@dataclass |
|
class Alice: |
|
channel: Channel |
|
sent: typing.List[bool] = field(default_factory=list) |
|
my_bases: typing.List[bool] = field(default_factory=list) |
|
key: typing.List[bool] = field(default_factory=list) |
|
correctness: typing.List[bool] = field(default_factory=list) |
|
|
|
def generate_key_and_send(self, n: int = 100): |
|
self.sent = [bool(r.randint(0, 1)) for _ in range(n)] |
|
self.my_bases = [bool(r.randint(0, 1)) for _ in range(n)] |
|
for key, basis in zip(self.sent, self.my_bases): |
|
self.channel.send_bit(key, basis) |
|
|
|
def generate_key(self, n: int = 100): |
|
self.generate_key_and_send(n) |
|
bobs_bases, bobs_correctness = bin_decode(self.channel.recv_classical(n, is_bob=False)), bin_decode( |
|
self.channel.recv_classical(n, is_bob=False)) |
|
self.correctness = [ |
|
my_basis == bobs_basis and bobs_correctness |
|
for my_basis, bobs_basis, bobs_correctness in |
|
zip(self.my_bases, bobs_bases, bobs_correctness)] |
|
self.channel.send_classical(bin_encode(self.correctness), is_bob=False) |
|
self.key = [k for k, c in zip(self.sent, self.correctness) if c]
|
|
|