summaryrefslogtreecommitdiff
path: root/src
diff options
context:
space:
mode:
Diffstat (limited to 'src')
-rw-r--r--src/plot/gnuplot.rs1
-rw-r--r--src/simulation/two_level.rs6
2 files changed, 3 insertions, 4 deletions
diff --git a/src/plot/gnuplot.rs b/src/plot/gnuplot.rs
index 8bb5ac6..fec64b7 100644
--- a/src/plot/gnuplot.rs
+++ b/src/plot/gnuplot.rs
@@ -29,6 +29,7 @@ fn write_data_file(path: &str, data: &[[f64; 6]]) -> Result<(), io::Error> {
line[0], line[1], line[2], line[3], line[4], line[5]
))?;
}
+ Ok(())
}
impl Gnuplot {
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()
}