py/pykms/pykmsbase.cpp

   1 #include <pybind11/pybind11.h>
   2 #include <pybind11/stl.h>
   3 #include <kms++/kms++.h>
   4
   5 namespace py = pybind11;
   6
   7 using namespace kms;
   8 using namespace std;
   9
  10 void init_pykmsbase(py::module &m)
  11 {
  12         py::class_<Card>(m, "Card")
  13                         .def(py::init<>())
  14                         .def_property_readonly("fd", &Card::fd)
  15                         .def("get_first_connected_connector", &Card::get_first_connected_connector)
  16                         .def_property_readonly("connectors", &Card::get_connectors)
  17                         .def_property_readonly("crtcs", &Card::get_crtcs)
  18                         .def_property_readonly("encoders", &Card::get_encoders)
  19                         .def_property_readonly("planes", &Card::get_planes)
  20                         .def_property_readonly("has_atomic", &Card::has_atomic)
  21                         .def("get_prop", (Property* (Card::*)(uint32_t) const)&Card::get_prop)
  22                         ;
  23
  24         py::class_<DrmObject, DrmObject*>(m, "DrmObject")
  25                         .def_property_readonly("id", &DrmObject::id)
  26                         .def_property_readonly("card", &DrmObject::card)
  27                         ;
  28
  29         py::class_<DrmPropObject, DrmPropObject*>(m, "DrmPropObject", py::base<DrmObject>())
  30                         .def("refresh_props", &DrmPropObject::refresh_props)
  31                         .def_property_readonly("prop_map", &DrmPropObject::get_prop_map)
  32                         .def("get_prop_value", (uint64_t (DrmPropObject::*)(const string&) const)&DrmPropObject::get_prop_value)
  33                         .def("set_prop_value",(int (DrmPropObject::*)(const string&, uint64_t)) &DrmPropObject::set_prop_value)
  34                         .def("get_prop_value_as_blob", &DrmPropObject::get_prop_value_as_blob)
  35                         ;
  36
  37         py::class_<Connector, Connector*>(m, "Connector",  py::base<DrmPropObject>())
  38                         .def_property_readonly("fullname", &Connector::fullname)
  39                         .def("get_default_mode", &Connector::get_default_mode)
  40                         .def("get_current_crtc", &Connector::get_current_crtc)
  41                         .def("get_possible_crtcs", &Connector::get_possible_crtcs)
  42                         .def("get_modes", &Connector::get_modes)
  43                         .def("get_mode", (Videomode (Connector::*)(const string& mode) const)&Connector::get_mode)
  44                         .def("get_mode", (Videomode (Connector::*)(unsigned xres, unsigned yres, float refresh, bool ilace) const)&Connector::get_mode)
  45                         .def("__repr__", [](const Connector& o) { return "<pykms.Connector " + to_string(o.id()) + ">"; })
  46                         .def("refresh", &Connector::refresh)
  47                         ;
  48
  49         py::class_<Crtc, Crtc*>(m, "Crtc",  py::base<DrmPropObject>())
  50                         .def("set_mode", &Crtc::set_mode)
  51                         .def("page_flip",
  52                              [](Crtc* self, Framebuffer& fb, py::object ob)
  53                                 {
  54                                         // This adds a ref to the object, and must be unpacked with __ob_unpack_helper()
  55                                         PyObject* pob = ob.ptr();
  56                                         Py_XINCREF(pob);
  57                                         self->page_flip(fb, pob);
  58                                 })
  59                         .def("set_plane", &Crtc::set_plane)
  60                         .def_property_readonly("possible_planes", &Crtc::get_possible_planes)
  61                         .def_property_readonly("primary_plane", &Crtc::get_primary_plane)
  62                         .def_property_readonly("mode", &Crtc::mode)
  63                         .def_property_readonly("mode_valid", &Crtc::mode_valid)
  64                         .def("__repr__", [](const Crtc& o) { return "<pykms.Crtc " + to_string(o.id()) + ">"; })
  65                         .def("refresh", &Crtc::refresh)
  66                         ;
  67
  68         py::class_<Encoder, Encoder*>(m, "Encoder",  py::base<DrmPropObject>())
  69                         .def("refresh", &Encoder::refresh)
  70                         ;
  71
  72         py::class_<Plane, Plane*>(m, "Plane",  py::base<DrmPropObject>())
  73                         .def("supports_crtc", &Plane::supports_crtc)
  74                         .def_property_readonly("plane_type", &Plane::plane_type)
  75                         .def("__repr__", [](const Plane& o) { return "<pykms.Plane " + to_string(o.id()) + ">"; })
  76                         ;
  77
  78         py::enum_<PlaneType>(m, "PlaneType")
  79                         .value("Overlay", PlaneType::Overlay)
  80                         .value("Primary", PlaneType::Primary)
  81                         .value("Cursor", PlaneType::Cursor)
  82                         ;
  83
  84         py::class_<Property, Property*>(m, "Property",  py::base<DrmObject>())
  85                         .def_property_readonly("name", &Property::name)
  86                         ;
  87
  88         py::class_<Blob>(m, "Blob", py::base<DrmObject>())
  89                         .def("__init__", [](Blob& instance, Card& card, py::buffer buf) {
  90                                 py::buffer_info info = buf.request();
  91                                 if (info.ndim != 1)
  92                                         throw std::runtime_error("Incompatible buffer dimension!");
  93
  94                                 new (&instance) Blob(card, info.ptr, info.size * info.itemsize);
  95                         })
  96                         .def_property_readonly("data", &Blob::data)
  97                         ;
  98
  99         py::class_<Framebuffer>(m, "Framebuffer", py::base<DrmObject>())
 100                         ;
 101
 102         py::class_<MappedFramebuffer>(m, "MappedFramebuffer", py::base<Framebuffer>())
 103                         .def_property_readonly("width", &MappedFramebuffer::width)
 104                         .def_property_readonly("height", &MappedFramebuffer::height)
 105                         ;
 106
 107         py::class_<DumbFramebuffer>(m, "DumbFramebuffer", py::base<MappedFramebuffer>())
 108                         .def(py::init<Card&, uint32_t, uint32_t, const string&>(),
 109                              py::keep_alive<1, 2>())    // Keep Card alive until this is destructed
 110                         .def(py::init<Card&, uint32_t, uint32_t, PixelFormat>(),
 111                              py::keep_alive<1, 2>())    // Keep Card alive until this is destructed
 112                         .def_property_readonly("format", &DumbFramebuffer::format)
 113                         .def_property_readonly("num_planes", &DumbFramebuffer::num_planes)
 114                         .def("fd", &DumbFramebuffer::prime_fd)
 115                         .def("stride", &DumbFramebuffer::stride)
 116                         .def("offset", &DumbFramebuffer::offset)
 117                         ;
 118
 119         py::class_<ExtFramebuffer>(m, "ExtFramebuffer", py::base<MappedFramebuffer>())
 120                         .def(py::init<Card&, uint32_t, uint32_t, PixelFormat, vector<int>, vector<uint32_t>, vector<uint32_t>>(),
 121                              py::keep_alive<1, 2>())    // Keep Card alive until this is destructed
 122                         ;
 123
 124         py::enum_<PixelFormat>(m, "PixelFormat")
 125                         .value("Undefined", PixelFormat::Undefined)
 126
 127                         .value("NV12", PixelFormat::NV12)
 128                         .value("NV21", PixelFormat::NV21)
 129
 130                         .value("UYVY", PixelFormat::UYVY)
 131                         .value("YUYV", PixelFormat::YUYV)
 132                         .value("YVYU", PixelFormat::YVYU)
 133                         .value("VYUY", PixelFormat::VYUY)
 134
 135                         .value("XRGB8888", PixelFormat::XRGB8888)
 136                         .value("XBGR8888", PixelFormat::XBGR8888)
 137                         .value("ARGB8888", PixelFormat::ARGB8888)
 138                         .value("ABGR8888", PixelFormat::ABGR8888)
 139
 140                         .value("RGB888", PixelFormat::RGB888)
 141
 142                         .value("RGB565", PixelFormat::RGB565)
 143                         ;
 144
 145         py::class_<Videomode>(m, "Videomode")
 146                         .def(py::init<>())
 147
 148                         .def_readwrite("name", &Videomode::name)
 149
 150                         .def_readwrite("clock", &Videomode::clock)
 151
 152                         .def_readwrite("hdisplay", &Videomode::hdisplay)
 153                         .def_readwrite("hsync_start", &Videomode::hsync_start)
 154                         .def_readwrite("hsync_end", &Videomode::hsync_end)
 155                         .def_readwrite("htotal", &Videomode::htotal)
 156
 157                         .def_readwrite("vdisplay", &Videomode::vdisplay)
 158                         .def_readwrite("vsync_start", &Videomode::vsync_start)
 159                         .def_readwrite("vsync_end", &Videomode::vsync_end)
 160                         .def_readwrite("vtotal", &Videomode::vtotal)
 161
 162                         .def_readwrite("vrefresh", &Videomode::vrefresh)
 163
 164                         .def_readwrite("flags", &Videomode::flags)
 165                         .def_readwrite("type", &Videomode::type)
 166
 167                         .def("__repr__", [](const Videomode& vm) { return "<pykms.Videomode " + to_string(vm.hdisplay) + "x" + to_string(vm.vdisplay) + ">"; })
 168                         ;
 169
 170         py::class_<AtomicReq>(m, "AtomicReq")
 171                         .def(py::init<Card&>(),
 172                              py::keep_alive<1, 2>())    // Keep Card alive until this is destructed
 173                         .def("add", (void (AtomicReq::*)(DrmPropObject*, const string&, uint64_t)) &AtomicReq::add)
 174                         .def("add", (void (AtomicReq::*)(DrmPropObject*, const map<string, uint64_t>&)) &AtomicReq::add)
 175                         .def("test", &AtomicReq::test, py::arg("allow_modeset") = false)
 176                         .def("commit",
 177                              [](AtomicReq* self, py::object ob, bool allow)
 178                                 {
 179                                         // This adds a ref to the object, and must be unpacked with __ob_unpack_helper()
 180                                         PyObject* pob = ob.ptr();
 181                                         Py_XINCREF(pob);
 182                                         return self->commit(pob, allow);
 183                                 }, py::arg("data"), py::arg("allow_modeset") = false)
 184                         .def("commit_sync", &AtomicReq::commit_sync, py::arg("allow_modeset") = false)
 185                         ;
 186 }