#include "PyLoopLevel.h"

namespace Halide {
namespace PythonBindings {

void define_loop_level(py::module &m) {
    // Note that the public-but-only-intended-for-internal-use methods
    // are deliberately omitted.
    auto looplevel_class = py::class_<LoopLevel>(m, "LoopLevel")
        .def(py::init<>())
        .def(py::init<const Func &, VarOrRVar, int>(),
             py::arg("func"), py::arg("var"), py::arg("stage_index") = -1)
        .def("stage_index", &LoopLevel::stage_index)
        .def("set", &LoopLevel::set)
        .def_static("inlined", &LoopLevel::inlined)
        .def_static("root", &LoopLevel::root)
        .def("__repr__", [](const LoopLevel &b) -> std::string {
            std::ostringstream o;
            o << "<halide.LoopLevel " << (b.defined() ? b.to_string() : "UNDEF") << ">";
            return o.str();
        })
    ;
}

}  // namespace PythonBindings
}  // namespace Halide