gst/gstbufferpool.c

   1 /* GStreamer
   2  * Copyright (C) 2010 Wim Taymans <wim.taymans@gmail.com>
   3  *
   4  * gstbufferpool.c: GstBufferPool baseclass
   5  *
   6  * This library is free software; you can redistribute it and/or
   7  * modify it under the terms of the GNU Library General Public
   8  * License as published by the Free Software Foundation; either
   9  * version 2 of the License, or (at your option) any later version.
  10  *
  11  * This library is distributed in the hope that it will be useful,
  12  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  13  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
  14  * Library General Public License for more details.
  15  *
  16  * You should have received a copy of the GNU Library General Public
  17  * License along with this library; if not, write to the
  18  * Free Software Foundation, Inc., 51 Franklin St, Fifth Floor,
  19  * Boston, MA 02110-1301, USA.
  20  */
  21
  22 /**
  23  * SECTION:gstbufferpool
  24  * @short_description: Pool for buffers
  25  * @see_also: #GstBuffer
  26  *
  27  * A #GstBufferPool is an object that can be used to pre-allocate and recycle
  28  * buffers of the same size and with the same properties.
  29  *
  30  * A #GstBufferPool is created with gst_buffer_pool_new().
  31  *
  32  * Once a pool is created, it needs to be configured. A call to
  33  * gst_buffer_pool_get_config() returns the current configuration structure from
  34  * the pool. With gst_buffer_pool_config_set_params() and
  35  * gst_buffer_pool_config_set_allocator() the bufferpool parameters and
  36  * allocator can be configured. Other properties can be configured in the pool
  37  * depending on the pool implementation.
  38  *
  39  * A bufferpool can have extra options that can be enabled with
  40  * gst_buffer_pool_config_add_option(). The available options can be retrieved
  41  * with gst_buffer_pool_get_options(). Some options allow for additional
  42  * configuration properties to be set.
  43  *
  44  * After the configuration structure has been configured,
  45  * gst_buffer_pool_set_config() updates the configuration in the pool. This can
  46  * fail when the configuration structure is not accepted.
  47  *
  48  * After the a pool has been configured, it can be activated with
  49  * gst_buffer_pool_set_active(). This will preallocate the configured resources
  50  * in the pool.
  51  *
  52  * When the pool is active, gst_buffer_pool_acquire_buffer() can be used to
  53  * retrieve a buffer from the pool.
  54  *
  55  * Buffers allocated from a bufferpool will automatically be returned to the
  56  * pool with gst_buffer_pool_release_buffer() when their refcount drops to 0.
  57  *
  58  * The bufferpool can be deactivated again with gst_buffer_pool_set_active().
  59  * All further gst_buffer_pool_acquire_buffer() calls will return an error. When
  60  * all buffers are returned to the pool they will be freed.
  61  *
  62  * Use gst_object_unref() to release the reference to a bufferpool. If the
  63  * refcount of the pool reaches 0, the pool will be freed.
  64  */
  65
  66 #include "gst_private.h"
  67 #include "glib-compat-private.h"
  68
  69 #include <errno.h>
  70 #ifdef HAVE_UNISTD_H
  71 #  include <unistd.h>
  72 #endif
  73 #include <sys/types.h>
  74
  75 #include "gstatomicqueue.h"
  76 #include "gstpoll.h"
  77 #include "gstinfo.h"
  78 #include "gstquark.h"
  79 #include "gstvalue.h"
  80
  81 #include "gstbufferpool.h"
  82
  83 GST_DEBUG_CATEGORY_STATIC (gst_buffer_pool_debug);
  84 #define GST_CAT_DEFAULT gst_buffer_pool_debug
  85
  86 #define GST_BUFFER_POOL_GET_PRIVATE(obj)  \
  87    (G_TYPE_INSTANCE_GET_PRIVATE ((obj), GST_TYPE_BUFFER_POOL, GstBufferPoolPrivate))
  88
  89 #define GST_BUFFER_POOL_LOCK(pool)   (g_rec_mutex_lock(&pool->priv->rec_lock))
  90 #define GST_BUFFER_POOL_UNLOCK(pool) (g_rec_mutex_unlock(&pool->priv->rec_lock))
  91
  92 struct _GstBufferPoolPrivate
  93 {
  94   GstAtomicQueue *queue;
  95   GstPoll *poll;
  96
  97   GRecMutex rec_lock;
  98
  99   gboolean started;
 100   gboolean active;
 101   gint outstanding;             /* number of buffers that are in use */
 102
 103   gboolean configured;
 104   GstStructure *config;
 105
 106   guint size;
 107   guint min_buffers;
 108   guint max_buffers;
 109   guint cur_buffers;
 110   GstAllocator *allocator;
 111   GstAllocationParams params;
 112 };
 113
 114 static void gst_buffer_pool_finalize (GObject * object);
 115
 116 G_DEFINE_TYPE (GstBufferPool, gst_buffer_pool, GST_TYPE_OBJECT);
 117
 118 static gboolean default_start (GstBufferPool * pool);
 119 static gboolean default_stop (GstBufferPool * pool);
 120 static gboolean default_set_config (GstBufferPool * pool,
 121     GstStructure * config);
 122 static GstFlowReturn default_alloc_buffer (GstBufferPool * pool,
 123     GstBuffer ** buffer, GstBufferPoolAcquireParams * params);
 124 static GstFlowReturn default_acquire_buffer (GstBufferPool * pool,
 125     GstBuffer ** buffer, GstBufferPoolAcquireParams * params);
 126 static void default_reset_buffer (GstBufferPool * pool, GstBuffer * buffer);
 127 static void default_free_buffer (GstBufferPool * pool, GstBuffer * buffer);
 128 static void default_release_buffer (GstBufferPool * pool, GstBuffer * buffer);
 129
 130 static void
 131 gst_buffer_pool_class_init (GstBufferPoolClass * klass)
 132 {
 133   GObjectClass *gobject_class = (GObjectClass *) klass;
 134
 135   g_type_class_add_private (klass, sizeof (GstBufferPoolPrivate));
 136
 137   gobject_class->finalize = gst_buffer_pool_finalize;
 138
 139   klass->start = default_start;
 140   klass->stop = default_stop;
 141   klass->set_config = default_set_config;
 142   klass->acquire_buffer = default_acquire_buffer;
 143   klass->reset_buffer = default_reset_buffer;
 144   klass->alloc_buffer = default_alloc_buffer;
 145   klass->release_buffer = default_release_buffer;
 146   klass->free_buffer = default_free_buffer;
 147
 148   GST_DEBUG_CATEGORY_INIT (gst_buffer_pool_debug, "bufferpool", 0,
 149       "bufferpool debug");
 150 }
 151
 152 static void
 153 gst_buffer_pool_init (GstBufferPool * pool)
 154 {
 155   GstBufferPoolPrivate *priv;
 156
 157   priv = pool->priv = GST_BUFFER_POOL_GET_PRIVATE (pool);
 158
 159   g_rec_mutex_init (&priv->rec_lock);
 160
 161   priv->poll = gst_poll_new_timer ();
 162   priv->queue = gst_atomic_queue_new (16);
 163   pool->flushing = 1;
 164   priv->active = FALSE;
 165   priv->configured = FALSE;
 166   priv->started = FALSE;
 167   priv->config = gst_structure_new_id_empty (GST_QUARK (BUFFER_POOL_CONFIG));
 168   gst_buffer_pool_config_set_params (priv->config, NULL, 0, 0, 0);
 169   priv->allocator = NULL;
 170   gst_allocation_params_init (&priv->params);
 171   gst_buffer_pool_config_set_allocator (priv->config, priv->allocator,
 172       &priv->params);
 173   /* 1 control write for flushing */
 174   gst_poll_write_control (priv->poll);
 175   /* 1 control write for marking that we are not waiting for poll */
 176   gst_poll_write_control (priv->poll);
 177
 178   GST_DEBUG_OBJECT (pool, "created");
 179 }
 180
 181 static void
 182 gst_buffer_pool_finalize (GObject * object)
 183 {
 184   GstBufferPool *pool;
 185   GstBufferPoolPrivate *priv;
 186
 187   pool = GST_BUFFER_POOL_CAST (object);
 188   priv = pool->priv;
 189
 190   GST_DEBUG_OBJECT (pool, "finalize");
 191
 192   gst_buffer_pool_set_active (pool, FALSE);
 193   gst_atomic_queue_unref (priv->queue);
 194   gst_poll_free (priv->poll);
 195   gst_structure_free (priv->config);
 196   g_rec_mutex_clear (&priv->rec_lock);
 197   if (priv->allocator)
 198     gst_object_unref (priv->allocator);
 199
 200   G_OBJECT_CLASS (gst_buffer_pool_parent_class)->finalize (object);
 201 }
 202
 203 /**
 204  * gst_buffer_pool_new:
 205  *
 206  * Creates a new #GstBufferPool instance.
 207  *
 208  * Returns: (transfer floating): a new #GstBufferPool instance
 209  */
 210 GstBufferPool *
 211 gst_buffer_pool_new (void)
 212 {
 213   GstBufferPool *result;
 214
 215   result = g_object_newv (GST_TYPE_BUFFER_POOL, 0, NULL);
 216   GST_DEBUG_OBJECT (result, "created new buffer pool");
 217
 218   return result;
 219 }
 220
 221 static GstFlowReturn
 222 default_alloc_buffer (GstBufferPool * pool, GstBuffer ** buffer,
 223     GstBufferPoolAcquireParams * params)
 224 {
 225   GstBufferPoolPrivate *priv = pool->priv;
 226
 227   *buffer =
 228       gst_buffer_new_allocate (priv->allocator, priv->size, &priv->params);
 229
 230   return GST_FLOW_OK;
 231 }
 232
 233 static gboolean
 234 mark_meta_pooled (GstBuffer * buffer, GstMeta ** meta, gpointer user_data)
 235 {
 236   GstBufferPool *pool = user_data;
 237
 238   GST_DEBUG_OBJECT (pool, "marking meta %p as POOLED in buffer %p", *meta,
 239       buffer);
 240   GST_META_FLAG_SET (*meta, GST_META_FLAG_POOLED);
 241   GST_META_FLAG_SET (*meta, GST_META_FLAG_LOCKED);
 242
 243   return TRUE;
 244 }
 245
 246 static GstFlowReturn
 247 do_alloc_buffer (GstBufferPool * pool, GstBuffer ** buffer,
 248     GstBufferPoolAcquireParams * params)
 249 {
 250   GstBufferPoolPrivate *priv = pool->priv;
 251   GstFlowReturn result;
 252   gint cur_buffers, max_buffers;
 253   GstBufferPoolClass *pclass;
 254
 255   pclass = GST_BUFFER_POOL_GET_CLASS (pool);
 256
 257   if (G_UNLIKELY (!pclass->alloc_buffer))
 258     goto no_function;
 259
 260   max_buffers = priv->max_buffers;
 261
 262   /* increment the allocation counter */
 263   cur_buffers = g_atomic_int_add (&priv->cur_buffers, 1);
 264   if (max_buffers && cur_buffers >= max_buffers)
 265     goto max_reached;
 266
 267   result = pclass->alloc_buffer (pool, buffer, params);
 268   if (G_UNLIKELY (result != GST_FLOW_OK))
 269     goto alloc_failed;
 270
 271   /* lock all metadata and mark as pooled, we want this to remain on
 272    * the buffer and we want to remove any other metadata that gets added
 273    * later */
 274   gst_buffer_foreach_meta (*buffer, mark_meta_pooled, pool);
 275
 276   /* un-tag memory, this is how we expect the buffer when it is
 277    * released again */
 278   GST_BUFFER_FLAG_UNSET (*buffer, GST_BUFFER_FLAG_TAG_MEMORY);
 279
 280   GST_LOG_OBJECT (pool, "allocated buffer %d/%d, %p", cur_buffers,
 281       max_buffers, *buffer);
 282
 283   return result;
 284
 285   /* ERRORS */
 286 no_function:
 287   {
 288     GST_ERROR_OBJECT (pool, "no alloc function");
 289     return GST_FLOW_NOT_SUPPORTED;
 290   }
 291 max_reached:
 292   {
 293     GST_DEBUG_OBJECT (pool, "max buffers reached");
 294     g_atomic_int_add (&priv->cur_buffers, -1);
 295     return GST_FLOW_EOS;
 296   }
 297 alloc_failed:
 298   {
 299     GST_WARNING_OBJECT (pool, "alloc function failed");
 300     g_atomic_int_add (&priv->cur_buffers, -1);
 301     return result;
 302   }
 303 }
 304
 305 /* the default implementation for preallocating the buffers in the pool */
 306 static gboolean
 307 default_start (GstBufferPool * pool)
 308 {
 309   guint i;
 310   GstBufferPoolPrivate *priv = pool->priv;
 311   GstBufferPoolClass *pclass;
 312
 313   pclass = GST_BUFFER_POOL_GET_CLASS (pool);
 314
 315   /* we need to prealloc buffers */
 316   for (i = 0; i < priv->min_buffers; i++) {
 317     GstBuffer *buffer;
 318
 319     if (do_alloc_buffer (pool, &buffer, NULL) != GST_FLOW_OK)
 320       goto alloc_failed;
 321
 322     /* release to the queue, we call the vmethod directly, we don't need to do
 323      * the other refcount handling right now. */
 324     if (G_LIKELY (pclass->release_buffer))
 325       pclass->release_buffer (pool, buffer);
 326   }
 327   return TRUE;
 328
 329   /* ERRORS */
 330 alloc_failed:
 331   {
 332     GST_WARNING_OBJECT (pool, "failed to allocate buffer");
 333     return FALSE;
 334   }
 335 }
 336
 337 /* must be called with the lock */
 338 static gboolean
 339 do_start (GstBufferPool * pool)
 340 {
 341   GstBufferPoolPrivate *priv = pool->priv;
 342
 343   if (!priv->started) {
 344     GstBufferPoolClass *pclass;
 345
 346     pclass = GST_BUFFER_POOL_GET_CLASS (pool);
 347
 348     GST_LOG_OBJECT (pool, "starting");
 349     /* start the pool, subclasses should allocate buffers and put them
 350      * in the queue */
 351     if (G_LIKELY (pclass->start)) {
 352       if (!pclass->start (pool))
 353         return FALSE;
 354     }
 355     priv->started = TRUE;
 356   }
 357   return TRUE;
 358 }
 359
 360
 361 static void
 362 default_free_buffer (GstBufferPool * pool, GstBuffer * buffer)
 363 {
 364   gst_buffer_unref (buffer);
 365 }
 366
 367 static void
 368 do_free_buffer (GstBufferPool * pool, GstBuffer * buffer)
 369 {
 370   GstBufferPoolPrivate *priv;
 371   GstBufferPoolClass *pclass;
 372
 373   priv = pool->priv;
 374   pclass = GST_BUFFER_POOL_GET_CLASS (pool);
 375
 376   g_atomic_int_add (&priv->cur_buffers, -1);
 377   GST_LOG_OBJECT (pool, "freeing buffer %p (%u left)", buffer,
 378       priv->cur_buffers);
 379
 380   if (G_LIKELY (pclass->free_buffer))
 381     pclass->free_buffer (pool, buffer);
 382 }
 383
 384 /* must be called with the lock */
 385 static gboolean
 386 default_stop (GstBufferPool * pool)
 387 {
 388   GstBufferPoolPrivate *priv = pool->priv;
 389   GstBuffer *buffer;
 390
 391   /* clear the pool */
 392   while ((buffer = gst_atomic_queue_pop (priv->queue))) {
 393     gst_poll_read_control (priv->poll);
 394     do_free_buffer (pool, buffer);
 395   }
 396   return priv->cur_buffers == 0;
 397 }
 398
 399 /* must be called with the lock */
 400 static gboolean
 401 do_stop (GstBufferPool * pool)
 402 {
 403   GstBufferPoolPrivate *priv = pool->priv;
 404
 405   if (priv->started) {
 406     GstBufferPoolClass *pclass;
 407
 408     pclass = GST_BUFFER_POOL_GET_CLASS (pool);
 409
 410     GST_LOG_OBJECT (pool, "stopping");
 411     if (G_LIKELY (pclass->stop)) {
 412       if (!pclass->stop (pool))
 413         return FALSE;
 414     }
 415     priv->started = FALSE;
 416   }
 417   return TRUE;
 418 }
 419
 420 /**
 421  * gst_buffer_pool_set_active:
 422  * @pool: a #GstBufferPool
 423  * @active: the new active state
 424  *
 425  * Control the active state of @pool. When the pool is inactive, new calls to
 426  * gst_buffer_pool_acquire_buffer() will return with %GST_FLOW_FLUSHING.
 427  *
 428  * Activating the bufferpool will preallocate all resources in the pool based on
 429  * the configuration of the pool.
 430  *
 431  * Deactivating will free the resources again when there are no outstanding
 432  * buffers. When there are outstanding buffers, they will be freed as soon as
 433  * they are all returned to the pool.
 434  *
 435  * Returns: %FALSE when the pool was not configured or when preallocation of the
 436  * buffers failed.
 437  */
 438 gboolean
 439 gst_buffer_pool_set_active (GstBufferPool * pool, gboolean active)
 440 {
 441   gboolean res = TRUE;
 442   GstBufferPoolPrivate *priv;
 443
 444   g_return_val_if_fail (GST_IS_BUFFER_POOL (pool), FALSE);
 445
 446   GST_LOG_OBJECT (pool, "active %d", active);
 447
 448   priv = pool->priv;
 449
 450   GST_BUFFER_POOL_LOCK (pool);
 451   /* just return if we are already in the right state */
 452   if (priv->active == active)
 453     goto was_ok;
 454
 455   /* we need to be configured */
 456   if (!priv->configured)
 457     goto not_configured;
 458
 459   if (active) {
 460     if (!do_start (pool))
 461       goto start_failed;
 462
 463     /* unset the flushing state now */
 464     gst_poll_read_control (priv->poll);
 465     g_atomic_int_set (&pool->flushing, 0);
 466   } else {
 467     gint outstanding;
 468
 469     /* set to flushing first */
 470     g_atomic_int_set (&pool->flushing, 1);
 471     gst_poll_write_control (priv->poll);
 472
 473     /* when all buffers are in the pool, free them. Else they will be
 474      * freed when they are released */
 475     outstanding = g_atomic_int_get (&priv->outstanding);
 476     GST_LOG_OBJECT (pool, "outstanding buffers %d", outstanding);
 477     if (outstanding == 0) {
 478       if (!do_stop (pool))
 479         goto stop_failed;
 480     }
 481   }
 482   priv->active = active;
 483   GST_BUFFER_POOL_UNLOCK (pool);
 484
 485   return res;
 486
 487 was_ok:
 488   {
 489     GST_DEBUG_OBJECT (pool, "pool was in the right state");
 490     GST_BUFFER_POOL_UNLOCK (pool);
 491     return TRUE;
 492   }
 493 not_configured:
 494   {
 495     GST_ERROR_OBJECT (pool, "pool was not configured");
 496     GST_BUFFER_POOL_UNLOCK (pool);
 497     return FALSE;
 498   }
 499 start_failed:
 500   {
 501     GST_ERROR_OBJECT (pool, "start failed");
 502     GST_BUFFER_POOL_UNLOCK (pool);
 503     return FALSE;
 504   }
 505 stop_failed:
 506   {
 507     GST_WARNING_OBJECT (pool, "stop failed");
 508     GST_BUFFER_POOL_UNLOCK (pool);
 509     return FALSE;
 510   }
 511 }
 512
 513 /**
 514  * gst_buffer_pool_is_active:
 515  * @pool: a #GstBufferPool
 516  *
 517  * Check if @pool is active. A pool can be activated with the
 518  * gst_buffer_pool_set_active() call.
 519  *
 520  * Returns: %TRUE when the pool is active.
 521  */
 522 gboolean
 523 gst_buffer_pool_is_active (GstBufferPool * pool)
 524 {
 525   gboolean res;
 526
 527   GST_BUFFER_POOL_LOCK (pool);
 528   res = pool->priv->active;
 529   GST_BUFFER_POOL_UNLOCK (pool);
 530
 531   return res;
 532 }
 533
 534 static gboolean
 535 default_set_config (GstBufferPool * pool, GstStructure * config)
 536 {
 537   GstBufferPoolPrivate *priv = pool->priv;
 538   GstCaps *caps;
 539   guint size, min_buffers, max_buffers;
 540   GstAllocator *allocator;
 541   GstAllocationParams params;
 542
 543   /* parse the config and keep around */
 544   if (!gst_buffer_pool_config_get_params (config, &caps, &size, &min_buffers,
 545           &max_buffers))
 546     goto wrong_config;
 547
 548   if (!gst_buffer_pool_config_get_allocator (config, &allocator, &params))
 549     goto wrong_config;
 550
 551   GST_DEBUG_OBJECT (pool, "config %" GST_PTR_FORMAT, config);
 552
 553   priv->size = size;
 554   priv->min_buffers = min_buffers;
 555   priv->max_buffers = max_buffers;
 556   priv->cur_buffers = 0;
 557
 558   if (priv->allocator)
 559     gst_object_unref (priv->allocator);
 560   if ((priv->allocator = allocator))
 561     gst_object_ref (allocator);
 562   priv->params = params;
 563
 564   return TRUE;
 565
 566 wrong_config:
 567   {
 568     GST_WARNING_OBJECT (pool, "invalid config %" GST_PTR_FORMAT, config);
 569     return FALSE;
 570   }
 571 }
 572
 573 /**
 574  * gst_buffer_pool_set_config:
 575  * @pool: a #GstBufferPool
 576  * @config: (transfer full): a #GstStructure
 577  *
 578  * Set the configuration of the pool. If the pool is already configured, and
 579  * the configuration haven't change, this function will return %TRUE. If the
 580  * pool is active, this function will try deactivating it. Buffers allocated
 581  * form this pool must be returned or else this function will do nothing and
 582  * return FALSE.
 583  *
 584  * @config is a #GstStructure that contains the configuration parameters for
 585  * the pool. A default and mandatory set of parameters can be configured with
 586  * gst_buffer_pool_config_set_params(), gst_buffer_pool_config_set_allocator()
 587  * and gst_buffer_pool_config_add_option().
 588  *
 589  * If the parameters in @config can not be set exactly, this function returns
 590  * %FALSE and will try to update as much state as possible. The new state can
 591  * then be retrieved and refined with gst_buffer_pool_get_config().
 592  *
 593  * This function takes ownership of @config.
 594  *
 595  * Returns: %TRUE when the configuration could be set.
 596  */
 597 gboolean
 598 gst_buffer_pool_set_config (GstBufferPool * pool, GstStructure * config)
 599 {
 600   gboolean result;
 601   GstBufferPoolClass *pclass;
 602   GstBufferPoolPrivate *priv;
 603
 604   g_return_val_if_fail (GST_IS_BUFFER_POOL (pool), FALSE);
 605   g_return_val_if_fail (config != NULL, FALSE);
 606
 607   priv = pool->priv;
 608
 609   GST_BUFFER_POOL_LOCK (pool);
 610
 611   /* nothing to do if config is unchanged */
 612   if (priv->configured && gst_structure_is_equal (config, priv->config))
 613     goto config_unchanged;
 614
 615   /* can't change the settings when active */
 616   if (priv->active) {
 617     GST_BUFFER_POOL_UNLOCK (pool);
 618     if (!gst_buffer_pool_set_active (pool, FALSE)) {
 619       GST_BUFFER_POOL_LOCK (pool);
 620       goto was_active;
 621     }
 622     GST_BUFFER_POOL_LOCK (pool);
 623
 624     /* not likely but as we released the lock */
 625     if (priv->active)
 626       goto was_active;
 627   }
 628
 629   /* we can't change when outstanding buffers */
 630   if (g_atomic_int_get (&priv->outstanding) != 0)
 631     goto have_outstanding;
 632
 633   pclass = GST_BUFFER_POOL_GET_CLASS (pool);
 634
 635   /* set the new config */
 636   if (G_LIKELY (pclass->set_config))
 637     result = pclass->set_config (pool, config);
 638   else
 639     result = FALSE;
 640
 641   /* save the config regardless of the result so user can read back the
 642    * modified config and evaluate if the changes are acceptable */
 643   if (priv->config)
 644     gst_structure_free (priv->config);
 645   priv->config = config;
 646
 647   if (result) {
 648     /* now we are configured */
 649     priv->configured = TRUE;
 650   }
 651   GST_BUFFER_POOL_UNLOCK (pool);
 652
 653   return result;
 654
 655 config_unchanged:
 656   {
 657     gst_structure_free (config);
 658     GST_BUFFER_POOL_UNLOCK (pool);
 659     return TRUE;
 660   }
 661   /* ERRORS */
 662 was_active:
 663   {
 664     gst_structure_free (config);
 665     GST_WARNING_OBJECT (pool, "can't change config, we are active");
 666     GST_BUFFER_POOL_UNLOCK (pool);
 667     return FALSE;
 668   }
 669 have_outstanding:
 670   {
 671     gst_structure_free (config);
 672     GST_WARNING_OBJECT (pool, "can't change config, have outstanding buffers");
 673     GST_BUFFER_POOL_UNLOCK (pool);
 674     return FALSE;
 675   }
 676 }
 677
 678 /**
 679  * gst_buffer_pool_get_config:
 680  * @pool: a #GstBufferPool
 681  *
 682  * Get a copy of the current configuration of the pool. This configuration
 683  * can either be modified and used for the gst_buffer_pool_set_config() call
 684  * or it must be freed after usage.
 685  *
 686  * Returns: (transfer full): a copy of the current configuration of @pool. use
 687  * gst_structure_free() after usage or gst_buffer_pool_set_config().
 688  */
 689 GstStructure *
 690 gst_buffer_pool_get_config (GstBufferPool * pool)
 691 {
 692   GstStructure *result;
 693
 694   g_return_val_if_fail (GST_IS_BUFFER_POOL (pool), NULL);
 695
 696   GST_BUFFER_POOL_LOCK (pool);
 697   result = gst_structure_copy (pool->priv->config);
 698   GST_BUFFER_POOL_UNLOCK (pool);
 699
 700   return result;
 701 }
 702
 703 static const gchar *empty_option[] = { NULL };
 704
 705 /**
 706  * gst_buffer_pool_get_options:
 707  * @pool: a #GstBufferPool
 708  *
 709  * Get a %NULL terminated array of string with supported bufferpool options for
 710  * @pool. An option would typically be enabled with
 711  * gst_buffer_pool_config_add_option().
 712  *
 713  * Returns: (array zero-terminated=1) (transfer none): a %NULL terminated array
 714  *          of strings.
 715  */
 716 const gchar **
 717 gst_buffer_pool_get_options (GstBufferPool * pool)
 718 {
 719   GstBufferPoolClass *pclass;
 720   const gchar **result;
 721
 722   g_return_val_if_fail (GST_IS_BUFFER_POOL (pool), NULL);
 723
 724   pclass = GST_BUFFER_POOL_GET_CLASS (pool);
 725
 726   if (G_LIKELY (pclass->get_options)) {
 727     if ((result = pclass->get_options (pool)) == NULL)
 728       goto invalid_result;
 729   } else
 730     result = empty_option;
 731
 732   return result;
 733
 734   /* ERROR */
 735 invalid_result:
 736   {
 737     g_warning ("bufferpool subclass returned NULL options");
 738     return empty_option;
 739   }
 740 }
 741
 742 /**
 743  * gst_buffer_pool_has_option:
 744  * @pool: a #GstBufferPool
 745  * @option: an option
 746  *
 747  * Check if the bufferpool supports @option.
 748  *
 749  * Returns: a %NULL terminated array of strings.
 750  */
 751 gboolean
 752 gst_buffer_pool_has_option (GstBufferPool * pool, const gchar * option)
 753 {
 754   guint i;
 755   const gchar **options;
 756
 757   g_return_val_if_fail (GST_IS_BUFFER_POOL (pool), FALSE);
 758   g_return_val_if_fail (option != NULL, FALSE);
 759
 760   options = gst_buffer_pool_get_options (pool);
 761
 762   for (i = 0; options[i]; i++) {
 763     if (g_str_equal (options[i], option))
 764       return TRUE;
 765   }
 766   return FALSE;
 767 }
 768
 769 /**
 770  * gst_buffer_pool_config_set_params:
 771  * @config: a #GstBufferPool configuration
 772  * @caps: caps for the buffers
 773  * @size: the size of each buffer, not including prefix and padding
 774  * @min_buffers: the minimum amount of buffers to allocate.
 775  * @max_buffers: the maximum amount of buffers to allocate or 0 for unlimited.
 776  *
 777  * Configure @config with the given parameters.
 778  */
 779 void
 780 gst_buffer_pool_config_set_params (GstStructure * config, GstCaps * caps,
 781     guint size, guint min_buffers, guint max_buffers)
 782 {
 783   g_return_if_fail (config != NULL);
 784   g_return_if_fail (max_buffers == 0 || min_buffers <= max_buffers);
 785   g_return_if_fail (caps == NULL || gst_caps_is_fixed (caps));
 786
 787   gst_structure_id_set (config,
 788       GST_QUARK (CAPS), GST_TYPE_CAPS, caps,
 789       GST_QUARK (SIZE), G_TYPE_UINT, size,
 790       GST_QUARK (MIN_BUFFERS), G_TYPE_UINT, min_buffers,
 791       GST_QUARK (MAX_BUFFERS), G_TYPE_UINT, max_buffers, NULL);
 792 }
 793
 794 /**
 795  * gst_buffer_pool_config_set_allocator:
 796  * @config: a #GstBufferPool configuration
 797  * @allocator: a #GstAllocator
 798  * @params: #GstAllocationParams
 799  *
 800  * Set the @allocator and @params on @config.
 801  *
 802  * One of @allocator and @params can be %NULL, but not both. When @allocator
 803  * is %NULL, the default allocator of the pool will use the values in @param
 804  * to perform its allocation. When @param is %NULL, the pool will use the
 805  * provided @allocator with its default #GstAllocationParams.
 806  *
 807  * A call to gst_buffer_pool_set_config() can update the allocator and params
 808  * with the values that it is able to do. Some pools are, for example, not able
 809  * to operate with different allocators or cannot allocate with the values
 810  * specified in @params. Use gst_buffer_pool_get_config() to get the currently
 811  * used values.
 812  */
 813 void
 814 gst_buffer_pool_config_set_allocator (GstStructure * config,
 815     GstAllocator * allocator, const GstAllocationParams * params)
 816 {
 817   g_return_if_fail (config != NULL);
 818   g_return_if_fail (allocator != NULL || params != NULL);
 819
 820   gst_structure_id_set (config,
 821       GST_QUARK (ALLOCATOR), GST_TYPE_ALLOCATOR, allocator,
 822       GST_QUARK (PARAMS), GST_TYPE_ALLOCATION_PARAMS, params, NULL);
 823 }
 824
 825 /**
 826  * gst_buffer_pool_config_add_option:
 827  * @config: a #GstBufferPool configuration
 828  * @option: an option to add
 829  *
 830  * Enabled the option in @config. This will instruct the @bufferpool to enable
 831  * the specified option on the buffers that it allocates.
 832  *
 833  * The supported options by @pool can be retrieved with gst_buffer_pool_get_options().
 834  */
 835 void
 836 gst_buffer_pool_config_add_option (GstStructure * config, const gchar * option)
 837 {
 838   const GValue *value;
 839   GValue option_value = { 0, };
 840   guint i, len;
 841
 842   g_return_if_fail (config != NULL);
 843
 844   value = gst_structure_id_get_value (config, GST_QUARK (OPTIONS));
 845   if (value) {
 846     len = gst_value_array_get_size (value);
 847     for (i = 0; i < len; ++i) {
 848       const GValue *nth_val = gst_value_array_get_value (value, i);
 849
 850       if (g_str_equal (option, g_value_get_string (nth_val)))
 851         return;
 852     }
 853   } else {
 854     GValue new_array_val = { 0, };
 855
 856     g_value_init (&new_array_val, GST_TYPE_ARRAY);
 857     gst_structure_id_take_value (config, GST_QUARK (OPTIONS), &new_array_val);
 858     value = gst_structure_id_get_value (config, GST_QUARK (OPTIONS));
 859   }
 860   g_value_init (&option_value, G_TYPE_STRING);
 861   g_value_set_string (&option_value, option);
 862   gst_value_array_append_and_take_value ((GValue *) value, &option_value);
 863 }
 864
 865 /**
 866  * gst_buffer_pool_config_n_options:
 867  * @config: a #GstBufferPool configuration
 868  *
 869  * Retrieve the number of values currently stored in the options array of the
 870  * @config structure.
 871  *
 872  * Returns: the options array size as a #guint.
 873  */
 874 guint
 875 gst_buffer_pool_config_n_options (GstStructure * config)
 876 {
 877   const GValue *value;
 878   guint size = 0;
 879
 880   g_return_val_if_fail (config != NULL, 0);
 881
 882   value = gst_structure_id_get_value (config, GST_QUARK (OPTIONS));
 883   if (value) {
 884     size = gst_value_array_get_size (value);
 885   }
 886   return size;
 887 }
 888
 889 /**
 890  * gst_buffer_pool_config_get_option:
 891  * @config: a #GstBufferPool configuration
 892  * @index: position in the option array to read
 893  *
 894  * Parse an available @config and get the option at @index of the options API
 895  * array.
 896  *
 897  * Returns: a #gchar of the option at @index.
 898  */
 899 const gchar *
 900 gst_buffer_pool_config_get_option (GstStructure * config, guint index)
 901 {
 902   const GValue *value;
 903   const gchar *ret = NULL;
 904
 905   g_return_val_if_fail (config != NULL, 0);
 906
 907   value = gst_structure_id_get_value (config, GST_QUARK (OPTIONS));
 908   if (value) {
 909     const GValue *option_value;
 910
 911     option_value = gst_value_array_get_value (value, index);
 912     if (option_value)
 913       ret = g_value_get_string (option_value);
 914   }
 915   return ret;
 916 }
 917
 918 /**
 919  * gst_buffer_pool_config_has_option:
 920  * @config: a #GstBufferPool configuration
 921  * @option: an option
 922  *
 923  * Check if @config contains @option.
 924  *
 925  * Returns: %TRUE if the options array contains @option.
 926  */
 927 gboolean
 928 gst_buffer_pool_config_has_option (GstStructure * config, const gchar * option)
 929 {
 930   const GValue *value;
 931   guint i, len;
 932
 933   g_return_val_if_fail (config != NULL, 0);
 934
 935   value = gst_structure_id_get_value (config, GST_QUARK (OPTIONS));
 936   if (value) {
 937     len = gst_value_array_get_size (value);
 938     for (i = 0; i < len; ++i) {
 939       const GValue *nth_val = gst_value_array_get_value (value, i);
 940
 941       if (g_str_equal (option, g_value_get_string (nth_val)))
 942         return TRUE;
 943     }
 944   }
 945   return FALSE;
 946 }
 947
 948 /**
 949  * gst_buffer_pool_config_get_params:
 950  * @config: (transfer none): a #GstBufferPool configuration
 951  * @caps: (out) (transfer none) (allow-none): the caps of buffers
 952  * @size: (out) (allow-none): the size of each buffer, not including prefix and padding
 953  * @min_buffers: (out) (allow-none): the minimum amount of buffers to allocate.
 954  * @max_buffers: (out) (allow-none): the maximum amount of buffers to allocate or 0 for unlimited.
 955  *
 956  * Get the configuration values from @config.
 957  *
 958  * Returns: %TRUE if all parameters could be fetched.
 959  */
 960 gboolean
 961 gst_buffer_pool_config_get_params (GstStructure * config, GstCaps ** caps,
 962     guint * size, guint * min_buffers, guint * max_buffers)
 963 {
 964   g_return_val_if_fail (config != NULL, FALSE);
 965
 966   if (caps) {
 967     *caps = g_value_get_boxed (gst_structure_id_get_value (config,
 968             GST_QUARK (CAPS)));
 969   }
 970   return gst_structure_id_get (config,
 971       GST_QUARK (SIZE), G_TYPE_UINT, size,
 972       GST_QUARK (MIN_BUFFERS), G_TYPE_UINT, min_buffers,
 973       GST_QUARK (MAX_BUFFERS), G_TYPE_UINT, max_buffers, NULL);
 974 }
 975
 976 /**
 977  * gst_buffer_pool_config_get_allocator:
 978  * @config: (transfer none): a #GstBufferPool configuration
 979  * @allocator: (transfer none): a #GstAllocator
 980  * @params: #GstAllocationParams
 981  *
 982  * Get the @allocator and @params from @config.
 983  *
 984  * Returns: %TRUE, if the values are set.
 985  */
 986 gboolean
 987 gst_buffer_pool_config_get_allocator (GstStructure * config,
 988     GstAllocator ** allocator, GstAllocationParams * params)
 989 {
 990   g_return_val_if_fail (config != NULL, FALSE);
 991
 992   if (allocator)
 993     *allocator = g_value_get_object (gst_structure_id_get_value (config,
 994             GST_QUARK (ALLOCATOR)));
 995   if (params) {
 996     GstAllocationParams *p;
 997
 998     p = g_value_get_boxed (gst_structure_id_get_value (config,
 999             GST_QUARK (PARAMS)));
1000     if (p) {
1001       *params = *p;
1002     } else {
1003       gst_allocation_params_init (params);
1004     }
1005   }
1006   return TRUE;
1007 }
1008
1009 /**
1010  * gst_buffer_pool_config_validate_params:
1011  * @config: (transfer none): a #GstBufferPool configuration
1012  * @caps: (transfer none): the excepted caps of buffers
1013  * @size: the expected size of each buffer, not including prefix and padding
1014  * @min_buffers: the expected minimum amount of buffers to allocate.
1015  * @max_buffers: the expect maximum amount of buffers to allocate or 0 for unlimited.
1016  *
1017  * Validate that changes made to @config are still valid in the context of the
1018  * expected parameters. This function is a helper that can be used to validate
1019  * changes made by a pool to a config when gst_buffer_pool_set_config()
1020  * returns %FALSE. This expects that @caps and @size haven't changed, and that
1021  * @min_buffers aren't lower then what we initially expected. This does not check
1022  * if options or allocator parameters.
1023  *
1024  * Since: 1.4
1025  *
1026  * Returns: %TRUE, if the parameters are valid in this context.
1027  */
1028 gboolean
1029 gst_buffer_pool_config_validate_params (GstStructure * config, GstCaps * caps,
1030     guint size, guint min_buffers, G_GNUC_UNUSED guint max_buffers)
1031 {
1032   GstCaps *newcaps;
1033   guint newsize, newmin;
1034   gboolean ret = FALSE;
1035
1036   g_return_val_if_fail (config != NULL, FALSE);
1037
1038   gst_buffer_pool_config_get_params (config, &newcaps, &newsize, &newmin, NULL);
1039
1040   if (gst_caps_is_equal (caps, newcaps) && (size == newsize)
1041       && (newmin >= min_buffers))
1042     ret = TRUE;
1043
1044   return ret;
1045 }
1046
1047 static GstFlowReturn
1048 default_acquire_buffer (GstBufferPool * pool, GstBuffer ** buffer,
1049     GstBufferPoolAcquireParams * params)
1050 {
1051   GstFlowReturn result;
1052   GstBufferPoolPrivate *priv = pool->priv;
1053
1054   while (TRUE) {
1055     if (G_UNLIKELY (GST_BUFFER_POOL_IS_FLUSHING (pool)))
1056       goto flushing;
1057
1058     /* try to get a buffer from the queue */
1059     *buffer = gst_atomic_queue_pop (priv->queue);
1060     if (G_LIKELY (*buffer)) {
1061       gst_poll_read_control (priv->poll);
1062       result = GST_FLOW_OK;
1063       GST_LOG_OBJECT (pool, "acquired buffer %p", *buffer);
1064       break;
1065     }
1066
1067     /* no buffer, try to allocate some more */
1068     GST_LOG_OBJECT (pool, "no buffer, trying to allocate");
1069     result = do_alloc_buffer (pool, buffer, NULL);
1070     if (G_LIKELY (result == GST_FLOW_OK))
1071       /* we have a buffer, return it */
1072       break;
1073
1074     if (G_UNLIKELY (result != GST_FLOW_EOS))
1075       /* something went wrong, return error */
1076       break;
1077
1078     /* check if we need to wait */
1079     if (params && (params->flags & GST_BUFFER_POOL_ACQUIRE_FLAG_DONTWAIT)) {
1080       GST_LOG_OBJECT (pool, "no more buffers");
1081       break;
1082     }
1083
1084     /* now we release the control socket, we wait for a buffer release or
1085      * flushing */
1086     gst_poll_read_control (pool->priv->poll);
1087     GST_LOG_OBJECT (pool, "waiting for free buffers or flushing");
1088     gst_poll_wait (priv->poll, GST_CLOCK_TIME_NONE);
1089     gst_poll_write_control (pool->priv->poll);
1090   }
1091
1092   return result;
1093
1094   /* ERRORS */
1095 flushing:
1096   {
1097     GST_DEBUG_OBJECT (pool, "we are flushing");
1098     return GST_FLOW_FLUSHING;
1099   }
1100 }
1101
1102 static inline void
1103 dec_outstanding (GstBufferPool * pool)
1104 {
1105   if (g_atomic_int_dec_and_test (&pool->priv->outstanding)) {
1106     /* all buffers are returned to the pool, see if we need to free them */
1107     if (GST_BUFFER_POOL_IS_FLUSHING (pool)) {
1108       /* take the lock so that set_active is not run concurrently */
1109       GST_BUFFER_POOL_LOCK (pool);
1110       /* recheck the flushing state in the lock, the pool could have been
1111        * set to active again */
1112       if (GST_BUFFER_POOL_IS_FLUSHING (pool))
1113         do_stop (pool);
1114
1115       GST_BUFFER_POOL_UNLOCK (pool);
1116     }
1117   }
1118 }
1119
1120 static gboolean
1121 remove_meta_unpooled (GstBuffer * buffer, GstMeta ** meta, gpointer user_data)
1122 {
1123   if (!GST_META_FLAG_IS_SET (*meta, GST_META_FLAG_POOLED)) {
1124     GST_META_FLAG_UNSET (*meta, GST_META_FLAG_LOCKED);
1125     *meta = NULL;
1126   }
1127   return TRUE;
1128 }
1129
1130 static void
1131 default_reset_buffer (GstBufferPool * pool, GstBuffer * buffer)
1132 {
1133   GST_BUFFER_FLAGS (buffer) &= GST_BUFFER_FLAG_TAG_MEMORY;
1134
1135   GST_BUFFER_PTS (buffer) = GST_CLOCK_TIME_NONE;
1136   GST_BUFFER_DTS (buffer) = GST_CLOCK_TIME_NONE;
1137   GST_BUFFER_DURATION (buffer) = GST_CLOCK_TIME_NONE;
1138   GST_BUFFER_OFFSET (buffer) = GST_BUFFER_OFFSET_NONE;
1139   GST_BUFFER_OFFSET_END (buffer) = GST_BUFFER_OFFSET_NONE;
1140
1141   /* remove all metadata without the POOLED flag */
1142   gst_buffer_foreach_meta (buffer, remove_meta_unpooled, pool);
1143 }
1144
1145 /**
1146  * gst_buffer_pool_acquire_buffer:
1147  * @pool: a #GstBufferPool
1148  * @buffer: (out): a location for a #GstBuffer
1149  * @params: (transfer none) (allow-none): parameters.
1150  *
1151  * Acquire a buffer from @pool. @buffer should point to a memory location that
1152  * can hold a pointer to the new buffer.
1153  *
1154  * @params can be %NULL or contain optional parameters to influence the
1155  * allocation.
1156  *
1157  * Returns: a #GstFlowReturn such as %GST_FLOW_FLUSHING when the pool is
1158  * inactive.
1159  */
1160 GstFlowReturn
1161 gst_buffer_pool_acquire_buffer (GstBufferPool * pool, GstBuffer ** buffer,
1162     GstBufferPoolAcquireParams * params)
1163 {
1164   GstBufferPoolClass *pclass;
1165   GstFlowReturn result;
1166
1167   g_return_val_if_fail (GST_IS_BUFFER_POOL (pool), GST_FLOW_ERROR);
1168   g_return_val_if_fail (buffer != NULL, GST_FLOW_ERROR);
1169
1170   pclass = GST_BUFFER_POOL_GET_CLASS (pool);
1171
1172   /* assume we'll have one more outstanding buffer we need to do that so
1173    * that concurrent set_active doesn't clear the buffers */
1174   g_atomic_int_inc (&pool->priv->outstanding);
1175
1176   if (G_LIKELY (pclass->acquire_buffer))
1177     result = pclass->acquire_buffer (pool, buffer, params);
1178   else
1179     result = GST_FLOW_NOT_SUPPORTED;
1180
1181   if (G_LIKELY (result == GST_FLOW_OK)) {
1182     /* all buffers from the pool point to the pool and have the refcount of the
1183      * pool incremented */
1184     (*buffer)->pool = gst_object_ref (pool);
1185   } else {
1186     dec_outstanding (pool);
1187   }
1188
1189   return result;
1190 }
1191
1192 static void
1193 default_release_buffer (GstBufferPool * pool, GstBuffer * buffer)
1194 {
1195   GST_LOG_OBJECT (pool, "released buffer %p %d", buffer,
1196       GST_MINI_OBJECT_FLAGS (buffer));
1197
1198   /* memory should be untouched */
1199   if (GST_BUFFER_FLAG_IS_SET (buffer, GST_BUFFER_FLAG_TAG_MEMORY))
1200     goto discard;
1201
1202   /* size should have been reset. This is not a catch all, pool with
1203    * size requirement per memory should do their own check. */
1204   if (gst_buffer_get_size (buffer) != pool->priv->size)
1205     goto discard;
1206
1207   /* all memory should be exclusive to this buffer (and thus be writable) */
1208   if (!gst_buffer_is_all_memory_writable (buffer))
1209     goto discard;
1210
1211   /* keep it around in our queue */
1212   gst_atomic_queue_push (pool->priv->queue, buffer);
1213   gst_poll_write_control (pool->priv->poll);
1214
1215   return;
1216
1217 discard:
1218   {
1219     do_free_buffer (pool, buffer);
1220     return;
1221   }
1222 }
1223
1224 /**
1225  * gst_buffer_pool_release_buffer:
1226  * @pool: a #GstBufferPool
1227  * @buffer: (transfer full): a #GstBuffer
1228  *
1229  * Release @buffer to @pool. @buffer should have previously been allocated from
1230  * @pool with gst_buffer_pool_acquire_buffer().
1231  *
1232  * This function is usually called automatically when the last ref on @buffer
1233  * disappears.
1234  */
1235 void
1236 gst_buffer_pool_release_buffer (GstBufferPool * pool, GstBuffer * buffer)
1237 {
1238   GstBufferPoolClass *pclass;
1239
1240   g_return_if_fail (GST_IS_BUFFER_POOL (pool));
1241   g_return_if_fail (buffer != NULL);
1242
1243   /* check that the buffer is ours, all buffers returned to the pool have the
1244    * pool member set to NULL and the pool refcount decreased */
1245   if (!g_atomic_pointer_compare_and_exchange (&buffer->pool, pool, NULL))
1246     return;
1247
1248   pclass = GST_BUFFER_POOL_GET_CLASS (pool);
1249
1250   /* reset the buffer when needed */
1251   if (G_LIKELY (pclass->reset_buffer))
1252     pclass->reset_buffer (pool, buffer);
1253
1254   if (G_LIKELY (pclass->release_buffer))
1255     pclass->release_buffer (pool, buffer);
1256
1257   dec_outstanding (pool);
1258
1259   /* decrease the refcount that the buffer had to us */
1260   gst_object_unref (pool);
1261 }