falkTX
/
FFmpeg
mirror of https://github.com/falkTX/FFmpeg.git
1
0
Fork 0
Code Issues 0 Releases 338 Wiki Activity
You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
29345 Commits
32 Branches
283MB
Tree: fbfcc8f1b7
Branches Tags
${ item.name }
Create branch ${ searchTerm }
from 'fbfcc8f1b7'
${ noResults }
FFmpeg/libavcodec/h264pred.c

190 lines
9.7KB
Raw Blame History 

  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. }