from pathlib import Path
from typing import Optional
from numpy import ndarray
import picamera
import picamera.array
from .base import BaseCamera
[docs]class PiCamera(BaseCamera):
[docs] def __init__(self, *, calibration_file: Optional[Path] = None) -> None:
super().__init__(calibration_file=calibration_file)
self.camera = picamera.PiCamera()
[docs] def capture_frame(self) -> ndarray:
with picamera.array.PiRGBArray(self.camera) as raw_capture:
self.camera.capture(raw_capture, format="bgr")
return raw_capture.array
[docs] def close(self) -> None:
super().close()
self.camera.close()
[docs]class PiSnapshotCamera(BaseCamera):
"""
A modified version of PiCamera optimised for single use.
- Doesn't keep the camera open between captures
"""
[docs] def capture_frame(self) -> ndarray:
with picamera.PiCamera() as camera:
with picamera.array.PiRGBArray(camera) as raw_capture:
camera.capture(raw_capture, format="bgr")
return raw_capture.array
[docs] def close(self) -> None:
super().close()