From de571765f161a49129fd9e34150f6a892f388bdc Mon Sep 17 00:00:00 2001
From: Dennis Kobert <dennis@kobert.dev>
Date: Fri, 13 Dec 2019 04:00:31 +0100
Subject: Add fibartion matrices

---
 src/simulation/two_level.rs | 8 ++++++--
 1 file changed, 6 insertions(+), 2 deletions(-)

(limited to 'src/simulation/two_level.rs')

diff --git a/src/simulation/two_level.rs b/src/simulation/two_level.rs
index 38a65d9..99336a5 100644
--- a/src/simulation/two_level.rs
+++ b/src/simulation/two_level.rs
@@ -15,8 +15,12 @@ impl TwoLevel {
 }
 
 impl State for TwoLevel {
-    fn fibrate(&self) -> Array1<c64> {
-        self.state.clone()
+    fn fibrate(&self) -> Vec<f64> {
+        self.state
+            .iter()
+            .flat_map(|c| [c.re, c.im])
+            .map(|v| *v)
+            .collect()
     }
 
     fn evolve<G: super::time_evolution::MatrixGen>(mut self, t: f64) -> Self {
-- 
cgit v1.2.3-54-g00ecf