1 #include <pybind11/pybind11.h>
2 #include <pybind11/stl.h>
3 #include <kms++/kms++.h>
5 namespace py = pybind11;
7 using namespace kms;
8 using namespace std;
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 ;
24 py::class_<DrmObject, DrmObject*>(m, "DrmObject")
25 .def_property_readonly("id", &DrmObject::id)
26 .def_property_readonly("card", &DrmObject::card)
27 ;
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 ;
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 ;
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 ;
68 py::class_<Encoder, Encoder*>(m, "Encoder", py::base<DrmPropObject>())
69 .def("refresh", &Encoder::refresh)
70 ;
72 py::class_<Plane, Plane*>(m, "Plane", py::base<DrmPropObject>())
73 .def("supports_crtc", &Plane::supports_crtc)
74 .def_property_readonly("formats", &Plane::get_formats)
75 .def_property_readonly("plane_type", &Plane::plane_type)
76 .def("__repr__", [](const Plane& o) { return "<pykms.Plane " + to_string(o.id()) + ">"; })
77 ;
79 py::enum_<PlaneType>(m, "PlaneType")
80 .value("Overlay", PlaneType::Overlay)
81 .value("Primary", PlaneType::Primary)
82 .value("Cursor", PlaneType::Cursor)
83 ;
85 py::class_<Property, Property*>(m, "Property", py::base<DrmObject>())
86 .def_property_readonly("name", &Property::name)
87 ;
89 py::class_<Blob>(m, "Blob", py::base<DrmObject>())
90 .def("__init__", [](Blob& instance, Card& card, py::buffer buf) {
91 py::buffer_info info = buf.request();
92 if (info.ndim != 1)
93 throw std::runtime_error("Incompatible buffer dimension!");
95 new (&instance) Blob(card, info.ptr, info.size * info.itemsize);
96 })
97 .def_property_readonly("data", &Blob::data)
98 ;
100 py::class_<Framebuffer>(m, "Framebuffer", py::base<DrmObject>())
101 ;
103 py::class_<MappedFramebuffer>(m, "MappedFramebuffer", py::base<Framebuffer>())
104 .def_property_readonly("width", &MappedFramebuffer::width)
105 .def_property_readonly("height", &MappedFramebuffer::height)
106 ;
108 py::class_<DumbFramebuffer>(m, "DumbFramebuffer", py::base<MappedFramebuffer>())
109 .def(py::init<Card&, uint32_t, uint32_t, const string&>(),
110 py::keep_alive<1, 2>()) // Keep Card alive until this is destructed
111 .def(py::init<Card&, uint32_t, uint32_t, PixelFormat>(),
112 py::keep_alive<1, 2>()) // Keep Card alive until this is destructed
113 .def_property_readonly("format", &DumbFramebuffer::format)
114 .def_property_readonly("num_planes", &DumbFramebuffer::num_planes)
115 .def("fd", &DumbFramebuffer::prime_fd)
116 .def("stride", &DumbFramebuffer::stride)
117 .def("offset", &DumbFramebuffer::offset)
118 ;
120 py::class_<ExtFramebuffer>(m, "ExtFramebuffer", py::base<MappedFramebuffer>())
121 .def(py::init<Card&, uint32_t, uint32_t, PixelFormat, vector<int>, vector<uint32_t>, vector<uint32_t>>(),
122 py::keep_alive<1, 2>()) // Keep Card alive until this is destructed
123 ;
125 py::enum_<PixelFormat>(m, "PixelFormat")
126 .value("Undefined", PixelFormat::Undefined)
128 .value("NV12", PixelFormat::NV12)
129 .value("NV21", PixelFormat::NV21)
131 .value("UYVY", PixelFormat::UYVY)
132 .value("YUYV", PixelFormat::YUYV)
133 .value("YVYU", PixelFormat::YVYU)
134 .value("VYUY", PixelFormat::VYUY)
136 .value("XRGB8888", PixelFormat::XRGB8888)
137 .value("XBGR8888", PixelFormat::XBGR8888)
138 .value("ARGB8888", PixelFormat::ARGB8888)
139 .value("ABGR8888", PixelFormat::ABGR8888)
141 .value("RGB888", PixelFormat::RGB888)
142 .value("BGR888", PixelFormat::BGR888)
144 .value("RGB565", PixelFormat::RGB565)
145 .value("BGR565", PixelFormat::BGR565)
146 ;
148 py::class_<Videomode>(m, "Videomode")
149 .def(py::init<>())
151 .def_readwrite("name", &Videomode::name)
153 .def_readwrite("clock", &Videomode::clock)
155 .def_readwrite("hdisplay", &Videomode::hdisplay)
156 .def_readwrite("hsync_start", &Videomode::hsync_start)
157 .def_readwrite("hsync_end", &Videomode::hsync_end)
158 .def_readwrite("htotal", &Videomode::htotal)
160 .def_readwrite("vdisplay", &Videomode::vdisplay)
161 .def_readwrite("vsync_start", &Videomode::vsync_start)
162 .def_readwrite("vsync_end", &Videomode::vsync_end)
163 .def_readwrite("vtotal", &Videomode::vtotal)
165 .def_readwrite("vrefresh", &Videomode::vrefresh)
167 .def_readwrite("flags", &Videomode::flags)
168 .def_readwrite("type", &Videomode::type)
170 .def("__repr__", [](const Videomode& vm) { return "<pykms.Videomode " + to_string(vm.hdisplay) + "x" + to_string(vm.vdisplay) + ">"; })
172 .def("to_blob", &Videomode::to_blob)
173 ;
175 py::class_<AtomicReq>(m, "AtomicReq")
176 .def(py::init<Card&>(),
177 py::keep_alive<1, 2>()) // Keep Card alive until this is destructed
178 .def("add", (void (AtomicReq::*)(DrmPropObject*, const string&, uint64_t)) &AtomicReq::add)
179 .def("add", (void (AtomicReq::*)(DrmPropObject*, const map<string, uint64_t>&)) &AtomicReq::add)
180 .def("test", &AtomicReq::test, py::arg("allow_modeset") = false)
181 .def("commit",
182 [](AtomicReq* self, py::object ob, bool allow)
183 {
184 // This adds a ref to the object, and must be unpacked with __ob_unpack_helper()
185 PyObject* pob = ob.ptr();
186 Py_XINCREF(pob);
187 return self->commit(pob, allow);
188 }, py::arg("data"), py::arg("allow_modeset") = false)
189 .def("commit_sync", &AtomicReq::commit_sync, py::arg("allow_modeset") = false)
190 ;
191 }