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)