2018-03-24 03:05:46 +01:00
|
|
|
import logging
|
|
|
|
import socket
|
|
|
|
import time
|
|
|
|
import picamera
|
|
|
|
|
|
|
|
from platypush.backend import Backend
|
|
|
|
|
|
|
|
class CameraPiBackend(Backend):
|
|
|
|
def __init__(self, listen_port, x_resolution=640, y_resolution=480,
|
2018-03-24 14:43:18 +01:00
|
|
|
framerate=24, hflip=False, vflip=False,
|
|
|
|
sharpness=0, contrast=0, brightness=50,
|
|
|
|
video_stabilization=False, ISO=0, exposure_compensation=0,
|
|
|
|
exposure_mode='auto', meter_mode='average', awb_mode='auto',
|
|
|
|
image_effect='none', color_effects=None, rotation=0,
|
|
|
|
crop=(0.0, 0.0, 1.0, 1.0), **kwargs):
|
|
|
|
""" See https://www.raspberrypi.org/documentation/usage/camera/python/README.md
|
|
|
|
for a detailed reference about the Pi camera options """
|
|
|
|
|
2018-03-24 03:05:46 +01:00
|
|
|
super().__init__(**kwargs)
|
|
|
|
|
|
|
|
self.listen_port = listen_port
|
|
|
|
self.server_socket = socket.socket()
|
|
|
|
self.server_socket.bind(('0.0.0.0', self.listen_port))
|
|
|
|
self.server_socket.listen(0)
|
|
|
|
|
|
|
|
self.camera = picamera.PiCamera()
|
2018-03-24 14:43:18 +01:00
|
|
|
self.camera.resolution = (x_resolution, y_resolution)
|
2018-03-24 03:05:46 +01:00
|
|
|
self.camera.framerate = framerate
|
2018-03-24 14:43:18 +01:00
|
|
|
self.camera.hflip = hflip
|
|
|
|
self.camera.vflip = vflip
|
|
|
|
self.camera.sharpness = sharpness
|
|
|
|
self.camera.contrast = contrast
|
|
|
|
self.camera.brightness = brightness
|
|
|
|
self.camera.video_stabilization = video_stabilization
|
|
|
|
self.camera.ISO = ISO
|
|
|
|
self.camera.exposure_compensation = exposure_compensation
|
|
|
|
self.camera.exposure_mode = exposure_mode
|
|
|
|
self.camera.meter_mode = meter_mode
|
|
|
|
self.camera.awb_mode = awb_mode
|
|
|
|
self.camera.image_effect = image_effect
|
|
|
|
self.camera.color_effects = color_effects
|
|
|
|
self.camera.rotation = rotation
|
|
|
|
self.camera.crop = crop
|
2018-03-24 03:05:46 +01:00
|
|
|
|
|
|
|
logging.info('Initialized Pi camera backend')
|
|
|
|
|
|
|
|
def send_message(self, msg):
|
|
|
|
pass
|
|
|
|
|
|
|
|
def run(self):
|
|
|
|
super().run()
|
|
|
|
|
|
|
|
while True:
|
|
|
|
connection = self.server_socket.accept()[0].makefile('wb')
|
|
|
|
|
|
|
|
try:
|
|
|
|
self.camera.start_recording(connection, format='h264')
|
|
|
|
while True:
|
|
|
|
self.camera.wait_recording(60)
|
|
|
|
except ConnectionError as e:
|
|
|
|
pass
|
|
|
|
finally:
|
|
|
|
try:
|
|
|
|
self.camera.stop_recording()
|
|
|
|
connection.close()
|
|
|
|
except:
|
|
|
|
pass
|
|
|
|
|
|
|
|
|
|
|
|
# vim:sw=4:ts=4:et:
|
|
|
|
|