diff options
Diffstat (limited to 'src/simulation')
| -rw-r--r-- | src/simulation/two_level.rs | 6 |
1 files changed, 2 insertions, 4 deletions
diff --git a/src/simulation/two_level.rs b/src/simulation/two_level.rs index 99336a5..2d9d459 100644 --- a/src/simulation/two_level.rs +++ b/src/simulation/two_level.rs @@ -16,10 +16,8 @@ impl TwoLevel { impl State for TwoLevel { fn fibrate(&self) -> Vec<f64> { - self.state - .iter() - .flat_map(|c| [c.re, c.im]) - .map(|v| *v) + self.state.iter() + .flat_map(|c| std::iter::once(c.re).chain(std::iter::once(c.im))) .collect() } |