diff --git a/py/pykms/pykmsutil.cpp b/py/pykms/pykmsutil.cpp
index cb09dea0ab96285d72c31a1dbb49c1470e208f47..a5a6041a22ebf40c38ce8ee1522973a8ffe7d942 100644 (file)
--- a/py/pykms/pykmsutil.cpp
+++ b/py/pykms/pykmsutil.cpp
py::arg("crtc"),
py::arg("format") = PixelFormat::Undefined)
;
+ py::enum_<YUVType>(m, "YUVType")
+ .value("BT601_Lim", YUVType::BT601_Lim)
+ .value("BT601_Full", YUVType::BT601_Full)
+ .value("BT709_Lim", YUVType::BT709_Lim)
+ .value("BT709_Full", YUVType::BT709_Full)
+ ;
// Use lambdas to handle IMappedFramebuffer
- m.def("draw_test_pattern", [](MappedFramebuffer& fb) { draw_test_pattern(fb); } );
+ m.def("draw_test_pattern", [](MappedFramebuffer& fb, YUVType yuvt) { draw_test_pattern(fb, yuvt); },
+ py::arg("fb"),
+ py::arg("yuvt") = YUVType::BT601_Lim);
m.def("draw_color_bar", [](MappedFramebuffer& fb, int old_xpos, int xpos, int width) {
draw_color_bar(fb, old_xpos, xpos, width);
} );