/*
  * thirdpel DSP functions
  *

  * This file is part of FFmpeg.

  *

  * FFmpeg is free software; you can redistribute it and/or

  * modify it under the terms of the GNU Lesser General Public
  * License as published by the Free Software Foundation; either
  * version 2.1 of the License, or (at your option) any later version.
  *

  * FFmpeg is distributed in the hope that it will be useful,

  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
  * Lesser General Public License for more details.
  *
  * You should have received a copy of the GNU Lesser General Public

  * License along with FFmpeg; if not, write to the Free Software

  * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
  */
 
 /**
  * @file
  * thirdpel DSP functions
  */
 
 #include <stdint.h>
 
 #include "libavutil/attributes.h"
 #include "tpeldsp.h"
 
 #define BIT_DEPTH 8

 #include "pel_template.c"

 
 static inline void put_tpel_pixels_mc00_c(uint8_t *dst, const uint8_t *src,
                                           int stride, int width, int height)
 {
     switch (width) {
     case 2:
         put_pixels2_8_c(dst, src, stride, height);
         break;
     case 4:
         put_pixels4_8_c(dst, src, stride, height);
         break;
     case 8:
         put_pixels8_8_c(dst, src, stride, height);
         break;
     case 16:
         put_pixels16_8_c(dst, src, stride, height);
         break;
     }
 }
 
 static inline void put_tpel_pixels_mc10_c(uint8_t *dst, const uint8_t *src,
                                           int stride, int width, int height)
 {
     int i, j;
 
     for (i = 0; i < height; i++) {
         for (j = 0; j < width; j++)
             dst[j] = ((2 * src[j] + src[j + 1] + 1) *
                       683) >> 11;
         src += stride;
         dst += stride;
     }
 }
 
 static inline void put_tpel_pixels_mc20_c(uint8_t *dst, const uint8_t *src,
                                           int stride, int width, int height)
 {
     int i, j;
 
     for (i = 0; i < height; i++) {
         for (j = 0; j < width; j++)
             dst[j] = ((src[j] + 2 * src[j + 1] + 1) *
                       683) >> 11;
         src += stride;
         dst += stride;
     }
 }
 
 static inline void put_tpel_pixels_mc01_c(uint8_t *dst, const uint8_t *src,
                                           int stride, int width, int height)
 {
     int i, j;
 
     for (i = 0; i < height; i++) {
         for (j = 0; j < width; j++)
             dst[j] = ((2 * src[j] + src[j + stride] + 1) *
                       683) >> 11;
         src += stride;
         dst += stride;
     }
 }
 
 static inline void put_tpel_pixels_mc11_c(uint8_t *dst, const uint8_t *src,
                                           int stride, int width, int height)
 {
     int i, j;
 
     for (i = 0; i < height; i++) {
         for (j = 0; j < width; j++)
             dst[j] = ((4 * src[j]          + 3 * src[j + 1] +
                        3 * src[j + stride] + 2 * src[j + stride + 1] + 6) *
                       2731) >> 15;
         src += stride;
         dst += stride;
     }
 }
 
 static inline void put_tpel_pixels_mc12_c(uint8_t *dst, const uint8_t *src,
                                           int stride, int width, int height)
 {
     int i, j;
 
     for (i = 0; i < height; i++) {
         for (j = 0; j < width; j++)
             dst[j] = ((3 * src[j]          + 2 * src[j + 1] +
                        4 * src[j + stride] + 3 * src[j + stride + 1] + 6) *
                       2731) >> 15;
         src += stride;
         dst += stride;
     }
 }
 
 static inline void put_tpel_pixels_mc02_c(uint8_t *dst, const uint8_t *src,
                                           int stride, int width, int height)
 {
     int i, j;
 
     for (i = 0; i < height; i++) {
         for (j = 0; j < width; j++)
             dst[j] = ((src[j] + 2 * src[j + stride] + 1) *
                       683) >> 11;
         src += stride;
         dst += stride;
     }
 }
 
 static inline void put_tpel_pixels_mc21_c(uint8_t *dst, const uint8_t *src,
                                           int stride, int width, int height)
 {
     int i, j;
 
     for (i = 0; i < height; i++) {
         for (j = 0; j < width; j++)
             dst[j] = ((3 * src[j]          + 4 * src[j + 1] +
                        2 * src[j + stride] + 3 * src[j + stride + 1] + 6) *
                       2731) >> 15;
         src += stride;
         dst += stride;
     }
 }
 
 static inline void put_tpel_pixels_mc22_c(uint8_t *dst, const uint8_t *src,
                                           int stride, int width, int height)
 {
     int i, j;
 
     for (i = 0; i < height; i++) {
         for (j = 0; j < width; j++)
             dst[j] = ((2 * src[j]          + 3 * src[j + 1] +
                        3 * src[j + stride] + 4 * src[j + stride + 1] + 6) *
                       2731) >> 15;
         src += stride;
         dst += stride;
     }
 }
 
 static inline void avg_tpel_pixels_mc00_c(uint8_t *dst, const uint8_t *src,
                                           int stride, int width, int height)
 {
     switch (width) {
     case 2:
         avg_pixels2_8_c(dst, src, stride, height);
         break;
     case 4:
         avg_pixels4_8_c(dst, src, stride, height);
         break;
     case 8:
         avg_pixels8_8_c(dst, src, stride, height);
         break;
     case 16:
         avg_pixels16_8_c(dst, src, stride, height);
         break;
     }
 }
 
 static inline void avg_tpel_pixels_mc10_c(uint8_t *dst, const uint8_t *src,
                                           int stride, int width, int height)
 {
     int i, j;
 
     for (i = 0; i < height; i++) {
         for (j = 0; j < width; j++)
             dst[j] = (dst[j] +
                       (((2 * src[j] + src[j + 1] + 1) *
                         683) >> 11) + 1) >> 1;
         src += stride;
         dst += stride;
     }
 }
 
 static inline void avg_tpel_pixels_mc20_c(uint8_t *dst, const uint8_t *src,
                                           int stride, int width, int height)
 {
     int i, j;
 
     for (i = 0; i < height; i++) {
         for (j = 0; j < width; j++)
             dst[j] = (dst[j] +
                       (((src[j] + 2 * src[j + 1] + 1) *
                         683) >> 11) + 1) >> 1;
         src += stride;
         dst += stride;
     }
 }
 
 static inline void avg_tpel_pixels_mc01_c(uint8_t *dst, const uint8_t *src,
                                           int stride, int width, int height)
 {
     int i, j;
 
     for (i = 0; i < height; i++) {
         for (j = 0; j < width; j++)
             dst[j] = (dst[j] +
                       (((2 * src[j] + src[j + stride] + 1) *
                         683) >> 11) + 1) >> 1;
         src += stride;
         dst += stride;
     }
 }
 
 static inline void avg_tpel_pixels_mc11_c(uint8_t *dst, const uint8_t *src,
                                           int stride, int width, int height)
 {
     int i, j;
 
     for (i = 0; i < height; i++) {
         for (j = 0; j < width; j++)
             dst[j] = (dst[j] +
                       (((4 * src[j]          + 3 * src[j + 1] +
                          3 * src[j + stride] + 2 * src[j + stride + 1] + 6) *
                         2731) >> 15) + 1) >> 1;
         src += stride;
         dst += stride;
     }
 }
 
 static inline void avg_tpel_pixels_mc12_c(uint8_t *dst, const uint8_t *src,
                                           int stride, int width, int height)
 {
     int i, j;
 
     for (i = 0; i < height; i++) {
         for (j = 0; j < width; j++)
             dst[j] = (dst[j] +
                       (((3 * src[j]          + 2 * src[j + 1] +
                          4 * src[j + stride] + 3 * src[j + stride + 1] + 6) *
                         2731) >> 15) + 1) >> 1;
         src += stride;
         dst += stride;
     }
 }
 
 static inline void avg_tpel_pixels_mc02_c(uint8_t *dst, const uint8_t *src,
                                           int stride, int width, int height)
 {
     int i, j;
 
     for (i = 0; i < height; i++) {
         for (j = 0; j < width; j++)
             dst[j] = (dst[j] +
                       (((src[j] + 2 * src[j + stride] + 1) *
                         683) >> 11) + 1) >> 1;
         src += stride;
         dst += stride;
     }
 }
 
 static inline void avg_tpel_pixels_mc21_c(uint8_t *dst, const uint8_t *src,
                                           int stride, int width, int height)
 {
     int i, j;
 
     for (i = 0; i < height; i++) {
         for (j = 0; j < width; j++)
             dst[j] = (dst[j] +
                       (((3 * src[j]          + 4 * src[j + 1] +
                          2 * src[j + stride] + 3 * src[j + stride + 1] + 6) *
                         2731) >> 15) + 1) >> 1;
         src += stride;
         dst += stride;
     }
 }
 
 static inline void avg_tpel_pixels_mc22_c(uint8_t *dst, const uint8_t *src,
                                           int stride, int width, int height)
 {
     int i, j;
 
     for (i = 0; i < height; i++) {
         for (j = 0; j < width; j++)
             dst[j] = (dst[j] +
                       (((2 * src[j]          + 3 * src[j + 1] +
                          3 * src[j + stride] + 4 * src[j + stride + 1] + 6) *
                         2731) >> 15) + 1) >> 1;
         src += stride;
         dst += stride;
     }
 }
 
 av_cold void ff_tpeldsp_init(TpelDSPContext *c)
 {
     c->put_tpel_pixels_tab[ 0] = put_tpel_pixels_mc00_c;
     c->put_tpel_pixels_tab[ 1] = put_tpel_pixels_mc10_c;
     c->put_tpel_pixels_tab[ 2] = put_tpel_pixels_mc20_c;
     c->put_tpel_pixels_tab[ 4] = put_tpel_pixels_mc01_c;
     c->put_tpel_pixels_tab[ 5] = put_tpel_pixels_mc11_c;
     c->put_tpel_pixels_tab[ 6] = put_tpel_pixels_mc21_c;
     c->put_tpel_pixels_tab[ 8] = put_tpel_pixels_mc02_c;
     c->put_tpel_pixels_tab[ 9] = put_tpel_pixels_mc12_c;
     c->put_tpel_pixels_tab[10] = put_tpel_pixels_mc22_c;
 
     c->avg_tpel_pixels_tab[ 0] = avg_tpel_pixels_mc00_c;
     c->avg_tpel_pixels_tab[ 1] = avg_tpel_pixels_mc10_c;
     c->avg_tpel_pixels_tab[ 2] = avg_tpel_pixels_mc20_c;
     c->avg_tpel_pixels_tab[ 4] = avg_tpel_pixels_mc01_c;
     c->avg_tpel_pixels_tab[ 5] = avg_tpel_pixels_mc11_c;
     c->avg_tpel_pixels_tab[ 6] = avg_tpel_pixels_mc21_c;
     c->avg_tpel_pixels_tab[ 8] = avg_tpel_pixels_mc02_c;
     c->avg_tpel_pixels_tab[ 9] = avg_tpel_pixels_mc12_c;
     c->avg_tpel_pixels_tab[10] = avg_tpel_pixels_mc22_c;
 }