mirror of
https://git.ffmpeg.org/ffmpeg.git
synced 2025-12-13 10:30:05 +01:00
avutil/threadmessage: add av_thread_message_flush()
This commit is contained in:
@@ -40,6 +40,7 @@ struct AVThreadMessageQueue {
|
||||
int err_send;
|
||||
int err_recv;
|
||||
unsigned elsize;
|
||||
void (*free_func)(void *msg);
|
||||
#else
|
||||
int dummy;
|
||||
#endif
|
||||
@@ -81,10 +82,17 @@ int av_thread_message_queue_alloc(AVThreadMessageQueue **mq,
|
||||
#endif /* HAVE_THREADS */
|
||||
}
|
||||
|
||||
void av_thread_message_queue_set_free_func(AVThreadMessageQueue *mq,
|
||||
void (*free_func)(void *msg))
|
||||
{
|
||||
mq->free_func = free_func;
|
||||
}
|
||||
|
||||
void av_thread_message_queue_free(AVThreadMessageQueue **mq)
|
||||
{
|
||||
#if HAVE_THREADS
|
||||
if (*mq) {
|
||||
av_thread_message_flush(*mq);
|
||||
av_fifo_freep(&(*mq)->fifo);
|
||||
pthread_cond_destroy(&(*mq)->cond);
|
||||
pthread_mutex_destroy(&(*mq)->lock);
|
||||
@@ -182,3 +190,26 @@ void av_thread_message_queue_set_err_recv(AVThreadMessageQueue *mq,
|
||||
pthread_mutex_unlock(&mq->lock);
|
||||
#endif /* HAVE_THREADS */
|
||||
}
|
||||
|
||||
static void free_func_wrap(void *arg, void *msg, int size)
|
||||
{
|
||||
AVThreadMessageQueue *mq = arg;
|
||||
mq->free_func(msg);
|
||||
}
|
||||
|
||||
void av_thread_message_flush(AVThreadMessageQueue *mq)
|
||||
{
|
||||
#if HAVE_THREADS
|
||||
int used, off;
|
||||
void *free_func = mq->free_func;
|
||||
|
||||
pthread_mutex_lock(&mq->lock);
|
||||
used = av_fifo_size(mq->fifo);
|
||||
if (free_func)
|
||||
for (off = 0; off < used; off += mq->elsize)
|
||||
av_fifo_generic_peek_at(mq->fifo, mq, off, mq->elsize, free_func_wrap);
|
||||
av_fifo_drain(mq->fifo, used);
|
||||
pthread_cond_broadcast(&mq->cond);
|
||||
pthread_mutex_unlock(&mq->lock);
|
||||
#endif /* HAVE_THREADS */
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user