2 * Copyright (C) 1999,2000 Erik Walthinsen <omega@cse.ogi.edu>
3 * 2005 Wim Taymans <wim@fluendo.com>
5 * gsttask.c: Streaming tasks
7 * This library is free software; you can redistribute it and/or
8 * modify it under the terms of the GNU Library General Public
9 * License as published by the Free Software Foundation; either
10 * version 2 of the License, or (at your option) any later version.
12 * This library is distributed in the hope that it will be useful,
13 * but WITHOUT ANY WARRANTY; without even the implied warranty of
14 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
15 * Library General Public License for more details.
17 * You should have received a copy of the GNU Library General Public
18 * License along with this library; if not, write to the
19 * Free Software Foundation, Inc., 59 Temple Place - Suite 330,
20 * Boston, MA 02111-1307, USA.
25 * @short_description: Abstraction of GStreamer streaming threads.
26 * @see_also: #GstElement, #GstPad
28 * #GstTask is used by #GstElement and #GstPad to provide the data passing
29 * threads in a #GstPipeline.
31 * A #GstPad will typically start a #GstTask to push or pull data to/from the
32 * peer pads. Most source elements start a #GstTask to push data. In some cases
33 * a demuxer element can start a #GstTask to pull data from a peer element. This
34 * is typically done when the demuxer can perform random access on the upstream
35 * peer element for improved performance.
37 * Although convenience functions exist on #GstPad to start/pause/stop tasks, it
38 * might sometimes be needed to create a #GstTask manually if it is not related to
41 * Before the #GstTask can be run, it needs a #GStaticRecMutex that can be set with
42 * gst_task_set_lock().
44 * The task can be started, paused and stopped with gst_task_start(), gst_task_pause()
45 * and gst_task_stop() respectively or with the gst_task_set_state() function.
47 * A #GstTask will repeatedly call the #GstTaskFunction with the user data
48 * that was provided when creating the task with gst_task_create(). While calling
49 * the function it will acquire the provided lock. The provided lock is released
50 * when the task pauses or stops.
52 * Stopping a task with gst_task_stop() will not immediately make sure the task is
53 * not running anymore. Use gst_task_join() to make sure the task is completely
54 * stopped and the thread is stopped.
56 * After creating a #GstTask, use gst_object_unref() to free its resources. This can
57 * only be done it the task is not running anymore.
59 * Last reviewed on 2006-02-13 (0.10.4)
62 #include "gst_private.h"
67 GST_DEBUG_CATEGORY_STATIC (task_debug);
68 #define GST_CAT_DEFAULT (task_debug)
70 #define GST_TASK_GET_PRIVATE(obj) \
71 (G_TYPE_INSTANCE_GET_PRIVATE ((obj), GST_TYPE_TASK, GstTaskPrivate))
73 struct _GstTaskPrivate
75 /* callbacks for managing the thread of this task */
76 GstTaskThreadCallbacks thr_callbacks;
77 gpointer thr_user_data;
78 GDestroyNotify thr_notify;
81 GThreadPriority priority;
86 /* remember the pool and id that is currently running. */
91 static void gst_task_class_init (GstTaskClass * klass);
92 static void gst_task_init (GstTask * task);
93 static void gst_task_finalize (GObject * object);
95 static void gst_task_func (GstTask * task);
97 static GStaticMutex pool_lock = G_STATIC_MUTEX_INIT;
101 GST_DEBUG_CATEGORY_INIT (task_debug, "task", 0, "Processing tasks"); \
104 G_DEFINE_TYPE_WITH_CODE (GstTask, gst_task, GST_TYPE_OBJECT, _do_init);
107 init_klass_pool (GstTaskClass * klass)
109 g_static_mutex_lock (&pool_lock);
111 gst_task_pool_cleanup (klass->pool);
112 gst_object_unref (klass->pool);
114 klass->pool = gst_task_pool_new ();
115 gst_task_pool_prepare (klass->pool, NULL);
116 g_static_mutex_unlock (&pool_lock);
120 gst_task_class_init (GstTaskClass * klass)
122 GObjectClass *gobject_class;
124 gobject_class = (GObjectClass *) klass;
126 g_type_class_add_private (klass, sizeof (GstTaskPrivate));
128 gobject_class->finalize = gst_task_finalize;
130 init_klass_pool (klass);
134 gst_task_init (GstTask * task)
138 klass = GST_TASK_GET_CLASS (task);
140 task->priv = GST_TASK_GET_PRIVATE (task);
141 task->running = FALSE;
142 task->abidata.ABI.thread = NULL;
144 task->cond = g_cond_new ();
145 task->state = GST_TASK_STOPPED;
146 task->priv->prio_set = FALSE;
148 /* use the default klass pool for this task, users can
149 * override this later */
150 g_static_mutex_lock (&pool_lock);
151 task->priv->pool = gst_object_ref (klass->pool);
152 g_static_mutex_unlock (&pool_lock);
156 gst_task_finalize (GObject * object)
158 GstTask *task = GST_TASK (object);
159 GstTaskPrivate *priv = task->priv;
161 GST_DEBUG ("task %p finalize", task);
163 if (priv->thr_notify)
164 priv->thr_notify (priv->thr_user_data);
165 priv->thr_notify = NULL;
166 priv->thr_user_data = NULL;
168 gst_object_unref (priv->pool);
170 /* task thread cannot be running here since it holds a ref
171 * to the task so that the finalize could not have happened */
172 g_cond_free (task->cond);
175 G_OBJECT_CLASS (gst_task_parent_class)->finalize (object);
179 gst_task_func (GstTask * task)
181 GStaticRecMutex *lock;
183 GstTaskPrivate *priv;
187 tself = g_thread_self ();
189 GST_DEBUG ("Entering task %p, thread %p", task, tself);
191 /* we have to grab the lock to get the mutex. We also
192 * mark our state running so that nobody can mess with
194 GST_OBJECT_LOCK (task);
195 if (task->state == GST_TASK_STOPPED)
197 lock = GST_TASK_GET_LOCK (task);
198 if (G_UNLIKELY (lock == NULL))
200 task->abidata.ABI.thread = tself;
201 /* only update the priority when it was changed */
203 g_thread_set_priority (tself, priv->priority);
204 GST_OBJECT_UNLOCK (task);
206 /* fire the enter_thread callback when we need to */
207 if (priv->thr_callbacks.enter_thread)
208 priv->thr_callbacks.enter_thread (task, tself, priv->thr_user_data);
210 /* locking order is TASK_LOCK, LOCK */
211 g_static_rec_mutex_lock (lock);
212 GST_OBJECT_LOCK (task);
213 while (G_LIKELY (task->state != GST_TASK_STOPPED)) {
214 while (G_UNLIKELY (task->state == GST_TASK_PAUSED)) {
217 t = g_static_rec_mutex_unlock_full (lock);
219 g_warning ("wrong STREAM_LOCK count %d", t);
221 GST_TASK_SIGNAL (task);
222 GST_TASK_WAIT (task);
223 GST_OBJECT_UNLOCK (task);
224 /* locking order.. */
226 g_static_rec_mutex_lock_full (lock, t);
228 GST_OBJECT_LOCK (task);
229 if (G_UNLIKELY (task->state == GST_TASK_STOPPED))
232 GST_OBJECT_UNLOCK (task);
234 task->func (task->data);
236 GST_OBJECT_LOCK (task);
239 GST_OBJECT_UNLOCK (task);
240 g_static_rec_mutex_unlock (lock);
242 GST_OBJECT_LOCK (task);
243 task->abidata.ABI.thread = NULL;
246 if (priv->thr_callbacks.leave_thread) {
247 /* fire the leave_thread callback when we need to. We need to do this before
248 * we signal the task and with the task lock released. */
249 GST_OBJECT_UNLOCK (task);
250 priv->thr_callbacks.leave_thread (task, tself, priv->thr_user_data);
251 GST_OBJECT_LOCK (task);
253 /* restore normal priority when releasing back into the pool, we will not
254 * touch the priority when a custom callback has been installed. */
255 g_thread_set_priority (tself, G_THREAD_PRIORITY_NORMAL);
257 /* now we allow messing with the lock again by setting the running flag to
258 * FALSE. Together with the SIGNAL this is the sign for the _join() to
260 * Note that we still have not dropped the final ref on the task. We could
261 * check here if there is a pending join() going on and drop the last ref
262 * before releasing the lock as we can be sure that a ref is held by the
263 * caller of the join(). */
264 task->running = FALSE;
265 GST_TASK_SIGNAL (task);
266 GST_OBJECT_UNLOCK (task);
268 GST_DEBUG ("Exit task %p, thread %p", task, g_thread_self ());
270 gst_object_unref (task);
275 g_warning ("starting task without a lock");
281 * gst_task_cleanup_all:
283 * Wait for all tasks to be stopped. This is mainly used internally
284 * to ensure proper cleanup of internal data structures in test suites.
289 gst_task_cleanup_all (void)
293 if ((klass = g_type_class_peek (GST_TYPE_TASK))) {
294 init_klass_pool (klass);
300 * @func: The #GstTaskFunction to use
301 * @data: User data to pass to @func
303 * Create a new Task that will repeatedly call the provided @func
304 * with @data as a parameter. Typically the task will run in
307 * The function cannot be changed after the task has been created. You
308 * must create a new #GstTask to change the function.
310 * This function will not yet create and start a thread. Use gst_task_start() or
311 * gst_task_pause() to create and start the GThread.
313 * Before the task can be used, a #GStaticRecMutex must be configured using the
314 * gst_task_set_lock() function. This lock will always be acquired while
317 * Returns: A new #GstTask.
322 gst_task_create (GstTaskFunction func, gpointer data)
326 task = g_object_new (GST_TYPE_TASK, NULL);
330 GST_DEBUG ("Created task %p", task);
337 * @task: The #GstTask to use
338 * @mutex: The #GMutex to use
340 * Set the mutex used by the task. The mutex will be acquired before
341 * calling the #GstTaskFunction.
343 * This function has to be called before calling gst_task_pause() or
349 gst_task_set_lock (GstTask * task, GStaticRecMutex * mutex)
351 GST_OBJECT_LOCK (task);
352 if (G_UNLIKELY (task->running))
354 GST_TASK_GET_LOCK (task) = mutex;
355 GST_OBJECT_UNLOCK (task);
362 GST_OBJECT_UNLOCK (task);
363 g_warning ("cannot call set_lock on a running task");
368 * gst_task_set_priority:
370 * @priority: a new priority for @task
372 * Changes the priority of @task to @priority.
374 * Note: try not to depend on task priorities.
381 gst_task_set_priority (GstTask * task, GThreadPriority priority)
383 GstTaskPrivate *priv;
386 g_return_if_fail (GST_IS_TASK (task));
390 GST_OBJECT_LOCK (task);
391 priv->prio_set = TRUE;
392 priv->priority = priority;
393 thread = task->abidata.ABI.thread;
394 if (thread != NULL) {
395 /* if this task already has a thread, we can configure the priority right
396 * away, else we do that when we assign a thread to the task. */
397 g_thread_set_priority (thread, priority);
399 GST_OBJECT_UNLOCK (task);
406 * Get the #GstTaskPool that this task will use for its streaming
411 * Returns: the #GstTaskPool used by @task. gst_object_unref()
417 gst_task_get_pool (GstTask * task)
420 GstTaskPrivate *priv;
422 g_return_val_if_fail (GST_IS_TASK (task), NULL);
426 GST_OBJECT_LOCK (task);
427 result = gst_object_ref (priv->pool);
428 GST_OBJECT_UNLOCK (task);
436 * @pool: a #GstTaskPool
438 * Set @pool as the new GstTaskPool for @task. Any new streaming threads that
439 * will be created by @task will now use @pool.
446 gst_task_set_pool (GstTask * task, GstTaskPool * pool)
449 GstTaskPrivate *priv;
451 g_return_if_fail (GST_IS_TASK (task));
452 g_return_if_fail (GST_IS_TASK_POOL (pool));
456 GST_OBJECT_LOCK (task);
457 if (priv->pool != pool) {
459 priv->pool = gst_object_ref (pool);
462 GST_OBJECT_UNLOCK (task);
465 gst_object_unref (old);
470 * gst_task_set_thread_callbacks:
471 * @task: The #GstTask to use
472 * @callbacks: a #GstTaskThreadCallbacks pointer
473 * @user_data: user data passed to the callbacks
474 * @notify: called when @user_data is no longer referenced
476 * Set callbacks which will be executed when a new thread is needed, the thread
477 * function is entered and left and when the thread is joined.
479 * By default a thread for @task will be created from a default thread pool.
481 * Objects can use custom GThreads or can perform additional configuration of
482 * the threads (such as changing the thread priority) by installing callbacks.
489 gst_task_set_thread_callbacks (GstTask * task,
490 GstTaskThreadCallbacks * callbacks, gpointer user_data,
491 GDestroyNotify notify)
493 GDestroyNotify old_notify;
495 g_return_if_fail (task != NULL);
496 g_return_if_fail (GST_IS_TASK (task));
497 g_return_if_fail (callbacks != NULL);
499 GST_OBJECT_LOCK (task);
500 old_notify = task->priv->thr_notify;
505 old_data = task->priv->thr_user_data;
507 task->priv->thr_user_data = NULL;
508 task->priv->thr_notify = NULL;
509 GST_OBJECT_UNLOCK (task);
511 old_notify (old_data);
513 GST_OBJECT_LOCK (task);
515 task->priv->thr_callbacks = *callbacks;
516 task->priv->thr_user_data = user_data;
517 task->priv->thr_notify = notify;
518 GST_OBJECT_UNLOCK (task);
522 * gst_task_get_state:
523 * @task: The #GstTask to query
525 * Get the current state of the task.
527 * Returns: The #GstTaskState of the task
532 gst_task_get_state (GstTask * task)
536 g_return_val_if_fail (GST_IS_TASK (task), GST_TASK_STOPPED);
538 GST_OBJECT_LOCK (task);
539 result = task->state;
540 GST_OBJECT_UNLOCK (task);
545 /* make sure the task is running and start a thread if it's not.
546 * This function must be called with the task LOCK. */
548 start_task (GstTask * task)
551 GError *error = NULL;
552 GstTaskPrivate *priv;
556 /* new task, We ref before so that it remains alive while
557 * the thread is running. */
558 gst_object_ref (task);
559 /* mark task as running so that a join will wait until we schedule
560 * and exit the task function. */
561 task->running = TRUE;
563 /* push on the thread pool, we remember the original pool because the user
564 * could change it later on and then we join to the wrong pool. */
565 priv->pool_id = gst_object_ref (priv->pool);
567 gst_task_pool_push (priv->pool_id, (GstTaskPoolFunction) gst_task_func,
571 g_warning ("failed to create thread: %s", error->message);
572 g_error_free (error);
580 * gst_task_set_state:
582 * @state: the new task state
584 * Sets the state of @task to @state.
586 * The @task must have a lock associated with it using
587 * gst_task_set_lock() when going to GST_TASK_STARTED or GST_TASK_PAUSED or
588 * this function will return %FALSE.
592 * Returns: %TRUE if the state could be changed.
597 gst_task_set_state (GstTask * task, GstTaskState state)
602 g_return_val_if_fail (GST_IS_TASK (task), FALSE);
604 GST_DEBUG_OBJECT (task, "Changing task %p to state %d", task, state);
606 GST_OBJECT_LOCK (task);
607 if (state != GST_TASK_STOPPED)
608 if (G_UNLIKELY (GST_TASK_GET_LOCK (task) == NULL))
611 /* if the state changed, do our thing */
616 case GST_TASK_STOPPED:
617 /* If the task already has a thread scheduled we don't have to do
619 if (G_UNLIKELY (!task->running))
620 res = start_task (task);
622 case GST_TASK_PAUSED:
623 /* when we are paused, signal to go to the new state */
624 GST_TASK_SIGNAL (task);
626 case GST_TASK_STARTED:
627 /* if we were started, we'll go to the new state after the next
632 GST_OBJECT_UNLOCK (task);
639 GST_WARNING_OBJECT (task, "state %d set on task without a lock", state);
640 GST_OBJECT_UNLOCK (task);
641 g_warning ("task without a lock can't be set to state %d", state);
648 * @task: The #GstTask to start
650 * Starts @task. The @task must have a lock associated with it using
651 * gst_task_set_lock() or this function will return %FALSE.
653 * Returns: %TRUE if the task could be started.
658 gst_task_start (GstTask * task)
660 return gst_task_set_state (task, GST_TASK_STARTED);
665 * @task: The #GstTask to stop
667 * Stops @task. This method merely schedules the task to stop and
668 * will not wait for the task to have completely stopped. Use
669 * gst_task_join() to stop and wait for completion.
671 * Returns: %TRUE if the task could be stopped.
676 gst_task_stop (GstTask * task)
678 return gst_task_set_state (task, GST_TASK_STOPPED);
683 * @task: The #GstTask to pause
685 * Pauses @task. This method can also be called on a task in the
686 * stopped state, in which case a thread will be started and will remain
687 * in the paused state. This function does not wait for the task to complete
690 * Returns: %TRUE if the task could be paused.
695 gst_task_pause (GstTask * task)
697 return gst_task_set_state (task, GST_TASK_PAUSED);
702 * @task: The #GstTask to join
704 * Joins @task. After this call, it is safe to unref the task
705 * and clean up the lock set with gst_task_set_lock().
707 * The task will automatically be stopped with this call.
709 * This function cannot be called from within a task function as this
710 * would cause a deadlock. The function will detect this and print a
713 * Returns: %TRUE if the task could be joined.
718 gst_task_join (GstTask * task)
721 GstTaskPrivate *priv;
723 GstTaskPool *pool = NULL;
727 g_return_val_if_fail (GST_IS_TASK (task), FALSE);
729 tself = g_thread_self ();
731 GST_DEBUG_OBJECT (task, "Joining task %p, thread %p", task, tself);
733 /* we don't use a real thread join here because we are using
735 GST_OBJECT_LOCK (task);
736 if (G_UNLIKELY (tself == task->abidata.ABI.thread))
738 task->state = GST_TASK_STOPPED;
739 /* signal the state change for when it was blocked in PAUSED. */
740 GST_TASK_SIGNAL (task);
741 /* we set the running flag when pushing the task on the thread pool.
742 * This means that the task function might not be called when we try
743 * to join it here. */
744 while (G_LIKELY (task->running))
745 GST_TASK_WAIT (task);
746 /* clean the thread */
747 task->abidata.ABI.thread = NULL;
748 /* get the id and pool to join */
749 pool = priv->pool_id;
751 priv->pool_id = NULL;
753 GST_OBJECT_UNLOCK (task);
757 gst_task_pool_join (pool, id);
758 gst_object_unref (pool);
761 GST_DEBUG_OBJECT (task, "Joined task %p", task);
768 GST_WARNING_OBJECT (task, "trying to join task from its thread");
769 GST_OBJECT_UNLOCK (task);
770 g_warning ("\nTrying to join task %p from its thread would deadlock.\n"
771 "You cannot change the state of an element from its streaming\n"
772 "thread. Use g_idle_add() or post a GstMessage on the bus to\n"
773 "schedule the state change from the main thread.\n", task);