AIprojekt-wozek/venv/Lib/site-packages/pygame/camera.py

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()