py/pykmsbase.cpp

   1 #include <pybind11/pybind11.h>
   2 #include <pybind11/stl.h>
   3 #include <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("call_page_flip_handlers", &Card::call_page_flip_handlers)
  22                         .def("get_prop", (Property* (Card::*)(uint32_t) const)&Card::get_prop)
  23                         .def("get_prop", (Property* (Card::*)(const string&) const)&Card::get_prop)
  24                         ;
  25
  26         py::class_<DrmObject, DrmObject*>(m, "DrmObject")
  27                         .def_property_readonly("id", &DrmObject::id)
  28                         .def_property_readonly("card", &DrmObject::card)
  29                         ;
  30
  31         py::class_<DrmPropObject, DrmPropObject*>(m, "DrmPropObject", py::base<DrmObject>())
  32                         .def("refresh_props", &DrmPropObject::refresh_props)
  33                         .def_property_readonly("prop_map", &DrmPropObject::get_prop_map)
  34                         ;
  35
  36         py::class_<Connector, Connector*>(m, "Connector",  py::base<DrmPropObject>())
  37                         .def_property_readonly("fullname", &Connector::fullname)
  38                         .def("get_default_mode", &Connector::get_default_mode)
  39                         .def("get_current_crtc", &Connector::get_current_crtc)
  40                         .def("get_modes", &Connector::get_modes)
  41                         .def("__repr__", [](const Connector& o) { return "<pykms.Connector " + to_string(o.id()) + ">"; })
  42                         ;
  43
  44         py::class_<Crtc, Crtc*>(m, "Crtc",  py::base<DrmPropObject>())
  45                         .def("set_mode", &Crtc::set_mode)
  46                         .def("page_flip", &Crtc::page_flip)
  47                         .def("set_plane", &Crtc::set_plane)
  48                         .def_property_readonly("possible_planes", &Crtc::get_possible_planes)
  49                         .def_property_readonly("primary_plane", &Crtc::get_primary_plane)
  50                         .def("__repr__", [](const Crtc& o) { return "<pykms.Crtc " + to_string(o.id()) + ">"; })
  51                         ;
  52
  53         py::class_<Encoder, Encoder*>(m, "Encoder",  py::base<DrmPropObject>())
  54                         ;
  55
  56         py::class_<Plane, Plane*>(m, "Plane",  py::base<DrmPropObject>())
  57                         .def("supports_crtc", &Plane::supports_crtc)
  58                         .def_property_readonly("plane_type", &Plane::plane_type)
  59                         .def("__repr__", [](const Plane& o) { return "<pykms.Plane " + to_string(o.id()) + ">"; })
  60                         ;
  61
  62         py::enum_<PlaneType>(m, "PlaneType")
  63                         .value("Overlay", PlaneType::Overlay)
  64                         .value("Primary", PlaneType::Primary)
  65                         .value("Cursor", PlaneType::Cursor)
  66                         ;
  67
  68         py::class_<Property, Property*>(m, "Property",  py::base<DrmObject>())
  69                         .def_property_readonly("name", &Property::name)
  70                         ;
  71
  72         py::class_<Framebuffer>(m, "Framebuffer",  py::base<DrmObject>())
  73                         ;
  74
  75         py::class_<DumbFramebuffer>(m, "DumbFramebuffer",  py::base<Framebuffer>())
  76                         .def(py::init<Card&, uint32_t, uint32_t, const string&>(),
  77                              py::keep_alive<1, 2>())    // Keep Card alive until this is destructed
  78                         .def_property_readonly("width", &DumbFramebuffer::width)
  79                         .def_property_readonly("height", &DumbFramebuffer::height)
  80                         ;
  81
  82         py::class_<Videomode>(m, "Videomode")
  83                         .def(py::init<>())
  84
  85                         .def_readwrite("name", &Videomode::name)
  86
  87                         .def_readwrite("clock", &Videomode::clock)
  88
  89                         .def_readwrite("hdisplay", &Videomode::hdisplay)
  90                         .def_readwrite("hsync_start", &Videomode::hsync_start)
  91                         .def_readwrite("hsync_end", &Videomode::hsync_end)
  92                         .def_readwrite("htotal", &Videomode::htotal)
  93
  94                         .def_readwrite("vdisplay", &Videomode::vdisplay)
  95                         .def_readwrite("vsync_start", &Videomode::vsync_start)
  96                         .def_readwrite("vsync_end", &Videomode::vsync_end)
  97                         .def_readwrite("vtotal", &Videomode::vtotal)
  98
  99                         .def_readwrite("vrefresh", &Videomode::vrefresh)
 100
 101                         .def_readwrite("flags", &Videomode::flags)
 102                         .def_readwrite("type", &Videomode::type)
 103                         ;
 104
 105         py::class_<AtomicReq>(m, "AtomicReq")
 106                         .def(py::init<Card&>(),
 107                              py::keep_alive<1, 2>())    // Keep Card alive until this is destructed
 108                         .def("add", (void (AtomicReq::*)(DrmObject*, const string&, uint64_t)) &AtomicReq::add)
 109                         .def("test", &AtomicReq::test)
 110                         .def("commit", &AtomicReq::commit)
 111                         .def("commit_sync", &AtomicReq::commit_sync)
 112                         ;
 113 }