利用ffmpeg解码h264流的代码

朴灿烈づ我的快乐病毒、 2022-08-13 13:55 291阅读 0赞

这里也直接给出代码：

h264dec.h:

**\[cpp\]** [view plain][][copy][view plain]

1.  \#pragma once  
2.  \#include "tdll.h"  
3.  \#include "avcodec.h"  
4.  \#include "postprocess.h"  
5.  //\#include "EMVideoCodec.h"  
6.    
7.  **class** h264dec /\*: public IH264Decoder\*/  
8.  \{  
9.  **public**:  
10.     **virtual** **bool** InitH264Deocder(**int** width,**int** height);  
11.     **virtual** **bool** H264Decode(unsigned **char** \* inbuf, **const** **int** & inlen,unsigned **char** \* outbuf,**int** & outlen);  
12.     **virtual** **void** StopH264Decoder();  
13.     **virtual** **void** ReleaseConnection();  
14.   
15. **public**:  
16.     h264dec(**void**);  
17.     **virtual** ~h264dec(**void**);  
18. **private**:  
19.     Tdll \*dll;  
20.   
21.     **bool** LoadDllFun();  
22.     **void** (\*avcodec\_init)(**void**);  
23.     **void** (\*avcodec\_register\_all)(**void**);  
24.     AVCodecContext\* (\*avcodec\_alloc\_context)(**void**);  
25.     AVFrame\* (\*avcodec\_alloc\_frame)(**void**);  
26.     AVCodec \*(\*avcodec\_find\_decoder)(**enum** CodecID id);  
27.     **int** (\*avcodec\_decode\_video)(AVCodecContext \*avctx, AVFrame \*picture,**int** \*got\_picture\_ptr,  
28.                                 uint8\_t \*buf, **int** buf\_size);  
29.     **int**  (\*avcodec\_open)(AVCodecContext \*avctx, AVCodec \*codec);  
30.     **int**  (\*avcodec\_close)(AVCodecContext \*avctx);  
31.     **void** (\*av\_free)(**void** \*ptr);       
32.       
33.     **bool** InitPostproc(**int** w,**int** h);  
34.     **void** ClosePostproc();  
35.     pp\_context\_t \*(\*pp\_get\_context)(**int** width, **int** height, **int** flags);  
36.     **void** (\*pp\_free\_context)(pp\_context\_t \*ppContext);  
37.     **void** (\*pp\_free\_mode)(pp\_mode\_t \*mode);  
38.     pp\_mode\_t \*(\*pp\_get\_mode\_by\_name\_and\_quality)(**char** \*name, **int** quality);  
39.     **void**  (\*pp\_postprocess)(uint8\_t \* src\[3\], **int** srcStride\[3\],  
40.                  uint8\_t \* dst\[3\], **int** dstStride\[3\],  
41.                  **int** horizontalSize, **int** verticalSize,  
42.                  QP\_STORE\_T \*QP\_store,  **int** QP\_stride,  
43.          pp\_mode\_t \*mode, pp\_context\_t \*ppContext, **int** pict\_type);  
44. **private**:  
45.     AVCodec         \*pdec;  
46.     AVCodecContext  \*pdecContext;  
47.     AVFrame         \*pdecFrame;  
48.     **int**             m\_width;  
49.     **int**             m\_height;  
50.   
51.     Tdll\* prodll;  
52.     pp\_context\_t \*pp\_context;  
53.     pp\_mode\_t    \*pp\_mode;  
54. \};

h264dec.cpp:

**\[cpp\]** [view plain][][copy][view plain]

