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 DWORDtypedef unsigned int DWORD;#endif#ifndef WORDtypedef unsigned short WORD;#endif#ifndef BYTEtypedef unsigned char BYTE;#endiftypedef 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_HASINDEX0x00000010 /* Index at end of file? */#define AVIF_MUSTUSEINDEX0x00000020#define AVIF_ISINTERLEAVED0x00000100#define AVIF_TRUSTCKTYPE0x00000800 /* 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_COPYRIGHTED0x00020000#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
评论