kopia lustrzana https://github.com/DL7AD/pecanpico9
95 wiersze
1.9 KiB
Python
95 wiersze
1.9 KiB
Python
#!/usr/bin/python
|
|
|
|
import serial,os,re,sys
|
|
import pygame
|
|
from pygame.locals import *
|
|
import pygame.time
|
|
from cStringIO import StringIO
|
|
|
|
send_to_server = False
|
|
SCREENX = 640
|
|
SCREENY = 480
|
|
pygame.font.init()
|
|
myfont = pygame.font.SysFont('Comic Sans MS', 20)
|
|
textsurface = myfont.render('Callsign: DL7AD2 Image ID: 07 Resolution: 640x480', False, (0, 255, 255))
|
|
pygame.init()
|
|
screen = pygame.display.set_mode((SCREENX, SCREENY))
|
|
background = pygame.Surface(screen.get_rect().size)
|
|
displaygroup = pygame.sprite.RenderUpdates()
|
|
updategroup = pygame.sprite.Group()
|
|
clock = pygame.time.Clock()
|
|
pygame.display.set_caption('PecanRXGui v.1.0.0 (Q)uit (s)end image')
|
|
|
|
|
|
try:
|
|
ser = serial.Serial(port='/dev/ttyACM1')
|
|
except:
|
|
sys.stderr.write('Error: Could not open serial port\n')
|
|
sys.exit(1)
|
|
|
|
|
|
ser.write('picture\r\n')
|
|
|
|
i=0
|
|
while True:
|
|
line = ser.readline()
|
|
m = re.search("\[(.*)\]\[(.*)\] DATA \> image\/jpeg\,(.*)", line)
|
|
try:
|
|
size = m.group(3)
|
|
except:
|
|
print line.strip()
|
|
continue
|
|
|
|
i += 1
|
|
|
|
if int(size) == 0:
|
|
ser.write('picture\r\n')
|
|
continue
|
|
|
|
imgbuf = ser.read(int(size))
|
|
|
|
|
|
for event in pygame.event.get():
|
|
if event.type == QUIT:
|
|
exit(0)
|
|
|
|
if event.type == pygame.KEYDOWN:
|
|
if event.key == pygame.K_q:
|
|
exit(0)
|
|
elif event.key == pygame.K_s:
|
|
send_to_server^=True
|
|
|
|
displaygroup.clear(screen, background)
|
|
updategroup.update()
|
|
try:
|
|
img=pygame.image.load(StringIO(imgbuf))
|
|
textsurface = myfont.render("ID: %s" % i, False, (0, 255, 255))
|
|
screen.blit(img,(0,0))
|
|
screen.blit(textsurface,(0,0))
|
|
pygame.display.flip()
|
|
pygame.display.update(displaygroup.draw(screen))
|
|
|
|
f = open('data/data'+str(i)+'.jpg','wb')
|
|
f.write(imgbuf)
|
|
f.close()
|
|
|
|
except Exception as e:
|
|
print str(e)
|
|
textsurface = myfont.render('Error %s' % (e), False, (255, 100, 100))
|
|
screen.blit(textsurface,(0,0))
|
|
pygame.display.flip()
|
|
pygame.display.update(displaygroup.draw(screen))
|
|
|
|
ser.write('picture\r\n')
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|