Merge pull request #10 from tqsd/homework_check_eqsn

added reading out the statevector

Author: benjione

eqsn/gates.py

@@ -2,7 +2,7 @@
 import logging
 import numpy as np
 from eqsn.qubit_thread import SINGLE_GATE, MERGE_SEND, MERGE_ACCEPT, MEASURE,\
-                MEASURE_NON_DESTRUCTIVE, DOUBLE_GATE, \
+                MEASURE_NON_DESTRUCTIVE, GIVE_STATEVECTOR, DOUBLE_GATE, \
                 CONTROLLED_GATE, NEW_QUBIT, ADD_MERGED_QUBITS_TO_DICT
 from eqsn.shared_dict import SharedDict
 from eqsn.worker_process import WorkerProcess
@@ -210,6 +210,13 @@ def cphase_gate(self, q_id1, q_id2):
         q = self.shared_dict.get_queues_for_ids([q_id1])[0]
         q.put([CONTROLLED_GATE, x, q_id1, q_id2])
 
+    def give_statevector_for(self, q_id):
+        ret = self.manager.Queue()
+        q = self.shared_dict.get_queues_for_ids([q_id])[0]
+        q.put([GIVE_STATEVECTOR, q_id, ret])
+        qubits, vector = ret.get()
+        return qubits, vector
+
     def custom_two_qubit_gate(self, q_id1, q_id2, gate):
         self.merge_qubits(q_id1, q_id2)
         q = self.shared_dict.get_queues_for_ids([q_id1])[0]

eqsn/qubit_thread.py

@@ -13,6 +13,7 @@
 MEASURE_NON_DESTRUCTIVE = 7
 NEW_QUBIT = 8
 ADD_MERGED_QUBITS_TO_DICT = 9
+GIVE_STATEVECTOR = 10
 DOUBLE_GATE = 11
 
 
@@ -60,6 +61,9 @@ def apply_single_gate(self, mat, q_id):
             apply_mat = np.kron(apply_mat, np.eye(2 ** after))
         self.qubit = np.dot(apply_mat, self.qubit)
 
+    def give_statevector(self, channel):
+        channel.put((dp(self.qubits), dp(self.qubit)))
+
     def apply_controlled_gate(self, mat, q_id1, q_id2):
         """
         Apply a controlled gate to
@@ -278,6 +282,8 @@ def run(self):
                 return
             elif item[0] == MEASURE_NON_DESTRUCTIVE:
                 self.measure_non_destructive(item[1], item[2])
+            elif item[0] == GIVE_STATEVECTOR:
+                self.give_statevector(item[1])
             elif item[0] == DOUBLE_GATE:
                 self.apply_two_qubit_gate(item[1], item[2], item[3])
             else:

eqsn/worker_process.py

@@ -4,7 +4,7 @@
 from queue import Queue
 from eqsn.shared_dict import SharedDict
 from eqsn.qubit_thread import SINGLE_GATE, MERGE_SEND, MERGE_ACCEPT, MEASURE,\
-                MEASURE_NON_DESTRUCTIVE, \
+                MEASURE_NON_DESTRUCTIVE, GIVE_STATEVECTOR, \
                 CONTROLLED_GATE, NEW_QUBIT, ADD_MERGED_QUBITS_TO_DICT, \
                 DOUBLE_GATE, QubitThread
 
@@ -43,6 +43,8 @@ def run(self):
                 self.measure_non_destructive(item[1], item[2])
             elif item[0] == ADD_MERGED_QUBITS_TO_DICT:
                 self.add_merged_qubits_to_thread(item[1], item[2])
+            elif item[0] == GIVE_STATEVECTOR:
+                self.give_statevector_for(item[1], item[2])
             elif item[0] == DOUBLE_GATE:
                 self.apply_two_qubit_gate(item[1], item[2], item[3])
             else:
@@ -99,6 +101,10 @@ def apply_single_gate(self, gate, q_id):
         q = self.shared_dict.get_queues_for_ids([q_id])[0]
         q.put([SINGLE_GATE, gate, q_id])
 
+    def give_statevector_for(self, q_id, channel):
+        q = self.shared_dict.get_queues_for_ids([q_id])[0]
+        q.put([GIVE_STATEVECTOR, channel])
+
     def apply_controlled_gate(self, gate, q_id1, q_id2):
         """
         Applies a controlled gate, where the gate is applied to