From 8a09c46c38b2c7622b75a11a4bf51a2631d1b278 Mon Sep 17 00:00:00 2001 From: George Hotz Date: Tue, 20 Oct 2020 16:07:05 -0700 Subject: [PATCH] support for halve in Window old-commit-hash: bb2e69faf339aaddf04e49d2dfd71367e72957de --- common/window.py | 8 +++++++- 1 file changed, 7 insertions(+), 1 deletion(-) diff --git a/common/window.py b/common/window.py index 62f8cb9e0b..983dc43984 100644 --- a/common/window.py +++ b/common/window.py @@ -3,14 +3,17 @@ import pygame # pylint: disable=import-error import cv2 # pylint: disable=import-error class Window(): - def __init__(self, w, h, caption="window", double=False): + def __init__(self, w, h, caption="window", double=False, halve=False): self.w = w self.h = h pygame.display.init() pygame.display.set_caption(caption) self.double = double + self.halve = halve if self.double: self.screen = pygame.display.set_mode((w*2, h*2)) + elif self.halve: + self.screen = pygame.display.set_mode((w//2, h//2)) else: self.screen = pygame.display.set_mode((w, h)) @@ -19,6 +22,9 @@ class Window(): if self.double: out2 = cv2.resize(out, (self.w*2, self.h*2)) pygame.surfarray.blit_array(self.screen, out2.swapaxes(0, 1)) + elif self.halve: + out2 = cv2.resize(out, (self.w//2, self.h//2)) + pygame.surfarray.blit_array(self.screen, out2.swapaxes(0, 1)) else: pygame.surfarray.blit_array(self.screen, out.swapaxes(0, 1)) pygame.display.flip()