1 files changed, 37 insertions, 14 deletions
diff --git a/single_dequantify.py b/single_dequantify.py
index 3cf7d3a..f43f9f1 100644
--- a/single_dequantify.py
+++ b/single_dequantify.py
@@ -5,10 +5,9 @@ import cmath as cmt
 import numpy as npy
 import scipy.linalg as sla
 
-alpha = 1
-beta  = 1
-dt = 0.1
-iterations = 61
+dt = 0.08
+iterations = 100
+state_num = 1
 
 H = npy.array([[1,0],[0,2]])
 
@@ -16,8 +15,13 @@ if len(sys.argv) == 3:
     alpha = complex(sys.argv[1])
     beta = complex(sys.argv[2])
 
-norm  = npy.linalg.norm([alpha, beta])
-state = npy.array([alpha / norm, beta / norm])
+def init_state(i):
+    alpha = 1
+    beta  = 1
+    if state_num > 1:
+        alpha = i / state_num
+        beta  = (state_num - i) / state_num
+    return (alpha, beta)
 
 def time_evolution(state, dt = dt):
     return npy.dot(state, sla.expm(-1j * dt * H))
@@ -36,16 +40,35 @@ def sphere2cart(phi, theta):
              -1 * mth.cos(theta)
         ]
 
-historie = npy.array([bloch_map(state)])
 f = open("data", "w")
 
-for i in range(iterations + 1):
-    historie = npy.vstack([historie,bloch_map(state)])
-    (phi, theta) = bloch_map(state)
-    coords = sphere2cart(phi, theta)
-    f.write(f"{coords[0]}; {coords[1]}; {coords[2]}; {(i + 1) / (iterations + 1)}\n")
-    state = time_evolution(state)
+states = []
+for i in range(state_num):
+    (alpha, beta) = init_state(i)
+    norm  = npy.linalg.norm([alpha, beta])
+    state = npy.array([alpha / norm, beta / norm])
+    states.append(state)
+
+
+for i in range(iterations):
+    for j in range(state_num):
+        (phi, theta) = bloch_map(states[j])
+        coords = sphere2cart(phi, theta)
+        colour = i / iterations
+        if state_num > 1:
+            colour = j / state_num
+        f.write(f"{coords[0]}; {coords[1]}; {coords[2]}; {colour}\n")
+        states[j] = time_evolution(states[j])
 
 f.close()
 
-subprocess.run(["gnuplot", "gnuplot.plt"])
+import os
+
+path = 'gnuplot/'
+if os.name == 'nt':
+    path.replace('/','\\')
+subprocess.run(["gnuplot",
+    "-e", f"states={state_num};",
+    "-e", f"start={iterations};" ,
+    "-e", f"end={iterations};",
+    "-c", path + "animation.plt"])