1 #include <pybind11/pybind11.h>
2 #include <pybind11/stl.h>
3 #include <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("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 ;
26 py::class_<DrmObject, DrmObject*>(m, "DrmObject")
27 .def_property_readonly("id", &DrmObject::id)
28 .def_property_readonly("card", &DrmObject::card)
29 ;
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 ;
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 ;
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 ;
53 py::class_<Encoder, Encoder*>(m, "Encoder", py::base<DrmPropObject>())
54 ;
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 ;
62 py::enum_<PlaneType>(m, "PlaneType")
63 .value("Overlay", PlaneType::Overlay)
64 .value("Primary", PlaneType::Primary)
65 .value("Cursor", PlaneType::Cursor)
66 ;
68 py::class_<Property, Property*>(m, "Property", py::base<DrmObject>())
69 .def_property_readonly("name", &Property::name)
70 ;
72 py::class_<Framebuffer>(m, "Framebuffer", py::base<DrmObject>())
73 ;
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 ;
82 py::class_<Videomode>(m, "Videomode")
83 .def(py::init<>())
85 .def_readwrite("name", &Videomode::name)
87 .def_readwrite("clock", &Videomode::clock)
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)
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)
99 .def_readwrite("vrefresh", &Videomode::vrefresh)
101 .def_readwrite("flags", &Videomode::flags)
102 .def_readwrite("type", &Videomode::type)
103 ;
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 }