Spaces:
Running
Running
File size: 3,139 Bytes
b8548e4 | 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 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 | from dataclasses import dataclass
import cv2
try:
from screeninfo import get_monitors
monitor_window = get_monitors()[0]
_monitor_width, _monitor_height = monitor_window.width, monitor_window.height
except Exception:
_monitor_width, _monitor_height = 1920, 1080
from src.logger import logger
from src.utils.image import ImageUtils
@dataclass
class ImageMetrics:
# TODO: Move TEXT_SIZE, etc here and find a better class name
window_width, window_height = _monitor_width, _monitor_height
# for positioning image windows
window_x, window_y = 0, 0
reset_pos = [0, 0]
class InteractionUtils:
"""Perform primary functions such as displaying images and reading responses"""
image_metrics = ImageMetrics()
@staticmethod
def show(name, origin, pause=1, resize=False, reset_pos=None, config=None):
image_metrics = InteractionUtils.image_metrics
if origin is None:
logger.info(f"'{name}' - NoneType image to show!")
if pause:
cv2.destroyAllWindows()
return
if resize:
if not config:
raise Exception("config not provided for resizing the image to show")
img = ImageUtils.resize_util(origin, config.dimensions.display_width)
else:
img = origin
if not is_window_available(name):
cv2.namedWindow(name)
cv2.imshow(name, img)
if reset_pos:
image_metrics.window_x = reset_pos[0]
image_metrics.window_y = reset_pos[1]
cv2.moveWindow(
name,
image_metrics.window_x,
image_metrics.window_y,
)
h, w = img.shape[:2]
# Set next window position
margin = 25
w += margin
h += margin
w, h = w // 2, h // 2
if image_metrics.window_x + w > image_metrics.window_width:
image_metrics.window_x = 0
if image_metrics.window_y + h > image_metrics.window_height:
image_metrics.window_y = 0
else:
image_metrics.window_y += h
else:
image_metrics.window_x += w
if pause:
logger.info(
f"Showing '{name}'\n\t Press Q on image to continue. Press Ctrl + C in terminal to exit"
)
wait_q()
InteractionUtils.image_metrics.window_x = 0
InteractionUtils.image_metrics.window_y = 0
@dataclass
class Stats:
# TODO Fill these for stats
# Move qbox_vals here?
# badThresholds = []
# veryBadPoints = []
files_moved = 0
files_not_moved = 0
def wait_q():
esc_key = 27
while cv2.waitKey(1) & 0xFF not in [ord("q"), esc_key]:
pass
cv2.destroyAllWindows()
def is_window_available(name: str) -> bool:
"""Checks if a window is available"""
try:
cv2.getWindowProperty(name, cv2.WND_PROP_VISIBLE)
return True
except Exception as e:
print(e)
return False
|