色度分量帧内预测的代码

JM模型中8x8块的色度分量帧内预测的代码 先保存下来 一边读一边注释

1. void IntraChromaPrediction8x8 (int *mb_up, int *mb_left, int*mb_up_left)
2. {
3. 
   
4.   Macroblock *currMB = &img->mb_data[img->current_mb_nr];
5.   int     s, s0, s1, s2, s3, i, j, k;
6.   pel_t** image;
7.   int     block_x, block_y;
8.   int     mb_nr             = img->current_mb_nr;
9.   int     mb_available_up;
10.   int     mb_available_left[2];
11.   int     mb_available_up_left;
12.   int     ih,iv;
13.   int     ib,ic,iaa;
14.   int     uv;
15.   int     hline[8], vline[9];
16.   int     mode;
17.   int     best_mode = DC_PRED_8;         //just an initilaization here, should always be overwritten
18.   int     cost;
19.   int     min_cost;
20.   int     diff[16];
21.   PixelPos up;       //!< pixel position p(0,-1)
22.   PixelPos left[9];  //!< pixel positions p(-1, -1..8)
23. 
    
24. 
    
25.   for (i=0;i<9;i++)
26.   {
27.     getNeighbour(mb_nr, -1 ,  i-1 , 0, &left[i]);
28.   }
29.   
30.   getNeighbour(mb_nr, 0     ,  -1 , 0, &up);
31. 
    
32. 
    
33.   mb_available_up                             = up.available;
34.   mb_available_up_left                        = left[0].available;
35.   mb_available_left[0] = mb_available_left[1] = left[1].available;
36. 
    
37.   if(input->UseConstrainedIntraPred)
38.   {
39.     mb_available_up      = up.available ? img->intra_block[up.mb_addr] : 0;
40.     for (i=1, mb_available_left[0]=1; i<5;i++)
41.       mb_available_left[0]  &= left[i].available ? img->intra_block[left[i].mb_addr]: 0;
42.     for (i=5, mb_available_left[1]=1; i<9;i++)
43.       mb_available_left[1]  &= left[i].available ? img->intra_block[left[i].mb_addr]: 0;
44.     mb_available_up_left = left[0].available ? img->intra_block[left[0].mb_addr]: 0;
45.   }
46. 
    
47.   if (mb_up)
48.     *mb_up = mb_available_up;
49.   if (mb_left)
50.     *mb_left = mb_available_left[0] && mb_available_left[1];
51.   if( mb_up_left )
52.     *mb_up_left = mb_available_up_left;
53. 
    
54.   // compute all chroma intra prediction modes for both U and V
55.   for (uv=0; uv<2; uv++)
56.   {
57.     image = enc_picture->imgUV[uv];
58. 
    
59.     // DC prediction
60.     for (block_y=0; block_y<8; block_y+=4)
61.     for (block_x=0; block_x<8; block_x+=4)
62.     {
63.       s=128;
64.       s0=s1=s2=s3=0;
65.       //===== get prediction value =====
66.       switch ((block_y>>1) + (block_x>>2))
67.       {
68.       case 0:  //===== TOP LEFT =====
69.         if      (mb_available_up)       for (i=0;i<4;i++)  s0 += image[up.pos_y][up.pos_x + i];
70.         if      (mb_available_left[0])  for (i=1;i<5;i++)  s2 += image[left[i].pos_y][left[i].pos_x];
71.         if      (mb_available_up && mb_available_left[0])  s  = (s0+s2+4) >> 3;
72.         else if (mb_available_up)                          s  = (s0   +2) >> 2;
73.         else if (mb_available_left[0])                     s  = (s2   +2) >> 2;
74.         break;
75.       case 1: //===== TOP RIGHT =====
76.         if      (mb_available_up)       for (i=4;i<8;i++)  s1 += image[up.pos_y][up.pos_x + i];
77.         else if (mb_available_left[0])  for (i=1;i<5;i++)  s2 += image[left[i].pos_y][left[i].pos_x];
78.         if      (mb_available_up)                          s  = (s1   +2) >> 2;
79.         else if (mb_available_left[0])                     s  = (s2   +2) >> 2;
80.         break;
81.       case 2: //===== BOTTOM LEFT =====
82.         if      (mb_available_left[1])  for (i=5;i<9;i++)  s3 += image[left[i].pos_y][left[i].pos_x];
83.         else if (mb_available_up)       for (i=0;i<4;i++)  s0 += image[up.pos_y][up.pos_x + i];
84.         if      (mb_available_left[1])                     s  = (s3   +2) >> 2;
85.         else if (mb_available_up)                          s  = (s0   +2) >> 2;
86.         break;
87.       case 3: //===== BOTTOM RIGHT =====
88.         if      (mb_available_up)       for (i=4;i<8;i++)  s1 += image[up.pos_y][up.pos_x + i];
89.         if      (mb_available_left[1])  for (i=5;i<9;i++)  s3 += image[left[i].pos_y][left[i].pos_x];
90.         if      (mb_available_up && mb_available_left[1])  s  = (s1+s3+4) >> 3;
91.         else if (mb_available_up)                          s  = (s1   +2) >> 2;
92.         else if (mb_available_left[1])                     s  = (s3   +2) >> 2;
93.         break;
94.       }
95. 
    
96. 
    
97.       //===== prediction =====
98.       for (j=block_y; j<block_y+4; j++)
99.       for (i=block_x; i<block_x+4; i++)
100.       {
101.         img->mprr_c[uv][DC_PRED_8][i][j] = s;
102.       }
103.     }
104. 
     
105.     // vertical prediction
106.     if (mb_available_up)
107.     {
108.       for (i=0; i<8; i++)
109.         hline[i] = image[up.pos_y][up.pos_x + i];
110.       for (i=0; i<8; i++)
111.       for (j=0; j<8; j++)
112.         img->mprr_c[uv][VERT_PRED_8][i][j] = hline[i];
113.     }
114. 
     
115.     // horizontal prediction 
116.     if (mb_available_left[0] && mb_available_left[1])
117.     {
118.       for (i=1; i<9; i++)
119.         vline[i] = image[left[i].pos_y][left[i].pos_x];
120.       for (i=0; i<8; i++)
121.       for (j=0; j<8; j++)
122.         img->mprr_c[uv][HOR_PRED_8][i][j] = vline[j+1]; 
123.     }
124. 
     
125.     // plane prediction 
126.     if (mb_available_left[0] && mb_available_left[1] && mb_available_up && mb_available_up_left)
127.     {
128.       ih = 4*(hline[7] - image[left[0].pos_y][left[0].pos_x]);
129.       iv = 4*(vline[7+1] - image[left[0].pos_y][left[0].pos_x]);
130.       for (i=1;i<4;i++)
131.       {
132.         ih += i*(hline[3+i] - hline[3-i]);
133.         iv += i*(vline[3+i+1] - vline[3-i+1]);
134.       }
135.       ib=(17*ih+16)>>5;
136.       ic=(17*iv+16)>>5;
137. 
     
138.       iaa=16*(hline[7]+vline[7+1]);
139.       for (j=0; j<8; j++)
140.       for (i=0; i<8; i++)
141.         img->mprr_c[uv][PLANE_8][i][j]=max(0,min(255,(iaa+(i-3)*ib +(j-3)*ic + 16)/32));// store plane prediction
142.     }
143.   }
144. 
     
145.   if (!input->rdopt) // the rd-opt part does not work correctly (see encode_one_macroblock)
146.   {                       // since ipredmodes could be overwritten => encoder-decoder-mismatches
147.     // pick lowest cost prediction mode
148.     min_cost = 1<<20;
149.     for (i=0;i<8;i++)
150.     {
151.       getNeighbour(mb_nr, 0 ,  i , 0, &left[i]);
152.     }
153.     for (mode=DC_PRED_8; mode<=PLANE_8; mode++)
154.     {
155.       if ((mode==VERT_PRED_8 && !mb_available_up) ||
156.           (mode==HOR_PRED_8 && (!mb_available_left[0] || !mb_available_left[1])) ||
157.           (mode==PLANE_8 && (!mb_available_left[0] || !mb_available_left[1] || !mb_available_up || !mb_available_up_left)))
158.         continue;
159. 
     
160.       cost = 0;
161.       for (uv=0; uv<2; uv++)
162.       {
163.         image = imgUV_org[uv];
164.         for (block_y=0; block_y<8; block_y+=4)
165.         for (block_x=0; block_x<8; block_x+=4)
166.         {
167.           for (k=0,j=block_y; j<block_y+4; j++)
168.           for (i=block_x; i<block_x+4; i++,k++)
169.           {
170.             diff[k] = image[left[j].pos_y][left[j].pos_x+i] - img->mprr_c[uv][mode][i][j];
171.           }
172.           cost += SATD(diff, input->hadamard);
173.         }
174.       }
175.       if (cost < min_cost)
176.       {
177.         best_mode = mode;
178.         min_cost = cost;
179.       }
180.     }
181. 
     
182.     currMB->c_ipred_mode = best_mode;
183.   }
184.  
185. }
186.