diff options
Diffstat (limited to 'libkmstest')
-rw-r--r-- | libkmstest/colorbar.cpp | 129 | ||||
-rw-r--r-- | libkmstest/colorbar.h | 6 | ||||
-rw-r--r-- | libkmstest/test.h | 1 |
3 files changed, 136 insertions, 0 deletions
diff --git a/libkmstest/colorbar.cpp b/libkmstest/colorbar.cpp new file mode 100644 index 0000000..bc9cdca --- /dev/null +++ b/libkmstest/colorbar.cpp @@ -0,0 +1,129 @@ +#include <cstdint> +#include <drm_fourcc.h> + +#include <kms++.h> + +#include "test.h" + +namespace kms +{ +static const RGB colors32[] = { + RGB(255, 255, 255), + RGB(255, 0, 0), + RGB(255, 255, 255), + RGB(0, 255, 0), + RGB(255, 255, 255), + RGB(0, 0, 255), + RGB(255, 255, 255), + RGB(200, 200, 200), + RGB(255, 255, 255), + RGB(100, 100, 100), + RGB(255, 255, 255), + RGB(50, 50, 50), +}; + +static const uint16_t colors16[] = { + colors32[0].rgb565(), + colors32[1].rgb565(), + colors32[2].rgb565(), + colors32[3].rgb565(), + colors32[4].rgb565(), + colors32[5].rgb565(), + colors32[6].rgb565(), + colors32[7].rgb565(), + colors32[8].rgb565(), + colors32[9].rgb565(), + colors32[10].rgb565(), + colors32[11].rgb565(), +}; + +static void drm_draw_color_bar_rgb888(DumbFramebuffer& buf, int old_xpos, int xpos, int width) +{ + for (unsigned y = 0; y < buf.height(); ++y) { + RGB bcol = colors32[y * ARRAY_SIZE(colors32) / buf.height()]; + uint32_t *line = (uint32_t*)(buf.map(0) + buf.stride(0) * y); + + if (old_xpos >= 0) { + for (int x = old_xpos; x < old_xpos + width; ++x) + line[x] = 0; + } + + for (int x = xpos; x < xpos + width; ++x) + line[x] = bcol.raw; + } +} + +static void drm_draw_color_bar_rgb565(DumbFramebuffer& buf, int old_xpos, int xpos, int width) +{ + static_assert(ARRAY_SIZE(colors32) == ARRAY_SIZE(colors16), "bad colors arrays"); + + for (unsigned y = 0; y < buf.height(); ++y) { + uint16_t bcol = colors16[y * ARRAY_SIZE(colors16) / buf.height()]; + uint16_t *line = (uint16_t*)(buf.map(0) + buf.stride(0) * y); + + if (old_xpos >= 0) { + for (int x = old_xpos; x < old_xpos + width; ++x) + line[x] = 0; + } + + for (int x = xpos; x < xpos + width; ++x) + line[x] = bcol; + } +} + +static void drm_draw_color_bar_semiplanar_yuv(DumbFramebuffer& buf, int old_xpos, int xpos, int width) +{ + const uint8_t colors[] = { + 0xff, + 0x00, + 0xff, + 0x20, + 0xff, + 0x40, + 0xff, + 0x80, + 0xff, + }; + + for (unsigned y = 0; y < buf.height(); ++y) { + unsigned int bcol = colors[y * ARRAY_SIZE(colors) / buf.height()]; + uint8_t *line = (uint8_t*)(buf.map(0) + buf.stride(0) * y); + + if (old_xpos >= 0) { + for (int x = old_xpos; x < old_xpos + width; ++x) + line[x] = 0; + } + + for (int x = xpos; x < xpos + width; ++x) + line[x] = bcol; + } +} + +void draw_color_bar(DumbFramebuffer& buf, int old_xpos, int xpos, int width) +{ + switch (buf.format()) { + case DRM_FORMAT_NV12: + case DRM_FORMAT_NV21: + // XXX not right but gets something on the screen + drm_draw_color_bar_semiplanar_yuv(buf, old_xpos, xpos, width); + break; + + case DRM_FORMAT_YUYV: + case DRM_FORMAT_UYVY: + // XXX not right but gets something on the screen + drm_draw_color_bar_rgb565(buf, old_xpos, xpos, width); + break; + + case DRM_FORMAT_RGB565: + drm_draw_color_bar_rgb565(buf, old_xpos, xpos, width); + break; + + case DRM_FORMAT_XRGB8888: + drm_draw_color_bar_rgb888(buf, old_xpos, xpos, width); + break; + + default: + ASSERT(false); + } +} +} diff --git a/libkmstest/colorbar.h b/libkmstest/colorbar.h new file mode 100644 index 0000000..a47d64d --- /dev/null +++ b/libkmstest/colorbar.h @@ -0,0 +1,6 @@ +#pragma once + +namespace kms +{ +void draw_color_bar(kms::DumbFramebuffer& buf, int old_xpos, int xpos, int width); +} diff --git a/libkmstest/test.h b/libkmstest/test.h index f4cec59..fbc4727 100644 --- a/libkmstest/test.h +++ b/libkmstest/test.h @@ -3,6 +3,7 @@ #include "color.h" #include "conv.h" #include "testpat.h" +#include "colorbar.h" #define ARRAY_SIZE(arr) (sizeof(arr) / sizeof((arr)[0])) |