initial site copied in

1 files changed, 28 insertions, 0 deletions

diff --git a/cli/app/utils/display_utils.py b/cli/app/utils/display_utils.py
new file mode 100644
index 0000000..7bc1782
--- /dev/null
+++ b/cli/app/utils/display_utils.py
@@ -0,0 +1,28 @@
+import sys
+import logging
+
+import cv2 as cv
+
+
+
+log = logging.getLogger('swimmer')
+
+def handle_keyboard(delay_amt=1):
+  '''Used with cv.imshow('title', image) to wait for keyboard press
+  '''
+  while True:
+    k = cv.waitKey(delay_amt) & 0xFF
+    if k == 27 or k == ord('q'):  # ESC
+      cv.destroyAllWindows()
+      sys.exit()
+    elif k == 32 or k == 83:  # 83 = right arrow
+      break
+    elif k != 255:
+      log.debug(f'k: {k}')
+
+def handle_keyboard_video(delay_amt=1):
+  key = cv.waitKey(1) & 0xFF
+  # if the `q` key was pressed, break from the loop
+  if key == ord("q"):
+    cv.destroyAllWindows()
+    sys.exit()