mirror of
https://github.com/hyperion-project/hyperion.ng.git
synced 2023-10-10 13:36:59 +02:00
Added setting of PAL/NTSC/NO_CHANGE and input
Former-commit-id: aee087b65b122e34e46b15854b94c720cbbec6b2
This commit is contained in:
parent
fdb140c1ae
commit
558ad6a11f
@ -19,62 +19,70 @@
|
|||||||
|
|
||||||
#define CLEAR(x) memset(&(x), 0, sizeof(x))
|
#define CLEAR(x) memset(&(x), 0, sizeof(x))
|
||||||
|
|
||||||
V4L2Grabber::V4L2Grabber() :
|
V4L2Grabber::V4L2Grabber(const std::string &device, int input, VideoStandard videoStandard, double fps) :
|
||||||
_deviceName("/dev/video0"),
|
_deviceName(device),
|
||||||
_ioMethod(IO_METHOD_MMAP),
|
_ioMethod(IO_METHOD_MMAP),
|
||||||
_fileDescriptor(-1),
|
_fileDescriptor(-1),
|
||||||
_buffers()
|
_buffers(),
|
||||||
|
_width(0),
|
||||||
|
_height(0)
|
||||||
{
|
{
|
||||||
|
open_device();
|
||||||
|
init_device(videoStandard, input);
|
||||||
}
|
}
|
||||||
|
|
||||||
V4L2Grabber::~V4L2Grabber()
|
V4L2Grabber::~V4L2Grabber()
|
||||||
{
|
{
|
||||||
|
uninit_device();
|
||||||
|
close_device();
|
||||||
}
|
}
|
||||||
|
|
||||||
void V4L2Grabber::start()
|
void V4L2Grabber::start()
|
||||||
{
|
{
|
||||||
open_device();
|
|
||||||
init_device();
|
|
||||||
start_capturing();
|
start_capturing();
|
||||||
|
}
|
||||||
|
|
||||||
int count = 100;
|
void V4L2Grabber::capture()
|
||||||
while (count-- > 0) {
|
{
|
||||||
for (;;) {
|
int count = 1;
|
||||||
fd_set fds;
|
while (count-- > 0) {
|
||||||
struct timeval tv;
|
for (;;) {
|
||||||
int r;
|
fd_set fds;
|
||||||
|
struct timeval tv;
|
||||||
|
int r;
|
||||||
|
|
||||||
FD_ZERO(&fds);
|
FD_ZERO(&fds);
|
||||||
FD_SET(_fileDescriptor, &fds);
|
FD_SET(_fileDescriptor, &fds);
|
||||||
|
|
||||||
/* Timeout. */
|
/* Timeout. */
|
||||||
tv.tv_sec = 2;
|
tv.tv_sec = 2;
|
||||||
tv.tv_usec = 0;
|
tv.tv_usec = 0;
|
||||||
|
|
||||||
r = select(_fileDescriptor + 1, &fds, NULL, NULL, &tv);
|
r = select(_fileDescriptor + 1, &fds, NULL, NULL, &tv);
|
||||||
|
|
||||||
if (-1 == r)
|
if (-1 == r)
|
||||||
{
|
{
|
||||||
if (EINTR == errno)
|
if (EINTR == errno)
|
||||||
continue;
|
continue;
|
||||||
errno_exit("select");
|
errno_exit("select");
|
||||||
}
|
}
|
||||||
|
|
||||||
if (0 == r)
|
if (0 == r)
|
||||||
{
|
{
|
||||||
fprintf(stderr, "select timeout\n");
|
fprintf(stderr, "select timeout\n");
|
||||||
exit(EXIT_FAILURE);
|
exit(EXIT_FAILURE);
|
||||||
}
|
}
|
||||||
|
|
||||||
if (read_frame())
|
if (read_frame())
|
||||||
break;
|
break;
|
||||||
/* EAGAIN - continue select loop. */
|
/* EAGAIN - continue select loop. */
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
}
|
||||||
|
|
||||||
stop_capturing();
|
void V4L2Grabber::stop()
|
||||||
uninit_device();
|
{
|
||||||
close_device();
|
stop_capturing();
|
||||||
}
|
}
|
||||||
|
|
||||||
void V4L2Grabber::open_device()
|
void V4L2Grabber::open_device()
|
||||||
@ -207,14 +215,10 @@ void V4L2Grabber::init_userp(unsigned int buffer_size)
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
void V4L2Grabber::init_device()
|
void V4L2Grabber::init_device(VideoStandard videoStandard, int input)
|
||||||
{
|
{
|
||||||
struct v4l2_capability cap;
|
struct v4l2_capability cap;
|
||||||
struct v4l2_cropcap cropcap;
|
if (-1 == xioctl(VIDIOC_QUERYCAP, &cap))
|
||||||
struct v4l2_crop crop;
|
|
||||||
struct v4l2_format fmt;
|
|
||||||
|
|
||||||
if (-1 == xioctl(VIDIOC_QUERYCAP, &cap))
|
|
||||||
{
|
{
|
||||||
if (EINVAL == errno) {
|
if (EINVAL == errno) {
|
||||||
fprintf(stderr, "%s is no V4L2 device\n", _deviceName.c_str());
|
fprintf(stderr, "%s is no V4L2 device\n", _deviceName.c_str());
|
||||||
@ -252,13 +256,14 @@ void V4L2Grabber::init_device()
|
|||||||
|
|
||||||
/* Select video input, video standard and tune here. */
|
/* Select video input, video standard and tune here. */
|
||||||
|
|
||||||
|
struct v4l2_cropcap cropcap;
|
||||||
CLEAR(cropcap);
|
CLEAR(cropcap);
|
||||||
|
|
||||||
cropcap.type = V4L2_BUF_TYPE_VIDEO_CAPTURE;
|
cropcap.type = V4L2_BUF_TYPE_VIDEO_CAPTURE;
|
||||||
|
|
||||||
if (0 == xioctl(VIDIOC_CROPCAP, &cropcap)) {
|
if (0 == xioctl(VIDIOC_CROPCAP, &cropcap)) {
|
||||||
crop.type = V4L2_BUF_TYPE_VIDEO_CAPTURE;
|
struct v4l2_crop crop;
|
||||||
|
crop.type = V4L2_BUF_TYPE_VIDEO_CAPTURE;
|
||||||
crop.c = cropcap.defrect; /* reset to default */
|
crop.c = cropcap.defrect; /* reset to default */
|
||||||
|
|
||||||
if (-1 == xioctl(VIDIOC_S_CROP, &crop)) {
|
if (-1 == xioctl(VIDIOC_S_CROP, &crop)) {
|
||||||
@ -275,24 +280,61 @@ void V4L2Grabber::init_device()
|
|||||||
/* Errors ignored. */
|
/* Errors ignored. */
|
||||||
}
|
}
|
||||||
|
|
||||||
|
// set input if needed
|
||||||
|
if (input >= 0)
|
||||||
|
{
|
||||||
|
if (-1 == xioctl(VIDIOC_S_INPUT, &input))
|
||||||
|
{
|
||||||
|
errno_exit("VIDIOC_S_INPUT");
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// set the video standard if needed
|
||||||
|
switch (videoStandard)
|
||||||
|
{
|
||||||
|
case PAL:
|
||||||
|
{
|
||||||
|
v4l2_std_id std_id = V4L2_STD_PAL;
|
||||||
|
if (-1 == xioctl(VIDIOC_S_STD, &std_id))
|
||||||
|
{
|
||||||
|
errno_exit("VIDIOC_S_STD");
|
||||||
|
}
|
||||||
|
}
|
||||||
|
break;
|
||||||
|
case NTSC:
|
||||||
|
{
|
||||||
|
v4l2_std_id std_id = V4L2_STD_NTSC;
|
||||||
|
if (-1 == xioctl(VIDIOC_S_STD, &std_id))
|
||||||
|
{
|
||||||
|
errno_exit("VIDIOC_S_STD");
|
||||||
|
}
|
||||||
|
}
|
||||||
|
break;
|
||||||
|
case NO_CHANGE:
|
||||||
|
default:
|
||||||
|
// No change to device settings
|
||||||
|
break;
|
||||||
|
}
|
||||||
|
|
||||||
|
// get the current settings
|
||||||
|
struct v4l2_format fmt;
|
||||||
CLEAR(fmt);
|
CLEAR(fmt);
|
||||||
|
|
||||||
fmt.type = V4L2_BUF_TYPE_VIDEO_CAPTURE;
|
fmt.type = V4L2_BUF_TYPE_VIDEO_CAPTURE;
|
||||||
|
|
||||||
/* Preserve original settings as set by v4l2-ctl for example */
|
|
||||||
if (-1 == xioctl(VIDIOC_G_FMT, &fmt))
|
if (-1 == xioctl(VIDIOC_G_FMT, &fmt))
|
||||||
{
|
{
|
||||||
errno_exit("VIDIOC_G_FMT");
|
errno_exit("VIDIOC_G_FMT");
|
||||||
}
|
}
|
||||||
|
|
||||||
// /* Buggy driver paranoia. */
|
// check pixel format
|
||||||
// min = fmt.fmt.pix.width * 2;
|
if (fmt.fmt.pix.pixelformat != V4L2_PIX_FMT_UYVY)
|
||||||
// if (fmt.fmt.pix.bytesperline < min)
|
{
|
||||||
// fmt.fmt.pix.bytesperline = min;
|
exit(EXIT_FAILURE);
|
||||||
// min = fmt.fmt.pix.bytesperline * fmt.fmt.pix.height;
|
}
|
||||||
// if (fmt.fmt.pix.sizeimage < min)
|
|
||||||
// fmt.fmt.pix.sizeimage = min;
|
// store width & height
|
||||||
|
_width = fmt.fmt.pix.width;
|
||||||
|
_height = fmt.fmt.pix.height;
|
||||||
|
std::cout << "V4L2 width=" << _width << " height=" << _height << std::endl;
|
||||||
|
|
||||||
switch (_ioMethod) {
|
switch (_ioMethod) {
|
||||||
case IO_METHOD_READ:
|
case IO_METHOD_READ:
|
||||||
@ -403,8 +445,11 @@ int V4L2Grabber::read_frame()
|
|||||||
|
|
||||||
switch (_ioMethod) {
|
switch (_ioMethod) {
|
||||||
case IO_METHOD_READ:
|
case IO_METHOD_READ:
|
||||||
if (-1 == read(_fileDescriptor, _buffers[0].start, _buffers[0].length)) {
|
int size;
|
||||||
switch (errno) {
|
if ((size = read(_fileDescriptor, _buffers[0].start, _buffers[0].length)) == -1)
|
||||||
|
{
|
||||||
|
switch (errno)
|
||||||
|
{
|
||||||
case EAGAIN:
|
case EAGAIN:
|
||||||
return 0;
|
return 0;
|
||||||
|
|
||||||
@ -418,7 +463,7 @@ int V4L2Grabber::read_frame()
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
process_image(_buffers[0].start, _buffers[0].length);
|
process_image(_buffers[0].start, size);
|
||||||
break;
|
break;
|
||||||
|
|
||||||
case IO_METHOD_MMAP:
|
case IO_METHOD_MMAP:
|
||||||
@ -427,8 +472,10 @@ int V4L2Grabber::read_frame()
|
|||||||
buf.type = V4L2_BUF_TYPE_VIDEO_CAPTURE;
|
buf.type = V4L2_BUF_TYPE_VIDEO_CAPTURE;
|
||||||
buf.memory = V4L2_MEMORY_MMAP;
|
buf.memory = V4L2_MEMORY_MMAP;
|
||||||
|
|
||||||
if (-1 == xioctl(VIDIOC_DQBUF, &buf)) {
|
if (-1 == xioctl(VIDIOC_DQBUF, &buf))
|
||||||
switch (errno) {
|
{
|
||||||
|
switch (errno)
|
||||||
|
{
|
||||||
case EAGAIN:
|
case EAGAIN:
|
||||||
return 0;
|
return 0;
|
||||||
|
|
||||||
@ -444,10 +491,13 @@ int V4L2Grabber::read_frame()
|
|||||||
|
|
||||||
assert(buf.index < _buffers.size());
|
assert(buf.index < _buffers.size());
|
||||||
|
|
||||||
process_image(_buffers[buf.index].start, buf.bytesused);
|
process_image(_buffers[buf.index].start, buf.bytesused);
|
||||||
|
|
||||||
if (-1 == xioctl(VIDIOC_QBUF, &buf))
|
if (-1 == xioctl(VIDIOC_QBUF, &buf))
|
||||||
|
{
|
||||||
errno_exit("VIDIOC_QBUF");
|
errno_exit("VIDIOC_QBUF");
|
||||||
|
}
|
||||||
|
|
||||||
break;
|
break;
|
||||||
|
|
||||||
case IO_METHOD_USERPTR:
|
case IO_METHOD_USERPTR:
|
||||||
@ -456,8 +506,10 @@ int V4L2Grabber::read_frame()
|
|||||||
buf.type = V4L2_BUF_TYPE_VIDEO_CAPTURE;
|
buf.type = V4L2_BUF_TYPE_VIDEO_CAPTURE;
|
||||||
buf.memory = V4L2_MEMORY_USERPTR;
|
buf.memory = V4L2_MEMORY_USERPTR;
|
||||||
|
|
||||||
if (-1 == xioctl(VIDIOC_DQBUF, &buf)) {
|
if (-1 == xioctl(VIDIOC_DQBUF, &buf))
|
||||||
switch (errno) {
|
{
|
||||||
|
switch (errno)
|
||||||
|
{
|
||||||
case EAGAIN:
|
case EAGAIN:
|
||||||
return 0;
|
return 0;
|
||||||
|
|
||||||
@ -472,14 +524,19 @@ int V4L2Grabber::read_frame()
|
|||||||
}
|
}
|
||||||
|
|
||||||
for (size_t i = 0; i < _buffers.size(); ++i)
|
for (size_t i = 0; i < _buffers.size(); ++i)
|
||||||
if (buf.m.userptr == (unsigned long)_buffers[i].start
|
{
|
||||||
&& buf.length == _buffers[i].length)
|
if (buf.m.userptr == (unsigned long)_buffers[i].start && buf.length == _buffers[i].length)
|
||||||
|
{
|
||||||
break;
|
break;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
process_image((void *)buf.m.userptr, buf.bytesused);
|
process_image((void *)buf.m.userptr, buf.bytesused);
|
||||||
|
|
||||||
if (-1 == xioctl(VIDIOC_QBUF, &buf))
|
if (-1 == xioctl(VIDIOC_QBUF, &buf))
|
||||||
|
{
|
||||||
errno_exit("VIDIOC_QBUF");
|
errno_exit("VIDIOC_QBUF");
|
||||||
|
}
|
||||||
break;
|
break;
|
||||||
}
|
}
|
||||||
|
|
||||||
@ -488,38 +545,44 @@ int V4L2Grabber::read_frame()
|
|||||||
|
|
||||||
void V4L2Grabber::process_image(const void *p, int size)
|
void V4L2Grabber::process_image(const void *p, int size)
|
||||||
{
|
{
|
||||||
if (size != 2*720*480)
|
if (size != 2*_width*_height)
|
||||||
{
|
{
|
||||||
std::cout << "Frame too small: " << size << "<" << (2*720*480) << std::endl;
|
std::cout << "Frame too small: " << size << " != " << (2*_width*_height) << std::endl;
|
||||||
return;
|
}
|
||||||
}
|
else
|
||||||
|
{
|
||||||
|
process_image(reinterpret_cast<const uint8_t *>(p));
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
std::cout << "process image of size = " << size << std::endl;
|
void V4L2Grabber::process_image(const uint8_t * data)
|
||||||
|
{
|
||||||
|
std::cout << "process image" << std::endl;
|
||||||
|
|
||||||
const uint8_t * data = reinterpret_cast<const uint8_t *>(p);
|
QImage image(_width, _height, QImage::Format_RGB888);
|
||||||
|
|
||||||
|
|
||||||
QImage image(720, 480, QImage::Format_RGB888);
|
|
||||||
|
|
||||||
for (int y = 0; y < image.height(); ++y)
|
for (int y = 0; y < image.height(); ++y)
|
||||||
{
|
{
|
||||||
for (int x = 0; x < image.width(); ++x)
|
for (int x = 0; x < image.width(); ++x)
|
||||||
{
|
{
|
||||||
uint8_t value = data[(720 * y + x) * 2 + 1];
|
uint8_t value = data[(image.width() * y + x) * 2 + 1];
|
||||||
image.setPixel(x, y, qRgb(value, value, value));
|
//std::cout << "data = " << int(value) << std::endl;
|
||||||
|
image.setPixel(x, y, qRgb(value, value, value));
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
image.save("/home/pi/screenshot.png");
|
image.save("screenshot.png");
|
||||||
}
|
}
|
||||||
|
|
||||||
int V4L2Grabber::xioctl(int request, void *arg)
|
int V4L2Grabber::xioctl(int request, void *arg)
|
||||||
{
|
{
|
||||||
int r;
|
int r;
|
||||||
|
|
||||||
do {
|
do
|
||||||
r = ioctl(_fileDescriptor, request, arg);
|
{
|
||||||
} while (-1 == r && EINTR == errno);
|
r = ioctl(_fileDescriptor, request, arg);
|
||||||
|
}
|
||||||
|
while (-1 == r && EINTR == errno);
|
||||||
|
|
||||||
return r;
|
return r;
|
||||||
}
|
}
|
||||||
|
@ -10,11 +10,20 @@
|
|||||||
class V4L2Grabber
|
class V4L2Grabber
|
||||||
{
|
{
|
||||||
public:
|
public:
|
||||||
V4L2Grabber();
|
enum VideoStandard {
|
||||||
|
PAL, NTSC, NO_CHANGE
|
||||||
|
};
|
||||||
|
|
||||||
|
public:
|
||||||
|
V4L2Grabber(const std::string & device, int input, VideoStandard videoStandard, double fps);
|
||||||
virtual ~V4L2Grabber();
|
virtual ~V4L2Grabber();
|
||||||
|
|
||||||
void start();
|
void start();
|
||||||
|
|
||||||
|
void capture();
|
||||||
|
|
||||||
|
void stop();
|
||||||
|
|
||||||
private:
|
private:
|
||||||
void open_device();
|
void open_device();
|
||||||
|
|
||||||
@ -26,7 +35,7 @@ private:
|
|||||||
|
|
||||||
void init_userp(unsigned int buffer_size);
|
void init_userp(unsigned int buffer_size);
|
||||||
|
|
||||||
void init_device();
|
void init_device(VideoStandard videoStandard, int input);
|
||||||
|
|
||||||
void uninit_device();
|
void uninit_device();
|
||||||
|
|
||||||
@ -36,7 +45,9 @@ private:
|
|||||||
|
|
||||||
int read_frame();
|
int read_frame();
|
||||||
|
|
||||||
void process_image(const void *p, int size);
|
void process_image(const void *p, int size);
|
||||||
|
|
||||||
|
void process_image(const uint8_t *p);
|
||||||
|
|
||||||
int xioctl(int request, void *arg);
|
int xioctl(int request, void *arg);
|
||||||
|
|
||||||
@ -59,4 +70,7 @@ private:
|
|||||||
const io_method _ioMethod;
|
const io_method _ioMethod;
|
||||||
int _fileDescriptor;
|
int _fileDescriptor;
|
||||||
std::vector<buffer> _buffers;
|
std::vector<buffer> _buffers;
|
||||||
|
|
||||||
|
int _width;
|
||||||
|
int _height;
|
||||||
};
|
};
|
||||||
|
@ -40,9 +40,10 @@ int main(int argc, char** argv)
|
|||||||
return 1;
|
return 1;
|
||||||
}
|
}
|
||||||
|
|
||||||
V4L2Grabber grabber;
|
V4L2Grabber grabber("/dev/video0", 0, V4L2Grabber::PAL, 10.0);
|
||||||
grabber.start();
|
grabber.start();
|
||||||
|
grabber.capture();
|
||||||
|
grabber.stop();
|
||||||
|
|
||||||
return 0;
|
return 0;
|
||||||
}
|
}
|
||||||
|
|
||||||
|
Loading…
Reference in New Issue
Block a user