{
FILE * inpf;
int nWrite;
int i,p;
int nalLen;
unsigned char* Buf;
int got_picture, consumed_bytes;
unsigned char *DisplayBuf;
DisplayBuf=(unsigned char *)malloc(60000);
char outfile[] = "test.pgm";
//1.打开输入文件
inpf = fopen("test.264", "rb");
//outf = fopen("out.yuv", "wb");
if(!inpf)
{
goto Decodereturn;
}
nalLen = 0;
Buf = (unsigned char*)calloc ( 1000000, sizeof(char)); //准备解码文件缓冲
//2.注册解码器,并且找到H264解码器
avcodec_init();
avcodec_register_all();
codec = avcodec_find_decoder(CODEC_ID_H264);
if (!codec) {
return 0;
}
//allocate codec context
//分配解码器内存
c = avcodec_alloc_context();
if(!c){
return 0;
}
//open codec
//3.打开解码器
if (avcodec_open(c, codec) < 0) {
return 0;
}
//allocate frame buffer
//分配解码器用的帧缓冲
picture = avcodec_alloc_frame();
if(!picture){
return 0;
}
rgbdatanew = (unsigned char *)malloc(sizeof(unsigned char)*(3 * width * height));
while(!feof(inpf))
{
//4.获取下一个NAL的长度,并且将NAL放入Buf
nalLen = getNextNal(inpf, Buf);
//5.对改NAL解码,解码的YUV数据存在picture中
consumed_bytes= avcodec_decode_video(c, picture, &got_picture, Buf, nalLen);
if(consumed_bytes > 0)
{
//6.将picture中的YUV数据显示或者保存到文件
p=0;
for(i=0; i<c->height; i++)
{
memcpy(DisplayBuf+p,picture->data[0] + i * picture->linesize[0], c->width);
p+=c->width;
}
for(i=0; i<c->height/2; i++)
{
memcpy(DisplayBuf+p,picture->data[1] + i * picture->linesize[1], c->width/2);
p+=c->width/2;
}
for(i=0; i<c->height/2; i++)
{
memcpy(DisplayBuf+p,picture->data[2] + i * picture->linesize[2], c->width/2);
p+=c->width/2;
}
//显示画面
DisplayVideo(DisplayBuf);
}
}
//7.关闭输入文件
if(inpf)
fclose(inpf);
Decodereturn:
//8.关闭解码器,释放解码器内存
if(c) {
avcodec_close(c);
av_free(c);
c = NULL;
}
//9.释放解码画面内存
if(picture) {
av_free(picture);
picture = NULL;
}
//10.释放解码文件缓冲
if(Buf)
{
free(Buf);
Buf = NULL;
}
free(DisplayBuf);
}