diff options
Diffstat (limited to 'src/simulation/two_level.rs')
| -rw-r--r-- | src/simulation/two_level.rs | 8 |
1 files changed, 6 insertions, 2 deletions
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 { |