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., 51 Franklin St, Fifth Floor,
  22  * Boston, MA 02110-1301, USA.
  23  */
  24
  25 #ifdef HAVE_CONFIG_H
  26 #  include <config.h>
  27 #endif
  28
  29 #ifndef _GNU_SOURCE
  30 # define _GNU_SOURCE            /* O_CLOEXEC */
  31 #endif
  32 #include <fcntl.h>
  33
  34 #include <sys/mman.h>
  35 #include <string.h>
  36 #include <unistd.h>
  37
  38 #include "gst/video/video.h"
  39 #include "gst/video/gstvideometa.h"
  40 #include "gst/video/gstvideopool.h"
  41 #include "gst/allocators/gstdmabuf.h"
  42
  43 #include <gstv4l2bufferpool.h>
  44
  45 #include "v4l2_calls.h"
  46 #include "gst/gst-i18n-plugin.h"
  47 #include <gst/glib-compat-private.h>
  48
  49 GST_DEBUG_CATEGORY_STATIC (v4l2bufferpool_debug);
  50 GST_DEBUG_CATEGORY_EXTERN (GST_CAT_PERFORMANCE);
  51 #define GST_CAT_DEFAULT v4l2bufferpool_debug
  52
  53 #define GST_V4L2_IMPORT_QUARK gst_v4l2_buffer_pool_import_quark ()
  54
  55
  56 /*
  57  * GstV4l2BufferPool:
  58  */
  59 #define gst_v4l2_buffer_pool_parent_class parent_class
  60 G_DEFINE_TYPE (GstV4l2BufferPool, gst_v4l2_buffer_pool, GST_TYPE_BUFFER_POOL);
  61
  62 enum _GstV4l2BufferPoolAcquireFlags
  63 {
  64   GST_V4L2_BUFFER_POOL_ACQUIRE_FLAG_RESURRECT =
  65       GST_BUFFER_POOL_ACQUIRE_FLAG_LAST,
  66   GST_V4L2_BUFFER_POOL_ACQUIRE_FLAG_LAST
  67 };
  68
  69 static void gst_v4l2_buffer_pool_release_buffer (GstBufferPool * bpool,
  70     GstBuffer * buffer);
  71
  72 static gboolean
  73 gst_v4l2_is_buffer_valid (GstBuffer * buffer, GstV4l2MemoryGroup ** out_group)
  74 {
  75   GstMemory *mem = gst_buffer_peek_memory (buffer, 0);
  76   gboolean valid = FALSE;
  77
  78   if (GST_BUFFER_FLAG_IS_SET (buffer, GST_BUFFER_FLAG_TAG_MEMORY))
  79     goto done;
  80
  81   if (gst_is_dmabuf_memory (mem))
  82     mem = gst_mini_object_get_qdata (GST_MINI_OBJECT (mem),
  83         GST_V4L2_MEMORY_QUARK);
  84
  85   if (mem && gst_is_v4l2_memory (mem)) {
  86     GstV4l2Memory *vmem = (GstV4l2Memory *) mem;
  87     GstV4l2MemoryGroup *group = vmem->group;
  88     gint i;
  89
  90     if (group->n_mem != gst_buffer_n_memory (buffer))
  91       goto done;
  92
  93     for (i = 0; i < group->n_mem; i++) {
  94       if (group->mem[i] != gst_buffer_peek_memory (buffer, i))
  95         goto done;
  96
  97       if (!gst_memory_is_writable (group->mem[i]))
  98         goto done;
  99     }
 100
 101     valid = TRUE;
 102     if (out_group)
 103       *out_group = group;
 104   }
 105
 106 done:
 107   return valid;
 108 }
 109
 110 static GstFlowReturn
 111 gst_v4l2_buffer_pool_copy_buffer (GstV4l2BufferPool * pool, GstBuffer * dest,
 112     GstBuffer * src)
 113 {
 114   const GstVideoFormatInfo *finfo = pool->caps_info.finfo;
 115
 116   GST_LOG_OBJECT (pool, "copying buffer");
 117
 118   if (finfo && (finfo->format != GST_VIDEO_FORMAT_UNKNOWN &&
 119           finfo->format != GST_VIDEO_FORMAT_ENCODED)) {
 120     GstVideoFrame src_frame, dest_frame;
 121
 122     GST_DEBUG_OBJECT (pool, "copy video frame");
 123
 124     /* we have raw video, use videoframe copy to get strides right */
 125     if (!gst_video_frame_map (&src_frame, &pool->caps_info, src, GST_MAP_READ))
 126       goto invalid_buffer;
 127
 128     if (!gst_video_frame_map (&dest_frame, &pool->caps_info, dest,
 129             GST_MAP_WRITE)) {
 130       gst_video_frame_unmap (&src_frame);
 131       goto invalid_buffer;
 132     }
 133
 134     gst_video_frame_copy (&dest_frame, &src_frame);
 135
 136     gst_video_frame_unmap (&src_frame);
 137     gst_video_frame_unmap (&dest_frame);
 138   } else {
 139     GstMapInfo map;
 140
 141     GST_DEBUG_OBJECT (pool, "copy raw bytes");
 142
 143     if (!gst_buffer_map (src, &map, GST_MAP_READ))
 144       goto invalid_buffer;
 145
 146     gst_buffer_fill (dest, 0, map.data, gst_buffer_get_size (src));
 147
 148     gst_buffer_unmap (src, &map);
 149     gst_buffer_resize (dest, 0, gst_buffer_get_size (src));
 150   }
 151
 152   GST_CAT_LOG_OBJECT (GST_CAT_PERFORMANCE, pool, "slow copy into buffer %p",
 153       dest);
 154
 155   return GST_FLOW_OK;
 156
 157 invalid_buffer:
 158   {
 159     GST_ERROR_OBJECT (pool, "could not map buffer");
 160     return GST_FLOW_ERROR;
 161   }
 162 }
 163
 164 struct UserPtrData
 165 {
 166   GstBuffer *buffer;
 167   gboolean is_frame;
 168   GstVideoFrame frame;
 169   GstMapInfo map;
 170 };
 171
 172 static GQuark
 173 gst_v4l2_buffer_pool_import_quark (void)
 174 {
 175   static GQuark quark = 0;
 176
 177   if (quark == 0)
 178     quark = g_quark_from_string ("GstV4l2BufferPoolUsePtrData");
 179
 180   return quark;
 181 }
 182
 183 static void
 184 _unmap_userptr_frame (struct UserPtrData *data)
 185 {
 186   if (data->is_frame)
 187     gst_video_frame_unmap (&data->frame);
 188   else
 189     gst_buffer_unmap (data->buffer, &data->map);
 190
 191   if (data->buffer)
 192     gst_buffer_unref (data->buffer);
 193
 194   g_slice_free (struct UserPtrData, data);
 195 }
 196
 197 static GstFlowReturn
 198 gst_v4l2_buffer_pool_import_userptr (GstV4l2BufferPool * pool,
 199     GstBuffer * dest, GstBuffer * src)
 200 {
 201   GstFlowReturn ret = GST_FLOW_OK;
 202   GstV4l2MemoryGroup *group = NULL;
 203   GstMapFlags flags;
 204   const GstVideoFormatInfo *finfo = pool->caps_info.finfo;
 205   struct UserPtrData *data = NULL;
 206
 207   GST_LOG_OBJECT (pool, "importing userptr");
 208
 209   /* get the group */
 210   if (!gst_v4l2_is_buffer_valid (dest, &group))
 211     goto not_our_buffer;
 212
 213   if (V4L2_TYPE_IS_OUTPUT (pool->obj->type))
 214     flags = GST_MAP_READ;
 215   else
 216     flags = GST_MAP_WRITE;
 217
 218   data = g_slice_new0 (struct UserPtrData);
 219
 220   if (finfo && (finfo->format != GST_VIDEO_FORMAT_UNKNOWN &&
 221           finfo->format != GST_VIDEO_FORMAT_ENCODED)) {
 222     data->is_frame = TRUE;
 223
 224     if (!gst_video_frame_map (&data->frame, &pool->caps_info, src, flags))
 225       goto invalid_buffer;
 226
 227     if (!gst_v4l2_allocator_import_userptr (pool->vallocator, group,
 228             data->frame.info.size, finfo->n_planes, data->frame.data,
 229             data->frame.info.offset))
 230       goto import_failed;
 231   } else {
 232     gsize offset[1] = { 0 };
 233     gpointer ptr[1];
 234
 235     data->is_frame = FALSE;
 236
 237     if (!gst_buffer_map (src, &data->map, flags))
 238       goto invalid_buffer;
 239
 240     ptr[0] = data->map.data;
 241
 242     if (!gst_v4l2_allocator_import_userptr (pool->vallocator, group,
 243             data->map.size, 1, ptr, offset))
 244       goto import_failed;
 245   }
 246
 247   data->buffer = gst_buffer_ref (src);
 248
 249   gst_mini_object_set_qdata (GST_MINI_OBJECT (dest), GST_V4L2_IMPORT_QUARK,
 250       data, (GDestroyNotify) _unmap_userptr_frame);
 251
 252   return ret;
 253
 254 not_our_buffer:
 255   {
 256     GST_ERROR_OBJECT (pool, "destination buffer invalid or not from our pool");
 257     return GST_FLOW_ERROR;
 258   }
 259 invalid_buffer:
 260   {
 261     GST_ERROR_OBJECT (pool, "could not map buffer");
 262     g_slice_free (struct UserPtrData, data);
 263     return GST_FLOW_ERROR;
 264   }
 265 import_failed:
 266   {
 267     GST_ERROR_OBJECT (pool, "failed to import data");
 268     _unmap_userptr_frame (data);
 269     return GST_FLOW_ERROR;
 270   }
 271 }
 272
 273 static GstFlowReturn
 274 gst_v4l2_buffer_pool_import_dmabuf (GstV4l2BufferPool * pool,
 275     GstBuffer * dest, GstBuffer * src)
 276 {
 277   GstV4l2MemoryGroup *group = NULL;
 278   GstMemory *dma_mem[GST_VIDEO_MAX_PLANES] = { 0 };
 279   guint n_mem = gst_buffer_n_memory (src);
 280   gint i;
 281
 282   GST_LOG_OBJECT (pool, "importing dmabuf");
 283
 284   if (!gst_v4l2_is_buffer_valid (dest, &group))
 285     goto not_our_buffer;
 286
 287   if (n_mem > GST_VIDEO_MAX_PLANES)
 288     goto too_many_mems;
 289
 290   for (i = 0; i < n_mem; i++)
 291     dma_mem[i] = gst_buffer_peek_memory (src, i);
 292
 293   if (!gst_v4l2_allocator_import_dmabuf (pool->vallocator, group, n_mem,
 294           dma_mem))
 295     goto import_failed;
 296
 297   gst_mini_object_set_qdata (GST_MINI_OBJECT (dest), GST_V4L2_IMPORT_QUARK,
 298       gst_buffer_ref (src), (GDestroyNotify) gst_buffer_unref);
 299
 300   return GST_FLOW_OK;
 301
 302 not_our_buffer:
 303   {
 304     GST_ERROR_OBJECT (pool, "destination buffer invalid or not from our pool");
 305     return GST_FLOW_ERROR;
 306   }
 307 too_many_mems:
 308   {
 309     GST_ERROR_OBJECT (pool, "could not map buffer");
 310     return GST_FLOW_ERROR;
 311   }
 312 import_failed:
 313   {
 314     GST_ERROR_OBJECT (pool, "failed to import dmabuf");
 315     return GST_FLOW_ERROR;
 316   }
 317 }
 318
 319 static GstFlowReturn
 320 gst_v4l2_buffer_pool_prepare_buffer (GstV4l2BufferPool * pool,
 321     GstBuffer * dest, GstBuffer * src)
 322 {
 323   GstFlowReturn ret = GST_FLOW_OK;
 324   gboolean own_src = FALSE;
 325
 326   if (src == NULL) {
 327     if (pool->other_pool == NULL) {
 328       GST_ERROR_OBJECT (pool, "can't prepare buffer, source buffer missing");
 329       return GST_FLOW_ERROR;
 330     }
 331
 332     ret = gst_buffer_pool_acquire_buffer (pool->other_pool, &src, NULL);
 333     if (ret != GST_FLOW_OK) {
 334       GST_ERROR_OBJECT (pool, "failed to acquire buffer from downstream pool");
 335       goto done;
 336     }
 337
 338     own_src = TRUE;
 339   }
 340
 341   switch (pool->obj->mode) {
 342     case GST_V4L2_IO_MMAP:
 343     case GST_V4L2_IO_DMABUF:
 344       ret = gst_v4l2_buffer_pool_copy_buffer (pool, dest, src);
 345       break;
 346     case GST_V4L2_IO_USERPTR:
 347       ret = gst_v4l2_buffer_pool_import_userptr (pool, dest, src);
 348       break;
 349     case GST_V4L2_IO_DMABUF_IMPORT:
 350       ret = gst_v4l2_buffer_pool_import_dmabuf (pool, dest, src);
 351       break;
 352     default:
 353       break;
 354   }
 355
 356   if (own_src)
 357     gst_buffer_unref (src);
 358
 359 done:
 360   return ret;
 361 }
 362
 363 static GstFlowReturn
 364 gst_v4l2_buffer_pool_alloc_buffer (GstBufferPool * bpool, GstBuffer ** buffer,
 365     GstBufferPoolAcquireParams * params)
 366 {
 367   GstV4l2BufferPool *pool = GST_V4L2_BUFFER_POOL (bpool);
 368   GstV4l2MemoryGroup *group = NULL;
 369   GstBuffer *newbuf = NULL;
 370   GstV4l2Object *obj;
 371   GstVideoInfo *info;
 372
 373   obj = pool->obj;
 374   info = &obj->info;
 375
 376   switch (obj->mode) {
 377     case GST_V4L2_IO_RW:
 378       newbuf =
 379           gst_buffer_new_allocate (pool->allocator, pool->size, &pool->params);
 380       break;
 381     case GST_V4L2_IO_MMAP:
 382       group = gst_v4l2_allocator_alloc_mmap (pool->vallocator);
 383       break;
 384     case GST_V4L2_IO_DMABUF:
 385       group = gst_v4l2_allocator_alloc_dmabuf (pool->vallocator,
 386           pool->allocator);
 387       break;
 388     case GST_V4L2_IO_USERPTR:
 389       group = gst_v4l2_allocator_alloc_userptr (pool->vallocator);
 390       break;
 391     case GST_V4L2_IO_DMABUF_IMPORT:
 392       group = gst_v4l2_allocator_alloc_dmabufin (pool->vallocator);
 393       break;
 394     default:
 395       newbuf = NULL;
 396       g_assert_not_reached ();
 397       break;
 398   }
 399
 400   if (group != NULL) {
 401     gint i;
 402     newbuf = gst_buffer_new ();
 403
 404     for (i = 0; i < group->n_mem; i++)
 405       gst_buffer_append_memory (newbuf, group->mem[i]);
 406   } else if (newbuf == NULL) {
 407     goto allocation_failed;
 408   }
 409
 410   /* add metadata to raw video buffers */
 411   if (pool->add_videometa)
 412     gst_buffer_add_video_meta_full (newbuf, GST_VIDEO_FRAME_FLAG_NONE,
 413         GST_VIDEO_INFO_FORMAT (info), GST_VIDEO_INFO_WIDTH (info),
 414         GST_VIDEO_INFO_HEIGHT (info), GST_VIDEO_INFO_N_PLANES (info),
 415         info->offset, info->stride);
 416
 417   *buffer = newbuf;
 418
 419   return GST_FLOW_OK;
 420
 421   /* ERRORS */
 422 allocation_failed:
 423   {
 424     GST_ERROR_OBJECT (pool, "failed to allocate buffer");
 425     return GST_FLOW_ERROR;
 426   }
 427 }
 428
 429 static gboolean
 430 gst_v4l2_buffer_pool_set_config (GstBufferPool * bpool, GstStructure * config)
 431 {
 432   GstV4l2BufferPool *pool = GST_V4L2_BUFFER_POOL (bpool);
 433   GstV4l2Object *obj = pool->obj;
 434   GstCaps *caps;
 435   guint size, min_buffers, max_buffers;
 436   GstAllocator *allocator;
 437   GstAllocationParams params;
 438   gboolean can_allocate = FALSE;
 439   gboolean updated = FALSE;
 440   gboolean ret;
 441
 442   pool->add_videometa =
 443       gst_buffer_pool_config_has_option (config,
 444       GST_BUFFER_POOL_OPTION_VIDEO_META);
 445
 446   /* parse the config and keep around */
 447   if (!gst_buffer_pool_config_get_params (config, &caps, &size, &min_buffers,
 448           &max_buffers))
 449     goto wrong_config;
 450
 451   if (!gst_buffer_pool_config_get_allocator (config, &allocator, &params))
 452     goto wrong_config;
 453
 454   GST_DEBUG_OBJECT (pool, "config %" GST_PTR_FORMAT, config);
 455
 456   if (pool->allocator)
 457     gst_object_unref (pool->allocator);
 458   pool->allocator = NULL;
 459
 460   switch (obj->mode) {
 461     case GST_V4L2_IO_DMABUF:
 462       pool->allocator = gst_dmabuf_allocator_new ();
 463       can_allocate = GST_V4L2_ALLOCATOR_CAN_ALLOCATE (pool->vallocator, MMAP);
 464       break;
 465     case GST_V4L2_IO_MMAP:
 466       can_allocate = GST_V4L2_ALLOCATOR_CAN_ALLOCATE (pool->vallocator, MMAP);
 467       break;
 468     case GST_V4L2_IO_USERPTR:
 469       can_allocate =
 470           GST_V4L2_ALLOCATOR_CAN_ALLOCATE (pool->vallocator, USERPTR);
 471       break;
 472     case GST_V4L2_IO_DMABUF_IMPORT:
 473       can_allocate = GST_V4L2_ALLOCATOR_CAN_ALLOCATE (pool->vallocator, DMABUF);
 474       break;
 475     case GST_V4L2_IO_RW:
 476       if (allocator)
 477         pool->allocator = g_object_ref (allocator);
 478       pool->params = params;
 479       /* No need to change the configuration */
 480       goto done;
 481       break;
 482     default:
 483       g_assert_not_reached ();
 484       break;
 485   }
 486
 487   /* libv4l2 conversion code does not handle CREATE_BUFS, and may lead to
 488    * instability and crash, disable it for now */
 489   if (can_allocate && obj->fmtdesc->flags & V4L2_FMT_FLAG_EMULATED) {
 490     GST_WARNING_OBJECT (pool,
 491         "libv4l2 converter detected, disabling CREATE_BUFS");
 492     can_allocate = FALSE;
 493     GST_OBJECT_FLAG_UNSET (pool->vallocator,
 494         GST_V4L2_ALLOCATOR_FLAG_MMAP_CREATE_BUFS
 495         | GST_V4L2_ALLOCATOR_FLAG_USERPTR_CREATE_BUFS
 496         | GST_V4L2_ALLOCATOR_FLAG_DMABUF_CREATE_BUFS);
 497   }
 498
 499   if (min_buffers < GST_V4L2_MIN_BUFFERS) {
 500     updated = TRUE;
 501     min_buffers = GST_V4L2_MIN_BUFFERS;
 502     GST_INFO_OBJECT (pool, "increasing minimum buffers to %u", min_buffers);
 503   }
 504
 505   if (max_buffers > VIDEO_MAX_FRAME || max_buffers == 0) {
 506     updated = TRUE;
 507     max_buffers = VIDEO_MAX_FRAME;
 508     GST_INFO_OBJECT (pool, "reducing maximum buffers to %u", max_buffers);
 509   }
 510
 511   if (min_buffers > max_buffers) {
 512     updated = TRUE;
 513     min_buffers = max_buffers;
 514     GST_INFO_OBJECT (pool, "reducing minimum buffers to %u", min_buffers);
 515   } else if (min_buffers != max_buffers) {
 516     if (!can_allocate) {
 517       updated = TRUE;
 518       max_buffers = min_buffers;
 519       GST_INFO_OBJECT (pool, "can't allocate, setting maximum to minimum");
 520     }
 521   }
 522
 523   if (!pool->add_videometa && obj->need_video_meta) {
 524     GST_INFO_OBJECT (pool, "adding needed video meta");
 525     updated = TRUE;
 526     gst_buffer_pool_config_add_option (config,
 527         GST_BUFFER_POOL_OPTION_VIDEO_META);
 528   }
 529
 530   /* Always update the config to ensure the configured size matches */
 531   gst_buffer_pool_config_set_params (config, caps, obj->info.size, min_buffers,
 532       max_buffers);
 533
 534   /* keep a GstVideoInfo with defaults for the when we need to copy */
 535   gst_video_info_from_caps (&pool->caps_info, caps);
 536
 537 done:
 538   ret = GST_BUFFER_POOL_CLASS (parent_class)->set_config (bpool, config);
 539
 540   /* If anything was changed documentation recommand to return FALSE */
 541   return !updated && ret;
 542
 543   /* ERRORS */
 544 wrong_config:
 545   {
 546     GST_ERROR_OBJECT (pool, "invalid config %" GST_PTR_FORMAT, config);
 547     return FALSE;
 548   }
 549 }
 550
 551 static gboolean
 552 gst_v4l2_buffer_pool_streamon (GstV4l2BufferPool * pool)
 553 {
 554   GstV4l2Object *obj = pool->obj;
 555
 556   switch (obj->mode) {
 557     case GST_V4L2_IO_MMAP:
 558     case GST_V4L2_IO_USERPTR:
 559     case GST_V4L2_IO_DMABUF:
 560     case GST_V4L2_IO_DMABUF_IMPORT:
 561       if (!pool->streaming) {
 562         if (v4l2_ioctl (pool->video_fd, VIDIOC_STREAMON, &obj->type) < 0)
 563           goto streamon_failed;
 564
 565         pool->streaming = TRUE;
 566
 567         GST_DEBUG_OBJECT (pool, "Started streaming");
 568       }
 569       break;
 570     default:
 571       break;
 572   }
 573
 574   return TRUE;
 575
 576 streamon_failed:
 577   {
 578     GST_ERROR_OBJECT (pool, "error with STREAMON %d (%s)", errno,
 579         g_strerror (errno));
 580     return FALSE;
 581   }
 582 }
 583
 584 static void
 585 gst_v4l2_buffer_pool_streamoff (GstV4l2BufferPool * pool)
 586 {
 587   GstV4l2Object *obj = pool->obj;
 588
 589   switch (obj->mode) {
 590     case GST_V4L2_IO_MMAP:
 591     case GST_V4L2_IO_USERPTR:
 592     case GST_V4L2_IO_DMABUF:
 593     case GST_V4L2_IO_DMABUF_IMPORT:
 594       if (pool->streaming) {
 595         if (v4l2_ioctl (pool->video_fd, VIDIOC_STREAMOFF, &obj->type) < 0)
 596           GST_WARNING_OBJECT (pool, "STREAMOFF failed with errno %d (%s)",
 597               errno, g_strerror (errno));
 598
 599         pool->streaming = FALSE;
 600
 601         GST_DEBUG_OBJECT (pool, "Stopped streaming");
 602
 603         if (pool->vallocator)
 604           gst_v4l2_allocator_flush (pool->vallocator);
 605       }
 606       break;
 607     default:
 608       break;
 609   }
 610 }
 611
 612 static GstFlowReturn
 613 gst_v4l2_buffer_pool_resurect_buffer (GstV4l2BufferPool * pool)
 614 {
 615   GstBufferPoolAcquireParams params = { 0 };
 616   GstBuffer *buffer = NULL;
 617   GstFlowReturn ret;
 618
 619   GST_DEBUG_OBJECT (pool, "A buffer was lost, reallocating it");
 620
 621   params.flags =
 622       (GstBufferPoolAcquireFlags) GST_V4L2_BUFFER_POOL_ACQUIRE_FLAG_RESURRECT;
 623   ret =
 624       gst_buffer_pool_acquire_buffer (GST_BUFFER_POOL (pool), &buffer, &params);
 625
 626   if (ret == GST_FLOW_OK)
 627     gst_buffer_unref (buffer);
 628
 629   return ret;
 630 }
 631
 632 static gboolean
 633 gst_v4l2_buffer_pool_start (GstBufferPool * bpool)
 634 {
 635   GstV4l2BufferPool *pool = GST_V4L2_BUFFER_POOL (bpool);
 636   GstBufferPoolClass *pclass = GST_BUFFER_POOL_CLASS (parent_class);
 637   GstV4l2Object *obj = pool->obj;
 638   GstStructure *config;
 639   GstCaps *caps;
 640   guint size, min_buffers, max_buffers;
 641   guint max_latency, min_latency, copy_threshold = 0;
 642   gboolean can_allocate = FALSE;
 643
 644   GST_DEBUG_OBJECT (pool, "activating pool");
 645
 646   config = gst_buffer_pool_get_config (bpool);
 647   if (!gst_buffer_pool_config_get_params (config, &caps, &size, &min_buffers,
 648           &max_buffers))
 649     goto wrong_config;
 650
 651   min_latency = MAX (GST_V4L2_MIN_BUFFERS, obj->min_buffers);
 652
 653   switch (obj->mode) {
 654     case GST_V4L2_IO_RW:
 655       can_allocate = TRUE;
 656 #ifdef HAVE_LIBV4L2
 657       /* This workaround a unfixable bug in libv4l2 when RW is emulated on top
 658        * of MMAP. In this case, the first read initialize the queues, but the
 659        * poll before that will always fail. Doing an empty read, forces the
 660        * queue to be initialized now. We only do this if we have a streaming
 661        * driver. */
 662       if (obj->vcap.capabilities & V4L2_CAP_STREAMING)
 663         v4l2_read (obj->video_fd, NULL, 0);
 664 #endif
 665       break;
 666     case GST_V4L2_IO_DMABUF:
 667     case GST_V4L2_IO_MMAP:
 668     {
 669       guint count;
 670
 671       can_allocate = GST_V4L2_ALLOCATOR_CAN_ALLOCATE (pool->vallocator, MMAP);
 672
 673       /* first, lets request buffers, and see how many we can get: */
 674       GST_DEBUG_OBJECT (pool, "requesting %d MMAP buffers", min_buffers);
 675
 676       count = gst_v4l2_allocator_start (pool->vallocator, min_buffers,
 677           V4L2_MEMORY_MMAP);
 678
 679       if (count < GST_V4L2_MIN_BUFFERS) {
 680         min_buffers = count;
 681         goto no_buffers;
 682       }
 683
 684       /* V4L2 buffer pool are often very limited in the amount of buffers it
 685        * can offer. The copy_threshold will workaround this limitation by
 686        * falling back to copy if the pipeline needed more buffers. This also
 687        * prevent having to do REQBUFS(N)/REQBUFS(0) everytime configure is
 688        * called. */
 689       if (count != min_buffers || pool->enable_copy_threshold) {
 690         GST_WARNING_OBJECT (pool,
 691             "Uncertain or not enough buffers, enabling copy threshold");
 692         min_buffers = count;
 693         copy_threshold = min_latency;
 694       }
 695
 696       break;
 697     }
 698     case GST_V4L2_IO_USERPTR:
 699     {
 700       guint count;
 701
 702       can_allocate =
 703           GST_V4L2_ALLOCATOR_CAN_ALLOCATE (pool->vallocator, USERPTR);
 704
 705       GST_DEBUG_OBJECT (pool, "requesting %d USERPTR buffers", min_buffers);
 706
 707       count = gst_v4l2_allocator_start (pool->vallocator, min_buffers,
 708           V4L2_MEMORY_USERPTR);
 709
 710       /* There is no rational to not get what we asked */
 711       if (count < min_buffers) {
 712         min_buffers = count;
 713         goto no_buffers;
 714       }
 715
 716       min_buffers = count;
 717       break;
 718     }
 719     case GST_V4L2_IO_DMABUF_IMPORT:
 720     {
 721       guint count;
 722
 723       can_allocate = GST_V4L2_ALLOCATOR_CAN_ALLOCATE (pool->vallocator, DMABUF);
 724
 725       GST_DEBUG_OBJECT (pool, "requesting %d DMABUF buffers", min_buffers);
 726
 727       count = gst_v4l2_allocator_start (pool->vallocator, min_buffers,
 728           V4L2_MEMORY_DMABUF);
 729
 730       /* There is no rational to not get what we asked */
 731       if (count < min_buffers) {
 732         min_buffers = count;
 733         goto no_buffers;
 734       }
 735
 736       min_buffers = count;
 737       break;
 738     }
 739     default:
 740       min_buffers = 0;
 741       copy_threshold = 0;
 742       g_assert_not_reached ();
 743       break;
 744   }
 745
 746   if (can_allocate)
 747     max_latency = max_buffers;
 748   else
 749     max_latency = min_buffers;
 750
 751   pool->size = size;
 752   pool->copy_threshold = copy_threshold;
 753   pool->max_latency = max_latency;
 754   pool->min_latency = min_latency;
 755   pool->num_queued = 0;
 756
 757   if (max_buffers != 0 && max_buffers < min_buffers)
 758     max_buffers = min_buffers;
 759
 760   gst_buffer_pool_config_set_params (config, caps, size, min_buffers,
 761       max_buffers);
 762   pclass->set_config (bpool, config);
 763   gst_structure_free (config);
 764
 765   if (pool->other_pool)
 766     if (!gst_buffer_pool_set_active (pool->other_pool, TRUE))
 767       goto other_pool_failed;
 768
 769   /* now, allocate the buffers: */
 770   if (!pclass->start (bpool))
 771     goto start_failed;
 772
 773   if (!V4L2_TYPE_IS_OUTPUT (obj->type))
 774     pool->group_released_handler =
 775         g_signal_connect_swapped (pool->vallocator, "group-released",
 776         G_CALLBACK (gst_v4l2_buffer_pool_resurect_buffer), pool);
 777
 778   return TRUE;
 779
 780   /* ERRORS */
 781 wrong_config:
 782   {
 783     GST_ERROR_OBJECT (pool, "invalid config %" GST_PTR_FORMAT, config);
 784     gst_structure_free (config);
 785     return FALSE;
 786   }
 787 no_buffers:
 788   {
 789     GST_ERROR_OBJECT (pool,
 790         "we received %d buffer from device '%s', we want at least %d",
 791         min_buffers, obj->videodev, GST_V4L2_MIN_BUFFERS);
 792     gst_structure_free (config);
 793     return FALSE;
 794   }
 795 start_failed:
 796   {
 797     GST_ERROR_OBJECT (pool, "failed to start streaming");
 798     return FALSE;
 799   }
 800 other_pool_failed:
 801   {
 802     GST_ERROR_OBJECT (pool, "failed to active the other pool %"
 803         GST_PTR_FORMAT, pool->other_pool);
 804     return FALSE;
 805   }
 806 }
 807
 808 static gboolean
 809 gst_v4l2_buffer_pool_stop (GstBufferPool * bpool)
 810 {
 811   GstV4l2BufferPool *pool = GST_V4L2_BUFFER_POOL (bpool);
 812   GstBufferPoolClass *pclass = GST_BUFFER_POOL_CLASS (parent_class);
 813   gboolean ret;
 814   gint i;
 815
 816   GST_DEBUG_OBJECT (pool, "stopping pool");
 817
 818   if (pool->group_released_handler > 0) {
 819     g_signal_handler_disconnect (pool->vallocator,
 820         pool->group_released_handler);
 821     pool->group_released_handler = 0;
 822   }
 823
 824   if (pool->other_pool) {
 825     gst_buffer_pool_set_active (pool->other_pool, FALSE);
 826     gst_object_unref (pool->other_pool);
 827     pool->other_pool = NULL;
 828   }
 829
 830   gst_v4l2_buffer_pool_streamoff (pool);
 831
 832   for (i = 0; i < VIDEO_MAX_FRAME; i++) {
 833     if (pool->buffers[i]) {
 834       GstBuffer *buffer = pool->buffers[i];
 835
 836       pool->buffers[i] = NULL;
 837
 838       if (V4L2_TYPE_IS_OUTPUT (pool->obj->type))
 839         gst_v4l2_buffer_pool_release_buffer (bpool, buffer);
 840       else                      /* Don't re-enqueue capture buffer on stop */
 841         pclass->release_buffer (bpool, buffer);
 842
 843       g_atomic_int_add (&pool->num_queued, -1);
 844     }
 845   }
 846
 847   ret = GST_BUFFER_POOL_CLASS (parent_class)->stop (bpool);
 848
 849   if (ret && pool->vallocator) {
 850     GstV4l2Return vret;
 851
 852     vret = gst_v4l2_allocator_stop (pool->vallocator);
 853
 854     if (vret == GST_V4L2_BUSY)
 855       GST_WARNING_OBJECT (pool, "some buffers are still outstanding");
 856
 857     ret = (vret == GST_V4L2_OK);
 858   }
 859
 860   return ret;
 861 }
 862
 863 static void
 864 gst_v4l2_buffer_pool_flush_start (GstBufferPool * bpool)
 865 {
 866   GstV4l2BufferPool *pool = GST_V4L2_BUFFER_POOL (bpool);
 867
 868   GST_DEBUG_OBJECT (pool, "start flushing");
 869
 870   gst_poll_set_flushing (pool->poll, TRUE);
 871
 872   GST_OBJECT_LOCK (pool);
 873   pool->empty = FALSE;
 874   g_cond_broadcast (&pool->empty_cond);
 875   GST_OBJECT_UNLOCK (pool);
 876
 877   if (pool->other_pool)
 878     gst_buffer_pool_set_flushing (pool->other_pool, TRUE);
 879 }
 880
 881 static void
 882 gst_v4l2_buffer_pool_flush_stop (GstBufferPool * bpool)
 883 {
 884   GstV4l2BufferPool *pool = GST_V4L2_BUFFER_POOL (bpool);
 885   GstV4l2Object *obj = pool->obj;
 886   gint i;
 887
 888   GST_DEBUG_OBJECT (pool, "stop flushing");
 889
 890   /* If we haven't started streaming yet, simply call streamon */
 891   if (!pool->streaming)
 892     goto streamon;
 893
 894   if (pool->other_pool)
 895     gst_buffer_pool_set_flushing (pool->other_pool, FALSE);
 896
 897   gst_v4l2_buffer_pool_streamoff (pool);
 898
 899   /* Reset our state */
 900   switch (obj->mode) {
 901     case GST_V4L2_IO_RW:
 902       break;
 903     case GST_V4L2_IO_MMAP:
 904     case GST_V4L2_IO_USERPTR:
 905     case GST_V4L2_IO_DMABUF:
 906     case GST_V4L2_IO_DMABUF_IMPORT:
 907     {
 908       for (i = 0; i < VIDEO_MAX_FRAME; i++) {
 909         /* Re-enqueue buffers */
 910         if (pool->buffers[i]) {
 911           GstBufferPool *bpool = (GstBufferPool *) pool;
 912           GstBuffer *buffer = pool->buffers[i];
 913
 914           pool->buffers[i] = NULL;
 915
 916           /* Remove qdata, this will unmap any map data in
 917            * userptr/dmabuf-import */
 918           gst_mini_object_set_qdata (GST_MINI_OBJECT (buffer),
 919               GST_V4L2_IMPORT_QUARK, NULL, NULL);
 920
 921           if (buffer->pool == NULL)
 922             gst_v4l2_buffer_pool_release_buffer (bpool, buffer);
 923
 924           g_atomic_int_add (&pool->num_queued, -1);
 925         }
 926       }
 927
 928       break;
 929     }
 930     default:
 931       g_assert_not_reached ();
 932       break;
 933   }
 934
 935 streamon:
 936   /* Start streaming on capture device only */
 937   if (!V4L2_TYPE_IS_OUTPUT (obj->type))
 938     gst_v4l2_buffer_pool_streamon (pool);
 939
 940   gst_poll_set_flushing (pool->poll, FALSE);
 941 }
 942
 943 static GstFlowReturn
 944 gst_v4l2_buffer_pool_poll (GstV4l2BufferPool * pool)
 945 {
 946   gint ret;
 947
 948   /* In RW mode there is no queue, hence no need to wait while the queue is
 949    * empty */
 950   if (pool->obj->mode != GST_V4L2_IO_RW) {
 951     GST_OBJECT_LOCK (pool);
 952     while (pool->empty)
 953       g_cond_wait (&pool->empty_cond, GST_OBJECT_GET_LOCK (pool));
 954     GST_OBJECT_UNLOCK (pool);
 955   }
 956
 957   if (!pool->can_poll_device)
 958     goto done;
 959
 960   GST_LOG_OBJECT (pool, "polling device");
 961
 962 again:
 963   ret = gst_poll_wait (pool->poll, GST_CLOCK_TIME_NONE);
 964   if (G_UNLIKELY (ret < 0)) {
 965     switch (errno) {
 966       case EBUSY:
 967         goto stopped;
 968       case EAGAIN:
 969       case EINTR:
 970         goto again;
 971       case ENXIO:
 972         GST_WARNING_OBJECT (pool,
 973             "v4l2 device doesn't support polling. Disabling"
 974             " using libv4l2 in this case may cause deadlocks");
 975         pool->can_poll_device = FALSE;
 976         goto done;
 977       default:
 978         goto select_error;
 979     }
 980   }
 981
 982   if (gst_poll_fd_has_error (pool->poll, &pool->pollfd))
 983     goto select_error;
 984
 985 done:
 986   return GST_FLOW_OK;
 987
 988   /* ERRORS */
 989 stopped:
 990   {
 991     GST_DEBUG_OBJECT (pool, "stop called");
 992     return GST_FLOW_FLUSHING;
 993   }
 994 select_error:
 995   {
 996     GST_ELEMENT_ERROR (pool->obj->element, RESOURCE, READ, (NULL),
 997         ("poll error %d: %s (%d)", ret, g_strerror (errno), errno));
 998     return GST_FLOW_ERROR;
 999   }
1000 }
1001
1002 static GstFlowReturn
1003 gst_v4l2_buffer_pool_qbuf (GstV4l2BufferPool * pool, GstBuffer * buf)
1004 {
1005   GstV4l2MemoryGroup *group = NULL;
1006   gint index;
1007
1008   if (!gst_v4l2_is_buffer_valid (buf, &group)) {
1009     GST_LOG_OBJECT (pool, "unref copied/invalid buffer %p", buf);
1010     gst_buffer_unref (buf);
1011     return GST_FLOW_OK;
1012   }
1013
1014   index = group->buffer.index;
1015
1016   if (pool->buffers[index] != NULL)
1017     goto already_queued;
1018
1019   GST_LOG_OBJECT (pool, "queuing buffer %i", index);
1020
1021   g_atomic_int_inc (&pool->num_queued);
1022   pool->buffers[index] = buf;
1023
1024   if (!gst_v4l2_allocator_qbuf (pool->vallocator, group))
1025     goto queue_failed;
1026
1027   GST_OBJECT_LOCK (pool);
1028   pool->empty = FALSE;
1029   g_cond_signal (&pool->empty_cond);
1030   GST_OBJECT_UNLOCK (pool);
1031
1032   return GST_FLOW_OK;
1033
1034 already_queued:
1035   {
1036     GST_ERROR_OBJECT (pool, "the buffer %i was already queued", index);
1037     return GST_FLOW_ERROR;
1038   }
1039 queue_failed:
1040   {
1041     GST_ERROR_OBJECT (pool, "could not queue a buffer %i", index);
1042     /* Mark broken buffer to the allocator */
1043     GST_BUFFER_FLAG_SET (buf, GST_BUFFER_FLAG_TAG_MEMORY);
1044     g_atomic_int_add (&pool->num_queued, -1);
1045     pool->buffers[index] = NULL;
1046     return GST_FLOW_ERROR;
1047   }
1048 }
1049
1050 static GstFlowReturn
1051 gst_v4l2_buffer_pool_dqbuf (GstV4l2BufferPool * pool, GstBuffer ** buffer)
1052 {
1053   GstFlowReturn res;
1054   GstBuffer *outbuf;
1055   GstV4l2Object *obj = pool->obj;
1056   GstClockTime timestamp;
1057   GstV4l2MemoryGroup *group;
1058   gint i;
1059
1060   if ((res = gst_v4l2_buffer_pool_poll (pool)) != GST_FLOW_OK)
1061     goto poll_failed;
1062
1063   GST_LOG_OBJECT (pool, "dequeueing a buffer");
1064
1065   group = gst_v4l2_allocator_dqbuf (pool->vallocator);
1066   if (group == NULL)
1067     goto dqbuf_failed;
1068
1069   /* get our GstBuffer with that index from the pool, if the buffer was
1070    * outstanding we have a serious problem.
1071    */
1072   outbuf = pool->buffers[group->buffer.index];
1073   if (outbuf == NULL)
1074     goto no_buffer;
1075
1076   /* mark the buffer outstanding */
1077   pool->buffers[group->buffer.index] = NULL;
1078   if (g_atomic_int_dec_and_test (&pool->num_queued)) {
1079     GST_OBJECT_LOCK (pool);
1080     pool->empty = TRUE;
1081     GST_OBJECT_UNLOCK (pool);
1082   }
1083
1084   timestamp = GST_TIMEVAL_TO_TIME (group->buffer.timestamp);
1085
1086 #ifndef GST_DISABLE_GST_DEBUG
1087   for (i = 0; i < group->n_mem; i++) {
1088     GST_LOG_OBJECT (pool,
1089         "dequeued buffer %p seq:%d (ix=%d), mem %p used %d, plane=%d, flags %08x, ts %"
1090         GST_TIME_FORMAT ", pool-queued=%d, buffer=%p", outbuf,
1091         group->buffer.sequence, group->buffer.index, group->mem[i],
1092         group->planes[i].bytesused, i, group->buffer.flags,
1093         GST_TIME_ARGS (timestamp), pool->num_queued, outbuf);
1094   }
1095 #endif
1096
1097   /* set top/bottom field first if v4l2_buffer has the information */
1098   switch (group->buffer.field) {
1099     case V4L2_FIELD_NONE:
1100       GST_BUFFER_FLAG_UNSET (outbuf, GST_VIDEO_BUFFER_FLAG_INTERLACED);
1101       GST_BUFFER_FLAG_UNSET (outbuf, GST_VIDEO_BUFFER_FLAG_TFF);
1102       break;
1103     case V4L2_FIELD_INTERLACED_TB:
1104       GST_BUFFER_FLAG_SET (outbuf, GST_VIDEO_BUFFER_FLAG_INTERLACED);
1105       GST_BUFFER_FLAG_SET (outbuf, GST_VIDEO_BUFFER_FLAG_TFF);
1106       break;
1107     case V4L2_FIELD_INTERLACED_BT:
1108       GST_BUFFER_FLAG_SET (outbuf, GST_VIDEO_BUFFER_FLAG_INTERLACED);
1109       GST_BUFFER_FLAG_UNSET (outbuf, GST_VIDEO_BUFFER_FLAG_TFF);
1110       break;
1111     case V4L2_FIELD_INTERLACED:
1112       GST_BUFFER_FLAG_SET (outbuf, GST_VIDEO_BUFFER_FLAG_INTERLACED);
1113       if (obj->tv_norm == V4L2_STD_NTSC_M ||
1114           obj->tv_norm == V4L2_STD_NTSC_M_JP ||
1115           obj->tv_norm == V4L2_STD_NTSC_M_KR) {
1116         GST_BUFFER_FLAG_UNSET (outbuf, GST_VIDEO_BUFFER_FLAG_TFF);
1117       } else {
1118         GST_BUFFER_FLAG_SET (outbuf, GST_VIDEO_BUFFER_FLAG_TFF);
1119       }
1120       break;
1121     default:
1122       GST_BUFFER_FLAG_UNSET (outbuf, GST_VIDEO_BUFFER_FLAG_INTERLACED);
1123       GST_BUFFER_FLAG_UNSET (outbuf, GST_VIDEO_BUFFER_FLAG_TFF);
1124       GST_FIXME_OBJECT (pool,
1125           "Unhandled enum v4l2_field %d - treating as progressive",
1126           group->buffer.field);
1127   }
1128
1129   if (GST_VIDEO_INFO_FORMAT (&obj->info) == GST_VIDEO_FORMAT_ENCODED) {
1130     if (group->buffer.flags & V4L2_BUF_FLAG_KEYFRAME)
1131       GST_BUFFER_FLAG_UNSET (outbuf, GST_BUFFER_FLAG_DELTA_UNIT);
1132     else
1133       GST_BUFFER_FLAG_SET (outbuf, GST_BUFFER_FLAG_DELTA_UNIT);
1134   }
1135
1136   if (group->buffer.flags & V4L2_BUF_FLAG_ERROR)
1137     GST_BUFFER_FLAG_SET (outbuf, GST_BUFFER_FLAG_CORRUPTED);
1138
1139   GST_BUFFER_TIMESTAMP (outbuf) = timestamp;
1140
1141   *buffer = outbuf;
1142
1143   return GST_FLOW_OK;
1144
1145   /* ERRORS */
1146 poll_failed:
1147   {
1148     GST_DEBUG_OBJECT (pool, "poll error %s", gst_flow_get_name (res));
1149     return res;
1150   }
1151 dqbuf_failed:
1152   {
1153     return GST_FLOW_ERROR;
1154   }
1155 no_buffer:
1156   {
1157     GST_ERROR_OBJECT (pool, "No free buffer found in the pool at index %d.",
1158         group->buffer.index);
1159     return GST_FLOW_ERROR;
1160   }
1161 }
1162
1163 static GstFlowReturn
1164 gst_v4l2_buffer_pool_acquire_buffer (GstBufferPool * bpool, GstBuffer ** buffer,
1165     GstBufferPoolAcquireParams * params)
1166 {
1167   GstFlowReturn ret;
1168   GstV4l2BufferPool *pool = GST_V4L2_BUFFER_POOL (bpool);
1169   GstBufferPoolClass *pclass = GST_BUFFER_POOL_CLASS (parent_class);
1170   GstV4l2Object *obj = pool->obj;
1171
1172   GST_DEBUG_OBJECT (pool, "acquire");
1173
1174   /* If this is being called to resurect a lost buffer */
1175   if (params && params->flags & GST_V4L2_BUFFER_POOL_ACQUIRE_FLAG_RESURRECT) {
1176     ret = pclass->acquire_buffer (bpool, buffer, params);
1177     goto done;
1178   }
1179
1180   switch (obj->type) {
1181     case V4L2_BUF_TYPE_VIDEO_CAPTURE:
1182     case V4L2_BUF_TYPE_VIDEO_CAPTURE_MPLANE:
1183       /* capture, This function should return a buffer with new captured data */
1184       switch (obj->mode) {
1185         case GST_V4L2_IO_RW:
1186         {
1187           /* take empty buffer from the pool */
1188           ret = pclass->acquire_buffer (bpool, buffer, params);
1189           break;
1190         }
1191         case GST_V4L2_IO_DMABUF:
1192         case GST_V4L2_IO_MMAP:
1193         case GST_V4L2_IO_USERPTR:
1194         case GST_V4L2_IO_DMABUF_IMPORT:
1195         {
1196           /* just dequeue a buffer, we basically use the queue of v4l2 as the
1197            * storage for our buffers. This function does poll first so we can
1198            * interrupt it fine. */
1199           ret = gst_v4l2_buffer_pool_dqbuf (pool, buffer);
1200           break;
1201         }
1202         default:
1203           ret = GST_FLOW_ERROR;
1204           g_assert_not_reached ();
1205           break;
1206       }
1207       break;
1208
1209
1210     case V4L2_BUF_TYPE_VIDEO_OUTPUT:
1211     case V4L2_BUF_TYPE_VIDEO_OUTPUT_MPLANE:
1212       /* playback, This function should return an empty buffer */
1213       switch (obj->mode) {
1214         case GST_V4L2_IO_RW:
1215           /* get an empty buffer */
1216           ret = pclass->acquire_buffer (bpool, buffer, params);
1217           break;
1218
1219         case GST_V4L2_IO_MMAP:
1220         case GST_V4L2_IO_DMABUF:
1221         case GST_V4L2_IO_USERPTR:
1222         case GST_V4L2_IO_DMABUF_IMPORT:
1223           /* get a free unqueued buffer */
1224           ret = pclass->acquire_buffer (bpool, buffer, params);
1225           break;
1226
1227         default:
1228           ret = GST_FLOW_ERROR;
1229           g_assert_not_reached ();
1230           break;
1231       }
1232       break;
1233
1234     default:
1235       ret = GST_FLOW_ERROR;
1236       g_assert_not_reached ();
1237       break;
1238   }
1239 done:
1240   return ret;
1241 }
1242
1243 static void
1244 gst_v4l2_buffer_pool_release_buffer (GstBufferPool * bpool, GstBuffer * buffer)
1245 {
1246   GstV4l2BufferPool *pool = GST_V4L2_BUFFER_POOL (bpool);
1247   GstBufferPoolClass *pclass = GST_BUFFER_POOL_CLASS (parent_class);
1248   GstV4l2Object *obj = pool->obj;
1249
1250   GST_DEBUG_OBJECT (pool, "release buffer %p", buffer);
1251
1252   switch (obj->type) {
1253     case V4L2_BUF_TYPE_VIDEO_CAPTURE:
1254     case V4L2_BUF_TYPE_VIDEO_CAPTURE_MPLANE:
1255       /* capture, put the buffer back in the queue so that we can refill it
1256        * later. */
1257       switch (obj->mode) {
1258         case GST_V4L2_IO_RW:
1259           /* release back in the pool */
1260           pclass->release_buffer (bpool, buffer);
1261           break;
1262
1263         case GST_V4L2_IO_DMABUF:
1264         case GST_V4L2_IO_MMAP:
1265         case GST_V4L2_IO_USERPTR:
1266         case GST_V4L2_IO_DMABUF_IMPORT:
1267         {
1268           if (gst_v4l2_is_buffer_valid (buffer, NULL)) {
1269             /* queue back in the device */
1270             if (pool->other_pool)
1271               gst_v4l2_buffer_pool_prepare_buffer (pool, buffer, NULL);
1272             if (gst_v4l2_buffer_pool_qbuf (pool, buffer) != GST_FLOW_OK)
1273               pclass->release_buffer (bpool, buffer);
1274           } else {
1275             /* Simply release invalide/modified buffer, the allocator will
1276              * give it back later */
1277             GST_BUFFER_FLAG_SET (buffer, GST_BUFFER_FLAG_TAG_MEMORY);
1278             pclass->release_buffer (bpool, buffer);
1279           }
1280           break;
1281         }
1282         default:
1283           g_assert_not_reached ();
1284           break;
1285       }
1286       break;
1287
1288     case V4L2_BUF_TYPE_VIDEO_OUTPUT:
1289     case V4L2_BUF_TYPE_VIDEO_OUTPUT_MPLANE:
1290       switch (obj->mode) {
1291         case GST_V4L2_IO_RW:
1292           /* release back in the pool */
1293           pclass->release_buffer (bpool, buffer);
1294           break;
1295
1296         case GST_V4L2_IO_MMAP:
1297         case GST_V4L2_IO_DMABUF:
1298         case GST_V4L2_IO_USERPTR:
1299         case GST_V4L2_IO_DMABUF_IMPORT:
1300         {
1301           GstV4l2MemoryGroup *group;
1302           guint index;
1303
1304           if (!gst_v4l2_is_buffer_valid (buffer, &group)) {
1305             /* Simply release invalide/modified buffer, the allocator will
1306              * give it back later */
1307             GST_BUFFER_FLAG_SET (buffer, GST_BUFFER_FLAG_TAG_MEMORY);
1308             pclass->release_buffer (bpool, buffer);
1309             break;
1310           }
1311
1312           index = group->buffer.index;
1313
1314           if (pool->buffers[index] == NULL) {
1315             GST_LOG_OBJECT (pool, "buffer %u not queued, putting on free list",
1316                 index);
1317
1318             /* Remove qdata, this will unmap any map data in userptr */
1319             gst_mini_object_set_qdata (GST_MINI_OBJECT (buffer),
1320                 GST_V4L2_IMPORT_QUARK, NULL, NULL);
1321
1322             /* reset to default size */
1323             gst_v4l2_allocator_reset_group (pool->vallocator, group);
1324
1325             /* playback, put the buffer back in the queue to refill later. */
1326             pclass->release_buffer (bpool, buffer);
1327           } else {
1328             /* the buffer is queued in the device but maybe not played yet. We just
1329              * leave it there and not make it available for future calls to acquire
1330              * for now. The buffer will be dequeued and reused later. */
1331             GST_LOG_OBJECT (pool, "buffer %u is queued", index);
1332           }
1333           break;
1334         }
1335
1336         default:
1337           g_assert_not_reached ();
1338           break;
1339       }
1340       break;
1341
1342     default:
1343       g_assert_not_reached ();
1344       break;
1345   }
1346 }
1347
1348 static void
1349 gst_v4l2_buffer_pool_dispose (GObject * object)
1350 {
1351   GstV4l2BufferPool *pool = GST_V4L2_BUFFER_POOL (object);
1352
1353   if (pool->vallocator)
1354     gst_object_unref (pool->vallocator);
1355   pool->vallocator = NULL;
1356
1357   if (pool->allocator)
1358     gst_object_unref (pool->allocator);
1359   pool->allocator = NULL;
1360
1361   if (pool->other_pool)
1362     gst_object_unref (pool->other_pool);
1363   pool->other_pool = NULL;
1364
1365   G_OBJECT_CLASS (parent_class)->dispose (object);
1366 }
1367
1368 static void
1369 gst_v4l2_buffer_pool_finalize (GObject * object)
1370 {
1371   GstV4l2BufferPool *pool = GST_V4L2_BUFFER_POOL (object);
1372
1373   if (pool->video_fd >= 0)
1374     v4l2_close (pool->video_fd);
1375
1376   gst_poll_free (pool->poll);
1377
1378   /* FIXME Is this required to keep around ?
1379    * This can't be done in dispose method because we must not set pointer
1380    * to NULL as it is part of the v4l2object and dispose could be called
1381    * multiple times */
1382   gst_object_unref (pool->obj->element);
1383
1384   g_cond_clear (&pool->empty_cond);
1385
1386   /* FIXME have we done enough here ? */
1387
1388   G_OBJECT_CLASS (parent_class)->finalize (object);
1389 }
1390
1391 static void
1392 gst_v4l2_buffer_pool_init (GstV4l2BufferPool * pool)
1393 {
1394   pool->poll = gst_poll_new (TRUE);
1395   pool->can_poll_device = TRUE;
1396   g_cond_init (&pool->empty_cond);
1397   pool->empty = TRUE;
1398 }
1399
1400 static void
1401 gst_v4l2_buffer_pool_class_init (GstV4l2BufferPoolClass * klass)
1402 {
1403   GObjectClass *object_class = G_OBJECT_CLASS (klass);
1404   GstBufferPoolClass *bufferpool_class = GST_BUFFER_POOL_CLASS (klass);
1405
1406   object_class->dispose = gst_v4l2_buffer_pool_dispose;
1407   object_class->finalize = gst_v4l2_buffer_pool_finalize;
1408
1409   bufferpool_class->start = gst_v4l2_buffer_pool_start;
1410   bufferpool_class->stop = gst_v4l2_buffer_pool_stop;
1411   bufferpool_class->set_config = gst_v4l2_buffer_pool_set_config;
1412   bufferpool_class->alloc_buffer = gst_v4l2_buffer_pool_alloc_buffer;
1413   bufferpool_class->acquire_buffer = gst_v4l2_buffer_pool_acquire_buffer;
1414   bufferpool_class->release_buffer = gst_v4l2_buffer_pool_release_buffer;
1415   bufferpool_class->flush_start = gst_v4l2_buffer_pool_flush_start;
1416   bufferpool_class->flush_stop = gst_v4l2_buffer_pool_flush_stop;
1417
1418   GST_DEBUG_CATEGORY_INIT (v4l2bufferpool_debug, "v4l2bufferpool", 0,
1419       "V4L2 Buffer Pool");
1420 }
1421
1422 /**
1423  * gst_v4l2_buffer_pool_new:
1424  * @obj:  the v4l2 object owning the pool
1425  *
1426  * Construct a new buffer pool.
1427  *
1428  * Returns: the new pool, use gst_object_unref() to free resources
1429  */
1430 GstBufferPool *
1431 gst_v4l2_buffer_pool_new (GstV4l2Object * obj, GstCaps * caps)
1432 {
1433   GstV4l2BufferPool *pool;
1434   GstStructure *config;
1435   gchar *name, *parent_name;
1436   gint fd;
1437
1438   fd = v4l2_dup (obj->video_fd);
1439   if (fd < 0)
1440     goto dup_failed;
1441
1442   /* setting a significant unique name */
1443   parent_name = gst_object_get_name (GST_OBJECT (obj->element));
1444   name = g_strconcat (parent_name, ":", "pool:",
1445       V4L2_TYPE_IS_OUTPUT (obj->type) ? "sink" : "src", NULL);
1446   g_free (parent_name);
1447
1448   pool = (GstV4l2BufferPool *) g_object_new (GST_TYPE_V4L2_BUFFER_POOL,
1449       "name", name, NULL);
1450   g_free (name);
1451
1452   gst_poll_fd_init (&pool->pollfd);
1453   pool->pollfd.fd = fd;
1454   gst_poll_add_fd (pool->poll, &pool->pollfd);
1455   if (V4L2_TYPE_IS_OUTPUT (obj->type))
1456     gst_poll_fd_ctl_write (pool->poll, &pool->pollfd, TRUE);
1457   else
1458     gst_poll_fd_ctl_read (pool->poll, &pool->pollfd, TRUE);
1459
1460   pool->video_fd = fd;
1461   pool->obj = obj;
1462   pool->can_poll_device = TRUE;
1463
1464   pool->vallocator =
1465       gst_v4l2_allocator_new (GST_OBJECT (pool), obj->video_fd, &obj->format);
1466   if (pool->vallocator == NULL)
1467     goto allocator_failed;
1468
1469   gst_object_ref (obj->element);
1470
1471   config = gst_buffer_pool_get_config (GST_BUFFER_POOL_CAST (pool));
1472   gst_buffer_pool_config_set_params (config, caps, obj->info.size, 0, 0);
1473   /* This will simply set a default config, but will not configure the pool
1474    * because min and max are not valid */
1475   gst_buffer_pool_set_config (GST_BUFFER_POOL_CAST (pool), config);
1476
1477   return GST_BUFFER_POOL (pool);
1478
1479   /* ERRORS */
1480 dup_failed:
1481   {
1482     GST_ERROR ("failed to dup fd %d (%s)", errno, g_strerror (errno));
1483     return NULL;
1484   }
1485 allocator_failed:
1486   {
1487     GST_ERROR_OBJECT (pool, "Failed to create V4L2 allocator");
1488     gst_object_unref (pool);
1489     return NULL;
1490   }
1491 }
1492
1493 static GstFlowReturn
1494 gst_v4l2_do_read (GstV4l2BufferPool * pool, GstBuffer * buf)
1495 {
1496   GstFlowReturn res;
1497   GstV4l2Object *obj = pool->obj;
1498   gint amount;
1499   GstMapInfo map;
1500   gint toread;
1501
1502   toread = obj->info.size;
1503
1504   GST_LOG_OBJECT (pool, "reading %d bytes into buffer %p", toread, buf);
1505
1506   gst_buffer_map (buf, &map, GST_MAP_WRITE);
1507
1508   do {
1509     if ((res = gst_v4l2_buffer_pool_poll (pool)) != GST_FLOW_OK)
1510       goto poll_error;
1511
1512     amount = v4l2_read (obj->video_fd, map.data, toread);
1513
1514     if (amount == toread) {
1515       break;
1516     } else if (amount == -1) {
1517       if (errno == EAGAIN || errno == EINTR) {
1518         continue;
1519       } else
1520         goto read_error;
1521     } else {
1522       /* short reads can happen if a signal interrupts the read */
1523       continue;
1524     }
1525   } while (TRUE);
1526
1527   GST_LOG_OBJECT (pool, "read %d bytes", amount);
1528   gst_buffer_unmap (buf, &map);
1529   gst_buffer_resize (buf, 0, amount);
1530
1531   return GST_FLOW_OK;
1532
1533   /* ERRORS */
1534 poll_error:
1535   {
1536     GST_DEBUG ("poll error %s", gst_flow_get_name (res));
1537     goto cleanup;
1538   }
1539 read_error:
1540   {
1541     GST_ELEMENT_ERROR (obj->element, RESOURCE, READ,
1542         (_("Error reading %d bytes from device '%s'."),
1543             toread, obj->videodev), GST_ERROR_SYSTEM);
1544     res = GST_FLOW_ERROR;
1545     goto cleanup;
1546   }
1547 cleanup:
1548   {
1549     gst_buffer_unmap (buf, &map);
1550     gst_buffer_resize (buf, 0, 0);
1551     return res;
1552   }
1553 }
1554
1555 /**
1556  * gst_v4l2_buffer_pool_process:
1557  * @bpool: a #GstBufferPool
1558  * @buf: a #GstBuffer, maybe be replaced
1559  *
1560  * Process @buf in @bpool. For capture devices, this functions fills @buf with
1561  * data from the device. For output devices, this functions send the contents of
1562  * @buf to the device for playback.
1563  *
1564  * Returns: %GST_FLOW_OK on success.
1565  */
1566 GstFlowReturn
1567 gst_v4l2_buffer_pool_process (GstV4l2BufferPool * pool, GstBuffer ** buf)
1568 {
1569   GstFlowReturn ret = GST_FLOW_OK;
1570   GstBufferPool *bpool = GST_BUFFER_POOL_CAST (pool);
1571   GstV4l2Object *obj = pool->obj;
1572
1573   GST_DEBUG_OBJECT (pool, "process buffer %p", buf);
1574
1575   g_return_val_if_fail (gst_buffer_pool_is_active (bpool), GST_FLOW_ERROR);
1576
1577   if (GST_BUFFER_POOL_IS_FLUSHING (pool))
1578     return GST_FLOW_FLUSHING;
1579
1580   switch (obj->type) {
1581     case V4L2_BUF_TYPE_VIDEO_CAPTURE:
1582     case V4L2_BUF_TYPE_VIDEO_CAPTURE_MPLANE:
1583       /* capture */
1584       switch (obj->mode) {
1585         case GST_V4L2_IO_RW:
1586           /* capture into the buffer */
1587           ret = gst_v4l2_do_read (pool, *buf);
1588           break;
1589
1590         case GST_V4L2_IO_MMAP:
1591         case GST_V4L2_IO_DMABUF:
1592         {
1593           GstBuffer *tmp;
1594
1595           if ((*buf)->pool == bpool) {
1596             guint num_queued;
1597
1598             if (gst_buffer_get_size (*buf) == 0) {
1599               if (GST_BUFFER_FLAG_IS_SET (*buf, GST_BUFFER_FLAG_CORRUPTED))
1600                 goto buffer_corrupted;
1601               else
1602                 goto eos;
1603             }
1604
1605             num_queued = g_atomic_int_get (&pool->num_queued);
1606             GST_TRACE_OBJECT (pool, "Only %i buffer left in the capture queue.",
1607                 num_queued);
1608
1609             /* If we have no more buffer, and can allocate it time to do so */
1610             if (num_queued == 0) {
1611               if (GST_V4L2_ALLOCATOR_CAN_ALLOCATE (pool->vallocator, MMAP)) {
1612                 ret = gst_v4l2_buffer_pool_resurect_buffer (pool);
1613                 if (ret == GST_FLOW_OK)
1614                   goto done;
1615               }
1616             }
1617
1618             /* start copying buffers when we are running low on buffers */
1619             if (num_queued < pool->copy_threshold) {
1620               GstBuffer *copy;
1621
1622               if (GST_V4L2_ALLOCATOR_CAN_ALLOCATE (pool->vallocator, MMAP)) {
1623                 ret = gst_v4l2_buffer_pool_resurect_buffer (pool);
1624                 if (ret == GST_FLOW_OK)
1625                   goto done;
1626               }
1627
1628               /* copy the buffer */
1629               copy = gst_buffer_copy_region (*buf,
1630                   GST_BUFFER_COPY_ALL | GST_BUFFER_COPY_DEEP, 0, -1);
1631               GST_LOG_OBJECT (pool, "copy buffer %p->%p", *buf, copy);
1632
1633               /* and requeue so that we can continue capturing */
1634               gst_buffer_unref (*buf);
1635               *buf = copy;
1636             }
1637
1638             /* nothing, data was inside the buffer when we did _acquire() */
1639             goto done;
1640           }
1641
1642           /* buffer not from our pool, grab a frame and copy it into the target */
1643           if ((ret = gst_v4l2_buffer_pool_dqbuf (pool, &tmp)) != GST_FLOW_OK)
1644             goto done;
1645
1646           /* An empty buffer on capture indicates the end of stream */
1647           if (gst_buffer_get_size (tmp) == 0) {
1648             gst_v4l2_buffer_pool_release_buffer (bpool, tmp);
1649
1650             if (GST_BUFFER_FLAG_IS_SET (*buf, GST_BUFFER_FLAG_CORRUPTED))
1651               goto buffer_corrupted;
1652             else
1653               goto eos;
1654           }
1655
1656           ret = gst_v4l2_buffer_pool_copy_buffer (pool, *buf, tmp);
1657
1658           /* an queue the buffer again after the copy */
1659           gst_v4l2_buffer_pool_release_buffer (bpool, tmp);
1660
1661           if (ret != GST_FLOW_OK)
1662             goto copy_failed;
1663           break;
1664         }
1665
1666         case GST_V4L2_IO_USERPTR:
1667         {
1668           struct UserPtrData *data;
1669
1670           /* Replace our buffer with downstream allocated buffer */
1671           data = gst_mini_object_steal_qdata (GST_MINI_OBJECT (*buf),
1672               GST_V4L2_IMPORT_QUARK);
1673           gst_buffer_replace (buf, data->buffer);
1674           _unmap_userptr_frame (data);
1675           break;
1676         }
1677
1678         case GST_V4L2_IO_DMABUF_IMPORT:
1679         {
1680           GstBuffer *tmp;
1681
1682           /* Replace our buffer with downstream allocated buffer */
1683           tmp = gst_mini_object_steal_qdata (GST_MINI_OBJECT (*buf),
1684               GST_V4L2_IMPORT_QUARK);
1685           gst_buffer_replace (buf, tmp);
1686           gst_buffer_unref (tmp);
1687           break;
1688         }
1689
1690         default:
1691           g_assert_not_reached ();
1692           break;
1693       }
1694       break;
1695
1696     case V4L2_BUF_TYPE_VIDEO_OUTPUT:
1697     case V4L2_BUF_TYPE_VIDEO_OUTPUT_MPLANE:
1698       /* playback */
1699       switch (obj->mode) {
1700         case GST_V4L2_IO_RW:
1701           /* FIXME, do write() */
1702           GST_WARNING_OBJECT (pool, "implement write()");
1703           break;
1704
1705         case GST_V4L2_IO_USERPTR:
1706         case GST_V4L2_IO_DMABUF_IMPORT:
1707         case GST_V4L2_IO_DMABUF:
1708         case GST_V4L2_IO_MMAP:
1709         {
1710           GstBuffer *to_queue = NULL;
1711           GstV4l2MemoryGroup *group;
1712           gint index;
1713
1714           if ((*buf)->pool != bpool)
1715             goto copying;
1716
1717           if (!gst_v4l2_is_buffer_valid (*buf, &group))
1718             goto copying;
1719
1720           index = group->buffer.index;
1721
1722           GST_LOG_OBJECT (pool, "processing buffer %i from our pool", index);
1723
1724           index = group->buffer.index;
1725           if (pool->buffers[index] != NULL) {
1726             GST_LOG_OBJECT (pool, "buffer %i already queued, copying", index);
1727             goto copying;
1728           }
1729
1730           /* we can queue directly */
1731           to_queue = gst_buffer_ref (*buf);
1732
1733         copying:
1734           if (to_queue == NULL) {
1735             GstBufferPoolAcquireParams params = { 0 };
1736
1737             GST_LOG_OBJECT (pool, "alloc buffer from our pool");
1738
1739             /* this can return EOS if all buffers are outstanding which would
1740              * be strange because we would expect the upstream element to have
1741              * allocated them and returned to us.. */
1742             params.flags = GST_BUFFER_POOL_ACQUIRE_FLAG_DONTWAIT;
1743             ret = gst_buffer_pool_acquire_buffer (bpool, &to_queue, &params);
1744             if (ret != GST_FLOW_OK)
1745               goto acquire_failed;
1746
1747             ret = gst_v4l2_buffer_pool_prepare_buffer (pool, to_queue, *buf);
1748             if (ret != GST_FLOW_OK) {
1749               gst_buffer_unref (to_queue);
1750               goto prepare_failed;
1751             }
1752           }
1753
1754           if ((ret = gst_v4l2_buffer_pool_qbuf (pool, to_queue)) != GST_FLOW_OK)
1755             goto queue_failed;
1756
1757           /* if we are not streaming yet (this is the first buffer, start
1758            * streaming now */
1759           if (!gst_v4l2_buffer_pool_streamon (pool)) {
1760             /* don't check return value because qbuf would have failed */
1761             gst_v4l2_is_buffer_valid (to_queue, &group);
1762
1763             /* qbuf has stored to_queue buffer but we are not in
1764              * streaming state, so the flush logic won't be performed.
1765              * To avoid leaks, flush the allocator and restore the queued
1766              * buffer as non-queued */
1767             gst_v4l2_allocator_flush (pool->vallocator);
1768
1769             pool->buffers[group->buffer.index] = NULL;
1770
1771             gst_mini_object_set_qdata (GST_MINI_OBJECT (to_queue),
1772                 GST_V4L2_IMPORT_QUARK, NULL, NULL);
1773             gst_buffer_unref (to_queue);
1774             g_atomic_int_add (&pool->num_queued, -1);
1775             goto start_failed;
1776           }
1777
1778           /* Remove our ref, we will still hold this buffer in acquire as needed,
1779            * otherwise the pool will think it is outstanding and will refuse to stop. */
1780           gst_buffer_unref (to_queue);
1781
1782           if (g_atomic_int_get (&pool->num_queued) >= pool->min_latency) {
1783             GstBuffer *out;
1784             /* all buffers are queued, try to dequeue one and release it back
1785              * into the pool so that _acquire can get to it again. */
1786             ret = gst_v4l2_buffer_pool_dqbuf (pool, &out);
1787             if (ret == GST_FLOW_OK && out->pool == NULL)
1788               /* release the rendered buffer back into the pool. This wakes up any
1789                * thread waiting for a buffer in _acquire(). */
1790               gst_v4l2_buffer_pool_release_buffer (bpool, out);
1791           }
1792           break;
1793         }
1794         default:
1795           g_assert_not_reached ();
1796           break;
1797       }
1798       break;
1799     default:
1800       g_assert_not_reached ();
1801       break;
1802   }
1803 done:
1804   return ret;
1805
1806   /* ERRORS */
1807 copy_failed:
1808   {
1809     GST_ERROR_OBJECT (pool, "failed to copy buffer");
1810     return ret;
1811   }
1812 buffer_corrupted:
1813   {
1814     GST_WARNING_OBJECT (pool, "Dropping corrupted buffer without payload");
1815     gst_buffer_unref (*buf);
1816     *buf = NULL;
1817     return GST_V4L2_FLOW_CORRUPTED_BUFFER;
1818   }
1819 eos:
1820   {
1821     GST_DEBUG_OBJECT (pool, "end of stream reached");
1822     gst_buffer_unref (*buf);
1823     *buf = NULL;
1824     return GST_V4L2_FLOW_LAST_BUFFER;
1825   }
1826 acquire_failed:
1827   {
1828     if (ret == GST_FLOW_FLUSHING)
1829       GST_DEBUG_OBJECT (pool, "flushing");
1830     else
1831       GST_WARNING_OBJECT (pool, "failed to acquire a buffer: %s",
1832           gst_flow_get_name (ret));
1833     return ret;
1834   }
1835 prepare_failed:
1836   {
1837     GST_ERROR_OBJECT (pool, "failed to prepare data");
1838     return ret;
1839   }
1840 queue_failed:
1841   {
1842     GST_ERROR_OBJECT (pool, "failed to queue buffer");
1843     return ret;
1844   }
1845 start_failed:
1846   {
1847     GST_ERROR_OBJECT (pool, "failed to start streaming");
1848     return GST_FLOW_ERROR;
1849   }
1850 }
1851
1852 void
1853 gst_v4l2_buffer_pool_set_other_pool (GstV4l2BufferPool * pool,
1854     GstBufferPool * other_pool)
1855 {
1856   g_return_if_fail (!gst_buffer_pool_is_active (GST_BUFFER_POOL (pool)));
1857
1858   if (pool->other_pool)
1859     gst_object_unref (pool->other_pool);
1860   pool->other_pool = gst_object_ref (other_pool);
1861 }
1862
1863 void
1864 gst_v4l2_buffer_pool_copy_at_threshold (GstV4l2BufferPool * pool, gboolean copy)
1865 {
1866   GST_OBJECT_LOCK (pool);
1867   pool->enable_copy_threshold = copy;
1868   GST_OBJECT_UNLOCK (pool);
1869 }