| 1 | /* --------------------------------------------------------------------------- | 
|---|
| 2 | ** This software is in the public domain, furnished "as is", without technical | 
|---|
| 3 | ** support, and with no warranty, express or implied, as to its usefulness for | 
|---|
| 4 | ** any purpose. | 
|---|
| 5 | ** | 
|---|
| 6 | ** V4l2Device.cpp | 
|---|
| 7 | ** | 
|---|
| 8 | ** -------------------------------------------------------------------------*/ | 
|---|
| 9 |  | 
|---|
| 10 | #include <unistd.h> | 
|---|
| 11 | #include <errno.h> | 
|---|
| 12 | #include <string.h> | 
|---|
| 13 | #include <sys/ioctl.h> | 
|---|
| 14 | #include <sys/stat.h> | 
|---|
| 15 |  | 
|---|
| 16 | // libv4l2 | 
|---|
| 17 | #include <linux/videodev2.h> | 
|---|
| 18 |  | 
|---|
| 19 | #include "logger.h" | 
|---|
| 20 |  | 
|---|
| 21 | #include "V4l2Device.h" | 
|---|
| 22 |  | 
|---|
| 23 | std::string fourcc(unsigned int format) | 
|---|
| 24 | { | 
|---|
| 25 | char formatArray[] = { (char)(format&0xff), (char)((format>>8)&0xff), (char)((format>>16)&0xff), (char)((format>>24)&0xff), 0 }; | 
|---|
| 26 | return std::string(formatArray, strlen(formatArray)); | 
|---|
| 27 | } | 
|---|
| 28 |  | 
|---|
| 29 | // ----------------------------------------- | 
|---|
| 30 | //    V4L2Device | 
|---|
| 31 | // ----------------------------------------- | 
|---|
| 32 | V4l2Device::V4l2Device(const V4L2DeviceParameters&  params, v4l2_buf_type deviceType) : m_params(params), m_fd(-1), m_deviceType(deviceType), m_bufferSize(0), m_format(0) | 
|---|
| 33 | { | 
|---|
| 34 | } | 
|---|
| 35 |  | 
|---|
| 36 | V4l2Device::~V4l2Device() | 
|---|
| 37 | { | 
|---|
| 38 | this->close(); | 
|---|
| 39 | } | 
|---|
| 40 |  | 
|---|
| 41 | void V4l2Device::close() | 
|---|
| 42 | { | 
|---|
| 43 | if (m_fd != -1) | 
|---|
| 44 | ::close(m_fd); | 
|---|
| 45 |  | 
|---|
| 46 | m_fd = -1; | 
|---|
| 47 | } | 
|---|
| 48 |  | 
|---|
| 49 | // query current format | 
|---|
| 50 | void V4l2Device::queryFormat() | 
|---|
| 51 | { | 
|---|
| 52 | struct v4l2_format     fmt; | 
|---|
| 53 | memset(&fmt,0,sizeof(fmt)); | 
|---|
| 54 | fmt.type  = m_deviceType; | 
|---|
| 55 | if (0 == ioctl(m_fd,VIDIOC_G_FMT,&fmt)) | 
|---|
| 56 | { | 
|---|
| 57 | m_format     = fmt.fmt.pix.pixelformat; | 
|---|
| 58 | m_width      = fmt.fmt.pix.width; | 
|---|
| 59 | m_height     = fmt.fmt.pix.height; | 
|---|
| 60 | m_bufferSize = fmt.fmt.pix.sizeimage; | 
|---|
| 61 |  | 
|---|
| 62 | LOG(NOTICE) << m_params.m_devName << ":"<< fourcc(m_format) << " size:"<< m_width << "x"<< m_height << " bufferSize:"<< m_bufferSize; | 
|---|
| 63 | } | 
|---|
| 64 | } | 
|---|
| 65 |  | 
|---|
| 66 | // intialize the V4L2 connection | 
|---|
| 67 | bool V4l2Device::init(unsigned int mandatoryCapabilities) | 
|---|
| 68 | { | 
|---|
| 69 | struct stat sb; | 
|---|
| 70 | if ( (stat(m_params.m_devName.c_str(), &sb)==0) && ((sb.st_mode & S_IFMT) == S_IFCHR) ) | 
|---|
| 71 | { | 
|---|
| 72 | if (initdevice(m_params.m_devName.c_str(), mandatoryCapabilities) == -1) | 
|---|
| 73 | { | 
|---|
| 74 | LOG(ERROR) << "Cannot init device:"<< m_params.m_devName; | 
|---|
| 75 | } | 
|---|
| 76 | } | 
|---|
| 77 | else | 
|---|
| 78 | { | 
|---|
| 79 | // open a normal file | 
|---|
| 80 | m_fd = open(m_params.m_devName.c_str(), O_WRONLY | O_CREAT | O_TRUNC, S_IRWXU); | 
|---|
| 81 | } | 
|---|
| 82 | return (m_fd!=-1); | 
|---|
| 83 | } | 
|---|
| 84 |  | 
|---|
| 85 | // intialize the V4L2 device | 
|---|
| 86 | int V4l2Device::initdevice(const char *dev_name, unsigned int mandatoryCapabilities) | 
|---|
| 87 | { | 
|---|
| 88 | m_fd = open(dev_name,  m_params.m_openFlags); | 
|---|
| 89 | if (m_fd < 0) | 
|---|
| 90 | { | 
|---|
| 91 | LOG(ERROR) << "Cannot open device:"<< m_params.m_devName << " "<< strerror(errno); | 
|---|
| 92 | this->close(); | 
|---|
| 93 | return -1; | 
|---|
| 94 | } | 
|---|
| 95 | if (checkCapabilities(m_fd,mandatoryCapabilities) !=0) | 
|---|
| 96 | { | 
|---|
| 97 | this->close(); | 
|---|
| 98 | return -1; | 
|---|
| 99 | } | 
|---|
| 100 | if (configureFormat(m_fd) !=0) | 
|---|
| 101 | { | 
|---|
| 102 | this->close(); | 
|---|
| 103 | return -1; | 
|---|
| 104 | } | 
|---|
| 105 | if (configureParam(m_fd) !=0) | 
|---|
| 106 | { | 
|---|
| 107 | this->close(); | 
|---|
| 108 | return -1; | 
|---|
| 109 | } | 
|---|
| 110 |  | 
|---|
| 111 | return m_fd; | 
|---|
| 112 | } | 
|---|
| 113 |  | 
|---|
| 114 | // check needed V4L2 capabilities | 
|---|
| 115 | int V4l2Device::checkCapabilities(int fd, unsigned int mandatoryCapabilities) | 
|---|
| 116 | { | 
|---|
| 117 | struct v4l2_capability cap; | 
|---|
| 118 | memset(&(cap), 0, sizeof(cap)); | 
|---|
| 119 | if (-1 == ioctl(fd, VIDIOC_QUERYCAP, &cap)) | 
|---|
| 120 | { | 
|---|
| 121 | LOG(ERROR) << "Cannot get capabilities for device:"<< m_params.m_devName << " "<< strerror(errno); | 
|---|
| 122 | return -1; | 
|---|
| 123 | } | 
|---|
| 124 | LOG(NOTICE) << "driver:"<< cap.driver << " capabilities:"<< std::hex << cap.capabilities << " mandatory:"<< mandatoryCapabilities << std::dec; | 
|---|
| 125 |  | 
|---|
| 126 | if ((cap.capabilities & V4L2_CAP_VIDEO_OUTPUT))  LOG(NOTICE) << m_params.m_devName << " support output"; | 
|---|
| 127 | if ((cap.capabilities & V4L2_CAP_VIDEO_CAPTURE)) LOG(NOTICE) << m_params.m_devName << " support capture"; | 
|---|
| 128 |  | 
|---|
| 129 | if ((cap.capabilities & V4L2_CAP_READWRITE))     LOG(NOTICE) << m_params.m_devName << " support read/write"; | 
|---|
| 130 | if ((cap.capabilities & V4L2_CAP_STREAMING))     LOG(NOTICE) << m_params.m_devName << " support streaming"; | 
|---|
| 131 |  | 
|---|
| 132 | if ((cap.capabilities & V4L2_CAP_TIMEPERFRAME))  LOG(NOTICE) << m_params.m_devName << " support timeperframe"; | 
|---|
| 133 |  | 
|---|
| 134 | if ( (cap.capabilities & mandatoryCapabilities) != mandatoryCapabilities ) | 
|---|
| 135 | { | 
|---|
| 136 | LOG(ERROR) << "Mandatory capability not available for device:"<< m_params.m_devName; | 
|---|
| 137 | return -1; | 
|---|
| 138 | } | 
|---|
| 139 |  | 
|---|
| 140 | return 0; | 
|---|
| 141 | } | 
|---|
| 142 |  | 
|---|
| 143 | // configure capture format | 
|---|
| 144 | int V4l2Device::configureFormat(int fd) | 
|---|
| 145 | { | 
|---|
| 146 | // get current configuration | 
|---|
| 147 | this->queryFormat(); | 
|---|
| 148 |  | 
|---|
| 149 | unsigned int width = m_width; | 
|---|
| 150 | unsigned int height = m_height; | 
|---|
| 151 | if (m_params.m_width != 0)  { | 
|---|
| 152 | width= m_params.m_width; | 
|---|
| 153 | } | 
|---|
| 154 | if (m_params.m_height != 0)  { | 
|---|
| 155 | height= m_params.m_height; | 
|---|
| 156 | } | 
|---|
| 157 | if  ( (m_params.m_formatList.size()==0) && (m_format != 0) )  { | 
|---|
| 158 | m_params.m_formatList.push_back(m_format); | 
|---|
| 159 | } | 
|---|
| 160 |  | 
|---|
| 161 | // try to set format, widht, height | 
|---|
| 162 | std::list<unsigned int>::iterator it; | 
|---|
| 163 | for (it = m_params.m_formatList.begin(); it != m_params.m_formatList.end(); ++it) { | 
|---|
| 164 | unsigned int format = *it; | 
|---|
| 165 | if (this->configureFormat(fd, format, width, height)==0) { | 
|---|
| 166 | // format has been set | 
|---|
| 167 | // get the format again because calling SET-FMT return a bad buffersize using v4l2loopback | 
|---|
| 168 | this->queryFormat(); | 
|---|
| 169 | return 0; | 
|---|
| 170 | } | 
|---|
| 171 | } | 
|---|
| 172 | return -1; | 
|---|
| 173 | } | 
|---|
| 174 |  | 
|---|
| 175 | // configure capture format | 
|---|
| 176 | int V4l2Device::configureFormat(int fd, unsigned int format, unsigned int width, unsigned int height) | 
|---|
| 177 | { | 
|---|
| 178 | struct v4l2_format   fmt; | 
|---|
| 179 | memset(&(fmt), 0, sizeof(fmt)); | 
|---|
| 180 | fmt.type                = m_deviceType; | 
|---|
| 181 | fmt.fmt.pix.width       = width; | 
|---|
| 182 | fmt.fmt.pix.height      = height; | 
|---|
| 183 | fmt.fmt.pix.pixelformat = format; | 
|---|
| 184 | fmt.fmt.pix.field       = V4L2_FIELD_ANY; | 
|---|
| 185 |  | 
|---|
| 186 | if (ioctl(fd, VIDIOC_S_FMT, &fmt) == -1) | 
|---|
| 187 | { | 
|---|
| 188 | LOG(ERROR) << "Cannot set format:"<< fourcc(format) << " for device:"<< m_params.m_devName << " "<< strerror(errno); | 
|---|
| 189 | return -1; | 
|---|
| 190 | } | 
|---|
| 191 | if (fmt.fmt.pix.pixelformat != format) | 
|---|
| 192 | { | 
|---|
| 193 | LOG(ERROR) << "Cannot set pixelformat to:"<< fourcc(format) << " format is:"<< fourcc(fmt.fmt.pix.pixelformat); | 
|---|
| 194 | return -1; | 
|---|
| 195 | } | 
|---|
| 196 | if ((fmt.fmt.pix.width != width) || (fmt.fmt.pix.height != height)) | 
|---|
| 197 | { | 
|---|
| 198 | LOG(WARN) << "Cannot set size to:"<< width << "x"<< height << " size is:"<< fmt.fmt.pix.width << "x"<< fmt.fmt.pix.height; | 
|---|
| 199 | } | 
|---|
| 200 |  | 
|---|
| 201 | m_format     = fmt.fmt.pix.pixelformat; | 
|---|
| 202 | m_width      = fmt.fmt.pix.width; | 
|---|
| 203 | m_height     = fmt.fmt.pix.height; | 
|---|
| 204 | m_bufferSize = fmt.fmt.pix.sizeimage; | 
|---|
| 205 |  | 
|---|
| 206 | LOG(NOTICE) << m_params.m_devName << ":"<< fourcc(m_format) << " size:"<< m_width << "x"<< m_height << " bufferSize:"<< m_bufferSize; | 
|---|
| 207 |  | 
|---|
| 208 | return 0; | 
|---|
| 209 | } | 
|---|
| 210 |  | 
|---|
| 211 | // configure capture FPS | 
|---|
| 212 | int V4l2Device::configureParam(int fd) | 
|---|
| 213 | { | 
|---|
| 214 | if (m_params.m_fps!=0) | 
|---|
| 215 | { | 
|---|
| 216 | struct v4l2_streamparm   param; | 
|---|
| 217 | memset(&(param), 0, sizeof(param)); | 
|---|
| 218 | param.type = m_deviceType; | 
|---|
| 219 | param.parm.capture.timeperframe.numerator = 1; | 
|---|
| 220 | param.parm.capture.timeperframe.denominator = m_params.m_fps; | 
|---|
| 221 |  | 
|---|
| 222 | if (ioctl(fd, VIDIOC_S_PARM, ¶m) == -1) | 
|---|
| 223 | { | 
|---|
| 224 | LOG(WARN) << "Cannot set param for device:"<< m_params.m_devName << " "<< strerror(errno); | 
|---|
| 225 | } | 
|---|
| 226 |  | 
|---|
| 227 | LOG(NOTICE) << "fps:"<< param.parm.capture.timeperframe.numerator << "/"<< param.parm.capture.timeperframe.denominator; | 
|---|
| 228 | LOG(NOTICE) << "nbBuffer:"<< param.parm.capture.readbuffers; | 
|---|
| 229 | } | 
|---|
| 230 |  | 
|---|
| 231 | return 0; | 
|---|
| 232 | } | 
|---|
| 233 |  | 
|---|
| 234 |  | 
|---|
| 235 |  | 
|---|