开通博客赚积分
发布资源赚积分
分类

源码开发语言/平台
当前位置：首页 > 源码/资料 > 多媒体 > 流媒体/Mpeg4/MP4 > 查看源码
mpegvideo.c
资源名称：ffh264libxxxx.rar　[点击查看]
上传用户：shlianrong
上传日期：2022-07-08
资源大小：309k
文件大小：89k
源码类别：

流媒体/Mpeg4/MP4

开发平台：

Visual C++

mpegvideo.c：源码内容
  1. /*
  2.  * The simplest mpeg encoder (well, it was the simplest!)
  3.  * Copyright (c) 2000,2001 Fabrice Bellard.
  4.  * Copyright (c) 2002-2004 Michael Niedermayer <michaelni@gmx.at>
  5.  *
  6.  * 4MV & hq & B-frame encoding stuff by Michael Niedermayer <michaelni@gmx.at>
  7.  *
  8.  * This file is part of FFmpeg.
  9.  *
  10.  * FFmpeg is free software; you can redistribute it and/or
  11.  * modify it under the terms of the GNU Lesser General Public
  12.  * License as published by the Free Software Foundation; either
  13.  * version 2.1 of the License, or (at your option) any later version.
  14.  *
  15.  * FFmpeg is distributed in the hope that it will be useful,
  16.  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  17.  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
  18.  * Lesser General Public License for more details.
  19.  *
  20.  * You should have received a copy of the GNU Lesser General Public
  21.  * License along with FFmpeg; if not, write to the Free Software
  22.  * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
  23.  */
  25. /**
  26.  * @file mpegvideo.c
  27.  * The simplest mpeg encoder (well, it was the simplest!).
  28.  */
  30. #include "dsputil.h"
  31. #include "internal.h"
  32. //#include "integer.h"
  33. #include "mpegvideo.h"
  34. #include "mpegvideo_common.h"
  35. //#include "mjpegenc.h"
  36. //#include "msmpeg4.h"
  37. //#include "faandct.h"
  38. #include <limits.h>
  40. //#undef NDEBUG
  41. //#include <assert.h>
  43. static void dct_unquantize_mpeg1_intra_c(MpegEncContext *s,
  44.                                    DCTELEM *block, int n, int qscale);
  45. static void dct_unquantize_mpeg1_inter_c(MpegEncContext *s,
  46.                                    DCTELEM *block, int n, int qscale);
  47. static void dct_unquantize_mpeg2_intra_c(MpegEncContext *s,
  48.                                    DCTELEM *block, int n, int qscale);
  49. static void dct_unquantize_mpeg2_intra_bitexact(MpegEncContext *s,
  50.                                    DCTELEM *block, int n, int qscale);
  51. static void dct_unquantize_mpeg2_inter_c(MpegEncContext *s,
  52.                                    DCTELEM *block, int n, int qscale);
  53. static void dct_unquantize_h263_intra_c(MpegEncContext *s,
  54.                                   DCTELEM *block, int n, int qscale);
  55. static void dct_unquantize_h263_inter_c(MpegEncContext *s,
  56.                                   DCTELEM *block, int n, int qscale);
  58. int  XVMC_field_start(MpegEncContext*s, AVCodecContext *avctx);
  59. void XVMC_field_end(MpegEncContext *s);
  60. void XVMC_decode_mb(MpegEncContext *s);
  63. /* enable all paranoid tests for rounding, overflows, etc... */
  64. //#define PARANOID
  66. //#define DEBUG
  69. static const uint8_t ff_default_chroma_qscale_table[32]={
  70. //  0  1  2  3  4  5  6  7  8  9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31
  71.     0, 1, 2, 3, 4, 5, 6, 7, 8, 9,10,11,12,13,14,15,16,17,18,19,20,21,22,23,24,25,26,27,28,29,30,31
  72. };
  74. const uint8_t ff_mpeg1_dc_scale_table[128]={
  75. //  0  1  2  3  4  5  6  7  8  9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31
  76.     8, 8, 8, 8, 8, 8, 8, 8, 8, 8, 8, 8, 8, 8, 8, 8, 8, 8, 8, 8, 8, 8, 8, 8, 8, 8, 8, 8, 8, 8, 8, 8,
  77.     8, 8, 8, 8, 8, 8, 8, 8, 8, 8, 8, 8, 8, 8, 8, 8, 8, 8, 8, 8, 8, 8, 8, 8, 8, 8, 8, 8, 8, 8, 8, 8,
  78.     8, 8, 8, 8, 8, 8, 8, 8, 8, 8, 8, 8, 8, 8, 8, 8, 8, 8, 8, 8, 8, 8, 8, 8, 8, 8, 8, 8, 8, 8, 8, 8,
  79.     8, 8, 8, 8, 8, 8, 8, 8, 8, 8, 8, 8, 8, 8, 8, 8, 8, 8, 8, 8, 8, 8, 8, 8, 8, 8, 8, 8, 8, 8, 8, 8,
  80. };
  83. const uint8_t *ff_find_start_code(const uint8_t * restrict p, const uint8_t *end, uint32_t * restrict state){
  84.     int i;
  86.     assert(p<=end);
  87.     if(p>=end)
  88.         return end;
  90.     for(i=0; i<3; i++){
  91.         uint32_t tmp= *state << 8;
  92.         *state= tmp + *(p++);
  93.         if(tmp == 0x100 || p==end)
  94.             return p;
  95.     }
  97.     while(p<end){
  98.         if     (p[-1] > 1      ) p+= 3;
  99.         else if(p[-2]          ) p+= 2;
  100.         else if(p[-3]|(p[-1]-1)) p++;
  101.         else{
  102.             p++;
  103.             break;
  104.         }
  105.     }
  107.     p= FFMIN(p, end)-4;
  108.     *state= AV_RB32(p);
  110.     return p+4;
  111. }
  113. /* init common dct for both encoder and decoder */
  114. int ff_dct_common_init(MpegEncContext *s)
  115. {
  116.     s->dct_unquantize_h263_intra = dct_unquantize_h263_intra_c;
  117.     s->dct_unquantize_h263_inter = dct_unquantize_h263_inter_c;
  118.     s->dct_unquantize_mpeg1_intra = dct_unquantize_mpeg1_intra_c;
  119.     s->dct_unquantize_mpeg1_inter = dct_unquantize_mpeg1_inter_c;
  120.     s->dct_unquantize_mpeg2_intra = dct_unquantize_mpeg2_intra_c;
  121.     if(s->flags & CODEC_FLAG_BITEXACT)
  122.         s->dct_unquantize_mpeg2_intra = dct_unquantize_mpeg2_intra_bitexact;
  123.     s->dct_unquantize_mpeg2_inter = dct_unquantize_mpeg2_inter_c;
  125. #if defined(HAVE_MMX)
  126.     MPV_common_init_mmx(s);
  127. #elif defined(ARCH_ALPHA)
  128.     MPV_common_init_axp(s);
  129. #elif defined(CONFIG_MLIB)
  130.     MPV_common_init_mlib(s);
  131. #elif defined(HAVE_MMI)
  132.     MPV_common_init_mmi(s);
  133. #elif defined(ARCH_ARM)
  134.     MPV_common_init_arm(s);
  135. #elif defined(HAVE_ALTIVEC)
  136.     MPV_common_init_altivec(s);
  137. #elif defined(ARCH_BFIN)
  138.     MPV_common_init_bfin(s);
  139. #endif
  141.     /* load & permutate scantables
  142.        note: only wmv uses different ones
  143.     */
  144.     if(s->alternate_scan){
  145.         ff_init_scantable(s->dsp.idct_permutation, &s->inter_scantable  , ff_alternate_vertical_scan);
  146.         ff_init_scantable(s->dsp.idct_permutation, &s->intra_scantable  , ff_alternate_vertical_scan);
  147.     }else{
  148.         ff_init_scantable(s->dsp.idct_permutation, &s->inter_scantable  , ff_zigzag_direct);
  149.         ff_init_scantable(s->dsp.idct_permutation, &s->intra_scantable  , ff_zigzag_direct);
  150.     }
  151.     ff_init_scantable(s->dsp.idct_permutation, &s->intra_h_scantable, ff_alternate_horizontal_scan);
  152.     ff_init_scantable(s->dsp.idct_permutation, &s->intra_v_scantable, ff_alternate_vertical_scan);
  154.     return 0;
  155. }
  157. void ff_copy_picture(Picture *dst, Picture *src){
  158.     *dst = *src;
  159.     dst->type= FF_BUFFER_TYPE_COPY;
  160. }
  162. /**
  163.  * allocates a Picture
  164.  * The pixels are allocated/set by calling get_buffer() if shared=0
  165.  */
  166. int alloc_picture(MpegEncContext *s, Picture *pic, int shared){
  167.     const int big_mb_num= s->mb_stride*(s->mb_height+1) + 1; //the +1 is needed so memset(,,stride*height) does not sig11
  168.     const int mb_array_size= s->mb_stride*s->mb_height;
  169.     const int b8_array_size= s->b8_stride*s->mb_height*2;
  170.     const int b4_array_size= s->b4_stride*s->mb_height*4;
  171.     int i;
  172.     int r= -1;
  174.     if(shared){
  175.         assert(pic->data[0]);
  176.         assert(pic->type == 0 || pic->type == FF_BUFFER_TYPE_SHARED);
  177.         pic->type= FF_BUFFER_TYPE_SHARED;
  178.     }else{
  179.         assert(!pic->data[0]);
  181.         r= s->avctx->get_buffer(s->avctx, (AVFrame*)pic);
  183.         if(r<0 || !pic->age || !pic->type || !pic->data[0]){
  184.             av_log(s->avctx, AV_LOG_ERROR, "get_buffer() failed (%d %d %d %p)n", r, pic->age, pic->type, pic->data[0]);
  185.             return -1;
  186.         }
  188.         if(s->linesize && (s->linesize != pic->linesize[0] || s->uvlinesize != pic->linesize[1])){
  189.             av_log(s->avctx, AV_LOG_ERROR, "get_buffer() failed (stride changed)n");
  190.             s->avctx->release_buffer(s->avctx, (AVFrame*)pic);
  191.             return -1;
  192.         }
  194.         if(pic->linesize[1] != pic->linesize[2]){
  195.             av_log(s->avctx, AV_LOG_ERROR, "get_buffer() failed (uv stride mismatch)n");
  196.             s->avctx->release_buffer(s->avctx, (AVFrame*)pic);
  197.             return -1;
  198.         }
  200.         s->linesize  = pic->linesize[0];
  201.         s->uvlinesize= pic->linesize[1];
  202.     }
  204.     if(pic->qscale_table==NULL){
  205.         if (s->encoding) {
  206.             CHECKED_ALLOCZ(pic->mb_var   , mb_array_size * sizeof(int16_t))
  207.             CHECKED_ALLOCZ(pic->mc_mb_var, mb_array_size * sizeof(int16_t))
  208.             CHECKED_ALLOCZ(pic->mb_mean  , mb_array_size * sizeof(int8_t))
  209.         }
  211.         CHECKED_ALLOCZ(pic->mbskip_table , mb_array_size * sizeof(uint8_t)+2) //the +2 is for the slice end check
  212.         CHECKED_ALLOCZ(pic->qscale_table , mb_array_size * sizeof(uint8_t))
  213.         CHECKED_ALLOCZ(pic->mb_type_base , (big_mb_num + s->mb_stride) * sizeof(uint32_t))
  214.         pic->mb_type= pic->mb_type_base + 2*s->mb_stride+1;
  215.         if(s->out_format == FMT_H264){
  216.             for(i=0; i<2; i++){
  217.                 CHECKED_ALLOCZ(pic->motion_val_base[i], 2 * (b4_array_size+4)  * sizeof(int16_t))
  218.                 pic->motion_val[i]= pic->motion_val_base[i]+4;
  219.                 CHECKED_ALLOCZ(pic->ref_index[i], b8_array_size * sizeof(uint8_t))
  220.             }
  221.             pic->motion_subsample_log2= 2;
  222.         }else if(s->out_format == FMT_H263 || s->encoding || (s->avctx->debug&FF_DEBUG_MV) || (s->avctx->debug_mv)){
  223.             for(i=0; i<2; i++){
  224.                 CHECKED_ALLOCZ(pic->motion_val_base[i], 2 * (b8_array_size+4) * sizeof(int16_t))
  225.                 pic->motion_val[i]= pic->motion_val_base[i]+4;
  226.                 CHECKED_ALLOCZ(pic->ref_index[i], b8_array_size * sizeof(uint8_t))
  227.             }
  228.             pic->motion_subsample_log2= 3;
  229.         }
  230.         if(s->avctx->debug&FF_DEBUG_DCT_COEFF) {
  231.             CHECKED_ALLOCZ(pic->dct_coeff, 64 * mb_array_size * sizeof(DCTELEM)*6)
  232.         }
  233.         pic->qstride= s->mb_stride;
  234.         CHECKED_ALLOCZ(pic->pan_scan , 1 * sizeof(AVPanScan))
  235.     }
  237.     /* It might be nicer if the application would keep track of these
  238.      * but it would require an API change. */
  239.     memmove(s->prev_pict_types+1, s->prev_pict_types, PREV_PICT_TYPES_BUFFER_SIZE-1);
  240.     s->prev_pict_types[0]= s->dropable ? FF_B_TYPE : s->pict_type;
  241.     if(pic->age < PREV_PICT_TYPES_BUFFER_SIZE && s->prev_pict_types[pic->age] == FF_B_TYPE)
  242.         pic->age= INT_MAX; // Skipped MBs in B-frames are quite rare in MPEG-1/2 and it is a bit tricky to skip them anyway.
  244.     return 0;
  245. fail: //for the CHECKED_ALLOCZ macro
  246.     if(r>=0)
  247.         s->avctx->release_buffer(s->avctx, (AVFrame*)pic);
  248.     return -1;
  249. }
  251. /**
  252.  * deallocates a picture
  253.  */
  254. static void free_picture(MpegEncContext *s, Picture *pic){
  255.     int i;
  257.     if(pic->data[0] && pic->type!=FF_BUFFER_TYPE_SHARED){
  258.         s->avctx->release_buffer(s->avctx, (AVFrame*)pic);
  259.     }
  261.     av_freep(&pic->mb_var);
  262.     av_freep(&pic->mc_mb_var);
  263.     av_freep(&pic->mb_mean);
  264.     av_freep(&pic->mbskip_table);
  265.     av_freep(&pic->qscale_table);
  266.     av_freep(&pic->mb_type_base);
  267.     av_freep(&pic->dct_coeff);
  268.     av_freep(&pic->pan_scan);
  269.     pic->mb_type= NULL;
  270.     for(i=0; i<2; i++){
  271.         av_freep(&pic->motion_val_base[i]);
  272.         av_freep(&pic->ref_index[i]);
  273.     }
  275.     if(pic->type == FF_BUFFER_TYPE_SHARED){
  276.         for(i=0; i<4; i++){
  277.             pic->base[i]=
  278.             pic->data[i]= NULL;
  279.         }
  280.         pic->type= 0;
  281.     }
  282. }
  284. static int init_duplicate_context(MpegEncContext *s, MpegEncContext *base){
  285.     int i;
  287.     // edge emu needs blocksize + filter length - 1 (=17x17 for halfpel / 21x21 for h264)
  288.     CHECKED_ALLOCZ(s->allocated_edge_emu_buffer, (s->width+64)*2*21*2); //(width + edge + align)*interlaced*MBsize*tolerance
  289.     s->edge_emu_buffer= s->allocated_edge_emu_buffer + (s->width+64)*2*21;
  291.      //FIXME should be linesize instead of s->width*2 but that is not known before get_buffer()
  292.     CHECKED_ALLOCZ(s->me.scratchpad,  (s->width+64)*4*16*2*sizeof(uint8_t))
  293.     s->me.temp=         s->me.scratchpad;
  294.     s->rd_scratchpad=   s->me.scratchpad;
  295.     s->b_scratchpad=    s->me.scratchpad;
  296.     s->obmc_scratchpad= s->me.scratchpad + 16;
  297.     if (s->encoding) {
  298.         CHECKED_ALLOCZ(s->me.map      , ME_MAP_SIZE*sizeof(uint32_t))
  299.         CHECKED_ALLOCZ(s->me.score_map, ME_MAP_SIZE*sizeof(uint32_t))
  300.         if(s->avctx->noise_reduction){
  301.             CHECKED_ALLOCZ(s->dct_error_sum, 2 * 64 * sizeof(int))
  302.         }
  303.     }
  304.     CHECKED_ALLOCZ(s->blocks, 64*12*2 * sizeof(DCTELEM))
  305.     s->block= s->blocks[0];
  307.     for(i=0;i<12;i++){
  308.         s->pblocks[i] = (short *)(&s->block[i]);
  309.     }
  310.     return 0;
  311. fail:
  312.     return -1; //free() through MPV_common_end()
  313. }
  315. static void free_duplicate_context(MpegEncContext *s){
  316.     if(s==NULL) return;
  318.     av_freep(&s->allocated_edge_emu_buffer); s->edge_emu_buffer= NULL;
  319.     av_freep(&s->me.scratchpad);
  320.     s->me.temp=
  321.     s->rd_scratchpad=
  322.     s->b_scratchpad=
  323.     s->obmc_scratchpad= NULL;
  325.     av_freep(&s->dct_error_sum);
  326.     av_freep(&s->me.map);
  327.     av_freep(&s->me.score_map);
  328.     av_freep(&s->blocks);
  329.     s->block= NULL;
  330. }
  332. static void backup_duplicate_context(MpegEncContext *bak, MpegEncContext *src){
  333. #define COPY(a) bak->a= src->a
  334.     COPY(allocated_edge_emu_buffer);
  335.     COPY(edge_emu_buffer);
  336.     COPY(me.scratchpad);
  337.     COPY(me.temp);
  338.     COPY(rd_scratchpad);
  339.     COPY(b_scratchpad);
  340.     COPY(obmc_scratchpad);
  341.     COPY(me.map);
  342.     COPY(me.score_map);
  343.     COPY(blocks);
  344.     COPY(block);
  345.     COPY(start_mb_y);
  346.     COPY(end_mb_y);
  347.     COPY(me.map_generation);
  348.     COPY(pb);
  349.     COPY(dct_error_sum);
  350.     COPY(dct_count[0]);
  351.     COPY(dct_count[1]);
  352. #undef COPY
  353. }
  355. void ff_update_duplicate_context(MpegEncContext *dst, MpegEncContext *src){
  356.     MpegEncContext bak;
  357.     int i;
  358.     //FIXME copy only needed parts
  359. //START_TIMER
  360.     backup_duplicate_context(&bak, dst);
  361.     memcpy(dst, src, sizeof(MpegEncContext));
  362.     backup_duplicate_context(dst, &bak);
  363.     for(i=0;i<12;i++){
  364.         dst->pblocks[i] = (short *)(&dst->block[i]);
  365.     }
  366. //STOP_TIMER("update_duplicate_context") //about 10k cycles / 0.01 sec for 1000frames on 1ghz with 2 threads
  367. }
  369. /**
  370.  * sets the given MpegEncContext to common defaults (same for encoding and decoding).
  371.  * the changed fields will not depend upon the prior state of the MpegEncContext.
  372.  */
  373. void MPV_common_defaults(MpegEncContext *s){
  374.     s->y_dc_scale_table=
  375.     s->c_dc_scale_table= ff_mpeg1_dc_scale_table;
  376.     s->chroma_qscale_table= ff_default_chroma_qscale_table;
  377.     s->progressive_frame= 1;
  378.     s->progressive_sequence= 1;
  379.     s->picture_structure= PICT_FRAME;
  381.     s->coded_picture_number = 0;
  382.     s->picture_number = 0;
  383.     s->input_picture_number = 0;
  385.     s->picture_in_gop_number = 0;
  387.     s->f_code = 1;
  388.     s->b_code = 1;
  389. }
  391. /**
  392.  * sets the given MpegEncContext to defaults for decoding.
  393.  * the changed fields will not depend upon the prior state of the MpegEncContext.
  394.  */
  395. void MPV_decode_defaults(MpegEncContext *s){
  396.     MPV_common_defaults(s);
  397. }
  399. /**
  400.  * init common structure for both encoder and decoder.
  401.  * this assumes that some variables like width/height are already set
  402.  */
  403. int MPV_common_init(MpegEncContext *s)
  404. {
  405.     int y_size, c_size, yc_size, i, mb_array_size, mv_table_size, x, y, threads;
  407.     s->mb_height = (s->height + 15) / 16;
  409.     if(s->avctx->thread_count > MAX_THREADS || (s->avctx->thread_count > s->mb_height && s->mb_height)){
  410.         av_log(s->avctx, AV_LOG_ERROR, "too many threadsn");
  411.         return -1;
  412.     }
  414.     if((s->width || s->height) && avcodec_check_dimensions(s->avctx, s->width, s->height))
  415.         return -1;
  417.     dsputil_init(&s->dsp, s->avctx);
  418.     ff_dct_common_init(s);
  420.     s->flags= s->avctx->flags;
  421.     s->flags2= s->avctx->flags2;
  423.     s->mb_width  = (s->width  + 15) / 16;
  424.     s->mb_stride = s->mb_width + 1;
  425.     s->b8_stride = s->mb_width*2 + 1;
  426.     s->b4_stride = s->mb_width*4 + 1;
  427.     mb_array_size= s->mb_height * s->mb_stride;
  428.     mv_table_size= (s->mb_height+2) * s->mb_stride + 1;
  430.     /* set chroma shifts */
  431.     avcodec_get_chroma_sub_sample(s->avctx->pix_fmt,&(s->chroma_x_shift),
  432.                                                     &(s->chroma_y_shift) );
  434.     /* set default edge pos, will be overriden in decode_header if needed */
  435.     s->h_edge_pos= s->mb_width*16;
  436.     s->v_edge_pos= s->mb_height*16;
  438.     s->mb_num = s->mb_width * s->mb_height;
  440.     s->block_wrap[0]=
  441.     s->block_wrap[1]=
  442.     s->block_wrap[2]=
  443.     s->block_wrap[3]= s->b8_stride;
  444.     s->block_wrap[4]=
  445.     s->block_wrap[5]= s->mb_stride;
  447.     y_size = s->b8_stride * (2 * s->mb_height + 1);
  448.     c_size = s->mb_stride * (s->mb_height + 1);
  449.     yc_size = y_size + 2 * c_size;
  451.     /* convert fourcc to upper case */
  452.     s->codec_tag=          toupper( s->avctx->codec_tag     &0xFF)
  453.                         + (toupper((s->avctx->codec_tag>>8 )&0xFF)<<8 )
  454.                         + (toupper((s->avctx->codec_tag>>16)&0xFF)<<16)
  455.                         + (toupper((s->avctx->codec_tag>>24)&0xFF)<<24);
  457.     s->stream_codec_tag=          toupper( s->avctx->stream_codec_tag     &0xFF)
  458.                                + (toupper((s->avctx->stream_codec_tag>>8 )&0xFF)<<8 )
  459.                                + (toupper((s->avctx->stream_codec_tag>>16)&0xFF)<<16)
  460.                                + (toupper((s->avctx->stream_codec_tag>>24)&0xFF)<<24);
  462.     s->avctx->coded_frame= (AVFrame*)&s->current_picture;
  464.     CHECKED_ALLOCZ(s->mb_index2xy, (s->mb_num+1)*sizeof(int)) //error ressilience code looks cleaner with this
  465.     for(y=0; y<s->mb_height; y++){
  466.         for(x=0; x<s->mb_width; x++){
  467.             s->mb_index2xy[ x + y*s->mb_width ] = x + y*s->mb_stride;
  468.         }
  469.     }
  470.     s->mb_index2xy[ s->mb_height*s->mb_width ] = (s->mb_height-1)*s->mb_stride + s->mb_width; //FIXME really needed?
  472.     if (s->encoding) {
  473.         /* Allocate MV tables */
  474.         CHECKED_ALLOCZ(s->p_mv_table_base            , mv_table_size * 2 * sizeof(int16_t))
  475.         CHECKED_ALLOCZ(s->b_forw_mv_table_base       , mv_table_size * 2 * sizeof(int16_t))
  476.         CHECKED_ALLOCZ(s->b_back_mv_table_base       , mv_table_size * 2 * sizeof(int16_t))
  477.         CHECKED_ALLOCZ(s->b_bidir_forw_mv_table_base , mv_table_size * 2 * sizeof(int16_t))
  478.         CHECKED_ALLOCZ(s->b_bidir_back_mv_table_base , mv_table_size * 2 * sizeof(int16_t))
  479.         CHECKED_ALLOCZ(s->b_direct_mv_table_base     , mv_table_size * 2 * sizeof(int16_t))
  480.         s->p_mv_table           = s->p_mv_table_base            + s->mb_stride + 1;
  481.         s->b_forw_mv_table      = s->b_forw_mv_table_base       + s->mb_stride + 1;
  482.         s->b_back_mv_table      = s->b_back_mv_table_base       + s->mb_stride + 1;
  483.         s->b_bidir_forw_mv_table= s->b_bidir_forw_mv_table_base + s->mb_stride + 1;
  484.         s->b_bidir_back_mv_table= s->b_bidir_back_mv_table_base + s->mb_stride + 1;
  485.         s->b_direct_mv_table    = s->b_direct_mv_table_base     + s->mb_stride + 1;
  487.         if(s->msmpeg4_version){
  488.             CHECKED_ALLOCZ(s->ac_stats, 2*2*(MAX_LEVEL+1)*(MAX_RUN+1)*2*sizeof(int));
  489.         }
  490.         CHECKED_ALLOCZ(s->avctx->stats_out, 256);
  492.         /* Allocate MB type table */
  493.         CHECKED_ALLOCZ(s->mb_type  , mb_array_size * sizeof(uint16_t)) //needed for encoding
  495.         CHECKED_ALLOCZ(s->lambda_table, mb_array_size * sizeof(int))
  497.         CHECKED_ALLOCZ(s->q_intra_matrix, 64*32 * sizeof(int))
  498.         CHECKED_ALLOCZ(s->q_inter_matrix, 64*32 * sizeof(int))
  499.         CHECKED_ALLOCZ(s->q_intra_matrix16, 64*32*2 * sizeof(uint16_t))
  500.         CHECKED_ALLOCZ(s->q_inter_matrix16, 64*32*2 * sizeof(uint16_t))
  501.         CHECKED_ALLOCZ(s->input_picture, MAX_PICTURE_COUNT * sizeof(Picture*))
  502.         CHECKED_ALLOCZ(s->reordered_input_picture, MAX_PICTURE_COUNT * sizeof(Picture*))
  504.         if(s->avctx->noise_reduction){
  505.             CHECKED_ALLOCZ(s->dct_offset, 2 * 64 * sizeof(uint16_t))
  506.         }
  507.     }
  508.     CHECKED_ALLOCZ(s->picture, MAX_PICTURE_COUNT * sizeof(Picture))
  510.     CHECKED_ALLOCZ(s->error_status_table, mb_array_size*sizeof(uint8_t))
  512.     if(s->codec_id==CODEC_ID_MPEG4 || (s->flags & CODEC_FLAG_INTERLACED_ME)){
  513.         /* interlaced direct mode decoding tables */
  514.             for(i=0; i<2; i++){
  515.                 int j, k;
  516.                 for(j=0; j<2; j++){
  517.                     for(k=0; k<2; k++){
  518.                         CHECKED_ALLOCZ(s->b_field_mv_table_base[i][j][k]     , mv_table_size * 2 * sizeof(int16_t))
  519.                         s->b_field_mv_table[i][j][k]    = s->b_field_mv_table_base[i][j][k]     + s->mb_stride + 1;
  520.                     }
  521.                     CHECKED_ALLOCZ(s->b_field_select_table[i][j]     , mb_array_size * 2 * sizeof(uint8_t))
  522.                     CHECKED_ALLOCZ(s->p_field_mv_table_base[i][j]     , mv_table_size * 2 * sizeof(int16_t))
  523.                     s->p_field_mv_table[i][j]    = s->p_field_mv_table_base[i][j]     + s->mb_stride + 1;
  524.                 }
  525.                 CHECKED_ALLOCZ(s->p_field_select_table[i]      , mb_array_size * 2 * sizeof(uint8_t))
  526.             }
  527.     }
  528.     if (s->out_format == FMT_H263) {
  529.         /* ac values */
  530.         CHECKED_ALLOCZ(s->ac_val_base, yc_size * sizeof(int16_t) * 16);
  531.         s->ac_val[0] = s->ac_val_base + s->b8_stride + 1;
  532.         s->ac_val[1] = s->ac_val_base + y_size + s->mb_stride + 1;
  533.         s->ac_val[2] = s->ac_val[1] + c_size;
  535.         /* cbp values */
  536.         CHECKED_ALLOCZ(s->coded_block_base, y_size);
  537.         s->coded_block= s->coded_block_base + s->b8_stride + 1;
  539.         /* cbp, ac_pred, pred_dir */
  540.         CHECKED_ALLOCZ(s->cbp_table  , mb_array_size * sizeof(uint8_t))
  541.         CHECKED_ALLOCZ(s->pred_dir_table, mb_array_size * sizeof(uint8_t))
  542.     }
  544.     if (s->h263_pred || s->h263_plus || !s->encoding) {
  545.         /* dc values */
  546.         //MN: we need these for error resilience of intra-frames
  547.         CHECKED_ALLOCZ(s->dc_val_base, yc_size * sizeof(int16_t));
  548.         s->dc_val[0] = s->dc_val_base + s->b8_stride + 1;
  549.         s->dc_val[1] = s->dc_val_base + y_size + s->mb_stride + 1;
  550.         s->dc_val[2] = s->dc_val[1] + c_size;
  551.         for(i=0;i<yc_size;i++)
  552.             s->dc_val_base[i] = 1024;
  553.     }
  555.     /* which mb is a intra block */
  556.     CHECKED_ALLOCZ(s->mbintra_table, mb_array_size);
  557.     memset(s->mbintra_table, 1, mb_array_size);
  559.     /* init macroblock skip table */
  560.     CHECKED_ALLOCZ(s->mbskip_table, mb_array_size+2);
  561.     //Note the +1 is for a quicker mpeg4 slice_end detection
  562.     CHECKED_ALLOCZ(s->prev_pict_types, PREV_PICT_TYPES_BUFFER_SIZE);
  564.     s->parse_context.state= -1;
  565.     if((s->avctx->debug&(FF_DEBUG_VIS_QP|FF_DEBUG_VIS_MB_TYPE)) || (s->avctx->debug_mv)){
  566.        s->visualization_buffer[0] = av_malloc((s->mb_width*16 + 2*EDGE_WIDTH) * s->mb_height*16 + 2*EDGE_WIDTH);
  567.        s->visualization_buffer[1] = av_malloc((s->mb_width*16 + 2*EDGE_WIDTH) * s->mb_height*16 + 2*EDGE_WIDTH);
  568.        s->visualization_buffer[2] = av_malloc((s->mb_width*16 + 2*EDGE_WIDTH) * s->mb_height*16 + 2*EDGE_WIDTH);
  569.     }
  571.     s->context_initialized = 1;
  573.     s->thread_context[0]= s;
  574.     threads = s->avctx->thread_count;
  576.     for(i=1; i<threads; i++){
  577.         s->thread_context[i]= av_malloc(sizeof(MpegEncContext));
  578.         memcpy(s->thread_context[i], s, sizeof(MpegEncContext));
  579.     }
  581.     for(i=0; i<threads; i++){
  582.         if(init_duplicate_context(s->thread_context[i], s) < 0)
  583.            goto fail;
  584.         s->thread_context[i]->start_mb_y= (s->mb_height*(i  ) + s->avctx->thread_count/2) / s->avctx->thread_count;
  585.         s->thread_context[i]->end_mb_y  = (s->mb_height*(i+1) + s->avctx->thread_count/2) / s->avctx->thread_count;
  586.     }
  588.     return 0;
  589.  fail:
  590.     MPV_common_end(s);
  591.     return -1;
  592. }
  594. /* init common structure for both encoder and decoder */
  595. void MPV_common_end(MpegEncContext *s)
  596. {
  597.     int i, j, k;
  599.     for(i=0; i<s->avctx->thread_count; i++){
  600.         free_duplicate_context(s->thread_context[i]);
  601.     }
  602.     for(i=1; i<s->avctx->thread_count; i++){
  603.         av_freep(&s->thread_context[i]);
  604.     }
  606.     av_freep(&s->parse_context.buffer);
  607.     s->parse_context.buffer_size=0;
  609.     av_freep(&s->mb_type);
  610.     av_freep(&s->p_mv_table_base);
  611.     av_freep(&s->b_forw_mv_table_base);
  612.     av_freep(&s->b_back_mv_table_base);
  613.     av_freep(&s->b_bidir_forw_mv_table_base);
  614.     av_freep(&s->b_bidir_back_mv_table_base);
  615.     av_freep(&s->b_direct_mv_table_base);
  616.     s->p_mv_table= NULL;
  617.     s->b_forw_mv_table= NULL;
  618.     s->b_back_mv_table= NULL;
  619.     s->b_bidir_forw_mv_table= NULL;
  620.     s->b_bidir_back_mv_table= NULL;
  621.     s->b_direct_mv_table= NULL;
  622.     for(i=0; i<2; i++){
  623.         for(j=0; j<2; j++){
  624.             for(k=0; k<2; k++){
  625.                 av_freep(&s->b_field_mv_table_base[i][j][k]);
  626.                 s->b_field_mv_table[i][j][k]=NULL;
  627.             }
  628.             av_freep(&s->b_field_select_table[i][j]);
  629.             av_freep(&s->p_field_mv_table_base[i][j]);
  630.             s->p_field_mv_table[i][j]=NULL;
  631.         }
  632.         av_freep(&s->p_field_select_table[i]);
  633.     }
  635.     av_freep(&s->dc_val_base);
  636.     av_freep(&s->ac_val_base);
  637.     av_freep(&s->coded_block_base);
  638.     av_freep(&s->mbintra_table);
  639.     av_freep(&s->cbp_table);
  640.     av_freep(&s->pred_dir_table);
  642.     av_freep(&s->mbskip_table);
  643.     av_freep(&s->prev_pict_types);
  644.     av_freep(&s->bitstream_buffer);
  645.     s->allocated_bitstream_buffer_size=0;
  647.     av_freep(&s->avctx->stats_out);
  648.     av_freep(&s->ac_stats);
  649.     av_freep(&s->error_status_table);
  650.     av_freep(&s->mb_index2xy);
  651.     av_freep(&s->lambda_table);
  652.     av_freep(&s->q_intra_matrix);
  653.     av_freep(&s->q_inter_matrix);
  654.     av_freep(&s->q_intra_matrix16);
  655.     av_freep(&s->q_inter_matrix16);
  656.     av_freep(&s->input_picture);
  657.     av_freep(&s->reordered_input_picture);
  658.     av_freep(&s->dct_offset);
  660.     if(s->picture){
  661.         for(i=0; i<MAX_PICTURE_COUNT; i++){
  662.             free_picture(s, &s->picture[i]);
  663.         }
  664.     }
  665.     av_freep(&s->picture);
  666.     s->context_initialized = 0;
  667.     s->last_picture_ptr=
  668.     s->next_picture_ptr=
  669.     s->current_picture_ptr= NULL;
  670.     s->linesize= s->uvlinesize= 0;
  672.     for(i=0; i<3; i++)
  673.         av_freep(&s->visualization_buffer[i]);
  675.     avcodec_default_free_buffers(s->avctx);
  676. }
  678. void init_rl(RLTable *rl, uint8_t static_store[2][2*MAX_RUN + MAX_LEVEL + 3])
  679. {
  680.     int8_t max_level[MAX_RUN+1], max_run[MAX_LEVEL+1];
  681.     uint8_t index_run[MAX_RUN+1];
  682.     int last, run, level, start, end, i;
  684.     /* If table is static, we can quit if rl->max_level[0] is not NULL */
  685.     if(static_store && rl->max_level[0])
  686.         return;
  688.     /* compute max_level[], max_run[] and index_run[] */
  689.     for(last=0;last<2;last++) {
  690.         if (last == 0) {
  691.             start = 0;
  692.             end = rl->last;
  693.         } else {
  694.             start = rl->last;
  695.             end = rl->n;
  696.         }
  698.         memset(max_level, 0, MAX_RUN + 1);
  699.         memset(max_run, 0, MAX_LEVEL + 1);
  700.         memset(index_run, rl->n, MAX_RUN + 1);
  701.         for(i=start;i<end;i++) {
  702.             run = rl->table_run[i];
  703.             level = rl->table_level[i];
  704.             if (index_run[run] == rl->n)
  705.                 index_run[run] = i;
  706.             if (level > max_level[run])
  707.                 max_level[run] = level;
  708.             if (run > max_run[level])
  709.                 max_run[level] = run;
  710.         }
  711.         if(static_store)
  712.             rl->max_level[last] = static_store[last];
  713.         else
  714.             rl->max_level[last] = av_malloc(MAX_RUN + 1);
  715.         memcpy(rl->max_level[last], max_level, MAX_RUN + 1);
  716.         if(static_store)
  717.             rl->max_run[last] = static_store[last] + MAX_RUN + 1;
  718.         else
  719.             rl->max_run[last] = av_malloc(MAX_LEVEL + 1);
  720.         memcpy(rl->max_run[last], max_run, MAX_LEVEL + 1);
  721.         if(static_store)
  722.             rl->index_run[last] = static_store[last] + MAX_RUN + MAX_LEVEL + 2;
  723.         else
  724.             rl->index_run[last] = av_malloc(MAX_RUN + 1);
  725.         memcpy(rl->index_run[last], index_run, MAX_RUN + 1);
  726.     }
  727. }
  729. void init_vlc_rl(RLTable *rl)
  730. {
  731.     int i, q;
  733.     for(q=0; q<32; q++){
  734.         int qmul= q*2;
  735.         int qadd= (q-1)|1;
  737.         if(q==0){
  738.             qmul=1;
  739.             qadd=0;
  740.         }
  741.         for(i=0; i<rl->vlc.table_size; i++){
  742.             int code= rl->vlc.table[i][0];
  743.             int len = rl->vlc.table[i][1];
  744.             int level, run;
  746.             if(len==0){ // illegal code
  747.                 run= 66;
  748.                 level= MAX_LEVEL;
  749.             }else if(len<0){ //more bits needed
  750.                 run= 0;
  751.                 level= code;
  752.             }else{
  753.                 if(code==rl->n){ //esc
  754.                     run= 66;
  755.                     level= 0;
  756.                 }else{
  757.                     run=   rl->table_run  [code] + 1;
  758.                     level= rl->table_level[code] * qmul + qadd;
  759.                     if(code >= rl->last) run+=192;
  760.                 }
  761.             }
  762.             rl->rl_vlc[q][i].len= len;
  763.             rl->rl_vlc[q][i].level= level;
  764.             rl->rl_vlc[q][i].run= run;
  765.         }
  766.     }
  767. }
  769. int ff_find_unused_picture(MpegEncContext *s, int shared){
  770.     int i;
  772.     if(shared){
  773.         for(i=0; i<MAX_PICTURE_COUNT; i++){
  774.             if(s->picture[i].data[0]==NULL && s->picture[i].type==0) return i;
  775.         }
  776.     }else{
  777.         for(i=0; i<MAX_PICTURE_COUNT; i++){
  778.             if(s->picture[i].data[0]==NULL && s->picture[i].type!=0) return i; //FIXME
  779.         }
  780.         for(i=0; i<MAX_PICTURE_COUNT; i++){
  781.             if(s->picture[i].data[0]==NULL) return i;
  782.         }
  783.     }
  785.     av_log(s->avctx, AV_LOG_FATAL, "Internal error, picture buffer overflown");
  786.     /* We could return -1, but the codec would crash trying to draw into a
  787.      * non-existing frame anyway. This is safer than waiting for a random crash.
  788.      * Also the return of this is never useful, an encoder must only allocate
  789.      * as much as allowed in the specification. This has no relationship to how
  790.      * much libavcodec could allocate (and MAX_PICTURE_COUNT is always large
  791.      * enough for such valid streams).
  792.      * Plus, a decoder has to check stream validity and remove frames if too
  793.      * many reference frames are around. Waiting for "OOM" is not correct at
  794.      * all. Similarly, missing reference frames have to be replaced by
  795.      * interpolated/MC frames, anything else is a bug in the codec ...
  796.      */
  797.     abort();
  798.     return -1;
  799. }
  801. static void update_noise_reduction(MpegEncContext *s){
  802.     int intra, i;
  804.     for(intra=0; intra<2; intra++){
  805.         if(s->dct_count[intra] > (1<<16)){
  806.             for(i=0; i<64; i++){
  807.                 s->dct_error_sum[intra][i] >>=1;
  808.             }
  809.             s->dct_count[intra] >>= 1;
  810.         }
  812.         for(i=0; i<64; i++){
  813.             s->dct_offset[intra][i]= (s->avctx->noise_reduction * s->dct_count[intra] + s->dct_error_sum[intra][i]/2) / (s->dct_error_sum[intra][i]+1);
  814.         }
  815.     }
  816. }
  818. /**
  819.  * generic function for encode/decode called after coding/decoding the header and before a frame is coded/decoded
  820.  */
  821. int MPV_frame_start(MpegEncContext *s, AVCodecContext *avctx)
  822. {
  823.     int i;
  824.     AVFrame *pic;
  825.     s->mb_skipped = 0;
  827.     assert(s->last_picture_ptr==NULL || s->out_format != FMT_H264 || s->codec_id == CODEC_ID_SVQ3);
  829.     /* mark&release old frames */
  830.     if (s->pict_type != FF_B_TYPE && s->last_picture_ptr && s->last_picture_ptr != s->next_picture_ptr && s->last_picture_ptr->data[0]) {
  831.       if(s->out_format != FMT_H264 || s->codec_id == CODEC_ID_SVQ3){
  832.         avctx->release_buffer(avctx, (AVFrame*)s->last_picture_ptr);
  834.         /* release forgotten pictures */
  835.         /* if(mpeg124/h263) */
  836.         if(!s->encoding){
  837.             for(i=0; i<MAX_PICTURE_COUNT; i++){
  838.                 if(s->picture[i].data[0] && &s->picture[i] != s->next_picture_ptr && s->picture[i].reference){
  839.                     av_log(avctx, AV_LOG_ERROR, "releasing zombie picturen");
  840.                     avctx->release_buffer(avctx, (AVFrame*)&s->picture[i]);
  841.                 }
  842.             }
  843.         }
  844.       }
  845.     }
  846. alloc:
  847.     if(!s->encoding){
  848.         /* release non reference frames */
  849.         for(i=0; i<MAX_PICTURE_COUNT; i++){
  850.             if(s->picture[i].data[0] && !s->picture[i].reference /*&& s->picture[i].type!=FF_BUFFER_TYPE_SHARED*/){
  851.                 s->avctx->release_buffer(s->avctx, (AVFrame*)&s->picture[i]);
  852.             }
  853.         }
  855.         if(s->current_picture_ptr && s->current_picture_ptr->data[0]==NULL)
  856.             pic= (AVFrame*)s->current_picture_ptr; //we already have a unused image (maybe it was set before reading the header)
  857.         else{
  858.             i= ff_find_unused_picture(s, 0);
  859.             pic= (AVFrame*)&s->picture[i];
  860.         }
  862.         pic->reference= 0;
  863.         if (!s->dropable){
  864.             if (s->codec_id == CODEC_ID_H264)
  865.                 pic->reference = s->picture_structure;
  866.             else if (s->pict_type != FF_B_TYPE)
  867.                 pic->reference = 3;
  868.         }
  870.         pic->coded_picture_number= s->coded_picture_number++;
  872.         if( alloc_picture(s, (Picture*)pic, 0) < 0)
  873.             return -1;
  875.         s->current_picture_ptr= (Picture*)pic;
  876.         s->current_picture_ptr->top_field_first= s->top_field_first; //FIXME use only the vars from current_pic
  877.         s->current_picture_ptr->interlaced_frame= !s->progressive_frame && !s->progressive_sequence;
  878.     }
  880.     s->current_picture_ptr->pict_type= s->pict_type;
  881. //    if(s->flags && CODEC_FLAG_QSCALE)
  882.   //      s->current_picture_ptr->quality= s->new_picture_ptr->quality;
  883.     s->current_picture_ptr->key_frame= s->pict_type == FF_I_TYPE;
  885.     ff_copy_picture(&s->current_picture, s->current_picture_ptr);
  887.     if (s->pict_type != FF_B_TYPE) {
  888.         s->last_picture_ptr= s->next_picture_ptr;
  889.         if(!s->dropable)
  890.             s->next_picture_ptr= s->current_picture_ptr;
  891.     }
  892. /*    av_log(s->avctx, AV_LOG_DEBUG, "L%p N%p C%p L%p N%p C%p type:%d drop:%dn", s->last_picture_ptr, s->next_picture_ptr,s->current_picture_ptr,
  893.         s->last_picture_ptr    ? s->last_picture_ptr->data[0] : NULL,
  894.         s->next_picture_ptr    ? s->next_picture_ptr->data[0] : NULL,
  895.         s->current_picture_ptr ? s->current_picture_ptr->data[0] : NULL,
  896.         s->pict_type, s->dropable);*/
  898.     if(s->last_picture_ptr) ff_copy_picture(&s->last_picture, s->last_picture_ptr);
  899.     if(s->next_picture_ptr) ff_copy_picture(&s->next_picture, s->next_picture_ptr);
  901.     if(s->pict_type != FF_I_TYPE && (s->last_picture_ptr==NULL || s->last_picture_ptr->data[0]==NULL) && !s->dropable && s->codec_id != CODEC_ID_H264){
  902.         av_log(avctx, AV_LOG_ERROR, "warning: first frame is no keyframen");
  903.         assert(s->pict_type != FF_B_TYPE); //these should have been dropped if we don't have a reference
  904.         goto alloc;
  905.     }
  907.     assert(s->pict_type == FF_I_TYPE || (s->last_picture_ptr && s->last_picture_ptr->data[0]));
  909.     if(s->picture_structure!=PICT_FRAME && s->out_format != FMT_H264){
  910.         int i;
  911.         for(i=0; i<4; i++){
  912.             if(s->picture_structure == PICT_BOTTOM_FIELD){
  913.                  s->current_picture.data[i] += s->current_picture.linesize[i];
  914.             }
  915.             s->current_picture.linesize[i] *= 2;
  916.             s->last_picture.linesize[i] *=2;
  917.             s->next_picture.linesize[i] *=2;
  918.         }
  919.     }
  921.     s->hurry_up= s->avctx->hurry_up;
  922.     s->error_recognition= avctx->error_recognition;
  924.     /* set dequantizer, we can't do it during init as it might change for mpeg4
  925.        and we can't do it in the header decode as init is not called for mpeg4 there yet */
  926.     if(s->mpeg_quant || s->codec_id == CODEC_ID_MPEG2VIDEO){
  927.         s->dct_unquantize_intra = s->dct_unquantize_mpeg2_intra;
  928.         s->dct_unquantize_inter = s->dct_unquantize_mpeg2_inter;
  929.     }else if(s->out_format == FMT_H263 || s->out_format == FMT_H261){
  930.         s->dct_unquantize_intra = s->dct_unquantize_h263_intra;
  931.         s->dct_unquantize_inter = s->dct_unquantize_h263_inter;
  932.     }else{
  933.         s->dct_unquantize_intra = s->dct_unquantize_mpeg1_intra;
  934.         s->dct_unquantize_inter = s->dct_unquantize_mpeg1_inter;
  935.     }
  937.     if(s->dct_error_sum){
  938.         assert(s->avctx->noise_reduction && s->encoding);
  940.         update_noise_reduction(s);
  941.     }
  943. #ifdef CONFIG_XVMC
  944.     if(s->avctx->xvmc_acceleration)
  945.         return XVMC_field_start(s, avctx);
  946. #endif
  947.     return 0;
  948. }
  950. /* generic function for encode/decode called after a frame has been coded/decoded */
  951. void MPV_frame_end(MpegEncContext *s)
  952. {
  953.     int i;
  954.     /* draw edge for correct motion prediction if outside */
  955. #ifdef CONFIG_XVMC
  956. //just to make sure that all data is rendered.
  957.     if(s->avctx->xvmc_acceleration){
  958.         XVMC_field_end(s);
  959.     }else
  960. #endif
  961.     if(!(s->avctx->codec->capabilities&CODEC_CAP_HWACCEL_VDPAU)
  962.        && s->unrestricted_mv
  963.        && s->current_picture.reference
  964.        && !s->intra_only
  965.        && !(s->flags&CODEC_FLAG_EMU_EDGE)) {
  966.             s->dsp.draw_edges(s->current_picture.data[0], s->linesize  , s->h_edge_pos   , s->v_edge_pos   , EDGE_WIDTH  );
  967.             s->dsp.draw_edges(s->current_picture.data[1], s->uvlinesize, s->h_edge_pos>>1, s->v_edge_pos>>1, EDGE_WIDTH/2);
  968.             s->dsp.draw_edges(s->current_picture.data[2], s->uvlinesize, s->h_edge_pos>>1, s->v_edge_pos>>1, EDGE_WIDTH/2);
  969.     }
  970.     emms_c();
  972.     s->last_pict_type    = s->pict_type;
  973.     s->last_lambda_for[s->pict_type]= s->current_picture_ptr->quality;
  974.     if(s->pict_type!=FF_B_TYPE){
  975.         s->last_non_b_pict_type= s->pict_type;
  976.     }
  977. #if 0
  978.         /* copy back current_picture variables */
  979.     for(i=0; i<MAX_PICTURE_COUNT; i++){
  980.         if(s->picture[i].data[0] == s->current_picture.data[0]){
  981.             s->picture[i]= s->current_picture;
  982.             break;
  983.         }
  984.     }
  985.     assert(i<MAX_PICTURE_COUNT);
  986. #endif
  988.     if(s->encoding){
  989.         /* release non-reference frames */
  990.         for(i=0; i<MAX_PICTURE_COUNT; i++){
  991.             if(s->picture[i].data[0] && !s->picture[i].reference /*&& s->picture[i].type!=FF_BUFFER_TYPE_SHARED*/){
  992.                 s->avctx->release_buffer(s->avctx, (AVFrame*)&s->picture[i]);
  993.             }
  994.         }
  995.     }
  996.     // clear copies, to avoid confusion
  997. #if 0
  998.     memset(&s->last_picture, 0, sizeof(Picture));
  999.     memset(&s->next_picture, 0, sizeof(Picture));
  1000.     memset(&s->current_picture, 0, sizeof(Picture));
  1001. #endif
  1002.     s->avctx->coded_frame= (AVFrame*)s->current_picture_ptr;
  1003. }
  1005. /**
  1006.  * draws an line from (ex, ey) -> (sx, sy).
  1007.  * @param w width of the image
  1008.  * @param h height of the image
  1009.  * @param stride stride/linesize of the image
  1010.  * @param color color of the arrow
  1011.  */
  1012. /*
  1013. static void draw_line(uint8_t *buf, int sx, int sy, int ex, int ey, int w, int h, int stride, int color){
  1014.     int x, y, fr, f;
  1016.     sx= av_clip(sx, 0, w-1);
  1017.     sy= av_clip(sy, 0, h-1);
  1018.     ex= av_clip(ex, 0, w-1);
  1019.     ey= av_clip(ey, 0, h-1);
  1021.     buf[sy*stride + sx]+= color;
  1023.     if(FFABS(ex - sx) > FFABS(ey - sy)){
  1024.         if(sx > ex){
  1025.             FFSWAP(int, sx, ex);
  1026.             FFSWAP(int, sy, ey);
  1027.         }
  1028.         buf+= sx + sy*stride;
  1029.         ex-= sx;
  1030.         f= ((ey-sy)<<16)/ex;
  1031.         for(x= 0; x <= ex; x++){
  1032.             y = (x*f)>>16;
  1033.             fr= (x*f)&0xFFFF;
  1034.             buf[ y   *stride + x]+= (color*(0x10000-fr))>>16;
  1035.             buf[(y+1)*stride + x]+= (color*         fr )>>16;
  1036.         }
  1037.     }else{
  1038.         if(sy > ey){
  1039.             FFSWAP(int, sx, ex);
  1040.             FFSWAP(int, sy, ey);
  1041.         }
  1042.         buf+= sx + sy*stride;
  1043.         ey-= sy;
  1044.         if(ey) f= ((ex-sx)<<16)/ey;
  1045.         else   f= 0;
  1046.         for(y= 0; y <= ey; y++){
  1047.             x = (y*f)>>16;
  1048.             fr= (y*f)&0xFFFF;
  1049.             buf[y*stride + x  ]+= (color*(0x10000-fr))>>16;
  1050.             buf[y*stride + x+1]+= (color*         fr )>>16;
  1051.         }
  1052.     }
  1053. }
  1054. */
  1055. /**
  1056.  * draws an arrow from (ex, ey) -> (sx, sy).
  1057.  * @param w width of the image
  1058.  * @param h height of the image
  1059.  * @param stride stride/linesize of the image
  1060.  * @param color color of the arrow
  1061.  */
  1062. /*
  1063. static void draw_arrow(uint8_t *buf, int sx, int sy, int ex, int ey, int w, int h, int stride, int color){
  1064.     int dx,dy;
  1066.     sx= av_clip(sx, -100, w+100);
  1067.     sy= av_clip(sy, -100, h+100);
  1068.     ex= av_clip(ex, -100, w+100);
  1069.     ey= av_clip(ey, -100, h+100);
  1071.     dx= ex - sx;
  1072.     dy= ey - sy;
  1074.     if(dx*dx + dy*dy > 3*3){
  1075.         int rx=  dx + dy;
  1076.         int ry= -dx + dy;
  1077.         int length= ff_sqrt((rx*rx + ry*ry)<<8);
  1079.         //FIXME subpixel accuracy
  1080.         rx= ROUNDED_DIV(rx*3<<4, length);
  1081.         ry= ROUNDED_DIV(ry*3<<4, length);
  1083.         draw_line(buf, sx, sy, sx + rx, sy + ry, w, h, stride, color);
  1084.         draw_line(buf, sx, sy, sx - ry, sy + rx, w, h, stride, color);
  1085.     }
  1086.     draw_line(buf, sx, sy, ex, ey, w, h, stride, color);
  1087. }
  1088. */
  1089. /**
  1090.  * prints debuging info for the given picture.
  1091.  */
  1092. /*
  1093. void ff_print_debug_info(MpegEncContext *s, AVFrame *pict){
  1095.     if(!pict || !pict->mb_type) return;
  1097.     if(s->avctx->debug&(FF_DEBUG_SKIP | FF_DEBUG_QP | FF_DEBUG_MB_TYPE)){
  1098.         int x,y;
  1100.         av_log(s->avctx,AV_LOG_DEBUG,"New frame, type: ");
  1101.         switch (pict->pict_type) {
  1102.             case FF_I_TYPE: av_log(s->avctx,AV_LOG_DEBUG,"In"); break;
  1103.             case FF_P_TYPE: av_log(s->avctx,AV_LOG_DEBUG,"Pn"); break;
  1104.             case FF_B_TYPE: av_log(s->avctx,AV_LOG_DEBUG,"Bn"); break;
  1105.             case FF_S_TYPE: av_log(s->avctx,AV_LOG_DEBUG,"Sn"); break;
  1106.             case FF_SI_TYPE: av_log(s->avctx,AV_LOG_DEBUG,"SIn"); break;
  1107.             case FF_SP_TYPE: av_log(s->avctx,AV_LOG_DEBUG,"SPn"); break;
  1108.         }
  1109.         for(y=0; y<s->mb_height; y++){
  1110.             for(x=0; x<s->mb_width; x++){
  1111.                 if(s->avctx->debug&FF_DEBUG_SKIP){
  1112.                     int count= s->mbskip_table[x + y*s->mb_stride];
  1113.                     if(count>9) count=9;
  1114.                     av_log(s->avctx, AV_LOG_DEBUG, "%1d", count);
  1115.                 }
  1116.                 if(s->avctx->debug&FF_DEBUG_QP){
  1117.                     av_log(s->avctx, AV_LOG_DEBUG, "%2d", pict->qscale_table[x + y*s->mb_stride]);
  1118.                 }
  1119.                 if(s->avctx->debug&FF_DEBUG_MB_TYPE){
  1120.                     int mb_type= pict->mb_type[x + y*s->mb_stride];
  1121.                     //Type & MV direction
  1122.                     if(IS_PCM(mb_type))
  1123.                         av_log(s->avctx, AV_LOG_DEBUG, "P");
  1124.                     else if(IS_INTRA(mb_type) && IS_ACPRED(mb_type))
  1125.                         av_log(s->avctx, AV_LOG_DEBUG, "A");
  1126.                     else if(IS_INTRA4x4(mb_type))
  1127.                         av_log(s->avctx, AV_LOG_DEBUG, "i");
  1128.                     else if(IS_INTRA16x16(mb_type))
  1129.                         av_log(s->avctx, AV_LOG_DEBUG, "I");
  1130.                     else if(IS_DIRECT(mb_type) && IS_SKIP(mb_type))
  1131.                         av_log(s->avctx, AV_LOG_DEBUG, "d");
  1132.                     else if(IS_DIRECT(mb_type))
  1133.                         av_log(s->avctx, AV_LOG_DEBUG, "D");
  1134.                     else if(IS_GMC(mb_type) && IS_SKIP(mb_type))
  1135.                         av_log(s->avctx, AV_LOG_DEBUG, "g");
  1136.                     else if(IS_GMC(mb_type))
  1137.                         av_log(s->avctx, AV_LOG_DEBUG, "G");
  1138.                     else if(IS_SKIP(mb_type))
  1139.                         av_log(s->avctx, AV_LOG_DEBUG, "S");
  1140.                     else if(!USES_LIST(mb_type, 1))
  1141.                         av_log(s->avctx, AV_LOG_DEBUG, ">");
  1142.                     else if(!USES_LIST(mb_type, 0))
  1143.                         av_log(s->avctx, AV_LOG_DEBUG, "<");
  1144.                     else{
  1145.                         assert(USES_LIST(mb_type, 0) && USES_LIST(mb_type, 1));
  1146.                         av_log(s->avctx, AV_LOG_DEBUG, "X");
  1147.                     }
  1149.                     //segmentation
  1150.                     if(IS_8X8(mb_type))
  1151.                         av_log(s->avctx, AV_LOG_DEBUG, "+");
  1152.                     else if(IS_16X8(mb_type))
  1153.                         av_log(s->avctx, AV_LOG_DEBUG, "-");
  1154.                     else if(IS_8X16(mb_type))
  1155.                         av_log(s->avctx, AV_LOG_DEBUG, "|");
  1156.                     else if(IS_INTRA(mb_type) || IS_16X16(mb_type))
  1157.                         av_log(s->avctx, AV_LOG_DEBUG, " ");
  1158.                     else
  1159.                         av_log(s->avctx, AV_LOG_DEBUG, "?");
  1162.                     if(IS_INTERLACED(mb_type) && s->codec_id == CODEC_ID_H264)
  1163.                         av_log(s->avctx, AV_LOG_DEBUG, "=");
  1164.                     else
  1165.                         av_log(s->avctx, AV_LOG_DEBUG, " ");
  1166.                 }
  1167. //                av_log(s->avctx, AV_LOG_DEBUG, " ");
  1168.             }
  1169.             av_log(s->avctx, AV_LOG_DEBUG, "n");
  1170.         }
  1171.     }
  1173.     if((s->avctx->debug&(FF_DEBUG_VIS_QP|FF_DEBUG_VIS_MB_TYPE)) || (s->avctx->debug_mv)){
  1174.         const int shift= 1 + s->quarter_sample;
  1175.         int mb_y;
  1176.         uint8_t *ptr;
  1177.         int i;
  1178.         int h_chroma_shift, v_chroma_shift, block_height;
  1179.         const int width = s->avctx->width;
  1180.         const int height= s->avctx->height;
  1181.         const int mv_sample_log2= 4 - pict->motion_subsample_log2;
  1182.         const int mv_stride= (s->mb_width << mv_sample_log2) + (s->codec_id == CODEC_ID_H264 ? 0 : 1);
  1183.         s->low_delay=0; //needed to see the vectors without trashing the buffers
  1185.         avcodec_get_chroma_sub_sample(s->avctx->pix_fmt, &h_chroma_shift, &v_chroma_shift);
  1186.         for(i=0; i<3; i++){
  1187.             memcpy(s->visualization_buffer[i], pict->data[i], (i==0) ? pict->linesize[i]*height:pict->linesize[i]*height >> v_chroma_shift);
  1188.             pict->data[i]= s->visualization_buffer[i];
  1189.         }
  1190.         pict->type= FF_BUFFER_TYPE_COPY;
  1191.         ptr= pict->data[0];
  1192.         block_height = 16>>v_chroma_shift;
  1194.         for(mb_y=0; mb_y<s->mb_height; mb_y++){
  1195.             int mb_x;
  1196.             for(mb_x=0; mb_x<s->mb_width; mb_x++){
  1197.                 const int mb_index= mb_x + mb_y*s->mb_stride;
  1198.                 if((s->avctx->debug_mv) && pict->motion_val){
  1199.                   int type;
  1200.                   for(type=0; type<3; type++){
  1201.                     int direction = 0;
  1202.                     switch (type) {
  1203.                       case 0: if ((!(s->avctx->debug_mv&FF_DEBUG_VIS_MV_P_FOR)) || (pict->pict_type!=FF_P_TYPE))
  1204.                                 continue;
  1205.                               direction = 0;
  1206.                               break;
  1207.                       case 1: if ((!(s->avctx->debug_mv&FF_DEBUG_VIS_MV_B_FOR)) || (pict->pict_type!=FF_B_TYPE))
  1208.                                 continue;
  1209.                               direction = 0;
  1210.                               break;
  1211.                       case 2: if ((!(s->avctx->debug_mv&FF_DEBUG_VIS_MV_B_BACK)) || (pict->pict_type!=FF_B_TYPE))
  1212.                                 continue;
  1213.                               direction = 1;
  1214.                               break;
  1215.                     }
  1216.                     if(!USES_LIST(pict->mb_type[mb_index], direction))
  1217.                         continue;
  1219.                     if(IS_8X8(pict->mb_type[mb_index])){
  1220.                       int i;
  1221.                       for(i=0; i<4; i++){
  1222.                         int sx= mb_x*16 + 4 + 8*(i&1);
  1223.                         int sy= mb_y*16 + 4 + 8*(i>>1);
  1224.                         int xy= (mb_x*2 + (i&1) + (mb_y*2 + (i>>1))*mv_stride) << (mv_sample_log2-1);
  1225.                         int mx= (pict->motion_val[direction][xy][0]>>shift) + sx;
  1226.                         int my= (pict->motion_val[direction][xy][1]>>shift) + sy;
  1227.                         draw_arrow(ptr, sx, sy, mx, my, width, height, s->linesize, 100);
  1228.                       }
  1229.                     }else if(IS_16X8(pict->mb_type[mb_index])){
  1230.                       int i;
  1231.                       for(i=0; i<2; i++){
  1232.                         int sx=mb_x*16 + 8;
  1233.                         int sy=mb_y*16 + 4 + 8*i;
  1234.                         int xy= (mb_x*2 + (mb_y*2 + i)*mv_stride) << (mv_sample_log2-1);
  1235.                         int mx=(pict->motion_val[direction][xy][0]>>shift);
  1236.                         int my=(pict->motion_val[direction][xy][1]>>shift);
  1238.                         if(IS_INTERLACED(pict->mb_type[mb_index]))
  1239.                             my*=2;
  1241.                         draw_arrow(ptr, sx, sy, mx+sx, my+sy, width, height, s->linesize, 100);
  1242.                       }
  1243.                     }else if(IS_8X16(pict->mb_type[mb_index])){
  1244.                       int i;
  1245.                       for(i=0; i<2; i++){
  1246.                         int sx=mb_x*16 + 4 + 8*i;
  1247.                         int sy=mb_y*16 + 8;
  1248.                         int xy= (mb_x*2 + i + mb_y*2*mv_stride) << (mv_sample_log2-1);
  1249.                         int mx=(pict->motion_val[direction][xy][0]>>shift);
  1250.                         int my=(pict->motion_val[direction][xy][1]>>shift);
  1252.                         if(IS_INTERLACED(pict->mb_type[mb_index]))
  1253.                             my*=2;
  1255.                         draw_arrow(ptr, sx, sy, mx+sx, my+sy, width, height, s->linesize, 100);
  1256.                       }
  1257.                     }else{
  1258.                       int sx= mb_x*16 + 8;
  1259.                       int sy= mb_y*16 + 8;
  1260.                       int xy= (mb_x + mb_y*mv_stride) << mv_sample_log2;
  1261.                       int mx= (pict->motion_val[direction][xy][0]>>shift) + sx;
  1262.                       int my= (pict->motion_val[direction][xy][1]>>shift) + sy;
  1263.                       draw_arrow(ptr, sx, sy, mx, my, width, height, s->linesize, 100);
  1264.                     }
  1265.                   }
  1266.                 }
  1267.                 if((s->avctx->debug&FF_DEBUG_VIS_QP) && pict->motion_val){
  1268.                     uint64_t c= (pict->qscale_table[mb_index]*128/31) * 0x0101010101010101ULL;
  1269.                     int y;
  1270.                     for(y=0; y<block_height; y++){
  1271.                         *(uint64_t*)(pict->data[1] + 8*mb_x + (block_height*mb_y + y)*pict->linesize[1])= c;
  1272.                         *(uint64_t*)(pict->data[2] + 8*mb_x + (block_height*mb_y + y)*pict->linesize[2])= c;
  1273.                     }
  1274.                 }
  1275.                 if((s->avctx->debug&FF_DEBUG_VIS_MB_TYPE) && pict->motion_val){
  1276.                     int mb_type= pict->mb_type[mb_index];
  1277.                     uint64_t u,v;
  1278.                     int y;
  1279. #define COLOR(theta, r)
  1280. u= (int)(128 + r*cos(theta*3.141592/180));
  1281. v= (int)(128 + r*sin(theta*3.141592/180));
  1284.                     u=v=128;
  1285.                     if(IS_PCM(mb_type)){
  1286.                         COLOR(120,48)
  1287.                     }else if((IS_INTRA(mb_type) && IS_ACPRED(mb_type)) || IS_INTRA16x16(mb_type)){
  1288.                         COLOR(30,48)
  1289.                     }else if(IS_INTRA4x4(mb_type)){
  1290.                         COLOR(90,48)
  1291.                     }else if(IS_DIRECT(mb_type) && IS_SKIP(mb_type)){
  1292. //                        COLOR(120,48)
  1293.                     }else if(IS_DIRECT(mb_type)){
  1294.                         COLOR(150,48)
  1295.                     }else if(IS_GMC(mb_type) && IS_SKIP(mb_type)){
  1296.                         COLOR(170,48)
  1297.                     }else if(IS_GMC(mb_type)){
  1298.                         COLOR(190,48)
  1299.                     }else if(IS_SKIP(mb_type)){
  1300. //                        COLOR(180,48)
  1301.                     }else if(!USES_LIST(mb_type, 1)){
  1302.                         COLOR(240,48)
  1303.                     }else if(!USES_LIST(mb_type, 0)){
  1304.                         COLOR(0,48)
  1305.                     }else{
  1306.                         assert(USES_LIST(mb_type, 0) && USES_LIST(mb_type, 1));
  1307.                         COLOR(300,48)
  1308.                     }
  1310.                     u*= 0x0101010101010101ULL;
  1311.                     v*= 0x0101010101010101ULL;
  1312.                     for(y=0; y<block_height; y++){
  1313.                         *(uint64_t*)(pict->data[1] + 8*mb_x + (block_height*mb_y + y)*pict->linesize[1])= u;
  1314.                         *(uint64_t*)(pict->data[2] + 8*mb_x + (block_height*mb_y + y)*pict->linesize[2])= v;
  1315.                     }
  1317.                     //segmentation
  1318.                     if(IS_8X8(mb_type) || IS_16X8(mb_type)){
  1319.                         *(uint64_t*)(pict->data[0] + 16*mb_x + 0 + (16*mb_y + 8)*pict->linesize[0])^= 0x8080808080808080ULL;
  1320.                         *(uint64_t*)(pict->data[0] + 16*mb_x + 8 + (16*mb_y + 8)*pict->linesize[0])^= 0x8080808080808080ULL;
  1321.                     }
  1322.                     if(IS_8X8(mb_type) || IS_8X16(mb_type)){
  1323.                         for(y=0; y<16; y++)
  1324.                             pict->data[0][16*mb_x + 8 + (16*mb_y + y)*pict->linesize[0]]^= 0x80;
  1325.                     }
  1326.                     if(IS_8X8(mb_type) && mv_sample_log2 >= 2){
  1327.                         int dm= 1 << (mv_sample_log2-2);
  1328.                         for(i=0; i<4; i++){
  1329.                             int sx= mb_x*16 + 8*(i&1);
  1330.                             int sy= mb_y*16 + 8*(i>>1);
  1331.                             int xy= (mb_x*2 + (i&1) + (mb_y*2 + (i>>1))*mv_stride) << (mv_sample_log2-1);
  1332.                             //FIXME bidir
  1333.                             int32_t *mv = (int32_t*)&pict->motion_val[0][xy];
  1334.                             if(mv[0] != mv[dm] || mv[dm*mv_stride] != mv[dm*(mv_stride+1)])
  1335.                                 for(y=0; y<8; y++)
  1336.                                     pict->data[0][sx + 4 + (sy + y)*pict->linesize[0]]^= 0x80;
  1337.                             if(mv[0] != mv[dm*mv_stride] || mv[dm] != mv[dm*(mv_stride+1)])
  1338.                                 *(uint64_t*)(pict->data[0] + sx + (sy + 4)*pict->linesize[0])^= 0x8080808080808080ULL;
  1339.                         }
  1340.                     }
  1342.                     if(IS_INTERLACED(mb_type) && s->codec_id == CODEC_ID_H264){
  1343.                         // hmm
  1344.                     }
  1345.                 }
  1346.                 s->mbskip_table[mb_index]=0;
  1347.             }
  1348.         }
  1349.     }
  1350. }*/
  1352. static inline int hpel_motion_lowres(MpegEncContext *s,
  1353.                                   uint8_t *dest, uint8_t *src,
  1354.                                   int field_based, int field_select,
  1355.                                   int src_x, int src_y,
  1356.                                   int width, int height, int stride,
  1357.                                   int h_edge_pos, int v_edge_pos,
  1358.                                   int w, int h, h264_chroma_mc_func *pix_op,
  1359.                                   int motion_x, int motion_y)
  1360. {
  1361.     const int lowres= s->avctx->lowres;
  1362.     const int s_mask= (2<<lowres)-1;
  1363.     int emu=0;
  1364.     int sx, sy;
  1366.     if(s->quarter_sample){
  1367.         motion_x/=2;
  1368.         motion_y/=2;
  1369.     }
  1371.     sx= motion_x & s_mask;
  1372.     sy= motion_y & s_mask;
  1373.     src_x += motion_x >> (lowres+1);
  1374.     src_y += motion_y >> (lowres+1);
  1376.     src += src_y * stride + src_x;
  1378.     if(   src_x > h_edge_pos                 - (!!sx) - w
  1379.        || src_y >(v_edge_pos >> field_based) - (!!sy) - h){
  1380.         ff_emulated_edge_mc(s->edge_emu_buffer, src, s->linesize, w+1, (h+1)<<field_based,
  1381.                             src_x, src_y<<field_based, h_edge_pos, v_edge_pos);
  1382.         src= s->edge_emu_buffer;
  1383.         emu=1;
  1384.     }
  1386.     sx <<= 2 - lowres;
  1387.     sy <<= 2 - lowres;
  1388.     if(field_select)
  1389.         src += s->linesize;
  1390.     pix_op[lowres](dest, src, stride, h, sx, sy);
  1391.     return emu;
  1392. }
  1394. /* apply one mpeg motion vector to the three components */
  1395. static av_always_inline void mpeg_motion_lowres(MpegEncContext *s,
  1396.                                uint8_t *dest_y, uint8_t *dest_cb, uint8_t *dest_cr,
  1397.                                int field_based, int bottom_field, int field_select,
  1398.                                uint8_t **ref_picture, h264_chroma_mc_func *pix_op,
  1399.                                int motion_x, int motion_y, int h)
  1400. {
  1401.     uint8_t *ptr_y, *ptr_cb, *ptr_cr;
  1402.     int mx, my, src_x, src_y, uvsrc_x, uvsrc_y, uvlinesize, linesize, sx, sy, uvsx, uvsy;
  1403.     const int lowres= s->avctx->lowres;
  1404.     const int block_s= 8>>lowres;
  1405.     const int s_mask= (2<<lowres)-1;
  1406.     const int h_edge_pos = s->h_edge_pos >> lowres;
  1407.     const int v_edge_pos = s->v_edge_pos >> lowres;
  1408.     linesize   = s->current_picture.linesize[0] << field_based;
  1409.     uvlinesize = s->current_picture.linesize[1] << field_based;
  1411.     if(s->quarter_sample){ //FIXME obviously not perfect but qpel will not work in lowres anyway
  1412.         motion_x/=2;
  1413.         motion_y/=2;
  1414.     }
  1416.     if(field_based){
  1417.         motion_y += (bottom_field - field_select)*((1<<lowres)-1);
  1418.     }
  1420.     sx= motion_x & s_mask;
  1421.     sy= motion_y & s_mask;
  1422.     src_x = s->mb_x*2*block_s               + (motion_x >> (lowres+1));
  1423.     src_y =(s->mb_y*2*block_s>>field_based) + (motion_y >> (lowres+1));
  1425.     if (s->out_format == FMT_H263) {
  1426.         uvsx = ((motion_x>>1) & s_mask) | (sx&1);
  1427.         uvsy = ((motion_y>>1) & s_mask) | (sy&1);
  1428.         uvsrc_x = src_x>>1;
  1429.         uvsrc_y = src_y>>1;
  1430.     }else if(s->out_format == FMT_H261){//even chroma mv's are full pel in H261
  1431.         mx = motion_x / 4;
  1432.         my = motion_y / 4;
  1433.         uvsx = (2*mx) & s_mask;
  1434.         uvsy = (2*my) & s_mask;
  1435.         uvsrc_x = s->mb_x*block_s               + (mx >> lowres);
  1436.         uvsrc_y = s->mb_y*block_s               + (my >> lowres);
  1437.     } else {
  1438.         mx = motion_x / 2;
  1439.         my = motion_y / 2;
  1440.         uvsx = mx & s_mask;
  1441.         uvsy = my & s_mask;
  1442.         uvsrc_x = s->mb_x*block_s               + (mx >> (lowres+1));
  1443.         uvsrc_y =(s->mb_y*block_s>>field_based) + (my >> (lowres+1));
  1444.     }
  1446.     ptr_y  = ref_picture[0] + src_y * linesize + src_x;
  1447.     ptr_cb = ref_picture[1] + uvsrc_y * uvlinesize + uvsrc_x;
  1448.     ptr_cr = ref_picture[2] + uvsrc_y * uvlinesize + uvsrc_x;
  1450.     if(   src_x > h_edge_pos                 - (!!sx) - 2*block_s
  1451.        || src_y >(v_edge_pos >> field_based) - (!!sy) - h){
  1452.             ff_emulated_edge_mc(s->edge_emu_buffer, ptr_y, s->linesize, 17, 17+field_based,
  1453.                              src_x, src_y<<field_based, h_edge_pos, v_edge_pos);
  1454.             ptr_y = s->edge_emu_buffer;
  1455.             if(!ENABLE_GRAY || !(s->flags&CODEC_FLAG_GRAY)){
  1456.                 uint8_t *uvbuf= s->edge_emu_buffer+18*s->linesize;
  1457.                 ff_emulated_edge_mc(uvbuf  , ptr_cb, s->uvlinesize, 9, 9+field_based,
  1458.                                  uvsrc_x, uvsrc_y<<field_based, h_edge_pos>>1, v_edge_pos>>1);
  1459.                 ff_emulated_edge_mc(uvbuf+16, ptr_cr, s->uvlinesize, 9, 9+field_based,
  1460.                                  uvsrc_x, uvsrc_y<<field_based, h_edge_pos>>1, v_edge_pos>>1);
  1461.                 ptr_cb= uvbuf;
  1462.                 ptr_cr= uvbuf+16;
  1463.             }
  1464.     }
  1466.     if(bottom_field){ //FIXME use this for field pix too instead of the obnoxious hack which changes picture.data
  1467.         dest_y += s->linesize;
  1468.         dest_cb+= s->uvlinesize;
  1469.         dest_cr+= s->uvlinesize;
  1470.     }
  1472.     if(field_select){
  1473.         ptr_y += s->linesize;
  1474.         ptr_cb+= s->uvlinesize;
  1475.         ptr_cr+= s->uvlinesize;
  1476.     }
  1478.     sx <<= 2 - lowres;
  1479.     sy <<= 2 - lowres;
  1480.     pix_op[lowres-1](dest_y, ptr_y, linesize, h, sx, sy);
  1482.     if(!ENABLE_GRAY || !(s->flags&CODEC_FLAG_GRAY)){
  1483.         uvsx <<= 2 - lowres;
  1484.         uvsy <<= 2 - lowres;
  1485.         pix_op[lowres](dest_cb, ptr_cb, uvlinesize, h >> s->chroma_y_shift, uvsx, uvsy);
  1486.         pix_op[lowres](dest_cr, ptr_cr, uvlinesize, h >> s->chroma_y_shift, uvsx, uvsy);
  1487.     }
  1488.     //FIXME h261 lowres loop filter
  1489. }
  1491. static inline void chroma_4mv_motion_lowres(MpegEncContext *s,
  1492.                                      uint8_t *dest_cb, uint8_t *dest_cr,
  1493.                                      uint8_t **ref_picture,
  1494.                                      h264_chroma_mc_func *pix_op,
  1495.                                      int mx, int my){
  1496.     const int lowres= s->avctx->lowres;
  1497.     const int block_s= 8>>lowres;
  1498.     const int s_mask= (2<<lowres)-1;
  1499.     const int h_edge_pos = s->h_edge_pos >> (lowres+1);
  1500.     const int v_edge_pos = s->v_edge_pos >> (lowres+1);
  1501.     int emu=0, src_x, src_y, offset, sx, sy;
  1502.     uint8_t *ptr;
  1504.     if(s->quarter_sample){
  1505.         mx/=2;
  1506.         my/=2;
  1507.     }
  1509.     /* In case of 8X8, we construct a single chroma motion vector
  1510.        with a special rounding */
  1511.     mx= ff_h263_round_chroma(mx);
  1512.     my= ff_h263_round_chroma(my);
  1514.     sx= mx & s_mask;
  1515.     sy= my & s_mask;
  1516.     src_x = s->mb_x*block_s + (mx >> (lowres+1));
  1517.     src_y = s->mb_y*block_s + (my >> (lowres+1));
  1519.     offset = src_y * s->uvlinesize + src_x;
  1520.     ptr = ref_picture[1] + offset;
  1521.     if(s->flags&CODEC_FLAG_EMU_EDGE){
  1522.         if(   src_x > h_edge_pos - (!!sx) - block_s
  1523.            || src_y > v_edge_pos - (!!sy) - block_s){
  1524.             ff_emulated_edge_mc(s->edge_emu_buffer, ptr, s->uvlinesize, 9, 9, src_x, src_y, h_edge_pos, v_edge_pos);
  1525.             ptr= s->edge_emu_buffer;
  1526.             emu=1;
  1527.         }
  1528.     }
  1529.     sx <<= 2 - lowres;
  1530.     sy <<= 2 - lowres;
  1531.     pix_op[lowres](dest_cb, ptr, s->uvlinesize, block_s, sx, sy);
  1533.     ptr = ref_picture[2] + offset;
  1534.     if(emu){
  1535.         ff_emulated_edge_mc(s->edge_emu_buffer, ptr, s->uvlinesize, 9, 9, src_x, src_y, h_edge_pos, v_edge_pos);
  1536.         ptr= s->edge_emu_buffer;
  1537.     }
  1538.     pix_op[lowres](dest_cr, ptr, s->uvlinesize, block_s, sx, sy);
  1539. }
  1541. /**
  1542.  * motion compensation of a single macroblock
  1543.  * @param s context
  1544.  * @param dest_y luma destination pointer
  1545.  * @param dest_cb chroma cb/u destination pointer
  1546.  * @param dest_cr chroma cr/v destination pointer
  1547.  * @param dir direction (0->forward, 1->backward)
  1548.  * @param ref_picture array[3] of pointers to the 3 planes of the reference picture
  1549.  * @param pic_op halfpel motion compensation function (average or put normally)
  1550.  * the motion vectors are taken from s->mv and the MV type from s->mv_type
  1551.  */
  1552. static inline void MPV_motion_lowres(MpegEncContext *s,
  1553.                               uint8_t *dest_y, uint8_t *dest_cb, uint8_t *dest_cr,
  1554.                               int dir, uint8_t **ref_picture,
  1555.                               h264_chroma_mc_func *pix_op)
  1556. {
  1557.     int mx, my;
  1558.     int mb_x, mb_y, i;
  1559.     const int lowres= s->avctx->lowres;
  1560.     const int block_s= 8>>lowres;
  1562.     mb_x = s->mb_x;
  1563.     mb_y = s->mb_y;
  1565.     switch(s->mv_type) {
  1566.     case MV_TYPE_16X16:
  1567.         mpeg_motion_lowres(s, dest_y, dest_cb, dest_cr,
  1568.                     0, 0, 0,
  1569.                     ref_picture, pix_op,
  1570.                     s->mv[dir][0][0], s->mv[dir][0][1], 2*block_s);
  1571.         break;
  1572.     case MV_TYPE_8X8:
  1573.         mx = 0;
  1574.         my = 0;
  1575.             for(i=0;i<4;i++) {
  1576.                 hpel_motion_lowres(s, dest_y + ((i & 1) + (i >> 1) * s->linesize)*block_s,
  1577.                             ref_picture[0], 0, 0,
  1578.                             (2*mb_x + (i & 1))*block_s, (2*mb_y + (i >>1))*block_s,
  1579.                             s->width, s->height, s->linesize,
  1580.                             s->h_edge_pos >> lowres, s->v_edge_pos >> lowres,
  1581.                             block_s, block_s, pix_op,
  1582.                             s->mv[dir][i][0], s->mv[dir][i][1]);
  1584.                 mx += s->mv[dir][i][0];
  1585.                 my += s->mv[dir][i][1];
  1586.             }
  1588.         if(!ENABLE_GRAY || !(s->flags&CODEC_FLAG_GRAY))
  1589.             chroma_4mv_motion_lowres(s, dest_cb, dest_cr, ref_picture, pix_op, mx, my);
  1590.         break;
  1591.     case MV_TYPE_FIELD:
  1592.         if (s->picture_structure == PICT_FRAME) {
  1593.             /* top field */
  1594.             mpeg_motion_lowres(s, dest_y, dest_cb, dest_cr,
  1595.                         1, 0, s->field_select[dir][0],
  1596.                         ref_picture, pix_op,
  1597.                         s->mv[dir][0][0], s->mv[dir][0][1], block_s);
  1598.             /* bottom field */
  1599.             mpeg_motion_lowres(s, dest_y, dest_cb, dest_cr,
  1600.                         1, 1, s->field_select[dir][1],
  1601.                         ref_picture, pix_op,
  1602.                         s->mv[dir][1][0], s->mv[dir][1][1], block_s);
  1603.         } else {
  1604.             if(s->picture_structure != s->field_select[dir][0] + 1 && s->pict_type != FF_B_TYPE && !s->first_field){
  1605.                 ref_picture= s->current_picture_ptr->data;
  1606.             }
  1608.             mpeg_motion_lowres(s, dest_y, dest_cb, dest_cr,
  1609.                         0, 0, s->field_select[dir][0],
  1610.                         ref_picture, pix_op,
  1611.                         s->mv[dir][0][0], s->mv[dir][0][1], 2*block_s);
  1612.         }
  1613.         break;
  1614.     case MV_TYPE_16X8:
  1615.         for(i=0; i<2; i++){
  1616.             uint8_t ** ref2picture;
  1618.             if(s->picture_structure == s->field_select[dir][i] + 1 || s->pict_type == FF_B_TYPE || s->first_field){
  1619.                 ref2picture= ref_picture;
  1620.             }else{
  1621.                 ref2picture= s->current_picture_ptr->data;
  1622.             }
  1624.             mpeg_motion_lowres(s, dest_y, dest_cb, dest_cr,
  1625.                         0, 0, s->field_select[dir][i],
  1626.                         ref2picture, pix_op,
  1627.                         s->mv[dir][i][0], s->mv[dir][i][1] + 2*block_s*i, block_s);
  1629.             dest_y += 2*block_s*s->linesize;
  1630.             dest_cb+= (2*block_s>>s->chroma_y_shift)*s->uvlinesize;
  1631.             dest_cr+= (2*block_s>>s->chroma_y_shift)*s->uvlinesize;
  1632.         }
  1633.         break;
  1634.     case MV_TYPE_DMV:
  1635.         if(s->picture_structure == PICT_FRAME){
  1636.             for(i=0; i<2; i++){
  1637.                 int j;
  1638.                 for(j=0; j<2; j++){
  1639.                     mpeg_motion_lowres(s, dest_y, dest_cb, dest_cr,
  1640.                                 1, j, j^i,
  1641.                                 ref_picture, pix_op,
  1642.                                 s->mv[dir][2*i + j][0], s->mv[dir][2*i + j][1], block_s);
  1643.                 }
  1644.                 pix_op = s->dsp.avg_h264_chroma_pixels_tab;
  1645.             }
  1646.         }else{
  1647.             for(i=0; i<2; i++){
  1648.                 mpeg_motion_lowres(s, dest_y, dest_cb, dest_cr,
  1649.                             0, 0, s->picture_structure != i+1,
  1650.                             ref_picture, pix_op,
  1651.                             s->mv[dir][2*i][0],s->mv[dir][2*i][1],2*block_s);
  1653.                 // after put we make avg of the same block
  1654.                 pix_op = s->dsp.avg_h264_chroma_pixels_tab;
  1656.                 //opposite parity is always in the same frame if this is second field
  1657.                 if(!s->first_field){
  1658.                     ref_picture = s->current_picture_ptr->data;
  1659.                 }
  1660.             }
  1661.         }
  1662.     break;
  1663.     default: assert(0);
  1664.     }
  1665. }
  1667. /* put block[] to dest[] */
  1668. static inline void put_dct(MpegEncContext *s,
  1669.                            DCTELEM *block, int i, uint8_t *dest, int line_size, int qscale)
  1670. {
  1671.     s->dct_unquantize_intra(s, block, i, qscale);
  1672.     s->dsp.idct_put (dest, line_size, block);
  1673. }
  1675. /* add block[] to dest[] */
  1676. static inline void add_dct(MpegEncContext *s,
  1677.                            DCTELEM *block, int i, uint8_t *dest, int line_size)
  1678. {
  1679.     if (s->block_last_index[i] >= 0) {
  1680.         s->dsp.idct_add (dest, line_size, block);
  1681.     }
  1682. }
  1683. /*
  1684. static inline void add_dequant_dct(MpegEncContext *s,
  1685.                            DCTELEM *block, int i, uint8_t *dest, int line_size, int qscale)
  1686. {
  1687.     if (s->block_last_index[i] >= 0) {
  1688.         s->dct_unquantize_inter(s, block, i, qscale);
  1690.         s->dsp.idct_add (dest, line_size, block);
  1691.     }
  1692. }
  1693. */
  1694. /**
  1695.  * cleans dc, ac, coded_block for the current non intra MB
  1696.  */
  1697. void ff_clean_intra_table_entries(MpegEncContext *s)
  1698. {
  1699.     int wrap = s->b8_stride;
  1700.     int xy = s->block_index[0];
  1702.     s->dc_val[0][xy           ] =
  1703.     s->dc_val[0][xy + 1       ] =
  1704.     s->dc_val[0][xy     + wrap] =
  1705.     s->dc_val[0][xy + 1 + wrap] = 1024;
  1706.     /* ac pred */
  1707.     memset(s->ac_val[0][xy       ], 0, 32 * sizeof(int16_t));
  1708.     memset(s->ac_val[0][xy + wrap], 0, 32 * sizeof(int16_t));
  1709.     if (s->msmpeg4_version>=3) {
  1710.         s->coded_block[xy           ] =
  1711.         s->coded_block[xy + 1       ] =
  1712.         s->coded_block[xy     + wrap] =
  1713.         s->coded_block[xy + 1 + wrap] = 0;
  1714.     }
  1715.     /* chroma */
  1716.     wrap = s->mb_stride;
  1717.     xy = s->mb_x + s->mb_y * wrap;
  1718.     s->dc_val[1][xy] =
  1719.     s->dc_val[2][xy] = 1024;
  1720.     /* ac pred */
  1721.     memset(s->ac_val[1][xy], 0, 16 * sizeof(int16_t));
  1722.     memset(s->ac_val[2][xy], 0, 16 * sizeof(int16_t));
  1724.     s->mbintra_table[xy]= 0;
  1725. }
  1727. /* generic function called after a macroblock has been parsed by the
  1728.    decoder or after it has been encoded by the encoder.
  1730.    Important variables used:
  1731.    s->mb_intra : true if intra macroblock
  1732.    s->mv_dir   : motion vector direction
  1733.    s->mv_type  : motion vector type
  1734.    s->mv       : motion vector
  1735.    s->interlaced_dct : true if interlaced dct used (mpeg2)
  1736.  */
  1737. static av_always_inline
  1738. void MPV_decode_mb_internal(MpegEncContext *s, DCTELEM block[12][64],
  1739.                             int lowres_flag, int is_mpeg12)
  1740. {
  1741.     int mb_x, mb_y;
  1742.     const int mb_xy = s->mb_y * s->mb_stride + s->mb_x;
  1743. #ifdef CONFIG_XVMC
  1744.     if(s->avctx->xvmc_acceleration){
  1745.         XVMC_decode_mb(s);//xvmc uses pblocks
  1746.         return;
  1747.     }
  1748. #endif
  1750.     mb_x = s->mb_x;
  1751.     mb_y = s->mb_y;
  1753.     if(s->avctx->debug&FF_DEBUG_DCT_COEFF) {
  1754.        /* save DCT coefficients */
  1755.        int i,j;
  1756.        DCTELEM *dct = &s->current_picture.dct_coeff[mb_xy*64*6];
  1757.        for(i=0; i<6; i++)
  1758.            for(j=0; j<64; j++)
  1759.                *dct++ = block[i][s->dsp.idct_permutation[j]];
  1760.     }
  1762.     s->current_picture.qscale_table[mb_xy]= s->qscale;
  1764.     /* update DC predictors for P macroblocks */
  1765.     if (!s->mb_intra) {
  1766.         if (!is_mpeg12 && (s->h263_pred || s->h263_aic)) {
  1767.             if(s->mbintra_table[mb_xy])
  1768.                 ff_clean_intra_table_entries(s);
  1769.         } else {
  1770.             s->last_dc[0] =
  1771.             s->last_dc[1] =
  1772.             s->last_dc[2] = 128 << s->intra_dc_precision;
  1773.         }
  1774.     }
  1775.     else if (!is_mpeg12 && (s->h263_pred || s->h263_aic))
  1776.         s->mbintra_table[mb_xy]=1;
  1778.     if ((s->flags&CODEC_FLAG_PSNR) || !(s->encoding && (s->intra_only || s->pict_type==FF_B_TYPE) && s->avctx->mb_decision != FF_MB_DECISION_RD)) { //FIXME precalc
  1779.         uint8_t *dest_y, *dest_cb, *dest_cr;
  1780.         int dct_linesize, dct_offset;
  1781.         op_pixels_func (*op_pix)[4];
  1782.         qpel_mc_func (*op_qpix)[16];
  1783.         const int linesize= s->current_picture.linesize[0]; //not s->linesize as this would be wrong for field pics
  1784.         const int uvlinesize= s->current_picture.linesize[1];
  1785.         const int readable= s->pict_type != FF_B_TYPE || s->encoding || s->avctx->draw_horiz_band || lowres_flag;
  1786.         const int block_size= lowres_flag ? 8>>s->avctx->lowres : 8;
  1788.         /* avoid copy if macroblock skipped in last frame too */
  1789.         /* skip only during decoding as we might trash the buffers during encoding a bit */
  1790.         if(!s->encoding){
  1791.             uint8_t *mbskip_ptr = &s->mbskip_table[mb_xy];
  1792.             const int age= s->current_picture.age;
  1794.             assert(age);
  1796.             if (s->mb_skipped) {
  1797.                 s->mb_skipped= 0;
  1798.                 assert(s->pict_type!=FF_I_TYPE);
  1800.                 (*mbskip_ptr) ++; /* indicate that this time we skipped it */
  1801.                 if(*mbskip_ptr >99) *mbskip_ptr= 99;
  1803.                 /* if previous was skipped too, then nothing to do !  */
  1804.                 if (*mbskip_ptr >= age && s->current_picture.reference){
  1805.                     return;
  1806.                 }
  1807.             } else if(!s->current_picture.reference){
  1808.                 (*mbskip_ptr) ++; /* increase counter so the age can be compared cleanly */
  1809.                 if(*mbskip_ptr >99) *mbskip_ptr= 99;
  1810.             } else{
  1811.                 *mbskip_ptr = 0; /* not skipped */
  1812.             }
  1813.         }
  1815.         dct_linesize = linesize << s->interlaced_dct;
  1816.         dct_offset =(s->interlaced_dct)? linesize : linesize*block_size;
  1818.         if(readable){
  1819.             dest_y=  s->dest[0];
  1820.             dest_cb= s->dest[1];
  1821.             dest_cr= s->dest[2];
  1822.         }else{
  1823.             dest_y = s->b_scratchpad;
  1824.             dest_cb= s->b_scratchpad+16*linesize;
  1825.             dest_cr= s->b_scratchpad+32*linesize;
  1826.         }
  1828.         if (!s->mb_intra) 
  1829. {
  1830.             /* motion handling */
  1831.             /* decoding or more than one mb_type (MC was already done otherwise) */
  1832.             if(!s->encoding)
  1833. {
  1834.                 if(lowres_flag)
  1835. {
  1836.                     h264_chroma_mc_func *op_pix = s->dsp.put_h264_chroma_pixels_tab;
  1838.                     if (s->mv_dir & MV_DIR_FORWARD) {
  1839.                         MPV_motion_lowres(s, dest_y, dest_cb, dest_cr, 0, s->last_picture.data, op_pix);
  1840.                         op_pix = s->dsp.avg_h264_chroma_pixels_tab;
  1841.                     }
  1842.                     if (s->mv_dir & MV_DIR_BACKWARD) {
  1843.                         MPV_motion_lowres(s, dest_y, dest_cb, dest_cr, 1, s->next_picture.data, op_pix);
  1844.                     }
  1845.                 }
  1846. else
  1847. {
  1848.                     op_qpix= s->me.qpel_put;
  1849.                     if ((!s->no_rounding) || s->pict_type==FF_B_TYPE){
  1850.                         op_pix = s->dsp.put_pixels_tab;
  1851.                     }else{
  1852.                         op_pix = s->dsp.put_no_rnd_pixels_tab;
  1853.                     }
  1854.                     if (s->mv_dir & MV_DIR_FORWARD) {
  1855.                         MPV_motion(s, dest_y, dest_cb, dest_cr, 0, s->last_picture.data, op_pix, op_qpix);
  1856.                         op_pix = s->dsp.avg_pixels_tab;
  1857.                         op_qpix= s->me.qpel_avg;
  1858.                     }
  1859.                     if (s->mv_dir & MV_DIR_BACKWARD) {
  1860.                         MPV_motion(s, dest_y, dest_cb, dest_cr, 1, s->next_picture.data, op_pix, op_qpix);
  1861.                     }
  1862.                 }
  1863.             }
  1865.             /* skip dequant / idct if we are really late ;) */
  1866.             if(s->hurry_up>1) goto skip_idct;
  1867.             if(s->avctx->skip_idct){
  1868.                 if(  (s->avctx->skip_idct >= AVDISCARD_NONREF && s->pict_type == FF_B_TYPE)
  1869.                    ||(s->avctx->skip_idct >= AVDISCARD_NONKEY && s->pict_type != FF_I_TYPE)
  1870.                    || s->avctx->skip_idct >= AVDISCARD_ALL)
  1871.                     goto skip_idct;
  1872.             }
  1874.             /* add dct residue */
  1875.             /*if(s->encoding 
  1876. || 
  1877. !(   s->h263_msmpeg4 
  1878. || 
  1879. s->codec_id==CODEC_ID_MPEG1VIDEO 
  1880. || 
  1881. s->codec_id==CODEC_ID_MPEG2VIDEO
  1882.                         || 
  1883. (s->codec_id==CODEC_ID_MPEG4 && !s->mpeg_quant)))
  1884. {
  1885.                 add_dequant_dct(s, block[0], 0, dest_y                          , dct_linesize, s->qscale);
  1886.                 add_dequant_dct(s, block[1], 1, dest_y              + block_size, dct_linesize, s->qscale);
  1887.                 add_dequant_dct(s, block[2], 2, dest_y + dct_offset             , dct_linesize, s->qscale);
  1888.                 add_dequant_dct(s, block[3], 3, dest_y + dct_offset + block_size, dct_linesize, s->qscale);
  1890.                 if(!ENABLE_GRAY || !(s->flags&CODEC_FLAG_GRAY)){
  1891.                     if (s->chroma_y_shift){
  1892.                         add_dequant_dct(s, block[4], 4, dest_cb, uvlinesize, s->chroma_qscale);
  1893.                         add_dequant_dct(s, block[5], 5, dest_cr, uvlinesize, s->chroma_qscale);
  1894.                     }else{
  1895.                         dct_linesize >>= 1;
  1896.                         dct_offset >>=1;
  1897.                         add_dequant_dct(s, block[4], 4, dest_cb,              dct_linesize, s->chroma_qscale);
  1898.                         add_dequant_dct(s, block[5], 5, dest_cr,              dct_linesize, s->chroma_qscale);
  1899.                         add_dequant_dct(s, block[6], 6, dest_cb + dct_offset, dct_linesize, s->chroma_qscale);
  1900.                         add_dequant_dct(s, block[7], 7, dest_cr + dct_offset, dct_linesize, s->chroma_qscale);
  1901.                     }
  1902.                 }
  1903.             } 
  1904. else*/ if(is_mpeg12 || (s->codec_id != CODEC_ID_WMV2))
  1905. {
  1906.                 add_dct(s, block[0], 0, dest_y                          , dct_linesize);
  1907.                 add_dct(s, block[1], 1, dest_y              + block_size, dct_linesize);
  1908.                 add_dct(s, block[2], 2, dest_y + dct_offset             , dct_linesize);
  1909.                 add_dct(s, block[3], 3, dest_y + dct_offset + block_size, dct_linesize);
  1911.                 if(!ENABLE_GRAY || !(s->flags&CODEC_FLAG_GRAY))
  1912. {
  1913.                     if(s->chroma_y_shift)
  1914. {//Chroma420
  1915.                         add_dct(s, block[4], 4, dest_cb, uvlinesize);
  1916.                         add_dct(s, block[5], 5, dest_cr, uvlinesize);
  1917.                     }
  1918. else
  1919. {
  1920.                         //chroma422
  1921.                         dct_linesize = uvlinesize << s->interlaced_dct;
  1922.                         dct_offset =(s->interlaced_dct)? uvlinesize : uvlinesize*8;
  1924.                         add_dct(s, block[4], 4, dest_cb, dct_linesize);
  1925.                         add_dct(s, block[5], 5, dest_cr, dct_linesize);
  1926.                         add_dct(s, block[6], 6, dest_cb+dct_offset, dct_linesize);
  1927.                         add_dct(s, block[7], 7, dest_cr+dct_offset, dct_linesize);
  1928.                         if(!s->chroma_x_shift){//Chroma444
  1929.                             add_dct(s, block[8], 8, dest_cb+8, dct_linesize);
  1930.                             add_dct(s, block[9], 9, dest_cr+8, dct_linesize);
  1931.                             add_dct(s, block[10], 10, dest_cb+8+dct_offset, dct_linesize);
  1932.                             add_dct(s, block[11], 11, dest_cr+8+dct_offset, dct_linesize);
  1933.                         }
  1934.                     }
  1935.                 }//fi gray
  1936.             }
  1937.             /*else if (ENABLE_WMV2) 
  1938. {
  1939.                 ff_wmv2_add_mb(s, block, dest_y, dest_cb, dest_cr);
  1940.             }*/
  1941.         } 
  1942. else 
  1943. {
  1944.             /* dct only in intra block */
  1945.             if(s->encoding || !(s->codec_id==CODEC_ID_MPEG1VIDEO || s->codec_id==CODEC_ID_MPEG2VIDEO)){
  1946.                 put_dct(s, block[0], 0, dest_y                          , dct_linesize, s->qscale);
  1947.                 put_dct(s, block[1], 1, dest_y              + block_size, dct_linesize, s->qscale);
  1948.                 put_dct(s, block[2], 2, dest_y + dct_offset             , dct_linesize, s->qscale);
  1949.                 put_dct(s, block[3], 3, dest_y + dct_offset + block_size, dct_linesize, s->qscale);
  1951.                 if(!ENABLE_GRAY || !(s->flags&CODEC_FLAG_GRAY)){
  1952.                     if(s->chroma_y_shift){
  1953.                         put_dct(s, block[4], 4, dest_cb, uvlinesize, s->chroma_qscale);
  1954.                         put_dct(s, block[5], 5, dest_cr, uvlinesize, s->chroma_qscale);
  1955.                     }else{
  1956.                         dct_offset >>=1;
  1957.                         dct_linesize >>=1;
  1958.                         put_dct(s, block[4], 4, dest_cb,              dct_linesize, s->chroma_qscale);
  1959.                         put_dct(s, block[5], 5, dest_cr,              dct_linesize, s->chroma_qscale);
  1960.                         put_dct(s, block[6], 6, dest_cb + dct_offset, dct_linesize, s->chroma_qscale);
  1961.                         put_dct(s, block[7], 7, dest_cr + dct_offset, dct_linesize, s->chroma_qscale);
  1962.                     }
  1963.                 }
  1964.             }else{
  1965.                 s->dsp.idct_put(dest_y                          , dct_linesize, block[0]);
  1966.                 s->dsp.idct_put(dest_y              + block_size, dct_linesize, block[1]);
  1967.                 s->dsp.idct_put(dest_y + dct_offset             , dct_linesize, block[2]);
  1968.                 s->dsp.idct_put(dest_y + dct_offset + block_size, dct_linesize, block[3]);
  1970.                 if(!ENABLE_GRAY || !(s->flags&CODEC_FLAG_GRAY)){
  1971.                     if(s->chroma_y_shift){
  1972.                         s->dsp.idct_put(dest_cb, uvlinesize, block[4]);
  1973.                         s->dsp.idct_put(dest_cr, uvlinesize, block[5]);
  1974.                     }else{
  1976.                         dct_linesize = uvlinesize << s->interlaced_dct;
  1977.                         dct_offset =(s->interlaced_dct)? uvlinesize : uvlinesize*8;
  1979.                         s->dsp.idct_put(dest_cb,              dct_linesize, block[4]);
  1980.                         s->dsp.idct_put(dest_cr,              dct_linesize, block[5]);
  1981.                         s->dsp.idct_put(dest_cb + dct_offset, dct_linesize, block[6]);
  1982.                         s->dsp.idct_put(dest_cr + dct_offset, dct_linesize, block[7]);
  1983.                         if(!s->chroma_x_shift){//Chroma444
  1984.                             s->dsp.idct_put(dest_cb + 8,              dct_linesize, block[8]);
  1985.                             s->dsp.idct_put(dest_cr + 8,              dct_linesize, block[9]);
  1986.                             s->dsp.idct_put(dest_cb + 8 + dct_offset, dct_linesize, block[10]);
  1987.                             s->dsp.idct_put(dest_cr + 8 + dct_offset, dct_linesize, block[11]);
  1988.                         }
  1989.                     }
  1990.                 }//gray
  1991.             }
  1992.         }
  1993. skip_idct:
  1994.         if(!readable){
  1995.             s->dsp.put_pixels_tab[0][0](s->dest[0], dest_y ,   linesize,16);
  1996.             s->dsp.put_pixels_tab[s->chroma_x_shift][0](s->dest[1], dest_cb, uvlinesize,16 >> s->chroma_y_shift);
  1997.             s->dsp.put_pixels_tab[s->chroma_x_shift][0](s->dest[2], dest_cr, uvlinesize,16 >> s->chroma_y_shift);
  1998.         }
  1999.     }
  2000. }
  2002. void MPV_decode_mb(MpegEncContext *s, DCTELEM block[12][64]){
  2003. #ifndef CONFIG_SMALL
  2004.     if(s->out_format == FMT_MPEG1) {
  2005.         if(s->avctx->lowres) MPV_decode_mb_internal(s, block, 1, 1);
  2006.         else                 MPV_decode_mb_internal(s, block, 0, 1);
  2007.     } else
  2008. #endif
  2009.     if(s->avctx->lowres) MPV_decode_mb_internal(s, block, 1, 0);
  2010.     else                  MPV_decode_mb_internal(s, block, 0, 0);
  2011. }
  2013. /**
  2014.  *
  2015.  * @param h is the normal height, this will be reduced automatically if needed for the last row
  2016.  */
  2017. void ff_draw_horiz_band(MpegEncContext *s, int y, int h){
  2018.     if (s->avctx->draw_horiz_band) {
  2019.         AVFrame *src;
  2020.         int offset[4];
  2022.         if(s->picture_structure != PICT_FRAME){
  2023.             h <<= 1;
  2024.             y <<= 1;
  2025.             if(s->first_field  && !(s->avctx->slice_flags&SLICE_FLAG_ALLOW_FIELD)) return;
  2026.         }
  2028.         h= FFMIN(h, s->avctx->height - y);
  2030.         if(s->pict_type==FF_B_TYPE || s->low_delay || (s->avctx->slice_flags&SLICE_FLAG_CODED_ORDER))
  2031.             src= (AVFrame*)s->current_picture_ptr;
  2032.         else if(s->last_picture_ptr)
  2033.             src= (AVFrame*)s->last_picture_ptr;
  2034.         else
  2035.             return;
  2037.         if(s->pict_type==FF_B_TYPE && s->picture_structure == PICT_FRAME && s->out_format != FMT_H264){
  2038.             offset[0]=
  2039.             offset[1]=
  2040.             offset[2]=
  2041.             offset[3]= 0;
  2042.         }else{
  2043.             offset[0]= y * s->linesize;
  2044.             offset[1]=
  2045.             offset[2]= (y >> s->chroma_y_shift) * s->uvlinesize;
  2046.             offset[3]= 0;
  2047.         }
  2049.         emms_c();
  2051.         s->avctx->draw_horiz_band(s->avctx, src, offset,
  2052.                                   y, s->picture_structure, h);
  2053.     }
  2054. }
  2056. void ff_init_block_index(MpegEncContext *s){ //FIXME maybe rename
  2057.     const int linesize= s->current_picture.linesize[0]; //not s->linesize as this would be wrong for field pics
  2058.     const int uvlinesize= s->current_picture.linesize[1];
  2059.     const int mb_size= 4 - s->avctx->lowres;
  2061.     s->block_index[0]= s->b8_stride*(s->mb_y*2    ) - 2 + s->mb_x*2;
  2062.     s->block_index[1]= s->b8_stride*(s->mb_y*2    ) - 1 + s->mb_x*2;
  2063.     s->block_index[2]= s->b8_stride*(s->mb_y*2 + 1) - 2 + s->mb_x*2;
  2064.     s->block_index[3]= s->b8_stride*(s->mb_y*2 + 1) - 1 + s->mb_x*2;
  2065.     s->block_index[4]= s->mb_stride*(s->mb_y + 1)                + s->b8_stride*s->mb_height*2 + s->mb_x - 1;
  2066.     s->block_index[5]= s->mb_stride*(s->mb_y + s->mb_height + 2) + s->b8_stride*s->mb_height*2 + s->mb_x - 1;
  2067.     //block_index is not used by mpeg2, so it is not affected by chroma_format
  2069.     s->dest[0] = s->current_picture.data[0] + ((s->mb_x - 1) << mb_size);
  2070.     s->dest[1] = s->current_picture.data[1] + ((s->mb_x - 1) << (mb_size - s->chroma_x_shift));
  2071.     s->dest[2] = s->current_picture.data[2] + ((s->mb_x - 1) << (mb_size - s->chroma_x_shift));
  2073.     if(!(s->pict_type==FF_B_TYPE && s->avctx->draw_horiz_band && s->picture_structure==PICT_FRAME))
  2074.     {
  2075.         s->dest[0] += s->mb_y *   linesize << mb_size;
  2076.         s->dest[1] += s->mb_y * uvlinesize << (mb_size - s->chroma_y_shift);
  2077.         s->dest[2] += s->mb_y * uvlinesize << (mb_size - s->chroma_y_shift);
  2078.     }
  2079. }
  2081. void ff_mpeg_flush(AVCodecContext *avctx){
  2082.     int i;
  2083.     MpegEncContext *s = avctx->priv_data;
  2085.     if(s==NULL || s->picture==NULL)
  2086.         return;
  2088.     for(i=0; i<MAX_PICTURE_COUNT; i++){
  2089.        if(s->picture[i].data[0] && (   s->picture[i].type == FF_BUFFER_TYPE_INTERNAL
  2090.                                     || s->picture[i].type == FF_BUFFER_TYPE_USER))
  2091.         avctx->release_buffer(avctx, (AVFrame*)&s->picture[i]);
  2092.     }
  2093.     s->current_picture_ptr = s->last_picture_ptr = s->next_picture_ptr = NULL;
  2095.     s->mb_x= s->mb_y= 0;
  2097.     s->parse_context.state= -1;
  2098.     s->parse_context.frame_start_found= 0;
  2099.     s->parse_context.overread= 0;
  2100.     s->parse_context.overread_index= 0;
  2101.     s->parse_context.index= 0;
  2102.     s->parse_context.last_index= 0;
  2103.     s->bitstream_buffer_size=0;
  2104.     s->pp_time=0;
  2105. }
  2107. static void dct_unquantize_mpeg1_intra_c(MpegEncContext *s,
  2108.                                    DCTELEM *block, int n, int qscale)
  2109. {
  2110.     int i, level, nCoeffs;
  2111.     const uint16_t *quant_matrix;
  2113.     nCoeffs= s->block_last_index[n];
  2115.     if (n < 4)
  2116.         block[0] = block[0] * s->y_dc_scale;
  2117.     else
  2118.         block[0] = block[0] * s->c_dc_scale;
  2119.     /* XXX: only mpeg1 */
  2120.     quant_matrix = s->intra_matrix;
  2121.     for(i=1;i<=nCoeffs;i++) {
  2122.         int j= s->intra_scantable.permutated[i];
  2123.         level = block[j];
  2124.         if (level) {
  2125.             if (level < 0) {
  2126.                 level = -level;
  2127.                 level = (int)(level * qscale * quant_matrix[j]) >> 3;
  2128.                 level = (level - 1) | 1;
  2129.                 level = -level;
  2130.             } else {
  2131.                 level = (int)(level * qscale * quant_matrix[j]) >> 3;
  2132.                 level = (level - 1) | 1;
  2133.             }
  2134.             block[j] = level;
  2135.         }
  2136.     }
  2137. }
  2139. static void dct_unquantize_mpeg1_inter_c(MpegEncContext *s,
  2140.                                    DCTELEM *block, int n, int qscale)
  2141. {
  2142.     int i, level, nCoeffs;
  2143.     const uint16_t *quant_matrix;
  2145.     nCoeffs= s->block_last_index[n];
  2147.     quant_matrix = s->inter_matrix;
  2148.     for(i=0; i<=nCoeffs; i++) {
  2149.         int j= s->intra_scantable.permutated[i];
  2150.         level = block[j];
  2151.         if (level) {
  2152.             if (level < 0) {
  2153.                 level = -level;
  2154.                 level = (((level << 1) + 1) * qscale *
  2155.                          ((int) (quant_matrix[j]))) >> 4;
  2156.                 level = (level - 1) | 1;
  2157.                 level = -level;
  2158.             } else {
  2159.                 level = (((level << 1) + 1) * qscale *
  2160.                          ((int) (quant_matrix[j]))) >> 4;
  2161.                 level = (level - 1) | 1;
  2162.             }
  2163.             block[j] = level;
  2164.         }
  2165.     }
  2166. }
  2168. static void dct_unquantize_mpeg2_intra_c(MpegEncContext *s,
  2169.                                    DCTELEM *block, int n, int qscale)
  2170. {
  2171.     int i, level, nCoeffs;
  2172.     const uint16_t *quant_matrix;
  2174.     if(s->alternate_scan) nCoeffs= 63;
  2175.     else nCoeffs= s->block_last_index[n];
  2177.     if (n < 4)
  2178.         block[0] = block[0] * s->y_dc_scale;
  2179.     else
  2180.         block[0] = block[0] * s->c_dc_scale;
  2181.     quant_matrix = s->intra_matrix;
  2182.     for(i=1;i<=nCoeffs;i++) {
  2183.         int j= s->intra_scantable.permutated[i];
  2184.         level = block[j];
  2185.         if (level) {
  2186.             if (level < 0) {
  2187.                 level = -level;
  2188.                 level = (int)(level * qscale * quant_matrix[j]) >> 3;
  2189.                 level = -level;
  2190.             } else {
  2191.                 level = (int)(level * qscale * quant_matrix[j]) >> 3;
  2192.             }
  2193.             block[j] = level;
  2194.         }
  2195.     }
  2196. }
  2198. static void dct_unquantize_mpeg2_intra_bitexact(MpegEncContext *s,
  2199.                                    DCTELEM *block, int n, int qscale)
  2200. {
  2201.     int i, level, nCoeffs;
  2202.     const uint16_t *quant_matrix;
  2203.     int sum=-1;
  2205.     if(s->alternate_scan) nCoeffs= 63;
  2206.     else nCoeffs= s->block_last_index[n];
  2208.     if (n < 4)
  2209.         block[0] = block[0] * s->y_dc_scale;
  2210.     else
  2211.         block[0] = block[0] * s->c_dc_scale;
  2212.     quant_matrix = s->intra_matrix;
  2213.     for(i=1;i<=nCoeffs;i++) {
  2214.         int j= s->intra_scantable.permutated[i];
  2215.         level = block[j];
  2216.         if (level) {
  2217.             if (level < 0) {
  2218.                 level = -level;
  2219.                 level = (int)(level * qscale * quant_matrix[j]) >> 3;
  2220.                 level = -level;
  2221.             } else {
  2222.                 level = (int)(level * qscale * quant_matrix[j]) >> 3;
  2223.             }
  2224.             block[j] = level;
  2225.             sum+=level;
  2226.         }
  2227.     }
  2228.     block[63]^=sum&1;
  2229. }
  2231. static void dct_unquantize_mpeg2_inter_c(MpegEncContext *s,
  2232.                                    DCTELEM *block, int n, int qscale)
  2233. {
  2234.     int i, level, nCoeffs;
  2235.     const uint16_t *quant_matrix;
  2236.     int sum=-1;
  2238.     if(s->alternate_scan) nCoeffs= 63;
  2239.     else nCoeffs= s->block_last_index[n];
  2241.     quant_matrix = s->inter_matrix;
  2242.     for(i=0; i<=nCoeffs; i++) {
  2243.         int j= s->intra_scantable.permutated[i];
  2244.         level = block[j];
  2245.         if (level) {
  2246.             if (level < 0) {
  2247.                 level = -level;
  2248.                 level = (((level << 1) + 1) * qscale *
  2249.                          ((int) (quant_matrix[j]))) >> 4;
  2250.                 level = -level;
  2251.             } else {
  2252.                 level = (((level << 1) + 1) * qscale *
  2253.                          ((int) (quant_matrix[j]))) >> 4;
  2254.             }
  2255.             block[j] = level;
  2256.             sum+=level;
  2257.         }
  2258.     }
  2259.     block[63]^=sum&1;
  2260. }
  2262. static void dct_unquantize_h263_intra_c(MpegEncContext *s,
  2263.                                   DCTELEM *block, int n, int qscale)
  2264. {
  2265.     int i, level, qmul, qadd;
  2266.     int nCoeffs;
  2268.     assert(s->block_last_index[n]>=0);
  2270.     qmul = qscale << 1;
  2272.     if (!s->h263_aic) {
  2273.         if (n < 4)
  2274.             block[0] = block[0] * s->y_dc_scale;
  2275.         else
  2276.             block[0] = block[0] * s->c_dc_scale;
  2277.         qadd = (qscale - 1) | 1;
  2278.     }else{
  2279.         qadd = 0;
  2280.     }
  2281.     if(s->ac_pred)
  2282.         nCoeffs=63;
  2283.     else
  2284.         nCoeffs= s->inter_scantable.raster_end[ s->block_last_index[n] ];
  2286.     for(i=1; i<=nCoeffs; i++) {
  2287.         level = block[i];
  2288.         if (level) {
  2289.             if (level < 0) {
  2290.                 level = level * qmul - qadd;
  2291.             } else {
  2292.                 level = level * qmul + qadd;
  2293.             }
  2294.             block[i] = level;
  2295.         }
  2296.     }
  2297. }
  2299. static void dct_unquantize_h263_inter_c(MpegEncContext *s,
  2300.                                   DCTELEM *block, int n, int qscale)
  2301. {
  2302.     int i, level, qmul, qadd;
  2303.     int nCoeffs;
  2305.     assert(s->block_last_index[n]>=0);
  2307.     qadd = (qscale - 1) | 1;
  2308.     qmul = qscale << 1;
  2310.     nCoeffs= s->inter_scantable.raster_end[ s->block_last_index[n] ];
  2312.     for(i=0; i<=nCoeffs; i++) {
  2313.         level = block[i];
  2314.         if (level) {
  2315.             if (level < 0) {
  2316.                 level = level * qmul - qadd;
  2317.             } else {
  2318.                 level = level * qmul + qadd;
  2319.             }
  2320.             block[i] = level;
  2321.         }
  2322.     }
  2323. }
  2325. /**
  2326.  * set qscale and update qscale dependent variables.
  2327.  */
  2328. void ff_set_qscale(MpegEncContext * s, int qscale)
  2329. {
  2330.     if (qscale < 1)
  2331.         qscale = 1;
  2332.     else if (qscale > 31)
  2333.         qscale = 31;
  2335.     s->qscale = qscale;
  2336.     s->chroma_qscale= s->chroma_qscale_table[qscale];
  2338.     s->y_dc_scale= s->y_dc_scale_table[ qscale ];
  2339.     s->c_dc_scale= s->c_dc_scale_table[ s->chroma_qscale ];
  2340. }