GST_FLOW_WRONG_STATE -> GST_FLOW_FLUSHING
[platform/upstream/gstreamer.git] / sys / v4l2 / gstv4l2bufferpool.c
1 /* GStreamer
2  *
3  * Copyright (C) 2001-2002 Ronald Bultje <rbultje@ronald.bitfreak.net>
4  *               2006 Edgard Lima <edgard.lima@indt.org.br>
5  *               2009 Texas Instruments, Inc - http://www.ti.com/
6  *
7  * gstv4l2bufferpool.c V4L2 buffer pool class
8  *
9  * This library is free software; you can redistribute it and/or
10  * modify it under the terms of the GNU Library General Public
11  * License as published by the Free Software Foundation; either
12  * version 2 of the License, or (at your option) any later version.
13  *
14  * This library is distributed in the hope that it will be useful,
15  * but WITHOUT ANY WARRANTY; without even the implied warranty of
16  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
17  * Library General Public License for more details.
18  *
19  * You should have received a copy of the GNU Library General Public
20  * License along with this library; if not, write to the
21  * Free Software Foundation, Inc., 59 Temple Place - Suite 330,
22  * Boston, MA 02111-1307, USA.
23  */
24
25 #ifdef HAVE_CONFIG_H
26 #  include <config.h>
27 #endif
28
29 #include <sys/mman.h>
30 #include <string.h>
31 #include <unistd.h>
32
33 #include "gst/video/video.h"
34 #include "gst/video/gstvideometa.h"
35 #include "gst/video/gstvideopool.h"
36
37 #include <gstv4l2bufferpool.h>
38
39 #include "gstv4l2src.h"
40 #include "gstv4l2sink.h"
41 #include "v4l2_calls.h"
42 #include "gst/gst-i18n-plugin.h"
43 #include <gst/glib-compat-private.h>
44
45 /* videodev2.h is not versioned and we can't easily check for the presence
46  * of enum values at compile time, but the V4L2_CAP_VIDEO_OUTPUT_OVERLAY define
47  * was added in the same commit as V4L2_FIELD_INTERLACED_{TB,BT} (b2787845) */
48 #ifndef V4L2_CAP_VIDEO_OUTPUT_OVERLAY
49 #define V4L2_FIELD_INTERLACED_TB 8
50 #define V4L2_FIELD_INTERLACED_BT 9
51 #endif
52
53
54 GST_DEBUG_CATEGORY_EXTERN (v4l2_debug);
55 #define GST_CAT_DEFAULT v4l2_debug
56
57 /*
58  * GstV4l2Buffer:
59  */
60 const GstMetaInfo *
61 gst_v4l2_meta_get_info (void)
62 {
63   static const GstMetaInfo *meta_info = NULL;
64
65   if (meta_info == NULL) {
66     meta_info =
67         gst_meta_register ("GstV4l2Meta", "GstV4l2Meta",
68         sizeof (GstV4l2Meta), (GstMetaInitFunction) NULL,
69         (GstMetaFreeFunction) NULL, (GstMetaCopyFunction) NULL,
70         (GstMetaTransformFunction) NULL);
71   }
72   return meta_info;
73 }
74
75 /*
76  * GstV4l2BufferPool:
77  */
78 #define gst_v4l2_buffer_pool_parent_class parent_class
79 G_DEFINE_TYPE (GstV4l2BufferPool, gst_v4l2_buffer_pool, GST_TYPE_BUFFER_POOL);
80
81 static void gst_v4l2_buffer_pool_release_buffer (GstBufferPool * bpool,
82     GstBuffer * buffer);
83
84 static void
85 gst_v4l2_buffer_pool_free_buffer (GstBufferPool * bpool, GstBuffer * buffer)
86 {
87   GstV4l2BufferPool *pool = GST_V4L2_BUFFER_POOL (bpool);
88   GstV4l2Object *obj;
89
90   obj = pool->obj;
91
92   switch (obj->mode) {
93     case GST_V4L2_IO_RW:
94       break;
95     case GST_V4L2_IO_MMAP:
96     {
97       GstV4l2Meta *meta;
98       gint index;
99
100       meta = GST_V4L2_META_GET (buffer);
101       g_assert (meta != NULL);
102
103       index = meta->vbuffer.index;
104       GST_LOG_OBJECT (pool,
105           "mmap buffer %p idx %d (data %p, len %u) freed, unmapping", buffer,
106           index, meta->mem, meta->vbuffer.length);
107
108       v4l2_munmap (meta->mem, meta->vbuffer.length);
109       pool->buffers[index] = NULL;
110       break;
111     }
112     case GST_V4L2_IO_USERPTR:
113     default:
114       g_assert_not_reached ();
115       break;
116   }
117   gst_buffer_unref (buffer);
118 }
119
120 static GstFlowReturn
121 gst_v4l2_buffer_pool_alloc_buffer (GstBufferPool * bpool, GstBuffer ** buffer,
122     GstBufferPoolParams * params)
123 {
124   GstV4l2BufferPool *pool = GST_V4L2_BUFFER_POOL (bpool);
125   GstBuffer *newbuf;
126   GstV4l2Meta *meta;
127   GstV4l2Object *obj;
128   GstVideoInfo *info;
129   guint index;
130
131   obj = pool->obj;
132   info = &obj->info;
133
134   switch (obj->mode) {
135     case GST_V4L2_IO_RW:
136     {
137       newbuf =
138           gst_buffer_new_allocate (pool->allocator, pool->size, pool->align);
139       break;
140     }
141     case GST_V4L2_IO_MMAP:
142     {
143       newbuf = gst_buffer_new ();
144       meta = GST_V4L2_META_ADD (newbuf);
145
146       index = pool->num_allocated;
147
148       GST_LOG_OBJECT (pool, "creating buffer %u, %p", index, newbuf);
149
150       meta->vbuffer.index = index;
151       meta->vbuffer.type = obj->type;
152       meta->vbuffer.memory = V4L2_MEMORY_MMAP;
153
154       if (v4l2_ioctl (pool->video_fd, VIDIOC_QUERYBUF, &meta->vbuffer) < 0)
155         goto querybuf_failed;
156
157       GST_LOG_OBJECT (pool, "  index:     %u", meta->vbuffer.index);
158       GST_LOG_OBJECT (pool, "  type:      %d", meta->vbuffer.type);
159       GST_LOG_OBJECT (pool, "  bytesused: %u", meta->vbuffer.bytesused);
160       GST_LOG_OBJECT (pool, "  flags:     %08x", meta->vbuffer.flags);
161       GST_LOG_OBJECT (pool, "  field:     %d", meta->vbuffer.field);
162       GST_LOG_OBJECT (pool, "  memory:    %d", meta->vbuffer.memory);
163       if (meta->vbuffer.memory == V4L2_MEMORY_MMAP)
164         GST_LOG_OBJECT (pool, "  MMAP offset:  %u", meta->vbuffer.m.offset);
165       GST_LOG_OBJECT (pool, "  length:    %u", meta->vbuffer.length);
166       GST_LOG_OBJECT (pool, "  input:     %u", meta->vbuffer.input);
167
168       meta->mem = v4l2_mmap (0, meta->vbuffer.length,
169           PROT_READ | PROT_WRITE, MAP_SHARED, pool->video_fd,
170           meta->vbuffer.m.offset);
171       if (meta->mem == MAP_FAILED)
172         goto mmap_failed;
173
174       gst_buffer_take_memory (newbuf, -1,
175           gst_memory_new_wrapped (0,
176               meta->mem, NULL, meta->vbuffer.length, 0, meta->vbuffer.length));
177
178       /* add metadata to raw video buffers */
179       if (pool->add_videometa && info->finfo) {
180         gsize offset[GST_VIDEO_MAX_PLANES];
181         gint stride[GST_VIDEO_MAX_PLANES];
182
183         offset[0] = 0;
184         stride[0] = obj->bytesperline;
185
186         GST_DEBUG_OBJECT (pool, "adding video meta, stride %d", stride[0]);
187         gst_buffer_add_video_meta_full (newbuf, info->flags,
188             GST_VIDEO_INFO_FORMAT (info), GST_VIDEO_INFO_WIDTH (info),
189             GST_VIDEO_INFO_HEIGHT (info), GST_VIDEO_INFO_N_PLANES (info),
190             offset, stride);
191       }
192       break;
193     }
194     case GST_V4L2_IO_USERPTR:
195     default:
196       g_assert_not_reached ();
197       break;
198   }
199
200   pool->num_allocated++;
201
202   *buffer = newbuf;
203
204   return GST_FLOW_OK;
205
206   /* ERRORS */
207 querybuf_failed:
208   {
209     gint errnosave = errno;
210
211     GST_WARNING ("Failed QUERYBUF: %s", g_strerror (errnosave));
212     gst_buffer_unref (newbuf);
213     errno = errnosave;
214     return GST_FLOW_ERROR;
215   }
216 mmap_failed:
217   {
218     gint errnosave = errno;
219
220     GST_WARNING ("Failed to mmap: %s", g_strerror (errnosave));
221     gst_buffer_unref (newbuf);
222     errno = errnosave;
223     return GST_FLOW_ERROR;
224   }
225 }
226
227 static gboolean
228 gst_v4l2_buffer_pool_set_config (GstBufferPool * bpool, GstStructure * config)
229 {
230   GstV4l2BufferPool *pool = GST_V4L2_BUFFER_POOL (bpool);
231   GstV4l2Object *obj = pool->obj;
232   const GstCaps *caps;
233   guint size, min_buffers, max_buffers;
234   guint prefix, align;
235
236   GST_DEBUG_OBJECT (pool, "set config");
237
238   pool->add_videometa =
239       gst_buffer_pool_config_has_option (config,
240       GST_BUFFER_POOL_OPTION_VIDEO_META);
241
242   if (!pool->add_videometa) {
243     gint stride;
244
245     /* we don't have video metadata, see if the strides are compatible */
246     stride = GST_VIDEO_INFO_PLANE_STRIDE (&obj->info, 0);
247
248     GST_DEBUG_OBJECT (pool, "no videometadata, checking strides %d and %u",
249         stride, obj->bytesperline);
250
251     if (stride != obj->bytesperline)
252       goto missing_video_api;
253   }
254
255   /* parse the config and keep around */
256   if (!gst_buffer_pool_config_get (config, &caps, &size, &min_buffers,
257           &max_buffers, &prefix, &align))
258     goto wrong_config;
259
260   GST_DEBUG_OBJECT (pool, "config %" GST_PTR_FORMAT, config);
261
262   pool->size = size;
263   pool->max_buffers = MAX (min_buffers, max_buffers);
264   pool->min_buffers = MIN (pool->max_buffers, min_buffers);
265   pool->prefix = prefix;
266   pool->align = align;
267
268   gst_buffer_pool_config_set (config, caps, size, min_buffers,
269       max_buffers, prefix, align);
270
271   return GST_BUFFER_POOL_CLASS (parent_class)->set_config (bpool, config);
272
273   /* ERRORS */
274 missing_video_api:
275   {
276     GST_ERROR_OBJECT (pool, "missing GstMetaVideo API in config, "
277         "default stride: %d, wanted stride %u",
278         GST_VIDEO_INFO_PLANE_STRIDE (&obj->info, 0), obj->bytesperline);
279     return FALSE;
280   }
281 wrong_config:
282   {
283     GST_ERROR_OBJECT (pool, "invalid config %" GST_PTR_FORMAT, config);
284     return FALSE;
285   }
286 }
287
288 static gboolean
289 start_streaming (GstV4l2BufferPool * pool)
290 {
291   GstV4l2Object *obj = pool->obj;
292
293   switch (obj->mode) {
294     case GST_V4L2_IO_RW:
295       break;
296     case GST_V4L2_IO_MMAP:
297     case GST_V4L2_IO_USERPTR:
298       GST_DEBUG_OBJECT (pool, "STREAMON");
299       if (v4l2_ioctl (pool->video_fd, VIDIOC_STREAMON, &obj->type) < 0)
300         goto start_failed;
301       break;
302     default:
303       g_assert_not_reached ();
304       break;
305   }
306
307   pool->streaming = TRUE;
308
309   return TRUE;
310
311   /* ERRORS */
312 start_failed:
313   {
314     GST_ERROR_OBJECT (pool, "error with STREAMON %d (%s)", errno,
315         g_strerror (errno));
316     return FALSE;
317   }
318 }
319
320 static gboolean
321 gst_v4l2_buffer_pool_start (GstBufferPool * bpool)
322 {
323   GstV4l2BufferPool *pool = GST_V4L2_BUFFER_POOL (bpool);
324   GstV4l2Object *obj = pool->obj;
325   gint n;
326   struct v4l2_requestbuffers breq;
327   gint min_buffers, max_buffers;
328
329   min_buffers = pool->min_buffers;
330   max_buffers = pool->max_buffers;
331
332   switch (obj->mode) {
333     case GST_V4L2_IO_RW:
334     {
335       break;
336     }
337     case GST_V4L2_IO_MMAP:
338     {
339       /* first, lets request buffers, and see how many we can get: */
340       GST_DEBUG_OBJECT (pool, "starting, requesting %d MMAP buffers",
341           max_buffers);
342
343       if (max_buffers == 0)
344         max_buffers = 4;
345
346       memset (&breq, 0, sizeof (struct v4l2_requestbuffers));
347       breq.type = obj->type;
348       breq.count = max_buffers;
349       breq.memory = V4L2_MEMORY_MMAP;
350
351       if (v4l2_ioctl (pool->video_fd, VIDIOC_REQBUFS, &breq) < 0)
352         goto reqbufs_failed;
353
354       GST_LOG_OBJECT (pool, " count:  %u", breq.count);
355       GST_LOG_OBJECT (pool, " type:   %d", breq.type);
356       GST_LOG_OBJECT (pool, " memory: %d", breq.memory);
357
358       if (breq.count < GST_V4L2_MIN_BUFFERS)
359         goto no_buffers;
360
361       if (max_buffers != breq.count) {
362         GST_WARNING_OBJECT (pool, "using %u buffers instead", breq.count);
363         max_buffers = breq.count;
364       }
365       break;
366     }
367     case GST_V4L2_IO_USERPTR:
368     default:
369       g_assert_not_reached ();
370       break;
371   }
372
373   pool->obj = obj;
374   pool->max_buffers = max_buffers;
375   pool->buffers = g_new0 (GstBuffer *, max_buffers);
376   pool->num_allocated = 0;
377
378   /* now, allocate the buffers: */
379   for (n = 0; n < min_buffers; n++) {
380     GstBuffer *buffer;
381
382     if (gst_v4l2_buffer_pool_alloc_buffer (bpool, &buffer, NULL) != GST_FLOW_OK)
383       goto buffer_new_failed;
384
385     gst_v4l2_buffer_pool_release_buffer (bpool, buffer);
386   }
387
388   /* we can start capturing now, we wait for the playback case until we queued
389    * the first buffer */
390   if (obj->type == V4L2_BUF_TYPE_VIDEO_CAPTURE)
391     if (!start_streaming (pool))
392       goto start_failed;
393
394   gst_poll_set_flushing (obj->poll, FALSE);
395
396   return TRUE;
397
398   /* ERRORS */
399 reqbufs_failed:
400   {
401     GST_ERROR_OBJECT (pool,
402         "error requesting %d buffers: %s", max_buffers, g_strerror (errno));
403     return FALSE;
404   }
405 no_buffers:
406   {
407     GST_ERROR_OBJECT (pool,
408         "we received %d from device '%s', we want at least %d",
409         breq.count, obj->videodev, GST_V4L2_MIN_BUFFERS);
410     return FALSE;
411   }
412 buffer_new_failed:
413   {
414     GST_ERROR_OBJECT (pool, "failed to create a buffer");
415     return FALSE;
416   }
417 start_failed:
418   {
419     GST_ERROR_OBJECT (pool, "failed to start streaming");
420     return FALSE;
421   }
422 }
423
424 static gboolean
425 gst_v4l2_buffer_pool_stop (GstBufferPool * bpool)
426 {
427   gboolean ret;
428   GstV4l2BufferPool *pool = GST_V4L2_BUFFER_POOL (bpool);
429   GstV4l2Object *obj = pool->obj;
430   guint n;
431
432   GST_DEBUG_OBJECT (pool, "stopping pool");
433
434   gst_poll_set_flushing (obj->poll, TRUE);
435
436   if (pool->streaming) {
437     switch (obj->mode) {
438       case GST_V4L2_IO_RW:
439         break;
440       case GST_V4L2_IO_MMAP:
441       case GST_V4L2_IO_USERPTR:
442         /* we actually need to sync on all queued buffers but not
443          * on the non-queued ones */
444         GST_DEBUG_OBJECT (pool, "STREAMOFF");
445         if (v4l2_ioctl (pool->video_fd, VIDIOC_STREAMOFF, &obj->type) < 0)
446           goto stop_failed;
447         break;
448       default:
449         g_assert_not_reached ();
450         break;
451     }
452     pool->streaming = FALSE;
453   }
454
455   /* first free the buffers in the queue */
456   ret = GST_BUFFER_POOL_CLASS (parent_class)->stop (bpool);
457
458   /* then free the remaining buffers */
459   for (n = 0; n < pool->num_allocated; n++) {
460     if (pool->buffers[n])
461       gst_v4l2_buffer_pool_free_buffer (bpool, pool->buffers[n]);
462   }
463   g_free (pool->buffers);
464   pool->buffers = NULL;
465
466   return ret;
467
468   /* ERRORS */
469 stop_failed:
470   {
471     GST_ERROR_OBJECT (pool, "error with STREAMOFF %d (%s)", errno,
472         g_strerror (errno));
473     return FALSE;
474   }
475 }
476
477 static GstFlowReturn
478 gst_v4l2_object_poll (GstV4l2Object * v4l2object)
479 {
480   gint ret;
481
482   if (v4l2object->can_poll_device) {
483     GST_LOG_OBJECT (v4l2object->element, "polling device");
484     ret = gst_poll_wait (v4l2object->poll, GST_CLOCK_TIME_NONE);
485     if (G_UNLIKELY (ret < 0)) {
486       if (errno == EBUSY)
487         goto stopped;
488       if (errno == ENXIO) {
489         GST_WARNING_OBJECT (v4l2object->element,
490             "v4l2 device doesn't support polling. Disabling");
491         v4l2object->can_poll_device = FALSE;
492       } else {
493         if (errno != EAGAIN && errno != EINTR)
494           goto select_error;
495       }
496     }
497   }
498   return GST_FLOW_OK;
499
500   /* ERRORS */
501 stopped:
502   {
503     GST_DEBUG ("stop called");
504     return GST_FLOW_FLUSHING;
505   }
506 select_error:
507   {
508     GST_ELEMENT_ERROR (v4l2object->element, RESOURCE, READ, (NULL),
509         ("poll error %d: %s (%d)", ret, g_strerror (errno), errno));
510     return GST_FLOW_ERROR;
511   }
512 }
513
514 static GstFlowReturn
515 gst_v4l2_buffer_pool_qbuf (GstV4l2BufferPool * pool, GstBuffer * buf)
516 {
517   GstV4l2Meta *meta;
518   gint index;
519
520   meta = GST_V4L2_META_GET (buf);
521   g_assert (meta != NULL);
522
523   index = meta->vbuffer.index;
524
525   GST_LOG_OBJECT (pool, "enqueue buffer %p, index:%d, queued:%d", buf,
526       index, pool->num_queued);
527
528   if (pool->buffers[index] != NULL)
529     goto already_queued;
530
531   if (v4l2_ioctl (pool->video_fd, VIDIOC_QBUF, &meta->vbuffer) < 0)
532     goto queue_failed;
533
534   pool->buffers[index] = buf;
535   pool->num_queued++;
536
537   return GST_FLOW_OK;
538
539   /* ERRORS */
540 already_queued:
541   {
542     GST_WARNING_OBJECT (pool, "the buffer was already queued");
543     return GST_FLOW_ERROR;
544   }
545 queue_failed:
546   {
547     GST_WARNING_OBJECT (pool, "could not queue a buffer %d (%s)", errno,
548         g_strerror (errno));
549     return GST_FLOW_ERROR;
550   }
551 }
552
553 static GstFlowReturn
554 gst_v4l2_buffer_pool_dqbuf (GstV4l2BufferPool * pool, GstBuffer ** buffer)
555 {
556   GstFlowReturn res;
557   GstBuffer *outbuf;
558   struct v4l2_buffer vbuffer;
559   GstV4l2Object *obj = pool->obj;
560
561   if (obj->type == V4L2_BUF_TYPE_VIDEO_CAPTURE) {
562     /* select works for input devices when data is available. According to the
563      * specs we can also poll to find out when a frame has been displayed but
564      * that just seems to lock up here */
565     if ((res = gst_v4l2_object_poll (obj)) != GST_FLOW_OK)
566       goto poll_error;
567   }
568
569   memset (&vbuffer, 0x00, sizeof (vbuffer));
570   vbuffer.type = obj->type;
571   vbuffer.memory = V4L2_MEMORY_MMAP;
572
573   GST_LOG_OBJECT (pool, "doing DQBUF");
574   if (v4l2_ioctl (pool->video_fd, VIDIOC_DQBUF, &vbuffer) < 0)
575     goto error;
576
577   /* get our GstBuffer with that index from the pool, if the buffer was
578    * outstanding we have a serious problem.
579    */
580   outbuf = pool->buffers[vbuffer.index];
581   if (outbuf == NULL)
582     goto no_buffer;
583
584   /* mark the buffer outstanding */
585   pool->buffers[vbuffer.index] = NULL;
586   pool->num_queued--;
587
588   GST_LOG_OBJECT (pool,
589       "dequeued buffer %p seq:%d (ix=%d), used %d, flags %08x, pool-queued=%d, buffer=%p",
590       outbuf, vbuffer.sequence, vbuffer.index, vbuffer.bytesused, vbuffer.flags,
591       pool->num_queued, outbuf);
592
593   /* set top/bottom field first if v4l2_buffer has the information */
594   if (vbuffer.field == V4L2_FIELD_INTERLACED_TB) {
595     GST_BUFFER_FLAG_SET (outbuf, GST_VIDEO_BUFFER_FLAG_TFF);
596     GST_BUFFER_FLAG_SET (outbuf, GST_VIDEO_BUFFER_FLAG_INTERLACED);
597   }
598   if (vbuffer.field == V4L2_FIELD_INTERLACED_BT) {
599     GST_BUFFER_FLAG_UNSET (outbuf, GST_VIDEO_BUFFER_FLAG_TFF);
600     GST_BUFFER_FLAG_SET (outbuf, GST_VIDEO_BUFFER_FLAG_INTERLACED);
601   }
602
603   /* this can change at every frame, esp. with jpeg */
604   if (obj->type == V4L2_BUF_TYPE_VIDEO_CAPTURE)
605     gst_buffer_resize (outbuf, 0, vbuffer.bytesused);
606   else
607     gst_buffer_resize (outbuf, 0, vbuffer.length);
608
609   *buffer = outbuf;
610
611   return GST_FLOW_OK;
612
613   /* ERRORS */
614 poll_error:
615   {
616     GST_DEBUG_OBJECT (pool, "poll error %s", gst_flow_get_name (res));
617     return res;
618   }
619 error:
620   {
621     GST_WARNING_OBJECT (pool,
622         "problem dequeuing frame %d (ix=%d), pool-ct=%d, buf.flags=%d",
623         vbuffer.sequence, vbuffer.index,
624         GST_MINI_OBJECT_REFCOUNT (pool), vbuffer.flags);
625
626     switch (errno) {
627       case EAGAIN:
628         GST_WARNING_OBJECT (pool,
629             "Non-blocking I/O has been selected using O_NONBLOCK and"
630             " no buffer was in the outgoing queue. device %s", obj->videodev);
631         break;
632       case EINVAL:
633         GST_ERROR_OBJECT (pool,
634             "The buffer type is not supported, or the index is out of bounds, "
635             "or no buffers have been allocated yet, or the userptr "
636             "or length are invalid. device %s", obj->videodev);
637         break;
638       case ENOMEM:
639         GST_ERROR_OBJECT (pool,
640             "insufficient memory to enqueue a user pointer buffer");
641         break;
642       case EIO:
643         GST_INFO_OBJECT (pool,
644             "VIDIOC_DQBUF failed due to an internal error."
645             " Can also indicate temporary problems like signal loss."
646             " Note the driver might dequeue an (empty) buffer despite"
647             " returning an error, or even stop capturing."
648             " device %s", obj->videodev);
649         /* have we de-queued a buffer ? */
650         if (!(vbuffer.flags & (V4L2_BUF_FLAG_QUEUED | V4L2_BUF_FLAG_DONE))) {
651           GST_DEBUG_OBJECT (pool, "reenqueing buffer");
652           /* FIXME ... should we do something here? */
653         }
654         break;
655       case EINTR:
656         GST_WARNING_OBJECT (pool,
657             "could not sync on a buffer on device %s", obj->videodev);
658         break;
659       default:
660         GST_WARNING_OBJECT (pool,
661             "Grabbing frame got interrupted on %s unexpectedly. %d: %s.",
662             obj->videodev, errno, g_strerror (errno));
663         break;
664     }
665     return GST_FLOW_ERROR;
666   }
667 no_buffer:
668   {
669     GST_ERROR_OBJECT (pool, "No free buffer found in the pool at index %d.",
670         vbuffer.index);
671     return GST_FLOW_ERROR;
672   }
673 }
674
675 static GstFlowReturn
676 gst_v4l2_buffer_pool_acquire_buffer (GstBufferPool * bpool, GstBuffer ** buffer,
677     GstBufferPoolParams * params)
678 {
679   GstFlowReturn ret;
680   GstV4l2BufferPool *pool = GST_V4L2_BUFFER_POOL (bpool);
681   GstV4l2Object *obj = pool->obj;
682
683   GST_DEBUG_OBJECT (pool, "acquire");
684
685   if (GST_BUFFER_POOL_IS_FLUSHING (bpool))
686     goto flushing;
687
688   switch (obj->type) {
689     case V4L2_BUF_TYPE_VIDEO_CAPTURE:
690       /* capture, This function should return a buffer with new captured data */
691       switch (obj->mode) {
692         case GST_V4L2_IO_RW:
693           /* take empty buffer from the pool */
694           ret = GST_BUFFER_POOL_CLASS (parent_class)->acquire_buffer (bpool,
695               buffer, params);
696           break;
697
698         case GST_V4L2_IO_MMAP:
699           /* just dequeue a buffer, we basically use the queue of v4l2 as the
700            * storage for our buffers. This function does poll first so we can
701            * interrupt it fine. */
702           ret = gst_v4l2_buffer_pool_dqbuf (pool, buffer);
703           break;
704
705         case GST_V4L2_IO_USERPTR:
706         default:
707           g_assert_not_reached ();
708           break;
709       }
710       break;
711
712     case V4L2_BUF_TYPE_VIDEO_OUTPUT:
713       /* playback, This function should return an empty buffer */
714       switch (obj->mode) {
715         case GST_V4L2_IO_RW:
716           /* get an empty buffer */
717           ret = GST_BUFFER_POOL_CLASS (parent_class)->acquire_buffer (bpool,
718               buffer, params);
719           break;
720
721         case GST_V4L2_IO_MMAP:
722           /* get a free unqueued buffer */
723           ret = GST_BUFFER_POOL_CLASS (parent_class)->acquire_buffer (bpool,
724               buffer, params);
725           break;
726
727         case GST_V4L2_IO_USERPTR:
728         default:
729           g_assert_not_reached ();
730           break;
731       }
732       break;
733
734     default:
735       g_assert_not_reached ();
736       break;
737   }
738   return ret;
739
740   /* ERRORS */
741 flushing:
742   {
743     GST_DEBUG_OBJECT (pool, "We are flushing");
744     return GST_FLOW_FLUSHING;
745   }
746 }
747
748 static void
749 gst_v4l2_buffer_pool_release_buffer (GstBufferPool * bpool, GstBuffer * buffer)
750 {
751   GstV4l2BufferPool *pool = GST_V4L2_BUFFER_POOL (bpool);
752   GstV4l2Object *obj = pool->obj;
753
754   GST_DEBUG_OBJECT (pool, "release buffer %p", buffer);
755
756   switch (obj->type) {
757     case V4L2_BUF_TYPE_VIDEO_CAPTURE:
758       /* capture, put the buffer back in the queue so that we can refill it
759        * later. */
760       switch (obj->mode) {
761         case GST_V4L2_IO_RW:
762           /* release back in the pool */
763           GST_BUFFER_POOL_CLASS (parent_class)->release_buffer (bpool, buffer);
764           break;
765
766         case GST_V4L2_IO_MMAP:
767           /* queue back in the device */
768           gst_v4l2_buffer_pool_qbuf (pool, buffer);
769           break;
770
771         case GST_V4L2_IO_USERPTR:
772         default:
773           g_assert_not_reached ();
774           break;
775       }
776       break;
777
778     case V4L2_BUF_TYPE_VIDEO_OUTPUT:
779       switch (obj->mode) {
780         case GST_V4L2_IO_RW:
781           /* release back in the pool */
782           GST_BUFFER_POOL_CLASS (parent_class)->release_buffer (bpool, buffer);
783           break;
784
785         case GST_V4L2_IO_MMAP:
786         {
787           GstV4l2Meta *meta;
788
789           meta = GST_V4L2_META_GET (buffer);
790           g_assert (meta != NULL);
791
792           if (pool->buffers[meta->vbuffer.index] == NULL) {
793             GST_LOG_OBJECT (pool, "buffer not queued, putting on free list");
794             /* playback, put the buffer back in the queue to refill later. */
795             GST_BUFFER_POOL_CLASS (parent_class)->release_buffer (bpool,
796                 buffer);
797           } else {
798             /* the buffer is queued in the device but maybe not played yet. We just
799              * leave it there and not make it available for future calls to acquire
800              * for now. The buffer will be dequeued and reused later. */
801             GST_LOG_OBJECT (pool, "buffer is queued");
802           }
803           break;
804         }
805
806         case GST_V4L2_IO_USERPTR:
807         default:
808           g_assert_not_reached ();
809           break;
810       }
811       break;
812
813     default:
814       g_assert_not_reached ();
815       break;
816   }
817 }
818
819 static void
820 gst_v4l2_buffer_pool_finalize (GObject * object)
821 {
822   GstV4l2BufferPool *pool = GST_V4L2_BUFFER_POOL (object);
823
824   if (pool->video_fd >= 0)
825     v4l2_close (pool->video_fd);
826
827   g_free (pool->buffers);
828
829   G_OBJECT_CLASS (parent_class)->finalize (object);
830 }
831
832 static void
833 gst_v4l2_buffer_pool_init (GstV4l2BufferPool * pool)
834 {
835 }
836
837 static void
838 gst_v4l2_buffer_pool_class_init (GstV4l2BufferPoolClass * klass)
839 {
840   GObjectClass *object_class = G_OBJECT_CLASS (klass);
841   GstBufferPoolClass *bufferpool_class = GST_BUFFER_POOL_CLASS (klass);
842
843   object_class->finalize = gst_v4l2_buffer_pool_finalize;
844
845   bufferpool_class->start = gst_v4l2_buffer_pool_start;
846   bufferpool_class->stop = gst_v4l2_buffer_pool_stop;
847   bufferpool_class->set_config = gst_v4l2_buffer_pool_set_config;
848   bufferpool_class->alloc_buffer = gst_v4l2_buffer_pool_alloc_buffer;
849   bufferpool_class->acquire_buffer = gst_v4l2_buffer_pool_acquire_buffer;
850   bufferpool_class->release_buffer = gst_v4l2_buffer_pool_release_buffer;
851   bufferpool_class->free_buffer = gst_v4l2_buffer_pool_free_buffer;
852 }
853
854 /**
855  * gst_v4l2_buffer_pool_new:
856  * @obj:  the v4l2 object owning the pool
857  *
858  * Construct a new buffer pool.
859  *
860  * Returns: the new pool, use gst_object_unref() to free resources
861  */
862 GstBufferPool *
863 gst_v4l2_buffer_pool_new (GstV4l2Object * obj, GstCaps * caps)
864 {
865   GstV4l2BufferPool *pool;
866   gint fd;
867
868   fd = v4l2_dup (obj->video_fd);
869   if (fd < 0)
870     goto dup_failed;
871
872   pool = (GstV4l2BufferPool *) g_object_new (GST_TYPE_V4L2_BUFFER_POOL, NULL);
873   pool->video_fd = fd;
874   pool->obj = obj;
875
876   gst_buffer_pool_config_set (GST_BUFFER_POOL_CAST (pool)->config, caps,
877       obj->sizeimage, 2, 0, 0, 0);
878
879   return GST_BUFFER_POOL (pool);
880
881   /* ERRORS */
882 dup_failed:
883   {
884     GST_DEBUG ("failed to dup fd %d (%s)", errno, g_strerror (errno));
885     return NULL;
886   }
887 }
888
889 static GstFlowReturn
890 gst_v4l2_do_read (GstV4l2BufferPool * pool, GstBuffer * buf)
891 {
892   GstFlowReturn res;
893   GstV4l2Object *obj = pool->obj;
894   gint amount;
895   GstMapInfo map;
896   gint toread;
897
898   toread = obj->sizeimage;
899
900   GST_LOG_OBJECT (pool, "reading %d bytes into buffer %p", toread, buf);
901
902   gst_buffer_map (buf, &map, GST_MAP_WRITE);
903
904   do {
905     if ((res = gst_v4l2_object_poll (obj)) != GST_FLOW_OK)
906       goto poll_error;
907
908     amount = v4l2_read (obj->video_fd, map.data, toread);
909
910     if (amount == toread) {
911       break;
912     } else if (amount == -1) {
913       if (errno == EAGAIN || errno == EINTR) {
914         continue;
915       } else
916         goto read_error;
917     } else {
918       /* short reads can happen if a signal interrupts the read */
919       continue;
920     }
921   } while (TRUE);
922
923   GST_LOG_OBJECT (pool, "read %d bytes", amount);
924   gst_buffer_unmap (buf, &map);
925   gst_buffer_resize (buf, 0, amount);
926
927   return GST_FLOW_OK;
928
929   /* ERRORS */
930 poll_error:
931   {
932     GST_DEBUG ("poll error %s", gst_flow_get_name (res));
933     goto cleanup;
934   }
935 read_error:
936   {
937     GST_ELEMENT_ERROR (obj->element, RESOURCE, READ,
938         (_("Error reading %d bytes from device '%s'."),
939             toread, obj->videodev), GST_ERROR_SYSTEM);
940     res = GST_FLOW_ERROR;
941     goto cleanup;
942   }
943 cleanup:
944   {
945     gst_buffer_unmap (buf, &map);
946     gst_buffer_resize (buf, 0, 0);
947     return res;
948   }
949 }
950
951 /**
952  * gst_v4l2_buffer_pool_process:
953  * @bpool: a #GstBufferPool
954  * @buf: a #GstBuffer
955  *
956  * Process @buf in @bpool. For capture devices, this functions fills @buf with
957  * data from the device. For output devices, this functions send the contents of
958  * @buf to the device for playback.
959  *
960  * Returns: %GST_FLOW_OK on success.
961  */
962 GstFlowReturn
963 gst_v4l2_buffer_pool_process (GstV4l2BufferPool * pool, GstBuffer * buf)
964 {
965   GstFlowReturn ret = GST_FLOW_OK;
966   GstBufferPool *bpool = GST_BUFFER_POOL_CAST (pool);
967   GstV4l2Object *obj = pool->obj;
968
969   GST_DEBUG_OBJECT (pool, "process buffer %p", buf);
970
971   switch (obj->type) {
972     case V4L2_BUF_TYPE_VIDEO_CAPTURE:
973       /* capture */
974       switch (obj->mode) {
975         case GST_V4L2_IO_RW:
976           /* capture into the buffer */
977           ret = gst_v4l2_do_read (pool, buf);
978           break;
979
980         case GST_V4L2_IO_MMAP:
981         {
982           GstBuffer *tmp;
983
984           if (buf->pool == bpool)
985             /* nothing, data was inside the buffer when we did _acquire() */
986             goto done;
987
988           /* buffer not from our pool, grab a frame and copy it into the target */
989           if ((ret = gst_v4l2_buffer_pool_dqbuf (pool, &tmp)) != GST_FLOW_OK)
990             goto done;
991
992           if (!gst_v4l2_object_copy (obj, buf, tmp))
993             goto copy_failed;
994
995           /* an queue the buffer again after the copy */
996           if ((ret = gst_v4l2_buffer_pool_qbuf (pool, tmp)) != GST_FLOW_OK)
997             goto done;
998           break;
999         }
1000
1001         case GST_V4L2_IO_USERPTR:
1002         default:
1003           g_assert_not_reached ();
1004           break;
1005       }
1006       break;
1007
1008     case V4L2_BUF_TYPE_VIDEO_OUTPUT:
1009       /* playback */
1010       switch (obj->mode) {
1011         case GST_V4L2_IO_RW:
1012           /* FIXME, do write() */
1013           GST_WARNING_OBJECT (pool, "implement write()");
1014           break;
1015
1016         case GST_V4L2_IO_MMAP:
1017         {
1018           GstBuffer *to_queue;
1019
1020           if (buf->pool == bpool) {
1021             /* nothing, we can queue directly */
1022             to_queue = buf;
1023             GST_LOG_OBJECT (pool, "processing buffer from our pool");
1024           } else {
1025             GST_LOG_OBJECT (pool, "alloc buffer from our pool");
1026             if (!gst_buffer_pool_is_active (bpool)) {
1027               GstStructure *config;
1028
1029               /* this pool was not activated, configure and activate */
1030               GST_DEBUG_OBJECT (pool, "activating pool");
1031
1032               config = gst_buffer_pool_get_config (bpool);
1033               gst_buffer_pool_config_add_option (config,
1034                   GST_BUFFER_POOL_OPTION_VIDEO_META);
1035               gst_buffer_pool_set_config (bpool, config);
1036
1037               if (!gst_buffer_pool_set_active (bpool, TRUE))
1038                 goto activate_failed;
1039             }
1040
1041             /* this can block if all buffers are outstanding which would be
1042              * strange because we would expect the upstream element to have
1043              * allocated them and returned to us.. */
1044             ret = GST_BUFFER_POOL_CLASS (parent_class)->acquire_buffer (bpool,
1045                 &to_queue, NULL);
1046             if (ret != GST_FLOW_OK)
1047               goto acquire_failed;
1048
1049             /* copy into it and queue */
1050             if (!gst_v4l2_object_copy (obj, to_queue, buf))
1051               goto copy_failed;
1052           }
1053
1054           if ((ret = gst_v4l2_buffer_pool_qbuf (pool, to_queue)) != GST_FLOW_OK)
1055             goto done;
1056
1057           /* if we are not streaming yet (this is the first buffer, start
1058            * streaming now */
1059           if (!pool->streaming)
1060             if (!start_streaming (pool))
1061               goto start_failed;
1062
1063           if (pool->num_queued == pool->num_allocated) {
1064             /* all buffers are queued, try to dequeue one and release it back
1065              * into the pool so that _acquire can get to it again. */
1066             ret = gst_v4l2_buffer_pool_dqbuf (pool, &to_queue);
1067             if (ret != GST_FLOW_OK)
1068               goto done;
1069
1070             /* release the rendered buffer back into the pool. This wakes up any
1071              * thread waiting for a buffer in _acquire() */
1072             gst_v4l2_buffer_pool_release_buffer (bpool, to_queue);
1073           }
1074           break;
1075         }
1076
1077         case GST_V4L2_IO_USERPTR:
1078         default:
1079           g_assert_not_reached ();
1080           break;
1081       }
1082       break;
1083     default:
1084       g_assert_not_reached ();
1085       break;
1086   }
1087 done:
1088   return ret;
1089
1090   /* ERRORS */
1091 activate_failed:
1092   {
1093     GST_ERROR_OBJECT (obj->element, "failed to activate pool");
1094     return GST_FLOW_ERROR;
1095   }
1096 acquire_failed:
1097   {
1098     GST_WARNING_OBJECT (obj->element, "failed to acquire a buffer: %s",
1099         gst_flow_get_name (ret));
1100     return ret;
1101   }
1102 copy_failed:
1103   {
1104     GST_ERROR_OBJECT (obj->element, "failed to copy data");
1105     return GST_FLOW_ERROR;
1106   }
1107 start_failed:
1108   {
1109     GST_ERROR_OBJECT (obj->element, "failed to start streaming");
1110     return GST_FLOW_ERROR;
1111   }
1112 }