1.  \#include "StdAfx.h"  
2.  \#include ".\\h264dec.h"  
3.    
4.  h264dec::h264dec(**void**)  
5.  :dll(NULL)  
6.  ,pdec(NULL)  
7.  ,pdecContext(NULL)  
8.  ,pdecFrame(NULL)  
9.  ,pp\_context(NULL)  
10. ,pp\_mode(NULL)  
11. ,prodll(NULL)  
12. \{  
13. \}  
14.   
15. h264dec::~h264dec(**void**)  
16. \{  
17. \}  
18.   
19. **bool** h264dec::LoadDllFun()  
20. \{  
21.     dll=**new** Tdll(L"libavcodec.dll");  
22.     dll->loadFunction((**void**\*\*)&avcodec\_init,"avcodec\_init");  
23.     dll->loadFunction((**void**\*\*)&avcodec\_register\_all,"avcodec\_register\_all");  
24.     dll->loadFunction((**void**\*\*)&avcodec\_alloc\_context,"avcodec\_alloc\_context");  
25.     dll->loadFunction((**void**\*\*)&avcodec\_alloc\_frame,"avcodec\_alloc\_frame");  
26.     dll->loadFunction((**void**\*\*)&avcodec\_find\_decoder,"avcodec\_find\_decoder");  
27.     dll->loadFunction((**void**\*\*)&avcodec\_open,"avcodec\_open");   
28.     dll->loadFunction((**void**\*\*)&avcodec\_decode\_video,"avcodec\_decode\_video");  
29.     dll->loadFunction((**void**\*\*)&avcodec\_close,"avcodec\_close");  
30.     dll->loadFunction((**void**\*\*)&av\_free,"av\_free");  
31.     **if** (!dll->ok)  
32.         **return** **false**;  
33.   
34.     prodll = **new** Tdll(L"postproc.dll");  
35.     prodll->loadFunction((**void**\*\*)&pp\_get\_context,"pp\_get\_context");  
36.     prodll->loadFunction((**void**\*\*)&pp\_free\_context,"pp\_free\_context");  
37.     prodll->loadFunction((**void**\*\*)&pp\_free\_mode,"pp\_free\_mode");  
38.     prodll->loadFunction((**void**\*\*)&pp\_get\_mode\_by\_name\_and\_quality,"pp\_get\_mode\_by\_name\_and\_quality");  
39.     prodll->loadFunction((**void**\*\*)&pp\_postprocess,"pp\_postprocess");  
40.     **if**(!prodll->ok)  
41.         **return** **false**;  
42.   
43.     avcodec\_init();  
44.     avcodec\_register\_all();  
45.     **return** **true**;  
46. \}  
47.   
48. **bool** h264dec::InitH264Deocder(**int** width,**int** height)  
49. \{  
50.     **if**(!LoadDllFun())  
51.         **return** **false**;  
52.     **if**(!InitPostproc(width,height))  
53.         **return** **false**;  
54.   
55.     m\_width=width;  
56.     m\_height=height;  
57.     pdec = avcodec\_find\_decoder(CODEC\_ID\_H264);  
58.     **if** (pdec == NULL )    
59.         **return** **false**;  
60.   
61.     pdecContext = avcodec\_alloc\_context();  
62.     pdecFrame = avcodec\_alloc\_frame();  
63.   
64.     pdecContext->width  = width;  
65.     pdecContext->height = height;  
66.     pdecContext->pix\_fmt = PIX\_FMT\_YUV420P;  
67.     /\* open it \*/  
68.     **if** (avcodec\_open(pdecContext, pdec) < 0)   
69.     \{  
70.         **return** **false**;  
71.     \}  
72.     **return** **true**;  
73. \}  
74.   
75. **bool** h264dec::InitPostproc(**int** w,**int** h)  
76. \{  
77.     **int** i\_flags = 0;  
78.     i\_flags |= PP\_CPU\_CAPS\_MMX | PP\_CPU\_CAPS\_MMX2 | PP\_FORMAT\_420;  
79.     pp\_context = pp\_get\_context( w, h, i\_flags );  
80.     **if**(!pp\_context)  
81.         **return** **false**;  
82.     pp\_mode = pp\_get\_mode\_by\_name\_and\_quality( "default", 6 );  
83.     **if**(!pp\_mode)  
84.         **return** **false**;  
85.     **return** **true**;  
86. \}  
87.   
88. **bool** h264dec::H264Decode(unsigned **char** \* inbuf, **const** **int** & inlen,unsigned **char** \* outbuf,**int** & outlen)  
89. \{  
90.     **int** got\_frame;  
91.     **BYTE**\* showImage\[3\];  
92.     **int** showheight\[3\],showLx\[3\];  
93.   
94.     **int** len;  
95.     len=avcodec\_decode\_video(pdecContext, pdecFrame, &got\_frame, inbuf, inlen);  
96.     **if**(len < 0)  
97.         **return** **false**;  
98.   
99.     **if**(got\_frame)  
100.     \{  
101.         showImage\[0\]=outbuf;  
102.         showImage\[1\]=showImage\[0\]+m\_width\*m\_height;  
103.         showImage\[2\]=showImage\[1\]+m\_width\*m\_height/4;  
104.         showLx\[0\]=m\_width;showLx\[1\]=m\_width>>1;showLx\[2\]=m\_width>>1;  
105.         showheight\[0\]=m\_height;showheight\[1\]=m\_height>>1;showheight\[2\]=m\_height>>1;  
106.         pp\_postprocess(pdecFrame->data,pdecFrame->linesize,showImage,showLx,m\_width,m\_height,pdecFrame->qscale\_table,  
107.             pdecFrame->qstride,pp\_mode,pp\_context,pdecFrame->pict\_type);  
108.         //GetImage( pdecFrame->data,  
109.         //          showImage,  
110.         //          pdecFrame->linesize,  
111.         //          showLx,  
112.         //          showheight);  
113.         outlen=m\_width\*m\_height\*3/2;  
114.     \}  
115.     **else**  
116.     \{  
117.         outlen = 0;  
118.     \}  
119.   
120.     **return** **true**;  
121. \}  
122.   
123. **void** h264dec::StopH264Decoder()  
124. \{  
125.     **if** (pdecContext != NULL)   
126.     \{       
127.         avcodec\_close(pdecContext);  
128.         av\_free( pdecContext );  
129.         pdecContext = NULL;  
130.         **if**(pdecFrame)\{  
131.             av\_free(pdecFrame);  
132.             pdecFrame = NULL;  
133.         \}  
134.     \}  
135.     **if**(dll)\{  
136.         **delete** dll;  
137.         dll=0;  
138.     \}  
139.   
140.     ClosePostproc();  
141. \}  
142.   
143. **void** h264dec::ClosePostproc()  
144. \{  
145.     **if**(pp\_mode)\{  
146.         pp\_free\_mode( pp\_mode );  
147.         pp\_mode=0;  
148.     \}  
149.     **if**(pp\_context)\{  
150.         pp\_free\_context(pp\_context);  
151.         pp\_context=0;  
152.     \}  
153.     **if**(prodll)\{  
154.         **delete** prodll;  
155.         prodll=0;  
156.     \}  
157. \}  
158.   
159. **void** h264dec::ReleaseConnection()  
160. \{  
161.     **delete** **this**;  
162. \}

