testpat: use stopwatch

[android/external-libkmsxx.git] / libkms++util / testpat.cpp

diff --git a/libkms++util/testpat.cpp b/libkms++util/testpat.cpp
index 68f438540d563e754d9a4dcc594c8aa8e398006e..6296b1925156c363d6b1335c61d45d3a0707aafd 100644 (file)
--- a/libkms++util/testpat.cpp
+++ b/libkms++util/testpat.cpp
@@ -1,16 +1,19 @@
 
 //#define DRAW_PERF_PRINT
 
-#include <chrono>
 #include <cstring>
 #include <cassert>
+#include <thread>
 
-#include "kms++.h"
-#include "test.h"
-#include "cpuframebuffer.h"
+#include <kms++.h>
+#include <kms++util.h>
+#include <cpuframebuffer.h>
+#include "stopwatch.h"
 
 #define ARRAY_SIZE(arr) (sizeof(arr) / sizeof((arr)[0]))
 
+using namespace std;
+
 namespace kms
 {
 static void draw_rgb_pixel(IMappedFramebuffer& buf, unsigned x, unsigned y, RGB color)
@@ -209,11 +212,10 @@ static RGB get_test_pattern_pixel(IMappedFramebuffer& fb, unsigned x, unsigned y
        }
 }
 
-static void draw_test_pattern_impl(IMappedFramebuffer& fb)
+static void draw_test_pattern_part(IMappedFramebuffer& fb, unsigned start_y, unsigned end_y)
 {
        unsigned x, y;
        unsigned w = fb.width();
-       unsigned h = fb.height();
 
        switch (fb.format()) {
        case PixelFormat::XRGB8888:
@@ -221,7 +223,7 @@ static void draw_test_pattern_impl(IMappedFramebuffer& fb)
        case PixelFormat::ARGB8888:
        case PixelFormat::ABGR8888:
        case PixelFormat::RGB565:
-               for (y = 0; y < h; y++) {
+               for (y = start_y; y < end_y; y++) {
                        for (x = 0; x < w; x++) {
                                RGB pixel = get_test_pattern_pixel(fb, x, y);
                                draw_rgb_pixel(fb, x, y, pixel);
@@ -233,7 +235,7 @@ static void draw_test_pattern_impl(IMappedFramebuffer& fb)
        case PixelFormat::YUYV:
        case PixelFormat::YVYU:
        case PixelFormat::VYUY:
-               for (y = 0; y < h; y++) {
+               for (y = start_y; y < end_y; y++) {
                        for (x = 0; x < w; x += 2) {
                                RGB pixel1 = get_test_pattern_pixel(fb, x, y);
                                RGB pixel2 = get_test_pattern_pixel(fb, x + 1, y);
@@ -244,7 +246,7 @@ static void draw_test_pattern_impl(IMappedFramebuffer& fb)
 
        case PixelFormat::NV12:
        case PixelFormat::NV21:
-               for (y = 0; y < h; y += 2) {
+               for (y = start_y; y < end_y; 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);
@@ -261,21 +263,44 @@ static void draw_test_pattern_impl(IMappedFramebuffer& fb)
        }
 }
 
+static void draw_test_pattern_impl(IMappedFramebuffer& fb)
+{
+       if (fb.height() < 20) {
+               draw_test_pattern_part(fb, 0, fb.height());
+               return;
+       }
+
+       unsigned num_threads = thread::hardware_concurrency();
+       vector<thread> workers;
+
+       unsigned part = (fb.height() / num_threads) & ~1;
+
+       for (unsigned n = 0; n < num_threads; ++n) {
+               unsigned start = n * part;
+               unsigned end = start + part;
+
+               if (n == num_threads - 1)
+                       end = fb.height();
+
+               workers.push_back(thread([&fb, start, end]() { draw_test_pattern_part(fb, start, end); }));
+       }
+
+       for (thread& t : workers)
+               t.join();
+}
+
 void draw_test_pattern(IMappedFramebuffer &fb)
 {
 #ifdef DRAW_PERF_PRINT
-       using namespace std::chrono;
-
-       auto t1 = high_resolution_clock::now();
+       Stopwatch sw;
+       sw.start();
 #endif
 
        draw_test_pattern_impl(fb);
 
 #ifdef DRAW_PERF_PRINT
-       auto t2 = high_resolution_clock::now();
-       auto time_span = duration_cast<microseconds>(t2 - t1);
-
-       printf("draw took %u us\n", (unsigned)time_span.count());
+       double us = sw.elapsed_us();
+       printf("draw took %u us\n", (unsigned)us);
 #endif
 }