2020-11-09 06:09:13 +00:00
|
|
|
/*
|
|
|
|
* Copyright (c) 2020
|
|
|
|
*
|
|
|
|
* This file is part of FFmpeg.
|
|
|
|
*
|
|
|
|
* FFmpeg is free software; you can redistribute it and/or
|
|
|
|
* modify it under the terms of the GNU Lesser General Public
|
|
|
|
* License as published by the Free Software Foundation; either
|
|
|
|
* version 2.1 of the License, or (at your option) any later version.
|
|
|
|
*
|
|
|
|
* FFmpeg is distributed in the hope that it will be useful,
|
|
|
|
* but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
|
|
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
|
|
|
|
* Lesser General Public License for more details.
|
|
|
|
*
|
|
|
|
* You should have received a copy of the GNU Lesser General Public
|
|
|
|
* License along with FFmpeg; if not, write to the Free Software
|
|
|
|
* Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
|
|
|
|
*/
|
|
|
|
|
|
|
|
#include <stdio.h>
|
|
|
|
#include "queue.h"
|
|
|
|
#include "safe_queue.h"
|
|
|
|
#include "libavutil/mem.h"
|
|
|
|
#include "libavutil/avassert.h"
|
|
|
|
#include "libavutil/thread.h"
|
|
|
|
|
2020-12-30 02:47:29 +00:00
|
|
|
#if HAVE_PTHREAD_CANCEL
|
|
|
|
#define DNNCond pthread_cond_t
|
|
|
|
#define dnn_cond_init pthread_cond_init
|
|
|
|
#define dnn_cond_destroy pthread_cond_destroy
|
|
|
|
#define dnn_cond_signal pthread_cond_signal
|
|
|
|
#define dnn_cond_wait pthread_cond_wait
|
|
|
|
#else
|
|
|
|
#define DNNCond char
|
|
|
|
static inline int dnn_cond_init(DNNCond *cond, const void *attr) { return 0; }
|
|
|
|
static inline int dnn_cond_destroy(DNNCond *cond) { return 0; }
|
|
|
|
static inline int dnn_cond_signal(DNNCond *cond) { return 0; }
|
|
|
|
static inline int dnn_cond_wait(DNNCond *cond, AVMutex *mutex)
|
|
|
|
{
|
|
|
|
av_assert0(!"should not reach here");
|
|
|
|
return 0;
|
|
|
|
}
|
|
|
|
#endif
|
|
|
|
|
2020-12-30 01:37:14 +00:00
|
|
|
struct FFSafeQueue {
|
|
|
|
FFQueue *q;
|
2020-12-30 02:47:29 +00:00
|
|
|
AVMutex mutex;
|
|
|
|
DNNCond cond;
|
2020-12-30 01:37:14 +00:00
|
|
|
};
|
2020-11-09 06:09:13 +00:00
|
|
|
|
2020-12-30 01:37:14 +00:00
|
|
|
FFSafeQueue *ff_safe_queue_create(void)
|
2020-11-09 06:09:13 +00:00
|
|
|
{
|
2020-12-30 01:37:14 +00:00
|
|
|
FFSafeQueue *sq = av_malloc(sizeof(*sq));
|
2020-11-09 06:09:13 +00:00
|
|
|
if (!sq)
|
|
|
|
return NULL;
|
|
|
|
|
2020-12-30 01:37:14 +00:00
|
|
|
sq->q = ff_queue_create();
|
2020-12-30 05:11:27 +00:00
|
|
|
if (!sq->q) {
|
|
|
|
av_freep(&sq);
|
2020-11-09 06:09:13 +00:00
|
|
|
return NULL;
|
2020-12-30 05:11:27 +00:00
|
|
|
}
|
2020-11-09 06:09:13 +00:00
|
|
|
|
2020-12-30 02:47:29 +00:00
|
|
|
ff_mutex_init(&sq->mutex, NULL);
|
|
|
|
dnn_cond_init(&sq->cond, NULL);
|
2020-11-09 06:09:13 +00:00
|
|
|
return sq;
|
|
|
|
}
|
|
|
|
|
2020-12-30 01:37:14 +00:00
|
|
|
void ff_safe_queue_destroy(FFSafeQueue *sq)
|
2020-11-09 06:09:13 +00:00
|
|
|
{
|
|
|
|
if (!sq)
|
|
|
|
return;
|
|
|
|
|
2020-12-30 01:37:14 +00:00
|
|
|
ff_queue_destroy(sq->q);
|
2020-12-30 02:47:29 +00:00
|
|
|
ff_mutex_destroy(&sq->mutex);
|
|
|
|
dnn_cond_destroy(&sq->cond);
|
2020-11-09 06:09:13 +00:00
|
|
|
av_freep(&sq);
|
|
|
|
}
|
|
|
|
|
2020-12-30 01:37:14 +00:00
|
|
|
size_t ff_safe_queue_size(FFSafeQueue *sq)
|
2020-11-09 06:09:13 +00:00
|
|
|
{
|
2020-12-30 01:37:14 +00:00
|
|
|
return sq ? ff_queue_size(sq->q) : 0;
|
2020-11-09 06:09:13 +00:00
|
|
|
}
|
|
|
|
|
2020-12-30 05:11:27 +00:00
|
|
|
int ff_safe_queue_push_front(FFSafeQueue *sq, void *v)
|
2020-11-09 06:09:13 +00:00
|
|
|
{
|
2020-12-30 05:11:27 +00:00
|
|
|
int ret;
|
2020-12-30 02:47:29 +00:00
|
|
|
ff_mutex_lock(&sq->mutex);
|
2020-12-30 05:11:27 +00:00
|
|
|
ret = ff_queue_push_front(sq->q, v);
|
2020-12-30 02:47:29 +00:00
|
|
|
dnn_cond_signal(&sq->cond);
|
|
|
|
ff_mutex_unlock(&sq->mutex);
|
2020-12-30 05:11:27 +00:00
|
|
|
return ret;
|
2020-11-09 06:09:13 +00:00
|
|
|
}
|
|
|
|
|
2020-12-30 05:11:27 +00:00
|
|
|
int ff_safe_queue_push_back(FFSafeQueue *sq, void *v)
|
2020-11-09 06:09:13 +00:00
|
|
|
{
|
2020-12-30 05:11:27 +00:00
|
|
|
int ret;
|
2020-12-30 02:47:29 +00:00
|
|
|
ff_mutex_lock(&sq->mutex);
|
2020-12-30 05:11:27 +00:00
|
|
|
ret = ff_queue_push_back(sq->q, v);
|
2020-12-30 02:47:29 +00:00
|
|
|
dnn_cond_signal(&sq->cond);
|
|
|
|
ff_mutex_unlock(&sq->mutex);
|
2020-12-30 05:11:27 +00:00
|
|
|
return ret;
|
2020-11-09 06:09:13 +00:00
|
|
|
}
|
|
|
|
|
2020-12-30 01:37:14 +00:00
|
|
|
void *ff_safe_queue_pop_front(FFSafeQueue *sq)
|
2020-11-09 06:09:13 +00:00
|
|
|
{
|
|
|
|
void *value;
|
2020-12-30 02:47:29 +00:00
|
|
|
ff_mutex_lock(&sq->mutex);
|
2020-12-30 01:37:14 +00:00
|
|
|
while (ff_queue_size(sq->q) == 0) {
|
2020-12-30 02:47:29 +00:00
|
|
|
dnn_cond_wait(&sq->cond, &sq->mutex);
|
2020-11-09 06:09:13 +00:00
|
|
|
}
|
2020-12-30 01:37:14 +00:00
|
|
|
value = ff_queue_pop_front(sq->q);
|
2020-12-30 02:47:29 +00:00
|
|
|
dnn_cond_signal(&sq->cond);
|
|
|
|
ff_mutex_unlock(&sq->mutex);
|
2020-11-09 06:09:13 +00:00
|
|
|
return value;
|
|
|
|
}
|