huffyuv.c
上传用户:wstnjxml
上传日期:2014-04-03
资源大小:7248k
文件大小:41k
源码类别:

Windows CE

开发平台:

C/C++

  1. /*
  2.  * huffyuv codec for libavcodec
  3.  *
  4.  * Copyright (c) 2002-2003 Michael Niedermayer <michaelni@gmx.at>
  5.  *
  6.  * This library is free software; you can redistribute it and/or
  7.  * modify it under the terms of the GNU Lesser General Public
  8.  * License as published by the Free Software Foundation; either
  9.  * version 2 of the License, or (at your option) any later version.
  10.  *
  11.  * This library is distributed in the hope that it will be useful,
  12.  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  13.  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
  14.  * Lesser General Public License for more details.
  15.  *
  16.  * You should have received a copy of the GNU Lesser General Public
  17.  * License along with this library; if not, write to the Free Software
  18.  * Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
  19.  *
  20.  * see http://www.pcisys.net/~melanson/codecs/huffyuv.txt for a description of
  21.  * the algorithm used 
  22.  */
  23.  
  24. /**
  25.  * @file huffyuv.c
  26.  * huffyuv codec for libavcodec.
  27.  */
  28. #include "common.h"
  29. #include "bitstream.h"
  30. #include "avcodec.h"
  31. #include "dsputil.h"
  32. #define VLC_BITS 11
  33. #ifdef WORDS_BIGENDIAN
  34. #define B 3
  35. #define G 2
  36. #define R 1
  37. #else
  38. #define B 0
  39. #define G 1
  40. #define R 2
  41. #endif
  42. typedef enum Predictor{
  43.     LEFT= 0,
  44.     PLANE,
  45.     MEDIAN,
  46. } Predictor;
  47.  
  48. typedef struct HYuvContext{
  49.     AVCodecContext *avctx;
  50.     Predictor predictor;
  51.     GetBitContext gb;
  52.     PutBitContext pb;
  53.     int interlaced;
  54.     int decorrelate;
  55.     int bitstream_bpp;
  56.     int version;
  57.     int yuy2;                               //use yuy2 instead of 422P
  58.     int bgr32;                              //use bgr32 instead of bgr24
  59.     int width, height;
  60.     int flags;
  61.     int context;
  62.     int picture_number;
  63.     int last_slice_end;
  64.     uint8_t *temp[3];
  65.     uint64_t stats[3][256];
  66.     uint8_t len[3][256];
  67.     uint32_t bits[3][256];
  68.     VLC vlc[3];
  69.     AVFrame picture;
  70.     uint8_t *bitstream_buffer;
  71.     int bitstream_buffer_size;
  72.     DSPContext dsp; 
  73. }HYuvContext;
  74. static const unsigned char classic_shift_luma[] = {
  75.   34,36,35,69,135,232,9,16,10,24,11,23,12,16,13,10,14,8,15,8,
  76.   16,8,17,20,16,10,207,206,205,236,11,8,10,21,9,23,8,8,199,70,
  77.   69,68, 0
  78. };
  79. static const unsigned char classic_shift_chroma[] = {
  80.   66,36,37,38,39,40,41,75,76,77,110,239,144,81,82,83,84,85,118,183,
  81.   56,57,88,89,56,89,154,57,58,57,26,141,57,56,58,57,58,57,184,119,
  82.   214,245,116,83,82,49,80,79,78,77,44,75,41,40,39,38,37,36,34, 0
  83. };
  84. static const unsigned char classic_add_luma[256] = {
  85.     3,  9,  5, 12, 10, 35, 32, 29, 27, 50, 48, 45, 44, 41, 39, 37,
  86.    73, 70, 68, 65, 64, 61, 58, 56, 53, 50, 49, 46, 44, 41, 38, 36,
  87.    68, 65, 63, 61, 58, 55, 53, 51, 48, 46, 45, 43, 41, 39, 38, 36,
  88.    35, 33, 32, 30, 29, 27, 26, 25, 48, 47, 46, 44, 43, 41, 40, 39,
  89.    37, 36, 35, 34, 32, 31, 30, 28, 27, 26, 24, 23, 22, 20, 19, 37,
  90.    35, 34, 33, 31, 30, 29, 27, 26, 24, 23, 21, 20, 18, 17, 15, 29,
  91.    27, 26, 24, 22, 21, 19, 17, 16, 14, 26, 25, 23, 21, 19, 18, 16,
  92.    15, 27, 25, 23, 21, 19, 17, 16, 14, 26, 25, 23, 21, 18, 17, 14,
  93.    12, 17, 19, 13,  4,  9,  2, 11,  1,  7,  8,  0, 16,  3, 14,  6,
  94.    12, 10,  5, 15, 18, 11, 10, 13, 15, 16, 19, 20, 22, 24, 27, 15,
  95.    18, 20, 22, 24, 26, 14, 17, 20, 22, 24, 27, 15, 18, 20, 23, 25,
  96.    28, 16, 19, 22, 25, 28, 32, 36, 21, 25, 29, 33, 38, 42, 45, 49,
  97.    28, 31, 34, 37, 40, 42, 44, 47, 49, 50, 52, 54, 56, 57, 59, 60,
  98.    62, 64, 66, 67, 69, 35, 37, 39, 40, 42, 43, 45, 47, 48, 51, 52,
  99.    54, 55, 57, 59, 60, 62, 63, 66, 67, 69, 71, 72, 38, 40, 42, 43,
  100.    46, 47, 49, 51, 26, 28, 30, 31, 33, 34, 18, 19, 11, 13,  7,  8,
  101. };
  102. static const unsigned char classic_add_chroma[256] = {
  103.     3,  1,  2,  2,  2,  2,  3,  3,  7,  5,  7,  5,  8,  6, 11,  9,
  104.     7, 13, 11, 10,  9,  8,  7,  5,  9,  7,  6,  4,  7,  5,  8,  7,
  105.    11,  8, 13, 11, 19, 15, 22, 23, 20, 33, 32, 28, 27, 29, 51, 77,
  106.    43, 45, 76, 81, 46, 82, 75, 55, 56,144, 58, 80, 60, 74,147, 63,
  107.   143, 65, 66, 67, 68, 69, 70, 71, 72, 73, 74, 75, 76, 77, 78, 79,
  108.    80, 81, 82, 83, 84, 85, 86, 87, 88, 89, 90, 91, 27, 30, 21, 22,
  109.    17, 14,  5,  6,100, 54, 47, 50, 51, 53,106,107,108,109,110,111,
  110.   112,113,114,115,  4,117,118, 92, 94,121,122,  3,124,103,  2,  1,
  111.     0,129,130,131,120,119,126,125,136,137,138,139,140,141,142,134,
  112.   135,132,133,104, 64,101, 62, 57,102, 95, 93, 59, 61, 28, 97, 96,
  113.    52, 49, 48, 29, 32, 25, 24, 46, 23, 98, 45, 44, 43, 20, 42, 41,
  114.    19, 18, 99, 40, 15, 39, 38, 16, 13, 12, 11, 37, 10,  9,  8, 36,
  115.     7,128,127,105,123,116, 35, 34, 33,145, 31, 79, 42,146, 78, 26,
  116.    83, 48, 49, 50, 44, 47, 26, 31, 30, 18, 17, 19, 21, 24, 25, 13,
  117.    14, 16, 17, 18, 20, 21, 12, 14, 15,  9, 10,  6,  9,  6,  5,  8,
  118.     6, 12,  8, 10,  7,  9,  6,  4,  6,  2,  2,  3,  3,  3,  3,  2,
  119. };
  120. static inline int add_left_prediction(uint8_t *dst, uint8_t *src, int w, int acc){
  121.     int i;
  122.     for(i=0; i<w-1; i++){
  123.         acc+= src[i];
  124.         dst[i]= acc;
  125.         i++;
  126.         acc+= src[i];
  127.         dst[i]= acc;
  128.     }
  129.     for(; i<w; i++){
  130.         acc+= src[i];
  131.         dst[i]= acc;
  132.     }
  133.     return acc;
  134. }
  135. static inline void add_median_prediction(uint8_t *dst, uint8_t *src1, uint8_t *diff, int w, int *left, int *left_top){
  136.     int i;
  137.     uint8_t l, lt;
  138.     l= *left;
  139.     lt= *left_top;
  140.     for(i=0; i<w; i++){
  141.         l= mid_pred(l, src1[i], (l + src1[i] - lt)&0xFF) + diff[i];
  142.         lt= src1[i];
  143.         dst[i]= l;
  144.     }    
  145.     *left= l;
  146.     *left_top= lt;
  147. }
  148. static inline void add_left_prediction_bgr32(uint8_t *dst, uint8_t *src, int w, int *red, int *green, int *blue){
  149.     int i;
  150.     int r,g,b;
  151.     r= *red;
  152.     g= *green;
  153.     b= *blue;
  154.     for(i=0; i<w; i++){
  155.         b+= src[4*i+B];
  156.         g+= src[4*i+G];
  157.         r+= src[4*i+R];
  158.         
  159.         dst[4*i+B]= b;
  160.         dst[4*i+G]= g;
  161.         dst[4*i+R]= r;
  162.     }
  163.     *red= r;
  164.     *green= g;
  165.     *blue= b;
  166. }
  167. static inline int sub_left_prediction(HYuvContext *s, uint8_t *dst, uint8_t *src, int w, int left){
  168.     int i;
  169.     if(w<32){
  170.         for(i=0; i<w; i++){
  171.             const int temp= src[i];
  172.             dst[i]= temp - left;
  173.             left= temp;
  174.         }
  175.         return left;
  176.     }else{
  177.         for(i=0; i<16; i++){
  178.             const int temp= src[i];
  179.             dst[i]= temp - left;
  180.             left= temp;
  181.         }
  182.         s->dsp.diff_bytes(dst+16, src+16, src+15, w-16);
  183.         return src[w-1];
  184.     }
  185. }
  186. static void read_len_table(uint8_t *dst, GetBitContext *gb){
  187.     int i, val, repeat;
  188.   
  189.     for(i=0; i<256;){
  190.         repeat= get_bits(gb, 3);
  191.         val   = get_bits(gb, 5);
  192.         if(repeat==0)
  193.             repeat= get_bits(gb, 8);
  194. //printf("%d %dn", val, repeat);
  195.         while (repeat--)
  196.             dst[i++] = val;
  197.     }
  198. }
  199. static int generate_bits_table(uint32_t *dst, uint8_t *len_table){
  200.     int len, index;
  201.     uint32_t bits=0;
  202.     for(len=32; len>0; len--){
  203.         for(index=0; index<256; index++){
  204.             if(len_table[index]==len)
  205.                 dst[index]= bits++;
  206.         }
  207.         if(bits & 1){
  208.             av_log(NULL, AV_LOG_ERROR, "Error generating huffman tablen");
  209.             return -1;
  210.         }
  211.         bits >>= 1;
  212.     }
  213.     return 0;
  214. }
  215. static void generate_len_table(uint8_t *dst, uint64_t *stats, int size){
  216.     uint64_t counts[2*size];
  217.     int up[2*size];
  218.     int offset, i, next;
  219.     
  220.     for(offset=1; ; offset<<=1){
  221.         for(i=0; i<size; i++){
  222.             counts[i]= stats[i] + offset - 1;
  223.         }
  224.         
  225.         for(next=size; next<size*2; next++){
  226.             uint64_t min1, min2;
  227.             int min1_i, min2_i;
  228.             
  229.             min1=min2= INT64_MAX;
  230.             min1_i= min2_i=-1;
  231.             
  232.             for(i=0; i<next; i++){
  233.                 if(min2 > counts[i]){
  234.                     if(min1 > counts[i]){
  235.                         min2= min1;
  236.                         min2_i= min1_i;
  237.                         min1= counts[i];
  238.                         min1_i= i;
  239.                     }else{
  240.                         min2= counts[i];
  241.                         min2_i= i;
  242.                     }
  243.                 }
  244.             }
  245.             
  246.             if(min2==INT64_MAX) break;
  247.             
  248.             counts[next]= min1 + min2;
  249.             counts[min1_i]=
  250.             counts[min2_i]= INT64_MAX;
  251.             up[min1_i]=
  252.             up[min2_i]= next;
  253.             up[next]= -1;
  254.         }
  255.         
  256.         for(i=0; i<size; i++){
  257.             int len;
  258.             int index=i;
  259.             
  260.             for(len=0; up[index] != -1; len++)
  261.                 index= up[index];
  262.                 
  263.             if(len >= 32) break;
  264.             
  265.             dst[i]= len;
  266.         }
  267.         if(i==size) break;
  268.     }
  269. }
  270. static int read_huffman_tables(HYuvContext *s, uint8_t *src, int length){
  271.     GetBitContext gb;
  272.     int i;
  273.     
  274.     init_get_bits(&gb, src, length*8);
  275.     
  276.     for(i=0; i<3; i++){
  277.         read_len_table(s->len[i], &gb);
  278.         
  279.         if(generate_bits_table(s->bits[i], s->len[i])<0){
  280.             return -1;
  281.         }
  282. #if 0
  283. for(j=0; j<256; j++){
  284. printf("%6X, %2d,  %3dn", s->bits[i][j], s->len[i][j], j);
  285. }
  286. #endif
  287.         free_vlc(&s->vlc[i]);
  288.         init_vlc(&s->vlc[i], VLC_BITS, 256, s->len[i], 1, 1, s->bits[i], 4, 4, 0);
  289.     }
  290.     
  291.     return (get_bits_count(&gb)+7)/8;
  292. }
  293. static int read_old_huffman_tables(HYuvContext *s){
  294. #if 1
  295.     GetBitContext gb;
  296.     int i;
  297.     init_get_bits(&gb, classic_shift_luma, sizeof(classic_shift_luma)*8);
  298.     read_len_table(s->len[0], &gb);
  299.     init_get_bits(&gb, classic_shift_chroma, sizeof(classic_shift_chroma)*8);
  300.     read_len_table(s->len[1], &gb);
  301.     
  302.     for(i=0; i<256; i++) s->bits[0][i] = classic_add_luma  [i];
  303.     for(i=0; i<256; i++) s->bits[1][i] = classic_add_chroma[i];
  304.     if(s->bitstream_bpp >= 24){
  305.         memcpy(s->bits[1], s->bits[0], 256*sizeof(uint32_t));
  306.         memcpy(s->len[1] , s->len [0], 256*sizeof(uint8_t));
  307.     }
  308.     memcpy(s->bits[2], s->bits[1], 256*sizeof(uint32_t));
  309.     memcpy(s->len[2] , s->len [1], 256*sizeof(uint8_t));
  310.     
  311.     for(i=0; i<3; i++){
  312.         free_vlc(&s->vlc[i]);
  313.         init_vlc(&s->vlc[i], VLC_BITS, 256, s->len[i], 1, 1, s->bits[i], 4, 4, 0);
  314.     }
  315.     
  316.     return 0;
  317. #else
  318.     fprintf(stderr, "v1 huffyuv is not supported n");
  319.     return -1;
  320. #endif
  321. }
  322. static void alloc_temp(HYuvContext *s){
  323.     int i;
  324.     
  325.     if(s->bitstream_bpp<24){
  326.         for(i=0; i<3; i++){
  327.             s->temp[i]= av_malloc(s->width + 16);
  328.         }
  329.     }else{
  330.         s->temp[0]= av_malloc(4*s->width + 16);
  331.     }
  332. }
  333. static int common_init(AVCodecContext *avctx){
  334.     HYuvContext *s = avctx->priv_data;
  335.     s->avctx= avctx;
  336.     s->flags= avctx->flags;
  337.         
  338.     dsputil_init(&s->dsp, avctx);
  339.     
  340.     s->width= avctx->width;
  341.     s->height= avctx->height;
  342.     assert(s->width>0 && s->height>0);
  343.         
  344.     return 0;
  345. }
  346. static int decode_init(AVCodecContext *avctx)
  347. {
  348.     HYuvContext *s = avctx->priv_data;
  349.     common_init(avctx);
  350.     memset(s->vlc, 0, 3*sizeof(VLC));
  351.     
  352.     avctx->coded_frame= &s->picture;
  353.     s->interlaced= s->height > 288;
  354. s->bgr32=1;
  355. //if(avctx->extradata)
  356. //  printf("extradata:%X, extradata_size:%dn", *(uint32_t*)avctx->extradata, avctx->extradata_size);
  357.     if(avctx->extradata_size){
  358.         if((avctx->bits_per_sample&7) && avctx->bits_per_sample != 12)
  359.             s->version=1; // do such files exist at all?
  360.         else
  361.             s->version=2;
  362.     }else
  363.         s->version=0;
  364.     
  365.     if(s->version==2){
  366.         int method, interlace;
  367.         method= ((uint8_t*)avctx->extradata)[0];
  368.         s->decorrelate= method&64 ? 1 : 0;
  369.         s->predictor= method&63;
  370.         s->bitstream_bpp= ((uint8_t*)avctx->extradata)[1];
  371.         if(s->bitstream_bpp==0) 
  372.             s->bitstream_bpp= avctx->bits_per_sample&~7;
  373.         interlace= (((uint8_t*)avctx->extradata)[2] & 0x30) >> 4;
  374.         s->interlaced= (interlace==1) ? 1 : (interlace==2) ? 0 : s->interlaced;
  375.         s->context= ((uint8_t*)avctx->extradata)[2] & 0x40 ? 1 : 0;
  376.             
  377.         if(read_huffman_tables(s, ((uint8_t*)avctx->extradata)+4, avctx->extradata_size) < 0)
  378.             return -1;
  379.     }else{
  380.         switch(avctx->bits_per_sample&7){
  381.         case 1:
  382.             s->predictor= LEFT;
  383.             s->decorrelate= 0;
  384.             break;
  385.         case 2:
  386.             s->predictor= LEFT;
  387.             s->decorrelate= 1;
  388.             break;
  389.         case 3:
  390.             s->predictor= PLANE;
  391.             s->decorrelate= avctx->bits_per_sample >= 24;
  392.             break;
  393.         case 4:
  394.             s->predictor= MEDIAN;
  395.             s->decorrelate= 0;
  396.             break;
  397.         default:
  398.             s->predictor= LEFT; //OLD
  399.             s->decorrelate= 0;
  400.             break;
  401.         }
  402.         s->bitstream_bpp= avctx->bits_per_sample & ~7;
  403.         s->context= 0;
  404.         
  405.         if(read_old_huffman_tables(s) < 0)
  406.             return -1;
  407.     }
  408.     
  409.     switch(s->bitstream_bpp){
  410.     case 12:
  411.         avctx->pix_fmt = PIX_FMT_YUV420P;
  412.         break;
  413.     case 16:
  414.         if(s->yuy2){
  415.             avctx->pix_fmt = PIX_FMT_YUV422;
  416.         }else{
  417.             avctx->pix_fmt = PIX_FMT_YUV422P;
  418.         }
  419.         break;
  420.     case 24:
  421.     case 32:
  422.         if(s->bgr32){
  423.             avctx->pix_fmt = PIX_FMT_RGBA32;
  424.         }else{
  425.             avctx->pix_fmt = PIX_FMT_BGR24;
  426.         }
  427.         break;
  428.     default:
  429.         assert(0);
  430.     }
  431.     
  432.     alloc_temp(s);
  433.     
  434. //    av_log(NULL, AV_LOG_DEBUG, "pred:%d bpp:%d hbpp:%d il:%dn", s->predictor, s->bitstream_bpp, avctx->bits_per_sample, s->interlaced);
  435.     return 0;
  436. }
  437. static int store_table(HYuvContext *s, uint8_t *len, uint8_t *buf){
  438.     int i;
  439.     int index= 0;
  440.     for(i=0; i<256;){
  441.         int val= len[i];
  442.         int repeat=0;
  443.         
  444.         for(; i<256 && len[i]==val && repeat<255; i++)
  445.             repeat++;
  446.         
  447.         assert(val < 32 && val >0 && repeat<256 && repeat>0);
  448.         if(repeat>7){
  449.             buf[index++]= val;
  450.             buf[index++]= repeat;
  451.         }else{
  452.             buf[index++]= val | (repeat<<5);
  453.         }
  454.     }
  455.     
  456.     return index;
  457. }
  458. static int encode_init(AVCodecContext *avctx)
  459. {
  460.     HYuvContext *s = avctx->priv_data;
  461.     int i, j;
  462.     common_init(avctx);
  463.     
  464.     avctx->extradata= av_mallocz(1024*30); // 256*3+4 == 772
  465.     avctx->stats_out= av_mallocz(1024*30); // 21*256*3(%llu ) + 3(n) + 1(0) = 16132
  466.     s->version=2;
  467.     
  468.     avctx->coded_frame= &s->picture;
  469.     
  470.     switch(avctx->pix_fmt){
  471.     case PIX_FMT_YUV420P:
  472.         s->bitstream_bpp= 12;
  473.         break;
  474.     case PIX_FMT_YUV422P:
  475.         s->bitstream_bpp= 16;
  476.         break;
  477.     default:
  478.         av_log(avctx, AV_LOG_ERROR, "format not supportedn");
  479.         return -1;
  480.     }
  481.     avctx->bits_per_sample= s->bitstream_bpp;
  482.     s->decorrelate= s->bitstream_bpp >= 24;
  483.     s->predictor= avctx->prediction_method;
  484.     s->interlaced= avctx->flags&CODEC_FLAG_INTERLACED_ME ? 1 : 0;
  485.     if(avctx->context_model==1){
  486.         s->context= avctx->context_model;
  487.         if(s->flags & (CODEC_FLAG_PASS1|CODEC_FLAG_PASS2)){
  488.             av_log(avctx, AV_LOG_ERROR, "context=1 is not compatible with 2 pass huffyuv encodingn");
  489.             return -1;
  490.         }
  491.     }else s->context= 0;
  492.     
  493.     if(avctx->codec->id==CODEC_ID_HUFFYUV){
  494.         if(avctx->pix_fmt==PIX_FMT_YUV420P){
  495.             av_log(avctx, AV_LOG_ERROR, "Error: YV12 is not supported by huffyuv; use vcodec=ffvhuff or format=422pn");
  496.             return -1;
  497.         }
  498.         if(avctx->context_model){
  499.             av_log(avctx, AV_LOG_ERROR, "Error: per-frame huffman tables are not supported by huffyuv; use vcodec=ffvhuffn");
  500.             return -1;
  501.         }
  502.         if(s->interlaced != ( s->height > 288 ))
  503.             av_log(avctx, AV_LOG_INFO, "using huffyuv 2.2.0 or newer interlacing flagn");
  504.     }else if(avctx->strict_std_compliance>FF_COMPLIANCE_EXPERIMENTAL){
  505.         av_log(avctx, AV_LOG_ERROR, "This codec is under development; files encoded with it may not be decodable with future versions!!! Set vstrict=-2 / -strict -2 to use it anyway.n");
  506.         return -1;
  507.     }
  508.     
  509.     ((uint8_t*)avctx->extradata)[0]= s->predictor;
  510.     ((uint8_t*)avctx->extradata)[1]= s->bitstream_bpp;
  511.     ((uint8_t*)avctx->extradata)[2]= s->interlaced ? 0x10 : 0x20;
  512.     if(s->context)
  513.         ((uint8_t*)avctx->extradata)[2]|= 0x40;
  514.     ((uint8_t*)avctx->extradata)[3]= 0;
  515.     s->avctx->extradata_size= 4;
  516.     
  517.     if(avctx->stats_in){
  518.         char *p= avctx->stats_in;
  519.     
  520.         for(i=0; i<3; i++)
  521.             for(j=0; j<256; j++)
  522.                 s->stats[i][j]= 1;
  523.         for(;;){
  524.             for(i=0; i<3; i++){
  525.                 char *next;
  526.                 for(j=0; j<256; j++){
  527.                     s->stats[i][j]+= strtol(p, &next, 0);
  528.                     if(next==p) return -1;
  529.                     p=next;
  530.                 }        
  531.             }
  532.             if(p[0]==0 || p[1]==0 || p[2]==0) break;
  533.         }
  534.     }else{
  535.         for(i=0; i<3; i++)
  536.             for(j=0; j<256; j++){
  537.                 int d= FFMIN(j, 256-j);
  538.                 
  539.                 s->stats[i][j]= 100000000/(d+1);
  540.             }
  541.     }
  542.     
  543.     for(i=0; i<3; i++){
  544.         generate_len_table(s->len[i], s->stats[i], 256);
  545.         if(generate_bits_table(s->bits[i], s->len[i])<0){
  546.             return -1;
  547.         }
  548.         
  549.         s->avctx->extradata_size+=
  550.         store_table(s, s->len[i], &((uint8_t*)s->avctx->extradata)[s->avctx->extradata_size]);
  551.     }
  552.     if(s->context){
  553.         for(i=0; i<3; i++){
  554.             int pels = s->width*s->height / (i?40:10);
  555.             for(j=0; j<256; j++){
  556.                 int d= FFMIN(j, 256-j);
  557.                 s->stats[i][j]= pels/(d+1);
  558.             }
  559.         }
  560.     }else{
  561.         for(i=0; i<3; i++)
  562.             for(j=0; j<256; j++)
  563.                 s->stats[i][j]= 0;
  564.     }
  565.     
  566. //    printf("pred:%d bpp:%d hbpp:%d il:%dn", s->predictor, s->bitstream_bpp, avctx->bits_per_sample, s->interlaced);
  567.     alloc_temp(s);
  568.     s->picture_number=0;
  569.     return 0;
  570. }
  571. static void decode_422_bitstream(HYuvContext *s, int count){
  572.     int i;
  573.     count/=2;
  574.     
  575.     for(i=0; i<count; i++){
  576.         s->temp[0][2*i  ]= get_vlc2(&s->gb, s->vlc[0].table, VLC_BITS, 3); 
  577.         s->temp[1][  i  ]= get_vlc2(&s->gb, s->vlc[1].table, VLC_BITS, 3); 
  578.         s->temp[0][2*i+1]= get_vlc2(&s->gb, s->vlc[0].table, VLC_BITS, 3); 
  579.         s->temp[2][  i  ]= get_vlc2(&s->gb, s->vlc[2].table, VLC_BITS, 3); 
  580.     }
  581. }
  582. static void decode_gray_bitstream(HYuvContext *s, int count){
  583.     int i;
  584.     
  585.     count/=2;
  586.     
  587.     for(i=0; i<count; i++){
  588.         s->temp[0][2*i  ]= get_vlc2(&s->gb, s->vlc[0].table, VLC_BITS, 3); 
  589.         s->temp[0][2*i+1]= get_vlc2(&s->gb, s->vlc[0].table, VLC_BITS, 3); 
  590.     }
  591. }
  592. static int encode_422_bitstream(HYuvContext *s, int count){
  593.     int i;
  594.     
  595.     if(s->pb.buf_end - s->pb.buf - (put_bits_count(&s->pb)>>3) < 2*4*count){
  596.         av_log(s->avctx, AV_LOG_ERROR, "encoded frame too largen");
  597.         return -1;
  598.     }
  599.     
  600.     count/=2;
  601.     if(s->flags&CODEC_FLAG_PASS1){
  602.         for(i=0; i<count; i++){
  603.             s->stats[0][ s->temp[0][2*i  ] ]++;
  604.             s->stats[1][ s->temp[1][  i  ] ]++;
  605.             s->stats[0][ s->temp[0][2*i+1] ]++;
  606.             s->stats[2][ s->temp[2][  i  ] ]++;
  607.         }
  608.     }
  609.     if(s->avctx->flags2&CODEC_FLAG2_NO_OUTPUT)
  610.         return 0;
  611.     if(s->context){
  612.         for(i=0; i<count; i++){
  613.             s->stats[0][ s->temp[0][2*i  ] ]++;
  614.             put_bits(&s->pb, s->len[0][ s->temp[0][2*i  ] ], s->bits[0][ s->temp[0][2*i  ] ]);
  615.             s->stats[1][ s->temp[1][  i  ] ]++;
  616.             put_bits(&s->pb, s->len[1][ s->temp[1][  i  ] ], s->bits[1][ s->temp[1][  i  ] ]);
  617.             s->stats[0][ s->temp[0][2*i+1] ]++;
  618.             put_bits(&s->pb, s->len[0][ s->temp[0][2*i+1] ], s->bits[0][ s->temp[0][2*i+1] ]);
  619.             s->stats[2][ s->temp[2][  i  ] ]++;
  620.             put_bits(&s->pb, s->len[2][ s->temp[2][  i  ] ], s->bits[2][ s->temp[2][  i  ] ]);
  621.         }
  622.     }else{
  623.         for(i=0; i<count; i++){
  624.             put_bits(&s->pb, s->len[0][ s->temp[0][2*i  ] ], s->bits[0][ s->temp[0][2*i  ] ]);
  625.             put_bits(&s->pb, s->len[1][ s->temp[1][  i  ] ], s->bits[1][ s->temp[1][  i  ] ]);
  626.             put_bits(&s->pb, s->len[0][ s->temp[0][2*i+1] ], s->bits[0][ s->temp[0][2*i+1] ]);
  627.             put_bits(&s->pb, s->len[2][ s->temp[2][  i  ] ], s->bits[2][ s->temp[2][  i  ] ]);
  628.         }
  629.     }
  630.     return 0;
  631. }
  632. static int encode_gray_bitstream(HYuvContext *s, int count){
  633.     int i;
  634.     
  635.     if(s->pb.buf_end - s->pb.buf - (put_bits_count(&s->pb)>>3) < 4*count){
  636.         av_log(s->avctx, AV_LOG_ERROR, "encoded frame too largen");
  637.         return -1;
  638.     }
  639.     count/=2;
  640.     if(s->flags&CODEC_FLAG_PASS1){
  641.         for(i=0; i<count; i++){
  642.             s->stats[0][ s->temp[0][2*i  ] ]++;
  643.             s->stats[0][ s->temp[0][2*i+1] ]++;
  644.         }
  645.     }
  646.     if(s->avctx->flags2&CODEC_FLAG2_NO_OUTPUT)
  647.         return 0;
  648.     
  649.     if(s->context){
  650.         for(i=0; i<count; i++){
  651.             s->stats[0][ s->temp[0][2*i  ] ]++;
  652.             put_bits(&s->pb, s->len[0][ s->temp[0][2*i  ] ], s->bits[0][ s->temp[0][2*i  ] ]);
  653.             s->stats[0][ s->temp[0][2*i+1] ]++;
  654.             put_bits(&s->pb, s->len[0][ s->temp[0][2*i+1] ], s->bits[0][ s->temp[0][2*i+1] ]);
  655.         }
  656.     }else{
  657.         for(i=0; i<count; i++){
  658.             put_bits(&s->pb, s->len[0][ s->temp[0][2*i  ] ], s->bits[0][ s->temp[0][2*i  ] ]);
  659.             put_bits(&s->pb, s->len[0][ s->temp[0][2*i+1] ], s->bits[0][ s->temp[0][2*i+1] ]);
  660.         }
  661.     }
  662.     return 0;
  663. }
  664. static void decode_bgr_bitstream(HYuvContext *s, int count){
  665.     int i;
  666.     if(s->decorrelate){
  667.         if(s->bitstream_bpp==24){
  668.             for(i=0; i<count; i++){
  669.                 s->temp[0][4*i+G]= get_vlc2(&s->gb, s->vlc[1].table, VLC_BITS, 3); 
  670.                 s->temp[0][4*i+B]= get_vlc2(&s->gb, s->vlc[0].table, VLC_BITS, 3) + s->temp[0][4*i+G];
  671.                 s->temp[0][4*i+R]= get_vlc2(&s->gb, s->vlc[2].table, VLC_BITS, 3) + s->temp[0][4*i+G];
  672.             }
  673.         }else{
  674.             for(i=0; i<count; i++){
  675.                 s->temp[0][4*i+G]= get_vlc2(&s->gb, s->vlc[1].table, VLC_BITS, 3); 
  676.                 s->temp[0][4*i+B]= get_vlc2(&s->gb, s->vlc[0].table, VLC_BITS, 3) + s->temp[0][4*i+G];
  677.                 s->temp[0][4*i+R]= get_vlc2(&s->gb, s->vlc[2].table, VLC_BITS, 3) + s->temp[0][4*i+G]; 
  678.                                    get_vlc2(&s->gb, s->vlc[2].table, VLC_BITS, 3); //?!
  679.             }
  680.         }
  681.     }else{
  682.         if(s->bitstream_bpp==24){
  683.             for(i=0; i<count; i++){
  684.                 s->temp[0][4*i+B]= get_vlc2(&s->gb, s->vlc[0].table, VLC_BITS, 3);
  685.                 s->temp[0][4*i+G]= get_vlc2(&s->gb, s->vlc[1].table, VLC_BITS, 3); 
  686.                 s->temp[0][4*i+R]= get_vlc2(&s->gb, s->vlc[2].table, VLC_BITS, 3); 
  687.             }
  688.         }else{
  689.             for(i=0; i<count; i++){
  690.                 s->temp[0][4*i+B]= get_vlc2(&s->gb, s->vlc[0].table, VLC_BITS, 3);
  691.                 s->temp[0][4*i+G]= get_vlc2(&s->gb, s->vlc[1].table, VLC_BITS, 3); 
  692.                 s->temp[0][4*i+R]= get_vlc2(&s->gb, s->vlc[2].table, VLC_BITS, 3); 
  693.                                    get_vlc2(&s->gb, s->vlc[2].table, VLC_BITS, 3); //?!
  694.             }
  695.         }
  696.     }
  697. }
  698. static void draw_slice(HYuvContext *s, int y){
  699.     int h, cy;
  700.     int offset[4];
  701.     
  702.     if(s->avctx->draw_horiz_band==NULL) 
  703.         return;
  704.         
  705.     h= y - s->last_slice_end;
  706.     y -= h;
  707.     
  708.     if(s->bitstream_bpp==12){
  709.         cy= y>>1;
  710.     }else{
  711.         cy= y;
  712.     }
  713.     offset[0] = s->picture.linesize[0]*y;
  714.     offset[1] = s->picture.linesize[1]*cy;
  715.     offset[2] = s->picture.linesize[2]*cy;
  716.     offset[3] = 0;
  717.     emms_c();
  718.     s->avctx->draw_horiz_band(s->avctx, &s->picture, offset, y, 3, h);
  719.     
  720.     s->last_slice_end= y + h;
  721. }
  722. static int decode_frame(AVCodecContext *avctx, void *data, int *data_size, uint8_t *buf, int buf_size){
  723.     HYuvContext *s = avctx->priv_data;
  724.     const int width= s->width;
  725.     const int width2= s->width>>1;
  726.     const int height= s->height;
  727.     int fake_ystride, fake_ustride, fake_vstride;
  728.     AVFrame * const p= &s->picture;
  729.     int table_size= 0;
  730.     AVFrame *picture = data;
  731.     s->bitstream_buffer= av_fast_realloc(s->bitstream_buffer, &s->bitstream_buffer_size, buf_size + FF_INPUT_BUFFER_PADDING_SIZE);
  732.     s->dsp.bswap_buf((uint32_t*)s->bitstream_buffer, (uint32_t*)buf, buf_size/4);
  733.     
  734.     if(p->data[0])
  735.         avctx->release_buffer(avctx, p);
  736.     p->reference= 0;
  737.     if(avctx->get_buffer(avctx, p) < 0){
  738.         av_log(avctx, AV_LOG_ERROR, "get_buffer() failedn");
  739.         return -1;
  740.     }
  741.     
  742.     if(s->context){
  743.         table_size = read_huffman_tables(s, s->bitstream_buffer, buf_size);
  744.         if(table_size < 0)
  745.             return -1;
  746.     }
  747.     init_get_bits(&s->gb, s->bitstream_buffer+table_size, (buf_size-table_size)*8);
  748.     fake_ystride= s->interlaced ? p->linesize[0]*2  : p->linesize[0];
  749.     fake_ustride= s->interlaced ? p->linesize[1]*2  : p->linesize[1];
  750.     fake_vstride= s->interlaced ? p->linesize[2]*2  : p->linesize[2];
  751.     
  752.     s->last_slice_end= 0;
  753.         
  754.     if(s->bitstream_bpp<24){
  755.         int y, cy;
  756.         int lefty, leftu, leftv;
  757.         int lefttopy, lefttopu, lefttopv;
  758.         
  759.         if(s->yuy2){
  760.             p->data[0][3]= get_bits(&s->gb, 8);
  761.             p->data[0][2]= get_bits(&s->gb, 8);
  762.             p->data[0][1]= get_bits(&s->gb, 8);
  763.             p->data[0][0]= get_bits(&s->gb, 8);
  764.             
  765.             av_log(avctx, AV_LOG_ERROR, "YUY2 output is not implemented yetn");
  766.             return -1;
  767.         }else{
  768.         
  769.             leftv= p->data[2][0]= get_bits(&s->gb, 8);
  770.             lefty= p->data[0][1]= get_bits(&s->gb, 8);
  771.             leftu= p->data[1][0]= get_bits(&s->gb, 8);
  772.                    p->data[0][0]= get_bits(&s->gb, 8);
  773.         
  774.             switch(s->predictor){
  775.             case LEFT:
  776.             case PLANE:
  777.                 decode_422_bitstream(s, width-2);
  778.                 lefty= add_left_prediction(p->data[0] + 2, s->temp[0], width-2, lefty);
  779.                 if(!(s->flags&CODEC_FLAG_GRAY)){
  780.                     leftu= add_left_prediction(p->data[1] + 1, s->temp[1], width2-1, leftu);
  781.                     leftv= add_left_prediction(p->data[2] + 1, s->temp[2], width2-1, leftv);
  782.                 }
  783.                 for(cy=y=1; y<s->height; y++,cy++){
  784.                     uint8_t *ydst, *udst, *vdst;
  785.                     
  786.                     if(s->bitstream_bpp==12){
  787.                         decode_gray_bitstream(s, width);
  788.                     
  789.                         ydst= p->data[0] + p->linesize[0]*y;
  790.                         lefty= add_left_prediction(ydst, s->temp[0], width, lefty);
  791.                         if(s->predictor == PLANE){
  792.                             if(y>s->interlaced)
  793.                                 s->dsp.add_bytes(ydst, ydst - fake_ystride, width);
  794.                         }
  795.                         y++;
  796.                         if(y>=s->height) break;
  797.                     }
  798.                     
  799.                     draw_slice(s, y);
  800.                     
  801.                     ydst= p->data[0] + p->linesize[0]*y;
  802.                     udst= p->data[1] + p->linesize[1]*cy;
  803.                     vdst= p->data[2] + p->linesize[2]*cy;
  804.                     
  805.                     decode_422_bitstream(s, width);
  806.                     lefty= add_left_prediction(ydst, s->temp[0], width, lefty);
  807.                     if(!(s->flags&CODEC_FLAG_GRAY)){
  808.                         leftu= add_left_prediction(udst, s->temp[1], width2, leftu);
  809.                         leftv= add_left_prediction(vdst, s->temp[2], width2, leftv);
  810.                     }
  811.                     if(s->predictor == PLANE){
  812.                         if(cy>s->interlaced){
  813.                             s->dsp.add_bytes(ydst, ydst - fake_ystride, width);
  814.                             if(!(s->flags&CODEC_FLAG_GRAY)){
  815.                                 s->dsp.add_bytes(udst, udst - fake_ustride, width2);
  816.                                 s->dsp.add_bytes(vdst, vdst - fake_vstride, width2);
  817.                             }
  818.                         }
  819.                     }
  820.                 }
  821.                 draw_slice(s, height);
  822.                 
  823.                 break;
  824.             case MEDIAN:
  825.                 /* first line except first 2 pixels is left predicted */
  826.                 decode_422_bitstream(s, width-2);
  827.                 lefty= add_left_prediction(p->data[0] + 2, s->temp[0], width-2, lefty);
  828.                 if(!(s->flags&CODEC_FLAG_GRAY)){
  829.                     leftu= add_left_prediction(p->data[1] + 1, s->temp[1], width2-1, leftu);
  830.                     leftv= add_left_prediction(p->data[2] + 1, s->temp[2], width2-1, leftv);
  831.                 }
  832.                 
  833.                 cy=y=1;
  834.                 
  835.                 /* second line is left predicted for interlaced case */
  836.                 if(s->interlaced){
  837.                     decode_422_bitstream(s, width);
  838.                     lefty= add_left_prediction(p->data[0] + p->linesize[0], s->temp[0], width, lefty);
  839.                     if(!(s->flags&CODEC_FLAG_GRAY)){
  840.                         leftu= add_left_prediction(p->data[1] + p->linesize[2], s->temp[1], width2, leftu);
  841.                         leftv= add_left_prediction(p->data[2] + p->linesize[1], s->temp[2], width2, leftv);
  842.                     }
  843.                     y++; cy++;
  844.                 }
  845.                 /* next 4 pixels are left predicted too */
  846.                 decode_422_bitstream(s, 4);
  847.                 lefty= add_left_prediction(p->data[0] + fake_ystride, s->temp[0], 4, lefty);
  848.                 if(!(s->flags&CODEC_FLAG_GRAY)){
  849.                     leftu= add_left_prediction(p->data[1] + fake_ustride, s->temp[1], 2, leftu);
  850.                     leftv= add_left_prediction(p->data[2] + fake_vstride, s->temp[2], 2, leftv);
  851.                 }
  852.                 /* next line except the first 4 pixels is median predicted */
  853.                 lefttopy= p->data[0][3];
  854.                 decode_422_bitstream(s, width-4);
  855.                 add_median_prediction(p->data[0] + fake_ystride+4, p->data[0]+4, s->temp[0], width-4, &lefty, &lefttopy);
  856.                 if(!(s->flags&CODEC_FLAG_GRAY)){
  857.                     lefttopu= p->data[1][1];
  858.                     lefttopv= p->data[2][1];
  859.                     add_median_prediction(p->data[1] + fake_ustride+2, p->data[1]+2, s->temp[1], width2-2, &leftu, &lefttopu);
  860.                     add_median_prediction(p->data[2] + fake_vstride+2, p->data[2]+2, s->temp[2], width2-2, &leftv, &lefttopv);
  861.                 }
  862.                 y++; cy++;
  863.                 
  864.                 for(; y<height; y++,cy++){
  865.                     uint8_t *ydst, *udst, *vdst;
  866.                     if(s->bitstream_bpp==12){
  867.                         while(2*cy > y){
  868.                             decode_gray_bitstream(s, width);
  869.                             ydst= p->data[0] + p->linesize[0]*y;
  870.                             add_median_prediction(ydst, ydst - fake_ystride, s->temp[0], width, &lefty, &lefttopy);
  871.                             y++;
  872.                         }
  873.                         if(y>=height) break;
  874.                     }
  875.                     draw_slice(s, y);
  876.                     decode_422_bitstream(s, width);
  877.                     ydst= p->data[0] + p->linesize[0]*y;
  878.                     udst= p->data[1] + p->linesize[1]*cy;
  879.                     vdst= p->data[2] + p->linesize[2]*cy;
  880.                     add_median_prediction(ydst, ydst - fake_ystride, s->temp[0], width, &lefty, &lefttopy);
  881.                     if(!(s->flags&CODEC_FLAG_GRAY)){
  882.                         add_median_prediction(udst, udst - fake_ustride, s->temp[1], width2, &leftu, &lefttopu);
  883.                         add_median_prediction(vdst, vdst - fake_vstride, s->temp[2], width2, &leftv, &lefttopv);
  884.                     }
  885.                 }
  886.                 draw_slice(s, height);
  887.                 break;
  888.             }
  889.         }
  890.     }else{
  891.         int y;
  892.         int leftr, leftg, leftb;
  893.         const int last_line= (height-1)*p->linesize[0];
  894.         
  895.         if(s->bitstream_bpp==32){
  896.             skip_bits(&s->gb, 8);
  897.             leftr= p->data[0][last_line+R]= get_bits(&s->gb, 8);
  898.             leftg= p->data[0][last_line+G]= get_bits(&s->gb, 8);
  899.             leftb= p->data[0][last_line+B]= get_bits(&s->gb, 8);
  900.         }else{
  901.             leftr= p->data[0][last_line+R]= get_bits(&s->gb, 8);
  902.             leftg= p->data[0][last_line+G]= get_bits(&s->gb, 8);
  903.             leftb= p->data[0][last_line+B]= get_bits(&s->gb, 8);
  904.             skip_bits(&s->gb, 8);
  905.         }
  906.         
  907.         if(s->bgr32){
  908.             switch(s->predictor){
  909.             case LEFT:
  910.             case PLANE:
  911.                 decode_bgr_bitstream(s, width-1);
  912.                 add_left_prediction_bgr32(p->data[0] + last_line+4, s->temp[0], width-1, &leftr, &leftg, &leftb);
  913.                 for(y=s->height-2; y>=0; y--){ //yes its stored upside down
  914.                     decode_bgr_bitstream(s, width);
  915.                     
  916.                     add_left_prediction_bgr32(p->data[0] + p->linesize[0]*y, s->temp[0], width, &leftr, &leftg, &leftb);
  917.                     if(s->predictor == PLANE){
  918.                         if((y&s->interlaced)==0 && y<s->height-1-s->interlaced){
  919.                             s->dsp.add_bytes(p->data[0] + p->linesize[0]*y, 
  920.                                              p->data[0] + p->linesize[0]*y + fake_ystride, fake_ystride);
  921.                         }
  922.                     }
  923.                 }
  924.                 draw_slice(s, height); // just 1 large slice as this is not possible in reverse order
  925.                 break;
  926.             default:
  927.                 av_log(avctx, AV_LOG_ERROR, "prediction type not supported!n");
  928.             }
  929.         }else{
  930.             av_log(avctx, AV_LOG_ERROR, "BGR24 output is not implemented yetn");
  931.             return -1;
  932.         }
  933.     }
  934.     emms_c();
  935.     
  936.     *picture= *p;
  937.     *data_size = sizeof(AVFrame);
  938.     
  939.     return (get_bits_count(&s->gb)+31)/32*4;
  940. }
  941. static int common_end(HYuvContext *s){
  942.     int i;
  943.     
  944.     for(i=0; i<3; i++){
  945.         av_freep(&s->temp[i]);
  946.     }
  947.     return 0;
  948. }
  949. static int decode_end(AVCodecContext *avctx)
  950. {
  951.     HYuvContext *s = avctx->priv_data;
  952.     int i;
  953.     
  954.     common_end(s);
  955.     av_freep(&s->bitstream_buffer);
  956.     
  957.     for(i=0; i<3; i++){
  958.         free_vlc(&s->vlc[i]);
  959.     }
  960.     return 0;
  961. }
  962. static int encode_frame(AVCodecContext *avctx, unsigned char *buf, int buf_size, void *data){
  963.     HYuvContext *s = avctx->priv_data;
  964.     AVFrame *pict = data;
  965.     const int width= s->width;
  966.     const int width2= s->width>>1;
  967.     const int height= s->height;
  968.     const int fake_ystride= s->interlaced ? pict->linesize[0]*2  : pict->linesize[0];
  969.     const int fake_ustride= s->interlaced ? pict->linesize[1]*2  : pict->linesize[1];
  970.     const int fake_vstride= s->interlaced ? pict->linesize[2]*2  : pict->linesize[2];
  971.     AVFrame * const p= &s->picture;
  972.     int i, j, size=0;
  973.     *p = *pict;
  974.     p->pict_type= FF_I_TYPE;
  975.     p->key_frame= 1;
  976.     
  977.     if(s->context){
  978.         for(i=0; i<3; i++){
  979.             generate_len_table(s->len[i], s->stats[i], 256);
  980.             if(generate_bits_table(s->bits[i], s->len[i])<0)
  981.                 return -1;
  982.             size+= store_table(s, s->len[i], &buf[size]);
  983.         }
  984.         for(i=0; i<3; i++)
  985.             for(j=0; j<256; j++)
  986.                 s->stats[i][j] >>= 1;
  987.     }
  988.     init_put_bits(&s->pb, buf+size, buf_size-size);
  989.     if(avctx->pix_fmt == PIX_FMT_YUV422P || avctx->pix_fmt == PIX_FMT_YUV420P){
  990.         int lefty, leftu, leftv, y, cy;
  991.         put_bits(&s->pb, 8, leftv= p->data[2][0]);
  992.         put_bits(&s->pb, 8, lefty= p->data[0][1]);
  993.         put_bits(&s->pb, 8, leftu= p->data[1][0]);
  994.         put_bits(&s->pb, 8,        p->data[0][0]);
  995.         
  996.         lefty= sub_left_prediction(s, s->temp[0], p->data[0]+2, width-2 , lefty);
  997.         leftu= sub_left_prediction(s, s->temp[1], p->data[1]+1, width2-1, leftu);
  998.         leftv= sub_left_prediction(s, s->temp[2], p->data[2]+1, width2-1, leftv);
  999.         
  1000.         encode_422_bitstream(s, width-2);
  1001.         
  1002.         if(s->predictor==MEDIAN){
  1003.             int lefttopy, lefttopu, lefttopv;
  1004.             cy=y=1;
  1005.             if(s->interlaced){
  1006.                 lefty= sub_left_prediction(s, s->temp[0], p->data[0]+p->linesize[0], width , lefty);
  1007.                 leftu= sub_left_prediction(s, s->temp[1], p->data[1]+p->linesize[1], width2, leftu);
  1008.                 leftv= sub_left_prediction(s, s->temp[2], p->data[2]+p->linesize[2], width2, leftv);
  1009.         
  1010.                 encode_422_bitstream(s, width);
  1011.                 y++; cy++;
  1012.             }
  1013.             
  1014.             lefty= sub_left_prediction(s, s->temp[0], p->data[0]+fake_ystride, 4, lefty);
  1015.             leftu= sub_left_prediction(s, s->temp[1], p->data[1]+fake_ustride, 2, leftu);
  1016.             leftv= sub_left_prediction(s, s->temp[2], p->data[2]+fake_vstride, 2, leftv);
  1017.         
  1018.             encode_422_bitstream(s, 4);
  1019.             lefttopy= p->data[0][3];
  1020.             lefttopu= p->data[1][1];
  1021.             lefttopv= p->data[2][1];
  1022.             s->dsp.sub_hfyu_median_prediction(s->temp[0], p->data[0]+4, p->data[0] + fake_ystride+4, width-4 , &lefty, &lefttopy);
  1023.             s->dsp.sub_hfyu_median_prediction(s->temp[1], p->data[1]+2, p->data[1] + fake_ustride+2, width2-2, &leftu, &lefttopu);
  1024.             s->dsp.sub_hfyu_median_prediction(s->temp[2], p->data[2]+2, p->data[2] + fake_vstride+2, width2-2, &leftv, &lefttopv);
  1025.             encode_422_bitstream(s, width-4);
  1026.             y++; cy++;
  1027.             for(; y<height; y++,cy++){
  1028.                 uint8_t *ydst, *udst, *vdst;
  1029.                     
  1030.                 if(s->bitstream_bpp==12){
  1031.                     while(2*cy > y){
  1032.                         ydst= p->data[0] + p->linesize[0]*y;
  1033.                         s->dsp.sub_hfyu_median_prediction(s->temp[0], ydst - fake_ystride, ydst, width , &lefty, &lefttopy);
  1034.                         encode_gray_bitstream(s, width);
  1035.                         y++;
  1036.                     }
  1037.                     if(y>=height) break;
  1038.                 }
  1039.                 ydst= p->data[0] + p->linesize[0]*y;
  1040.                 udst= p->data[1] + p->linesize[1]*cy;
  1041.                 vdst= p->data[2] + p->linesize[2]*cy;
  1042.                 s->dsp.sub_hfyu_median_prediction(s->temp[0], ydst - fake_ystride, ydst, width , &lefty, &lefttopy);
  1043.                 s->dsp.sub_hfyu_median_prediction(s->temp[1], udst - fake_ustride, udst, width2, &leftu, &lefttopu);
  1044.                 s->dsp.sub_hfyu_median_prediction(s->temp[2], vdst - fake_vstride, vdst, width2, &leftv, &lefttopv);
  1045.                 encode_422_bitstream(s, width);
  1046.             }
  1047.         }else{
  1048.             for(cy=y=1; y<height; y++,cy++){
  1049.                 uint8_t *ydst, *udst, *vdst;
  1050.                 
  1051.                 /* encode a luma only line & y++ */
  1052.                 if(s->bitstream_bpp==12){
  1053.                     ydst= p->data[0] + p->linesize[0]*y;
  1054.                     if(s->predictor == PLANE && s->interlaced < y){
  1055.                         s->dsp.diff_bytes(s->temp[1], ydst, ydst - fake_ystride, width);
  1056.                         lefty= sub_left_prediction(s, s->temp[0], s->temp[1], width , lefty);
  1057.                     }else{
  1058.                         lefty= sub_left_prediction(s, s->temp[0], ydst, width , lefty);
  1059.                     }
  1060.                     encode_gray_bitstream(s, width);
  1061.                     y++;
  1062.                     if(y>=height) break;
  1063.                 }
  1064.                 
  1065.                 ydst= p->data[0] + p->linesize[0]*y;
  1066.                 udst= p->data[1] + p->linesize[1]*cy;
  1067.                 vdst= p->data[2] + p->linesize[2]*cy;
  1068.                 if(s->predictor == PLANE && s->interlaced < cy){
  1069.                     s->dsp.diff_bytes(s->temp[1], ydst, ydst - fake_ystride, width);
  1070.                     s->dsp.diff_bytes(s->temp[2], udst, udst - fake_ustride, width2);
  1071.                     s->dsp.diff_bytes(s->temp[2] + width2, vdst, vdst - fake_vstride, width2);
  1072.                     lefty= sub_left_prediction(s, s->temp[0], s->temp[1], width , lefty);
  1073.                     leftu= sub_left_prediction(s, s->temp[1], s->temp[2], width2, leftu);
  1074.                     leftv= sub_left_prediction(s, s->temp[2], s->temp[2] + width2, width2, leftv);
  1075.                 }else{
  1076.                     lefty= sub_left_prediction(s, s->temp[0], ydst, width , lefty);
  1077.                     leftu= sub_left_prediction(s, s->temp[1], udst, width2, leftu);
  1078.                     leftv= sub_left_prediction(s, s->temp[2], vdst, width2, leftv);
  1079.                 }
  1080.                 encode_422_bitstream(s, width);
  1081.             }
  1082.         }        
  1083.     }else{
  1084.         av_log(avctx, AV_LOG_ERROR, "Format not supported!n");
  1085.     }
  1086.     emms_c();
  1087.     
  1088.     size+= (put_bits_count(&s->pb)+31)/8;
  1089.     size/= 4;
  1090.     
  1091.     if((s->flags&CODEC_FLAG_PASS1) && (s->picture_number&31)==0){
  1092.         int j;
  1093.         char *p= avctx->stats_out;
  1094.         char *end= p + 1024*30;
  1095.         for(i=0; i<3; i++){
  1096.             for(j=0; j<256; j++){
  1097.                 snprintf(p, end-p, "%llu ", s->stats[i][j]);
  1098.                 p+= strlen(p);
  1099.                 s->stats[i][j]= 0;
  1100.             }
  1101.             snprintf(p, end-p, "n");
  1102.             p++;
  1103.         }
  1104.     }
  1105.     if(!(s->avctx->flags2 & CODEC_FLAG2_NO_OUTPUT)){
  1106.         flush_put_bits(&s->pb);
  1107.         s->dsp.bswap_buf((uint32_t*)buf, (uint32_t*)buf, size);
  1108.         avctx->stats_out[0] = '';
  1109.     }
  1110.     
  1111.     s->picture_number++;
  1112.     return size*4;
  1113. }
  1114. static int encode_end(AVCodecContext *avctx)
  1115. {
  1116.     HYuvContext *s = avctx->priv_data;
  1117.     
  1118.     common_end(s);
  1119.     av_freep(&avctx->extradata);
  1120.     av_freep(&avctx->stats_out);
  1121.     
  1122.     return 0;
  1123. }
  1124. AVCodec huffyuv_decoder = {
  1125.     "huffyuv",
  1126.     CODEC_TYPE_VIDEO,
  1127.     CODEC_ID_HUFFYUV,
  1128.     sizeof(HYuvContext),
  1129.     decode_init,
  1130.     NULL,
  1131.     decode_end,
  1132.     decode_frame,
  1133.     CODEC_CAP_DR1 | CODEC_CAP_DRAW_HORIZ_BAND,
  1134.     NULL
  1135. };
  1136. AVCodec ffvhuff_decoder = {
  1137.     "ffvhuff",
  1138.     CODEC_TYPE_VIDEO,
  1139.     CODEC_ID_FFVHUFF,
  1140.     sizeof(HYuvContext),
  1141.     decode_init,
  1142.     NULL,
  1143.     decode_end,
  1144.     decode_frame,
  1145.     CODEC_CAP_DR1 | CODEC_CAP_DRAW_HORIZ_BAND,
  1146.     NULL
  1147. };
  1148. #ifdef CONFIG_ENCODERS
  1149. AVCodec huffyuv_encoder = {
  1150.     "huffyuv",
  1151.     CODEC_TYPE_VIDEO,
  1152.     CODEC_ID_HUFFYUV,
  1153.     sizeof(HYuvContext),
  1154.     encode_init,
  1155.     encode_frame,
  1156.     encode_end,
  1157. };
  1158. AVCodec ffvhuff_encoder = {
  1159.     "ffvhuff",
  1160.     CODEC_TYPE_VIDEO,
  1161.     CODEC_ID_FFVHUFF,
  1162.     sizeof(HYuvContext),
  1163.     encode_init,
  1164.     encode_frame,
  1165.     encode_end,
  1166. };
  1167. #endif //CONFIG_ENCODERS