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

   1 /*
   2  * Copyright (c) 2000, 2001 Fabrice Bellard
   3  * Copyright (c) 2002-2004 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 #include "libavutil/avassert.h"
  23
  24 #include "bit_depth_template.c"
  25
  26 #define H264_CHROMA_MC(OPNAME, OP)\
  27 static void FUNCC(OPNAME ## h264_chroma_mc1)(uint8_t *_dst/*align 8*/, uint8_t *_src/*align 1*/, int stride, int h, int x, int y){\
  28     pixel *dst = (pixel*)_dst;\
  29     pixel *src = (pixel*)_src;\
  30     const int A=(8-x)*(8-y);\
  31     const int B=(  x)*(8-y);\
  32     const int C=(8-x)*(  y);\
  33     const int D=(  x)*(  y);\
  34     int i;\
  35     stride >>= sizeof(pixel)-1;\
  36     \
  37     av_assert2(x<8 && y<8 && x>=0 && y>=0);\
  38 \
  39     if(D){\
  40         for(i=0; i<h; i++){\
  41             OP(dst[0], (A*src[0] + B*src[1] + C*src[stride+0] + D*src[stride+1]));\
  42             dst+= stride;\
  43             src+= stride;\
  44         }\
  45     }else{\
  46         const int E= B+C;\
  47         const int step= C ? stride : 1;\
  48         for(i=0; i<h; i++){\
  49             OP(dst[0], (A*src[0] + E*src[step+0]));\
  50             dst+= stride;\
  51             src+= stride;\
  52         }\
  53     }\
  54 }\
  55 static void FUNCC(OPNAME ## h264_chroma_mc2)(uint8_t *_dst/*align 8*/, uint8_t *_src/*align 1*/, int stride, int h, int x, int y){\
  56     pixel *dst = (pixel*)_dst;\
  57     pixel *src = (pixel*)_src;\
  58     const int A=(8-x)*(8-y);\
  59     const int B=(  x)*(8-y);\
  60     const int C=(8-x)*(  y);\
  61     const int D=(  x)*(  y);\
  62     int i;\
  63     stride >>= sizeof(pixel)-1;\
  64     \
  65     av_assert2(x<8 && y<8 && x>=0 && y>=0);\
  66 \
  67     if(D){\
  68         for(i=0; i<h; i++){\
  69             OP(dst[0], (A*src[0] + B*src[1] + C*src[stride+0] + D*src[stride+1]));\
  70             OP(dst[1], (A*src[1] + B*src[2] + C*src[stride+1] + D*src[stride+2]));\
  71             dst+= stride;\
  72             src+= stride;\
  73         }\
  74     }else{\
  75         const int E= B+C;\
  76         const int step= C ? stride : 1;\
  77         for(i=0; i<h; i++){\
  78             OP(dst[0], (A*src[0] + E*src[step+0]));\
  79             OP(dst[1], (A*src[1] + E*src[step+1]));\
  80             dst+= stride;\
  81             src+= stride;\
  82         }\
  83     }\
  84 }\
  85 \
  86 static void FUNCC(OPNAME ## h264_chroma_mc4)(uint8_t *_dst/*align 8*/, uint8_t *_src/*align 1*/, int stride, int h, int x, int y){\
  87     pixel *dst = (pixel*)_dst;\
  88     pixel *src = (pixel*)_src;\
  89     const int A=(8-x)*(8-y);\
  90     const int B=(  x)*(8-y);\
  91     const int C=(8-x)*(  y);\
  92     const int D=(  x)*(  y);\
  93     int i;\
  94     stride >>= sizeof(pixel)-1;\
  95     \
  96     av_assert2(x<8 && y<8 && x>=0 && y>=0);\
  97 \
  98     if(D){\
  99         for(i=0; i<h; i++){\
 100             OP(dst[0], (A*src[0] + B*src[1] + C*src[stride+0] + D*src[stride+1]));\
 101             OP(dst[1], (A*src[1] + B*src[2] + C*src[stride+1] + D*src[stride+2]));\
 102             OP(dst[2], (A*src[2] + B*src[3] + C*src[stride+2] + D*src[stride+3]));\
 103             OP(dst[3], (A*src[3] + B*src[4] + C*src[stride+3] + D*src[stride+4]));\
 104             dst+= stride;\
 105             src+= stride;\
 106         }\
 107     }else{\
 108         const int E= B+C;\
 109         const int step= C ? stride : 1;\
 110         for(i=0; i<h; i++){\
 111             OP(dst[0], (A*src[0] + E*src[step+0]));\
 112             OP(dst[1], (A*src[1] + E*src[step+1]));\
 113             OP(dst[2], (A*src[2] + E*src[step+2]));\
 114             OP(dst[3], (A*src[3] + E*src[step+3]));\
 115             dst+= stride;\
 116             src+= stride;\
 117         }\
 118     }\
 119 }\
 120 \
 121 static void FUNCC(OPNAME ## h264_chroma_mc8)(uint8_t *_dst/*align 8*/, uint8_t *_src/*align 1*/, int stride, int h, int x, int y){\
 122     pixel *dst = (pixel*)_dst;\
 123     pixel *src = (pixel*)_src;\
 124     const int A=(8-x)*(8-y);\
 125     const int B=(  x)*(8-y);\
 126     const int C=(8-x)*(  y);\
 127     const int D=(  x)*(  y);\
 128     int i;\
 129     stride >>= sizeof(pixel)-1;\
 130     \
 131     av_assert2(x<8 && y<8 && x>=0 && y>=0);\
 132 \
 133     if(D){\
 134         for(i=0; i<h; i++){\
 135             OP(dst[0], (A*src[0] + B*src[1] + C*src[stride+0] + D*src[stride+1]));\
 136             OP(dst[1], (A*src[1] + B*src[2] + C*src[stride+1] + D*src[stride+2]));\
 137             OP(dst[2], (A*src[2] + B*src[3] + C*src[stride+2] + D*src[stride+3]));\
 138             OP(dst[3], (A*src[3] + B*src[4] + C*src[stride+3] + D*src[stride+4]));\
 139             OP(dst[4], (A*src[4] + B*src[5] + C*src[stride+4] + D*src[stride+5]));\
 140             OP(dst[5], (A*src[5] + B*src[6] + C*src[stride+5] + D*src[stride+6]));\
 141             OP(dst[6], (A*src[6] + B*src[7] + C*src[stride+6] + D*src[stride+7]));\
 142             OP(dst[7], (A*src[7] + B*src[8] + C*src[stride+7] + D*src[stride+8]));\
 143             dst+= stride;\
 144             src+= stride;\
 145         }\
 146     }else{\
 147         const int E= B+C;\
 148         const int step= C ? stride : 1;\
 149         for(i=0; i<h; i++){\
 150             OP(dst[0], (A*src[0] + E*src[step+0]));\
 151             OP(dst[1], (A*src[1] + E*src[step+1]));\
 152             OP(dst[2], (A*src[2] + E*src[step+2]));\
 153             OP(dst[3], (A*src[3] + E*src[step+3]));\
 154             OP(dst[4], (A*src[4] + E*src[step+4]));\
 155             OP(dst[5], (A*src[5] + E*src[step+5]));\
 156             OP(dst[6], (A*src[6] + E*src[step+6]));\
 157             OP(dst[7], (A*src[7] + E*src[step+7]));\
 158             dst+= stride;\
 159             src+= stride;\
 160         }\
 161     }\
 162 }
 163
 164 #define op_avg(a, b) a = (((a)+(((b) + 32)>>6)+1)>>1)
 165 #define op_put(a, b) a = (((b) + 32)>>6)
 166
 167 H264_CHROMA_MC(put_       , op_put)
 168 H264_CHROMA_MC(avg_       , op_avg)
 169 #undef op_avg
 170 #undef op_put