62 lines
2.0 KiB
Python
62 lines
2.0 KiB
Python
from picam.picam import PiCam
|
|
|
|
from io import BytesIO
|
|
from picamera import PiCamera
|
|
|
|
class PiCam1(PiCam):
|
|
"""Implementation of PiCam class using legacy picamera module"""
|
|
def __init__(self, config: dict):
|
|
super().__init__(config)
|
|
|
|
self.init_camera()
|
|
|
|
def __del__(self):
|
|
"""Destructor - cleans up picamera"""
|
|
self.close_camera()
|
|
|
|
def set_camera_resolution(self, resolution: "tuple[int, int]") -> None:
|
|
"""Set resolution of active camera
|
|
|
|
:param resolution: Resolution to set camera to
|
|
"""
|
|
self.camera.resolution = resolution
|
|
|
|
def set_preview_resolution(self, resolution: "tuple[int, int]") -> None:
|
|
super().set_preview_resolution(resolution)
|
|
|
|
self.set_camera_resolution(self.preview_resolution)
|
|
|
|
def init_camera(self) -> None:
|
|
"""Initialize picamera camera object"""
|
|
self.camera = PiCamera(resolution=self.preview_resolution)
|
|
|
|
def close_camera(self) -> None:
|
|
"""Close picamera camera object"""
|
|
self.camera.close()
|
|
self.camera = None
|
|
|
|
def capture(self, preview: bool = False) -> bytearray:
|
|
# Bytes to hold output buffer
|
|
if preview:
|
|
# Multiply by 3 for individual R, G, B values
|
|
out = bytearray((self.preview_resolution[0] * self.preview_resolution[1]) * 3)
|
|
else:
|
|
out = bytearray((self.capture_resolution[0] * self.capture_resolution[1]) * 3)
|
|
self.set_camera_resolution(self.capture_resolution) # If preview we'll already have the right resolution
|
|
|
|
bytes_stream = BytesIO()
|
|
self.camera.capture(bytes_stream, format='rgb', use_video_port=preview)
|
|
|
|
bytes_stream.seek(0)
|
|
bytes_stream.readinto(out)
|
|
bytes_stream.close()
|
|
|
|
if not preview:
|
|
# If preview we'll already have the right resolution
|
|
self.set_camera_resolution(self.preview_resolution)
|
|
|
|
return out
|
|
|
|
def capture_preview(self) -> bytearray:
|
|
return self.capture(preview=True)
|