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
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
|
import os
import sys
import platform
import warnings
from abc import ABC, abstractmethod
_is_init = 0
def _setup_opencv_mac():
global list_cameras, Camera, colorspace
from pygame import _camera_opencv
try:
from pygame import _camera
except ImportError:
_camera = None
list_cameras = _camera_opencv.list_cameras_darwin
Camera = _camera_opencv.CameraMac
if _camera:
colorspace = _camera.colorspace
def _setup_opencv():
global list_cameras, Camera, colorspace
from pygame import _camera_opencv
try:
from pygame import _camera
except ImportError:
_camera = None
list_cameras = _camera_opencv.list_cameras
Camera = _camera_opencv.Camera
if _camera:
colorspace = _camera.colorspace
def _setup__camera():
global list_cameras, Camera, colorspace
from pygame import _camera
list_cameras = _camera.list_cameras
Camera = _camera.Camera
colorspace = _camera.colorspace
def _setup_vidcapture():
global list_cameras, Camera, colorspace
from pygame import _camera_vidcapture
try:
from pygame import _camera
except ImportError:
_camera = None
warnings.warn(
"The VideoCapture backend is not recommended and may be removed."
"For Python3 and Windows 8+, there is now a native Windows backend built into pygame.",
DeprecationWarning,
stacklevel=2,
)
_camera_vidcapture.init()
list_cameras = _camera_vidcapture.list_cameras
Camera = _camera_vidcapture.Camera
if _camera:
colorspace = _camera.colorspace
def get_backends():
possible_backends = []
if sys.platform == "win32" and int(platform.win32_ver()[0]) > 8:
possible_backends.append("_camera (MSMF)")
if "linux" in sys.platform:
possible_backends.append("_camera (V4L2)")
if "darwin" in sys.platform:
possible_backends.append("OpenCV-Mac")
possible_backends.append("OpenCV")
if sys.platform == "win32":
possible_backends.append("VidCapture")
# see if we have any user specified defaults in environments.
camera_env = os.environ.get("PYGAME_CAMERA", "")
if camera_env == "opencv": # prioritize opencv
if "OpenCV" in possible_backends:
possible_backends.remove("OpenCV")
possible_backends = ["OpenCV"] + possible_backends
if camera_env == "vidcapture": # prioritize vidcapture
if "VideoCapture" in possible_backends:
possible_backends.remove("VideoCapture")
possible_backends = ["VideoCapture"] + possible_backends
return possible_backends
backend_table = {
"opencv-mac": _setup_opencv_mac,
"opencv": _setup_opencv,
"_camera (msmf)": _setup__camera,
"_camera (v4l2)": _setup__camera,
"videocapture": _setup_vidcapture,
}
def init(backend=None):
global _is_init
# select the camera module to import here.
backends = get_backends()
if not backends:
_is_init = 1
return
backends = [b.lower() for b in backends]
if not backend:
backend = backends[0]
else:
backend = backend.lower()
if backend not in backend_table:
raise ValueError("unrecognized backend name")
if backend not in backends:
warnings.warn(
"We don't think this is a supported backend on this system, but we'll try it...",
Warning,
stacklevel=2,
)
backend_table[backend]()
_is_init = 1
def quit():
global _is_init
_is_init = 0
class AbstractCamera(ABC):
@abstractmethod
def __init__(self, device=0, size=(320, 200), mode="RGB"):
""" """
@abstractmethod
def set_resolution(self, width, height):
"""Sets the capture resolution. (without dialog)"""
@abstractmethod
def start(self):
""" """
@abstractmethod
def stop(self):
""" """
@abstractmethod
def get_buffer(self):
""" """
@abstractmethod
def set_controls(self, **kwargs):
""" """
@abstractmethod
def get_image(self, dest_surf=None):
""" """
@abstractmethod
def get_surface(self, dest_surf=None):
""" """
if __name__ == "__main__":
# try and use this camera stuff with the pygame camera example.
import pygame.examples.camera
# pygame.camera.Camera = Camera
# pygame.camera.list_cameras = list_cameras
pygame.examples.camera.main()
|
