Camera_streaming_rpi/camera.py at main · build-error/Camera_streaming_rpi · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
import cv2

try:
    from picamera2 import Picamera2
    PICAMERA2_AVAILABLE = True
except ImportError:
    PICAMERA2_AVAILABLE = False


class Camera:

    def __init__(
        self,
        width=640,
        height=480,
        use_picam=True,
        camera_id=0
    ):

        self.camera_type = None

        # ---------------------------------
        # Try Raspberry Pi Camera
        # ---------------------------------

        if use_picam and PICAMERA2_AVAILABLE:

            try:

                self.picam2 = Picamera2()

                self.picam2.configure(
                    self.picam2.create_preview_configuration(
                        main={
                            "size": (width, height),
                            "format": "BGR888"
                        }
                    )
                )

                self.picam2.start()

                self.camera_type = "picamera2"

                print("[INFO] Using Pi Camera")

                return

            except Exception as e:

                print(
                    "[WARN] Pi Camera unavailable:",
                    e
                )

        # ---------------------------------
        # Fall back to USB webcam
        # ---------------------------------

        self.cap = cv2.VideoCapture(camera_id)

        if not self.cap.isOpened():

            raise RuntimeError(
                "No camera detected."
            )

        self.cap.set(
            cv2.CAP_PROP_FRAME_WIDTH,
            width
        )

        self.cap.set(
            cv2.CAP_PROP_FRAME_HEIGHT,
            height
        )

        self.camera_type = "usb"

        print(
            f"[INFO] Using USB Webcam ({camera_id})"
        )

    # ---------------------------------
    # Grab frame
    # ---------------------------------

    def read(self):

        if self.camera_type == "picamera2":

            frame = self.picam2.capture_array()

            return True, frame

        ret, frame = self.cap.read()

        return ret, frame

    # ---------------------------------
    # Cleanup
    # ---------------------------------

    def release(self):

        if self.camera_type == "picamera2":

            self.picam2.stop()

        elif self.camera_type == "usb":

            self.cap.release()