12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849 |
- #!/usr/bin/python3
- import pygame
- import pygame.gfxdraw
- import sys
- import signal
- import readYUV422
- import argparse
- parser = argparse.ArgumentParser()
- parser.add_argument('-p','--port',metavar='port',default='/dev/ttyACM0',type=str,help='serial port')
- parser.add_argument('-c','--colour',action='store_true',help='enable colours')
- args = parser.parse_args()
- port = args.port
- colSelect = 1 if args.colour else 0
- def signal_handler(sig,frame):
- pygame.quit()
- sys.exit(0)
-
- signal.signal(signal.SIGINT, signal_handler)
- pygame.init()
- display_width = 320
- display_height = 240
- image = pygame.display.set_mode((display_width,display_height))
- pygame.display.set_caption("Image")
- def showImage():
- index = 0
- bitmap = readYUV422.readImage(port_dev=port,width=display_width,height=display_height,colour=colSelect)
- for y in range(display_height):
- for x in range(display_width):
- pygame.gfxdraw.pixel(image, x, y, bitmap[index])
- index += 1
- pygame.display.update()
-
- while True:
- showImage()
- for event in pygame.event.get():
- if event.type == pygame.QUIT:
- pygame.quit()
- sys.exit(0)
|