Skip to content
Snippets Groups Projects
ffmpeg.c 112 KiB
Newer Older
  • Learn to ignore specific revisions
  •                                         ost1 = ost_table[i];
                                            if (ost1->file_index == ost->file_index &&
                                                ost1->st->codec.codec_type == CODEC_TYPE_AUDIO) {
                                                audio_sync = ost1;
                                                break;
                                            }
                                        }
    
                                        do_video_out(os, ost, ist, &picture, &frame_size, audio_sync);
                                        if (do_vstats && frame_size)
                                            do_video_stats(os, ost, frame_size);
                                    }
                                    break;
                                default:
                                    av_abort();
                                }
                            } else {
                                AVFrame avframe;
                                                    
                                /* no reencoding needed : output the packet directly */
                                /* force the input stream PTS */
                            
    
                                avcodec_get_frame_defaults(&avframe);
    
                                ost->st->codec.coded_frame= &avframe;
                                avframe.key_frame = pkt->flags & PKT_FLAG_KEY; 
                            
                                av_write_frame(os, ost->index, data_buf, data_size);
                                ost->st->codec.frame_number++;
                                ost->frame_number++;
                            }
                        }
                    }
                av_free(buffer_to_free);
            }
     discard_packet:
        return 0;
     fail_decode:
        return -1;
    }
    
    
    
    Fabrice Bellard's avatar
    Fabrice Bellard committed
    /*
     * The following code is the main loop of the file converter
     */
    static int av_encode(AVFormatContext **output_files,
                         int nb_output_files,
                         AVFormatContext **input_files,
                         int nb_input_files,
                         AVStreamMap *stream_maps, int nb_stream_maps)
    {
    
        int ret, i, j, k, n, nb_istreams = 0, nb_ostreams = 0;
    
    Fabrice Bellard's avatar
    Fabrice Bellard committed
        AVFormatContext *is, *os;
        AVCodecContext *codec, *icodec;
        AVOutputStream *ost, **ost_table = NULL;
        AVInputStream *ist, **ist_table = NULL;
    
    Fabrice Bellard's avatar
    Fabrice Bellard committed
        AVInputFile *file_table;
    
        AVFormatContext *stream_no_data;
    
    Fabrice Bellard's avatar
    Fabrice Bellard committed
    
    
        file_table= (AVInputFile*) av_mallocz(nb_input_files * sizeof(AVInputFile));
    
    Fabrice Bellard's avatar
    Fabrice Bellard committed
        if (!file_table)
            goto fail;
    
    Fabrice Bellard's avatar
    Fabrice Bellard committed
    
        /* input stream init */
        j = 0;
        for(i=0;i<nb_input_files;i++) {
            is = input_files[i];
            file_table[i].ist_index = j;
    
            file_table[i].nb_streams = is->nb_streams;
    
    Fabrice Bellard's avatar
    Fabrice Bellard committed
            j += is->nb_streams;
        }
        nb_istreams = j;
    
        ist_table = av_mallocz(nb_istreams * sizeof(AVInputStream *));
        if (!ist_table)
    
    Fabrice Bellard's avatar
    Fabrice Bellard committed
            goto fail;
    
    Fabrice Bellard's avatar
    Fabrice Bellard committed
        
        for(i=0;i<nb_istreams;i++) {
            ist = av_mallocz(sizeof(AVInputStream));
            if (!ist)
                goto fail;
            ist_table[i] = ist;
        }
        j = 0;
        for(i=0;i<nb_input_files;i++) {
            is = input_files[i];
            for(k=0;k<is->nb_streams;k++) {
                ist = ist_table[j++];
                ist->st = is->streams[k];
                ist->file_index = i;
                ist->index = k;
                ist->discard = 1; /* the stream is discarded by default
                                     (changed later) */
    
    
                if (ist->st->codec.rate_emu) {
                    ist->start = av_gettime();
                    ist->frame = 0;
                }
    
    Fabrice Bellard's avatar
    Fabrice Bellard committed
            }
        }
    
        /* output stream init */
        nb_ostreams = 0;
        for(i=0;i<nb_output_files;i++) {
            os = output_files[i];
            nb_ostreams += os->nb_streams;
        }
        if (nb_stream_maps > 0 && nb_stream_maps != nb_ostreams) {
            fprintf(stderr, "Number of stream maps must match number of output streams\n");
            exit(1);
        }
    
    
        /* Sanity check the mapping args -- do the input files & streams exist? */
        for(i=0;i<nb_stream_maps;i++) {
            int fi = stream_maps[i].file_index;
            int si = stream_maps[i].stream_index;
            
            if (fi < 0 || fi > nb_input_files - 1 ||
                si < 0 || si > file_table[fi].nb_streams - 1) {
                fprintf(stderr,"Could not find input stream #%d.%d\n", fi, si);
                exit(1);
            }
        }
        
    
    Fabrice Bellard's avatar
    Fabrice Bellard committed
        ost_table = av_mallocz(sizeof(AVOutputStream *) * nb_ostreams);
        if (!ost_table)
            goto fail;
        for(i=0;i<nb_ostreams;i++) {
            ost = av_mallocz(sizeof(AVOutputStream));
            if (!ost)
                goto fail;
            ost_table[i] = ost;
        }
        
        n = 0;
        for(k=0;k<nb_output_files;k++) {
            os = output_files[k];
            for(i=0;i<os->nb_streams;i++) {
                int found;
                ost = ost_table[n++];
                ost->file_index = k;
                ost->index = i;
                ost->st = os->streams[i];
                if (nb_stream_maps > 0) {
                    ost->source_index = file_table[stream_maps[n-1].file_index].ist_index + 
                        stream_maps[n-1].stream_index;
    
                        
                    /* Sanity check that the stream types match */
                    if (ist_table[ost->source_index]->st->codec.codec_type != ost->st->codec.codec_type) {
                        fprintf(stderr, "Codec type mismatch for mapping #%d.%d -> #%d.%d\n",
                            stream_maps[n-1].file_index, stream_maps[n-1].stream_index,
                            ost->file_index, ost->index);
                        exit(1);
                    }
                    
    
    Fabrice Bellard's avatar
    Fabrice Bellard committed
                } else {
                    /* get corresponding input stream index : we select the first one with the right type */
                    found = 0;
                    for(j=0;j<nb_istreams;j++) {
                        ist = ist_table[j];
                        if (ist->discard && 
                            ist->st->codec.codec_type == ost->st->codec.codec_type) {
                            ost->source_index = j;
                            found = 1;
                        }
                    }
                    
                    if (!found) {
                        /* try again and reuse existing stream */
                        for(j=0;j<nb_istreams;j++) {
                            ist = ist_table[j];
                            if (ist->st->codec.codec_type == ost->st->codec.codec_type) {
                                ost->source_index = j;
                                found = 1;
                            }
                        }
                        if (!found) {
                            fprintf(stderr, "Could not find input stream matching output stream #%d.%d\n",
                                    ost->file_index, ost->index);
                            exit(1);
                        }
                    }
                }
                ist = ist_table[ost->source_index];
                ist->discard = 0;
            }
        }
    
        /* for each output stream, we compute the right encoding parameters */
        for(i=0;i<nb_ostreams;i++) {
            ost = ost_table[i];
            ist = ist_table[ost->source_index];
    
            codec = &ost->st->codec;
            icodec = &ist->st->codec;
    
    
            if (ost->st->stream_copy) {
                /* if stream_copy is selected, no need to decode or encode */
                codec->codec_id = icodec->codec_id;
                codec->codec_type = icodec->codec_type;
                codec->codec_tag = icodec->codec_tag;
                codec->bit_rate = icodec->bit_rate;
                switch(codec->codec_type) {
                case CODEC_TYPE_AUDIO:
                    codec->sample_rate = icodec->sample_rate;
                    codec->channels = icodec->channels;
    
                    codec->frame_size = icodec->frame_size;
    
                    break;
                case CODEC_TYPE_VIDEO:
                    codec->frame_rate = icodec->frame_rate;
    
                    codec->frame_rate_base = icodec->frame_rate_base;
    
                    codec->width = icodec->width;
                    codec->height = icodec->height;
                    break;
                default:
                    av_abort();
                }
            } else {
                switch(codec->codec_type) {
                case CODEC_TYPE_AUDIO:
    
    Fabrice Bellard's avatar
    Fabrice Bellard committed
                    if (fifo_init(&ost->fifo, 2 * MAX_AUDIO_PACKET_SIZE))
                        goto fail;
    
    Fabrice Bellard's avatar
    Fabrice Bellard committed
                    if (codec->channels == icodec->channels &&
                        codec->sample_rate == icodec->sample_rate) {
                        ost->audio_resample = 0;
                    } else {
    
                        if (codec->channels != icodec->channels &&
                            icodec->codec_id == CODEC_ID_AC3) {
                            /* Special case for 5:1 AC3 input */
                            /* and mono or stereo output      */
    
                            /* Request specific number of channels */
                            icodec->channels = codec->channels;
                            if (codec->sample_rate == icodec->sample_rate)
                                ost->audio_resample = 0;
                            else {
                                ost->audio_resample = 1;
                                ost->resample = audio_resample_init(codec->channels, icodec->channels,
                                                            codec->sample_rate, 
                                                            icodec->sample_rate);
    
    			    if(!ost->resample)
    			      {
    				printf("Can't resample.  Aborting.\n");
    				av_abort();
    			      }
    
                            /* Request specific number of channels */
                            icodec->channels = codec->channels;
                        } else {
                            ost->audio_resample = 1; 
                            ost->resample = audio_resample_init(codec->channels, icodec->channels,
    
    Fabrice Bellard's avatar
    Fabrice Bellard committed
                                                            codec->sample_rate, 
                                                            icodec->sample_rate);
    
    			if(!ost->resample)
    			  {
    			    printf("Can't resample.  Aborting.\n");
    			    av_abort();
    			  }
    
    Fabrice Bellard's avatar
    Fabrice Bellard committed
                    }
                    ist->decoding_needed = 1;
                    ost->encoding_needed = 1;
    
    Fabrice Bellard's avatar
    Fabrice Bellard committed
                    if (codec->width == icodec->width &&
    
                        codec->height == icodec->height &&
                        frame_topBand == 0 &&
                        frame_bottomBand == 0 &&
                        frame_leftBand == 0 &&
                        frame_rightBand == 0)
                    {
                        ost->video_resample = 0;
                        ost->video_crop = 0;
                    } else if ((codec->width == icodec->width -
                                    (frame_leftBand + frame_rightBand)) &&
                            (codec->height == icodec->height -
                                    (frame_topBand  + frame_bottomBand)))
                    {
    
    Fabrice Bellard's avatar
    Fabrice Bellard committed
                        ost->video_resample = 0;
    
                        ost->video_crop = 1;
                        ost->topBand = frame_topBand;
                        ost->leftBand = frame_leftBand;
    
    Fabrice Bellard's avatar
    Fabrice Bellard committed
                    } else {
                        ost->video_resample = 1;
    
                        ost->video_crop = 0; // cropping is handled as part of resample
    
    Michael Niedermayer's avatar
    Michael Niedermayer committed
                        if( avpicture_alloc( &ost->pict_tmp, PIX_FMT_YUV420P,
    
                                             codec->width, codec->height ) )
    
    Fabrice Bellard's avatar
    Fabrice Bellard committed
                            goto fail;
    
    
                        ost->img_resample_ctx = img_resample_full_init( 
    
    Fabrice Bellard's avatar
    Fabrice Bellard committed
                                          ost->st->codec.width, ost->st->codec.height,
    
                                          ist->st->codec.width, ist->st->codec.height,
                                          frame_topBand, frame_bottomBand,
                                          frame_leftBand, frame_rightBand);
    
    Fabrice Bellard's avatar
    Fabrice Bellard committed
                    }
                    ost->encoding_needed = 1;
                    ist->decoding_needed = 1;
    
    Fabrice Bellard's avatar
    Fabrice Bellard committed
                }
    
                /* two pass mode */
                if (ost->encoding_needed && 
                    (codec->flags & (CODEC_FLAG_PASS1 | CODEC_FLAG_PASS2))) {
                    char logfilename[1024];
                    FILE *f;
                    int size;
                    char *logbuffer;
                    
                    snprintf(logfilename, sizeof(logfilename), "%s-%d.log", 
                             pass_logfilename ? 
                             pass_logfilename : DEFAULT_PASS_LOGFILENAME, i);
                    if (codec->flags & CODEC_FLAG_PASS1) {
                        f = fopen(logfilename, "w");
                        if (!f) {
                            perror(logfilename);
                            exit(1);
                        }
                        ost->logfile = f;
                    } else {
                        /* read the log file */
                        f = fopen(logfilename, "r");
                        if (!f) {
                            perror(logfilename);
                            exit(1);
                        }
                        fseek(f, 0, SEEK_END);
                        size = ftell(f);
                        fseek(f, 0, SEEK_SET);
                        logbuffer = av_malloc(size + 1);
                        if (!logbuffer) {
                            fprintf(stderr, "Could not allocate log buffer\n");
                            exit(1);
                        }
    
                        fclose(f);
                        logbuffer[size] = '\0';
                        codec->stats_in = logbuffer;
    
    Fabrice Bellard's avatar
    Fabrice Bellard committed
        }
    
    
        /* dump the file output parameters - cannot be done before in case
           of stream copy */
        for(i=0;i<nb_output_files;i++) {
            dump_format(output_files[i], i, output_files[i]->filename, 1);
        }
    
        /* dump the stream mapping */
        fprintf(stderr, "Stream mapping:\n");
        for(i=0;i<nb_ostreams;i++) {
            ost = ost_table[i];
            fprintf(stderr, "  Stream #%d.%d -> #%d.%d\n",
                    ist_table[ost->source_index]->file_index,
                    ist_table[ost->source_index]->index,
                    ost->file_index, 
                    ost->index);
        }
    
    
    Fabrice Bellard's avatar
    Fabrice Bellard committed
        /* open each encoder */
        for(i=0;i<nb_ostreams;i++) {
            ost = ost_table[i];
            if (ost->encoding_needed) {
                AVCodec *codec;
                codec = avcodec_find_encoder(ost->st->codec.codec_id);
                if (!codec) {
                    fprintf(stderr, "Unsupported codec for output stream #%d.%d\n", 
                            ost->file_index, ost->index);
                    exit(1);
                }
                if (avcodec_open(&ost->st->codec, codec) < 0) {
                    fprintf(stderr, "Error while opening codec for stream #%d.%d - maybe incorrect parameters such as bit_rate, rate, width or height\n", 
                            ost->file_index, ost->index);
                    exit(1);
                }
            }
        }
    
        /* open each decoder */
        for(i=0;i<nb_istreams;i++) {
            ist = ist_table[i];
            if (ist->decoding_needed) {
                AVCodec *codec;
                codec = avcodec_find_decoder(ist->st->codec.codec_id);
                if (!codec) {
    
                    fprintf(stderr, "Unsupported codec (id=%d) for input stream #%d.%d\n", 
                            ist->st->codec.codec_id, ist->file_index, ist->index);
    
    Fabrice Bellard's avatar
    Fabrice Bellard committed
                    exit(1);
                }
                if (avcodec_open(&ist->st->codec, codec) < 0) {
                    fprintf(stderr, "Error while opening codec for input stream #%d.%d\n", 
                            ist->file_index, ist->index);
                    exit(1);
                }
    
                //if (ist->st->codec.codec_type == CODEC_TYPE_VIDEO)
                //    ist->st->codec.flags |= CODEC_FLAG_REPEAT_FIELD;
    
    Fabrice Bellard's avatar
    Fabrice Bellard committed
            }
        }
    
        /* init pts */
        for(i=0;i<nb_istreams;i++) {
            ist = ist_table[i];
    
    	is = input_files[ist->file_index];
    
            ist->pts = 0;
    
    Fabrice Bellard's avatar
    Fabrice Bellard committed
        }
        
        /* compute buffer size max (should use a complete heuristic) */
        for(i=0;i<nb_input_files;i++) {
            file_table[i].buffer_size_max = 2048;
        }
    
        /* open files and write file headers */
        for(i=0;i<nb_output_files;i++) {
            os = output_files[i];
    
            if (av_write_header(os) < 0) {
    
                fprintf(stderr, "Could not write header for output file #%d (incorrect codec paramters ?)\n", i);
                ret = -EINVAL;
                goto fail;
            }
    
    Fabrice Bellard's avatar
    Fabrice Bellard committed
        }
    
    
            fprintf(stderr, "Press [q] to stop encoding\n");
    
            url_set_interrupt_cb(decode_interrupt_cb);
        }
    
    Fabrice Bellard's avatar
    Fabrice Bellard committed
            int file_index, ist_index;
            AVPacket pkt;
    
    Fabrice Bellard's avatar
    Fabrice Bellard committed
        redo:
    
                /* read_key() returns 0 on EOF */
                key = read_key();
                if (key == 'q')
                    break;
            }
    
            /* select the stream that we must read now by looking at the
               smallest output pts */
    
    Fabrice Bellard's avatar
    Fabrice Bellard committed
            file_index = -1;
    
            pts_min = 1e10;
            for(i=0;i<nb_ostreams;i++) {
                double pts;
                ost = ost_table[i];
                os = output_files[ost->file_index];
                ist = ist_table[ost->source_index];
                pts = (double)ost->st->pts.val * os->pts_num / os->pts_den;
                if (!file_table[ist->file_index].eof_reached && 
                    pts < pts_min) {
                    pts_min = pts;
    
    Fabrice Bellard's avatar
    Fabrice Bellard committed
                    file_index = ist->file_index;
                }
            }
            /* if none, if is finished */
    
    Fabrice Bellard's avatar
    Fabrice Bellard committed
                break;
    
    Fabrice Bellard's avatar
    Fabrice Bellard committed
            /* finish if recording time exhausted */
    
            if (recording_time > 0 && pts_min >= (recording_time / 1000000.0))
    
    Fabrice Bellard's avatar
    Fabrice Bellard committed
                break;
    
            /* read a frame from it and output it in the fifo */
    
    Fabrice Bellard's avatar
    Fabrice Bellard committed
            is = input_files[file_index];
    
    Fabrice Bellard's avatar
    Fabrice Bellard committed
                file_table[file_index].eof_reached = 1;
                continue;
            }
    
            if (!pkt.size) {
                stream_no_data = is;
            } else {
                stream_no_data = 0;
            }
    
            if (do_pkt_dump) {
                av_pkt_dump(stdout, &pkt, do_hex_dump);
    
            /* the following test is needed in case new streams appear
               dynamically in stream : we ignore them */
            if (pkt.stream_index >= file_table[file_index].nb_streams)
    
                goto discard_packet;
    
    Fabrice Bellard's avatar
    Fabrice Bellard committed
            ist_index = file_table[file_index].ist_index + pkt.stream_index;
            ist = ist_table[ist_index];
    
            if (ist->discard)
                goto discard_packet;
    
    Fabrice Bellard's avatar
    Fabrice Bellard committed
    
    
            //fprintf(stderr,"read #%d.%d size=%d\n", ist->file_index, ist->index, pkt.size);
    
            if (output_packet(ist, ist_index, ost_table, nb_ostreams, &pkt) < 0) {
                fprintf(stderr, "Error while decoding stream #%d.%d\n",
                        ist->file_index, ist->index);
                av_free_packet(&pkt);
                goto redo;
    
    Michael Niedermayer's avatar
    Michael Niedermayer committed
            }
    
    Fabrice Bellard's avatar
    Fabrice Bellard committed
            av_free_packet(&pkt);
            
    
            /* dump report by using the output first video and audio streams */
            print_report(output_files, ost_table, nb_ostreams, 0);
    
    Fabrice Bellard's avatar
    Fabrice Bellard committed
        }
    
    
        /* at the end of stream, we must flush the decoder buffers */
        for(i=0;i<nb_istreams;i++) {
            ist = ist_table[i];
            if (ist->decoding_needed) {
                output_packet(ist, i, ost_table, nb_ostreams, NULL);
            }
        }
    
    
    Fabrice Bellard's avatar
    Fabrice Bellard committed
    
        /* dump report by using the first video and audio streams */
    
        print_report(output_files, ost_table, nb_ostreams, 1);
    
    
        /* write the trailer if needed and close file */
        for(i=0;i<nb_output_files;i++) {
            os = output_files[i];
            av_write_trailer(os);
        }
    
    
    Fabrice Bellard's avatar
    Fabrice Bellard committed
        /* close each encoder */
        for(i=0;i<nb_ostreams;i++) {
            ost = ost_table[i];
            if (ost->encoding_needed) {
    
                av_freep(&ost->st->codec.stats_in);
    
    Fabrice Bellard's avatar
    Fabrice Bellard committed
                avcodec_close(&ost->st->codec);
            }
        }
        
        /* close each decoder */
        for(i=0;i<nb_istreams;i++) {
            ist = ist_table[i];
            if (ist->decoding_needed) {
                avcodec_close(&ist->st->codec);
            }
        }
    
        /* finished ! */
        
        ret = 0;
     fail1:
    
    Fabrice Bellard's avatar
    Fabrice Bellard committed
        av_free(file_table);
    
    Fabrice Bellard's avatar
    Fabrice Bellard committed
    
    
    Fabrice Bellard's avatar
    Fabrice Bellard committed
        if (ist_table) {
            for(i=0;i<nb_istreams;i++) {
                ist = ist_table[i];
    
    Fabrice Bellard's avatar
    Fabrice Bellard committed
                av_free(ist);
    
    Fabrice Bellard's avatar
    Fabrice Bellard committed
            }
    
    Fabrice Bellard's avatar
    Fabrice Bellard committed
            av_free(ist_table);
    
    Fabrice Bellard's avatar
    Fabrice Bellard committed
        }
        if (ost_table) {
            for(i=0;i<nb_ostreams;i++) {
                ost = ost_table[i];
                if (ost) {
    
                    if (ost->logfile) {
                        fclose(ost->logfile);
                        ost->logfile = NULL;
                    }
    
                    fifo_free(&ost->fifo); /* works even if fifo is not
                                              initialized but set to zero */
    
    Fabrice Bellard's avatar
    Fabrice Bellard committed
                    av_free(ost->pict_tmp.data[0]);
    
    Fabrice Bellard's avatar
    Fabrice Bellard committed
                    if (ost->video_resample)
                        img_resample_close(ost->img_resample_ctx);
                    if (ost->audio_resample)
                        audio_resample_close(ost->resample);
    
    Fabrice Bellard's avatar
    Fabrice Bellard committed
                    av_free(ost);
    
    Fabrice Bellard's avatar
    Fabrice Bellard committed
                }
            }
    
    Fabrice Bellard's avatar
    Fabrice Bellard committed
            av_free(ost_table);
    
    Fabrice Bellard's avatar
    Fabrice Bellard committed
        }
        return ret;
     fail:
        ret = -ENOMEM;
        goto fail1;
    }
    
    #if 0
    int file_read(const char *filename)
    {
        URLContext *h;
        unsigned char buffer[1024];
        int len, i;
    
        if (url_open(&h, filename, O_RDONLY) < 0) {
            printf("could not open '%s'\n", filename);
            return -1;
        }
        for(;;) {
            len = url_read(h, buffer, sizeof(buffer));
            if (len <= 0)
                break;
            for(i=0;i<len;i++) putchar(buffer[i]);
        }
        url_close(h);
        return 0;
    }
    #endif
    
    
    static void opt_image_format(const char *arg)
    
    {
        AVImageFormat *f;
        
        for(f = first_image_format; f != NULL; f = f->next) {
            if (!strcmp(arg, f->name))
                break;
        }
        if (!f) {
            fprintf(stderr, "Unknown image format: '%s'\n", arg);
            exit(1);
        }
        image_format = f;
    }
    
    
    static void opt_format(const char *arg)
    
    Fabrice Bellard's avatar
    Fabrice Bellard committed
    {
    
        /* compatibility stuff for pgmyuv */
        if (!strcmp(arg, "pgmyuv")) {
            opt_image_format(arg);
            arg = "image";
        }
    
    
        file_iformat = av_find_input_format(arg);
        file_oformat = guess_format(arg, NULL, NULL);
        if (!file_iformat && !file_oformat) {
            fprintf(stderr, "Unknown input or output format: %s\n", arg);
    
    Fabrice Bellard's avatar
    Fabrice Bellard committed
            exit(1);
        }
    }
    
    
    static void opt_video_bitrate(const char *arg)
    
    Fabrice Bellard's avatar
    Fabrice Bellard committed
    {
        video_bit_rate = atoi(arg) * 1000;
    }
    
    
    static void opt_video_bitrate_tolerance(const char *arg)
    
    {
        video_bit_rate_tolerance = atoi(arg) * 1000;
    }
    
    
    static void opt_video_bitrate_max(const char *arg)
    
    {
        video_rc_max_rate = atoi(arg) * 1000;
    }
    
    
    static void opt_video_bitrate_min(const char *arg)
    
    {
        video_rc_min_rate = atoi(arg) * 1000;
    }
    
    
    static void opt_video_buffer_size(const char *arg)
    
    Michael Niedermayer's avatar
    Michael Niedermayer committed
    {
    
        video_rc_buffer_size = atoi(arg) * 8*1024;
    
    Michael Niedermayer's avatar
    Michael Niedermayer committed
    }
    
    
    static void opt_video_rc_eq(char *arg)
    
    {
        video_rc_eq = arg;
    }
    
    
    static void opt_video_rc_override_string(char *arg)
    
    static void opt_workaround_bugs(const char *arg)
    
    static void opt_dct_algo(const char *arg)
    
    static void opt_idct_algo(const char *arg)
    
    static void opt_error_resilience(const char *arg)
    
    {
        error_resilience = atoi(arg);
    }
    
    
    static void opt_error_concealment(const char *arg)
    
    static void opt_debug(const char *arg)
    
    Michael Niedermayer's avatar
    Michael Niedermayer committed
    {
        debug = atoi(arg);
    }
    
    static void opt_vismv(const char *arg)
    {
        debug_mv = atoi(arg);
    }
        
    
    static void opt_verbose(const char *arg)
    {
        verbose = atoi(arg);
    }
    
    
    static void opt_frame_rate(const char *arg)
    
    Fabrice Bellard's avatar
    Fabrice Bellard committed
    {
    
        if (parse_frame_rate(&frame_rate, &frame_rate_base, arg) < 0) {
            fprintf(stderr, "Incorrect frame rate\n");
    	exit(1);
        }
    
    Fabrice Bellard's avatar
    Fabrice Bellard committed
    }
    
    
    static void opt_frame_crop_top(const char *arg)
    
    {
        frame_topBand = atoi(arg); 
        if (frame_topBand < 0) {
            fprintf(stderr, "Incorrect top crop size\n");
            exit(1);
        }
        if ((frame_topBand % 2) != 0) {
            fprintf(stderr, "Top crop size must be a multiple of 2\n");
            exit(1);
        }
        if ((frame_topBand) >= frame_height){
        	fprintf(stderr, "Vertical crop dimensions are outside the range of the original image.\nRemember to crop first and scale second.\n");
            exit(1);
        }
        frame_height -= frame_topBand;
    }
    
    
    static void opt_frame_crop_bottom(const char *arg)
    
    {
        frame_bottomBand = atoi(arg);
        if (frame_bottomBand < 0) {
            fprintf(stderr, "Incorrect bottom crop size\n");
            exit(1);
        }
        if ((frame_bottomBand % 2) != 0) {
            fprintf(stderr, "Bottom crop size must be a multiple of 2\n");
            exit(1);        
        }
        if ((frame_bottomBand) >= frame_height){
        	fprintf(stderr, "Vertical crop dimensions are outside the range of the original image.\nRemember to crop first and scale second.\n");
            exit(1);
        }
        frame_height -= frame_bottomBand;
    }
    
    
    static void opt_frame_crop_left(const char *arg)
    
    {
        frame_leftBand = atoi(arg);
        if (frame_leftBand < 0) {
            fprintf(stderr, "Incorrect left crop size\n");
            exit(1);
        }
        if ((frame_leftBand % 2) != 0) {
            fprintf(stderr, "Left crop size must be a multiple of 2\n");
            exit(1);
        }
        if ((frame_leftBand) >= frame_width){
        	fprintf(stderr, "Horizontal crop dimensions are outside the range of the original image.\nRemember to crop first and scale second.\n");
            exit(1);
        }
        frame_width -= frame_leftBand;
    }
    
    
    static void opt_frame_crop_right(const char *arg)
    
    {
        frame_rightBand = atoi(arg);
        if (frame_rightBand < 0) {
            fprintf(stderr, "Incorrect right crop size\n");
            exit(1);
        }
        if ((frame_rightBand % 2) != 0) {
            fprintf(stderr, "Right crop size must be a multiple of 2\n");
            exit(1);        
        }
        if ((frame_rightBand) >= frame_width){
        	fprintf(stderr, "Horizontal crop dimensions are outside the range of the original image.\nRemember to crop first and scale second.\n");
            exit(1);
        }
        frame_width -= frame_rightBand;
    }
    
    
    static void opt_frame_size(const char *arg)
    
    Fabrice Bellard's avatar
    Fabrice Bellard committed
    {
    
        if (parse_image_size(&frame_width, &frame_height, arg) < 0) {
    
    Fabrice Bellard's avatar
    Fabrice Bellard committed
            fprintf(stderr, "Incorrect frame size\n");
            exit(1);
        }
        if ((frame_width % 2) != 0 || (frame_height % 2) != 0) {
            fprintf(stderr, "Frame size must be a multiple of 2\n");
            exit(1);
        }
    }
    
    
    static void opt_frame_pix_fmt(const char *arg)
    {
        frame_pix_fmt = avcodec_get_pix_fmt(arg);
    }
    
    
    static void opt_frame_aspect_ratio(const char *arg)
    {
        int x = 0, y = 0;
        double ar = 0;
        const char *p;
        
        p = strchr(arg, ':');
        if (p) {
            x = strtol(arg, (char **)&arg, 10);
    	if (arg == p)
    	    y = strtol(arg+1, (char **)&arg, 10);
    	if (x > 0 && y > 0)
    	    ar = (double)x / (double)y;
        } else
            ar = strtod(arg, (char **)&arg);
    
        if (!ar) {
            fprintf(stderr, "Incorrect aspect ratio specification.\n");
    	exit(1);
        }
        frame_aspect_ratio = ar;
    }
    
    
    static void opt_gop_size(const char *arg)
    
    Fabrice Bellard's avatar
    Fabrice Bellard committed
    {
        gop_size = atoi(arg);
    }
    
    
    static void opt_b_frames(const char *arg)
    
    {
        b_frames = atoi(arg);
        if (b_frames > FF_MAX_B_FRAMES) {
            fprintf(stderr, "\nCannot have more than %d B frames, increase FF_MAX_B_FRAMES.\n", FF_MAX_B_FRAMES);
            exit(1);
        } else if (b_frames < 1) {
            fprintf(stderr, "\nNumber of B frames must be higher than 0\n");
            exit(1);
        }
    }
    
    
    static void opt_mb_decision(const char *arg)
    {
        mb_decision = atoi(arg);
    }
    
    
    static void opt_mb_cmp(const char *arg)
    {
        mb_cmp = atoi(arg);
    }
    
    
    static void opt_ildct_cmp(const char *arg)
    {
        ildct_cmp = atoi(arg);
    }
    
    
    static void opt_sub_cmp(const char *arg)
    {
        sub_cmp = atoi(arg);
    }
    
    static void opt_cmp(const char *arg)
    {
        cmp = atoi(arg);
    }
    
    
    static void opt_pre_cmp(const char *arg)
    {
        pre_cmp = atoi(arg);
    }
    
    static void opt_pre_me(const char *arg)
    {
        pre_me = atoi(arg);
    }
    
    static void opt_lumi_mask(const char *arg)
    {
        lumi_mask = atof(arg);
    }
    
    static void opt_dark_mask(const char *arg)
    {
        dark_mask = atof(arg);
    }
    
    static void opt_scplx_mask(const char *arg)
    {
        scplx_mask = atof(arg);
    }
    
    static void opt_tcplx_mask(const char *arg)
    {
        tcplx_mask = atof(arg);
    }
    
    static void opt_p_mask(const char *arg)
    {
        p_mask = atof(arg);
    }
    
    
    static void opt_qscale(const char *arg)
    
    Fabrice Bellard's avatar
    Fabrice Bellard committed
    {
    
        video_qscale = atof(arg);
        if (video_qscale < 0.01 ||
            video_qscale > 255) {
            fprintf(stderr, "qscale must be >= 0.01 and <= 255\n");
    
    Fabrice Bellard's avatar
    Fabrice Bellard committed
            exit(1);
        }
    }
    
    
    static void opt_qmin(const char *arg)
    
    {
        video_qmin = atoi(arg);
        if (video_qmin < 0 ||
            video_qmin > 31) {
            fprintf(stderr, "qmin must be >= 1 and <= 31\n");
            exit(1);
        }
    }
    
    
    static void opt_qmax(const char *arg)
    
    {
        video_qmax = atoi(arg);
        if (video_qmax < 0 ||
            video_qmax > 31) {
            fprintf(stderr, "qmax must be >= 1 and <= 31\n");
            exit(1);
        }
    }
    
    
    static void opt_mb_qmin(const char *arg)
    
    Michael Niedermayer's avatar
    Michael Niedermayer committed
    {
        video_mb_qmin = atoi(arg);
        if (video_mb_qmin < 0 ||
            video_mb_qmin > 31) {
            fprintf(stderr, "qmin must be >= 1 and <= 31\n");
            exit(1);
        }
    }
    
    
    static void opt_mb_qmax(const char *arg)
    
    Michael Niedermayer's avatar
    Michael Niedermayer committed
    {
        video_mb_qmax = atoi(arg);
        if (video_mb_qmax < 0 ||
            video_mb_qmax > 31) {
            fprintf(stderr, "qmax must be >= 1 and <= 31\n");
            exit(1);
        }
    }
    
    
    static void opt_qdiff(const char *arg)
    
    {
        video_qdiff = atoi(arg);
        if (video_qdiff < 0 ||
            video_qdiff > 31) {
            fprintf(stderr, "qdiff must be >= 1 and <= 31\n");
            exit(1);
        }
    }
    
    
    static void opt_qblur(const char *arg)
    
    {
        video_qblur = atof(arg);
    }
    
    
    static void opt_qcomp(const char *arg)
    
    {
        video_qcomp = atof(arg);
    }
    
    Fabrice Bellard's avatar
    Fabrice Bellard committed
    
    
    static void opt_rc_initial_cplx(const char *arg)
    
    {
        video_rc_initial_cplx = atof(arg);
    }
    
    static void opt_b_qfactor(const char *arg)
    
    static void opt_i_qfactor(const char *arg)
    
    static void opt_b_qoffset(const char *arg)