git.sesse.net Git - ffmpeg/blob - libavcodec/h264pred.c

   1 /*
   2  * H.26L/H.264/AVC/JVT/14496-10/... encoder/decoder
   3  * Copyright (c) 2003 Michael Niedermayer <michaelni@gmx.at>
   4  *
   5  * This file is part of FFmpeg.
   6  *
   7  * FFmpeg is free software; you can redistribute it and/or
   8  * modify it under the terms of the GNU Lesser General Public
   9  * License as published by the Free Software Foundation; either
  10  * version 2.1 of the License, or (at your option) any later version.
  11  *
  12  * FFmpeg is distributed in the hope that it will be useful,
  13  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  14  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
  15  * Lesser General Public License for more details.
  16  *
  17  * You should have received a copy of the GNU Lesser General Public
  18  * License along with FFmpeg; if not, write to the Free Software
  19  * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
  20  */
  21
  22 /**
  23  * @file
  24  * H.264 / AVC / MPEG4 part10 prediction functions.
  25  * @author Michael Niedermayer <michaelni@gmx.at>
  26  */
  27
  28 #include "h264pred.h"
  29
  30 #define BIT_DEPTH 8
  31 #include "h264pred_internal.h"
  32 #undef BIT_DEPTH
  33
  34 #define BIT_DEPTH 9
  35 #include "h264pred_internal.h"
  36 #undef BIT_DEPTH
  37
  38 #define BIT_DEPTH 10
  39 #include "h264pred_internal.h"
  40 #undef BIT_DEPTH
  41
  42 /**
  43  * Set the intra prediction function pointers.
  44  */
  45 void ff_h264_pred_init(H264PredContext *h, int codec_id, const int bit_depth){
  46 //    MpegEncContext * const s = &h->s;
  47
  48 #undef FUNC
  49 #undef FUNCC
  50 #define FUNC(a, depth) a ## _ ## depth
  51 #define FUNCC(a, depth) a ## _ ## depth ## _c
  52
  53 #define H264_PRED(depth) \
  54     if(codec_id != CODEC_ID_RV40){\
  55         if(codec_id == CODEC_ID_VP8) {\
  56             h->pred4x4[VERT_PRED       ]= FUNCC(pred4x4_vertical_vp8      , depth);\
  57             h->pred4x4[HOR_PRED        ]= FUNCC(pred4x4_horizontal_vp8    , depth);\
  58         } else {\
  59             h->pred4x4[VERT_PRED       ]= FUNCC(pred4x4_vertical          , depth);\
  60             h->pred4x4[HOR_PRED        ]= FUNCC(pred4x4_horizontal        , depth);\
  61         }\
  62         h->pred4x4[DC_PRED             ]= FUNCC(pred4x4_dc                , depth);\
  63         if(codec_id == CODEC_ID_SVQ3)\
  64             h->pred4x4[DIAG_DOWN_LEFT_PRED ]= FUNCC(pred4x4_down_left_svq3, depth);\
  65         else\
  66             h->pred4x4[DIAG_DOWN_LEFT_PRED ]= FUNCC(pred4x4_down_left     , depth);\
  67         h->pred4x4[DIAG_DOWN_RIGHT_PRED]= FUNCC(pred4x4_down_right        , depth);\
  68         h->pred4x4[VERT_RIGHT_PRED     ]= FUNCC(pred4x4_vertical_right    , depth);\
  69         h->pred4x4[HOR_DOWN_PRED       ]= FUNCC(pred4x4_horizontal_down   , depth);\
  70         if (codec_id == CODEC_ID_VP8) {\
  71             h->pred4x4[VERT_LEFT_PRED  ]= FUNCC(pred4x4_vertical_left_vp8 , depth);\
  72         } else\
  73             h->pred4x4[VERT_LEFT_PRED  ]= FUNCC(pred4x4_vertical_left     , depth);\
  74         h->pred4x4[HOR_UP_PRED         ]= FUNCC(pred4x4_horizontal_up     , depth);\
  75         if(codec_id != CODEC_ID_VP8) {\
  76             h->pred4x4[LEFT_DC_PRED    ]= FUNCC(pred4x4_left_dc           , depth);\
  77             h->pred4x4[TOP_DC_PRED     ]= FUNCC(pred4x4_top_dc            , depth);\
  78             h->pred4x4[DC_128_PRED     ]= FUNCC(pred4x4_128_dc            , depth);\
  79         } else {\
  80             h->pred4x4[TM_VP8_PRED     ]= FUNCC(pred4x4_tm_vp8            , depth);\
  81             h->pred4x4[DC_127_PRED     ]= FUNCC(pred4x4_127_dc            , depth);\
  82             h->pred4x4[DC_129_PRED     ]= FUNCC(pred4x4_129_dc            , depth);\
  83             h->pred4x4[VERT_VP8_PRED   ]= FUNCC(pred4x4_vertical          , depth);\
  84             h->pred4x4[HOR_VP8_PRED    ]= FUNCC(pred4x4_horizontal        , depth);\
  85         }\
  86     }else{\
  87         h->pred4x4[VERT_PRED           ]= FUNCC(pred4x4_vertical          , depth);\
  88         h->pred4x4[HOR_PRED            ]= FUNCC(pred4x4_horizontal        , depth);\
  89         h->pred4x4[DC_PRED             ]= FUNCC(pred4x4_dc                , depth);\
  90         h->pred4x4[DIAG_DOWN_LEFT_PRED ]= FUNCC(pred4x4_down_left_rv40    , depth);\
  91         h->pred4x4[DIAG_DOWN_RIGHT_PRED]= FUNCC(pred4x4_down_right        , depth);\
  92         h->pred4x4[VERT_RIGHT_PRED     ]= FUNCC(pred4x4_vertical_right    , depth);\
  93         h->pred4x4[HOR_DOWN_PRED       ]= FUNCC(pred4x4_horizontal_down   , depth);\
  94         h->pred4x4[VERT_LEFT_PRED      ]= FUNCC(pred4x4_vertical_left_rv40, depth);\
  95         h->pred4x4[HOR_UP_PRED         ]= FUNCC(pred4x4_horizontal_up_rv40, depth);\
  96         h->pred4x4[LEFT_DC_PRED        ]= FUNCC(pred4x4_left_dc           , depth);\
  97         h->pred4x4[TOP_DC_PRED         ]= FUNCC(pred4x4_top_dc            , depth);\
  98         h->pred4x4[DC_128_PRED         ]= FUNCC(pred4x4_128_dc            , depth);\
  99         h->pred4x4[DIAG_DOWN_LEFT_PRED_RV40_NODOWN]= FUNCC(pred4x4_down_left_rv40_nodown, depth);\
 100         h->pred4x4[HOR_UP_PRED_RV40_NODOWN]= FUNCC(pred4x4_horizontal_up_rv40_nodown    , depth);\
 101         h->pred4x4[VERT_LEFT_PRED_RV40_NODOWN]= FUNCC(pred4x4_vertical_left_rv40_nodown , depth);\
 102     }\
 103 \
 104     h->pred8x8l[VERT_PRED           ]= FUNCC(pred8x8l_vertical            , depth);\
 105     h->pred8x8l[HOR_PRED            ]= FUNCC(pred8x8l_horizontal          , depth);\
 106     h->pred8x8l[DC_PRED             ]= FUNCC(pred8x8l_dc                  , depth);\
 107     h->pred8x8l[DIAG_DOWN_LEFT_PRED ]= FUNCC(pred8x8l_down_left           , depth);\
 108     h->pred8x8l[DIAG_DOWN_RIGHT_PRED]= FUNCC(pred8x8l_down_right          , depth);\
 109     h->pred8x8l[VERT_RIGHT_PRED     ]= FUNCC(pred8x8l_vertical_right      , depth);\
 110     h->pred8x8l[HOR_DOWN_PRED       ]= FUNCC(pred8x8l_horizontal_down     , depth);\
 111     h->pred8x8l[VERT_LEFT_PRED      ]= FUNCC(pred8x8l_vertical_left       , depth);\
 112     h->pred8x8l[HOR_UP_PRED         ]= FUNCC(pred8x8l_horizontal_up       , depth);\
 113     h->pred8x8l[LEFT_DC_PRED        ]= FUNCC(pred8x8l_left_dc             , depth);\
 114     h->pred8x8l[TOP_DC_PRED         ]= FUNCC(pred8x8l_top_dc              , depth);\
 115     h->pred8x8l[DC_128_PRED         ]= FUNCC(pred8x8l_128_dc              , depth);\
 116 \
 117     h->pred8x8[VERT_PRED8x8   ]= FUNCC(pred8x8_vertical                   , depth);\
 118     h->pred8x8[HOR_PRED8x8    ]= FUNCC(pred8x8_horizontal                 , depth);\
 119     if (codec_id != CODEC_ID_VP8) {\
 120         h->pred8x8[PLANE_PRED8x8]= FUNCC(pred8x8_plane                    , depth);\
 121     } else\
 122         h->pred8x8[PLANE_PRED8x8]= FUNCC(pred8x8_tm_vp8                   , depth);\
 123     if(codec_id != CODEC_ID_RV40 && codec_id != CODEC_ID_VP8){\
 124         h->pred8x8[DC_PRED8x8     ]= FUNCC(pred8x8_dc                     , depth);\
 125         h->pred8x8[LEFT_DC_PRED8x8]= FUNCC(pred8x8_left_dc                , depth);\
 126         h->pred8x8[TOP_DC_PRED8x8 ]= FUNCC(pred8x8_top_dc                 , depth);\
 127         h->pred8x8[ALZHEIMER_DC_L0T_PRED8x8 ]= FUNC(pred8x8_mad_cow_dc_l0t, depth);\
 128         h->pred8x8[ALZHEIMER_DC_0LT_PRED8x8 ]= FUNC(pred8x8_mad_cow_dc_0lt, depth);\
 129         h->pred8x8[ALZHEIMER_DC_L00_PRED8x8 ]= FUNC(pred8x8_mad_cow_dc_l00, depth);\
 130         h->pred8x8[ALZHEIMER_DC_0L0_PRED8x8 ]= FUNC(pred8x8_mad_cow_dc_0l0, depth);\
 131     }else{\
 132         h->pred8x8[DC_PRED8x8     ]= FUNCC(pred8x8_dc_rv40                , depth);\
 133         h->pred8x8[LEFT_DC_PRED8x8]= FUNCC(pred8x8_left_dc_rv40           , depth);\
 134         h->pred8x8[TOP_DC_PRED8x8 ]= FUNCC(pred8x8_top_dc_rv40            , depth);\
 135         if (codec_id == CODEC_ID_VP8) {\
 136             h->pred8x8[DC_127_PRED8x8]= FUNCC(pred8x8_127_dc              , depth);\
 137             h->pred8x8[DC_129_PRED8x8]= FUNCC(pred8x8_129_dc              , depth);\
 138         }\
 139     }\
 140     h->pred8x8[DC_128_PRED8x8 ]= FUNCC(pred8x8_128_dc                     , depth);\
 141 \
 142     h->pred16x16[DC_PRED8x8     ]= FUNCC(pred16x16_dc                     , depth);\
 143     h->pred16x16[VERT_PRED8x8   ]= FUNCC(pred16x16_vertical               , depth);\
 144     h->pred16x16[HOR_PRED8x8    ]= FUNCC(pred16x16_horizontal             , depth);\
 145     switch(codec_id){\
 146     case CODEC_ID_SVQ3:\
 147        h->pred16x16[PLANE_PRED8x8  ]= FUNCC(pred16x16_plane_svq3          , depth);\
 148        break;\
 149     case CODEC_ID_RV40:\
 150        h->pred16x16[PLANE_PRED8x8  ]= FUNCC(pred16x16_plane_rv40          , depth);\
 151        break;\
 152     case CODEC_ID_VP8:\
 153        h->pred16x16[PLANE_PRED8x8  ]= FUNCC(pred16x16_tm_vp8              , depth);\
 154        h->pred16x16[DC_127_PRED8x8]= FUNCC(pred16x16_127_dc               , depth);\
 155        h->pred16x16[DC_129_PRED8x8]= FUNCC(pred16x16_129_dc               , depth);\
 156        break;\
 157     default:\
 158        h->pred16x16[PLANE_PRED8x8  ]= FUNCC(pred16x16_plane               , depth);\
 159        break;\
 160     }\
 161     h->pred16x16[LEFT_DC_PRED8x8]= FUNCC(pred16x16_left_dc                , depth);\
 162     h->pred16x16[TOP_DC_PRED8x8 ]= FUNCC(pred16x16_top_dc                 , depth);\
 163     h->pred16x16[DC_128_PRED8x8 ]= FUNCC(pred16x16_128_dc                 , depth);\
 164 \
 165     /* special lossless h/v prediction for h264 */ \
 166     h->pred4x4_add  [VERT_PRED   ]= FUNCC(pred4x4_vertical_add            , depth);\
 167     h->pred4x4_add  [ HOR_PRED   ]= FUNCC(pred4x4_horizontal_add          , depth);\
 168     h->pred8x8l_add [VERT_PRED   ]= FUNCC(pred8x8l_vertical_add           , depth);\
 169     h->pred8x8l_add [ HOR_PRED   ]= FUNCC(pred8x8l_horizontal_add         , depth);\
 170     h->pred8x8_add  [VERT_PRED8x8]= FUNCC(pred8x8_vertical_add            , depth);\
 171     h->pred8x8_add  [ HOR_PRED8x8]= FUNCC(pred8x8_horizontal_add          , depth);\
 172     h->pred16x16_add[VERT_PRED8x8]= FUNCC(pred16x16_vertical_add          , depth);\
 173     h->pred16x16_add[ HOR_PRED8x8]= FUNCC(pred16x16_horizontal_add        , depth);\
 174
 175     switch (bit_depth) {
 176         case 9:
 177             H264_PRED(9)
 178             break;
 179         case 10:
 180             H264_PRED(10)
 181             break;
 182         default:
 183             H264_PRED(8)
 184             break;
 185     }
 186
 187     if (ARCH_ARM) ff_h264_pred_init_arm(h, codec_id, bit_depth);
 188     if (HAVE_MMX) ff_h264_pred_init_x86(h, codec_id, bit_depth);
 189 }