1. #include <stdlib.h>
2. #include <stdio.h>
3. #include <string.h>
4. typedef       void                Void;
5. typedef       bool                Bool;
6. typedef       int                 Int;
7. typedef       unsigned char       UChar;
8. typedef       UChar               Pel;        ///< 16-bit pixel type
9. class TComPicYuv
10. {
11. private:
12. // ------------------------------------------------------------------------------------------------
13. //  YUV buffer
14. // ------------------------------------------------------------------------------------------------
15. Pel*  m_apiPicBufY;           ///< Buffer (including margin)
16. Pel*  m_apiPicBufU;
17. Pel*  m_apiPicBufV;
18. // ------------------------------------------------------------------------------------------------
19. //  Parameter for general YUV buffer usage
20. // ------------------------------------------------------------------------------------------------
21. Int   m_iPicWidth;            ///< Width of picture
22. Int   m_iPicHeight;           ///< Height of picture
23. public:
24. TComPicYuv         ();
25. virtual ~TComPicYuv();
26. // ------------------------------------------------------------------------------------------------
27. //  Memory management
28. // ------------------------------------------------------------------------------------------------
29. Bool  create      ( Int iPicWidth, Int iPicHeight );
30. Void  destroy     ();
31. Void  setYuv      (Pel pixelY,Pel pixelU,Pel pixelV);
32. Void  setPixel    (int posX, int posY, int width, Pel pixelY,Pel pixelU,Pel pixelV);
33. // ------------------------------------------------------------------------------------------------
34. //  Get information of picture
35. // ------------------------------------------------------------------------------------------------
36. Pel*  getPicBufY  ()     { return m_apiPicBufY ; }
37. Pel*  getPicBufU  ()     { return m_apiPicBufU ; }
38. Pel*  getPicBufV  ()     { return m_apiPicBufV ; }
39. Int   getWidth    ()     { return  m_iPicWidth;    }
40. Int   getHeight   ()     { return  m_iPicHeight;   }
41. Int   getStride   ()     { return  m_iPicWidth ;    }
42. Int   getCStride  ()     { return  (m_iPicWidth >> 1); }
43. };// END CLASS DEFINITION TComPicYuv
44. TComPicYuv::TComPicYuv()
45. {
46. m_apiPicBufY      = NULL;   // Buffer (including margin)
47. m_apiPicBufU      = NULL;
48. m_apiPicBufV      = NULL;
49. }
50. TComPicYuv::~TComPicYuv()
51. {
52. }
53. Bool TComPicYuv::create( Int iPicWidth, Int iPicHeight )
54. {
55. m_iPicWidth       = iPicWidth;
56. m_iPicHeight      = iPicHeight;
57. m_apiPicBufY      = (Pel*)malloc(m_iPicWidth * m_iPicHeight * sizeof(Pel));
58. m_apiPicBufU      = (Pel*)malloc((m_iPicWidth >>1) * (m_iPicHeight>>1) * sizeof(Pel));
59. m_apiPicBufV      = (Pel*)malloc((m_iPicWidth >>1) * (m_iPicHeight>>1) * sizeof(Pel));
60. return true;
61. }
62. Void TComPicYuv::destroy()
63. {
64. if( m_apiPicBufY ){ free( m_apiPicBufY );    m_apiPicBufY = NULL; }
65. if( m_apiPicBufU ){ free( m_apiPicBufU );    m_apiPicBufU = NULL; }
66. if( m_apiPicBufV ){ free( m_apiPicBufV );    m_apiPicBufV = NULL; }
67. }
68. Void TComPicYuv::setYuv(Pel pixelY,Pel pixelU,Pel pixelV)
69. {
70. if( m_apiPicBufY ){ memset(m_apiPicBufY, pixelY,m_iPicWidth * m_iPicHeight * sizeof(Pel)) ; }
71. if( m_apiPicBufU ){ memset(m_apiPicBufU, pixelU,(m_iPicWidth >>1) * (m_iPicHeight>>1) * sizeof(Pel)) ; }
72. if( m_apiPicBufV ){ memset(m_apiPicBufV, pixelV,(m_iPicWidth >>1) * (m_iPicHeight>>1) * sizeof(Pel)) ; }
73. }
74. Void TComPicYuv::setPixel(int posX, int posY, int width, Pel pixelY,Pel pixelU,Pel pixelV)
75. {
76. Pel *Y = m_apiPicBufY + posY*m_iPicWidth+posX;
77. for(int h = 0; h<width; h++)
78. {
79. for(int w = 0; w<width;w++)
80. {
81. Y[h*m_iPicWidth + w] = pixelY;
82. }
83. }
84. Pel *U = m_apiPicBufU + (posY>>1)*(m_iPicWidth>>1)+(posX>>1);
85. Pel *V = m_apiPicBufV + (posY>>1)*(m_iPicWidth>>1)+(posX>>1);
86. for(int h = 0; h<(width>>1); h++)
87. {
88. for(int w = 0; w<(width>>1);w++)
89. {
90. U[h*(m_iPicWidth>>1) + w] = pixelU;
91. V[h*(m_iPicWidth>>1) + w] = pixelV;
92. }
93. }
94. }
95. int main()
96. {
97. TComPicYuv *frame = new TComPicYuv;
98. TComPicYuv *clourMap = new TComPicYuv;
99. frame->create(8,8);
100. clourMap->create(512,512);
101. FILE *FrameFile = fopen("D:\\clourFrame_8x8.yuv","wb");
102. FILE *clourFile = fopen("D:\\clourMap_512x512.yuv","wb");
103. for(int Y = 0; Y<256; Y++)
104. {
105. for(int U = 0; U<256; U++)
106. {
107. for(int V = 0; V<256; V++)
108. {
109. frame->setYuv((Pel)Y,(Pel)U,(Pel)V);
110. fwrite(frame->getPicBufY(),1,frame->getWidth()*frame->getHeight(),FrameFile);
111. fwrite(frame->getPicBufU(),1,(frame->getWidth()*frame->getHeight())>>2,FrameFile);
112. fwrite(frame->getPicBufV(),1,(frame->getWidth()*frame->getHeight())>>2,FrameFile);
113. int  count = Y*256*256 + U*256 + V;
114. if(count%1000000==0)
115. {
116. printf("=");
117. }
118. }
119. }
120. }
121. for(int Y = 0; Y<256; Y++)
122. {
123. for(int U = 0; U<256; U++)
124. {
125. for(int V = 0; V<256; V++)
126. {
127. clourMap->setPixel(U*2,V*2,2,Y,U,V);
128. }
129. }
130. fwrite(clourMap->getPicBufY(),1,clourMap->getWidth()*clourMap->getHeight(),clourFile);
131. fwrite(clourMap->getPicBufU(),1,(clourMap->getWidth()*clourMap->getHeight())>>2,clourFile);
132. fwrite(clourMap->getPicBufV(),1,(clourMap->getWidth()*clourMap->getHeight())>>2,clourFile);
133. printf("*");
134. }
135. fclose(clourFile);
136. fclose(FrameFile);
137. frame->destroy();
138. clourMap->destroy();
139. }

转载：http://blog.csdn.net/cabbage2008/article/details/50117671