예제 #1
0
파일: __init__.py 프로젝트: klngbhn/pupil
    def _init(self, cam, size=(640, 480), fps=None, timebase=None):
        # setting up device
        self.device = cam.device
        self.deviceSettings = vi.DeviceSettings()
        self.deviceSettings.symbolicLink = self.device.symbolicName
        self.deviceSettings.indexStream = 0
        self.deviceSettings.indexMediaType = 0
        self.captureSettings = vi.CaptureSettings()
        self.captureSettings.readMode = vi.ReadMode.SYNC
        self.captureSettings.videoFormat = vi.CaptureVideoFormat.RGB32
        self.stream = self.device.listStream[self.deviceSettings.indexStream]

        # collecting additional information
        if timebase == None:
            logger.debug("Capture will run with default system timebase")
            self.timebase = 0
        else:
            logger.debug("Capture will run with app wide adjustable timebase")
            self.timebase = timebase

        self.width = size[0]
        self.height = size[1]
        self.preferred_fps = fps
        self._initFrameRates()
        self._initMediaTypeId()

        self.context = _getVideoInputInstance()
        res = self.context.setupDevice(self.deviceSettings,
                                       self.captureSettings)
        if res != vi.ResultCode.OK:
            raise CameraCaptureError("Could not setup device. Error code: %d" %
                                     (res))

        # creating frame buffer and initializing capture settings
        frame = np.empty((self.actual_height * self.actual_width * 4),
                         dtype=np.uint8)
        self.readSetting = vi.ReadSetting()
        self.readSetting.symbolicLink = self.deviceSettings.symbolicLink
        self.readSetting.setNumpyArray(frame)
        frame.shape = (self.actual_height, self.actual_width, -1)
        self._frame = frame

        logger.debug(
            "Successfully set up device: %s @ %dx%dpx %dfps (mediatype %d)" %
            (self.name, self.actual_height, self.actual_width, self.frame_rate,
             self.deviceSettings.indexMediaType))
예제 #2
0
    def init_capture(self, uid, size=(640, 480), fps=None, timebase=None):
        self.uid = uid
        if uid is not None:
            # validate parameter UID
            devices = device_list(
            )  # TODO: read list only once (initially) to save runtime
            for device in devices:
                print
                if device['uid'] == uid:
                    break
            if device['uid'] != uid:
                msg = ERR_INIT_FAIL + "UID of camera was not found."
                logger.error(msg)
                self.init_capture(None, size, fps, timebase)
                return

            # validate parameter SIZE
            if not len(size) == 2:
                msg = ERR_INIT_FAIL + "Parameter 'size' must have length 2."
                logger.error(msg)
                self.init_capture(None, size, fps, timebase)
                return

            # setting up device
            self.device = device
            self.deviceSettings = vi.DeviceSettings()
            self.deviceSettings.symbolicLink = self.device['uid']
            self.deviceSettings.indexStream = 0
            self.deviceSettings.indexMediaType = 0
            self.captureSettings = vi.CaptureSettings()
            self.captureSettings.readMode = vi.ReadMode.SYNC
            self.captureSettings.videoFormat = vi.CaptureVideoFormat.RGB32
            self.stream = self.device['handle'].listStream[
                self.deviceSettings.indexStream]

            # set timebase
            if timebase == None:
                logger.debug("Capture will run with default system timebase")
                self.timebase = 0
            else:
                logger.debug(
                    "Capture will run with app wide adjustable timebase")
                self.timebase = timebase

            # set properties
            self.width = size[0]
            self.height = size[1]
            self.preferred_fps = fps
            self._initFrameRates()
            self._initMediaTypeId()

            # robust camera initialization
            self.context = _getVideoInputInstance()
            while True:
                res = self.context.setupDevice(self.deviceSettings,
                                               self.captureSettings)
                if res != vi.ResultCode.OK:
                    self._failed_inits += 1
                    msg = ERR_INIT_FAIL + "Fall back to Fake Capture. Error code: %d" % (
                        res)
                    if self._failed_inits < MAX_RETRY_INIT_CAMERA:
                        logger.warning(
                            "Retry initializing camera: {0}/{1}: ".format(
                                self._failed_inits, MAX_RETRY_INIT_CAMERA) +
                            msg)
                    else:
                        logger.error(msg)
                        self._failed_inits = 0
                        self.init_capture(None, size, fps, timebase)
                        return
                    sleep(0.25)
                else:
                    break

            # creating frame buffer and initializing capture settings
            frame = np.empty((self.actual_height * self.actual_width * 4),
                             dtype=np.uint8)
            self.readSetting = vi.ReadSetting()
            self.readSetting.symbolicLink = self.deviceSettings.symbolicLink
            self.readSetting.setNumpyArray(frame)
            frame.shape = (self.actual_height, self.actual_width, -1)
            self._frame = frame

            logger.debug(
                "Successfully set up device: %s @ %dx%dpx %dfps (mediatype %d)"
                % (self.name, self.actual_height, self.actual_width,
                   self.frame_rate, self.deviceSettings.indexMediaType))
        else:
            self.device = Fake_Capture()
            self.width = size[0]
            self.height = size[1]
            self.preferred_fps = fps
        self._is_initialized = True
        self._failed_inits = 0
