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, \
+                MEASURE_NON_DESTRUCTIVE, GIVE_STATEVECTOR, \
                 CONTROLLED_GATE, NEW_QUBIT, ADD_MERGED_QUBITS_TO_DICT
 from eqsn.shared_dict import SharedDict
 from eqsn.worker_process import WorkerProcess
@@ -203,6 +203,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 measure(self, q_id, non_destructive=False):
         """
         Measures a qubit with an id. If non_destructive is False, the qubit

eqsn/qubit_thread.py

@@ -13,6 +13,7 @@
 MEASURE_NON_DESTRUCTIVE = 7
 NEW_QUBIT = 8
 ADD_MERGED_QUBITS_TO_DICT = 9
+GIVE_STATEVECTOR = 10
 
 
 class QubitThread(object):
@@ -59,6 +60,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
@@ -235,5 +239,7 @@ 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])
             else:
                 raise ValueError("Command does not exist!")

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, QubitThread
 
 
@@ -42,6 +42,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])
             else:
                 raise ValueError("Command does not exist!")
 
@@ -91,6 +93,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