avi 封装 C语言代码

【实例简介】avi 封装 C语言代码

用纯C编写的AVI封装代码

【实例截图】

【核心代码】

int avi_write(uint8_t *p_data, int i_data, int b_key)

{

   int ret;

   int fd;

   avi_t *a = &avi;

   int64_t i_pos = ftell(a->f);

   /* chunk header */

   avi_write_fourcc(a, "00dc");

   avi_write_uint32(a, i_data);

   fd = fileno(a->f);

   ret = fwrite(p_data, i_data, 1, a->f);

   //printf("[%s]fwrite ret[%d].\n", __FUNCTION__, ret);

   if (i_data & 0x01)

   {

       /* pad */

       fputc(0, a->f);

   }

   /* Append idx chunk */

   if (a->i_idx_max <= a->i_frame)

   {

       a->i_idx_max = 1000;

       a->idx = realloc(a->idx, a->i_idx_max * 16);

   }

   memcpy(&a->idx[4 * a->i_frame 0], "00dc", 4);

   avi_set_dw(&a->idx[4 * a->i_frame 1], b_key ? AVIIF_KEYFRAME : 0);

   avi_set_dw(&a->idx[4 * a->i_frame 2], i_pos - (a->movi_begin_pos)); /* xiahaiming */

   avi_set_dw(&a->idx[4 * a->i_frame 3], i_data);

   a->i_frame ;

   return ((ret == 1) ? (i_data * 1) : (-1));

}

#ifndef __AVI_API_H__
#define __AVI_API_H__

/*AVI格式，勿删*/

typedef unsigned int FOURCC;
#ifndef DWORD
typedef unsigned int DWORD;
#endif
#ifndef WORD
typedef unsigned short WORD;
#endif

#ifndef BYTE
typedef unsigned char BYTE;
#endif

typedef struct MainAVIHeader
{
    DWORD dwMicroSecPerFrame;   /* 显示每帧所需的时间ns,定义avi的显示速率 */
    DWORD dwMaxBytesPerSec;     /* 最大的数据传输率 */
    DWORD dwPaddingGranularity; /* 记录块的长度需为此值的倍数，通常是2048 */
    DWORD dwFlages;             /* AVI文件的特殊属性，如是否包含索引块，音视频数据是否交叉存储 */
    DWORD dwTotalFrame;         /* 文件中的总帧数 */
    DWORD dwInitialFrames;      /* 说明在开始播放前需要多少帧 */
    DWORD dwStreams;            /* 文件中包含的数据流种类 */
    DWORD dwSuggestedBufferSize; /* 建议使用的缓冲区大小；通常为存储一帧图像以及同步声音所需要的数据之和 */
    DWORD dwWidth;              /* 图像宽 */
    DWORD dwHeight;             /* 图像高 */
    DWORD dwReserved[4];        /* 保留 */
} MAIN_AVI_HEADER, *MAIN_AVI_HEADER_PTR;


typedef struct AVIStreamHeader
{
    FOURCC fccType;             /* 4字节，表示数据流的种类,vids表示视频数据流，auds音频数据流 */
    FOURCC fccHandler;          /* 4字节，表示数据流解压缩的驱动程序代号 */
    DWORD dwFlags;              /* 数据流属性 */
    WORD wPriority;             /* 此数据流的播放优先级 */
    WORD wLanguage;
    DWORD dwInitalFrames;       /* 说明在开始播放前需要多少帧 */
    DWORD dwScale;              /* 数据量，视频每帧的大小或者音频的采样大小 */
    DWORD dwRate;               /* dwScale/dwRate=每秒的采样数 */
    DWORD dwStart;              /* 数据流开始播放的位置，以dwScale为单位 */
    DWORD dwLength;             /* 数据流的数据量，以dwScale为单位 */
    DWORD dwSuggestedBufferSize; /* 建议缓冲区的大小 */
    DWORD dwQuality;            /* 解压缩质量参数，值越大，质量越好 */
    DWORD dwSampleSize;         /* 音频的采样大小 */
    DWORD dwReserved;
    WORD wWidth;
    WORD wHeight;
} AVI_STREAM_HEADER, *AVI_STREAM_HEADER_PTR;


typedef struct bitmapInfoHeader /* 位图信息头部结构 */
{
    DWORD bisize;           /* 位图信息头部大小 */
    DWORD bitWidth;          /* 图像宽度 */
    DWORD biHeight;          /* 图像高度 */
    WORD biPlanes;          /* 目标设备位数，设为1 */
    WORD biBitCount;        /* 单位像素的位数，即图像的位深度 */
    DWORD biCompression;    /* 图像的压缩类型 */
    DWORD biSizeImage;      /* 图像的大小，以字节为单位 */
    DWORD biXPelsPerMeter;   /* 水平方向每米像素数 */
    DWORD biYPelsPerMeter;   /* 垂直方向每米像素数 */
    DWORD biClrUsed;        /* 实际使用的色彩表中的颜色索引数 */
} BITMAP_INFO_HEADER, *BITMAP_INFO_HEADER_PTR;

typedef struct tagRGBQUAD { // rgbq 
    BYTE    rgbBlue; 
    BYTE    rgbGreen; 
    BYTE    rgbRed; 
    BYTE    rgbReserved; 
} RGB_QUAD, *RGB_QUAD_PTR; 

typedef struct bitmapInfo
{
    BITMAP_INFO_HEADER bmiHeader;
    RGB_QUAD bmiColors[1];          /* 颜色表 */
} BITMAP_INFO, *BITMAP_INFO_PTR;


typedef struct
{
    FILE *f;

    float f_fps;
    char fcc[4];

    int i_width;
    int i_height;

    int64_t i_movi;
    int64_t i_movi_end;
    int64_t i_riff;

    int i_frame;
    int i_idx_max;
    uint32_t *idx;
    uint32_t movi_begin_pos;

} avi_t;


typedef struct
{
    pthread_mutex_t aviCtrlMutex;

    int frameRate;
    char *pEncTypeStr;

} AVI_CTRL_PARA, *AVI_CTRL_PARA_PTR;

/* Flags in avih */
#define AVIF_HASINDEX		0x00000010  /* Index at end of file? */
#define AVIF_MUSTUSEINDEX	0x00000020
#define AVIF_ISINTERLEAVED	0x00000100
#define AVIF_TRUSTCKTYPE	0x00000800  /* Use CKType to find key frames? */
#define AVIF_WASCAPTUREFILE 0x00010000  /* the AVI file is a specially allocated file used for capturing real-time video*/
#define AVIF_COPYRIGHTED	0x00020000

#define AVIIF_KEYFRAME 0x00000010L      /* this frame is a key frame.*/

void avi_init(FILE *, float, char fcc[4]);
void avi_end(int width, int height);
int avi_write(uint8_t *p_data, int i_data, int b_key);

#endif