예제 #3
0
    def _init(self, uid, size=(640,480), fps=None, timebase=None):

        devices = vi.DeviceList()
        for cam in _getVideoInputInstance().getListOfDevices(devices):
            if cam.symbolicName == uid:
                break
        if cam.symbolicName != uid:
            raise CameraCaptureError("uid for camera not found.")

        if not len(size) == 2:
            msg = ERR_INIT_FAIL + "Parameter 'size' must have length 2."
            logger.error(msg)
            raise CameraCaptureError(msg)
        # setting up device
        self.device = cam.device
        self.deviceSettings = vi.DeviceSettings()
        self.deviceSettings.symbolicLink = self.device.symbolicName
        self.deviceSettings.indexStream = 0
        self.deviceSettings.indexMediaType = 0
        self.captureSettings = vi.CaptureSettings()
        self.captureSettings.readMode = vi.ReadMode.SYNC
        self.captureSettings.videoFormat = vi.CaptureVideoFormat.RGB32
        self.stream = self.device.listStream[self.deviceSettings.indexStream]

        # collecting additional information
        if timebase == None:
            logger.debug("Capture will run with default system timebase")
            self.timebase = 0
        else:
            logger.debug("Capture will run with app wide adjustable timebase")
            self.timebase = timebase

        self.width = size[0]
        self.height = size[1]
        self.preferred_fps = fps
        self._initFrameRates()
        self._initMediaTypeId()

        self.context = _getVideoInputInstance()
        while True:
            res = self.context.setupDevice(self.deviceSettings, self.captureSettings)
            if res != vi.ResultCode.OK:
                self._failed_inits += 1
                msg = ERR_INIT_FAIL + "Error code: %d" %(res)
                if self._failed_inits < MAX_RETRY_INIT_CAMERA:
                    logger.error("Retry initializing camera: {0}/{1}: ".format(self._failed_inits, MAX_RETRY_INIT_CAMERA) + msg)
                else:
                    logger.error(msg)
                    raise CameraCaptureError(msg)
                sleep(1)
            else:
                break

        # creating frame buffer and initializing capture settings
        frame = np.empty((self.actual_height * self.actual_width * 4), dtype=np.uint8)
        self.readSetting = vi.ReadSetting()
        self.readSetting.symbolicLink = self.deviceSettings.symbolicLink
        self.readSetting.setNumpyArray(frame)
        frame.shape = (self.actual_height, self.actual_width, -1)
        self._frame = frame

        logger.debug("Successfully set up device: %s @ %dx%dpx %dfps (mediatype %d)" %(self.name, self.actual_height, self.actual_width, self.frame_rate, self.deviceSettings.indexMediaType))
        self._is_initialized = True
        self._failed_inits = 0
예제 #4
0
파일: test.py 프로젝트: uSEEet/pyvideoinput
print "LOADED VIDEOINPUT LIB"

vin = vi.videoInput_getInstance()
assert isinstance(vin, vi.videoInput)

devices = vi.DeviceList()
vin.getListOfDevices(devices)
for d in devices:
    assert isinstance(d, vi.Device)
    print d.friendlyName, d.symbolicName
    dev = d
dev = devices[1]
assert isinstance(dev, vi.Device)
print "LOADED CAPTURE DEVICES"

deviceSettings = vi.DeviceSettings()
deviceSettings.symbolicLink = dev.symbolicName
deviceSettings.indexStream = 0
deviceSettings.indexMediaType = 195
captureSettings = vi.CaptureSettings()
captureSettings.readMode = vi.ReadMode.SYNC
captureSettings.videoFormat = vi.CaptureVideoFormat.RGB24

stream = dev.listStream[0]
assert isinstance(stream, vi.Stream)
mediaType = stream.listMediaType[195]
for mt in stream.listMediaType:
    print mt.width, mt.height, mt.MF_MT_FRAME_RATE

assert isinstance(mediaType, vi.MediaType)
frame = np.zeros((mediaType.height, mediaType.width, 3), dtype=np.uint8)