Subversion Repositories Kolibri OS

Rev

Go to most recent revision | Blame | Last modification | View Log | RSS feed

  1. /*
  2.  * H.26L/H.264/AVC/JVT/14496-10/... encoder/decoder
  3.  * Copyright (c) 2003-2010 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 DSP functions.
  25.  * @author Michael Niedermayer <michaelni@gmx.at>
  26.  */
  27.  
  28. #include <stdint.h>
  29. #include "libavutil/attributes.h"
  30. #include "libavutil/avassert.h"
  31.  
  32. #include "avcodec.h"
  33. #include "h264dsp.h"
  34. #include "h264idct.h"
  35. #include "libavutil/common.h"
  36.  
  37. #define BIT_DEPTH 8
  38. #include "h264dsp_template.c"
  39. #undef BIT_DEPTH
  40.  
  41. #define BIT_DEPTH 9
  42. #include "h264dsp_template.c"
  43. #undef BIT_DEPTH
  44.  
  45. #define BIT_DEPTH 10
  46. #include "h264dsp_template.c"
  47. #undef BIT_DEPTH
  48.  
  49. #define BIT_DEPTH 12
  50. #include "h264dsp_template.c"
  51. #undef BIT_DEPTH
  52.  
  53. #define BIT_DEPTH 14
  54. #include "h264dsp_template.c"
  55. #undef BIT_DEPTH
  56.  
  57. #define BIT_DEPTH 8
  58. #include "h264addpx_template.c"
  59. #undef BIT_DEPTH
  60.  
  61. #define BIT_DEPTH 16
  62. #include "h264addpx_template.c"
  63. #undef BIT_DEPTH
  64.  
  65. static int h264_find_start_code_candidate_c(const uint8_t *buf, int size)
  66. {
  67.     int i = 0;
  68. #if HAVE_FAST_UNALIGNED
  69.     /* we check i < size instead of i + 3 / 7 because it is
  70.      * simpler and there must be FF_INPUT_BUFFER_PADDING_SIZE
  71.      * bytes at the end.
  72.      */
  73. #       if HAVE_FAST_64BIT
  74.     while (i < size &&
  75.             !((~*(const uint64_t *)(buf + i) &
  76.                     (*(const uint64_t *)(buf + i) - 0x0101010101010101ULL)) &
  77.                     0x8080808080808080ULL))
  78.         i += 8;
  79. #       else
  80.     while (i < size &&
  81.             !((~*(const uint32_t *)(buf + i) &
  82.                     (*(const uint32_t *)(buf + i) - 0x01010101U)) &
  83.                     0x80808080U))
  84.         i += 4;
  85. #       endif
  86. #endif
  87.     for (; i < size; i++)
  88.         if (!buf[i])
  89.             break;
  90.     return i;
  91. }
  92.  
  93. av_cold void ff_h264dsp_init(H264DSPContext *c, const int bit_depth,
  94.                              const int chroma_format_idc)
  95. {
  96. #undef FUNC
  97. #define FUNC(a, depth) a ## _ ## depth ## _c
  98.  
  99. #define ADDPX_DSP(depth) \
  100.     c->h264_add_pixels4_clear = FUNC(ff_h264_add_pixels4, depth);\
  101.     c->h264_add_pixels8_clear = FUNC(ff_h264_add_pixels8, depth)
  102.  
  103.     if (bit_depth > 8 && bit_depth <= 16) {
  104.         ADDPX_DSP(16);
  105.     } else {
  106.         ADDPX_DSP(8);
  107.     }
  108.  
  109. #define H264_DSP(depth) \
  110.     c->h264_idct_add= FUNC(ff_h264_idct_add, depth);\
  111.     c->h264_idct8_add= FUNC(ff_h264_idct8_add, depth);\
  112.     c->h264_idct_dc_add= FUNC(ff_h264_idct_dc_add, depth);\
  113.     c->h264_idct8_dc_add= FUNC(ff_h264_idct8_dc_add, depth);\
  114.     c->h264_idct_add16     = FUNC(ff_h264_idct_add16, depth);\
  115.     c->h264_idct8_add4     = FUNC(ff_h264_idct8_add4, depth);\
  116.     if (chroma_format_idc == 1)\
  117.         c->h264_idct_add8  = FUNC(ff_h264_idct_add8, depth);\
  118.     else\
  119.         c->h264_idct_add8  = FUNC(ff_h264_idct_add8_422, depth);\
  120.     c->h264_idct_add16intra= FUNC(ff_h264_idct_add16intra, depth);\
  121.     c->h264_luma_dc_dequant_idct= FUNC(ff_h264_luma_dc_dequant_idct, depth);\
  122.     if (chroma_format_idc == 1)\
  123.         c->h264_chroma_dc_dequant_idct= FUNC(ff_h264_chroma_dc_dequant_idct, depth);\
  124.     else\
  125.         c->h264_chroma_dc_dequant_idct= FUNC(ff_h264_chroma422_dc_dequant_idct, depth);\
  126. \
  127.     c->weight_h264_pixels_tab[0]= FUNC(weight_h264_pixels16, depth);\
  128.     c->weight_h264_pixels_tab[1]= FUNC(weight_h264_pixels8, depth);\
  129.     c->weight_h264_pixels_tab[2]= FUNC(weight_h264_pixels4, depth);\
  130.     c->weight_h264_pixels_tab[3]= FUNC(weight_h264_pixels2, depth);\
  131.     c->biweight_h264_pixels_tab[0]= FUNC(biweight_h264_pixels16, depth);\
  132.     c->biweight_h264_pixels_tab[1]= FUNC(biweight_h264_pixels8, depth);\
  133.     c->biweight_h264_pixels_tab[2]= FUNC(biweight_h264_pixels4, depth);\
  134.     c->biweight_h264_pixels_tab[3]= FUNC(biweight_h264_pixels2, depth);\
  135. \
  136.     c->h264_v_loop_filter_luma= FUNC(h264_v_loop_filter_luma, depth);\
  137.     c->h264_h_loop_filter_luma= FUNC(h264_h_loop_filter_luma, depth);\
  138.     c->h264_h_loop_filter_luma_mbaff= FUNC(h264_h_loop_filter_luma_mbaff, depth);\
  139.     c->h264_v_loop_filter_luma_intra= FUNC(h264_v_loop_filter_luma_intra, depth);\
  140.     c->h264_h_loop_filter_luma_intra= FUNC(h264_h_loop_filter_luma_intra, depth);\
  141.     c->h264_h_loop_filter_luma_mbaff_intra= FUNC(h264_h_loop_filter_luma_mbaff_intra, depth);\
  142.     c->h264_v_loop_filter_chroma= FUNC(h264_v_loop_filter_chroma, depth);\
  143.     if (chroma_format_idc == 1)\
  144.         c->h264_h_loop_filter_chroma= FUNC(h264_h_loop_filter_chroma, depth);\
  145.     else\
  146.         c->h264_h_loop_filter_chroma= FUNC(h264_h_loop_filter_chroma422, depth);\
  147.     if (chroma_format_idc == 1)\
  148.         c->h264_h_loop_filter_chroma_mbaff= FUNC(h264_h_loop_filter_chroma_mbaff, depth);\
  149.     else\
  150.         c->h264_h_loop_filter_chroma_mbaff= FUNC(h264_h_loop_filter_chroma422_mbaff, depth);\
  151.     c->h264_v_loop_filter_chroma_intra= FUNC(h264_v_loop_filter_chroma_intra, depth);\
  152.     if (chroma_format_idc == 1)\
  153.         c->h264_h_loop_filter_chroma_intra= FUNC(h264_h_loop_filter_chroma_intra, depth);\
  154.     else\
  155.         c->h264_h_loop_filter_chroma_intra= FUNC(h264_h_loop_filter_chroma422_intra, depth);\
  156.     if (chroma_format_idc == 1)\
  157.         c->h264_h_loop_filter_chroma_mbaff_intra= FUNC(h264_h_loop_filter_chroma_mbaff_intra, depth);\
  158.     else\
  159.         c->h264_h_loop_filter_chroma_mbaff_intra= FUNC(h264_h_loop_filter_chroma422_mbaff_intra, depth);\
  160.     c->h264_loop_filter_strength= NULL;
  161.  
  162.     switch (bit_depth) {
  163.     case 9:
  164.         H264_DSP(9);
  165.         break;
  166.     case 10:
  167.         H264_DSP(10);
  168.         break;
  169.     case 12:
  170.         H264_DSP(12);
  171.         break;
  172.     case 14:
  173.         H264_DSP(14);
  174.         break;
  175.     default:
  176.         av_assert0(bit_depth<=8);
  177.         H264_DSP(8);
  178.         break;
  179.     }
  180.     c->h264_find_start_code_candidate = h264_find_start_code_candidate_c;
  181.  
  182.     if (ARCH_ARM) ff_h264dsp_init_arm(c, bit_depth, chroma_format_idc);
  183.     if (ARCH_PPC) ff_h264dsp_init_ppc(c, bit_depth, chroma_format_idc);
  184.     if (ARCH_X86) ff_h264dsp_init_x86(c, bit_depth, chroma_format_idc);
  185. }
  186.