From 3d6262d56406e17163deddafbe2b928168784594 Mon Sep 17 00:00:00 2001 From: Laurent Pinchart Date: Mon, 17 Jun 2019 02:34:29 +0300 Subject: utils: Add a dump_framebuffer() method Add a new method to write the contents of a framebuffer to a file descriptor. This can be used to capture frames from writeback connectors. Signed-off-by: Laurent Pinchart --- kms++util/inc/kms++util/kms++util.h | 1 + kms++util/src/drawing.cpp | 7 +++++++ py/pykms/pykmsutil.cpp | 3 +++ 3 files changed, 11 insertions(+) diff --git a/kms++util/inc/kms++util/kms++util.h b/kms++util/inc/kms++util/kms++util.h index 52b6ce9..0d14c99 100644 --- a/kms++util/inc/kms++util/kms++util.h +++ b/kms++util/inc/kms++util/kms++util.h @@ -29,6 +29,7 @@ void draw_text(IFramebuffer& buf, uint32_t x, uint32_t y, const std::string& str void draw_color_bar(IFramebuffer& buf, int old_xpos, int xpos, int width); void draw_test_pattern(IFramebuffer& fb, YUVType yuvt = YUVType::BT601_Lim); +void dump_framebuffer(IFramebuffer& fb, int fd); } // namespace kms #define ARRAY_SIZE(arr) (sizeof(arr) / sizeof((arr)[0])) diff --git a/kms++util/src/drawing.cpp b/kms++util/src/drawing.cpp index 79e0d90..8d82ca4 100644 --- a/kms++util/src/drawing.cpp +++ b/kms++util/src/drawing.cpp @@ -1,5 +1,6 @@ #include +#include #include #include @@ -554,4 +555,10 @@ void draw_text(IFramebuffer& buf, uint32_t x, uint32_t y, const string& str, RGB draw_char(buf, (x + 8 * i), y, str[i], color); } +void dump_framebuffer(IFramebuffer& fb, int fd) +{ + for (unsigned int i = 0; i < fb.num_planes(); ++i) + ::write(fd, fb.map(i), fb.size(i)); +} + } // namespace kms diff --git a/py/pykms/pykmsutil.cpp b/py/pykms/pykmsutil.cpp index 666cbdc..6cf79ee 100644 --- a/py/pykms/pykmsutil.cpp +++ b/py/pykms/pykmsutil.cpp @@ -59,4 +59,7 @@ void init_pykmstest(py::module& m) draw_circle(fb, xCenter, yCenter, radius, color); }); m.def("draw_text", [](Framebuffer& fb, uint32_t x, uint32_t y, const string& str, RGB color) { draw_text(fb, x, y, str, color); }); + m.def("dump_framebuffer", [](Framebuffer& fb, int fd) { + dump_framebuffer(fb, fd); + }); } -- cgit v1.2.3