Skip to content
Snippets Groups Projects
ffmpeg.c 215 KiB
Newer Older
  • Learn to ignore specific revisions
  •         return codec;
        } else
            return avcodec_find_decoder(st->codec->codec_id);
    
    /**
     * Add all the streams from the given input file to the global
     * list of input streams.
     */
    
    static void add_input_streams(OptionsContext *o, AVFormatContext *ic)
    
        char *next, *codec_tag = NULL;
    
    
        for (i = 0; i < ic->nb_streams; i++) {
            AVStream *st = ic->streams[i];
            AVCodecContext *dec = st->codec;
    
            InputStream *ist = av_mallocz(sizeof(*ist));
    
            if (!ist)
                exit_program(1);
    
    
            input_streams = grow_array(input_streams, sizeof(*input_streams), &nb_input_streams, nb_input_streams + 1);
    
            input_streams[nb_input_streams - 1] = ist;
    
    
            ist->st = st;
            ist->file_index = nb_input_files;
            ist->discard = 1;
    
            st->discard  = AVDISCARD_ALL;
    
            ist->opts = filter_codec_opts(codec_opts, choose_decoder(o, ic, st), ic, st);
    
            ist->ts_scale = 1.0;
            MATCH_PER_STREAM_OPT(ts_scale, dbl, ist->ts_scale, ic, st);
    
            MATCH_PER_STREAM_OPT(codec_tags, str, codec_tag, ic, st);
            if (codec_tag) {
                uint32_t tag = strtol(codec_tag, &next, 0);
                if (*next)
                    tag = AV_RL32(codec_tag);
                st->codec->codec_tag = tag;
            }
    
    
            ist->dec = choose_decoder(o, ic, st);
    
    
            switch (dec->codec_type) {
            case AVMEDIA_TYPE_VIDEO:
                if(!ist->dec)
                    ist->dec = avcodec_find_decoder(dec->codec_id);
                if (dec->lowres) {
                    dec->flags |= CODEC_FLAG_EMU_EDGE;
                }
    
    
                ist->resample_height  = dec->height;
                ist->resample_width   = dec->width;
                ist->resample_pix_fmt = dec->pix_fmt;
    
    
            case AVMEDIA_TYPE_AUDIO:
    
                guess_input_channel_layout(ist);
    
                ist->resample_sample_fmt     = dec->sample_fmt;
                ist->resample_sample_rate    = dec->sample_rate;
                ist->resample_channels       = dec->channels;
                ist->resample_channel_layout = dec->channel_layout;
    
                break;
    
            case AVMEDIA_TYPE_DATA:
            case AVMEDIA_TYPE_SUBTITLE:
                if(!ist->dec)
                    ist->dec = avcodec_find_decoder(dec->codec_id);
                break;
            case AVMEDIA_TYPE_ATTACHMENT:
            case AVMEDIA_TYPE_UNKNOWN:
                break;
            default:
                abort();
            }
        }
    }
    
    
    static void assert_file_overwrite(const char *filename)
    
        if ((!file_overwrite || no_file_overwrite) &&
    
            (strchr(filename, ':') == NULL || filename[1] == ':' ||
             av_strstart(filename, "file:", NULL))) {
            if (avio_check(filename, 0) == 0) {
    
                if (!using_stdin && (!no_file_overwrite || file_overwrite)) {
    
                    fprintf(stderr,"File '%s' already exists. Overwrite ? [y/N] ", filename);
                    fflush(stderr);
                    term_exit();
    
                    signal(SIGINT, SIG_DFL);
    
                    if (!read_yesno()) {
    
                        av_log(NULL, AV_LOG_FATAL, "Not overwriting - exiting\n");
    
                        exit_program(1);
                    }
                    term_init();
                }
                else {
    
                    av_log(NULL, AV_LOG_FATAL, "File '%s' already exists. Exiting.\n", filename);
    
    static void dump_attachment(AVStream *st, const char *filename)
    
        int ret;
        AVIOContext *out = NULL;
        AVDictionaryEntry *e;
    
        if (!st->codec->extradata_size) {
            av_log(NULL, AV_LOG_WARNING, "No extradata to dump in stream #%d:%d.\n",
                   nb_input_files - 1, st->index);
            return;
        }
        if (!*filename && (e = av_dict_get(st->metadata, "filename", NULL, 0)))
            filename = e->value;
        if (!*filename) {
            av_log(NULL, AV_LOG_FATAL, "No filename specified and no 'filename' tag"
                   "in stream #%d:%d.\n", nb_input_files - 1, st->index);
    
    
        assert_file_overwrite(filename);
    
    
        if ((ret = avio_open2(&out, filename, AVIO_FLAG_WRITE, &int_cb, NULL)) < 0) {
    
            av_log(NULL, AV_LOG_FATAL, "Could not open file %s for writing.\n",
                   filename);
    
    
        avio_write(out, st->codec->extradata, st->codec->extradata_size);
        avio_flush(out);
        avio_close(out);
    
    static int opt_input_file(OptionsContext *o, const char *opt, const char *filename)
    
    Fabrice Bellard's avatar
    Fabrice Bellard committed
    {
        AVFormatContext *ic;
    
        AVInputFormat *file_iformat = NULL;
    
        int err, i, ret;
    
        int64_t timestamp;
    
        uint8_t buf[128];
    
        AVDictionary **opts;
        int orig_nb_streams;                     // number of streams before avformat_find_stream_info
    
    Fabrice Bellard's avatar
    Fabrice Bellard committed
    
    
        if (o->format) {
            if (!(file_iformat = av_find_input_format(o->format))) {
    
                av_log(NULL, AV_LOG_FATAL, "Unknown input format: '%s'\n", o->format);
    
                exit_program(1);
    
        if (!strcmp(filename, "-"))
            filename = "pipe:";
    
    
        using_stdin |= !strncmp(filename, "pipe:", 5) ||
    
    Fabrice Bellard's avatar
    Fabrice Bellard committed
        /* get default parameters from command line */
    
        if (!ic) {
            print_error(filename, AVERROR(ENOMEM));
    
            exit_program(1);
    
        if (o->nb_audio_sample_rate) {
            snprintf(buf, sizeof(buf), "%d", o->audio_sample_rate[o->nb_audio_sample_rate - 1].u.i);
    
            av_dict_set(&format_opts, "sample_rate", buf, 0);
        }
    
        if (o->nb_audio_channels) {
    
            /* because we set audio_channels based on both the "ac" and
             * "channel_layout" options, we need to check that the specified
             * demuxer actually has the "channels" option before setting it */
            if (file_iformat && file_iformat->priv_class &&
                av_opt_find(&file_iformat->priv_class, "channels", NULL, 0,
                            AV_OPT_SEARCH_FAKE_OBJ)) {
                snprintf(buf, sizeof(buf), "%d",
                         o->audio_channels[o->nb_audio_channels - 1].u.i);
                av_dict_set(&format_opts, "channels", buf, 0);
            }
    
        if (o->nb_frame_rates) {
            av_dict_set(&format_opts, "framerate", o->frame_rates[o->nb_frame_rates - 1].u.str, 0);
    
        if (o->nb_frame_sizes) {
            av_dict_set(&format_opts, "video_size", o->frame_sizes[o->nb_frame_sizes - 1].u.str, 0);
    
        if (o->nb_frame_pix_fmts)
            av_dict_set(&format_opts, "pixel_format", o->frame_pix_fmts[o->nb_frame_pix_fmts - 1].u.str, 0);
    
        ic->video_codec_id   = video_codec_name ?
            find_codec_or_die(video_codec_name   , AVMEDIA_TYPE_VIDEO   , 0)->id : CODEC_ID_NONE;
        ic->audio_codec_id   = audio_codec_name ?
            find_codec_or_die(audio_codec_name   , AVMEDIA_TYPE_AUDIO   , 0)->id : CODEC_ID_NONE;
        ic->subtitle_codec_id= subtitle_codec_name ?
            find_codec_or_die(subtitle_codec_name, AVMEDIA_TYPE_SUBTITLE, 0)->id : CODEC_ID_NONE;
    
        ic->flags |= AVFMT_FLAG_NONBLOCK;
    
        ic->interrupt_callback = int_cb;
    
        /* open the input file with generic avformat function */
    
        err = avformat_open_input(&ic, filename, file_iformat, &format_opts);
    
    Fabrice Bellard's avatar
    Fabrice Bellard committed
        if (err < 0) {
    
            exit_program(1);
    
    Fabrice Bellard's avatar
    Fabrice Bellard committed
        }
    
        assert_avoptions(format_opts);
    
    
        /* apply forced codec ids */
        for (i = 0; i < ic->nb_streams; i++)
    
            choose_decoder(o, ic, ic->streams[i]);
    
        /* Set AVCodecContext options for avformat_find_stream_info */
    
        opts = setup_find_stream_info_opts(ic, codec_opts);
    
        orig_nb_streams = ic->nb_streams;
    
        /* If not enough info to get the stream parameters, we decode the
           first frames to get it. (used in mpeg case for example) */
    
        ret = avformat_find_stream_info(ic, opts);
    
        if (ret < 0) {
            av_log(NULL, AV_LOG_FATAL, "%s: could not find codec parameters\n", filename);
    
            avformat_close_input(&ic);
    
            exit_program(1);
    
    Fabrice Bellard's avatar
    Fabrice Bellard committed
        }
    
    
        timestamp = o->start_time;
    
        /* add the stream start time */
        if (ic->start_time != AV_NOPTS_VALUE)
            timestamp += ic->start_time;
    
    
        /* if seeking requested, we execute it */
    
        if (o->start_time != 0) {
    
            ret = av_seek_frame(ic, -1, timestamp, AVSEEK_FLAG_BACKWARD);
    
                av_log(NULL, AV_LOG_WARNING, "%s: could not seek to position %0.3f\n",
                       filename, (double)timestamp / AV_TIME_BASE);
    
    Fabrice Bellard's avatar
    Fabrice Bellard committed
        /* update the current parameters so that they match the one of the input stream */
    
        add_input_streams(o, ic);
    
    Fabrice Bellard's avatar
    Fabrice Bellard committed
        /* dump the file content */
    
        av_dump_format(ic, nb_input_files, filename, 0);
    
        input_files = grow_array(input_files, sizeof(*input_files), &nb_input_files, nb_input_files + 1);
    
        if (!(input_files[nb_input_files - 1] = av_mallocz(sizeof(*input_files[0]))))
            exit_program(1);
    
        input_files[nb_input_files - 1]->ctx        = ic;
        input_files[nb_input_files - 1]->ist_index  = nb_input_streams - ic->nb_streams;
        input_files[nb_input_files - 1]->ts_offset  = o->input_ts_offset - (copy_ts ? 0 : timestamp);
        input_files[nb_input_files - 1]->nb_streams = ic->nb_streams;
        input_files[nb_input_files - 1]->rate_emu   = o->rate_emu;
    
        for (i = 0; i < o->nb_dump_attachment; i++) {
            int j;
    
            for (j = 0; j < ic->nb_streams; j++) {
                AVStream *st = ic->streams[j];
    
                if (check_stream_specifier(ic, st, o->dump_attachment[i].specifier) == 1)
                    dump_attachment(st, o->dump_attachment[i].u.str);
            }
        }
    
        for (i = 0; i < orig_nb_streams; i++)
            av_dict_free(&opts[i]);
        av_freep(&opts);
    
        return 0;
    
    Fabrice Bellard's avatar
    Fabrice Bellard committed
    }
    
    
    static void parse_forced_key_frames(char *kf, OutputStream *ost)
    
    {
        char *p;
        int n = 1, i;
    
        for (p = kf; *p; p++)
            if (*p == ',')
                n++;
        ost->forced_kf_count = n;
    
        ost->forced_kf_pts   = av_malloc(sizeof(*ost->forced_kf_pts) * n);
    
        if (!ost->forced_kf_pts) {
            av_log(NULL, AV_LOG_FATAL, "Could not allocate forced key frames array.\n");
            exit_program(1);
        }
    
        for (i = 0; i < n; i++) {
    
            char *next = strchr(p, ',');
            if (next) *next++ = 0;
    
            ost->forced_kf_pts[i] = parse_time_or_die("force_key_frames", p, 1);
    
    static uint8_t *get_line(AVIOContext *s)
    {
        AVIOContext *line;
        uint8_t *buf;
        char c;
    
        if (avio_open_dyn_buf(&line) < 0) {
            av_log(NULL, AV_LOG_FATAL, "Could not alloc buffer for reading preset.\n");
            exit_program(1);
        }
    
        while ((c = avio_r8(s)) && c != '\n')
            avio_w8(line, c);
        avio_w8(line, 0);
        avio_close_dyn_buf(line, &buf);
    
        return buf;
    }
    
    static int get_preset_file_2(const char *preset_name, const char *codec_name, AVIOContext **s)
    {
        int i, ret = 1;
        char filename[1000];
        const char *base[3] = { getenv("AVCONV_DATADIR"),
                                getenv("HOME"),
                                AVCONV_DATADIR,
                                };
    
        for (i = 0; i < FF_ARRAY_ELEMS(base) && ret; i++) {
            if (!base[i])
                continue;
            if (codec_name) {
                snprintf(filename, sizeof(filename), "%s%s/%s-%s.avpreset", base[i],
                         i != 1 ? "" : "/.avconv", codec_name, preset_name);
    
                ret = avio_open2(s, filename, AVIO_FLAG_READ, &int_cb, NULL);
    
            }
            if (ret) {
                snprintf(filename, sizeof(filename), "%s%s/%s.avpreset", base[i],
                         i != 1 ? "" : "/.avconv", preset_name);
    
                ret = avio_open2(s, filename, AVIO_FLAG_READ, &int_cb, NULL);
    
    static void choose_encoder(OptionsContext *o, AVFormatContext *s, OutputStream *ost)
    {
        char *codec_name = NULL;
    
        MATCH_PER_STREAM_OPT(codec_names, str, codec_name, s, ost->st);
        if (!codec_name) {
            ost->st->codec->codec_id = av_guess_codec(s->oformat, NULL, s->filename,
                                                      NULL, ost->st->codec->codec_type);
            ost->enc = avcodec_find_encoder(ost->st->codec->codec_id);
        } else if (!strcmp(codec_name, "copy"))
            ost->stream_copy = 1;
        else {
            ost->enc = find_codec_or_die(codec_name, ost->st->codec->codec_type, 1);
            ost->st->codec->codec_id = ost->enc->id;
        }
    }
    
    
    static OutputStream *new_output_stream(OptionsContext *o, AVFormatContext *oc, enum AVMediaType type, int source_index)
    
    {
        OutputStream *ost;
    
        AVStream *st = avformat_new_stream(oc, NULL);
    
        int idx      = oc->nb_streams - 1, ret = 0;
    
        char *bsf = NULL, *next, *codec_tag = NULL;
    
        AVBitStreamFilterContext *bsfc, *bsfc_prev = NULL;
    
        char *buf = NULL, *arg = NULL, *preset = NULL;
        AVIOContext *s = NULL;
    
            av_log(NULL, AV_LOG_FATAL, "Could not alloc stream.\n");
    
            exit_program(1);
        }
    
    
        if (oc->nb_streams - 1 < o->nb_streamid_map)
            st->id = o->streamid_map[oc->nb_streams - 1];
    
    
        output_streams = grow_array(output_streams, sizeof(*output_streams), &nb_output_streams,
                                    nb_output_streams + 1);
    
        if (!(ost = av_mallocz(sizeof(*ost))))
            exit_program(1);
        output_streams[nb_output_streams - 1] = ost;
    
    
        ost->file_index = nb_output_files;
    
        ost->index      = idx;
        ost->st         = st;
    
        st->codec->codec_type = type;
    
        choose_encoder(o, oc, ost);
    
        if (ost->enc) {
    
            ost->opts  = filter_codec_opts(codec_opts, ost->enc, oc, st);
    
        }
    
        avcodec_get_context_defaults3(st->codec, ost->enc);
        st->codec->codec_type = type; // XXX hack, avcodec_get_context_defaults2() sets type to unknown for stream copy
    
    
        MATCH_PER_STREAM_OPT(presets, str, preset, oc, st);
        if (preset && (!(ret = get_preset_file_2(preset, ost->enc->name, &s)))) {
            do  {
                buf = get_line(s);
                if (!buf[0] || buf[0] == '#') {
                    av_free(buf);
                    continue;
                }
                if (!(arg = strchr(buf, '='))) {
                    av_log(NULL, AV_LOG_FATAL, "Invalid line found in the preset file.\n");
                    exit_program(1);
                }
                *arg++ = 0;
                av_dict_set(&ost->opts, buf, arg, AV_DICT_DONT_OVERWRITE);
                av_free(buf);
            } while (!s->eof_reached);
            avio_close(s);
        }
        if (ret) {
            av_log(NULL, AV_LOG_FATAL,
                   "Preset %s specified for stream %d:%d, but could not be opened.\n",
                   preset, ost->file_index, ost->index);
            exit_program(1);
        }
    
    
        ost->max_frames = INT64_MAX;
        MATCH_PER_STREAM_OPT(max_frames, i64, ost->max_frames, oc, st);
    
    
        MATCH_PER_STREAM_OPT(bitstream_filters, str, bsf, oc, st);
        while (bsf) {
            if (next = strchr(bsf, ','))
                *next++ = 0;
            if (!(bsfc = av_bitstream_filter_init(bsf))) {
    
                av_log(NULL, AV_LOG_FATAL, "Unknown bitstream filter %s\n", bsf);
    
            if (bsfc_prev)
                bsfc_prev->next = bsfc;
            else
                ost->bitstream_filters = bsfc;
    
            bsfc_prev = bsfc;
            bsf       = next;
    
        MATCH_PER_STREAM_OPT(codec_tags, str, codec_tag, oc, st);
        if (codec_tag) {
            uint32_t tag = strtol(codec_tag, &next, 0);
            if (*next)
                tag = AV_RL32(codec_tag);
            st->codec->codec_tag = tag;
    
        MATCH_PER_STREAM_OPT(qscale, dbl, qscale, oc, st);
        if (qscale >= 0 || same_quant) {
            st->codec->flags |= CODEC_FLAG_QSCALE;
            st->codec->global_quality = FF_QP2LAMBDA * qscale;
        }
    
        if (oc->oformat->flags & AVFMT_GLOBALHEADER)
            st->codec->flags |= CODEC_FLAG_GLOBAL_HEADER;
    
        av_opt_get_int(sws_opts, "sws_flags", 0, &ost->sws_flags);
    
        av_opt_get_int   (swr_opts, "dither_method", 0, &ost->swr_dither_method);
        av_opt_get_double(swr_opts, "dither_scale" , 0, &ost->swr_dither_scale);
    
    
        ost->source_index = source_index;
        if (source_index >= 0) {
    
            ost->sync_ist = input_streams[source_index];
            input_streams[source_index]->discard = 0;
            input_streams[source_index]->st->discard = AVDISCARD_NONE;
    
        ost->pix_fmts[0] = ost->pix_fmts[1] = PIX_FMT_NONE;
    
    
    static void parse_matrix_coeffs(uint16_t *dest, const char *str)
    {
        int i;
        const char *p = str;
    
                break;
            p = strchr(p, ',');
    
                av_log(NULL, AV_LOG_FATAL, "Syntax error in matrix \"%s\" at coeff %d\n", str, i);
    
    static OutputStream *new_video_stream(OptionsContext *o, AVFormatContext *oc, int source_index)
    
    Fabrice Bellard's avatar
    Fabrice Bellard committed
    {
        AVStream *st;
    
        OutputStream *ost;
    
        ost = new_output_stream(o, oc, AVMEDIA_TYPE_VIDEO, source_index);
    
        if (!ost->stream_copy) {
    
            const char *p = NULL;
            char *forced_key_frames = NULL, *frame_rate = NULL, *frame_size = NULL;
            char *frame_aspect_ratio = NULL, *frame_pix_fmt = NULL;
            char *intra_matrix = NULL, *inter_matrix = NULL, *filters = NULL;
    
    
            MATCH_PER_STREAM_OPT(frame_rates, str, frame_rate, oc, st);
            if (frame_rate && av_parse_video_rate(&ost->frame_rate, frame_rate) < 0) {
    
                av_log(NULL, AV_LOG_FATAL, "Invalid framerate value: %s\n", frame_rate);
    
            MATCH_PER_STREAM_OPT(frame_sizes, str, frame_size, oc, st);
            if (frame_size && av_parse_video_size(&video_enc->width, &video_enc->height, frame_size) < 0) {
    
                av_log(NULL, AV_LOG_FATAL, "Invalid frame size: %s.\n", frame_size);
    
            MATCH_PER_STREAM_OPT(frame_aspect_ratios, str, frame_aspect_ratio, oc, st);
    
            if (frame_aspect_ratio) {
                AVRational q;
                if (av_parse_ratio(&q, frame_aspect_ratio, 255, 0, NULL) < 0 ||
                    q.num <= 0 || q.den <= 0) {
                    av_log(NULL, AV_LOG_FATAL, "Invalid aspect ratio: %s\n", frame_aspect_ratio);
                    exit_program(1);
                }
                ost->frame_aspect_ratio = av_q2d(q);
            }
    
            video_enc->bits_per_raw_sample = frame_bits_per_raw_sample;
    
            MATCH_PER_STREAM_OPT(frame_pix_fmts, str, frame_pix_fmt, oc, st);
    
            if (frame_pix_fmt && *frame_pix_fmt == '+') {
                ost->keep_pix_fmt = 1;
                if (!*++frame_pix_fmt)
                    frame_pix_fmt = NULL;
            }
    
            if (frame_pix_fmt && (video_enc->pix_fmt = av_get_pix_fmt(frame_pix_fmt)) == PIX_FMT_NONE) {
    
                av_log(NULL, AV_LOG_FATAL, "Unknown pixel format requested: %s.\n", frame_pix_fmt);
    
            st->sample_aspect_ratio = video_enc->sample_aspect_ratio;
    
            MATCH_PER_STREAM_OPT(intra_matrices, str, intra_matrix, oc, st);
            if (intra_matrix) {
                if (!(video_enc->intra_matrix = av_mallocz(sizeof(*video_enc->intra_matrix) * 64))) {
    
                    av_log(NULL, AV_LOG_FATAL, "Could not allocate memory for intra matrix.\n");
    
                    exit_program(1);
                }
                parse_matrix_coeffs(video_enc->intra_matrix, intra_matrix);
            }
            MATCH_PER_STREAM_OPT(inter_matrices, str, inter_matrix, oc, st);
            if (inter_matrix) {
                if (!(video_enc->inter_matrix = av_mallocz(sizeof(*video_enc->inter_matrix) * 64))) {
    
                    av_log(NULL, AV_LOG_FATAL, "Could not allocate memory for inter matrix.\n");
    
                    exit_program(1);
                }
                parse_matrix_coeffs(video_enc->inter_matrix, inter_matrix);
    
            MATCH_PER_STREAM_OPT(rc_overrides, str, p, oc, st);
    
            for (i = 0; p; i++) {
    
                int e = sscanf(p, "%d,%d,%d", &start, &end, &q);
                if (e != 3) {
    
                    av_log(NULL, AV_LOG_FATAL, "error parsing rc_override\n");
    
                    exit_program(1);
    
                /* FIXME realloc failure */
    
                video_enc->rc_override =
    
                    av_realloc(video_enc->rc_override,
    
                               sizeof(RcOverride) * (i + 1));
                video_enc->rc_override[i].start_frame = start;
                video_enc->rc_override[i].end_frame   = end;
                if (q > 0) {
                    video_enc->rc_override[i].qscale         = q;
                    video_enc->rc_override[i].quality_factor = 1.0;
    
                else {
                    video_enc->rc_override[i].qscale         = 0;
                    video_enc->rc_override[i].quality_factor = -q/100.0;
    
                p = strchr(p, '/');
                if (p) p++;
    
            video_enc->rc_override_count = i;
    
            if (!video_enc->rc_initial_buffer_occupancy)
    
                video_enc->rc_initial_buffer_occupancy = video_enc->rc_buffer_size * 3 / 4;
            video_enc->intra_dc_precision = intra_dc_precision - 8;
    
            if (do_psnr)
                video_enc->flags|= CODEC_FLAG_PSNR;
    
    
                if (do_pass & 1) {
    
                }
                if (do_pass & 2) {
    
            MATCH_PER_STREAM_OPT(forced_key_frames, str, forced_key_frames, oc, st);
    
            if (forced_key_frames)
    
                parse_forced_key_frames(forced_key_frames, ost);
    
            MATCH_PER_STREAM_OPT(force_fps, i, ost->force_fps, oc, st);
    
            ost->top_field_first = -1;
            MATCH_PER_STREAM_OPT(top_field_first, i, ost->top_field_first, oc, st);
    
    
            MATCH_PER_STREAM_OPT(filters, str, filters, oc, st);
            if (filters)
                ost->avfilter = av_strdup(filters);
    
        } else {
            MATCH_PER_STREAM_OPT(copy_initial_nonkeyframes, i, ost->copy_initial_nonkeyframes, oc ,st);
    
        return ost;
    
    static OutputStream *new_audio_stream(OptionsContext *o, AVFormatContext *oc, int source_index)
    
        OutputStream *ost;
    
        ost = new_output_stream(o, oc, AVMEDIA_TYPE_AUDIO, source_index);
    
        audio_enc->codec_type = AVMEDIA_TYPE_AUDIO;
    
        if (!ost->stream_copy) {
    
    Michael Niedermayer's avatar
    Michael Niedermayer committed
            char *sample_fmt = NULL, *filters = NULL;
    
            MATCH_PER_STREAM_OPT(audio_channels, i, audio_enc->channels, oc, st);
    
    
            MATCH_PER_STREAM_OPT(sample_fmts, str, sample_fmt, oc, st);
            if (sample_fmt &&
                (audio_enc->sample_fmt = av_get_sample_fmt(sample_fmt)) == AV_SAMPLE_FMT_NONE) {
    
                av_log(NULL, AV_LOG_FATAL, "Invalid sample format '%s'\n", sample_fmt);
    
    
            MATCH_PER_STREAM_OPT(audio_sample_rate, i, audio_enc->sample_rate, oc, st);
    
            MATCH_PER_STREAM_OPT(filters, str, filters, oc, st);
            if (filters)
                ost->avfilter = av_strdup(filters);
    
    
            /* check for channel mapping for this audio stream */
            for (n = 0; n < o->nb_audio_channel_maps; n++) {
                AudioChannelMap *map = &o->audio_channel_maps[n];
                InputStream *ist = input_streams[ost->source_index];
                if ((map->channel_idx == -1 || (ist->file_index == map->file_idx && ist->st->index == map->stream_idx)) &&
                    (map->ofile_idx   == -1 || ost->file_index == map->ofile_idx) &&
                    (map->ostream_idx == -1 || ost->st->index  == map->ostream_idx)) {
                    if (ost->audio_channels_mapped < FF_ARRAY_ELEMS(ost->audio_channels_map))
                        ost->audio_channels_map[ost->audio_channels_mapped++] = map->channel_idx;
                    else
                        av_log(NULL, AV_LOG_FATAL, "Max channel mapping for output %d.%d reached\n",
                               ost->file_index, ost->st->index);
                }
            }
    
        return ost;
    
    static OutputStream *new_data_stream(OptionsContext *o, AVFormatContext *oc, int source_index)
    
        ost = new_output_stream(o, oc, AVMEDIA_TYPE_DATA, source_index);
    
        if (!ost->stream_copy) {
    
            av_log(NULL, AV_LOG_FATAL, "Data stream encoding not supported yet (only streamcopy)\n");
    
            exit_program(1);
    
    static OutputStream *new_attachment_stream(OptionsContext *o, AVFormatContext *oc, int source_index)
    
        OutputStream *ost = new_output_stream(o, oc, AVMEDIA_TYPE_ATTACHMENT, source_index);
    
        ost->stream_copy = 1;
    
        return ost;
    
    static OutputStream *new_subtitle_stream(OptionsContext *o, AVFormatContext *oc, int source_index)
    
        OutputStream *ost;
    
        ost = new_output_stream(o, oc, AVMEDIA_TYPE_SUBTITLE, source_index);
    
        subtitle_enc->codec_type = AVMEDIA_TYPE_SUBTITLE;
    
        MATCH_PER_STREAM_OPT(copy_initial_nonkeyframes, i, ost->copy_initial_nonkeyframes, oc, st);
    
    
        return ost;
    
    /* arg format is "output-stream-index:streamid-value". */
    
    static int opt_streamid(OptionsContext *o, const char *opt, const char *arg)
    
        av_strlcpy(idx_str, arg, sizeof(idx_str));
    
            av_log(NULL, AV_LOG_FATAL,
                   "Invalid value '%s' for option '%s', required syntax is 'index:value'\n",
                   arg, opt);
    
            exit_program(1);
    
        idx = parse_number_or_die(opt, idx_str, OPT_INT, 0, MAX_STREAMS-1);
    
        o->streamid_map = grow_array(o->streamid_map, sizeof(*o->streamid_map), &o->nb_streamid_map, idx+1);
        o->streamid_map[idx] = parse_number_or_die(opt, p, OPT_INT, 0, INT_MAX);
    
    static int copy_chapters(InputFile *ifile, OutputFile *ofile, int copy_metadata)
    
        AVFormatContext *is = ifile->ctx;
        AVFormatContext *os = ofile->ctx;
    
        int i;
    
        for (i = 0; i < is->nb_chapters; i++) {
            AVChapter *in_ch = is->chapters[i], *out_ch;
    
            int64_t ts_off   = av_rescale_q(ofile->start_time - ifile->ts_offset,
    
                                           AV_TIME_BASE_Q, in_ch->time_base);
    
            int64_t rt       = (ofile->recording_time == INT64_MAX) ? INT64_MAX :
                               av_rescale_q(ofile->recording_time, AV_TIME_BASE_Q, in_ch->time_base);
    
    
    
            if (in_ch->end < ts_off)
                continue;
            if (rt != INT64_MAX && in_ch->start > rt + ts_off)
                break;
    
            out_ch = av_mallocz(sizeof(AVChapter));
            if (!out_ch)
                return AVERROR(ENOMEM);
    
            out_ch->id        = in_ch->id;
            out_ch->time_base = in_ch->time_base;
            out_ch->start     = FFMAX(0,  in_ch->start - ts_off);
            out_ch->end       = FFMIN(rt, in_ch->end   - ts_off);
    
    
                av_dict_copy(&out_ch->metadata, in_ch->metadata, 0);
    
            os->nb_chapters++;
    
            os->chapters = av_realloc_f(os->chapters, os->nb_chapters, sizeof(AVChapter));
    
            if (!os->chapters)
                return AVERROR(ENOMEM);
            os->chapters[os->nb_chapters - 1] = out_ch;
        }
        return 0;
    }
    
    
    static int read_ffserver_streams(OptionsContext *o, AVFormatContext *s, const char *filename)
    {
        int i, err;
    
        AVFormatContext *ic = avformat_alloc_context();
    
        ic->interrupt_callback = int_cb;
    
        err = avformat_open_input(&ic, filename, NULL, NULL);
        if (err < 0)
            return err;
        /* copy stream format */
        for(i=0;i<ic->nb_streams;i++) {
            AVStream *st;
            OutputStream *ost;
            AVCodec *codec;
    
            AVCodecContext *avctx;
    
    
            codec = avcodec_find_encoder(ic->streams[i]->codec->codec_id);
    
            ost   = new_output_stream(o, s, codec->type, -1);
    
    
            // FIXME: a more elegant solution is needed
            memcpy(st, ic->streams[i], sizeof(AVStream));
    
            st->info = av_malloc(sizeof(*st->info));
            memcpy(st->info, ic->streams[i]->info, sizeof(*st->info));
    
            avcodec_copy_context(st->codec, ic->streams[i]->codec);
    
    
            if (st->codec->codec_type == AVMEDIA_TYPE_AUDIO && !ost->stream_copy)
    
                choose_sample_fmt(st, codec);
    
            else if (st->codec->codec_type == AVMEDIA_TYPE_VIDEO && !ost->stream_copy)
    
                choose_pixel_fmt(st, codec, st->codec->pix_fmt);
    
    static void init_output_filter(OutputFilter *ofilter, OptionsContext *o,
                                   AVFormatContext *oc)
    {
        OutputStream *ost;
    
        if (ofilter->out_tmp->filter_ctx->output_pads[ofilter->out_tmp->pad_idx].type != AVMEDIA_TYPE_VIDEO) {
            av_log(NULL, AV_LOG_FATAL, "Only video filters are supported currently.\n");
            exit_program(1);
        }
    
    
        ost               = new_video_stream(o, oc, -1);
    
        ost->source_index = -1;
        ost->filter       = ofilter;
    
        ofilter->ost      = ost;
    
    
        if (ost->stream_copy) {
            av_log(NULL, AV_LOG_ERROR, "Streamcopy requested for output stream %d:%d, "
                   "which is fed from a complex filtergraph. Filtering and streamcopy "
                   "cannot be used together.\n", ost->file_index, ost->index);
            exit_program(1);
        }
    
    
        if (configure_output_filter(ofilter->graph, ofilter, ofilter->out_tmp) < 0) {
            av_log(NULL, AV_LOG_FATAL, "Error configuring filter.\n");
            exit_program(1);
        }
        avfilter_inout_free(&ofilter->out_tmp);
    }
    
    
    static void opt_output_file(void *optctx, const char *filename)
    
        OptionsContext *o = optctx;
    
        AVOutputFormat *file_oformat;
    
        OutputStream *ost;
        InputStream  *ist;
    
    Fabrice Bellard's avatar
    Fabrice Bellard committed
    
    
        if (configure_complex_filters() < 0) {
            av_log(NULL, AV_LOG_FATAL, "Error configuring filters.\n");
            exit_program(1);
        }
    
    
    Fabrice Bellard's avatar
    Fabrice Bellard committed
        if (!strcmp(filename, "-"))
            filename = "pipe:";
    
    
        err = avformat_alloc_output_context2(&oc, NULL, o->format, filename);
    
            exit_program(1);
    
        oc->interrupt_callback = int_cb;
    
    Fabrice Bellard's avatar
    Fabrice Bellard committed
    
    
        /* create streams for all unlabeled output pads */
        for (i = 0; i < nb_filtergraphs; i++) {
            FilterGraph *fg = filtergraphs[i];
            for (j = 0; j < fg->nb_outputs; j++) {
                OutputFilter *ofilter = fg->outputs[j];
    
                if (!ofilter->out_tmp || ofilter->out_tmp->name)
                    continue;
    
                switch (ofilter->out_tmp->filter_ctx->output_pads[ofilter->out_tmp->pad_idx].type) {
                case AVMEDIA_TYPE_VIDEO:    o->video_disable    = 1; break;
                case AVMEDIA_TYPE_AUDIO:    o->audio_disable    = 1; break;
                case AVMEDIA_TYPE_SUBTITLE: o->subtitle_disable = 1; break;
                }
                init_output_filter(ofilter, o, oc);
            }
        }
    
    
        if (!strcmp(file_oformat->name, "ffm") &&
    
            av_strstart(filename, "http:", NULL)) {
    
    Fabrice Bellard's avatar
    Fabrice Bellard committed
            /* special case for files sent to ffserver: we get the stream
               parameters from ffserver */
    
            int err = read_ffserver_streams(o, oc, filename);
    
            if (err < 0) {
                print_error(filename, err);
    
                exit_program(1);
    
    Fabrice Bellard's avatar
    Fabrice Bellard committed
            }
    
            for(j = nb_output_streams - oc->nb_streams; j < nb_output_streams; j++) {
    
                    if(ist->st->codec->codec_type == ost->st->codec->codec_type){
                        ost->sync_ist= ist;
                        ost->source_index= i;
                        ist->discard = 0;
    
                        ist->st->discard = AVDISCARD_NONE;
    
                if(!ost->sync_ist){
                    av_log(NULL, AV_LOG_FATAL, "Missing %s stream which is required by this ffm\n", av_get_media_type_string(ost->st->codec->codec_type));
                    exit_program(1);
                }
    
        } else if (!o->nb_stream_maps) {
    
            /* pick the "best" stream of each type */
    
            /* video: highest resolution */
    
            if (!o->video_disable && oc->oformat->video_codec != CODEC_ID_NONE) {
    
                int area = 0, idx = -1;
                for (i = 0; i < nb_input_streams; i++) {
    
                    if (ist->st->codec->codec_type == AVMEDIA_TYPE_VIDEO &&
                        ist->st->codec->width * ist->st->codec->height > area) {
                        area = ist->st->codec->width * ist->st->codec->height;
                        idx = i;
                    }
                }
    
                if (idx >= 0)
                    new_video_stream(o, oc, idx);
    
            }
    
            /* audio: most channels */
    
            if (!o->audio_disable && oc->oformat->audio_codec != CODEC_ID_NONE) {
    
                int channels = 0, idx = -1;
                for (i = 0; i < nb_input_streams; i++) {
    
                    if (ist->st->codec->codec_type == AVMEDIA_TYPE_AUDIO &&
                        ist->st->codec->channels > channels) {
                        channels = ist->st->codec->channels;
                        idx = i;
                    }
                }
    
                if (idx >= 0)
                    new_audio_stream(o, oc, idx);
    
            }
    
            /* subtitles: pick first */
    
            if (!o->subtitle_disable && (oc->oformat->subtitle_codec != CODEC_ID_NONE || subtitle_codec_name)) {
    
                for (i = 0; i < nb_input_streams; i++)
    
                    if (input_streams[i]->st->codec->codec_type == AVMEDIA_TYPE_SUBTITLE) {
    
                        break;
                    }
            }
            /* do something with data? */
    
    Fabrice Bellard's avatar
    Fabrice Bellard committed
        } else {
    
            for (i = 0; i < o->nb_stream_maps; i++) {
                StreamMap *map = &o->stream_maps[i];
    
                int src_idx = input_files[map->file_index]->ist_index + map->stream_index;
    
                if (map->disabled)
                    continue;
    
                if (map->linklabel) {
                    FilterGraph *fg;
                    OutputFilter *ofilter = NULL;
                    int j, k;
    
                    for (j = 0; j < nb_filtergraphs; j++) {
                        fg = filtergraphs[j];
                        for (k = 0; k < fg->nb_outputs; k++) {
                            AVFilterInOut *out = fg->outputs[k]->out_tmp;
                            if (out && !strcmp(out->name, map->linklabel)) {
                                ofilter = fg->outputs[k];
                                goto loop_end;
                            }
                        }
                    }
    loop_end:
                    if (!ofilter) {
                        av_log(NULL, AV_LOG_FATAL, "Output with label '%s' does not exist "
                               "in any defined filter graph.\n", map->linklabel);
                        exit_program(1);
                    }
                    init_output_filter(ofilter, o, oc);
                } else {
    
    Anton Khirnov's avatar
    Anton Khirnov committed
                    ist = input_streams[input_files[map->file_index]->ist_index + map->stream_index];
    
                    if(o->subtitle_disable && ist->st->codec->codec_type == AVMEDIA_TYPE_SUBTITLE)
                        continue;
                    if(o->   audio_disable && ist->st->codec->codec_type == AVMEDIA_TYPE_AUDIO)
                        continue;
                    if(o->   video_disable && ist->st->codec->codec_type == AVMEDIA_TYPE_VIDEO)
                        continue;
                    if(o->    data_disable && ist->st->codec->codec_type == AVMEDIA_TYPE_DATA)
                        continue;
    
    
    Anton Khirnov's avatar
    Anton Khirnov committed
                    switch (ist->st->codec->codec_type) {
    
                    case AVMEDIA_TYPE_VIDEO:      ost = new_video_stream     (o, oc, src_idx); break;
                    case AVMEDIA_TYPE_AUDIO:      ost = new_audio_stream     (o, oc, src_idx); break;