[cvs] / xvidcore / src / utils / mbtransquant.c Repository:
ViewVC logotype

Diff of /xvidcore/src/utils/mbtransquant.c

Parent Directory Parent Directory | Revision Log Revision Log | View Patch Patch

revision 1.21.2.8, Sun Apr 13 16:18:09 2003 UTC revision 1.21.2.18, Tue Oct 7 13:02:35 2003 UTC
# Line 25  Line 25 
25   *   *
26   ****************************************************************************/   ****************************************************************************/
27    
28  #include <string.h>  #include <stdio.h>
29  #include <stdlib.h>  #include <stdlib.h>
30    #include <string.h>
31    
32  #include "../portab.h"  #include "../portab.h"
33  #include "mbfunctions.h"  #include "mbfunctions.h"
# Line 34  Line 35 
35  #include "../global.h"  #include "../global.h"
36  #include "mem_transfer.h"  #include "mem_transfer.h"
37  #include "timer.h"  #include "timer.h"
38    #include "../bitstream/mbcoding.h"
39    #include "../bitstream/zigzag.h"
40  #include "../dct/fdct.h"  #include "../dct/fdct.h"
41  #include "../dct/idct.h"  #include "../dct/idct.h"
42  #include "../quant/quant_mpeg4.h"  #include "../quant/quant.h"
 #include "../quant/quant_h263.h"  
43  #include "../encoder.h"  #include "../encoder.h"
44    
45  #include "../image/reduced.h"  #include "../image/reduced.h"
# Line 115  Line 117 
117  /* Quantize all blocks -- Intra mode */  /* Quantize all blocks -- Intra mode */
118  static __inline void  static __inline void
119  MBQuantIntra(const MBParam * pParam,  MBQuantIntra(const MBParam * pParam,
120                             const FRAMEINFO * const frame,
121                           const MACROBLOCK * pMB,                           const MACROBLOCK * pMB,
122                           int16_t qcoeff[6 * 64],                           int16_t qcoeff[6 * 64],
123                           int16_t data[6*64])                           int16_t data[6*64])
124  {  {
125          int i;          int mpeg;
126            int scaler_lum, scaler_chr;
127    
128          for (i = 0; i < 6; i++) {          quant_intraFuncPtr const quant[2] =
129                  uint32_t iDcScaler = get_dc_scaler(pMB->quant, i < 4);                  {
130                            quant_h263_intra,
131                            quant_mpeg_intra
132                    };
133    
134            mpeg = !!(pParam->vol_flags & XVID_VOL_MPEGQUANT);
135            scaler_lum = get_dc_scaler(pMB->quant, 1);
136            scaler_chr = get_dc_scaler(pMB->quant, 0);
137    
138                  /* Quantize the block */                  /* Quantize the block */
139                  start_timer();                  start_timer();
140                  if (!(pParam->vol_flags & XVID_VOL_MPEGQUANT))          quant[mpeg](&data[0 * 64], &qcoeff[0 * 64], pMB->quant, scaler_lum);
141                          quant_intra(&data[i * 64], &qcoeff[i * 64], pMB->quant, iDcScaler);          quant[mpeg](&data[1 * 64], &qcoeff[1 * 64], pMB->quant, scaler_lum);
142                  else          quant[mpeg](&data[2 * 64], &qcoeff[2 * 64], pMB->quant, scaler_lum);
143                          quant4_intra(&data[i * 64], &qcoeff[i * 64], pMB->quant, iDcScaler);          quant[mpeg](&data[3 * 64], &qcoeff[3 * 64], pMB->quant, scaler_lum);
144            quant[mpeg](&data[4 * 64], &qcoeff[4 * 64], pMB->quant, scaler_chr);
145            quant[mpeg](&data[5 * 64], &qcoeff[5 * 64], pMB->quant, scaler_chr);
146                  stop_quant_timer();                  stop_quant_timer();
147          }          }
 }  
