-
Notifications
You must be signed in to change notification settings - Fork 0
/
cam.py
96 lines (87 loc) · 3.35 KB
/
cam.py
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
import datetime
import atexit
import io
import os
import picamera
import pygame,pigame
import menu
import subprocess, sys
from time import sleep
import yuv2rgb
import RPi.GPIO as GPIO
from pygame.locals import *
def deploy():
# Where your .py file is located
current_path = os.path.dirname(__file__)
GPIO.setmode(GPIO.BCM)
GPIO.setup(17, GPIO.IN, pull_up_down=GPIO.PUD_UP)
GPIO.setup(22, GPIO.IN, pull_up_down=GPIO.PUD_UP)
GPIO.setup(23, GPIO.IN, pull_up_down=GPIO.PUD_UP)
GPIO.setup(27, GPIO.IN, pull_up_down=GPIO.PUD_UP)
screenMode = 3
sizeMode = 0
sizeData = [[(1440, 1080), (320, 240), (0.2222, 0.2222, 0.5556, 0.5556)]]
os.putenv('SDL_VIDEODRIVER', 'fbcon')
os.putenv('SDL_FBDEV' , '/dev/fb1')
rgb = bytearray(320 * 240 * 3)
yuv = bytearray(320 * 240 * 3 / 2)
pygame.init()
pitft = pigame.PiTft()
pygame.mouse.set_visible(False)
screen = pygame.display.set_mode((0,0), pygame.FULLSCREEN)
camera = picamera.PiCamera()
atexit.register(camera.close)
camera.resolution = sizeData[sizeMode][1]
camera.crop = (0.0, 0.0, 1.0, 1.0)
camera.rotation = 270
try:
while(True):
pitft.update()
event = pygame.event.poll()
stream = io.BytesIO()
camera.capture(stream, use_video_port=True, format='raw')
stream.seek(0)
stream.readinto(yuv)
stream.close()
yuv2rgb.convert(yuv, rgb, sizeData[sizeMode][1][0],
sizeData[sizeMode][1][1])
img = pygame.image.frombuffer(rgb[0:
(sizeData[sizeMode][1][0] * sizeData[sizeMode][1][1] * 3)],
sizeData[sizeMode][1], 'RGB')
if img:
screen.blit(img,
((320 - img.get_width() ) / 2,
(240 - img.get_height()) / 2))
for event in pygame.event.get():
if(event.type is MOUSEBUTTONDOWN):
x,y = pygame.mouse.get_pos()
elif(event.type is MOUSEBUTTONUP):
x,y = pygame.mouse.get_pos()
if y > 200:
if x < 50:
print("Cross")
subprocess.Popen(["python", "/home/pi/camera23/menu.py"])
exit()
#menu.deploy()
else:
print("Photo")
pygame.draw.rect(screen,(255,255,255), (0,0,320,240))
pygame.display.update()
now = datetime.datetime.now()
image_date = now.strftime("pics/%Y-%m-%d_%H%M%S.jpg")
image_path = os.path.join(current_path, image_date)
camera.capture(image_path)
img = pygame.image.load(image_path)
screen.blit(img, (0,0))
pygame.display.update()
sleep(1)
img_cross = pygame.image.load('resources/cross.png')
screen.blit(img_cross, (280,5))
pygame.display.update()
sleep(0.1)
except KeyboardInterrupt:
pass
finally:
del(pitft)
if __name__ == '__main__':
deploy()