tdll.h:

**\[cpp\]** [view plain][][copy][view plain]

1.  \#ifndef \_TDLL\_  
2.  \#define \_TDLL\_  
3.    
4.  **class** Tdll  
5.  \{  
6.  **private**:  
7.      **HMODULE** hdll;  
8.      **void** loadDll(**const** **char** \*dllName);  
9.  **public**:  
10.     **bool** ok;  
11.     Tdll(**const** **TCHAR** \*dllName1)  
12.     \{  
13.         hdll=LoadLibrary(dllName1);  
14.         **if** (!hdll)  
15.         \{  
16.             hdll=NULL;  
17.         \}  
18.         ok=(hdll!=NULL);  
19.     \};  
20.     ~Tdll()  
21.     \{  
22.         **if** (hdll)  
23.             FreeLibrary(hdll);  
24.     \}  
25.     **void** loadFunction(**void** \*\*fnc,**const** **char** \*name)  
26.     \{  
27.         \*fnc=GetProcAddress(hdll,name);  
28.         ok&=(\*fnc!=NULL);  
29.     \};  
30. \};    
31.   
32. \#endif

main.cpp:

**\[cpp\]** [view plain][][copy][view plain]

1.  \#include "stdafx.h"  
2.  \#include "h264dec.h"  
3.  \#include "postprocess.h"  
4.    
5.  \#define INBUF\_SIZE 100 \* 1024;  
6.    
7.    
8.  **static** **int** FindStartCode (unsigned **char** \*Buf, **int** zeros\_in\_startcode)  
9.  \{  
10.     **int** info;  
11.     **int** i;  
12.   
13.     info = 1;  
14.     **for** (i = 0; i < zeros\_in\_startcode; i++)  
15.     \{  
16.         **if**(Buf\[i\] != 0)  
17.             info = 0;  
18.     \}  
19.   
20.     **if**(Buf\[i\] != 1)  
21.         info = 0;  
22.     **return** info;  
23. \}  
24.   
25. **static** **bool** Check\_StartCode(unsigned **char** \*Buf, **int** pos)  
26. \{  
27.     **int** info3 = 0;  
28.   
29.     info3 = FindStartCode(&Buf\[pos-4\], 3);  
30.     **return** info3 == 1;  
31.   
32. \}  
33.   
34. **static** **int** getNextNal(**FILE**\* inpf, unsigned **char**\* Buf)  
35. \{  
36.     **int** pos = 0;  
37.     **int** StartCodeFound = 0;  
38.     **int** info2 = 0;  
39.     **int** info3 = 0;  
40.   
41.     **int** nCount = 0;  
42.     **while**(!feof(inpf) && ++nCount <= 4)  
43.     \{  
44.         Buf\[pos++\]=fgetc(inpf);  
45.     \}  
46.   
47.     **if**(!Check\_StartCode(Buf, pos))  
48.     \{  
49.         **return** 0;  
50.     \}  
51.   
52.   
53.     **while**(!feof(inpf) && (Buf\[pos++\]=fgetc(inpf))==0);  
54.   
55.     **while** (!StartCodeFound)  
56.     \{  
57.         **if** (feof (inpf))  
58.         \{  
59.             //          return -1;  
60.             **return** pos-1;  
61.         \}  
62.         Buf\[pos++\] = fgetc (inpf);  
63.   
64.         StartCodeFound = Check\_StartCode(Buf, pos);  
65.     \}  
66.   
67.     fseek (inpf, -4, SEEK\_CUR);  
68.     **return** pos - 4;  
69. \}  
70.   
71. **int** main(**int** argc, **char**\* argv\[\])  
72. \{  
73.     **if** (argc != 5)  
74.     \{  
75.         printf("please input: PP\_Demo.exe filename1\[input\] Width Height filename2\[output\]\\n");  
76.     \}  
77.       
78.     //params set  
79.     unsigned **short** usWidth = atoi(argv\[2\]);  
80.     unsigned **short** usHeight = atoi(argv\[3\]);  
81.       
82.     //create dec&pp  
83.     h264dec \*pdec = **new** h264dec;  
84.     **if**(!pdec->InitH264Deocder(usWidth, usHeight))  
85.     \{  
86.         **return** **false**;  
87.     \}  
88.   
89.   
90.   
91.     unsigned **char** \*p\_In\_Frame = **new** unsigned **char**\[usWidth \* usHeight \* 3/2\];  
92.     unsigned **char** \*p\_Out\_Frame = **new** unsigned **char**\[usWidth \* usHeight \* 3/2\];  
93.     **FILE**\* ifp = fopen(argv\[1\],"rb");  
94.     **FILE**\* ofp = fopen(argv\[4\],"wb");  
95.   
96.     **bool** b\_continue = **true**;  
97.     **int** nReadUnit = usWidth \* usHeight \* 3/2;  
98.     **while**(!feof(ifp))  
99.     \{  
100.         **int** nCount = getNextNal(ifp, p\_In\_Frame);  
101.   
102.         **if**(nCount == 0)  
103.         \{  
104.             **continue**;  
105.         \}  
106.           
107.         unsigned **char** \*ptr = p\_In\_Frame;  
108.         **int** n\_Outlen = 0;  
109.         pdec->H264Decode(ptr, nCount, p\_Out\_Frame, n\_Outlen);  
110.           
111.         **if**(n\_Outlen > 0)  
112.         \{  
113.             fwrite(p\_Out\_Frame, 1, n\_Outlen, ofp);  
114.         \}  
115.     \}  
116.   
117.   
118.     //realse  
119.     **delete** \[\]p\_In\_Frame;  
120.     **delete** \[\]p\_Out\_Frame;  
121.     pdec->StopH264Decoder();  
122.     pdec->ReleaseConnection();  
123.     fclose(ifp);  
124.     fclose(ofp);  
125.   
126.     **return** 0;  
127. \}  
128.

[view plain]: http://blog.csdn.net/wchm_seu/article/details/7736650#