Added a new selection GUI and a Result viewer

pyidi/GUIs/__init__.py

@@ -0,0 +1,29 @@
+import typing
+
+try:
+    import PyQt6
+
+    HAS_PYQT6 = True
+except ImportError:
+    HAS_PYQT6 = False
+
+if HAS_PYQT6 or typing.TYPE_CHECKING:
+    from .subset_selection import SelectionGUI
+    from .result_viewer import ResultViewer
+else:
+    class SelectionGUI:
+        def __init__(self, video):
+            pass
+
+        def show_displacement(self, data):
+            raise RuntimeError("SelectionGUI requires PyQt6: pip install pyidi[qt]")
+
+    class ResultViewer:
+        def __init__(self):
+            pass
+
+        def show_displacement(self, data):
+            raise RuntimeError("ResultViewer requires PyQt6: pip install pyidi[qt]")
+
+from .selection import SubsetSelection
+from .gui import GUI

pyidi/gui.py → pyidi/GUIs/gui.py

@@ -6,10 +6,10 @@
 import warnings
 warnings.simplefilter("default")
 
-from . import tools
+from .. import tools
 from . import selection
-from .methods import SimplifiedOpticalFlow
-from .methods import LucasKanade
+from ..methods import SimplifiedOpticalFlow
+from ..methods import LucasKanade
 
 NO_METHOD = '---'
 add_vertical_stretch = True