kms++util/src/testpat.cpp

   1
   2 //#define DRAW_PERF_PRINT
   3
   4 #include <cstring>
   5 #include <cassert>
   6 #include <thread>
   7
   8 #include <kms++/kms++.h>
   9 #include <kms++util/kms++util.h>
  10
  11 using namespace std;
  12
  13 namespace kms
  14 {
  15
  16 static RGB get_test_pattern_pixel(IMappedFramebuffer& fb, unsigned x, unsigned y)
  17 {
  18         const unsigned w = fb.width();
  19         const unsigned h = fb.height();
  20
  21         const unsigned mw = 20;
  22
  23         const unsigned xm1 = mw;
  24         const unsigned xm2 = w - mw - 1;
  25         const unsigned ym1 = mw;
  26         const unsigned ym2 = h - mw - 1;
  27
  28         // white margin lines
  29         if (x == xm1 || x == xm2 || y == ym1 || y == ym2)
  30                 return RGB(255, 255, 255);
  31         // white box in top left corner
  32         else if (x < xm1 && y < ym1)
  33                 return RGB(255, 255, 255);
  34         // white box outlines to corners
  35         else if ((x == 0 || x == w - 1) && (y < ym1 || y > ym2))
  36                 return RGB(255, 255, 255);
  37         // white box outlines to corners
  38         else if ((y == 0 || y == h - 1) && (x < xm1 || x > xm2))
  39                 return RGB(255, 255, 255);
  40         // blue bar on the left
  41         else if (x < xm1 && (y > ym1 && y < ym2))
  42                 return RGB(0, 0, 255);
  43         // blue bar on the top
  44         else if (y < ym1 && (x > xm1 && x < xm2))
  45                 return RGB(0, 0, 255);
  46         // red bar on the right
  47         else if (x > xm2 && (y > ym1 && y < ym2))
  48                 return RGB(255, 0, 0);
  49         // red bar on the bottom
  50         else if (y > ym2 && (x > xm1 && x < xm2))
  51                 return RGB(255, 0, 0);
  52         // inside the margins
  53         else if (x > xm1 && x < xm2 && y > ym1 && y < ym2) {
  54                 // diagonal line
  55                 if (x == y || w - x == h - y)
  56                         return RGB(255, 255, 255);
  57                 // diagonal line
  58                 else if (w - x == y || x == h - y)
  59                         return RGB(255, 255, 255);
  60                 else {
  61                         int t = (x - xm1 - 1) * 8 / (xm2 - xm1 - 1);
  62                         unsigned r = 0, g = 0, b = 0;
  63
  64                         unsigned c = (y - ym1 - 1) % 256;
  65
  66                         switch (t) {
  67                         case 0:
  68                                 r = c;
  69                                 break;
  70                         case 1:
  71                                 g = c;
  72                                 break;
  73                         case 2:
  74                                 b = c;
  75                                 break;
  76                         case 3:
  77                                 g = b = c;
  78                                 break;
  79                         case 4:
  80                                 r = b = c;
  81                                 break;
  82                         case 5:
  83                                 r = g = c;
  84                                 break;
  85                         case 6:
  86                                 r = g = b = c;
  87                                 break;
  88                         case 7:
  89                                 break;
  90                         }
  91
  92                         return RGB(r, g, b);
  93                 }
  94         } else {
  95                 // black corners
  96                 return RGB(0, 0, 0);
  97         }
  98 }
  99
 100 static void draw_test_pattern_part(IMappedFramebuffer& fb, unsigned start_y, unsigned end_y, YUVType yuvt)
 101 {
 102         unsigned x, y;
 103         unsigned w = fb.width();
 104
 105         switch (fb.format()) {
 106         case PixelFormat::XRGB8888:
 107         case PixelFormat::XBGR8888:
 108         case PixelFormat::ARGB8888:
 109         case PixelFormat::ABGR8888:
 110         case PixelFormat::RGB888:
 111         case PixelFormat::BGR888:
 112         case PixelFormat::RGB565:
 113         case PixelFormat::BGR565:
 114                 for (y = start_y; y < end_y; y++) {
 115                         for (x = 0; x < w; x++) {
 116                                 RGB pixel = get_test_pattern_pixel(fb, x, y);
 117                                 draw_rgb_pixel(fb, x, y, pixel);
 118                         }
 119                 }
 120                 break;
 121
 122         case PixelFormat::UYVY:
 123         case PixelFormat::YUYV:
 124         case PixelFormat::YVYU:
 125         case PixelFormat::VYUY:
 126                 for (y = start_y; y < end_y; y++) {
 127                         for (x = 0; x < w; x += 2) {
 128                                 RGB pixel1 = get_test_pattern_pixel(fb, x, y);
 129                                 RGB pixel2 = get_test_pattern_pixel(fb, x + 1, y);
 130                                 draw_yuv422_macropixel(fb, x, y, pixel1.yuv(yuvt), pixel2.yuv(yuvt));
 131                         }
 132                 }
 133                 break;
 134
 135         case PixelFormat::NV12:
 136         case PixelFormat::NV21:
 137                 for (y = start_y; y < end_y; y += 2) {
 138                         for (x = 0; x < w; x += 2) {
 139                                 RGB pixel00 = get_test_pattern_pixel(fb, x, y);
 140                                 RGB pixel10 = get_test_pattern_pixel(fb, x + 1, y);
 141                                 RGB pixel01 = get_test_pattern_pixel(fb, x, y + 1);
 142                                 RGB pixel11 = get_test_pattern_pixel(fb, x + 1, y + 1);
 143                                 draw_yuv420_macropixel(fb, x, y,
 144                                                        pixel00.yuv(yuvt), pixel10.yuv(yuvt),
 145                                                        pixel01.yuv(yuvt), pixel11.yuv(yuvt));
 146                         }
 147                 }
 148                 break;
 149         default:
 150                 throw std::invalid_argument("unknown pixelformat");
 151         }
 152 }
 153
 154 static void draw_test_pattern_impl(IMappedFramebuffer& fb, YUVType yuvt)
 155 {
 156         if (fb.height() < 20) {
 157                 draw_test_pattern_part(fb, 0, fb.height(), yuvt);
 158                 return;
 159         }
 160
 161         // Create the mmaps before starting the threads
 162         for (unsigned i = 0; i < fb.num_planes(); ++i)
 163                 fb.map(0);
 164
 165         unsigned num_threads = thread::hardware_concurrency();
 166         vector<thread> workers;
 167
 168         unsigned part = (fb.height() / num_threads) & ~1;
 169
 170         for (unsigned n = 0; n < num_threads; ++n) {
 171                 unsigned start = n * part;
 172                 unsigned end = start + part;
 173
 174                 if (n == num_threads - 1)
 175                         end = fb.height();
 176
 177                 workers.push_back(thread([&fb, start, end, yuvt]() { draw_test_pattern_part(fb, start, end, yuvt); }));
 178         }
 179
 180         for (thread& t : workers)
 181                 t.join();
 182 }
 183
 184 void draw_test_pattern(IMappedFramebuffer &fb, YUVType yuvt)
 185 {
 186 #ifdef DRAW_PERF_PRINT
 187         Stopwatch sw;
 188         sw.start();
 189 #endif
 190
 191         draw_test_pattern_impl(fb, yuvt);
 192
 193 #ifdef DRAW_PERF_PRINT
 194         double us = sw.elapsed_us();
 195         printf("draw took %u us\n", (unsigned)us);
 196 #endif
 197 }
 198
 199 }