148    
149  /* DeQuantize all blocks -- Intra mode */  /* DeQuantize all blocks -- Intra mode */
150  static __inline void  static __inline void
# Line 141  Line 153 
153                             int16_t qcoeff[6 * 64],                             int16_t qcoeff[6 * 64],
154                             int16_t data[6*64])                             int16_t data[6*64])
155  {  {
156          int i;          int mpeg;
157            int scaler_lum, scaler_chr;
158    
159          for (i = 0; i < 6; i++) {          quant_intraFuncPtr const dequant[2] =
160                  uint32_t iDcScaler = get_dc_scaler(iQuant, i < 4);                  {
161                            dequant_h263_intra,
162                            dequant_mpeg_intra
163                    };
164    
165            mpeg = !!(pParam->vol_flags & XVID_VOL_MPEGQUANT);
166            scaler_lum = get_dc_scaler(iQuant, 1);
167            scaler_chr = get_dc_scaler(iQuant, 0);
168    
169                  start_timer();                  start_timer();
170                  if (!(pParam->vol_flags & XVID_VOL_MPEGQUANT))          dequant[mpeg](&qcoeff[0 * 64], &data[0 * 64], iQuant, scaler_lum);
171                          dequant_intra(&qcoeff[i * 64], &data[i * 64], iQuant, iDcScaler);          dequant[mpeg](&qcoeff[1 * 64], &data[1 * 64], iQuant, scaler_lum);
172                  else          dequant[mpeg](&qcoeff[2 * 64], &data[2 * 64], iQuant, scaler_lum);
173                          dequant4_intra(&qcoeff[i * 64], &data[i * 64], iQuant, iDcScaler);          dequant[mpeg](&qcoeff[3 * 64], &data[3 * 64], iQuant, scaler_lum);
174            dequant[mpeg](&qcoeff[4 * 64], &data[4 * 64], iQuant, scaler_chr);
175            dequant[mpeg](&qcoeff[5 * 64], &data[5 * 64], iQuant, scaler_chr);
176                  stop_iquant_timer();                  stop_iquant_timer();
177          }          }
178  }  
179    
180    typedef int (*trellis_func_ptr_t)(int16_t *const Out,
181                                                                      const int16_t *const In,
182                                                                      int Q,
183                                                                      const uint16_t * const Zigzag,
184                                                                      int Non_Zero);
185    
186    static int
187    dct_quantize_trellis_h263_c(int16_t *const Out,
188                                                            const int16_t *const In,
189                                                            int Q,
190                                                            const uint16_t * const Zigzag,
191                                                            int Non_Zero);
192    
193    static int
194    dct_quantize_trellis_mpeg_c(int16_t *const Out,
195                                                            const int16_t *const In,
196                                                            int Q,
197                                                            const uint16_t * const Zigzag,
198                                                            int Non_Zero);
199    
200  /* Quantize all blocks -- Inter mode */  /* Quantize all blocks -- Inter mode */
201  static __inline uint8_t  static __inline uint8_t
202  MBQuantInter(const MBParam * pParam,  MBQuantInter(const MBParam * pParam,
203                             const FRAMEINFO * const frame,
204                           const MACROBLOCK * pMB,                           const MACROBLOCK * pMB,
205                           int16_t data[6 * 64],                           int16_t data[6 * 64],
206                           int16_t qcoeff[6 * 64],                           int16_t qcoeff[6 * 64],
# Line 168  Line 211 
211          int i;          int i;
212          uint8_t cbp = 0;          uint8_t cbp = 0;
213          int sum;          int sum;
214          int code_block;          int code_block, mpeg;
215    
216            quant_interFuncPtr const quant[2] =
217                    {
218                            quant_h263_inter,
219                            quant_mpeg_inter
220                    };
221    
222            trellis_func_ptr_t const trellis[2] =
223                    {
224                            dct_quantize_trellis_h263_c,
225                            dct_quantize_trellis_mpeg_c
226                    };
227    
228            mpeg = !!(pParam->vol_flags & XVID_VOL_MPEGQUANT);
229    
230          for (i = 0; i < 6; i++) {          for (i = 0; i < 6; i++) {
231    
232                  /* Quantize the block */                  /* Quantize the block */
233                  start_timer();                  start_timer();
234                  if (!(pParam->vol_flags & XVID_VOL_MPEGQUANT))  
235                          sum = quant_inter(&qcoeff[i * 64], &data[i * 64], pMB->quant);                  sum = quant[mpeg](&qcoeff[i*64], &data[i*64], pMB->quant);
236                  else  
237                          sum = quant4_inter(&qcoeff[i * 64], &data[i * 64], pMB->quant);                  if(sum && (frame->vop_flags & XVID_VOP_TRELLISQUANT)) {
238                            sum = trellis[mpeg](&qcoeff[i*64], &data[i*64], pMB->quant, &scan_tables[0][0], 63);
239                    }
240                  stop_quant_timer();                  stop_quant_timer();
241    
242                  /*                  /*
# Line 202  Line 261 
261    
262                  /* Set the corresponding cbp bit */                  /* Set the corresponding cbp bit */
263                  cbp |= code_block << (5 - i);                  cbp |= code_block << (5 - i);
   
264          }          }
265    
266          return(cbp);          return(cbp);
# Line 216  Line 274 
274                             int16_t qcoeff[6 * 64],                             int16_t qcoeff[6 * 64],
275                             const uint8_t cbp)                             const uint8_t cbp)
276  {  {
277          int i;          int mpeg;
278    
279            quant_interFuncPtr const dequant[2] =
280                    {
281                            dequant_h263_inter,
282                            dequant_mpeg_inter
283                    };
284    
285            mpeg = !!(pParam->vol_flags & XVID_VOL_MPEGQUANT);
286    
         for (i = 0; i < 6; i++) {  
                 if (cbp & (1 << (5 - i))) {  
287                          start_timer();                          start_timer();
288                          if (!(pParam->vol_flags & XVID_VOL_MPEGQUANT))          if(cbp & (1 << (5 - 0))) dequant[mpeg](&data[0 * 64], &qcoeff[0 * 64], iQuant);
289                                  dequant_inter(&data[i * 64], &qcoeff[i * 64], iQuant);          if(cbp & (1 << (5 - 1))) dequant[mpeg](&data[1 * 64], &qcoeff[1 * 64], iQuant);
290                          else          if(cbp & (1 << (5 - 2))) dequant[mpeg](&data[2 * 64], &qcoeff[2 * 64], iQuant);
291                                  dequant4_inter(&data[i * 64], &qcoeff[i * 64], iQuant);          if(cbp & (1 << (5 - 3))) dequant[mpeg](&data[3 * 64], &qcoeff[3 * 64], iQuant);
292            if(cbp & (1 << (5 - 4))) dequant[mpeg](&data[4 * 64], &qcoeff[4 * 64], iQuant);
293            if(cbp & (1 << (5 - 5))) dequant[mpeg](&data[5 * 64], &qcoeff[5 * 64], iQuant);
294                          stop_iquant_timer();                          stop_iquant_timer();
295                  }                  }
         }  
 }  
296    
297  typedef void (transfer_operation_8to16_t) (int16_t *Dst, const uint8_t *Src, int BpS);  typedef void (transfer_operation_8to16_t) (int16_t *Dst, const uint8_t *Src, int BpS);
298  typedef void (transfer_operation_16to8_t) (uint8_t *Dst, const int16_t *Src, int BpS);  typedef void (transfer_operation_16to8_t) (uint8_t *Dst, const int16_t *Src, int BpS);
# Line 246  Line 310 
310          uint32_t stride2 = stride / 2;          uint32_t stride2 = stride / 2;
311          uint32_t next_block = stride * 8;          uint32_t next_block = stride * 8;
312          int32_t cst;          int32_t cst;
313            int vop_reduced;
314          uint8_t *pY_Cur, *pU_Cur, *pV_Cur;          uint8_t *pY_Cur, *pU_Cur, *pV_Cur;
315          const IMAGE * const pCurrent = &frame->image;          const IMAGE * const pCurrent = &frame->image;
316            transfer_operation_8to16_t * const functions[2] =
317                    {
318                            (transfer_operation_8to16_t *)transfer_8to16copy,
319                            (transfer_operation_8to16_t *)filter_18x18_to_8x8
320                    };
321          transfer_operation_8to16_t *transfer_op = NULL;          transfer_operation_8to16_t *transfer_op = NULL;
322    
323          if ((frame->vop_flags & XVID_VOP_REDUCED)) {          vop_reduced = !!(frame->vop_flags & XVID_VOP_REDUCED);
   
                 /* Image pointers */  
                 pY_Cur = pCurrent->y + (y_pos << 5) * stride  + (x_pos << 5);  
                 pU_Cur = pCurrent->u + (y_pos << 4) * stride2 + (x_pos << 4);  
                 pV_Cur = pCurrent->v + (y_pos << 4) * stride2 + (x_pos << 4);  
   
                 /* Block size */  
                 cst = 16;  
   
                 /* Operation function */  
                 transfer_op = (transfer_operation_8to16_t*)filter_18x18_to_8x8;  
         } else {  
324    
325                  /* Image pointers */                  /* Image pointers */
326                  pY_Cur = pCurrent->y + (y_pos << 4) * stride  + (x_pos << 4);          pY_Cur = pCurrent->y + (y_pos << (4+vop_reduced)) * stride  + (x_pos << (4+vop_reduced));
327                  pU_Cur = pCurrent->u + (y_pos << 3) * stride2 + (x_pos << 3);          pU_Cur = pCurrent->u + (y_pos << (3+vop_reduced)) * stride2 + (x_pos << (3+vop_reduced));
328                  pV_Cur = pCurrent->v + (y_pos << 3) * stride2 + (x_pos << 3);          pV_Cur = pCurrent->v + (y_pos << (3+vop_reduced)) * stride2 + (x_pos << (3+vop_reduced));
329    
330                  /* Block size */                  /* Block size */
331                  cst = 8;          cst = 8<<vop_reduced;
332    
333                  /* Operation function */                  /* Operation function */
334                  transfer_op = (transfer_operation_8to16_t*)transfer_8to16copy;          transfer_op = functions[vop_reduced];
         }  
