testpat: add NV12/NV21 support

[android/external-libkmsxx.git] / libkmstest / testpat.cpp

diff --git a/libkmstest/testpat.cpp b/libkmstest/testpat.cpp
index b0b67a49a445ef8c28ee92b957d268088cb03783..f9cf907fd392349461b770f6eef6fb4edde1183b 100644 (file)
--- a/libkmstest/testpat.cpp
+++ b/libkmstest/testpat.cpp
@@ -87,6 +87,48 @@ static void draw_yuv422_macropixel(DumbFramebuffer& buf, unsigned x, unsigned y,
        }
 }
 
+static void draw_yuv420_macropixel(DumbFramebuffer& buf, unsigned x, unsigned y,
+                                  YUV yuv1, YUV yuv2, YUV yuv3, YUV yuv4)
+{
+       ASSERT((x & 1) == 0);
+       ASSERT((y & 1) == 0);
+
+       uint8_t *py1 = (uint8_t*)(buf.map(0) + buf.stride(0) * (y + 0) + x);
+       uint8_t *py2 = (uint8_t*)(buf.map(0) + buf.stride(0) * (y + 1) + x);
+
+       uint8_t *puv = (uint8_t*)(buf.map(1) + buf.stride(1) * (y / 2) + x);
+
+       uint8_t y0 = yuv1.y;
+       uint8_t y1 = yuv2.y;
+       uint8_t y2 = yuv3.y;
+       uint8_t y3 = yuv4.y;
+       uint8_t u = (yuv1.u + yuv2.u + yuv3.u + yuv4.u) / 4;
+       uint8_t v = (yuv1.v + yuv2.v + yuv3.v + yuv4.v) / 4;
+
+       switch (buf.format()) {
+       case PixelFormat::NV12:
+               py1[0] = y0;
+               py1[1] = y1;
+               py2[0] = y2;
+               py2[1] = y3;
+               puv[0] = u;
+               puv[1] = v;
+               break;
+
+       case PixelFormat::NV21:
+               py1[0] = y0;
+               py1[1] = y1;
+               py2[0] = y2;
+               py2[1] = y3;
+               puv[0] = v;
+               puv[1] = u;
+               break;
+
+       default:
+               throw std::invalid_argument("invalid pixelformat");
+       }
+}
+
 static RGB get_test_pattern_pixel(DumbFramebuffer& fb, unsigned x, unsigned y)
 {
        unsigned w = fb.width();
@@ -184,6 +226,21 @@ static void draw_test_pattern_impl(DumbFramebuffer& fb)
                        }
                }
                break;
+
+       case PixelFormat::NV12:
+       case PixelFormat::NV21:
+               for (y = 0; y < h; y += 2) {
+                       for (x = 0; x < w; x += 2) {
+                               RGB pixel00 = get_test_pattern_pixel(fb, x, y);
+                               RGB pixel10 = get_test_pattern_pixel(fb, x + 1, y);
+                               RGB pixel01 = get_test_pattern_pixel(fb, x, y + 1);
+                               RGB pixel11 = get_test_pattern_pixel(fb, x + 1, y + 1);
+                               draw_yuv420_macropixel(fb, x, y,
+                                                      pixel00.yuv(), pixel10.yuv(),
+                                                      pixel01.yuv(), pixel11.yuv());
+                       }
+               }
+               break;
        default:
                throw std::invalid_argument("unknown pixelformat");
        }