Skip to content
Snippets Groups Projects
ffmpeg.c 179 KiB
Newer Older
  • Learn to ignore specific revisions
  • Fabrice Bellard's avatar
    Fabrice Bellard committed
    /*
    
     * ffmpeg main
    
    Fabrice Bellard's avatar
    Fabrice Bellard committed
     * Copyright (c) 2000-2003 Fabrice Bellard
    
    Fabrice Bellard's avatar
    Fabrice Bellard committed
     *
    
     * 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.
    
    Fabrice Bellard's avatar
    Fabrice Bellard committed
     *
    
     * FFmpeg is distributed in the hope that it will be useful,
    
    Fabrice Bellard's avatar
    Fabrice Bellard committed
     * 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.
    
    Fabrice Bellard's avatar
    Fabrice Bellard committed
     *
    
     * 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
    
    Fabrice Bellard's avatar
    Fabrice Bellard committed
     */
    
    #include "config.h"
    #include <ctype.h>
    #include <string.h>
    #include <math.h>
    #include <stdlib.h>
    #include <errno.h>
    
    #include <signal.h>
    
    #include <limits.h>
    
    #include "libavformat/avformat.h"
    #include "libavdevice/avdevice.h"
    #include "libswscale/swscale.h"
    
    #include "libavutil/opt.h"
    
    #include "libavcodec/audioconvert.h"
    
    #include "libavutil/audioconvert.h"
    #include "libavutil/parseutils.h"
    #include "libavutil/samplefmt.h"
    
    #include "libavutil/colorspace.h"
    
    #include "libavutil/fifo.h"
    
    #include "libavutil/intreadwrite.h"
    
    #include "libavutil/dict.h"
    
    #include "libavutil/mathematics.h"
    
    #include "libavutil/avstring.h"
    
    #include "libavutil/libm.h"
    
    #include "libavformat/os_support.h"
    
    Michael Niedermayer's avatar
    Michael Niedermayer committed
    #include "libswresample/swresample.h"
    
    #include "libavformat/ffm.h" // not public API
    
    
    #if CONFIG_AVFILTER
    
    # include "libavfilter/avcodec.h"
    
    # include "libavfilter/avfilter.h"
    # include "libavfilter/avfiltergraph.h"
    
    # include "libavfilter/vsrc_buffer.h"
    #endif
    
    
    #if HAVE_SYS_RESOURCE_H
    
    #include <sys/resource.h>
    
    #elif HAVE_GETPROCESSTIMES
    
    #include <windows.h>
    #endif
    
    #if HAVE_GETPROCESSMEMORYINFO
    #include <windows.h>
    #include <psapi.h>
    #endif
    
    #if HAVE_SYS_SELECT_H
    
    #include <sys/select.h>
    #endif
    
    
    #if HAVE_TERMIOS_H
    #include <fcntl.h>
    #include <sys/ioctl.h>
    #include <sys/time.h>
    #include <termios.h>
    #elif HAVE_KBHIT
    
    #include <conio.h>
    
    Fabrice Bellard's avatar
    Fabrice Bellard committed
    #endif
    
    #include <time.h>
    
    Fabrice Bellard's avatar
    Fabrice Bellard committed
    
    
    Fabrice Bellard's avatar
    Fabrice Bellard committed
    #include "cmdutils.h"
    
    
    #include "libavutil/avassert.h"
    
    const char program_name[] = "ffmpeg";
    
    const int program_birth_year = 2000;
    
    Fabrice Bellard's avatar
    Fabrice Bellard committed
    /* select an input stream for an output stream */
    
    typedef struct StreamMap {
    
        int disabled;           /** 1 is this mapping is disabled by a negative map */
    
    Fabrice Bellard's avatar
    Fabrice Bellard committed
        int file_index;
        int stream_index;
    
        int sync_file_index;
        int sync_stream_index;
    
    Fabrice Bellard's avatar
    Fabrice Bellard committed
    
    
    typedef struct {
        int  file_idx,  stream_idx,  channel_idx; // input
        int ofile_idx, ostream_idx;               // output
    } AudioChannelMap;
    
    
    /**
     * select an input file for an output file
     */
    
    typedef struct MetadataMap {
    
        int  file;      ///< file index
        char type;      ///< type of metadata to copy -- (g)lobal, (s)tream, (c)hapter or (p)rogram
        int  index;     ///< stream/chapter/program number
    
    Fabrice Bellard's avatar
    Fabrice Bellard committed
    
    
    #define MAX_STREAMS 1024    /* arbitrary sanity check value */
    
    static int frame_bits_per_raw_sample = 0;
    
    static int video_discard = 0;
    
    static int same_quant = 0;
    
    static int do_deinterlace = 0;
    
    Michael Niedermayer's avatar
    Michael Niedermayer committed
    static int intra_dc_precision = 8;
    
    static int loop_input = 0;
    
    static int loop_output = AVFMT_NOOUTPUTLOOP;
    
    static int qp_hist = 0;
    
    Fabrice Bellard's avatar
    Fabrice Bellard committed
    static int intra_only = 0;
    
    static const char *video_codec_name    = NULL;
    static const char *audio_codec_name    = NULL;
    static const char *subtitle_codec_name = NULL;
    
    Fabrice Bellard's avatar
    Fabrice Bellard committed
    
    static int file_overwrite = 0;
    
    Fabrice Bellard's avatar
    Fabrice Bellard committed
    static int do_benchmark = 0;
    
    static int do_hex_dump = 0;
    
    static int do_pass = 0;
    
    static const char *pass_logfilename_prefix;
    
    static int video_sync_method= -1;
    
    static int audio_sync_method= 0;
    
    static float audio_drift_threshold= 0.1;
    
    static int copy_ts= 0;
    
    static int opt_shortest = 0;
    
    static char *vstats_filename;
    
    static FILE *vstats_file;
    
    static int copy_initial_nonkeyframes = 0;
    
    static int run_as_daemon  = 0;
    
    static volatile int received_nb_signals = 0;
    
    static int64_t video_size = 0;
    static int64_t audio_size = 0;
    static int64_t extra_size = 0;
    
    static int nb_frames_dup = 0;
    static int nb_frames_drop = 0;
    
    static float dts_delta_threshold = 10;
    
    static int print_stats = 1;
    
    static uint8_t *audio_buf;
    static uint8_t *audio_out;
    
    static unsigned int allocated_audio_out_size, allocated_audio_buf_size;
    
    static void *samples;
    
    static uint8_t *input_tmp= NULL;
    
    
    Stefano Sabatini's avatar
    Stefano Sabatini committed
    #define DEFAULT_PASS_LOGFILENAME_PREFIX "ffmpeg2pass"
    
    Fabrice Bellard's avatar
    Fabrice Bellard committed
    
    
    typedef struct InputStream {
        int file_index;
        AVStream *st;
        int discard;             /* true if stream data should be discarded */
        int decoding_needed;     /* true if the packets must be decoded in 'raw_fifo' */
        AVCodec *dec;
    
        int64_t       start;     /* time when read started */
        int64_t       next_pts;  /* synthetic pts for cases where pkt.pts
                                    is not defined */
        int64_t       pts;       /* current pts */
        double ts_scale;
        int is_start;            /* is 1 at the start and after a discontinuity */
        int showed_multi_packet_warning;
        AVDictionary *opts;
    } InputStream;
    
    typedef struct InputFile {
        AVFormatContext *ctx;
        int eof_reached;      /* true if eof reached */
    
        int ist_index;        /* index of first stream in input_streams */
    
        int buffer_size;      /* current total buffer size */
        int64_t ts_offset;
    
        int nb_streams;       /* number of stream that ffmpeg is aware of; may be different
    
                                 from ctx.nb_streams if new streams appear during av_read_frame() */
    
    } InputFile;
    
    typedef struct OutputStream {
    
    Fabrice Bellard's avatar
    Fabrice Bellard committed
        int file_index;          /* file index */
        int index;               /* stream index in the output file */
    
        int source_index;        /* InputStream index */
    
    Fabrice Bellard's avatar
    Fabrice Bellard committed
        AVStream *st;            /* stream in the output file */
    
        int encoding_needed;     /* true if encoding needed for this stream */
        int frame_number;
        /* input pts and corresponding output pts
           for A/V sync */
    
        //double sync_ipts;        /* dts from the AVPacket of the demuxer in second units */
    
        struct InputStream *sync_ist; /* input stream to sync against */
    
        int64_t sync_opts;       /* output frame counter, could be changed to some true timestamp */ //FIXME look at frame_number
    
        AVBitStreamFilterContext *bitstream_filters;
    
        int64_t max_frames;
    
    Fabrice Bellard's avatar
    Fabrice Bellard committed
        /* video only */
    
        AVFrame resample_frame;              /* temporary frame for image resampling */
    
        struct SwsContext *img_resample_ctx; /* for image resampling */
        int resample_height;
    
        int resample_width;
    
        int force_fps;
        int top_field_first;
    
        float frame_aspect_ratio;
    
        /* forced key frames */
        int64_t *forced_kf_pts;
        int forced_kf_count;
        int forced_kf_index;
    
    
    Fabrice Bellard's avatar
    Fabrice Bellard committed
        /* audio only */
        int audio_resample;
    
        int audio_channels_map[SWR_CH_MAX];  ///< list of the channels id to pick from the source stream
        int audio_channels_mapped;           ///< number of channels in audio_channels_map
    
        int resample_sample_fmt;
        int resample_channels;
        int resample_sample_rate;
    
        AVFifoBuffer *fifo;     /* for compression: one audio fifo per codec */
    
    Michael Niedermayer's avatar
    Michael Niedermayer committed
        struct SwrContext *swr;
    
    
    #if CONFIG_AVFILTER
        AVFilterContext *output_video_filter;
        AVFilterContext *input_video_filter;
        AVFilterBufferRef *picref;
        char *avfilter;
        AVFilterGraph *graph;
    #endif
    
       int is_past_recording_time;
    
       const char *attachment_filename;
    
    } OutputStream;
    
    Fabrice Bellard's avatar
    Fabrice Bellard committed
    
    
    
    #if HAVE_TERMIOS_H
    
    /* init terminal so that we can grab keys */
    static struct termios oldtty;
    #endif
    
    
    typedef struct OutputFile {
        AVFormatContext *ctx;
        AVDictionary *opts;
    
        int ost_index;       /* index of the first stream in output_streams */
    
        int64_t recording_time; /* desired length of the resulting file in microseconds */
    
        int64_t start_time;     /* start time in microseconds */
    
        uint64_t limit_filesize;
    
    static InputStream *input_streams = NULL;
    
    static int         nb_input_streams = 0;
    
    static InputFile   *input_files   = NULL;
    
    static int         nb_input_files   = 0;
    
    
    static OutputStream *output_streams = NULL;
    static int        nb_output_streams = 0;
    
    static OutputFile   *output_files   = NULL;
    static int        nb_output_files   = 0;
    
    
    typedef struct OptionsContext {
        /* input/output options */
        int64_t start_time;
        const char *format;
    
    
        SpecifierOpt *codec_names;
        int        nb_codec_names;
    
        SpecifierOpt *audio_channels;
        int        nb_audio_channels;
    
        SpecifierOpt *audio_sample_rate;
        int        nb_audio_sample_rate;
    
        SpecifierOpt *rematrix_volume;
        int        nb_rematrix_volume;
    
        SpecifierOpt *frame_rates;
        int        nb_frame_rates;
        SpecifierOpt *frame_sizes;
        int        nb_frame_sizes;
        SpecifierOpt *frame_pix_fmts;
        int        nb_frame_pix_fmts;
    
        /* input options */
        int64_t input_ts_offset;
    
        int rate_emu;
    
        SpecifierOpt *ts_scale;
        int        nb_ts_scale;
    
        SpecifierOpt *dump_attachment;
        int        nb_dump_attachment;
    
    
        /* output options */
        StreamMap *stream_maps;
        int     nb_stream_maps;
    
        AudioChannelMap *audio_channel_maps; ///< one info entry per -map_channel
        int           nb_audio_channel_maps; ///< number of (valid) -map_channel settings
    
        /* first item specifies output metadata, second is input */
        MetadataMap (*meta_data_maps)[2];
        int nb_meta_data_maps;
        int metadata_global_manual;
        int metadata_streams_manual;
        int metadata_chapters_manual;
    
        const char **attachments;
        int       nb_attachments;
    
    
        int chapters_input_file;
    
    
        int64_t recording_time;
        uint64_t limit_filesize;
    
        float mux_preload;
        float mux_max_delay;
    
    
        int video_disable;
        int audio_disable;
        int subtitle_disable;
        int data_disable;
    
    
        /* indexed by output file stream index */
        int   *streamid_map;
        int nb_streamid_map;
    
    
        SpecifierOpt *metadata;
        int        nb_metadata;
        SpecifierOpt *max_frames;
        int        nb_max_frames;
        SpecifierOpt *bitstream_filters;
        int        nb_bitstream_filters;
    
        SpecifierOpt *codec_tags;
        int        nb_codec_tags;
    
        SpecifierOpt *sample_fmts;
        int        nb_sample_fmts;
        SpecifierOpt *qscale;
        int        nb_qscale;
        SpecifierOpt *forced_key_frames;
        int        nb_forced_key_frames;
        SpecifierOpt *force_fps;
        int        nb_force_fps;
        SpecifierOpt *frame_aspect_ratios;
        int        nb_frame_aspect_ratios;
        SpecifierOpt *rc_overrides;
        int        nb_rc_overrides;
        SpecifierOpt *intra_matrices;
        int        nb_intra_matrices;
        SpecifierOpt *inter_matrices;
        int        nb_inter_matrices;
        SpecifierOpt *top_field_first;
        int        nb_top_field_first;
    
        SpecifierOpt *presets;
        int        nb_presets;
    
    #if CONFIG_AVFILTER
        SpecifierOpt *filters;
        int        nb_filters;
    #endif
    
    #define MATCH_PER_STREAM_OPT(name, type, outvar, fmtctx, st)\
    {\
        int i, ret;\
        for (i = 0; i < o->nb_ ## name; i++) {\
            char *spec = o->name[i].specifier;\
            if ((ret = check_stream_specifier(fmtctx, st, spec)) > 0)\
                outvar = o->name[i].u.type;\
            else if (ret < 0)\
                exit_program(1);\
        }\
    }
    
    
    static void reset_options(OptionsContext *o, int is_input)
    
    {
        const OptionDef *po = options;
    
    
        /* all OPT_SPEC and OPT_STRING can be freed in generic way */
        while (po->name) {
            void *dst = (uint8_t*)o + po->u.off;
    
            if (po->flags & OPT_SPEC) {
                SpecifierOpt **so = dst;
                int i, *count = (int*)(so + 1);
                for (i = 0; i < *count; i++) {
                    av_freep(&(*so)[i].specifier);
                    if (po->flags & OPT_STRING)
                        av_freep(&(*so)[i].u.str);
                }
                av_freep(so);
                *count = 0;
            } else if (po->flags & OPT_OFFSET && po->flags & OPT_STRING)
                av_freep(dst);
            po++;
        }
    
        av_freep(&o->stream_maps);
    
        av_freep(&o->audio_channel_maps);
    
        av_freep(&o->meta_data_maps);
    
        av_freep(&o->streamid_map);
    
        if(is_input) o->recording_time = bak.recording_time;
        else         o->recording_time = INT64_MAX;
    
        o->mux_max_delay  = 0.7;
        o->limit_filesize = UINT64_MAX;
        o->chapters_input_file = INT_MAX;
    
    #if CONFIG_AVFILTER
    
    
    static int configure_video_filters(InputStream *ist, OutputStream *ost)
    
        AVFilterContext *last_filter, *filter;
    
        /** filter graph containing all filters including input & output */
        AVCodecContext *codec = ost->st->codec;
        AVCodecContext *icodec = ist->st->codec;
    
        enum PixelFormat pix_fmts[] = { codec->pix_fmt, PIX_FMT_NONE };
    
        AVBufferSinkParams *buffersink_params = av_buffersink_params_alloc();
    
        AVRational sample_aspect_ratio;
    
        char args[255];
    
        ost->graph = avfilter_graph_alloc();
    
        if (ist->st->sample_aspect_ratio.num){
            sample_aspect_ratio = ist->st->sample_aspect_ratio;
        }else
            sample_aspect_ratio = ist->st->codec->sample_aspect_ratio;
    
        snprintf(args, 255, "%d:%d:%d:%d:%d:%d:%d", ist->st->codec->width,
                 ist->st->codec->height, ist->st->codec->pix_fmt, 1, AV_TIME_BASE,
                 sample_aspect_ratio.num, sample_aspect_ratio.den);
    
    
        ret = avfilter_graph_create_filter(&ost->input_video_filter, avfilter_get_by_name("buffer"),
                                           "src", args, NULL, ost->graph);
    
        if (ret < 0)
    
    #if FF_API_OLD_VSINK_API
    
        ret = avfilter_graph_create_filter(&ost->output_video_filter, avfilter_get_by_name("buffersink"),
                                           "out", NULL, pix_fmts, ost->graph);
    
    #else
        buffersink_params->pixel_fmts = pix_fmts;
        ret = avfilter_graph_create_filter(&ost->output_video_filter, avfilter_get_by_name("buffersink"),
                                           "out", NULL, buffersink_params, ost->graph);
    #endif
        av_freep(&buffersink_params);
    
        if (ret < 0)
    
        last_filter = ost->input_video_filter;
    
    Stefano Sabatini's avatar
    Stefano Sabatini committed
        if (codec->width  != icodec->width || codec->height != icodec->height) {
    
            snprintf(args, 255, "%d:%d:flags=0x%X",
    
                     codec->width,
                     codec->height,
    
                     (unsigned)ost->sws_flags);
    
            if ((ret = avfilter_graph_create_filter(&filter, avfilter_get_by_name("scale"),
    
                                                    NULL, args, NULL, ost->graph)) < 0)
    
                return ret;
            if ((ret = avfilter_link(last_filter, 0, filter, 0)) < 0)
                return ret;
    
        snprintf(args, sizeof(args), "flags=0x%X", (unsigned)ost->sws_flags);
    
        ost->graph->scale_sws_opts = av_strdup(args);
    
            AVFilterInOut *outputs = avfilter_inout_alloc();
            AVFilterInOut *inputs  = avfilter_inout_alloc();
    
    
            outputs->name    = av_strdup("in");
    
            outputs->pad_idx = 0;
            outputs->next    = NULL;
    
            inputs->name    = av_strdup("out");
    
            inputs->filter_ctx = ost->output_video_filter;
    
            inputs->pad_idx = 0;
            inputs->next    = NULL;
    
    
            if ((ret = avfilter_graph_parse(ost->graph, ost->avfilter, &inputs, &outputs, NULL)) < 0)
    
            av_freep(&ost->avfilter);
    
            if ((ret = avfilter_link(last_filter, 0, ost->output_video_filter, 0)) < 0)
    
        if ((ret = avfilter_graph_config(ost->graph, NULL)) < 0)
    
        codec->width  = ost->output_video_filter->inputs[0]->w;
        codec->height = ost->output_video_filter->inputs[0]->h;
    
        codec->sample_aspect_ratio = ost->st->sample_aspect_ratio =
    
            ost->frame_aspect_ratio ? // overridden by the -aspect cli option
    
            av_d2q(ost->frame_aspect_ratio*codec->height/codec->width, 255) :
    
            ost->output_video_filter->inputs[0]->sample_aspect_ratio;
    
    
        return 0;
    }
    #endif /* CONFIG_AVFILTER */
    
    
    Fabrice Bellard's avatar
    Fabrice Bellard committed
    static void term_exit(void)
    {
    
        av_log(NULL, AV_LOG_QUIET, "%s", "");
    
            tcsetattr (0, TCSANOW, &oldtty);
    
    Fabrice Bellard's avatar
    Fabrice Bellard committed
    
    
    static volatile int received_sigterm = 0;
    
    static void sigterm_handler(int sig)
    
    Fabrice Bellard's avatar
    Fabrice Bellard committed
    static void term_init(void)
    {
    
        struct termios tty;
    
        tcgetattr (0, &tty);
        oldtty = tty;
        atexit(term_exit);
    
        tty.c_iflag &= ~(IGNBRK|BRKINT|PARMRK|ISTRIP
                              |INLCR|IGNCR|ICRNL|IXON);
        tty.c_oflag |= OPOST;
        tty.c_lflag &= ~(ECHO|ECHONL|ICANON|IEXTEN);
        tty.c_cflag &= ~(CSIZE|PARENB);
        tty.c_cflag |= CS8;
        tty.c_cc[VMIN] = 1;
        tty.c_cc[VTIME] = 0;
    
        tcsetattr (0, TCSANOW, &tty);
        signal(SIGQUIT, sigterm_handler); /* Quit (POSIX).  */
    
        signal(SIGINT , sigterm_handler); /* Interrupt (ANSI).  */
        signal(SIGTERM, sigterm_handler); /* Termination (ANSI).  */
    
    Måns Rullgård's avatar
    Måns Rullgård committed
    #ifdef SIGXCPU
        signal(SIGXCPU, sigterm_handler);
    #endif
    
    Fabrice Bellard's avatar
    Fabrice Bellard committed
    }
    
    /* read a key without blocking */
    static int read_key(void)
    {
    
    #if HAVE_TERMIOS_H
        int n = 1;
        struct timeval tv;
        fd_set rfds;
    
        FD_ZERO(&rfds);
        FD_SET(0, &rfds);
        tv.tv_sec = 0;
        tv.tv_usec = 0;
        n = select(1, &rfds, NULL, NULL, &tv);
        if (n > 0) {
            n = read(0, &ch, 1);
            if (n == 1)
                return ch;
    
            return n;
        }
    #elif HAVE_KBHIT
    
    #    if HAVE_PEEKNAMEDPIPE
        static int is_pipe;
        static HANDLE input_handle;
        DWORD dw, nchars;
        if(!input_handle){
            input_handle = GetStdHandle(STD_INPUT_HANDLE);
            is_pipe = !GetConsoleMode(input_handle, &dw);
        }
    
        if (stdin->_cnt > 0) {
            read(0, &ch, 1);
            return ch;
        }
        if (is_pipe) {
            /* When running under a GUI, you will end here. */
            if (!PeekNamedPipe(input_handle, NULL, 0, NULL, &nchars, NULL))
                return -1;
            //Read it
            if(nchars != 0) {
                read(0, &ch, 1);
                return ch;
            }else{
                return -1;
            }
        }
    #    endif
    
        if(kbhit())
            return(getch());
    
    Fabrice Bellard's avatar
    Fabrice Bellard committed
        return -1;
    }
    
    
        return received_nb_signals > 1;
    
    {
        int i;
    
        /* close files */
        for(i=0;i<nb_output_files;i++) {
    
            AVFormatContext *s = output_files[i].ctx;
    
            if (!(s->oformat->flags & AVFMT_NOFILE) && s->pb)
    
            avformat_free_context(s);
    
        for(i=0;i<nb_input_files;i++) {
    
            av_close_input_file(input_files[i].ctx);
    
        for (i = 0; i < nb_input_streams; i++)
            av_dict_free(&input_streams[i].opts);
    
    
        if (vstats_file)
            fclose(vstats_file);
        av_free(vstats_filename);
    
    
        av_freep(&input_streams);
        av_freep(&input_files);
    
        av_free(audio_buf);
        av_free(audio_out);
    
        allocated_audio_buf_size= allocated_audio_out_size= 0;
    
    #if CONFIG_AVFILTER
        avfilter_uninit();
    #endif
    
    
        av_freep(&input_tmp);
    
    
        if (received_sigterm) {
    
            av_log(NULL, AV_LOG_INFO, "Received signal %d: terminating.\n",
                   (int) received_sigterm);
    
        exit(ret); /* not all OS-es handle main() return value */
    
    static void assert_avoptions(AVDictionary *m)
    {
        AVDictionaryEntry *t;
        if ((t = av_dict_get(m, "", NULL, AV_DICT_IGNORE_SUFFIX))) {
    
            av_log(NULL, AV_LOG_FATAL, "Option %s not found.\n", t->key);
    
            exit_program(1);
    
    static void assert_codec_experimental(AVCodecContext *c, int encoder)
    {
        const char *codec_string = encoder ? "encoder" : "decoder";
        AVCodec *codec;
        if (c->codec->capabilities & CODEC_CAP_EXPERIMENTAL &&
            c->strict_std_compliance > FF_COMPLIANCE_EXPERIMENTAL) {
    
            av_log(NULL, AV_LOG_FATAL, "%s '%s' is experimental and might produce bad "
    
                    "results.\nAdd '-strict experimental' if you want to use it.\n",
                    codec_string, c->codec->name);
    
            codec = encoder ? avcodec_find_encoder(c->codec->id) : avcodec_find_decoder(c->codec->id);
    
            if (!(codec->capabilities & CODEC_CAP_EXPERIMENTAL))
    
                av_log(NULL, AV_LOG_FATAL, "Or use the non experimental %s '%s'.\n",
    
                       codec_string, codec->name);
    
            exit_program(1);
    
    static void choose_sample_fmt(AVStream *st, AVCodec *codec)
    {
        if(codec && codec->sample_fmts){
    
            const enum AVSampleFormat *p= codec->sample_fmts;
    
            for(; *p!=-1; p++){
                if(*p == st->codec->sample_fmt)
                    break;
            }
    
                if((codec->capabilities & CODEC_CAP_LOSSLESS) && av_get_sample_fmt_name(st->codec->sample_fmt) > av_get_sample_fmt_name(codec->sample_fmts[0]))
    
                    av_log(NULL, AV_LOG_ERROR, "Conversion will not be lossless.\n");
    
                if(av_get_sample_fmt_name(st->codec->sample_fmt))
    
                av_log(NULL, AV_LOG_WARNING,
                       "Incompatible sample format '%s' for codec '%s', auto-selecting format '%s'\n",
                       av_get_sample_fmt_name(st->codec->sample_fmt),
                       codec->name,
                       av_get_sample_fmt_name(codec->sample_fmts[0]));
    
                st->codec->sample_fmt = codec->sample_fmts[0];
    
    static void choose_sample_rate(AVStream *st, AVCodec *codec)
    {
        if(codec && codec->supported_samplerates){
            const int *p= codec->supported_samplerates;
    
            int best=0;
    
            int best_dist=INT_MAX;
            for(; *p; p++){
                int dist= abs(st->codec->sample_rate - *p);
                if(dist < best_dist){
                    best_dist= dist;
                    best= *p;
                }
            }
    
            if(best_dist){
                av_log(st->codec, AV_LOG_WARNING, "Requested sampling rate unsupported using closest supported (%d)\n", best);
            }
    
    static void choose_pixel_fmt(AVStream *st, AVCodec *codec)
    {
        if(codec && codec->pix_fmts){
            const enum PixelFormat *p= codec->pix_fmts;
    
            if(st->codec->strict_std_compliance <= FF_COMPLIANCE_UNOFFICIAL){
                if(st->codec->codec_id==CODEC_ID_MJPEG){
                    p= (const enum PixelFormat[]){PIX_FMT_YUVJ420P, PIX_FMT_YUVJ422P, PIX_FMT_YUV420P, PIX_FMT_YUV422P, PIX_FMT_NONE};
                }else if(st->codec->codec_id==CODEC_ID_LJPEG){
                    p= (const enum PixelFormat[]){PIX_FMT_YUVJ420P, PIX_FMT_YUVJ422P, PIX_FMT_YUVJ444P, PIX_FMT_YUV420P, PIX_FMT_YUV422P, PIX_FMT_YUV444P, PIX_FMT_BGRA, PIX_FMT_NONE};
                }
            }
    
            for(; *p!=-1; p++){
                if(*p == st->codec->pix_fmt)
                    break;
            }
    
                if(st->codec->pix_fmt != PIX_FMT_NONE)
    
                    av_log(NULL, AV_LOG_WARNING,
                            "Incompatible pixel format '%s' for codec '%s', auto-selecting format '%s'\n",
                            av_pix_fmt_descriptors[st->codec->pix_fmt].name,
                            codec->name,
                            av_pix_fmt_descriptors[codec->pix_fmts[0]].name);
    
                st->codec->pix_fmt = codec->pix_fmts[0];
    
    static double get_sync_ipts(const OutputStream *ost)
    
        const InputStream *ist = ost->sync_ist;
    
        OutputFile *of = &output_files[ost->file_index];
        return (double)(ist->pts - of->start_time)/AV_TIME_BASE;
    
    static void write_frame(AVFormatContext *s, AVPacket *pkt, AVCodecContext *avctx, AVBitStreamFilterContext *bsfc)
    {
    
        while(bsfc){
            AVPacket new_pkt= *pkt;
            int a= av_bitstream_filter_filter(bsfc, avctx, NULL,
                                              &new_pkt.data, &new_pkt.size,
                                              pkt->data, pkt->size,
    
                                              pkt->flags & AV_PKT_FLAG_KEY);
    
                av_free_packet(pkt);
                new_pkt.destruct= av_destruct_packet;
    
            } else if(a<0){
    
                av_log(NULL, AV_LOG_ERROR, "%s failed for stream %d, codec %s",
                       bsfc->filter->name, pkt->stream_index,
                       avctx->codec ? avctx->codec->name : "copy");
    
                print_error("", a);
    
                    exit_program(1);
    
        ret= av_interleaved_write_frame(s, pkt);
        if(ret < 0){
            print_error("av_interleaved_write_frame()", ret);
    
            exit_program(1);
    
    static void do_audio_out(AVFormatContext *s,
    
                             OutputStream *ost,
                             InputStream *ist,
    
    Fabrice Bellard's avatar
    Fabrice Bellard committed
                             unsigned char *buf, int size)
    {
    
        uint8_t *buftmp;
    
        int64_t allocated_for_size= size;
    
        int size_out, frame_bytes, ret, resample_changed;
    
        AVCodecContext *dec= ist->st->codec;
    
        int osize = av_get_bytes_per_sample(enc->sample_fmt);
        int isize = av_get_bytes_per_sample(dec->sample_fmt);
    
        const int coded_bps = av_get_bits_per_sample(enc->codec->id);
    
    
    need_realloc:
        audio_buf_size= (allocated_for_size + isize*dec->channels - 1) / (isize*dec->channels);
    
        audio_buf_size= (audio_buf_size*enc->sample_rate + dec->sample_rate) / dec->sample_rate;
    
    Vitor Sessak's avatar
    Vitor Sessak committed
        audio_buf_size= audio_buf_size*2 + 10000; //safety factors for the deprecated resampling API
    
        audio_buf_size= FFMAX(audio_buf_size, enc->frame_size);
    
        audio_buf_size*= osize*enc->channels;
    
        audio_out_size= FFMAX(audio_buf_size, enc->frame_size * osize * enc->channels);
        if(coded_bps > 8*osize)
            audio_out_size= audio_out_size * coded_bps / (8*osize);
        audio_out_size += FF_MIN_BUFFER_SIZE;
    
        if(audio_out_size > INT_MAX || audio_buf_size > INT_MAX){
    
            av_log(NULL, AV_LOG_FATAL, "Buffer sizes too large\n");
    
            exit_program(1);
    
    Fabrice Bellard's avatar
    Fabrice Bellard committed
    
    
        av_fast_malloc(&audio_buf, &allocated_audio_buf_size, audio_buf_size);
        av_fast_malloc(&audio_out, &allocated_audio_out_size, audio_out_size);
    
        if (!audio_buf || !audio_out){
    
            av_log(NULL, AV_LOG_FATAL, "Out of memory in do_audio_out\n");
    
            exit_program(1);
    
    Michael Niedermayer's avatar
    Michael Niedermayer committed
        if (enc->channels != dec->channels
         || enc->sample_fmt != dec->sample_fmt)
    
        resample_changed = ost->resample_sample_fmt  != dec->sample_fmt ||
                           ost->resample_channels    != dec->channels   ||
                           ost->resample_sample_rate != dec->sample_rate;
    
    
        if ((ost->audio_resample && !ost->swr) || resample_changed || ost->audio_channels_mapped) {
    
            if (resample_changed) {
                av_log(NULL, AV_LOG_INFO, "Input stream #%d.%d frame changed from rate:%d fmt:%s ch:%d to rate:%d fmt:%s ch:%d\n",
    
                       ist->file_index, ist->st->index,
    
                       ost->resample_sample_rate, av_get_sample_fmt_name(ost->resample_sample_fmt), ost->resample_channels,
                       dec->sample_rate, av_get_sample_fmt_name(dec->sample_fmt), dec->channels);
                ost->resample_sample_fmt  = dec->sample_fmt;
                ost->resample_channels    = dec->channels;
                ost->resample_sample_rate = dec->sample_rate;
    
    Michael Niedermayer's avatar
    Michael Niedermayer committed
                swr_free(&ost->swr);
    
            /* if audio_sync_method is >1 the resampler is needed for audio drift compensation */
    
            if (audio_sync_method <= 1 && !ost->audio_channels_mapped &&
    
                ost->resample_sample_fmt  == enc->sample_fmt &&
    
                ost->resample_channels    == enc->channels   &&
                ost->resample_sample_rate == enc->sample_rate) {
    
    Michael Niedermayer's avatar
    Michael Niedermayer committed
                //ost->swr = NULL;
    
    Michael Niedermayer's avatar
    Michael Niedermayer committed
                ost->swr = swr_alloc2(ost->swr,
                                      enc->channel_layout, enc->sample_fmt, enc->sample_rate,
                                      dec->channel_layout, dec->sample_fmt, dec->sample_rate,
    
                                      ost->audio_channels_mapped ? ost->audio_channels_map : NULL,
    
    Michael Niedermayer's avatar
    Michael Niedermayer committed
                                      0, NULL);
    
                av_set_double(ost->swr, "rmvol", ost->rematrix_volume);
    
                if (ost->audio_channels_mapped) {
                    av_set_int(ost->swr, "icl", av_get_default_channel_layout(ost->audio_channels_mapped));
                    av_set_int(ost->swr, "uch", ost->audio_channels_mapped);
                }
    
    Michael Niedermayer's avatar
    Michael Niedermayer committed
                av_set_int(ost->swr, "ich", dec->channels);
                av_set_int(ost->swr, "och", enc->channels);
                if(audio_sync_method>1) av_set_int(ost->swr, "flags", SWR_FLAG_RESAMPLE);
                if(ost->swr && swr_init(ost->swr) < 0){
    
                    av_log(NULL, AV_LOG_FATAL, "swr_init() failed\n");
    
    Michael Niedermayer's avatar
    Michael Niedermayer committed
                    swr_free(&ost->swr);
                }
    
                if (!ost->swr) {
    
                    av_log(NULL, AV_LOG_FATAL, "Can not resample %d channels @ %d Hz to %d channels @ %d Hz\n",
    
    Stefano Sabatini's avatar
    Stefano Sabatini committed
                            dec->channels, dec->sample_rate,
                            enc->channels, enc->sample_rate);
    
                    exit_program(1);
    
    Stefano Sabatini's avatar
    Stefano Sabatini committed
                }
    
    Michael Niedermayer's avatar
    Michael Niedermayer committed
        av_assert0(ost->audio_resample || dec->sample_fmt==enc->sample_fmt);
    
        if(audio_sync_method){
    
            double delta = get_sync_ipts(ost) * enc->sample_rate - ost->sync_opts
    
                    - av_fifo_size(ost->fifo)/(enc->channels * 2);
            double idelta= delta*dec->sample_rate / enc->sample_rate;
    
            int byte_delta= ((int)idelta)*2*dec->channels;
    
            //FIXME resample delay
            if(fabs(delta) > 50){
    
                if(ist->is_start || fabs(delta) > audio_drift_threshold*enc->sample_rate){
    
                    if(byte_delta < 0){
    
                        size += byte_delta;
                        buf  -= byte_delta;
    
                        av_log(NULL, AV_LOG_VERBOSE, "discarding %d audio samples\n", (int)-delta);
    
                        if(!size)
                            return;
                        ist->is_start=0;
                    }else{
                        input_tmp= av_realloc(input_tmp, byte_delta + size);
    
    
                        if(byte_delta > allocated_for_size - size){
                            allocated_for_size= byte_delta + (int64_t)size;
                            goto need_realloc;
                        }
                        ist->is_start=0;
    
    
                        memset(input_tmp, 0, byte_delta);
                        memcpy(input_tmp + byte_delta, buf, size);
                        buf= input_tmp;
                        size += byte_delta;
    
                        av_log(NULL, AV_LOG_VERBOSE, "adding %d audio samples of silence\n", (int)delta);
    
                    }
                }else if(audio_sync_method>1){
    
                    int comp= av_clip(delta, -audio_sync_method, audio_sync_method);
    
    Michael Niedermayer's avatar
    Michael Niedermayer committed
                    av_assert0(ost->audio_resample);
    
                    av_log(NULL, AV_LOG_VERBOSE, "compensating audio timestamp drift:%f compensation:%d in:%d\n",
                           delta, comp, enc->sample_rate);
    
    //                fprintf(stderr, "drift:%f len:%d opts:%"PRId64" ipts:%"PRId64" fifo:%d\n", delta, -1, ost->sync_opts, (int64_t)(get_sync_ipts(ost) * enc->sample_rate), av_fifo_size(ost->fifo)/(ost->st->codec->channels * 2));
    
    Michael Niedermayer's avatar
    Michael Niedermayer committed
                    swr_compensate(ost->swr, comp, enc->sample_rate);
    
            ost->sync_opts= lrintf(get_sync_ipts(ost) * enc->sample_rate)
    
                            - av_fifo_size(ost->fifo)/(enc->channels * 2); //FIXME wrong
    
    Fabrice Bellard's avatar
    Fabrice Bellard committed
    
        if (ost->audio_resample) {
            buftmp = audio_buf;
    
    Michael Niedermayer's avatar
    Michael Niedermayer committed
            size_out = swr_convert(ost->swr, (      uint8_t*[]){buftmp}, audio_buf_size / (enc->channels * osize),
                                             (const uint8_t*[]){buf   }, size / (dec->channels * isize));
    
            size_out = size_out * enc->channels * osize;
    
    Fabrice Bellard's avatar
    Fabrice Bellard committed
        } else {
            buftmp = buf;
            size_out = size;
        }
    
    
    Michael Niedermayer's avatar
    Michael Niedermayer committed
        av_assert0(ost->audio_resample || dec->sample_fmt==enc->sample_fmt);
    
    Fabrice Bellard's avatar
    Fabrice Bellard committed
        /* now encode as many frames as possible */
    
        if (enc->frame_size > 1) {
    
    Fabrice Bellard's avatar
    Fabrice Bellard committed
            /* output resampled raw samples */
    
            if (av_fifo_realloc2(ost->fifo, av_fifo_size(ost->fifo) + size_out) < 0) {
    
                av_log(NULL, AV_LOG_FATAL, "av_fifo_realloc2() failed\n");
    
                exit_program(1);
    
            av_fifo_generic_write(ost->fifo, buftmp, size_out, NULL);
    
    Fabrice Bellard's avatar
    Fabrice Bellard committed
    
    
            frame_bytes = enc->frame_size * osize * enc->channels;
    
            while (av_fifo_size(ost->fifo) >= frame_bytes) {
    
                AVPacket pkt;
                av_init_packet(&pkt);
    
    
                av_fifo_generic_read(ost->fifo, audio_buf, frame_bytes, NULL);
    
                //FIXME pass ost->sync_opts as AVFrame.pts in avcodec_encode_audio()
    
    
                ret = avcodec_encode_audio(enc, audio_out, audio_out_size,
    
                    av_log(NULL, AV_LOG_FATAL, "Audio encoding failed\n");
    
                    exit_program(1);