206 lines
7.3 KiB
C
206 lines
7.3 KiB
C
#ifndef __AVI_H
|
||
#define __AVI_H
|
||
#include <stm32f4xx.h>
|
||
//////////////////////////////////////////////////////////////////////////////////
|
||
// 本程序只供学习使用,未经作者许可,不得用于其它任何用途
|
||
// ALIENTEK STM32F407开发板
|
||
// AVI视频格式解析 代码
|
||
// 正点原子@ALIENTEK
|
||
// 技术论坛:www.openedv.com
|
||
// 创建日期:2014/7/20
|
||
// 版本:V1.0
|
||
// 版权所有,盗版必究。
|
||
// Copyright(C) 广州市星翼电子科技有限公司 2009-2019
|
||
// All rights reserved
|
||
//*******************************************************************************
|
||
// 修改信息
|
||
// 无
|
||
//////////////////////////////////////////////////////////////////////////////////
|
||
|
||
// 错误类型
|
||
typedef enum {
|
||
AVI_OK = 0, // 0,成功
|
||
AVI_RIFF_ERR, // 1,RIFF ID读取失败
|
||
AVI_AVI_ERR, // 2,AVI ID读取失败
|
||
AVI_LIST_ERR, // 3,LIST ID读取失败
|
||
AVI_HDRL_ERR, // 4,HDRL ID读取失败
|
||
AVI_AVIH_ERR, // 5,AVIH ID读取失败
|
||
AVI_STRL_ERR, // 6,STRL ID读取失败
|
||
AVI_STRH_ERR, // 7,STRH ID读取失败
|
||
AVI_STRF_ERR, // 8,STRF ID读取失败
|
||
AVI_MOVI_ERR, // 9,MOVI ID读取失败
|
||
AVI_FORMAT_ERR, // 10,格式错误
|
||
AVI_STREAM_ERR, // 11,流错误
|
||
} AVISTATUS;
|
||
|
||
#define AVI_RIFF_ID 0X46464952
|
||
#define AVI_AVI_ID 0X20495641
|
||
#define AVI_LIST_ID 0X5453494C
|
||
#define AVI_HDRL_ID 0X6C726468 // 信息块标志
|
||
#define AVI_MOVI_ID 0X69766F6D // 数据块标志
|
||
#define AVI_STRL_ID 0X6C727473 // strl标志
|
||
|
||
#define AVI_AVIH_ID 0X68697661 // avih子块∈AVI_HDRL_ID
|
||
#define AVI_STRH_ID 0X68727473 // strh(流头)子块∈AVI_STRL_ID
|
||
#define AVI_STRF_ID 0X66727473 // strf(流格式)子块∈AVI_STRL_ID
|
||
#define AVI_STRD_ID 0X64727473 // strd子块∈AVI_STRL_ID (可选的)
|
||
|
||
#define AVI_VIDS_STREAM 0X73646976 // 视频流
|
||
#define AVI_AUDS_STREAM 0X73647561 // 音频流
|
||
|
||
#define AVI_VIDS_FLAG 0X6463 // 视频流标志
|
||
#define AVI_AUDS_FLAG 0X7762 // 音频流标志
|
||
//////////////////////////////////////////////////////////////////////////////////////////
|
||
|
||
#define AVI_FORMAT_MJPG 0X47504A4D
|
||
|
||
// AVI 信息结构体
|
||
// 将一些重要的数据,存放在这里,方便解码
|
||
typedef struct __packed {
|
||
u32 SecPerFrame; // 视频帧间隔时间(单位为us)
|
||
u32 TotalFrame; // 文件总帧数
|
||
u32 Width; // 图像宽
|
||
u32 Height; // 图像高
|
||
u32 SampleRate; // 音频采样率
|
||
u16 Channels; // 声道数,一般为2,表示立体声
|
||
u16 AudioBufSize; // 音频缓冲区大小
|
||
u16 AudioType; // 音频类型:0X0001=PCM;0X0050=MP2;0X0055=MP3;0X2000=AC3;
|
||
u16 StreamID; // 流类型ID,StreamID=='dc'==0X6463 /StreamID=='wb'==0X7762
|
||
u32 StreamSize; // 流大小,必须是偶数,如果读取到为奇数,则加1.补为偶数.
|
||
u8 *VideoFLAG; // 视频帧标记,VideoFLAG="00dc"/"01dc"
|
||
u8 *AudioFLAG; // 音频帧标记,AudioFLAG="00wb"/"01wb"
|
||
} AVI_INFO;
|
||
|
||
// extern AVI_INFO avix; //avi文件相关信息
|
||
|
||
//////////////////////////////////////////////////////////////////////////////////////////
|
||
// AVI 块信息
|
||
typedef struct {
|
||
u32 RiffID; // RiffID=='RIFF'==0X61766968
|
||
u32 FileSize; // AVI文件大小(不包含最初的8字节,也RIFFID和FileSize不计算在内)
|
||
u32 AviID; // AviID=='AVI '==0X41564920
|
||
} AVI_HEADER;
|
||
|
||
// AVI 块信息
|
||
typedef struct {
|
||
u32 FrameID; // 帧ID,FrameID=='RIFF'==0X61766968
|
||
u32 FrameSize; // 帧大小
|
||
} FRAME_HEADER;
|
||
|
||
// LIST 块信息
|
||
typedef struct {
|
||
u32 ListID; // ListID=='LIST'==0X4c495354
|
||
u32 BlockSize; // 块大小(不包含最初的8字节,也ListID和BlockSize不计算在内)
|
||
u32 ListType; // LIST子块类型:hdrl(信息块)/movi(数据块)/idxl(索引块,非必须,是可选的)
|
||
} LIST_HEADER;
|
||
|
||
// avih 子块信息
|
||
typedef struct {
|
||
u32 BlockID; // 块标志:avih==0X61766968
|
||
u32 BlockSize; // 块大小(不包含最初的8字节,也就是BlockID和BlockSize不计算在内)
|
||
u32 SecPerFrame; // 视频帧间隔时间(单位为us)
|
||
u32 MaxByteSec; // 最大数据传输率,字节/秒
|
||
u32 PaddingGranularity; // 数据填充的粒度
|
||
u32 Flags; // AVI文件的全局标记,比如是否含有索引块等
|
||
u32 TotalFrame; // 文件总帧数
|
||
u32 InitFrames; // 为交互格式指定初始帧数(非交互格式应该指定为0)
|
||
u32 Streams; // 包含的数据流种类个数,通常为2
|
||
u32 RefBufSize; // 建议读取本文件的缓存大小(应能容纳最大的块)默认可能是1M字节!!!
|
||
u32 Width; // 图像宽
|
||
u32 Height; // 图像高
|
||
u32 Reserved[4]; // 保留
|
||
} AVIH_HEADER;
|
||
|
||
// strh 流头子块信息(strh∈strl)
|
||
typedef struct {
|
||
u32 BlockID; // 块标志:strh==0X73747268
|
||
u32 BlockSize; // 块大小(不包含最初的8字节,也就是BlockID和BlockSize不计算在内)
|
||
u32 StreamType; // 数据流种类,vids(0X73646976):视频;auds(0X73647561):音频
|
||
u32 Handler; // 指定流的处理者,对于音视频来说就是解码器,比如MJPG/H264之类的.
|
||
u32 Flags; // 标记:是否允许这个流输出?调色板是否变化?
|
||
u16 Priority; // 流的优先级(当有多个相同类型的流时优先级最高的为默认流)
|
||
u16 Language; // 音频的语言代号
|
||
u32 InitFrames; // 为交互格式指定初始帧数
|
||
u32 Scale; // 数据量, 视频每桢的大小或者音频的采样大小
|
||
u32 Rate; // Scale/Rate=每秒采样数
|
||
u32 Start; // 数据流开始播放的位置,单位为Scale
|
||
u32 Length; // 数据流的数据量,单位为Scale
|
||
u32 RefBufSize; // 建议使用的缓冲区大小
|
||
u32 Quality; // 解压缩质量参数,值越大,质量越好
|
||
u32 SampleSize; // 音频的样本大小
|
||
struct // 视频帧所占的矩形
|
||
{
|
||
short Left;
|
||
short Top;
|
||
short Right;
|
||
short Bottom;
|
||
} Frame;
|
||
} STRH_HEADER;
|
||
|
||
// BMP结构体
|
||
typedef struct {
|
||
u32 BmpSize; // bmp结构体大小,包含(BmpSize在内)
|
||
long Width; // 图像宽
|
||
long Height; // 图像高
|
||
u16 Planes; // 平面数,必须为1
|
||
u16 BitCount; // 像素位数,0X0018表示24位
|
||
u32 Compression; // 压缩类型,比如:MJPG/H264等
|
||
u32 SizeImage; // 图像大小
|
||
long XpixPerMeter; // 水平分辨率
|
||
long YpixPerMeter; // 垂直分辨率
|
||
u32 ClrUsed; // 实际使用了调色板中的颜色数,压缩格式中不使用
|
||
u32 ClrImportant; // 重要的颜色
|
||
} BMP_HEADER;
|
||
|
||
// 颜色表
|
||
typedef struct {
|
||
u8 rgbBlue; // 蓝色的亮度(值范围为0-255)
|
||
u8 rgbGreen; // 绿色的亮度(值范围为0-255)
|
||
u8 rgbRed; // 红色的亮度(值范围为0-255)
|
||
u8 rgbReserved; // 保留,必须为0
|
||
} AVIRGBQUAD;
|
||
|
||
// 对于strh,如果是视频流,strf(流格式)使STRH_BMPHEADER块
|
||
typedef struct {
|
||
u32 BlockID; // 块标志,strf==0X73747266
|
||
u32 BlockSize; // 块大小(不包含最初的8字节,也就是BlockID和本BlockSize不计算在内)
|
||
BMP_HEADER bmiHeader; // 位图信息头
|
||
AVIRGBQUAD bmColors[1]; // 颜色表
|
||
} STRF_BMPHEADER;
|
||
|
||
// 对于strh,如果是音频流,strf(流格式)使STRH_WAVHEADER块
|
||
typedef struct {
|
||
u32 BlockID; // 块标志,strf==0X73747266
|
||
u32 BlockSize; // 块大小(不包含最初的8字节,也就是BlockID和本BlockSize不计算在内)
|
||
u16 FormatTag; // 格式标志:0X0001=PCM,0X0055=MP3...
|
||
u16 Channels; // 声道数,一般为2,表示立体声
|
||
u32 SampleRate; // 音频采样率
|
||
u32 BaudRate; // 波特率
|
||
u16 BlockAlign; // 数据块对齐标志
|
||
u16 Size; // 该结构大小
|
||
} STRF_WAVHEADER;
|
||
|
||
#define MAKEWORD(ptr) \
|
||
(u16)(((u16) * ((u8 *)(ptr)) << 8) | (u16) * (u8 *)((ptr) + 1))
|
||
#define MAKEDWORD(ptr) \
|
||
(u32)(((u16) * (u8 *)(ptr) | (((u16) * (u8 *)(ptr + 1)) << 8) | \
|
||
(((u16) * (u8 *)(ptr + 2)) << 16) | \
|
||
(((u16) * (u8 *)(ptr + 3)) << 24)))
|
||
|
||
AVISTATUS avi_init(u8 *buf, int size); // 初始化avi解码器
|
||
u16 avi_srarch_id(u8 *buf, int size, u8 *id); // 查找ID,ID必须是4个字节长度
|
||
AVISTATUS avi_get_streaminfo(u8 *buf); // 获取流信息
|
||
|
||
// 播放视频,loop==1,循环播放
|
||
int video_play(char *pname, int loop);
|
||
|
||
u8 Avih_Parser(u8 *buffer);
|
||
u8 Strl_Parser(u8 *buffer);
|
||
u8 Strf_Parser(u8 *buffer);
|
||
|
||
u16 Search_Movi(u8 *buffer);
|
||
u16 Search_Fram(u8 *buffer);
|
||
u32 ReadUnit(u8 *buffer, u8 index, u8 Bytes, u8 Format);
|
||
|
||
#endif
|