From cde05ddcf3c08a0852959d298e500c726ae21a25 Mon Sep 17 00:00:00 2001
From: Daniel Tsvetkov <danieltcv@gmail.com>
Date: Thu, 30 Jan 2020 13:28:39 +0100
Subject: [PATCH] added cirq for sim

---
 06_krisis.py | 99 ++++++++++++++++++++++++++++++++++++++++++++++------
 1 file changed, 89 insertions(+), 10 deletions(-)

diff --git a/06_krisis.py b/06_krisis.py
index e83eade..07250cc 100644
--- a/06_krisis.py
+++ b/06_krisis.py
@@ -1,3 +1,4 @@
+import cirq
 import numpy as np
 from collections import defaultdict
 
@@ -12,7 +13,37 @@ def from_angles_1(theta, phi):
     return State(m.m)
 
 
-def krisi(q2func, iterations=100, sample_count=1):
+class MemoizedExp(object):
+    def __init__(self):
+        self.theta = None
+        self.phi = None
+
+    def gen_exp_for_cirq_0(self):
+        if not (self.theta and self.phi):
+            self.theta = round(np.random.uniform(0, 1), 10)
+            self.phi = round(np.random.uniform(0, 1), 10)
+        return self.theta, self.phi
+
+    def reset(self):
+        self.theta = None
+        self.phi = None
+
+
+def gen_exp_for_cirq_1():
+    """TODO: How to generate the exponents for the second case"""
+    theta = round(np.random.uniform(0, 1), 10)
+    phi = round(np.random.uniform(0, 1), 10)
+    return theta, phi
+
+
+def print_all_samples(all_samples):
+    print("------------- ALL SAMPLES for cirq")
+    for k, v in sorted(all_samples.items(), key=lambda x: x[0]):
+        print("{}: {}".format(k, v))
+    print("==============================================")
+
+
+def math_sim(q2func=from_angles_1, iterations=1000, sample_count=1):
     all_samples = defaultdict(int)
     for i in range(iterations):
         # print("Running iteration {}".format(i))
@@ -31,19 +62,67 @@ def krisi(q2func, iterations=100, sample_count=1):
         this_samples = qp.get_sample(sample_count)
         for k, v in this_samples.items():
             all_samples[k] += v
-    print("------------- ALL SAMPLES for {}".format(q2func.__name__))
-    for k, v in sorted(all_samples.items(), key=lambda x: x[0]):
-        print("{}: {}".format(k, v))
-    print("==============================================")
+    print_all_samples(all_samples)
 
 
-def krisi_0():
-    krisi(q2func=State.from_angles)
+def cirq_sim(q1func, q2func, memoized_exp, iterations=1000, sample_count=1):
+    all_samples = defaultdict(int)
+    for i in range(iterations):
+        theta1, phi1 = q1func()
+        theta2, phi2 = q2func()
+        memoized_exp.reset()
+
+        q1 = cirq.GridQubit(0, 0)
+        q2 = cirq.GridQubit(1, 0)
+
+        # Create a circuit
+        circuit = cirq.Circuit(
+            cirq.XPowGate(exponent=theta1).on(q1),
+            cirq.ZPowGate(exponent=phi1).on(q1),
+            cirq.XPowGate(exponent=theta2).on(q2),
+            cirq.ZPowGate(exponent=phi2).on(q2),
+            cirq.CNOT(q1, q2),
+            cirq.H(q1),
+
+            cirq.measure(q1, key='q1'),  # Measurement.
+            cirq.measure(q2, key='q2')  # Measurement.
+        )
+        # print(circuit)
+
+        # Simulate the circuit several times.
+        simulator = cirq.Simulator()
+        result = simulator.run(circuit, repetitions=sample_count)
+        for rep in range(len(result.measurements['q1'])):
+            rq1 = result.measurements['q1'][rep][0]
+            rq2 = result.measurements['q2'][rep][0]
+            k = "{}{}".format(rq1, rq2)
+            all_samples[k] += 1
+    print_all_samples(all_samples)
 
 
-def krisi_1():
-    krisi(q2func=from_angles_1)
+def math_sim_0(*args, **kwargs):
+    math_sim(q2func=State.from_angles, *args, **kwargs)
+
+
+def math_sim_1(*args, **kwargs):
+    math_sim(q2func=from_angles_1, *args, **kwargs)
+
+
+def cirq_sim_0(*args, **kwargs):
+    memoized_exp = MemoizedExp()
+    cirq_sim(q1func=memoized_exp.gen_exp_for_cirq_0,
+             q2func=memoized_exp.gen_exp_for_cirq_0,
+             memoized_exp=memoized_exp,
+             *args, **kwargs)
+
+
+def cirq_sim_1(*args, **kwargs):
+    memoized_exp = MemoizedExp()
+    cirq_sim(q1func=memoized_exp.gen_exp_for_cirq_0,
+             q2func=gen_exp_for_cirq_1,
+             memoized_exp=memoized_exp,
+             *args, **kwargs)
 
 
 if __name__ == "__main__":
-    krisi_0()
+    cirq_sim_0()