From d8e826e37903d79834f7f2680d5bda6e4083f7d1 Mon Sep 17 00:00:00 2001
From: KIU Shueng Chuan <nixchuan@gmail.com>
Date: Mon, 22 Feb 2021 16:13:53 +0800
Subject: [PATCH] disable opengl vsync for benchmarking

---
 examples/VideoSpeedTest.py | 16 ++++++++++++----
 1 file changed, 12 insertions(+), 4 deletions(-)

diff --git a/examples/VideoSpeedTest.py b/examples/VideoSpeedTest.py
index a6c20cb7..c8c05d8e 100644
--- a/examples/VideoSpeedTest.py
+++ b/examples/VideoSpeedTest.py
@@ -29,6 +29,11 @@ except ImportError:
     cp = None
     _has_cupy = False
 
+try:
+    from pyqtgraph.widgets.RawImageWidget import RawImageGLWidget
+except ImportError:
+    RawImageGLWidget = None
+
 parser = argparse.ArgumentParser(description="Benchmark for testing video performance")
 parser.add_argument('--cuda', default=False, action='store_true', help="Use CUDA to process on the GPU", dest="cuda")
 parser.add_argument('--dtype', default='uint8', choices=['uint8', 'uint16', 'float'], help="Image dtype (uint8, uint16, or float)")
@@ -40,6 +45,12 @@ parser.add_argument('--lut-alpha', default=False, action='store_true', help="Use
 parser.add_argument('--size', default='512x512', type=lambda s: tuple([int(x) for x in s.split('x')]), help="WxH image dimensions default='512x512'")
 args = parser.parse_args(sys.argv[1:])
 
+if RawImageGLWidget is not None:
+    # don't limit frame rate to vsync
+    sfmt = QtGui.QSurfaceFormat()
+    sfmt.setSwapInterval(0)
+    QtGui.QSurfaceFormat.setDefaultFormat(sfmt)
+
 app = pg.mkQApp("Video Speed Test Example")
 
 win = QtGui.QMainWindow()
@@ -48,10 +59,7 @@ ui = ui_template.Ui_MainWindow()
 ui.setupUi(win)
 win.show()
 
-try:
-    from pyqtgraph.widgets.RawImageWidget import RawImageGLWidget
-except ImportError:
-    RawImageGLWidget = None
+if RawImageGLWidget is None:
     ui.rawGLRadio.setEnabled(False)
     ui.rawGLRadio.setText(ui.rawGLRadio.text() + " (OpenGL not available)")
 else: