Optimize widget redraw

Signed-off-by: Stanislaw Halik <sthalik@misaki.pl>

1 files changed, 20 insertions, 6 deletions

diff --git a/ftnoir_tracker_aruco/ar_video_widget.cpp b/ftnoir_tracker_aruco/ar_video_widget.cpp
index 6a4572a0..b727679b 100644
--- a/ftnoir_tracker_aruco/ar_video_widget.cpp
+++ b/ftnoir_tracker_aruco/ar_video_widget.cpp
@@ -14,17 +14,31 @@ using namespace std;
 void ArucoVideoWidget::update_image(unsigned char *frame, int width, int height)
 {
     QMutexLocker foo(&mtx);
+    memcpy(fb, frame, width * height * 3);
+    this->width = width;
+    this->height = height;
+}
+
+void ArucoVideoWidget::update_and_repaint()
+{
+    QMutexLocker foo(&mtx);
+    if (width*height <= 0)
+        return;
     QImage qframe = QImage(width, height, QImage::Format_RGB888);
     uchar* data = qframe.bits();
     const int pitch = qframe.bytesPerLine();
     for (int y = 0; y < height; y++)
+    {
+        const int part = y*width;
         for (int x = 0; x < width; x++)
         {
-            const int pos = 3 * (y*width + x);
-            data[y * pitch + x * 3 + 0] = frame[pos + 2];
-            data[y * pitch + x * 3 + 1] = frame[pos + 1];
-            data[y * pitch + x * 3 + 2] = frame[pos + 0];
+            const int pos = 3 * (part + x);
+            data[y * pitch + x * 3 + 0] = fb[pos + 2];
+            data[y * pitch + x * 3 + 1] = fb[pos + 1];
+            data[y * pitch + x * 3 + 2] = fb[pos + 0];
         }
-    qframe = qframe.scaled(size(), Qt::IgnoreAspectRatio, Qt::FastTransformation);
-    pixmap = QPixmap::fromImage(qframe);
+    }
+    auto qframe2 = qframe.scaled(size(), Qt::IgnoreAspectRatio, Qt::FastTransformation);
+    pixmap = QPixmap::fromImage(qframe2);
+    update();
 }