diff options
author | Anton Khirnov <anton@khirnov.net> | 2022-08-03 14:08:44 +0200 |
---|---|---|
committer | Anton Khirnov <anton@khirnov.net> | 2022-08-08 16:20:58 +0200 |
commit | 7b6622705e552b1dd5284b743974b66abe04954b (patch) | |
tree | 92dc1a8fe1d2015071937a6357d141bdac761f8c /fftools/ffmpeg_demux.c | |
parent | c3b8d59d2f40550369dbfb2505f2f8415f1e37ec (diff) | |
download | ffmpeg-7b6622705e552b1dd5284b743974b66abe04954b.tar.gz |
fftools/ffmpeg: move the input thread into its own file
It will contain more demuxing-specific code in the future.
Diffstat (limited to 'fftools/ffmpeg_demux.c')
-rw-r--r-- | fftools/ffmpeg_demux.c | 136 |
1 files changed, 136 insertions, 0 deletions
diff --git a/fftools/ffmpeg_demux.c b/fftools/ffmpeg_demux.c new file mode 100644 index 0000000000..2629af5950 --- /dev/null +++ b/fftools/ffmpeg_demux.c @@ -0,0 +1,136 @@ +/* + * 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 "ffmpeg.h" + +#include "libavutil/error.h" +#include "libavutil/time.h" +#include "libavutil/thread.h" +#include "libavutil/threadmessage.h" + +#include "libavcodec/packet.h" + +#include "libavformat/avformat.h" + +static void *input_thread(void *arg) +{ + InputFile *f = arg; + AVPacket *pkt = f->pkt, *queue_pkt; + unsigned flags = f->non_blocking ? AV_THREAD_MESSAGE_NONBLOCK : 0; + int ret = 0; + + while (1) { + ret = av_read_frame(f->ctx, pkt); + + if (ret == AVERROR(EAGAIN)) { + av_usleep(10000); + continue; + } + if (ret < 0) { + av_thread_message_queue_set_err_recv(f->in_thread_queue, ret); + break; + } + queue_pkt = av_packet_alloc(); + if (!queue_pkt) { + av_packet_unref(pkt); + av_thread_message_queue_set_err_recv(f->in_thread_queue, AVERROR(ENOMEM)); + break; + } + av_packet_move_ref(queue_pkt, pkt); + ret = av_thread_message_queue_send(f->in_thread_queue, &queue_pkt, flags); + if (flags && ret == AVERROR(EAGAIN)) { + flags = 0; + ret = av_thread_message_queue_send(f->in_thread_queue, &queue_pkt, flags); + av_log(f->ctx, AV_LOG_WARNING, + "Thread message queue blocking; consider raising the " + "thread_queue_size option (current value: %d)\n", + f->thread_queue_size); + } + if (ret < 0) { + if (ret != AVERROR_EOF) + av_log(f->ctx, AV_LOG_ERROR, + "Unable to send packet to main thread: %s\n", + av_err2str(ret)); + av_packet_free(&queue_pkt); + av_thread_message_queue_set_err_recv(f->in_thread_queue, ret); + break; + } + } + + return NULL; +} + +void free_input_thread(int i) +{ + InputFile *f = input_files[i]; + AVPacket *pkt; + + if (!f || !f->in_thread_queue) + return; + av_thread_message_queue_set_err_send(f->in_thread_queue, AVERROR_EOF); + while (av_thread_message_queue_recv(f->in_thread_queue, &pkt, 0) >= 0) + av_packet_free(&pkt); + + pthread_join(f->thread, NULL); + av_thread_message_queue_free(&f->in_thread_queue); +} + +void free_input_threads(void) +{ + int i; + + for (i = 0; i < nb_input_files; i++) + free_input_thread(i); +} + +int init_input_thread(int i) +{ + int ret; + InputFile *f = input_files[i]; + + if (f->thread_queue_size <= 0) + f->thread_queue_size = (nb_input_files > 1 ? 8 : 1); + + if (f->ctx->pb ? !f->ctx->pb->seekable : + strcmp(f->ctx->iformat->name, "lavfi")) + f->non_blocking = 1; + ret = av_thread_message_queue_alloc(&f->in_thread_queue, + f->thread_queue_size, sizeof(f->pkt)); + if (ret < 0) + return ret; + + if ((ret = pthread_create(&f->thread, NULL, input_thread, f))) { + av_log(NULL, AV_LOG_ERROR, "pthread_create failed: %s. Try to increase `ulimit -v` or decrease `ulimit -s`.\n", strerror(ret)); + av_thread_message_queue_free(&f->in_thread_queue); + return AVERROR(ret); + } + + return 0; +} + +int init_input_threads(void) +{ + int i, ret; + + for (i = 0; i < nb_input_files; i++) { + ret = init_input_thread(i); + if (ret < 0) + return ret; + } + return 0; +} |