forked from pi3d/pi3d_demos
-
Notifications
You must be signed in to change notification settings - Fork 0
/
camera_minimal_2d.py
76 lines (65 loc) · 1.88 KB
/
camera_minimal_2d.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
#!/usr/bin/python
from __future__ import absolute_import, division, print_function, unicode_literals
""" Example showing what can be left out. ESC to quit"""
import demo
import pi3d
import numpy as np
import picamera
import picamera.array
import threading
import time
npa = None # this is the array for the camera to fill
new_pic = False # this is the flag to signal when array refilled
def get_pics():
# function to run in thread
global npa, new_pic
with picamera.PiCamera() as camera:
camera.resolution = (320, 240)
with picamera.array.PiRGBArray(camera) as output:
while True: # loop for ever
output.truncate(0)
camera.capture(output, 'rgb')
if npa is None: # do this once only
npa = np.zeros(output.array.shape[:2] + (4,), dtype=np.uint8)
npa[:,:,3] = 255 # fill alpha value
npa[:,:,0:3] = output.array # copy in rgb bytes
new_pic = True
time.sleep(0.05)
t = threading.Thread(target=get_pics) # set up and start capture thread
t.daemon = True
t.start()
while not new_pic: # wait for array to be filled first time
time.sleep(0.1)
########################################################################
DISPLAY = pi3d.Display.create(x=150, y=150, frames_per_second=30)
shader = pi3d.Shader("uv_flat")
CAMERA = pi3d.Camera(is_3d=False)
tex = pi3d.Texture(npa)
sprite = pi3d.Sprite(w=tex.ix, h=tex.iy, z=5.0)
sprite.set_draw_details(shader, [tex])
mykeys = pi3d.Keyboard()
xloc = 100.0
dx = 6.1
yloc = 100.0
dy = 1.13
while DISPLAY.loop_running():
if new_pic:
tex.update_ndarray(npa)
new_pic = False
sprite.draw()
sprite.rotateIncZ(1)
sprite.position(xloc, yloc, 5.0)
if xloc > 300.0:
dx = -2.1
elif xloc < -300.0:
dx = 2.1
if yloc > 300.0:
dy = -1.13
elif yloc < -300.0:
dy = 1.13
xloc += dx
yloc += dy
if mykeys.read() == 27:
mykeys.close()
DISPLAY.destroy()
break