mirror of https://github.com/FFmpeg/FFmpeg.git
It will contain more demuxing-specific code in the future.pull/388/head
Anton Khirnov
2 years ago
parent
c3b8d59d2f
commit
7b6622705e
4 changed files with 142 additions and 110 deletions
@ -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; |
||||||
|
} |
||||||
Loading…
Reference in new issue