/**

* @file gridspec.h

* @brief corresponding header for matplotlib.axes

**/

#pragma once

#include <matplotlibcpp17/common.h>

#include <pybind11/pybind11.h>

namespace matplotlibcpp17::gridspec {

/**

* @brief A wrapper class for matplotlib.gridspec.SubplotSpec

**/

struct DECL_STRUCT_ATTR SubplotSpec : public BaseWrapper {

public:

SubplotSpec(const pybind11::object &subplotspec) { self = subplotspec; }

SubplotSpec(pybind11::object &&subplotspec) { self = std::move(subplotspec); }

};

/**

* @brief A wrapper class for matplotlib.gridspec.GridSpec

**/

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