Skip to content
Snippets Groups Projects
ffmpeg.c 70.9 KiB
Newer Older
  • Learn to ignore specific revisions
  • Fabrice Bellard's avatar
    Fabrice Bellard committed
            }
        
            if (use_audio) {
                AVCodecContext *audio_enc;
    
                st = av_mallocz(sizeof(AVStream));
                if (!st) {
                    fprintf(stderr, "Could not alloc stream\n");
                    exit(1);
                }
                audio_enc = &st->codec;
                codec_id = file_format->audio_codec;
                if (audio_codec_id != CODEC_ID_NONE)
                    codec_id = audio_codec_id;
                audio_enc->codec_id = codec_id;
                audio_enc->codec_type = CODEC_TYPE_AUDIO;
                
                audio_enc->bit_rate = audio_bit_rate;
                audio_enc->sample_rate = audio_sample_rate;
                audio_enc->channels = audio_channels;
                oc->streams[nb_streams] = st;
                nb_streams++;
            }
    
            oc->nb_streams = nb_streams;
    
            if (!nb_streams) {
    
                fprintf(stderr, "No audio or video streams available\n");
    
    Fabrice Bellard's avatar
    Fabrice Bellard committed
                exit(1);
            }
    
            if (str_title)
                nstrcpy(oc->title, sizeof(oc->title), str_title);
            if (str_author)
                nstrcpy(oc->author, sizeof(oc->author), str_author);
            if (str_copyright)
                nstrcpy(oc->copyright, sizeof(oc->copyright), str_copyright);
            if (str_comment)
                nstrcpy(oc->comment, sizeof(oc->comment), str_comment);
        }
    
        output_files[nb_output_files] = oc;
        /* dump the file content */
        dump_format(oc, nb_output_files, filename, 1);
        nb_output_files++;
    
        strcpy(oc->filename, filename);
    
    
        /* check filename in case of an image number is expected */
        if (oc->format->flags & AVFMT_NEEDNUMBER) {
            if (filename_number_test(oc->filename) < 0)
                exit(1);
        }
    
    
    Fabrice Bellard's avatar
    Fabrice Bellard committed
        if (!(oc->format->flags & AVFMT_NOFILE)) {
            /* test if it already exists to avoid loosing precious files */
            if (!file_overwrite && 
                (strchr(filename, ':') == NULL ||
                 strstart(filename, "file:", NULL))) {
                if (url_exist(filename)) {
                    int c;
                    
                    printf("File '%s' already exists. Overwrite ? [y/N] ", filename);
                    fflush(stdout);
                    c = getchar();
                    if (toupper(c) != 'Y') {
                        fprintf(stderr, "Not overwriting - exiting\n");
                        exit(1);
                    }
                }
            }
            
            /* open the file */
            if (url_fopen(&oc->pb, filename, URL_WRONLY) < 0) {
                fprintf(stderr, "Could not open '%s'\n", filename);
                exit(1);
            }
        }
    
        /* reset some options */
        file_format = NULL;
        audio_disable = 0;
        video_disable = 0;
        audio_codec_id = CODEC_ID_NONE;
        video_codec_id = CODEC_ID_NONE;
    }
    
    
    Fabrice Bellard's avatar
    Fabrice Bellard committed
    #ifndef CONFIG_WIN32
    
    Fabrice Bellard's avatar
    Fabrice Bellard committed
    INT64 getutime(void)
    {
        struct rusage rusage;
    
        getrusage(RUSAGE_SELF, &rusage);
        return (rusage.ru_utime.tv_sec * 1000000LL) + rusage.ru_utime.tv_usec;
    }
    
    Fabrice Bellard's avatar
    Fabrice Bellard committed
    #else
    INT64 getutime(void)
    {
      return gettime();
    }
    #endif
    
    Fabrice Bellard's avatar
    Fabrice Bellard committed
    void show_formats(void)
    {
        AVFormat *f;
        URLProtocol *up;
        AVCodec *p;
        const char **pp;
    
        printf("File formats:\n");
        printf("  Encoding:");
        for(f = first_format; f != NULL; f = f->next) {
            if (f->write_header)
                printf(" %s", f->name);
        }
        printf("\n");
        printf("  Decoding:");
        for(f = first_format; f != NULL; f = f->next) {
            if (f->read_header)
                printf(" %s", f->name);
        }
        printf("\n");
    
        printf("Codecs:\n");
        printf("  Encoders:");
        for(p = first_avcodec; p != NULL; p = p->next) {
            if (p->encode)
                printf(" %s", p->name);
        }
        printf("\n");
    
        printf("  Decoders:");
        for(p = first_avcodec; p != NULL; p = p->next) {
            if (p->decode)
                printf(" %s", p->name);
        }
        printf("\n");
    
        printf("Supported file protocols:");
        for(up = first_protocol; up != NULL; up = up->next)
            printf(" %s:", up->name);
        printf("\n");
        
        printf("Frame size abbreviations: sqcif qcif cif 4cif\n");
        printf("Motion estimation methods:");
        pp = motion_str;
        while (*pp) {
            printf(" %s", *pp);
            if ((pp - motion_str) == ME_ZERO) 
                printf("(fastest)");
            else if ((pp - motion_str) == ME_FULL) 
                printf("(slowest)");
            else if ((pp - motion_str) == ME_LOG) 
                printf("(default)");
            pp++;
        }
        printf("\n");
        exit(1);
    }
    
    void show_help(void)
    {
        const OptionDef *po;
        int i, expert;
    
        printf("ffmpeg version " FFMPEG_VERSION ", Copyright (c) 2000,2001 Gerard Lantau\n"
               "usage: ffmpeg [[options] -i input_file]... {[options] outfile}...\n"
               "Hyper fast MPEG1/MPEG4/H263/RV and AC3/MPEG audio encoder\n"
               "\n"
               "Main options are:\n");
        for(i=0;i<2;i++) {
            if (i == 1)
                printf("\nAdvanced options are:\n");
            for(po = options; po->name != NULL; po++) {
                char buf[64];
                expert = (po->flags & OPT_EXPERT) != 0;
                if (expert == i) {
                    strcpy(buf, po->name);
                    if (po->flags & HAS_ARG) {
                        strcat(buf, " ");
                        strcat(buf, po->argname);
                    }
                    printf("-%-17s  %s\n", buf, po->help);
                }
            }
        }
    
        exit(1);
    }
    
    const OptionDef options[] = {
    
    Fabrice Bellard's avatar
    Fabrice Bellard committed
        { "L", 0, {(void*)show_licence}, "show license" },
        { "h", 0, {(void*)show_help}, "show help" },
        { "formats", 0, {(void*)show_formats}, "show available formats, codecs, protocols, ..." },
        { "f", HAS_ARG, {(void*)opt_format}, "force format", "fmt" },
        { "i", HAS_ARG, {(void*)opt_input_file}, "input file name", "filename" },
        { "y", OPT_BOOL, {(void*)&file_overwrite}, "overwrite output files" },
        { "map", HAS_ARG | OPT_EXPERT, {(void*)opt_map}, "set input stream mapping", "file:stream" },
        { "t", HAS_ARG, {(void*)opt_recording_time}, "set the recording time", "duration" },
        { "title", HAS_ARG | OPT_STRING, {(void*)&str_title}, "set the title", "string" },
        { "author", HAS_ARG | OPT_STRING, {(void*)&str_author}, "set the author", "string" },
        { "copyright", HAS_ARG | OPT_STRING, {(void*)&str_copyright}, "set the copyright", "string" },
        { "comment", HAS_ARG | OPT_STRING, {(void*)&str_comment}, "set the comment", "string" },
    
    Fabrice Bellard's avatar
    Fabrice Bellard committed
        /* video options */
    
    Fabrice Bellard's avatar
    Fabrice Bellard committed
        { "b", HAS_ARG, {(void*)opt_video_bitrate}, "set video bitrate (in kbit/s)", "bitrate" },
        { "r", HAS_ARG, {(void*)opt_frame_rate}, "set frame rate (in Hz)", "rate" },
        { "s", HAS_ARG, {(void*)opt_frame_size}, "set frame size (WxH or abbreviation)", "size" },
        { "g", HAS_ARG | OPT_EXPERT, {(void*)opt_gop_size}, "set the group of picture size", "gop_size" },
        { "intra", OPT_BOOL | OPT_EXPERT, {(void*)&intra_only}, "use only intra frames"},
        { "vn", OPT_BOOL, {(void*)&video_disable}, "disable video" },
        { "qscale", HAS_ARG | OPT_EXPERT, {(void*)opt_qscale}, "use fixed video quantiser scale (VBR)", "q" },
    #ifdef CONFIG_GRAB
        { "vd", HAS_ARG | OPT_EXPERT, {(void*)opt_video_device}, "set video device", "device" },
    #endif
        { "vcodec", HAS_ARG | OPT_EXPERT, {(void*)opt_video_codec}, "force video codec", "codec" },
        { "me", HAS_ARG | OPT_EXPERT, {(void*)opt_motion_estimation}, "set motion estimation method", 
    
    Fabrice Bellard's avatar
    Fabrice Bellard committed
          "method" },
    
    Fabrice Bellard's avatar
    Fabrice Bellard committed
        { "sameq", OPT_BOOL, {(void*)&same_quality}, 
    
    Fabrice Bellard's avatar
    Fabrice Bellard committed
          "use same video quality as source (implies VBR)" },
        /* audio options */
    
    Fabrice Bellard's avatar
    Fabrice Bellard committed
        { "ab", HAS_ARG, {(void*)opt_audio_bitrate}, "set audio bitrate (in kbit/s)", "bitrate", },
        { "ar", HAS_ARG, {(void*)opt_audio_rate}, "set audio sampling rate (in Hz)", "rate" },
        { "ac", HAS_ARG, {(void*)opt_audio_channels}, "set number of audio channels", "channels" },
        { "an", OPT_BOOL, {(void*)&audio_disable}, "disable audio" },
    #ifdef CONFIG_GRAB
        { "ad", HAS_ARG | OPT_EXPERT, {(void*)opt_audio_device}, "set audio device", "device" },
    #endif
        { "acodec", HAS_ARG | OPT_EXPERT, {(void*)opt_audio_codec}, "force audio codec", "codec" },
        { "deinterlace", OPT_BOOL | OPT_EXPERT, {(void*)&do_deinterlace}, 
    
    Fabrice Bellard's avatar
    Fabrice Bellard committed
        { "benchmark", OPT_BOOL | OPT_EXPERT, {(void*)&do_benchmark}, 
    
    Fabrice Bellard's avatar
    Fabrice Bellard committed
          "add timings for benchmarking" },
    
    Fabrice Bellard's avatar
    Fabrice Bellard committed
    
        { NULL, },
    };
    
    int main(int argc, char **argv)
    {
        int optindex, i;
        const char *opt, *arg;
        const OptionDef *po;
        
        register_all();
    
        if (argc <= 1)
            show_help();
    
        optindex = 1;
        while (optindex < argc) {
            opt = argv[optindex++];
            
            if (opt[0] == '-' && opt[1] != '\0') {
                po = options;
                while (po->name != NULL) {
                    if (!strcmp(opt + 1, po->name))
                        break;
                    po++;
                }
                if (!po->name) {
                    fprintf(stderr, "%s: unrecognized option '%s'\n", argv[0], opt);
                    exit(1);
                }
                arg = NULL;
                if (po->flags & HAS_ARG)
                    arg = argv[optindex++];
                if (po->flags & OPT_STRING) {
                    char *str;
                    str = strdup(arg);
                    *po->u.str_arg = str;
                } else if (po->flags & OPT_BOOL) {
                    *po->u.int_arg = 1;
                } else {
                    po->u.func_arg(arg);
                }
            } else {
                opt_output_file(opt);
            }
        }
    
    
        if (nb_input_files == 0) {
    
    Fabrice Bellard's avatar
    Fabrice Bellard committed
    #ifdef CONFIG_GRAB
    
    Fabrice Bellard's avatar
    Fabrice Bellard committed
            if (nb_output_files != 1) {
                fprintf(stderr, "Only one output file supported when grabbing\n");
                exit(1);
            }
            av_grab(output_files[0]);
    
    Fabrice Bellard's avatar
    Fabrice Bellard committed
    #else
            fprintf(stderr, "Must supply at least one input file\n");
    #endif
    
    Fabrice Bellard's avatar
    Fabrice Bellard committed
        } else {
    
    Fabrice Bellard's avatar
    Fabrice Bellard committed
            INT64 ti;
    
    
    Fabrice Bellard's avatar
    Fabrice Bellard committed
            if (nb_output_files <= 0) {
                fprintf(stderr, "Must supply at least one output file\n");
                exit(1);
            }
    
    Fabrice Bellard's avatar
    Fabrice Bellard committed
            ti = getutime();
    
    Fabrice Bellard's avatar
    Fabrice Bellard committed
            av_encode(output_files, nb_output_files, input_files, nb_input_files, 
                      stream_maps, nb_stream_maps);
    
    Fabrice Bellard's avatar
    Fabrice Bellard committed
            ti = getutime() - ti;
            if (do_benchmark) {
                printf("bench: utime=%0.3fs\n", ti / 1000000.0);
            }
    
    Fabrice Bellard's avatar
    Fabrice Bellard committed
        }
    
        /* close files */
        for(i=0;i<nb_output_files;i++) {
            if (!(output_files[i]->format->flags & AVFMT_NOFILE)) 
                url_fclose(&output_files[i]->pb);
        }
        for(i=0;i<nb_input_files;i++) {
            if (!(input_files[i]->format->flags & AVFMT_NOFILE)) 
                url_fclose(&input_files[i]->pb);
        }
    
        return 0;
    }