335    
336          /* Do the transfer */          /* Do the transfer */
337          start_timer();          start_timer();
# Line 294  Line 351 
351                           const uint32_t x_pos,                           const uint32_t x_pos,
352                           const uint32_t y_pos,                           const uint32_t y_pos,
353                           int16_t data[6 * 64],                           int16_t data[6 * 64],
354                           const uint32_t add,                           const uint32_t add, /* Must be 1 or 0 */
355                           const uint8_t cbp)                           const uint8_t cbp)
356  {  {
357          uint8_t *pY_Cur, *pU_Cur, *pV_Cur;          uint8_t *pY_Cur, *pU_Cur, *pV_Cur;
# Line 302  Line 359 
359          uint32_t stride2 = stride / 2;          uint32_t stride2 = stride / 2;
360          uint32_t next_block = stride * 8;          uint32_t next_block = stride * 8;
361          uint32_t cst;          uint32_t cst;
362            int vop_reduced;
363          const IMAGE * const pCurrent = &frame->image;          const IMAGE * const pCurrent = &frame->image;
364    
365            /* Array of function pointers, indexed by [vop_reduced<<1+add] */
366            transfer_operation_16to8_t  * const functions[4] =
367                    {
368                            (transfer_operation_16to8_t*)transfer_16to8copy,
369                            (transfer_operation_16to8_t*)transfer_16to8add,
370                            (transfer_operation_16to8_t*)copy_upsampled_8x8_16to8,
371                            (transfer_operation_16to8_t*)add_upsampled_8x8_16to8
372                    };
373    
374          transfer_operation_16to8_t *transfer_op = NULL;          transfer_operation_16to8_t *transfer_op = NULL;
375    
376          if (pMB->field_dct) {          if (pMB->field_dct) {
# Line 310  Line 378 
378                  stride *= 2;                  stride *= 2;
379          }          }
380    
381          if ((frame->vop_flags & XVID_VOP_REDUCED)) {          /* Makes this vars booleans */
382            vop_reduced = !!(frame->vop_flags & XVID_VOP_REDUCED);
                 /* Image pointers */  
                 pY_Cur = pCurrent->y + (y_pos << 5) * stride  + (x_pos << 5);  
                 pU_Cur = pCurrent->u + (y_pos << 4) * stride2 + (x_pos << 4);  
                 pV_Cur = pCurrent->v + (y_pos << 4) * stride2 + (x_pos << 4);  
   
                 /* Block size */  
                 cst = 16;  
   
                 /* Operation function */  
                 if(add)  
                         transfer_op = (transfer_operation_16to8_t*)add_upsampled_8x8_16to8;  
                 else  
                         transfer_op = (transfer_operation_16to8_t*)copy_upsampled_8x8_16to8;  
         } else {  
383    
384                  /* Image pointers */                  /* Image pointers */
385                  pY_Cur = pCurrent->y + (y_pos << 4) * stride  + (x_pos << 4);          pY_Cur = pCurrent->y + (y_pos << (4+vop_reduced)) * stride  + (x_pos << (4+vop_reduced));
386                  pU_Cur = pCurrent->u + (y_pos << 3) * stride2 + (x_pos << 3);          pU_Cur = pCurrent->u + (y_pos << (3+vop_reduced)) * stride2 + (x_pos << (3+vop_reduced));
387                  pV_Cur = pCurrent->v + (y_pos << 3) * stride2 + (x_pos << 3);          pV_Cur = pCurrent->v + (y_pos << (3+vop_reduced)) * stride2 + (x_pos << (3+vop_reduced));
388    
389                  /* Block size */                  /* Block size */
390                  cst = 8;          cst = 8<<vop_reduced;
391    
392                  /* Operation function */                  /* Operation function */
393                  if(add)          transfer_op = functions[(vop_reduced<<1) + add];
                         transfer_op = (transfer_operation_16to8_t*)transfer_16to8add;  
                 else  
                         transfer_op = (transfer_operation_16to8_t*)transfer_16to8copy;  
         }  
394    
395          /* Do the operation */          /* Do the operation */
396          start_timer();          start_timer();
# Line 374  Line 424 
424          MBfDCT(pParam, frame, pMB, x_pos, y_pos, data);          MBfDCT(pParam, frame, pMB, x_pos, y_pos, data);
425    
426          /* Quantize the block */          /* Quantize the block */
427          MBQuantIntra(pParam, pMB, data, qcoeff);          MBQuantIntra(pParam, frame, pMB, data, qcoeff);
428    
429          /* DeQuantize the block */          /* DeQuantize the block */
430          MBDeQuantIntra(pParam, pMB->quant, data, qcoeff);          MBDeQuantIntra(pParam, pMB->quant, data, qcoeff);
# Line 399  Line 449 
449          uint8_t cbp;          uint8_t cbp;
450          uint32_t limit;          uint32_t limit;
451    
452          /*          /* There is no MBTrans8to16 for Inter block, that's done in motion compensation
453           * There is no MBTrans8to16 for Inter block, that's done in motion compensation           * already */
          * already  
          */  
454    
455          /* Perform DCT (and field decision) */          /* Perform DCT (and field decision) */
456          MBfDCT(pParam, frame, pMB, x_pos, y_pos, data);          MBfDCT(pParam, frame, pMB, x_pos, y_pos, data);
# Line 410  Line 458 
458          /* Set the limit threshold */          /* Set the limit threshold */
459          limit = PVOP_TOOSMALL_LIMIT + ((pMB->quant == 1)? 1 : 0);          limit = PVOP_TOOSMALL_LIMIT + ((pMB->quant == 1)? 1 : 0);
460    
461            if (frame->vop_flags & XVID_VOP_CARTOON)
462                    limit *= 3;
463    
464          /* Quantize the block */          /* Quantize the block */
465          cbp = MBQuantInter(pParam, pMB, data, qcoeff, 0, limit);          cbp = MBQuantInter(pParam, frame, pMB, data, qcoeff, 0, limit);
466    
467          /* DeQuantize the block */          /* DeQuantize the block */
468          MBDeQuantInter(pParam, pMB->quant, data, qcoeff, cbp);          MBDeQuantInter(pParam, pMB->quant, data, qcoeff, cbp);
# Line 437  Line 488 
488          uint8_t cbp;          uint8_t cbp;
489          uint32_t limit;          uint32_t limit;
490    
491          /*          /* There is no MBTrans8to16 for Inter block, that's done in motion compensation
492           * There is no MBTrans8to16 for Inter block, that's done in motion compensation           * already */
          * already  
          */  
493    
494          /* Perform DCT (and field decision) */          /* Perform DCT (and field decision) */
495          MBfDCT(pParam, frame, pMB, x_pos, y_pos, data);          MBfDCT(pParam, frame, pMB, x_pos, y_pos, data);
# Line 448  Line 497 
497          /* Set the limit threshold */          /* Set the limit threshold */
498          limit = BVOP_TOOSMALL_LIMIT;          limit = BVOP_TOOSMALL_LIMIT;
499    
500            if (frame->vop_flags & XVID_VOP_CARTOON)
501                    limit *= 2;
502    
503          /* Quantize the block */          /* Quantize the block */
504          cbp = MBQuantInter(pParam, pMB, data, qcoeff, 1, limit);          cbp = MBQuantInter(pParam, frame, pMB, data, qcoeff, 1, limit);
505    
506          /*          /*
507           * History comment:           * History comment:
508           * We don't have to DeQuant, iDCT and Transfer back data for B-frames.           * We don't have to DeQuant, iDCT and Transfer back data for B-frames.
509           *           *
510           * BUT some plugins require the original frame to be passed so we have           * BUT some plugins require the rebuilt original frame to be passed so we
511           * to take care of that here           * have to take care of that here
512           */           */
513          if((pParam->plugin_flags & XVID_REQORIGINAL)) {          if((pParam->plugin_flags & XVID_REQORIGINAL)) {
514    
# Line 526  Line 578 
578    
579          /* left blocks */          /* left blocks */
580    
581          // 1=2, 2=4, 4=8, 8=1          /* 1=2, 2=4, 4=8, 8=1 */
582          MOVLINE(tmp, LINE(0, 1));          MOVLINE(tmp, LINE(0, 1));
583          MOVLINE(LINE(0, 1), LINE(0, 2));          MOVLINE(LINE(0, 1), LINE(0, 2));
584          MOVLINE(LINE(0, 2), LINE(0, 4));          MOVLINE(LINE(0, 2), LINE(0, 4));
585          MOVLINE(LINE(0, 4), LINE(2, 0));          MOVLINE(LINE(0, 4), LINE(2, 0));
586          MOVLINE(LINE(2, 0), tmp);          MOVLINE(LINE(2, 0), tmp);
587    
588          // 3=6, 6=12, 12=9, 9=3          /* 3=6, 6=12, 12=9, 9=3 */
589          MOVLINE(tmp, LINE(0, 3));          MOVLINE(tmp, LINE(0, 3));
590          MOVLINE(LINE(0, 3), LINE(0, 6));          MOVLINE(LINE(0, 3), LINE(0, 6));
591          MOVLINE(LINE(0, 6), LINE(2, 4));          MOVLINE(LINE(0, 6), LINE(2, 4));
592          MOVLINE(LINE(2, 4), LINE(2, 1));          MOVLINE(LINE(2, 4), LINE(2, 1));
593          MOVLINE(LINE(2, 1), tmp);          MOVLINE(LINE(2, 1), tmp);
594    
595          // 5=10, 10=5          /* 5=10, 10=5 */
596          MOVLINE(tmp, LINE(0, 5));          MOVLINE(tmp, LINE(0, 5));
597          MOVLINE(LINE(0, 5), LINE(2, 2));          MOVLINE(LINE(0, 5), LINE(2, 2));
598          MOVLINE(LINE(2, 2), tmp);          MOVLINE(LINE(2, 2), tmp);
599    
600          // 7=14, 14=13, 13=11, 11=7          /* 7=14, 14=13, 13=11, 11=7 */
601          MOVLINE(tmp, LINE(0, 7));          MOVLINE(tmp, LINE(0, 7));
602          MOVLINE(LINE(0, 7), LINE(2, 6));          MOVLINE(LINE(0, 7), LINE(2, 6));
603          MOVLINE(LINE(2, 6), LINE(2, 5));          MOVLINE(LINE(2, 6), LINE(2, 5));
# Line 554  Line 606 
606    
607          /* right blocks */          /* right blocks */
608    
609          // 1=2, 2=4, 4=8, 8=1          /* 1=2, 2=4, 4=8, 8=1 */
610          MOVLINE(tmp, LINE(1, 1));          MOVLINE(tmp, LINE(1, 1));
611          MOVLINE(LINE(1, 1), LINE(1, 2));          MOVLINE(LINE(1, 1), LINE(1, 2));
612          MOVLINE(LINE(1, 2), LINE(1, 4));          MOVLINE(LINE(1, 2), LINE(1, 4));
613          MOVLINE(LINE(1, 4), LINE(3, 0));          MOVLINE(LINE(1, 4), LINE(3, 0));
614          MOVLINE(LINE(3, 0), tmp);          MOVLINE(LINE(3, 0), tmp);
615    
616          // 3=6, 6=12, 12=9, 9=3          /* 3=6, 6=12, 12=9, 9=3 */
617          MOVLINE(tmp, LINE(1, 3));          MOVLINE(tmp, LINE(1, 3));
618          MOVLINE(LINE(1, 3), LINE(1, 6));          MOVLINE(LINE(1, 3), LINE(1, 6));
619          MOVLINE(LINE(1, 6), LINE(3, 4));          MOVLINE(LINE(1, 6), LINE(3, 4));
620          MOVLINE(LINE(3, 4), LINE(3, 1));          MOVLINE(LINE(3, 4), LINE(3, 1));
621          MOVLINE(LINE(3, 1), tmp);          MOVLINE(LINE(3, 1), tmp);
622    
623          // 5=10, 10=5          /* 5=10, 10=5 */
624          MOVLINE(tmp, LINE(1, 5));          MOVLINE(tmp, LINE(1, 5));
625          MOVLINE(LINE(1, 5), LINE(3, 2));          MOVLINE(LINE(1, 5), LINE(3, 2));
626          MOVLINE(LINE(3, 2), tmp);          MOVLINE(LINE(3, 2), tmp);
627    
628          // 7=14, 14=13, 13=11, 11=7          /* 7=14, 14=13, 13=11, 11=7 */
629          MOVLINE(tmp, LINE(1, 7));          MOVLINE(tmp, LINE(1, 7));
630          MOVLINE(LINE(1, 7), LINE(3, 6));          MOVLINE(LINE(1, 7), LINE(3, 6));
631          MOVLINE(LINE(3, 6), LINE(3, 5));          MOVLINE(LINE(3, 6), LINE(3, 5));
632          MOVLINE(LINE(3, 5), LINE(3, 3));          MOVLINE(LINE(3, 5), LINE(3, 3));
633          MOVLINE(LINE(3, 3), tmp);          MOVLINE(LINE(3, 3), tmp);
634  }  }
635    
636    /*****************************************************************************
637     *               Trellis based R-D optimal quantization
638     *
639     *   Trellis Quant code (C) 2003 Pascal Massimino skal(at)planet-d.net
640     *
641     ****************************************************************************/
642    
643    /*----------------------------------------------------------------------------
644     *
645     *        Trellis-Based quantization
646     *
647     * So far I understand this paper:
648     *
649     *  "Trellis-Based R-D Optimal Quantization in H.263+"
650     *    J.Wen, M.Luttrell, J.Villasenor
651     *    IEEE Transactions on Image Processing, Vol.9, No.8, Aug. 2000.
652     *
653     * we are at stake with a simplified Bellmand-Ford / Dijkstra Single
654     * Source Shorted Path algo. But due to the underlying graph structure
655     * ("Trellis"), it can be turned into a dynamic programming algo,
656     * partially saving the explicit graph's nodes representation. And
657     * without using a heap, since the open frontier of the DAG is always
658     * known, and of fixed sized.
659     *--------------------------------------------------------------------------*/
660    
661    
662    
663    /* Codes lengths for relevant levels. */
664    
665    /* let's factorize: */
666    static const uint8_t Code_Len0[64] = {
667            30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,
668            30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30 };
669    static const uint8_t Code_Len1[64] = {
670            20,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,
671            30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30 };
672    static const uint8_t Code_Len2[64] = {
673            19,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,
674            30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30 };
675    static const uint8_t Code_Len3[64] = {
676            18,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,
677            30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30 };
678    static const uint8_t Code_Len4[64] = {
679            17,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,
680            30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30 };
681    static const uint8_t Code_Len5[64] = {
682            16,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,
683            30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30 };
684    static const uint8_t Code_Len6[64] = {
685            15,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,
686            30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30 };
687    static const uint8_t Code_Len7[64] = {
688            13,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,
689            30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30 };
690    static const uint8_t Code_Len8[64] = {
691            11,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,
692            30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30 };
693    static const uint8_t Code_Len9[64] = {
694            12,21,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,
695            30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30 };
696    static const uint8_t Code_Len10[64] = {
697            12,20,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,
698            30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30 };
699    static const uint8_t Code_Len11[64] = {
700            12,19,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,
701            30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30 };
702    static const uint8_t Code_Len12[64] = {
703            11,17,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,
704            30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30 };
705    static const uint8_t Code_Len13[64] = {
706            11,15,21,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,
707            30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30 };
708    static const uint8_t Code_Len14[64] = {
709            10,12,19,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,
710            30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30 };
711    static const uint8_t Code_Len15[64] = {
712            10,13,17,19,21,21,21,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,
713            30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30 };
714    static const uint8_t Code_Len16[64] = {
715            9,12,13,18,18,19,19,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,
716            30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30};
717    static const uint8_t Code_Len17[64] = {
718            8,11,13,14,14,14,15,19,19,19,21,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,
719            30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30 };
720    static const uint8_t Code_Len18[64] = {
721            7, 9,11,11,13,13,13,15,15,15,16,22,22,22,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,
722            30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30 };
723    static const uint8_t Code_Len19[64] = {
724            5, 7, 9,10,10,11,11,11,11,11,13,14,16,17,17,18,18,18,18,18,18,18,18,20,20,21,21,30,30,30,30,30,
725            30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30 };
726    static const uint8_t Code_Len20[64] = {
727            3, 4, 5, 6, 6, 6, 7, 7, 7, 7, 8, 8, 8, 9, 9,10,10,10,10,10,10,10,10,12,12,13,13,12,13,14,15,15,
728            15,16,16,16,16,17,17,17,18,18,19,19,19,19,19,19,19,19,21,21,22,22,30,30,30,30,30,30,30,30,30,30 };
729    
730    /* a few more table for LAST table: */
731    static const uint8_t Code_Len21[64] = {
732            13,20,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,
733            30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30};
734    static const uint8_t Code_Len22[64] = {
735            12,15,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,
736            30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30};
737    static const uint8_t Code_Len23[64] = {
738            10,12,15,15,15,16,16,16,16,17,17,17,17,17,17,17,17,18,18,18,18,18,18,18,18,19,19,19,19,20,20,20,
739            20,21,21,21,21,21,21,21,21,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30,30};
740    static const uint8_t Code_Len24[64] = {
741            5, 7, 7, 7, 7, 8, 8, 8, 8, 9, 9, 9, 9, 9, 9, 9, 9,10,10,10,10,10,10,10,10,11,11,11,11,12,12,12,
742            12,13,13,13,13,13,13,13,13,14,16,16,16,16,17,17,17,17,18,18,18,18,18,18,18,18,19,19,19,19,19,19};
743    
744    
745    static const uint8_t * const B16_17_Code_Len[24] = { /* levels [1..24] */
746            Code_Len20,Code_Len19,Code_Len18,Code_Len17,
747            Code_Len16,Code_Len15,Code_Len14,Code_Len13,
748            Code_Len12,Code_Len11,Code_Len10,Code_Len9,
749            Code_Len8, Code_Len7 ,Code_Len6 ,Code_Len5,
750            Code_Len4, Code_Len3, Code_Len3 ,Code_Len2,
751            Code_Len2, Code_Len1, Code_Len1, Code_Len1,
752    };
753    
754    static const uint8_t * const B16_17_Code_Len_Last[6] = { /* levels [1..6] */
755            Code_Len24,Code_Len23,Code_Len22,Code_Len21, Code_Len3, Code_Len1,
756    };
757    
758    #define TL(q) 0xfe00/(q*q)
759    
760    static const int Trellis_Lambda_Tabs[31] = {
761            TL( 1),TL( 2),TL( 3),TL( 4),TL( 5),TL( 6), TL( 7),
762            TL( 8),TL( 9),TL(10),TL(11),TL(12),TL(13),TL(14), TL(15),
763            TL(16),TL(17),TL(18),TL(19),TL(20),TL(21),TL(22), TL(23),
764            TL(24),TL(25),TL(26),TL(27),TL(28),TL(29),TL(30), TL(31)
765    };
766    #undef TL
767    
768    static int __inline
769    Find_Last(const int16_t *C, const uint16_t *Zigzag, int i)
770    {
771            while(i>=0)
772                    if (C[Zigzag[i]])
773                            return i;
774                    else i--;
775            return -1;
776    }
777    
778    static int __inline
779    Compute_Sum(const int16_t *C, int last)
780    {
781            int sum = 0;
782    
783            while(last--)
784                    sum += abs(C[last]);
785    
786            return(sum);
787    }
788    /* this routine has been strippen of all debug code */
789    
790    static int
791    dct_quantize_trellis_h263_c(int16_t *const Out, const int16_t *const In, int Q, const uint16_t * const Zigzag, int Non_Zero)
792    {
793    
794        /*
795             * Note: We should search last non-zero coeffs on *real* DCT input coeffs (In[]),
796             * not quantized one (Out[]). However, it only improves the result *very*
797             * slightly (~0.01dB), whereas speed drops to crawling level :)
798             * Well, actually, taking 1 more coeff past Non_Zero into account sometimes helps.
799             */
800            typedef struct { int16_t Run, Level; } NODE;
801    
802            NODE Nodes[65], Last;
803            uint32_t Run_Costs0[64+1];
804            uint32_t * const Run_Costs = Run_Costs0 + 1;
805            const int Mult = 2*Q;
806            const int Bias = (Q-1) | 1;
807            const int Lev0 = Mult + Bias;
808            const int Lambda = Trellis_Lambda_Tabs[Q-1];    /* it's 1/lambda, actually */
809    
810            int Run_Start = -1;
811            uint32_t Min_Cost = 2<<16;
812    
813            int Last_Node = -1;
814            uint32_t Last_Cost = 0;
815    
816            int i, j, sum;
817            Run_Costs[-1] = 2<<16;                          /* source (w/ CBP penalty) */
818    
819            Non_Zero = Find_Last(Out, Zigzag, Non_Zero);
820            if (Non_Zero<0)
821                    return 0; /* Sum is zero if there are only zero coeffs */
822    
823            for(i=0; i<=Non_Zero; i++) {
824                    const int AC = In[Zigzag[i]];
825                    const int Level1 = Out[Zigzag[i]];
826                    const int Dist0 = Lambda* AC*AC;
827                    uint32_t Best_Cost = 0xf0000000;
828                    Last_Cost += Dist0;
829    
830                    /* very specialized loop for -1,0,+1 */
831                    if ((uint32_t)(Level1+1)<3) {
832                            int dQ;
833                            int Run;
834                            uint32_t Cost0;
835    
836                            if (AC<0) {
837                                    Nodes[i].Level = -1;
838                                    dQ = Lev0 + AC;
839                            } else {
840                                    Nodes[i].Level = 1;
841                                    dQ = Lev0 - AC;
842                            }
843                            Cost0 = Lambda*dQ*dQ;
844    
845                            Nodes[i].Run = 1;
846                            Best_Cost = (Code_Len20[0]<<16) + Run_Costs[i-1]+Cost0;
847                            for(Run=i-Run_Start; Run>0; --Run) {
848                                    const uint32_t Cost_Base = Cost0 + Run_Costs[i-Run];
849                                    const uint32_t Cost = Cost_Base + (Code_Len20[Run-1]<<16);
850                                    const uint32_t lCost = Cost_Base + (Code_Len24[Run-1]<<16);
851    
852                                    /*
853                                     * TODO: what about tie-breaks? Should we favor short runs or
854                                     * long runs? Although the error is the same, it would not be
855                                     * spread the same way along high and low frequencies...
856                                     */
857    
858                                    /* (I'd say: favour short runs => hifreq errors (HVS) -- gruel ) */
859    
860                                    if (Cost<Best_Cost) {
861                                            Best_Cost    = Cost;
862                                            Nodes[i].Run = Run;
863                                    }
864    
865                                    if (lCost<Last_Cost) {
866                                            Last_Cost  = lCost;
867                                            Last.Run   = Run;
868                                            Last_Node  = i;
869                                    }
870                            }
871                            if (Last_Node==i)
872                                    Last.Level = Nodes[i].Level;
873                    } else { /* "big" levels */
874                            const uint8_t *Tbl_L1, *Tbl_L2, *Tbl_L1_Last, *Tbl_L2_Last;
875                            int Level2;
876                            int dQ1, dQ2;
877                            int Run;
878                            uint32_t Dist1,Dist2;
879                            int dDist21;
880    
881                            if (Level1>1) {
882                                    dQ1 = Level1*Mult-AC + Bias;
883                                    dQ2 = dQ1 - Mult;
884                                    Level2 = Level1-1;
885                                    Tbl_L1      = (Level1<=24) ? B16_17_Code_Len[Level1-1]     : Code_Len0;
886                                    Tbl_L2      = (Level2<=24) ? B16_17_Code_Len[Level2-1]     : Code_Len0;
887                                    Tbl_L1_Last = (Level1<=6) ? B16_17_Code_Len_Last[Level1-1] : Code_Len0;
888                                    Tbl_L2_Last = (Level2<=6) ? B16_17_Code_Len_Last[Level2-1] : Code_Len0;
889                            } else { /* Level1<-1 */
890                                    dQ1 = Level1*Mult-AC - Bias;
891                                    dQ2 = dQ1 + Mult;
892                                    Level2 = Level1 + 1;
893                                    Tbl_L1      = (Level1>=-24) ? B16_17_Code_Len[Level1^-1]      : Code_Len0;
894                                    Tbl_L2      = (Level2>=-24) ? B16_17_Code_Len[Level2^-1]      : Code_Len0;
895                                    Tbl_L1_Last = (Level1>=- 6) ? B16_17_Code_Len_Last[Level1^-1] : Code_Len0;
896                                    Tbl_L2_Last = (Level2>=- 6) ? B16_17_Code_Len_Last[Level2^-1] : Code_Len0;
897                            }
898    
899                            Dist1 = Lambda*dQ1*dQ1;
900                            Dist2 = Lambda*dQ2*dQ2;
901                            dDist21 = Dist2-Dist1;
902    
903                            for(Run=i-Run_Start; Run>0; --Run)
904                            {
905                                    const uint32_t Cost_Base = Dist1 + Run_Costs[i-Run];
906                                    uint32_t Cost1, Cost2;
907                                    int bLevel;
908    
909                                    /*
910                                     * for sub-optimal (but slightly worth it, speed-wise) search, uncomment the following:
911                                     *      if (Cost_Base>=Best_Cost) continue;
912                                     * (? doesn't seem to have any effect -- gruel )
913                                     */
914    
915                                    Cost1 = Cost_Base + (Tbl_L1[Run-1]<<16);
916                                    Cost2 = Cost_Base + (Tbl_L2[Run-1]<<16) + dDist21;
917    
918                                    if (Cost2<Cost1) {
919                                            Cost1 = Cost2;
920                                            bLevel = Level2;
921                                    } else {
922                                            bLevel = Level1;
923                                    }
924    
925                                    if (Cost1<Best_Cost) {
926                                            Best_Cost = Cost1;
927                                            Nodes[i].Run   = Run;
928                                            Nodes[i].Level = bLevel;
929                                    }
930    
931                                    Cost1 = Cost_Base + (Tbl_L1_Last[Run-1]<<16);
932                                    Cost2 = Cost_Base + (Tbl_L2_Last[Run-1]<<16) + dDist21;
933    
934                                    if (Cost2<Cost1) {
935                                            Cost1 = Cost2;
936                                            bLevel = Level2;
937                                    } else {
938                                            bLevel = Level1;
939                                    }
940    
941                                    if (Cost1<Last_Cost) {
942                                            Last_Cost  = Cost1;
943                                            Last.Run   = Run;
944                                            Last.Level = bLevel;
945                                            Last_Node  = i;
946                                    }
947                            } /* end of "for Run" */
948    
949                    }
950    
951                    Run_Costs[i] = Best_Cost;
952    
953                    if (Best_Cost < Min_Cost + Dist0) {
954                            Min_Cost = Best_Cost;
955                            Run_Start = i;
956                    } else {
957                            /*
958                             * as noticed by Michael Niedermayer (michaelni at gmx.at), there's
959                             * a code shorter by 1 bit for a larger run (!), same level. We give
960                             * it a chance by not moving the left barrier too much.
961                             */
962    
963                            while( Run_Costs[Run_Start]>Min_Cost+(1<<16) )
964                                    Run_Start++;
965    
966                            /* spread on preceding coeffs the cost incurred by skipping this one */
967                            for(j=Run_Start; j<i; ++j) Run_Costs[j] += Dist0;
968                            Min_Cost += Dist0;
969                    }
970            }
971    
972            /* It seems trellis doesn't give good results... just compute the Out sum and
973             * quit (even if we did not modify it, upperlayer relies on this data) */
974            if (Last_Node<0)
975                    return Compute_Sum(Out, Non_Zero);
976    
977            /* reconstruct optimal sequence backward with surviving paths */
978            memset(Out, 0x00, 64*sizeof(*Out));
979            Out[Zigzag[Last_Node]] = Last.Level;
980            i = Last_Node - Last.Run;
981            sum = 0;
982            while(i>=0) {
983                    Out[Zigzag[i]] = Nodes[i].Level;
984                    sum += abs(Nodes[i].Level);
985                    i -= Nodes[i].Run;
986            }
987    
988            return sum;
989    }
990    
991    static int
992    dct_quantize_trellis_mpeg_c(int16_t *const Out, const int16_t *const In, int Q, const uint16_t * const Zigzag, int Non_Zero)
993    {
994            /* ToDo: Ok ok it's just a place holder for Gruel -- damn write this one :-) */
995            return Compute_Sum(Out, 63);
996    }
997    
998    /* original version including heavy debugging info */
999    
1000    #ifdef DBGTRELL
1001    
1002    #define DBG 0
1003    
1004    static __inline uint32_t Evaluate_Cost(const int16_t *C, int Mult, int Bias,
1005                                                                               const uint16_t * Zigzag, int Max, int Lambda)
1006    {
1007    #if (DBG>0)
1008            const int16_t * const Ref = C + 6*64;
1009            int Last = Max;
1010            int Bits = 0;
1011            int Dist = 0;
1012            int i;
1013            uint32_t Cost;
1014    
1015            while(Last>=0 && C[Zigzag[Last]]==0)
1016                    Last--;
1017    
1018            if (Last>=0) {
1019                    int j=0, j0=0;
1020                    int Run, Level;
1021    
1022                    Bits = 2;   /* CBP */
1023                    while(j<Last) {
1024                            while(!C[Zigzag[j]])
1025                                    j++;
1026                            if (j==Last)
1027                                    break;
1028                            Level=C[Zigzag[j]];
1029                            Run = j - j0;
1030                            j0 = ++j;
1031                            if (Level>=-24 && Level<=24)
1032                                    Bits += B16_17_Code_Len[(Level<0) ? -Level-1 : Level-1][Run];
1033                            else
1034                                    Bits += 30;
1035                    }
1036                    Level = C[Zigzag[Last]];
1037                    Run = j - j0;
1038                    if (Level>=-6 && Level<=6)
1039                            Bits += B16_17_Code_Len_Last[(Level<0) ? -Level-1 : Level-1][Run];
1040                    else
1041                            Bits += 30;
1042            }
1043    
1044            for(i=0; i<=Last; ++i) {
1045                    int V = C[Zigzag[i]]*Mult;
1046                    if (V>0)
1047                            V += Bias;
1048                    else
1049                            if (V<0)
1050                                    V -= Bias;
1051                    V -= Ref[Zigzag[i]];
1052                    Dist += V*V;
1053            }
1054            Cost = Lambda*Dist + (Bits<<16);
1055            if (DBG==1)
1056                    printf( " Last:%2d/%2d Cost = [(Bits=%5.0d) + Lambda*(Dist=%6.0d) = %d ] >>12= %d ", Last,Max, Bits, Dist, Cost, Cost>>12 );
1057            return Cost;
1058    
1059    #else
1060            return 0;
1061    #endif
1062    }
1063    
1064    
1065    static int
1066    dct_quantize_trellis_h263_c(int16_t *const Out, const int16_t *const In, int Q, const uint16_t * const Zigzag, int Non_Zero)
1067    {
1068    
1069        /*
1070             * Note: We should search last non-zero coeffs on *real* DCT input coeffs (In[]),
1071             * not quantized one (Out[]). However, it only improves the result *very*
1072             * slightly (~0.01dB), whereas speed drops to crawling level :)
1073             * Well, actually, taking 1 more coeff past Non_Zero into account sometimes helps.
1074             */
1075            typedef struct { int16_t Run, Level; } NODE;
1076    
1077            NODE Nodes[65], Last;
1078            uint32_t Run_Costs0[64+1];
1079            uint32_t * const Run_Costs = Run_Costs0 + 1;
1080            const int Mult = 2*Q;
1081            const int Bias = (Q-1) | 1;
1082            const int Lev0 = Mult + Bias;
1083            const int Lambda = Trellis_Lambda_Tabs[Q-1];    /* it's 1/lambda, actually */
1084    
1085            int Run_Start = -1;
1086            Run_Costs[-1] = 2<<16;                          /* source (w/ CBP penalty) */
1087            uint32_t Min_Cost = 2<<16;
1088    
1089            int Last_Node = -1;
1090            uint32_t Last_Cost = 0;
1091    
1092            int i, j;
1093    
1094    #if (DBG>0)
1095            Last.Level = 0; Last.Run = -1; /* just initialize to smthg */
1096    #endif
1097    
1098            Non_Zero = Find_Last(Out, Zigzag, Non_Zero);
1099            if (Non_Zero<0)
1100                    return -1;
1101    
1102            for(i=0; i<=Non_Zero; i++)
1103            {
1104                    const int AC = In[Zigzag[i]];
1105                    const int Level1 = Out[Zigzag[i]];
1106                    const int Dist0 = Lambda* AC*AC;
1107                    uint32_t Best_Cost = 0xf0000000;
1108                    Last_Cost += Dist0;
1109    
1110                    if ((uint32_t)(Level1+1)<3)                 /* very specialized loop for -1,0,+1 */
1111                    {
1112                            int dQ;
1113                            int Run;
1114                            uint32_t Cost0;
1115    
1116                            if (AC<0) {
1117                                    Nodes[i].Level = -1;
1118                                    dQ = Lev0 + AC;
1119                            } else {
1120                                    Nodes[i].Level = 1;
1121                                    dQ = Lev0 - AC;
1122                            }
1123                            Cost0 = Lambda*dQ*dQ;
1124    
1125                            Nodes[i].Run = 1;
1126                            Best_Cost = (Code_Len20[0]<<16) + Run_Costs[i-1]+Cost0;
1127                            for(Run=i-Run_Start; Run>0; --Run)
1128                            {
1129                                    const uint32_t Cost_Base = Cost0 + Run_Costs[i-Run];
1130                                    const uint32_t Cost = Cost_Base + (Code_Len20[Run-1]<<16);
1131                                    const uint32_t lCost = Cost_Base + (Code_Len24[Run-1]<<16);
1132    
1133                                    /*
1134                                     * TODO: what about tie-breaks? Should we favor short runs or
1135                                     * long runs? Although the error is the same, it would not be
1136                                     * spread the same way along high and low frequencies...
1137                                     */
1138                                    if (Cost<Best_Cost) {
1139                                            Best_Cost    = Cost;
1140                                            Nodes[i].Run = Run;
1141                                    }
1142    
1143                                    if (lCost<Last_Cost) {
1144                                            Last_Cost  = lCost;
1145                                            Last.Run   = Run;
1146                                            Last_Node  = i;
1147                                    }
1148                            }
1149                            if (Last_Node==i)
1150                                    Last.Level = Nodes[i].Level;
1151    
1152                            if (DBG==1) {
1153                                    Run_Costs[i] = Best_Cost;
1154                                    printf( "Costs #%2d: ", i);
1155                                    for(j=-1;j<=Non_Zero;++j) {
1156                                            if (j==Run_Start)            printf( " %3.0d|", Run_Costs[j]>>12 );
1157                                            else if (j>Run_Start && j<i) printf( " %3.0d|", Run_Costs[j]>>12 );
1158                                            else if (j==i)               printf( "(%3.0d)", Run_Costs[j]>>12 );
1159                                            else                         printf( "  - |" );
1160                                    }
1161                                    printf( "<%3.0d %2d %d>", Min_Cost>>12, Nodes[i].Level, Nodes[i].Run );
1162                                    printf( "  Last:#%2d {%3.0d %2d %d}", Last_Node, Last_Cost>>12, Last.Level, Last.Run );
1163                                    printf( " AC:%3.0d Dist0:%3d Dist(%d)=%d", AC, Dist0>>12, Nodes[i].Level, Cost0>>12 );
1164                                    printf( "\n" );
1165                            }
1166                    }
1167                    else                      /* "big" levels */
1168                    {
1169                            const uint8_t *Tbl_L1, *Tbl_L2, *Tbl_L1_Last, *Tbl_L2_Last;
1170                            int Level2;
1171                            int dQ1, dQ2;
1172                            int Run;
1173                            uint32_t Dist1,Dist2;
1174                            int dDist21;
1175    
1176                            if (Level1>1) {
1177                                    dQ1 = Level1*Mult-AC + Bias;
1178                                    dQ2 = dQ1 - Mult;
1179                                    Level2 = Level1-1;
1180                                    Tbl_L1      = (Level1<=24) ? B16_17_Code_Len[Level1-1]     : Code_Len0;
1181                                    Tbl_L2      = (Level2<=24) ? B16_17_Code_Len[Level2-1]     : Code_Len0;
1182                                    Tbl_L1_Last = (Level1<=6) ? B16_17_Code_Len_Last[Level1-1] : Code_Len0;
1183                                    Tbl_L2_Last = (Level2<=6) ? B16_17_Code_Len_Last[Level2-1] : Code_Len0;
1184                            } else { /* Level1<-1 */
1185                                    dQ1 = Level1*Mult-AC - Bias;
1186                                    dQ2 = dQ1 + Mult;
1187                                    Level2 = Level1 + 1;
1188                                    Tbl_L1      = (Level1>=-24) ? B16_17_Code_Len[Level1^-1]      : Code_Len0;
1189                                    Tbl_L2      = (Level2>=-24) ? B16_17_Code_Len[Level2^-1]      : Code_Len0;
1190                                    Tbl_L1_Last = (Level1>=- 6) ? B16_17_Code_Len_Last[Level1^-1] : Code_Len0;
1191                                    Tbl_L2_Last = (Level2>=- 6) ? B16_17_Code_Len_Last[Level2^-1] : Code_Len0;
1192                            }
1193                            Dist1 = Lambda*dQ1*dQ1;
1194                            Dist2 = Lambda*dQ2*dQ2;
1195                            dDist21 = Dist2-Dist1;
1196    
1197                            for(Run=i-Run_Start; Run>0; --Run)
1198                            {
1199                                    const uint32_t Cost_Base = Dist1 + Run_Costs[i-Run];
1200                                    uint32_t Cost1, Cost2;
1201                                    int bLevel;
1202    
1203    /*
1204     * for sub-optimal (but slightly worth it, speed-wise) search, uncomment the following:
1205     *        if (Cost_Base>=Best_Cost) continue;
1206     */
1207                                    Cost1 = Cost_Base + (Tbl_L1[Run-1]<<16);
1208                                    Cost2 = Cost_Base + (Tbl_L2[Run-1]<<16) + dDist21;
1209    
1210                                    if (Cost2<Cost1) {
1211                                            Cost1 = Cost2;
1212                                            bLevel = Level2;
1213                                    } else
1214                                            bLevel = Level1;
1215    
1216                                    if (Cost1<Best_Cost) {
1217                                            Best_Cost = Cost1;
1218                                            Nodes[i].Run   = Run;
1219                                            Nodes[i].Level = bLevel;
1220                                    }
1221    
1222                                    Cost1 = Cost_Base + (Tbl_L1_Last[Run-1]<<16);
1223                                    Cost2 = Cost_Base + (Tbl_L2_Last[Run-1]<<16) + dDist21;
1224    
1225                                    if (Cost2<Cost1) {
1226                                            Cost1 = Cost2;
1227                                            bLevel = Level2;
1228                                    } else
1229                                            bLevel = Level1;
1230    
1231                                    if (Cost1<Last_Cost) {
1232                                            Last_Cost  = Cost1;
1233                                            Last.Run   = Run;
1234                                            Last.Level = bLevel;
1235                                            Last_Node  = i;
1236                                    }
1237                            } /* end of "for Run" */
1238    
1239                            if (DBG==1) {
1240                                    Run_Costs[i] = Best_Cost;
1241                                    printf( "Costs #%2d: ", i);
1242                                    for(j=-1;j<=Non_Zero;++j) {
1243                                            if (j==Run_Start)            printf( " %3.0d|", Run_Costs[j]>>12 );
1244                                            else if (j>Run_Start && j<i) printf( " %3.0d|", Run_Costs[j]>>12 );
1245                                            else if (j==i)               printf( "(%3.0d)", Run_Costs[j]>>12 );
1246                                            else                         printf( "  - |" );
1247                                    }
1248                                    printf( "<%3.0d %2d %d>", Min_Cost>>12, Nodes[i].Level, Nodes[i].Run );
1249                                    printf( "  Last:#%2d {%3.0d %2d %d}", Last_Node, Last_Cost>>12, Last.Level, Last.Run );
1250                                    printf( " AC:%3.0d Dist0:%3d Dist(%2d):%3d Dist(%2d):%3d", AC, Dist0>>12, Level1, Dist1>>12, Level2, Dist2>>12 );
1251                                    printf( "\n" );
1252                            }
1253                    }
1254    
1255                    Run_Costs[i] = Best_Cost;
1256    
1257                    if (Best_Cost < Min_Cost + Dist0) {
1258                            Min_Cost = Best_Cost;
1259                            Run_Start = i;
1260                    }
1261                    else
1262                    {
1263                            /*
1264                             * as noticed by Michael Niedermayer (michaelni at gmx.at), there's
1265                             * a code shorter by 1 bit for a larger run (!), same level. We give
1266                             * it a chance by not moving the left barrier too much.
1267                             */
1268    
1269                            while( Run_Costs[Run_Start]>Min_Cost+(1<<16) )
1270                                    Run_Start++;
1271    
1272                            /* spread on preceding coeffs the cost incurred by skipping this one */
1273                            for(j=Run_Start; j<i; ++j) Run_Costs[j] += Dist0;
1274                            Min_Cost += Dist0;
1275                    }
1276            }
1277    
1278            if (DBG) {
1279                    Last_Cost = Evaluate_Cost(Out,Mult,Bias, Zigzag,Non_Zero, Lambda);
1280                    if (DBG==1) {
1281                            printf( "=> " );
1282                            for(i=0; i<=Non_Zero; ++i) printf( "[%3.0d] ", Out[Zigzag[i]] );
1283                            printf( "\n" );
1284                    }
1285            }
1286    
1287            if (Last_Node<0)
1288                    return -1;
1289    
1290            /* reconstruct optimal sequence backward with surviving paths */
1291            memset(Out, 0x00, 64*sizeof(*Out));
1292            Out[Zigzag[Last_Node]] = Last.Level;
1293            i = Last_Node - Last.Run;
1294            while(i>=0) {
1295                    Out[Zigzag[i]] = Nodes[i].Level;
1296                    i -= Nodes[i].Run;
1297            }
1298    
1299            if (DBG) {
1300                    uint32_t Cost = Evaluate_Cost(Out,Mult,Bias, Zigzag,Non_Zero, Lambda);
1301                    if (DBG==1) {
1302                            printf( "<= " );
1303                            for(i=0; i<=Last_Node; ++i) printf( "[%3.0d] ", Out[Zigzag[i]] );
1304                            printf( "\n--------------------------------\n" );
1305                    }
1306                    if (Cost>Last_Cost) printf( "!!! %u > %u\n", Cost, Last_Cost );
1307            }
1308            return Last_Node;
1309    }
1310    
1311    #undef DBG
1312    
1313    #endif
Legend:
Removed from v.1.21.2.8  
changed lines
  Added in v.1.21.2.18
No admin address has been configured
 ViewVC Help
Powered by ViewVC 1.0.4