apiexample.c
上传用户:lctgjx
上传日期:2022-06-04
资源大小:8887k
文件大小:12k
源码类别:

流媒体/Mpeg4/MP4

开发平台:

Visual C++

  1. /*
  2.  * copyright (c) 2001 Fabrice Bellard
  3.  *
  4.  * This file is part of FFmpeg.
  5.  *
  6.  * FFmpeg is free software; you can redistribute it and/or
  7.  * modify it under the terms of the GNU Lesser General Public
  8.  * License as published by the Free Software Foundation; either
  9.  * version 2.1 of the License, or (at your option) any later version.
  10.  *
  11.  * FFmpeg is distributed in the hope that it will be useful,
  12.  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  13.  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
  14.  * Lesser General Public License for more details.
  15.  *
  16.  * You should have received a copy of the GNU Lesser General Public
  17.  * License along with FFmpeg; if not, write to the Free Software
  18.  * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
  19.  */
  20. /**
  21.  * @file apiexample.c
  22.  * avcodec API use example.
  23.  *
  24.  * Note that this library only handles codecs (mpeg, mpeg4, etc...),
  25.  * not file formats (avi, vob, etc...). See library 'libavformat' for the
  26.  * format handling
  27.  */
  28. #include <stdlib.h>
  29. #include <stdio.h>
  30. #include <string.h>
  31. #include <math.h>
  32. #define PI 3.14159265358979323846
  33. #ifdef HAVE_AV_CONFIG_H
  34. #undef HAVE_AV_CONFIG_H
  35. #endif
  36. #include "avcodec.h"
  37. #define INBUF_SIZE 4096
  38. /*
  39.  * Audio encoding example
  40.  */
  41. void audio_encode_example(const char *filename)
  42. {
  43.     AVCodec *codec;
  44.     AVCodecContext *c= NULL;
  45.     int frame_size, i, j, out_size, outbuf_size;
  46.     FILE *f;
  47.     short *samples;
  48.     float t, tincr;
  49.     uint8_t *outbuf;
  50.     printf("Audio encodingn");
  51.     /* find the MP2 encoder */
  52.     codec = avcodec_find_encoder(CODEC_ID_MP2);
  53.     if (!codec) {
  54.         fprintf(stderr, "codec not foundn");
  55.         exit(1);
  56.     }
  57.     c= avcodec_alloc_context();
  58.     /* put sample parameters */
  59.     c->bit_rate = 64000;
  60.     c->sample_rate = 44100;
  61.     c->channels = 2;
  62.     /* open it */
  63.     if (avcodec_open(c, codec) < 0) {
  64.         fprintf(stderr, "could not open codecn");
  65.         exit(1);
  66.     }
  67.     /* the codec gives us the frame size, in samples */
  68.     frame_size = c->frame_size;
  69.     samples = malloc(frame_size * 2 * c->channels);
  70.     outbuf_size = 10000;
  71.     outbuf = malloc(outbuf_size);
  72.     f = fopen(filename, "wb");
  73.     if (!f) {
  74.         fprintf(stderr, "could not open %sn", filename);
  75.         exit(1);
  76.     }
  77.     /* encode a single tone sound */
  78.     t = 0;
  79.     tincr = 2 * PI * 440.0 / c->sample_rate;
  80.     for(i=0;i<200;i++) {
  81.         for(j=0;j<frame_size;j++) {
  82.             samples[2*j] = (int)(sin(t) * 10000);
  83.             samples[2*j+1] = samples[2*j];
  84.             t += tincr;
  85.         }
  86.         /* encode the samples */
  87.         out_size = avcodec_encode_audio(c, outbuf, outbuf_size, samples);
  88.         fwrite(outbuf, 1, out_size, f);
  89.     }
  90.     fclose(f);
  91.     free(outbuf);
  92.     free(samples);
  93.     avcodec_close(c);
  94.     av_free(c);
  95. }
  96. /*
  97.  * Audio decoding.
  98.  */
  99. void audio_decode_example(const char *outfilename, const char *filename)
  100. {
  101.     AVCodec *codec;
  102.     AVCodecContext *c= NULL;
  103.     int out_size, size, len;
  104.     FILE *f, *outfile;
  105.     uint8_t *outbuf;
  106.     uint8_t inbuf[INBUF_SIZE + FF_INPUT_BUFFER_PADDING_SIZE], *inbuf_ptr;
  107.     printf("Audio decodingn");
  108.     /* find the mpeg audio decoder */
  109.     codec = avcodec_find_decoder(CODEC_ID_MP2);
  110.     if (!codec) {
  111.         fprintf(stderr, "codec not foundn");
  112.         exit(1);
  113.     }
  114.     c= avcodec_alloc_context();
  115.     /* open it */
  116.     if (avcodec_open(c, codec) < 0) {
  117.         fprintf(stderr, "could not open codecn");
  118.         exit(1);
  119.     }
  120.     outbuf = malloc(AVCODEC_MAX_AUDIO_FRAME_SIZE);
  121.     f = fopen(filename, "rb");
  122.     if (!f) {
  123.         fprintf(stderr, "could not open %sn", filename);
  124.         exit(1);
  125.     }
  126.     outfile = fopen(outfilename, "wb");
  127.     if (!outfile) {
  128.         av_free(c);
  129.         exit(1);
  130.     }
  131.     /* decode until eof */
  132.     inbuf_ptr = inbuf;
  133.     for(;;) {
  134.         size = fread(inbuf, 1, INBUF_SIZE, f);
  135.         if (size == 0)
  136.             break;
  137.         inbuf_ptr = inbuf;
  138.         while (size > 0) {
  139.             len = avcodec_decode_audio(c, (short *)outbuf, &out_size,
  140.                                        inbuf_ptr, size);
  141.             if (len < 0) {
  142.                 fprintf(stderr, "Error while decodingn");
  143.                 exit(1);
  144.             }
  145.             if (out_size > 0) {
  146.                 /* if a frame has been decoded, output it */
  147.                 fwrite(outbuf, 1, out_size, outfile);
  148.             }
  149.             size -= len;
  150.             inbuf_ptr += len;
  151.         }
  152.     }
  153.     fclose(outfile);
  154.     fclose(f);
  155.     free(outbuf);
  156.     avcodec_close(c);
  157.     av_free(c);
  158. }
  159. /*
  160.  * Video encoding example
  161.  */
  162. void video_encode_example(const char *filename)
  163. {
  164.     AVCodec *codec;
  165.     AVCodecContext *c= NULL;
  166.     int i, out_size, size, x, y, outbuf_size;
  167.     FILE *f;
  168.     AVFrame *picture;
  169.     uint8_t *outbuf, *picture_buf;
  170.     printf("Video encodingn");
  171.     /* find the mpeg1 video encoder */
  172.     codec = avcodec_find_encoder(CODEC_ID_MPEG1VIDEO);
  173.     if (!codec) {
  174.         fprintf(stderr, "codec not foundn");
  175.         exit(1);
  176.     }
  177.     c= avcodec_alloc_context();
  178.     picture= avcodec_alloc_frame();
  179.     /* put sample parameters */
  180.     c->bit_rate = 400000;
  181.     /* resolution must be a multiple of two */
  182.     c->width = 352;
  183.     c->height = 288;
  184.     /* frames per second */
  185.     c->time_base= (AVRational){1,25};
  186.     c->gop_size = 10; /* emit one intra frame every ten frames */
  187.     c->max_b_frames=1;
  188.     c->pix_fmt = PIX_FMT_YUV420P;
  189.     /* open it */
  190.     if (avcodec_open(c, codec) < 0) {
  191.         fprintf(stderr, "could not open codecn");
  192.         exit(1);
  193.     }
  194.     f = fopen(filename, "wb");
  195.     if (!f) {
  196.         fprintf(stderr, "could not open %sn", filename);
  197.         exit(1);
  198.     }
  199.     /* alloc image and output buffer */
  200.     outbuf_size = 100000;
  201.     outbuf = malloc(outbuf_size);
  202.     size = c->width * c->height;
  203.     picture_buf = malloc((size * 3) / 2); /* size for YUV 420 */
  204.     picture->data[0] = picture_buf;
  205.     picture->data[1] = picture->data[0] + size;
  206.     picture->data[2] = picture->data[1] + size / 4;
  207.     picture->linesize[0] = c->width;
  208.     picture->linesize[1] = c->width / 2;
  209.     picture->linesize[2] = c->width / 2;
  210.     /* encode 1 second of video */
  211.     for(i=0;i<25;i++) {
  212.         fflush(stdout);
  213.         /* prepare a dummy image */
  214.         /* Y */
  215.         for(y=0;y<c->height;y++) {
  216.             for(x=0;x<c->width;x++) {
  217.                 picture->data[0][y * picture->linesize[0] + x] = x + y + i * 3;
  218.             }
  219.         }
  220.         /* Cb and Cr */
  221.         for(y=0;y<c->height/2;y++) {
  222.             for(x=0;x<c->width/2;x++) {
  223.                 picture->data[1][y * picture->linesize[1] + x] = 128 + y + i * 2;
  224.                 picture->data[2][y * picture->linesize[2] + x] = 64 + x + i * 5;
  225.             }
  226.         }
  227.         /* encode the image */
  228.         out_size = avcodec_encode_video(c, outbuf, outbuf_size, picture);
  229.         printf("encoding frame %3d (size=%5d)n", i, out_size);
  230.         fwrite(outbuf, 1, out_size, f);
  231.     }
  232.     /* get the delayed frames */
  233.     for(; out_size; i++) {
  234.         fflush(stdout);
  235.         out_size = avcodec_encode_video(c, outbuf, outbuf_size, NULL);
  236.         printf("write frame %3d (size=%5d)n", i, out_size);
  237.         fwrite(outbuf, 1, out_size, f);
  238.     }
  239.     /* add sequence end code to have a real mpeg file */
  240.     outbuf[0] = 0x00;
  241.     outbuf[1] = 0x00;
  242.     outbuf[2] = 0x01;
  243.     outbuf[3] = 0xb7;
  244.     fwrite(outbuf, 1, 4, f);
  245.     fclose(f);
  246.     free(picture_buf);
  247.     free(outbuf);
  248.     avcodec_close(c);
  249.     av_free(c);
  250.     av_free(picture);
  251.     printf("n");
  252. }
  253. /*
  254.  * Video decoding example
  255.  */
  256. void pgm_save(unsigned char *buf,int wrap, int xsize,int ysize,char *filename)
  257. {
  258.     FILE *f;
  259.     int i;
  260.     f=fopen(filename,"w");
  261.     fprintf(f,"P5n%d %dn%dn",xsize,ysize,255);
  262.     for(i=0;i<ysize;i++)
  263.         fwrite(buf + i * wrap,1,xsize,f);
  264.     fclose(f);
  265. }
  266. void video_decode_example(const char *outfilename, const char *filename)
  267. {
  268.     AVCodec *codec;
  269.     AVCodecContext *c= NULL;
  270.     int frame, size, got_picture, len;
  271.     FILE *f;
  272.     AVFrame *picture;
  273.     uint8_t inbuf[INBUF_SIZE + FF_INPUT_BUFFER_PADDING_SIZE], *inbuf_ptr;
  274.     char buf[1024];
  275.     /* set end of buffer to 0 (this ensures that no overreading happens for damaged mpeg streams) */
  276.     memset(inbuf + INBUF_SIZE, 0, FF_INPUT_BUFFER_PADDING_SIZE);
  277.     printf("Video decodingn");
  278.     /* find the mpeg1 video decoder */
  279.     codec = avcodec_find_decoder(CODEC_ID_MPEG1VIDEO);
  280.     if (!codec) {
  281.         fprintf(stderr, "codec not foundn");
  282.         exit(1);
  283.     }
  284.     c= avcodec_alloc_context();
  285.     picture= avcodec_alloc_frame();
  286.     if(codec->capabilities&CODEC_CAP_TRUNCATED)
  287.         c->flags|= CODEC_FLAG_TRUNCATED; /* we do not send complete frames */
  288.     /* For some codecs, such as msmpeg4 and mpeg4, width and height
  289.        MUST be initialized there because this information is not
  290.        available in the bitstream. */
  291.     /* open it */
  292.     if (avcodec_open(c, codec) < 0) {
  293.         fprintf(stderr, "could not open codecn");
  294.         exit(1);
  295.     }
  296.     /* the codec gives us the frame size, in samples */
  297.     f = fopen(filename, "rb");
  298.     if (!f) {
  299.         fprintf(stderr, "could not open %sn", filename);
  300.         exit(1);
  301.     }
  302.     frame = 0;
  303.     for(;;) {
  304.         size = fread(inbuf, 1, INBUF_SIZE, f);
  305.         if (size == 0)
  306.             break;
  307.         /* NOTE1: some codecs are stream based (mpegvideo, mpegaudio)
  308.            and this is the only method to use them because you cannot
  309.            know the compressed data size before analysing it.
  310.            BUT some other codecs (msmpeg4, mpeg4) are inherently frame
  311.            based, so you must call them with all the data for one
  312.            frame exactly. You must also initialize 'width' and
  313.            'height' before initializing them. */
  314.         /* NOTE2: some codecs allow the raw parameters (frame size,
  315.            sample rate) to be changed at any frame. We handle this, so
  316.            you should also take care of it */
  317.         /* here, we use a stream based decoder (mpeg1video), so we
  318.            feed decoder and see if it could decode a frame */
  319.         inbuf_ptr = inbuf;
  320.         while (size > 0) {
  321.             len = avcodec_decode_video(c, picture, &got_picture,
  322.                                        inbuf_ptr, size);
  323.             if (len < 0) {
  324.                 fprintf(stderr, "Error while decoding frame %dn", frame);
  325.                 exit(1);
  326.             }
  327.             if (got_picture) {
  328.                 printf("saving frame %3dn", frame);
  329.                 fflush(stdout);
  330.                 /* the picture is allocated by the decoder. no need to
  331.                    free it */
  332.                 snprintf(buf, sizeof(buf), outfilename, frame);
  333.                 pgm_save(picture->data[0], picture->linesize[0],
  334.                          c->width, c->height, buf);
  335.                 frame++;
  336.             }
  337.             size -= len;
  338.             inbuf_ptr += len;
  339.         }
  340.     }
  341.     /* some codecs, such as MPEG, transmit the I and P frame with a
  342.        latency of one frame. You must do the following to have a
  343.        chance to get the last frame of the video */
  344.     len = avcodec_decode_video(c, picture, &got_picture,
  345.                                NULL, 0);
  346.     if (got_picture) {
  347.         printf("saving last frame %3dn", frame);
  348.         fflush(stdout);
  349.         /* the picture is allocated by the decoder. no need to
  350.            free it */
  351.         snprintf(buf, sizeof(buf), outfilename, frame);
  352.         pgm_save(picture->data[0], picture->linesize[0],
  353.                  c->width, c->height, buf);
  354.         frame++;
  355.     }
  356.     fclose(f);
  357.     avcodec_close(c);
  358.     av_free(c);
  359.     av_free(picture);
  360.     printf("n");
  361. }
  362. int main(int argc, char **argv)
  363. {
  364.     const char *filename;
  365.     /* must be called before using avcodec lib */
  366.     avcodec_init();
  367.     /* register all the codecs */
  368.     avcodec_register_all();
  369.     if (argc <= 1) {
  370.         audio_encode_example("/tmp/test.mp2");
  371.         audio_decode_example("/tmp/test.sw", "/tmp/test.mp2");
  372.         video_encode_example("/tmp/test.mpg");
  373.         filename = "/tmp/test.mpg";
  374.     } else {
  375.         filename = argv[1];
  376.     }
  377.     //    audio_decode_example("/tmp/test.sw", filename);
  378.     video_decode_example("/tmp/test%d.pgm", filename);
  379.     return 0;
  380. }