summaryrefslogtreecommitdiff
path: root/megapixels/app/utils/display_utils.py
blob: 43328ae9fe0a0481c9723b65d805db0e53584ee7 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
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}')