65 lines
2.2 KiB
Python
65 lines
2.2 KiB
Python
from typing import Optional
|
|
|
|
from PIL import Image
|
|
from PIL.Image import Image as ImageType
|
|
|
|
from platypush.plugins.camera import CameraPlugin
|
|
from platypush.plugins.camera.gstreamer.model import GStreamerCamera, Pipeline, Loop
|
|
|
|
|
|
class CameraGstreamerPlugin(CameraPlugin):
|
|
"""
|
|
Plugin to interact with a camera over GStreamer.
|
|
|
|
Requires:
|
|
|
|
* **gst-python** (``pip install gst-python``)
|
|
|
|
"""
|
|
|
|
_camera_class = GStreamerCamera
|
|
|
|
def __init__(self, device: Optional[str] = '/dev/video0', **opts):
|
|
"""
|
|
:param device: Path to the camera device (default ``/dev/video0``).
|
|
:param opts: Camera options - see constructor of :class:`platypush.plugins.camera.CameraPlugin`.
|
|
"""
|
|
super().__init__(device=device, **opts)
|
|
|
|
def prepare_device(self, camera: GStreamerCamera) -> Pipeline:
|
|
pipeline = Pipeline()
|
|
src = pipeline.add('v4l2src', device=camera.info.device)
|
|
convert = pipeline.add('videoconvert')
|
|
video_filter = pipeline.add(
|
|
'capsfilter', caps='video/x-raw,format=RGB,width={width},height={height},framerate={fps}/1'.format(
|
|
width=camera.info.resolution[0], height=camera.info.resolution[1], fps=camera.info.fps))
|
|
|
|
sink = pipeline.add_sink('appsink', name='appsink', sync=False)
|
|
pipeline.link(src, convert, video_filter, sink)
|
|
return pipeline
|
|
|
|
def start_camera(self, camera: GStreamerCamera, preview: bool = False, *args, **kwargs):
|
|
super().start_camera(*args, camera=camera, preview=preview, **kwargs)
|
|
if camera.object:
|
|
camera.object.play()
|
|
|
|
def release_device(self, camera: GStreamerCamera):
|
|
if camera.object:
|
|
camera.object.stop()
|
|
|
|
def capture_frame(self, camera: GStreamerCamera, *args, **kwargs) -> Optional[ImageType]:
|
|
timed_out = not camera.object.data_ready.wait(timeout=5 + (1. / camera.info.fps))
|
|
if timed_out:
|
|
self.logger.warning('Frame capture timeout')
|
|
return
|
|
|
|
data = camera.object.data
|
|
camera.object.data_ready.clear()
|
|
if not data and len(data) != camera.info.resolution[0] * camera.info.resolution[1] * 3:
|
|
return
|
|
|
|
return Image.frombytes('RGB', camera.info.resolution, data)
|
|
|
|
|
|
# vim:sw=4:ts=4:et:
|