Add a basic version of stack plots

Author: Chris00

examples/plot_types.rs

@@ -16,6 +16,7 @@ fn main() -> anyhow::Result<()> {
     bar()?;
     stem()?;
     fill_between()?;
+    stackplot()?;
     Ok(())
 }
 
@@ -108,3 +109,22 @@ fn fill_between() -> anyhow::Result<()> {
     fig.save().to_file(format!("{BASE}fill_between.pdf"))?;
     Ok(())
 }
+
+fn stackplot() -> anyhow::Result<()> {
+    use ndarray::{arr1, Axis, stack};
+
+    let x = Array1::range(0., 10., 2.);
+    let ay = arr1(&[1., 1.25, 2., 2.75, 3.]);
+    let by = arr1(&[1., 1., 1., 1., 1.]);
+    let cy = arr1(&[2., 1., 2., 1., 2.]);
+    let y = stack(Axis(0), &[ay.view(), by.view(), cy.view()]).unwrap();
+
+    let (fig, [[mut ax]]) = figure::subplots()?;
+
+    ax.stack(&x, &y).plot();
+
+    ax.set_xlim(0., 8.) .set_xticks((1..8).map(f64::from));
+    ax.set_ylim(0., 8.) .set_yticks((1..8).map(f64::from));
+    fig.save().to_file(format!("{BASE}stackplot.pdf"))?;
+    Ok(())
+}

src/axes.rs

@@ -282,6 +282,14 @@ impl Axes {
         FillBetween::new(self, x, y1, y2)
     }
 
+    pub fn stack<'a>(
+        &'a mut self,
+        x: &'a impl Vector<f64>,
+        y: &'a ndarray::Array2<f64>,
+    ) -> Stack<'a> {
+        Stack::new(self, x, y)
+    }
+
     /// Set the title to `txt` for the Axes.
     pub fn set_title(&mut self, txt: impl AsRef<str>) -> &mut Self {
         meth!(self.ax, set_title, (txt.as_ref(),)).unwrap();
@@ -1204,6 +1212,35 @@ impl<'a> FillBetween<'a> {
     }
 }
 
+pub struct Stack<'a> {
+    axes: &'a Axes,
+    x: &'a dyn Vector<f64>,
+    y: &'a Array2<f64>,
+    // FIXME: there are optional arguments.
+}
+
+impl<'a> Stack<'a> {
+    fn new(
+        axes: &'a Axes,
+        x: &'a impl Vector<f64>,
+        y: &'a Array2<f64>,
+    ) -> Self {
+        Self { axes, x, y }
+    }
+
+    pub fn plot(self) {
+        Python::attach(|py| {
+            let x = self.x.to_pyvector(py);
+            let y = self.y.to_pyarray(py);
+            self.axes.ax.bind(py)
+                .call_method(intern!(py, "stackplot"),
+                    (x, y),
+                    None)
+                .unwrap();
+        })
+    }
+}
+
 pub struct QuadContourSet {
     contours: Py<PyAny>,
 }