modules/video_capture/linux/device_info_linux.cc - src.git - Git at Google

 /*
  *  Copyright (c) 2012 The WebRTC project authors. All Rights Reserved.
  *
  *  Use of this source code is governed by a BSD-style license
  *  that can be found in the LICENSE file in the root of the source
  *  tree. An additional intellectual property rights grant can be found
  *  in the file PATENTS.  All contributing project authors may
  *  be found in the AUTHORS file in the root of the source tree.
  */

 #include "modules/video_capture/linux/device_info_linux.h"

 #include <errno.h>
 #include <fcntl.h>
 #include <stdio.h>
 #include <stdlib.h>
 #include <sys/ioctl.h>
 #include <sys/stat.h>
 #include <unistd.h>
 // v4l includes
 #include <linux/videodev2.h>

 #include "rtc_base/logging.h"

 namespace webrtc {
 namespace videocapturemodule {
 VideoCaptureModule::DeviceInfo* VideoCaptureImpl::CreateDeviceInfo() {
   return new videocapturemodule::DeviceInfoLinux();
 }

 DeviceInfoLinux::DeviceInfoLinux() : DeviceInfoImpl() {}

 int32_t DeviceInfoLinux::Init() {
   return 0;
 }

 DeviceInfoLinux::~DeviceInfoLinux() {}

 uint32_t DeviceInfoLinux::NumberOfDevices() {
   RTC_LOG(LS_INFO) << __FUNCTION__;

   uint32_t count = 0;
   char device[20];
   int fd = -1;

   /* detect /dev/video [0-63]VideoCaptureModule entries */
   for (int n = 0; n < 64; n++) {
     sprintf(device, "/dev/video%d", n);
     if ((fd = open(device, O_RDONLY)) != -1) {
       close(fd);
       count++;
     }
   }

   return count;
 }

 int32_t DeviceInfoLinux::GetDeviceName(uint32_t deviceNumber,
                                        char* deviceNameUTF8,
                                        uint32_t deviceNameLength,
                                        char* deviceUniqueIdUTF8,
                                        uint32_t deviceUniqueIdUTF8Length,
                                        char* /*productUniqueIdUTF8*/,
                                        uint32_t /*productUniqueIdUTF8Length*/) {
   RTC_LOG(LS_INFO) << __FUNCTION__;

   // Travel through /dev/video [0-63]
   uint32_t count = 0;
   char device[20];
   int fd = -1;
   bool found = false;
   for (int n = 0; n < 64; n++) {
     sprintf(device, "/dev/video%d", n);
     if ((fd = open(device, O_RDONLY)) != -1) {
       if (count == deviceNumber) {
         // Found the device
         found = true;
         break;
       } else {
         close(fd);
         count++;
       }
     }
   }

   if (!found)
     return -1;

   // query device capabilities
   struct v4l2_capability cap;
   if (ioctl(fd, VIDIOC_QUERYCAP, &cap) < 0) {
     RTC_LOG(LS_INFO) << "error in querying the device capability for device "
                      << device << ". errno = " << errno;
     close(fd);
     return -1;
   }

   close(fd);

   char cameraName[64];
   memset(deviceNameUTF8, 0, deviceNameLength);
   memcpy(cameraName, cap.card, sizeof(cap.card));

   if (deviceNameLength >= strlen(cameraName)) {
     memcpy(deviceNameUTF8, cameraName, strlen(cameraName));
   } else {
     RTC_LOG(LS_INFO) << "buffer passed is too small";
     return -1;
   }

   if (cap.bus_info[0] != 0)  // may not available in all drivers
   {
     // copy device id
     if (deviceUniqueIdUTF8Length >= strlen((const char*)cap.bus_info)) {
       memset(deviceUniqueIdUTF8, 0, deviceUniqueIdUTF8Length);
       memcpy(deviceUniqueIdUTF8, cap.bus_info,
              strlen((const char*)cap.bus_info));
     } else {
       RTC_LOG(LS_INFO) << "buffer passed is too small";
       return -1;
     }
   }

   return 0;
 }

 int32_t DeviceInfoLinux::CreateCapabilityMap(const char* deviceUniqueIdUTF8) {
   int fd;
   char device[32];
   bool found = false;

   const int32_t deviceUniqueIdUTF8Length =
       (int32_t)strlen((char*)deviceUniqueIdUTF8);
   if (deviceUniqueIdUTF8Length > kVideoCaptureUniqueNameLength) {
     RTC_LOG(LS_INFO) << "Device name too long";
     return -1;
   }
   RTC_LOG(LS_INFO) << "CreateCapabilityMap called for device "
                    << deviceUniqueIdUTF8;

   /* detect /dev/video [0-63] entries */
   for (int n = 0; n < 64; ++n) {
     sprintf(device, "/dev/video%d", n);
     fd = open(device, O_RDONLY);
     if (fd == -1)
       continue;

     // query device capabilities
     struct v4l2_capability cap;
     if (ioctl(fd, VIDIOC_QUERYCAP, &cap) == 0) {
       if (cap.bus_info[0] != 0) {
         if (strncmp((const char*)cap.bus_info, (const char*)deviceUniqueIdUTF8,
                     strlen((const char*)deviceUniqueIdUTF8)) ==
             0)  // match with device id
         {
           found = true;
           break;  // fd matches with device unique id supplied
         }
       } else  // match for device name
       {
         if (IsDeviceNameMatches((const char*)cap.card,
                                 (const char*)deviceUniqueIdUTF8)) {
           found = true;
           break;
         }
       }
     }
     close(fd);  // close since this is not the matching device
   }

   if (!found) {
     RTC_LOG(LS_INFO) << "no matching device found";
     return -1;
   }

   // now fd will point to the matching device
   // reset old capability list.
   _captureCapabilities.clear();

   int size = FillCapabilities(fd);
   close(fd);

   // Store the new used device name
   _lastUsedDeviceNameLength = deviceUniqueIdUTF8Length;
   _lastUsedDeviceName =
       (char*)realloc(_lastUsedDeviceName, _lastUsedDeviceNameLength + 1);
   memcpy(_lastUsedDeviceName, deviceUniqueIdUTF8,
          _lastUsedDeviceNameLength + 1);

   RTC_LOG(LS_INFO) << "CreateCapabilityMap " << _captureCapabilities.size();

   return size;
 }

 bool DeviceInfoLinux::IsDeviceNameMatches(const char* name,
                                           const char* deviceUniqueIdUTF8) {
   if (strncmp(deviceUniqueIdUTF8, name, strlen(name)) == 0)
     return true;
   return false;
 }

 int32_t DeviceInfoLinux::FillCapabilities(int fd) {
   // set image format
   struct v4l2_format video_fmt;
   memset(&video_fmt, 0, sizeof(struct v4l2_format));

   video_fmt.type = V4L2_BUF_TYPE_VIDEO_CAPTURE;
   video_fmt.fmt.pix.sizeimage = 0;

   int totalFmts = 4;
   unsigned int videoFormats[] = {V4L2_PIX_FMT_MJPEG, V4L2_PIX_FMT_YUV420,
                                  V4L2_PIX_FMT_YUYV, V4L2_PIX_FMT_UYVY};

   int sizes = 13;
   unsigned int size[][2] = {{128, 96},   {160, 120},  {176, 144},  {320, 240},
                             {352, 288},  {640, 480},  {704, 576},  {800, 600},
                             {960, 720},  {1280, 720}, {1024, 768}, {1440, 1080},
                             {1920, 1080}};

   int index = 0;
   for (int fmts = 0; fmts < totalFmts; fmts++) {
     for (int i = 0; i < sizes; i++) {
       video_fmt.fmt.pix.pixelformat = videoFormats[fmts];
       video_fmt.fmt.pix.width = size[i][0];
       video_fmt.fmt.pix.height = size[i][1];

       if (ioctl(fd, VIDIOC_TRY_FMT, &video_fmt) >= 0) {
         if ((video_fmt.fmt.pix.width == size[i][0]) &&
             (video_fmt.fmt.pix.height == size[i][1])) {
           VideoCaptureCapability cap;
           cap.width = video_fmt.fmt.pix.width;
           cap.height = video_fmt.fmt.pix.height;
           if (videoFormats[fmts] == V4L2_PIX_FMT_YUYV) {
             cap.videoType = VideoType::kYUY2;
           } else if (videoFormats[fmts] == V4L2_PIX_FMT_YUV420) {
             cap.videoType = VideoType::kI420;
           } else if (videoFormats[fmts] == V4L2_PIX_FMT_MJPEG) {
             cap.videoType = VideoType::kMJPEG;
           } else if (videoFormats[fmts] == V4L2_PIX_FMT_UYVY) {
             cap.videoType = VideoType::kUYVY;
           }

           // get fps of current camera mode
           // V4l2 does not have a stable method of knowing so we just guess.
           if (cap.width >= 800 && cap.videoType != VideoType::kMJPEG) {
             cap.maxFPS = 15;
           } else {
             cap.maxFPS = 30;
           }

           _captureCapabilities.push_back(cap);
           index++;
           RTC_LOG(LS_VERBOSE) << "Camera capability, width:" << cap.width
                               << " height:" << cap.height
                               << " type:" << static_cast<int32_t>(cap.videoType)
                               << " fps:" << cap.maxFPS;
         }
       }
     }
   }

   RTC_LOG(LS_INFO) << "CreateCapabilityMap " << _captureCapabilities.size();
   return _captureCapabilities.size();
 }

 }  // namespace videocapturemodule
 }  // namespace webrtc
	/*
	* Copyright (c) 2012 The WebRTC project authors. All Rights Reserved.
	*
	* Use of this source code is governed by a BSD-style license
	* that can be found in the LICENSE file in the root of the source
	* tree. An additional intellectual property rights grant can be found
	* in the file PATENTS. All contributing project authors may
	* be found in the AUTHORS file in the root of the source tree.
	*/

	#include "modules/video_capture/linux/device_info_linux.h"

	#include <errno.h>
	#include <fcntl.h>
	#include <stdio.h>
	#include <stdlib.h>
	#include <sys/ioctl.h>
	#include <sys/stat.h>
	#include <unistd.h>
	// v4l includes
	#include <linux/videodev2.h>

	#include "rtc_base/logging.h"

	namespace webrtc {
	namespace videocapturemodule {
	VideoCaptureModule::DeviceInfo* VideoCaptureImpl::CreateDeviceInfo() {
	return new videocapturemodule::DeviceInfoLinux();
	}

	DeviceInfoLinux::DeviceInfoLinux() : DeviceInfoImpl() {}

	int32_t DeviceInfoLinux::Init() {
	return 0;
	}

	DeviceInfoLinux::~DeviceInfoLinux() {}

	uint32_t DeviceInfoLinux::NumberOfDevices() {
	RTC_LOG(LS_INFO) << __FUNCTION__;

	uint32_t count = 0;
	char device[20];
	int fd = -1;

	/* detect /dev/video [0-63]VideoCaptureModule entries */
	for (int n = 0; n < 64; n++) {
	sprintf(device, "/dev/video%d", n);
	if ((fd = open(device, O_RDONLY)) != -1) {
	close(fd);
	count++;
	}
	}

	return count;
	}

	int32_t DeviceInfoLinux::GetDeviceName(uint32_t deviceNumber,
	char* deviceNameUTF8,
	uint32_t deviceNameLength,
	char* deviceUniqueIdUTF8,
	uint32_t deviceUniqueIdUTF8Length,
	char* /productUniqueIdUTF8/,
	uint32_t /productUniqueIdUTF8Length/) {
	RTC_LOG(LS_INFO) << __FUNCTION__;

	// Travel through /dev/video [0-63]
	uint32_t count = 0;
	char device[20];
	int fd = -1;
	bool found = false;
	for (int n = 0; n < 64; n++) {
	sprintf(device, "/dev/video%d", n);
	if ((fd = open(device, O_RDONLY)) != -1) {
	if (count == deviceNumber) {
	// Found the device
	found = true;
	break;
	} else {
	close(fd);
	count++;
	}
	}
	}

	if (!found)
	return -1;

	// query device capabilities
	struct v4l2_capability cap;
	if (ioctl(fd, VIDIOC_QUERYCAP, &cap) < 0) {
	RTC_LOG(LS_INFO) << "error in querying the device capability for device "
	<< device << ". errno = " << errno;
	close(fd);
	return -1;
	}

	close(fd);

	char cameraName[64];
	memset(deviceNameUTF8, 0, deviceNameLength);
	memcpy(cameraName, cap.card, sizeof(cap.card));

	if (deviceNameLength >= strlen(cameraName)) {
	memcpy(deviceNameUTF8, cameraName, strlen(cameraName));
	} else {
	RTC_LOG(LS_INFO) << "buffer passed is too small";
	return -1;
	}

	if (cap.bus_info[0] != 0) // may not available in all drivers
	{
	// copy device id
	if (deviceUniqueIdUTF8Length >= strlen((const char*)cap.bus_info)) {
	memset(deviceUniqueIdUTF8, 0, deviceUniqueIdUTF8Length);
	memcpy(deviceUniqueIdUTF8, cap.bus_info,
	strlen((const char*)cap.bus_info));
	} else {
	RTC_LOG(LS_INFO) << "buffer passed is too small";
	return -1;
	}
	}

	return 0;
	}

	int32_t DeviceInfoLinux::CreateCapabilityMap(const char* deviceUniqueIdUTF8) {
	int fd;
	char device[32];
	bool found = false;

	const int32_t deviceUniqueIdUTF8Length =
	(int32_t)strlen((char*)deviceUniqueIdUTF8);
	if (deviceUniqueIdUTF8Length > kVideoCaptureUniqueNameLength) {
	RTC_LOG(LS_INFO) << "Device name too long";
	return -1;
	}
	RTC_LOG(LS_INFO) << "CreateCapabilityMap called for device "
	<< deviceUniqueIdUTF8;

	/* detect /dev/video [0-63] entries */
	for (int n = 0; n < 64; ++n) {
	sprintf(device, "/dev/video%d", n);
	fd = open(device, O_RDONLY);
	if (fd == -1)
	continue;

	// query device capabilities
	struct v4l2_capability cap;
	if (ioctl(fd, VIDIOC_QUERYCAP, &cap) == 0) {
	if (cap.bus_info[0] != 0) {
	if (strncmp((const char)cap.bus_info, (const char)deviceUniqueIdUTF8,
	strlen((const char*)deviceUniqueIdUTF8)) ==
	0) // match with device id
	{
	found = true;
	break; // fd matches with device unique id supplied
	}
	} else // match for device name
	{
	if (IsDeviceNameMatches((const char*)cap.card,
	(const char*)deviceUniqueIdUTF8)) {
	found = true;
	break;
	}
	}
	}
	close(fd); // close since this is not the matching device
	}

	if (!found) {
	RTC_LOG(LS_INFO) << "no matching device found";
	return -1;
	}

	// now fd will point to the matching device
	// reset old capability list.
	_captureCapabilities.clear();

	int size = FillCapabilities(fd);
	close(fd);

	// Store the new used device name
	_lastUsedDeviceNameLength = deviceUniqueIdUTF8Length;
	_lastUsedDeviceName =
	(char*)realloc(_lastUsedDeviceName, _lastUsedDeviceNameLength + 1);
	memcpy(_lastUsedDeviceName, deviceUniqueIdUTF8,
	_lastUsedDeviceNameLength + 1);

	RTC_LOG(LS_INFO) << "CreateCapabilityMap " << _captureCapabilities.size();

	return size;
	}

	bool DeviceInfoLinux::IsDeviceNameMatches(const char* name,
	const char* deviceUniqueIdUTF8) {
	if (strncmp(deviceUniqueIdUTF8, name, strlen(name)) == 0)
	return true;
	return false;
	}

	int32_t DeviceInfoLinux::FillCapabilities(int fd) {
	// set image format
	struct v4l2_format video_fmt;
	memset(&video_fmt, 0, sizeof(struct v4l2_format));

	video_fmt.type = V4L2_BUF_TYPE_VIDEO_CAPTURE;
	video_fmt.fmt.pix.sizeimage = 0;

	int totalFmts = 4;
	unsigned int videoFormats[] = {V4L2_PIX_FMT_MJPEG, V4L2_PIX_FMT_YUV420,
	V4L2_PIX_FMT_YUYV, V4L2_PIX_FMT_UYVY};

	int sizes = 13;
	unsigned int size[][2] = {{128, 96}, {160, 120}, {176, 144}, {320, 240},
	{352, 288}, {640, 480}, {704, 576}, {800, 600},
	{960, 720}, {1280, 720}, {1024, 768}, {1440, 1080},
	{1920, 1080}};

	int index = 0;
	for (int fmts = 0; fmts < totalFmts; fmts++) {
	for (int i = 0; i < sizes; i++) {
	video_fmt.fmt.pix.pixelformat = videoFormats[fmts];
	video_fmt.fmt.pix.width = size[i][0];
	video_fmt.fmt.pix.height = size[i][1];

	if (ioctl(fd, VIDIOC_TRY_FMT, &video_fmt) >= 0) {
	if ((video_fmt.fmt.pix.width == size[i][0]) &&
	(video_fmt.fmt.pix.height == size[i][1])) {
	VideoCaptureCapability cap;
	cap.width = video_fmt.fmt.pix.width;
	cap.height = video_fmt.fmt.pix.height;
	if (videoFormats[fmts] == V4L2_PIX_FMT_YUYV) {
	cap.videoType = VideoType::kYUY2;
	} else if (videoFormats[fmts] == V4L2_PIX_FMT_YUV420) {
	cap.videoType = VideoType::kI420;
	} else if (videoFormats[fmts] == V4L2_PIX_FMT_MJPEG) {
	cap.videoType = VideoType::kMJPEG;
	} else if (videoFormats[fmts] == V4L2_PIX_FMT_UYVY) {
	cap.videoType = VideoType::kUYVY;
	}

	// get fps of current camera mode
	// V4l2 does not have a stable method of knowing so we just guess.
	if (cap.width >= 800 && cap.videoType != VideoType::kMJPEG) {
	cap.maxFPS = 15;
	} else {
	cap.maxFPS = 30;
	}

	_captureCapabilities.push_back(cap);
	index++;
	RTC_LOG(LS_VERBOSE) << "Camera capability, width:" << cap.width
	<< " height:" << cap.height
	<< " type:" << static_cast<int32_t>(cap.videoType)
	<< " fps:" << cap.maxFPS;
	}
	}
	}
	}

	RTC_LOG(LS_INFO) << "CreateCapabilityMap " << _captureCapabilities.size();
	return _captureCapabilities.size();
	}

	} // namespace videocapturemodule
	} // namespace webrtc