summaryrefslogtreecommitdiff
path: root/cli/app/utils/display_utils.py
blob: 8e265ae78b872acb8a9de00e6859b81c7de80652 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
import sys

import cv2 as cv

from app.utils.logger_utils import Logger


log = Logger.getLogger()

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()