gridspec.h

 
#pragma once
 
#include <matplotlibcpp17/common.h>
 
#include <pybind11/pybind11.h>
 
namespace matplotlibcpp17::gridspec {
 
struct DECL_STRUCT_ATTR SubplotSpec : public BaseWrapper {
public:
  SubplotSpec(const pybind11::object &subplotspec) { self = subplotspec; }
  SubplotSpec(pybind11::object &&subplotspec) { self = std::move(subplotspec); }
};
 
struct DECL_STRUCT_ATTR GridSpec : public BaseWrapper {
public:
  GridSpec(int nrow_, int ncol_,
           const pybind11::dict &kwargs = pybind11::dict()) {
    nrow = nrow_;
    ncol = ncol_;
    gridspec_attr =
        pybind11::module::import("matplotlib.gridspec").attr("GridSpec");
    self = gridspec_attr(nrow, ncol, **kwargs);
  }
  template <typename Rows, typename Cols>
  SubplotSpec operator()(const Rows &r, const Cols &c) {
    pybind11::object obj = self[pybind11::make_tuple(r, c)];
    return SubplotSpec(std::move(obj));
  }
 
private:
  int nrow, ncol;
  pybind11::object gridspec_attr;
};
 
} // namespace matplotlibcpp17::gridspec