update

common/window.py

@@ -1,5 +1,6 @@
 import sys
 import pygame
+import cv2
 
 class Window():
   def __init__(self, w, h, caption="window", double=False):
@@ -9,20 +10,20 @@ class Window():
     pygame.display.set_caption(caption)
     self.double = double
     if self.double:
-      self.screen = pygame.display.set_mode((w*2,h*2), pygame.DOUBLEBUF)
+      self.screen = pygame.display.set_mode((w*2,h*2))
     else:
-      self.screen = pygame.display.set_mode((w,h), pygame.DOUBLEBUF)
-    self.camera_surface = pygame.surface.Surface((w,h), 0, 24).convert()
+      self.screen = pygame.display.set_mode((w,h))
 
   def draw(self, out):
-    pygame.surfarray.blit_array(self.camera_surface, out.swapaxes(0,1))
+    pygame.event.pump()
     if self.double:
-      camera_surface_2x = pygame.transform.scale2x(self.camera_surface)
-      self.screen.blit(camera_surface_2x, (0, 0))
+      out2 = cv2.resize(out, (self.w*2, self.h*2))
+      pygame.surfarray.blit_array(self.screen, out2.swapaxes(0,1))
     else:
-      self.screen.blit(self.camera_surface, (0, 0))
+      pygame.surfarray.blit_array(self.screen, out.swapaxes(0,1))
     pygame.display.flip()
 
+
   def getkey(self):
     while 1:
       event = pygame.event.wait()
@@ -40,7 +41,7 @@ class Window():
 
 if __name__ == "__main__":
   import numpy as np
-  win = Window(200, 200)
+  win = Window(200, 200, double=True)
   img = np.zeros((200,200,3), np.uint8)
   while 1:
     print("draw")