getvlc.c
上传用户:hkgotone
上传日期:2013-02-17
资源大小:293k
文件大小:15k
源码类别:

Windows Mobile

开发平台:

C/C++

  1. /* getvlc.c, variable length decoding                                       */
  2. /* Copyright (C) 1996, MPEG Software Simulation Group. All Rights Reserved. */
  3. /*
  4.  * Disclaimer of Warranty
  5.  *
  6.  * These software programs are available to the user without any license fee or
  7.  * royalty on an "as is" basis.  The MPEG Software Simulation Group disclaims
  8.  * any and all warranties, whether express, implied, or statuary, including any
  9.  * implied warranties or merchantability or of fitness for a particular
  10.  * purpose.  In no event shall the copyright-holder be liable for any
  11.  * incidental, punitive, or consequential damages of any kind whatsoever
  12.  * arising from the use of these programs.
  13.  *
  14.  * This disclaimer of warranty extends to the user of these programs and user's
  15.  * customers, employees, agents, transferees, successors, and assigns.
  16.  *
  17.  * The MPEG Software Simulation Group does not represent or warrant that the
  18.  * programs furnished hereunder are free of infringement of any third-party
  19.  * patents.
  20.  *
  21.  * Commercial implementations of MPEG-1 and MPEG-2 video, including shareware,
  22.  * are subject to royalty fees to patent holders.  Many of these patents are
  23.  * general enough such that they are unavoidable regardless of implementation
  24.  * design.
  25.  *
  26.  */
  27. #include <stdio.h>
  28. #include "config.h"
  29. #include "global.h"
  30. #include "getvlc.h"
  31. /* private prototypes */
  32. /* generic picture macroblock type processing functions */
  33. static int Get_I_macroblock_type _ANSI_ARGS_((void));
  34. static int Get_P_macroblock_type _ANSI_ARGS_((void));
  35. static int Get_B_macroblock_type _ANSI_ARGS_((void));
  36. static int Get_D_macroblock_type _ANSI_ARGS_((void));
  37. /* spatial picture macroblock type processing functions */
  38. static int Get_I_Spatial_macroblock_type _ANSI_ARGS_((void));
  39. static int Get_P_Spatial_macroblock_type _ANSI_ARGS_((void));
  40. static int Get_B_Spatial_macroblock_type _ANSI_ARGS_((void));
  41. static int Get_SNR_macroblock_type _ANSI_ARGS_((void));
  42. int Get_macroblock_type()
  43. {
  44.   int macroblock_type = 0;
  45.   if (ld->scalable_mode==SC_SNR)
  46.     macroblock_type = Get_SNR_macroblock_type();
  47.   else
  48.   {
  49.     switch (picture_coding_type)
  50.     {
  51.     case I_TYPE:
  52.       macroblock_type = ld->pict_scal ? Get_I_Spatial_macroblock_type() : Get_I_macroblock_type();
  53.       break;
  54.     case P_TYPE:
  55.       macroblock_type = ld->pict_scal ? Get_P_Spatial_macroblock_type() : Get_P_macroblock_type();
  56.       break;
  57.     case B_TYPE:
  58.       macroblock_type = ld->pict_scal ? Get_B_Spatial_macroblock_type() : Get_B_macroblock_type();
  59.       break;
  60.     case D_TYPE:
  61.       macroblock_type = Get_D_macroblock_type();
  62.       break;
  63.     default:
  64.       printf("Get_macroblock_type(): unrecognized picture coding typen");
  65.       break;
  66.     }
  67.   }
  68.   return macroblock_type;
  69. }
  70. static int Get_I_macroblock_type()
  71. {
  72. #ifdef TRACE
  73.   if (Trace_Flag)
  74.     printf("macroblock_type(I) ");
  75. #endif /* TRACE */
  76.   if (Get_Bits1())
  77.   {
  78. #ifdef TRACE
  79.     if (Trace_Flag)
  80.       printf("(1): Intra (1)n");
  81. #endif /* TRACE */
  82.     return 1;
  83.   }
  84.   if (!Get_Bits1())
  85.   {
  86.     if (!Quiet_Flag)
  87.       printf("Invalid macroblock_type coden");
  88.     Fault_Flag = 1;
  89.   }
  90. #ifdef TRACE
  91.   if (Trace_Flag)
  92.     printf("(01): Intra, Quant (17)n");
  93. #endif /* TRACE */
  94.   return 17;
  95. }
  96. static char *MBdescr[]={
  97.   "",                  "Intra",        "No MC, Coded",         "",
  98.   "Bwd, Not Coded",    "",             "Bwd, Coded",           "",
  99.   "Fwd, Not Coded",    "",             "Fwd, Coded",           "",
  100.   "Interp, Not Coded", "",             "Interp, Coded",        "",
  101.   "",                  "Intra, Quant", "No MC, Coded, Quant",  "",
  102.   "",                  "",             "Bwd, Coded, Quant",    "",
  103.   "",                  "",             "Fwd, Coded, Quant",    "",
  104.   "",                  "",             "Interp, Coded, Quant", ""
  105. };
  106. static int Get_P_macroblock_type()
  107. {
  108.   int code;
  109. #ifdef TRACE
  110.   if (Trace_Flag)
  111.     printf("macroblock_type(P) (");
  112. #endif /* TRACE */
  113.   if ((code = Show_Bits(6))>=8)
  114.   {
  115.     code >>= 3;
  116.     Flush_Buffer(PMBtab0[code].len);
  117. #ifdef TRACE
  118.     if (Trace_Flag)
  119.     {
  120.       Print_Bits(code,3,PMBtab0[code].len);
  121.       printf("): %s (%d)n",MBdescr[(int)PMBtab0[code].val],PMBtab0[code].val);
  122.     }
  123. #endif /* TRACE */
  124.     return PMBtab0[code].val;
  125.   }
  126.   if (code==0)
  127.   {
  128.     if (!Quiet_Flag)
  129.       printf("Invalid macroblock_type coden");
  130.     Fault_Flag = 1;
  131.     return 0;
  132.   }
  133.   Flush_Buffer(PMBtab1[code].len);
  134. #ifdef TRACE
  135.   if (Trace_Flag)
  136.   {
  137.     Print_Bits(code,6,PMBtab1[code].len);
  138.     printf("): %s (%d)n",MBdescr[(int)PMBtab1[code].val],PMBtab1[code].val);
  139.   }
  140. #endif /* TRACE */
  141.   return PMBtab1[code].val;
  142. }
  143. static int Get_B_macroblock_type()
  144. {
  145.   int code;
  146. #ifdef TRACE
  147.   if (Trace_Flag)
  148.     printf("macroblock_type(B) (");
  149. #endif /* TRACE */
  150.   if ((code = Show_Bits(6))>=8)
  151.   {
  152.     code >>= 2;
  153.     Flush_Buffer(BMBtab0[code].len);
  154. #ifdef TRACE
  155.     if (Trace_Flag)
  156.     {
  157.       Print_Bits(code,4,BMBtab0[code].len);
  158.       printf("): %s (%d)n",MBdescr[(int)BMBtab0[code].val],BMBtab0[code].val);
  159.     }
  160. #endif /* TRACE */
  161.     return BMBtab0[code].val;
  162.   }
  163.   if (code==0)
  164.   {
  165.     if (!Quiet_Flag)
  166.       printf("Invalid macroblock_type coden");
  167.     Fault_Flag = 1;
  168.     return 0;
  169.   }
  170.   Flush_Buffer(BMBtab1[code].len);
  171. #ifdef TRACE
  172.   if (Trace_Flag)
  173.   {
  174.     Print_Bits(code,6,BMBtab1[code].len);
  175.     printf("): %s (%d)n",MBdescr[(int)BMBtab1[code].val],BMBtab1[code].val);
  176.   }
  177. #endif /* TRACE */
  178.   return BMBtab1[code].val;
  179. }
  180. static int Get_D_macroblock_type()
  181. {
  182.   if (!Get_Bits1())
  183.   {
  184.     if (!Quiet_Flag)
  185.       printf("Invalid macroblock_type coden");
  186.     Fault_Flag=1;
  187.   }
  188.   return 1;
  189. }
  190. /* macroblock_type for pictures with spatial scalability */
  191. static int Get_I_Spatial_macroblock_type()
  192. {
  193.   int code;
  194. #ifdef TRACE
  195.   if (Trace_Flag)
  196.     printf("macroblock_type(I,spat) (");
  197. #endif /* TRACE */
  198.   code = Show_Bits(4);
  199.   if (code==0)
  200.   {
  201.     if (!Quiet_Flag)
  202.       printf("Invalid macroblock_type coden");
  203.     Fault_Flag = 1;
  204.     return 0;
  205.   }
  206. #ifdef TRACE
  207.   if (Trace_Flag)
  208.   {
  209.     Print_Bits(code,4,spIMBtab[code].len);
  210.     printf("): %02xn",spIMBtab[code].val);
  211.   }
  212. #endif /* TRACE */
  213.   Flush_Buffer(spIMBtab[code].len);
  214.   return spIMBtab[code].val;
  215. }
  216. static int Get_P_Spatial_macroblock_type()
  217. {
  218.   int code;
  219. #ifdef TRACE
  220.   if (Trace_Flag)
  221.     printf("macroblock_type(P,spat) (");
  222. #endif /* TRACE */
  223.   code = Show_Bits(7);
  224.   if (code<2)
  225.   {
  226.     if (!Quiet_Flag)
  227.       printf("Invalid macroblock_type coden");
  228.     Fault_Flag = 1;
  229.     return 0;
  230.   }
  231.   if (code>=16)
  232.   {
  233.     code >>= 3;
  234.     Flush_Buffer(spPMBtab0[code].len);
  235. #ifdef TRACE
  236.     if (Trace_Flag)
  237.     {
  238.       Print_Bits(code,4,spPMBtab0[code].len);
  239.       printf("): %02xn",spPMBtab0[code].val);
  240.     }
  241. #endif /* TRACE */
  242.     return spPMBtab0[code].val;
  243.   }
  244.   Flush_Buffer(spPMBtab1[code].len);
  245. #ifdef TRACE
  246.   if (Trace_Flag)
  247.   {
  248.     Print_Bits(code,7,spPMBtab1[code].len);
  249.     printf("): %02xn",spPMBtab1[code].val);
  250.   }
  251. #endif /* TRACE */
  252.   return spPMBtab1[code].val;
  253. }
  254. static int Get_B_Spatial_macroblock_type()
  255. {
  256.   int code;
  257.   VLCtab *p;
  258. #ifdef TRACE
  259.   if (Trace_Flag)
  260.     printf("macroblock_type(B,spat) (");
  261. #endif /* TRACE */
  262.   code = Show_Bits(9);
  263.   if (code>=64)
  264.     p = &spBMBtab0[(code>>5)-2];
  265.   else if (code>=16)
  266.     p = &spBMBtab1[(code>>2)-4];
  267.   else if (code>=8)
  268.     p = &spBMBtab2[code-8];
  269.   else
  270.   {
  271.     if (!Quiet_Flag)
  272.       printf("Invalid macroblock_type coden");
  273.     Fault_Flag = 1;
  274.     return 0;
  275.   }
  276.   Flush_Buffer(p->len);
  277. #ifdef TRACE
  278.   if (Trace_Flag)
  279.   {
  280.     Print_Bits(code,9,p->len);
  281.     printf("): %02xn",p->val);
  282.   }
  283. #endif /* TRACE */
  284.   return p->val;
  285. }
  286. static int Get_SNR_macroblock_type()
  287. {
  288.   int code;
  289. #ifdef TRACE /* *CH* */
  290.   if (Trace_Flag)
  291.     printf("macroblock_type(SNR) (");
  292. #endif TRACE
  293.   code = Show_Bits(3);
  294.   if (code==0)
  295.   {
  296.     if (!Quiet_Flag)
  297.       printf("Invalid macroblock_type coden");
  298.     Fault_Flag = 1;
  299.     return 0;
  300.   }
  301.   Flush_Buffer(SNRMBtab[code].len);
  302. #ifdef TRACE /* *CH* */
  303.   if (Trace_Flag)
  304.   {
  305.     Print_Bits(code,3,SNRMBtab[code].len);
  306.     printf("): %s (%d)n",MBdescr[(int)SNRMBtab[code].val],SNRMBtab[code].val);
  307.   }
  308. #endif TRACE
  309.   return SNRMBtab[code].val;
  310. }
  311. int Get_motion_code()
  312. {
  313.   int code;
  314. #ifdef TRACE
  315.   if (Trace_Flag)
  316.     printf("motion_code (");
  317. #endif /* TRACE */
  318.   if (Get_Bits1())
  319.   {
  320. #ifdef TRACE
  321.     if (Trace_Flag)
  322.       printf("0): 0n");
  323. #endif /* TRACE */
  324.     return 0;
  325.   }
  326.   if ((code = Show_Bits(9))>=64)
  327.   {
  328.     code >>= 6;
  329.     Flush_Buffer(MVtab0[code].len);
  330. #ifdef TRACE
  331.     if (Trace_Flag)
  332.     {
  333.       Print_Bits(code,3,MVtab0[code].len);
  334.       printf("%d): %dn",
  335.         Show_Bits(1),Show_Bits(1)?-MVtab0[code].val:MVtab0[code].val);
  336.     }
  337. #endif /* TRACE */
  338.     return Get_Bits1()?-MVtab0[code].val:MVtab0[code].val;
  339.   }
  340.   if (code>=24)
  341.   {
  342.     code >>= 3;
  343.     Flush_Buffer(MVtab1[code].len);
  344. #ifdef TRACE
  345.     if (Trace_Flag)
  346.     {
  347.       Print_Bits(code,6,MVtab1[code].len);
  348.       printf("%d): %dn",
  349.         Show_Bits(1),Show_Bits(1)?-MVtab1[code].val:MVtab1[code].val);
  350.     }
  351. #endif /* TRACE */
  352.     return Get_Bits1()?-MVtab1[code].val:MVtab1[code].val;
  353.   }
  354.   if ((code-=12)<0)
  355.   {
  356.     if (!Quiet_Flag)
  357. /* HACK */
  358.       printf("Invalid motion_vector code (MBA %d, pic %d)n", global_MBA, global_pic);
  359.     Fault_Flag=1;
  360.     return 0;
  361.   }
  362.   Flush_Buffer(MVtab2[code].len);
  363. #ifdef TRACE
  364.   if (Trace_Flag)
  365.   {
  366.     Print_Bits(code+12,9,MVtab2[code].len);
  367.     printf("%d): %dn",
  368.       Show_Bits(1),Show_Bits(1)?-MVtab2[code].val:MVtab2[code].val);
  369.   }
  370. #endif /* TRACE */
  371.   return Get_Bits1() ? -MVtab2[code].val : MVtab2[code].val;
  372. }
  373. /* get differential motion vector (for dual prime prediction) */
  374. int Get_dmvector()
  375. {
  376. #ifdef TRACE
  377.   if (Trace_Flag)
  378.     printf("dmvector (");
  379. #endif /* TRACE */
  380.   if (Get_Bits(1))
  381.   {
  382. #ifdef TRACE
  383.     if (Trace_Flag)
  384.       printf(Show_Bits(1) ? "11): -1n" : "10): 1n");
  385. #endif /* TRACE */
  386.     return Get_Bits(1) ? -1 : 1;
  387.   }
  388.   else
  389.   {
  390. #ifdef TRACE
  391.     if (Trace_Flag)
  392.       printf("0): 0n");
  393. #endif /* TRACE */
  394.     return 0;
  395.   }
  396. }
  397. int Get_coded_block_pattern()
  398. {
  399.   int code;
  400. #ifdef TRACE
  401.   if (Trace_Flag)
  402.     printf("coded_block_pattern_420 (");
  403. #endif /* TRACE */
  404.   if ((code = Show_Bits(9))>=128)
  405.   {
  406.     code >>= 4;
  407.     Flush_Buffer(CBPtab0[code].len);
  408. #ifdef TRACE
  409.     if (Trace_Flag)
  410.     {
  411.       Print_Bits(code,5,CBPtab0[code].len);
  412.       printf("): ");
  413.       Print_Bits(CBPtab0[code].val,6,6);
  414.       printf(" (%d)n",CBPtab0[code].val);
  415.     }
  416. #endif /* TRACE */
  417.     return CBPtab0[code].val;
  418.   }
  419.   if (code>=8)
  420.   {
  421.     code >>= 1;
  422.     Flush_Buffer(CBPtab1[code].len);
  423. #ifdef TRACE
  424.     if (Trace_Flag)
  425.     {
  426.       Print_Bits(code,8,CBPtab1[code].len);
  427.       printf("): ");
  428.       Print_Bits(CBPtab1[code].val,6,6);
  429.       printf(" (%d)n",CBPtab1[code].val);
  430.     }
  431. #endif /* TRACE */
  432.     return CBPtab1[code].val;
  433.   }
  434.   if (code<1)
  435.   {
  436.     if (!Quiet_Flag)
  437.       printf("Invalid coded_block_pattern coden");
  438.     Fault_Flag = 1;
  439.     return 0;
  440.   }
  441.   Flush_Buffer(CBPtab2[code].len);
  442. #ifdef TRACE
  443.   if (Trace_Flag)
  444.   {
  445.     Print_Bits(code,9,CBPtab2[code].len);
  446.     printf("): ");
  447.     Print_Bits(CBPtab2[code].val,6,6);
  448.     printf(" (%d)n",CBPtab2[code].val);
  449.   }
  450. #endif /* TRACE */
  451.   return CBPtab2[code].val;
  452. }
  453. int Get_macroblock_address_increment()
  454. {
  455.   int code, val;
  456. #ifdef TRACE
  457.   if (Trace_Flag)
  458.     printf("macroblock_address_increment (");
  459. #endif /* TRACE */
  460.   val = 0;
  461.   while ((code = Show_Bits(11))<24)
  462.   {
  463.     if (code!=15) /* if not macroblock_stuffing */
  464.     {
  465.       if (code==8) /* if macroblock_escape */
  466.       {
  467. #ifdef TRACE
  468.         if (Trace_Flag)
  469.           printf("00000001000 ");
  470. #endif /* TRACE */
  471.         val+= 33;
  472.       }
  473.       else
  474.       {
  475.         if (!Quiet_Flag)
  476.           printf("Invalid macroblock_address_increment coden");
  477.         Fault_Flag = 1;
  478.         return 1;
  479.       }
  480.     }
  481.     else /* macroblock suffing */
  482.     {
  483. #ifdef TRACE
  484.       if (Trace_Flag)
  485.         printf("00000001111 ");
  486. #endif /* TRACE */
  487.     }
  488.     Flush_Buffer(11);
  489.   }
  490.   /* macroblock_address_increment == 1 */
  491.   /* ('1' is in the MSB position of the lookahead) */
  492.   if (code>=1024)
  493.   {
  494.     Flush_Buffer(1);
  495. #ifdef TRACE
  496.     if (Trace_Flag)
  497.       printf("1): %dn",val+1);
  498. #endif /* TRACE */
  499.     return val + 1;
  500.   }
  501.   /* codes 00010 ... 011xx */
  502.   if (code>=128)
  503.   {
  504.     /* remove leading zeros */
  505.     code >>= 6;
  506.     Flush_Buffer(MBAtab1[code].len);
  507. #ifdef TRACE
  508.     if (Trace_Flag)
  509.     {
  510.       Print_Bits(code,5,MBAtab1[code].len);
  511.       printf("): %dn",val+MBAtab1[code].val);
  512.     }
  513. #endif /* TRACE */
  514.     
  515.     return val + MBAtab1[code].val;
  516.   }
  517.   
  518.   /* codes 00000011000 ... 0000111xxxx */
  519.   code-= 24; /* remove common base */
  520.   Flush_Buffer(MBAtab2[code].len);
  521. #ifdef TRACE
  522.   if (Trace_Flag)
  523.   {
  524.     Print_Bits(code+24,11,MBAtab2[code].len);
  525.     printf("): %dn",val+MBAtab2[code].val);
  526.   }
  527. #endif /* TRACE */
  528.   return val + MBAtab2[code].val;
  529. }
  530. /* combined MPEG-1 and MPEG-2 stage. parse VLC and 
  531.    perform dct_diff arithmetic.
  532.    MPEG-1:  ISO/IEC 11172-2 section
  533.    MPEG-2:  ISO/IEC 13818-2 section 7.2.1 
  534.    
  535.    Note: the arithmetic here is presented more elegantly than
  536.    the spec, yet the results, dct_diff, are the same.
  537. */
  538. int Get_Luma_DC_dct_diff()
  539. {
  540.   int code, size, dct_diff;
  541. #ifdef TRACE
  542. /*
  543.   if (Trace_Flag)
  544.     printf("dct_dc_size_luminance: (");
  545. */
  546. #endif /* TRACE */
  547.   /* decode length */
  548.   code = Show_Bits(5);
  549.   if (code<31)
  550.   {
  551.     size = DClumtab0[code].val;
  552.     Flush_Buffer(DClumtab0[code].len);
  553. #ifdef TRACE
  554. /*
  555.     if (Trace_Flag)
  556.     {
  557.       Print_Bits(code,5,DClumtab0[code].len);
  558.       printf("): %d",size);
  559.     }
  560. */
  561. #endif /* TRACE */
  562.   }
  563.   else
  564.   {
  565.     code = Show_Bits(9) - 0x1f0;
  566.     size = DClumtab1[code].val;
  567.     Flush_Buffer(DClumtab1[code].len);
  568. #ifdef TRACE
  569. /*
  570.     if (Trace_Flag)
  571.     {
  572.       Print_Bits(code+0x1f0,9,DClumtab1[code].len);
  573.       printf("): %d",size);
  574.     }
  575. */
  576. #endif /* TRACE */
  577.   }
  578. #ifdef TRACE
  579. /*
  580.   if (Trace_Flag)
  581.     printf(", dct_dc_differential (");
  582. */
  583. #endif /* TRACE */
  584.   if (size==0)
  585.     dct_diff = 0;
  586.   else
  587.   {
  588.     dct_diff = Get_Bits(size);
  589. #ifdef TRACE
  590. /*
  591.     if (Trace_Flag)
  592.       Print_Bits(dct_diff,size,size);
  593. */
  594. #endif /* TRACE */
  595.     if ((dct_diff & (1<<(size-1)))==0)
  596.       dct_diff-= (1<<size) - 1;
  597.   }
  598. #ifdef TRACE
  599. /*
  600.   if (Trace_Flag)
  601.     printf("): %dn",dct_diff);
  602. */
  603. #endif /* TRACE */
  604.   return dct_diff;
  605. }
  606. int Get_Chroma_DC_dct_diff()
  607. {
  608.   int code, size, dct_diff;
  609. #ifdef TRACE
  610. /*
  611.   if (Trace_Flag)
  612.     printf("dct_dc_size_chrominance: (");
  613. */
  614. #endif /* TRACE */
  615.   /* decode length */
  616.   code = Show_Bits(5);
  617.   if (code<31)
  618.   {
  619.     size = DCchromtab0[code].val;
  620.     Flush_Buffer(DCchromtab0[code].len);
  621. #ifdef TRACE
  622. /*
  623.     if (Trace_Flag)
  624.     {
  625.       Print_Bits(code,5,DCchromtab0[code].len);
  626.       printf("): %d",size);
  627.     }
  628. */
  629. #endif /* TRACE */
  630.   }
  631.   else
  632.   {
  633.     code = Show_Bits(10) - 0x3e0;
  634.     size = DCchromtab1[code].val;
  635.     Flush_Buffer(DCchromtab1[code].len);
  636. #ifdef TRACE
  637. /*
  638.     if (Trace_Flag)
  639.     {
  640.       Print_Bits(code+0x3e0,10,DCchromtab1[code].len);
  641.       printf("): %d",size);
  642.     }
  643. */
  644. #endif /* TRACE */
  645.   }
  646. #ifdef TRACE
  647. /* 
  648.   if (Trace_Flag)
  649.     printf(", dct_dc_differential (");
  650. */
  651. #endif /* TRACE */
  652.   if (size==0)
  653.     dct_diff = 0;
  654.   else
  655.   {
  656.     dct_diff = Get_Bits(size);
  657. #ifdef TRACE
  658. /*
  659.     if (Trace_Flag)
  660.       Print_Bits(dct_diff,size,size);
  661. */
  662. #endif /* TRACE */
  663.     if ((dct_diff & (1<<(size-1)))==0)
  664.       dct_diff-= (1<<size) - 1;
  665.   }
  666. #ifdef TRACE
  667. /*
  668.   if (Trace_Flag)
  669.     printf("): %dn",dct_diff);
  670. */
  671. #endif /* TRACE */
  672.   return dct_diff;
  673. }