You've already forked FFmpeg
mirror of
https://github.com/FFmpeg/FFmpeg.git
synced 2025-08-10 06:10:52 +02:00
fftools/ffmpeg: move the input thread into its own file
It will contain more demuxing-specific code in the future.
This commit is contained in:
@@ -10,6 +10,7 @@ ALLAVPROGS = $(AVBASENAMES:%=%$(PROGSSUF)$(EXESUF))
|
|||||||
ALLAVPROGS_G = $(AVBASENAMES:%=%$(PROGSSUF)_g$(EXESUF))
|
ALLAVPROGS_G = $(AVBASENAMES:%=%$(PROGSSUF)_g$(EXESUF))
|
||||||
|
|
||||||
OBJS-ffmpeg += \
|
OBJS-ffmpeg += \
|
||||||
|
fftools/ffmpeg_demux.o \
|
||||||
fftools/ffmpeg_filter.o \
|
fftools/ffmpeg_filter.o \
|
||||||
fftools/ffmpeg_hw.o \
|
fftools/ffmpeg_hw.o \
|
||||||
fftools/ffmpeg_mux.o \
|
fftools/ffmpeg_mux.o \
|
||||||
|
110
fftools/ffmpeg.c
110
fftools/ffmpeg.c
@@ -161,8 +161,6 @@ static struct termios oldtty;
|
|||||||
static int restore_tty;
|
static int restore_tty;
|
||||||
#endif
|
#endif
|
||||||
|
|
||||||
static void free_input_threads(void);
|
|
||||||
|
|
||||||
/* sub2video hack:
|
/* sub2video hack:
|
||||||
Convert subtitles to video with alpha to insert them in filter graphs.
|
Convert subtitles to video with alpha to insert them in filter graphs.
|
||||||
This is a temporary solution until libavfilter gets real subtitles support.
|
This is a temporary solution until libavfilter gets real subtitles support.
|
||||||
@@ -3636,114 +3634,6 @@ static int check_keyboard_interaction(int64_t cur_time)
|
|||||||
return 0;
|
return 0;
|
||||||
}
|
}
|
||||||
|
|
||||||
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;
|
|
||||||
}
|
|
||||||
|
|
||||||
static 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);
|
|
||||||
}
|
|
||||||
|
|
||||||
static void free_input_threads(void)
|
|
||||||
{
|
|
||||||
int i;
|
|
||||||
|
|
||||||
for (i = 0; i < nb_input_files; i++)
|
|
||||||
free_input_thread(i);
|
|
||||||
}
|
|
||||||
|
|
||||||
static 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;
|
|
||||||
}
|
|
||||||
|
|
||||||
static 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;
|
|
||||||
}
|
|
||||||
|
|
||||||
static int get_input_packet(InputFile *f, AVPacket **pkt)
|
static int get_input_packet(InputFile *f, AVPacket **pkt)
|
||||||
{
|
{
|
||||||
if (f->readrate || f->rate_emu) {
|
if (f->readrate || f->rate_emu) {
|
||||||
|
@@ -710,4 +710,9 @@ int64_t of_filesize(OutputFile *of);
|
|||||||
AVChapter * const *
|
AVChapter * const *
|
||||||
of_get_chapters(OutputFile *of, unsigned int *nb_chapters);
|
of_get_chapters(OutputFile *of, unsigned int *nb_chapters);
|
||||||
|
|
||||||
|
int init_input_threads(void);
|
||||||
|
int init_input_thread(int i);
|
||||||
|
void free_input_threads(void);
|
||||||
|
void free_input_thread(int i);
|
||||||
|
|
||||||
#endif /* FFTOOLS_FFMPEG_H */
|
#endif /* FFTOOLS_FFMPEG_H */
|
||||||
|
136
fftools/ffmpeg_demux.c
Normal file
136
fftools/ffmpeg_demux.c
Normal file
@@ -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;
|
||||||
|
}
|
Reference in New Issue
Block a user