h264pred.c
上传用户:jylinhe
上传日期:2022-07-11
资源大小:334k
文件大小:40k
源码类别:

多媒体编程

开发平台:

Visual 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.  * @file h264pred.c
  23.  * H.264 / AVC / MPEG4 part10 prediction functions.
  24.  * @author Michael Niedermayer <michaelni@gmx.at>
  25.  */
  26. #include "dsputil.h"
  27. #include "mpegvideo.h"
  28. #include "h264pred.h"
  29. static void pred4x4_vertical_c(uint8_t *src, uint8_t *topright, int stride){
  30.     const uint32_t a= ((uint32_t*)(src-stride))[0];
  31.     ((uint32_t*)(src+0*stride))[0]= a;
  32.     ((uint32_t*)(src+1*stride))[0]= a;
  33.     ((uint32_t*)(src+2*stride))[0]= a;
  34.     ((uint32_t*)(src+3*stride))[0]= a;
  35. }
  36. static void pred4x4_horizontal_c(uint8_t *src, uint8_t *topright, int stride){
  37.     ((uint32_t*)(src+0*stride))[0]= src[-1+0*stride]*0x01010101;
  38.     ((uint32_t*)(src+1*stride))[0]= src[-1+1*stride]*0x01010101;
  39.     ((uint32_t*)(src+2*stride))[0]= src[-1+2*stride]*0x01010101;
  40.     ((uint32_t*)(src+3*stride))[0]= src[-1+3*stride]*0x01010101;
  41. }
  42. static void pred4x4_dc_c(uint8_t *src, uint8_t *topright, int stride){
  43.     const int dc= (  src[-stride] + src[1-stride] + src[2-stride] + src[3-stride]
  44.                    + src[-1+0*stride] + src[-1+1*stride] + src[-1+2*stride] + src[-1+3*stride] + 4) >>3;
  45.     ((uint32_t*)(src+0*stride))[0]=
  46.     ((uint32_t*)(src+1*stride))[0]=
  47.     ((uint32_t*)(src+2*stride))[0]=
  48.     ((uint32_t*)(src+3*stride))[0]= dc* 0x01010101;
  49. }
  50. static void pred4x4_left_dc_c(uint8_t *src, uint8_t *topright, int stride){
  51.     const int dc= (  src[-1+0*stride] + src[-1+1*stride] + src[-1+2*stride] + src[-1+3*stride] + 2) >>2;
  52.     ((uint32_t*)(src+0*stride))[0]=
  53.     ((uint32_t*)(src+1*stride))[0]=
  54.     ((uint32_t*)(src+2*stride))[0]=
  55.     ((uint32_t*)(src+3*stride))[0]= dc* 0x01010101;
  56. }
  57. static void pred4x4_top_dc_c(uint8_t *src, uint8_t *topright, int stride){
  58.     const int dc= (  src[-stride] + src[1-stride] + src[2-stride] + src[3-stride] + 2) >>2;
  59.     ((uint32_t*)(src+0*stride))[0]=
  60.     ((uint32_t*)(src+1*stride))[0]=
  61.     ((uint32_t*)(src+2*stride))[0]=
  62.     ((uint32_t*)(src+3*stride))[0]= dc* 0x01010101;
  63. }
  64. static void pred4x4_128_dc_c(uint8_t *src, uint8_t *topright, int stride){
  65.     ((uint32_t*)(src+0*stride))[0]=
  66.     ((uint32_t*)(src+1*stride))[0]=
  67.     ((uint32_t*)(src+2*stride))[0]=
  68.     ((uint32_t*)(src+3*stride))[0]= 128U*0x01010101U;
  69. }
  70. #define LOAD_TOP_RIGHT_EDGE
  71.      int av_unused t4= topright[0];
  72.      int av_unused t5= topright[1];
  73.      int av_unused t6= topright[2];
  74.      int av_unused t7= topright[3];
  75. #define LOAD_DOWN_LEFT_EDGE
  76.      int av_unused l4= src[-1+4*stride];
  77.      int av_unused l5= src[-1+5*stride];
  78.      int av_unused l6= src[-1+6*stride];
  79.      int av_unused l7= src[-1+7*stride];
  80. #define LOAD_LEFT_EDGE
  81.      int av_unused l0= src[-1+0*stride];
  82.      int av_unused l1= src[-1+1*stride];
  83.      int av_unused l2= src[-1+2*stride];
  84.      int av_unused l3= src[-1+3*stride];
  85. #define LOAD_TOP_EDGE
  86.      int av_unused t0= src[ 0-1*stride];
  87.      int av_unused t1= src[ 1-1*stride];
  88.      int av_unused t2= src[ 2-1*stride];
  89.      int av_unused t3= src[ 3-1*stride];
  90. static void pred4x4_down_right_c(uint8_t *src, uint8_t *topright, int stride){
  91.     const int lt= src[-1-1*stride];
  92.     LOAD_TOP_EDGE
  93.     LOAD_LEFT_EDGE
  94.     src[0+3*stride]=(l3 + 2*l2 + l1 + 2)>>2;
  95.     src[0+2*stride]=
  96.     src[1+3*stride]=(l2 + 2*l1 + l0 + 2)>>2;
  97.     src[0+1*stride]=
  98.     src[1+2*stride]=
  99.     src[2+3*stride]=(l1 + 2*l0 + lt + 2)>>2;
  100.     src[0+0*stride]=
  101.     src[1+1*stride]=
  102.     src[2+2*stride]=
  103.     src[3+3*stride]=(l0 + 2*lt + t0 + 2)>>2;
  104.     src[1+0*stride]=
  105.     src[2+1*stride]=
  106.     src[3+2*stride]=(lt + 2*t0 + t1 + 2)>>2;
  107.     src[2+0*stride]=
  108.     src[3+1*stride]=(t0 + 2*t1 + t2 + 2)>>2;
  109.     src[3+0*stride]=(t1 + 2*t2 + t3 + 2)>>2;
  110. }
  111. static void pred4x4_down_left_c(uint8_t *src, uint8_t *topright, int stride){
  112.     LOAD_TOP_EDGE
  113.     LOAD_TOP_RIGHT_EDGE
  114. //    LOAD_LEFT_EDGE
  115.     src[0+0*stride]=(t0 + t2 + 2*t1 + 2)>>2;
  116.     src[1+0*stride]=
  117.     src[0+1*stride]=(t1 + t3 + 2*t2 + 2)>>2;
  118.     src[2+0*stride]=
  119.     src[1+1*stride]=
  120.     src[0+2*stride]=(t2 + t4 + 2*t3 + 2)>>2;
  121.     src[3+0*stride]=
  122.     src[2+1*stride]=
  123.     src[1+2*stride]=
  124.     src[0+3*stride]=(t3 + t5 + 2*t4 + 2)>>2;
  125.     src[3+1*stride]=
  126.     src[2+2*stride]=
  127.     src[1+3*stride]=(t4 + t6 + 2*t5 + 2)>>2;
  128.     src[3+2*stride]=
  129.     src[2+3*stride]=(t5 + t7 + 2*t6 + 2)>>2;
  130.     src[3+3*stride]=(t6 + 3*t7 + 2)>>2;
  131. }
  132. static void pred4x4_down_left_svq3_c(uint8_t *src, uint8_t *topright, int stride){
  133.     LOAD_TOP_EDGE
  134.     LOAD_LEFT_EDGE
  135.     //const av_unused int unu0= t0;
  136.     //const av_unused int unu1= l0;
  137. t0 = 0; l0 = 0;
  138.     src[0+0*stride]=(l1 + t1)>>1;
  139.     src[1+0*stride]=
  140.     src[0+1*stride]=(l2 + t2)>>1;
  141.     src[2+0*stride]=
  142.     src[1+1*stride]=
  143.     src[0+2*stride]=
  144.     src[3+0*stride]=
  145.     src[2+1*stride]=
  146.     src[1+2*stride]=
  147.     src[0+3*stride]=
  148.     src[3+1*stride]=
  149.     src[2+2*stride]=
  150.     src[1+3*stride]=
  151.     src[3+2*stride]=
  152.     src[2+3*stride]=
  153.     src[3+3*stride]=(l3 + t3)>>1;
  154. }
  155. static void pred4x4_down_left_rv40_c(uint8_t *src, uint8_t *topright, int stride){
  156.     LOAD_TOP_EDGE
  157.     LOAD_TOP_RIGHT_EDGE
  158.     LOAD_LEFT_EDGE
  159.     LOAD_DOWN_LEFT_EDGE
  160.     src[0+0*stride]=(t0 + t2 + 2*t1 + 2 + l0 + l2 + 2*l1 + 2)>>3;
  161.     src[1+0*stride]=
  162.     src[0+1*stride]=(t1 + t3 + 2*t2 + 2 + l1 + l3 + 2*l2 + 2)>>3;
  163.     src[2+0*stride]=
  164.     src[1+1*stride]=
  165.     src[0+2*stride]=(t2 + t4 + 2*t3 + 2 + l2 + l4 + 2*l3 + 2)>>3;
  166.     src[3+0*stride]=
  167.     src[2+1*stride]=
  168.     src[1+2*stride]=
  169.     src[0+3*stride]=(t3 + t5 + 2*t4 + 2 + l3 + l5 + 2*l4 + 2)>>3;
  170.     src[3+1*stride]=
  171.     src[2+2*stride]=
  172.     src[1+3*stride]=(t4 + t6 + 2*t5 + 2 + l4 + l6 + 2*l5 + 2)>>3;
  173.     src[3+2*stride]=
  174.     src[2+3*stride]=(t5 + t7 + 2*t6 + 2 + l5 + l7 + 2*l6 + 2)>>3;
  175.     src[3+3*stride]=(t6 + t7 + 1 + l6 + l7 + 1)>>2;
  176. }
  177. static void pred4x4_down_left_rv40_nodown_c(uint8_t *src, uint8_t *topright, int stride){
  178.     LOAD_TOP_EDGE
  179.     LOAD_TOP_RIGHT_EDGE
  180.     LOAD_LEFT_EDGE
  181.     src[0+0*stride]=(t0 + t2 + 2*t1 + 2 + l0 + l2 + 2*l1 + 2)>>3;
  182.     src[1+0*stride]=
  183.     src[0+1*stride]=(t1 + t3 + 2*t2 + 2 + l1 + l3 + 2*l2 + 2)>>3;
  184.     src[2+0*stride]=
  185.     src[1+1*stride]=
  186.     src[0+2*stride]=(t2 + t4 + 2*t3 + 2 + l2 + 3*l3 + 2)>>3;
  187.     src[3+0*stride]=
  188.     src[2+1*stride]=
  189.     src[1+2*stride]=
  190.     src[0+3*stride]=(t3 + t5 + 2*t4 + 2 + l3*4 + 2)>>3;
  191.     src[3+1*stride]=
  192.     src[2+2*stride]=
  193.     src[1+3*stride]=(t4 + t6 + 2*t5 + 2 + l3*4 + 2)>>3;
  194.     src[3+2*stride]=
  195.     src[2+3*stride]=(t5 + t7 + 2*t6 + 2 + l3*4 + 2)>>3;
  196.     src[3+3*stride]=(t6 + t7 + 1 + 2*l3 + 1)>>2;
  197. }
  198. static void pred4x4_vertical_right_c(uint8_t *src, uint8_t *topright, int stride){
  199.     const int lt= src[-1-1*stride];
  200.     LOAD_TOP_EDGE
  201.     LOAD_LEFT_EDGE
  202. l3=0;
  203.     src[0+0*stride]=
  204.     src[1+2*stride]=(lt + t0 + 1)>>1;
  205.     src[1+0*stride]=
  206.     src[2+2*stride]=(t0 + t1 + 1)>>1;
  207.     src[2+0*stride]=
  208.     src[3+2*stride]=(t1 + t2 + 1)>>1;
  209.     src[3+0*stride]=(t2 + t3 + 1)>>1;
  210.     src[0+1*stride]=
  211.     src[1+3*stride]=(l0 + 2*lt + t0 + 2)>>2;
  212.     src[1+1*stride]=
  213.     src[2+3*stride]=(lt + 2*t0 + t1 + 2)>>2;
  214.     src[2+1*stride]=
  215.     src[3+3*stride]=(t0 + 2*t1 + t2 + 2)>>2;
  216.     src[3+1*stride]=(t1 + 2*t2 + t3 + 2)>>2;
  217.     src[0+2*stride]=(lt + 2*l0 + l1 + 2)>>2;
  218.     src[0+3*stride]=(l0 + 2*l1 + l2 + 2)>>2;
  219. }
  220. static void pred4x4_vertical_left_c(uint8_t *src, uint8_t *topright, int stride){
  221.     LOAD_TOP_EDGE
  222.     LOAD_TOP_RIGHT_EDGE
  223. t7=0;
  224.     src[0+0*stride]=(t0 + t1 + 1)>>1;
  225.     src[1+0*stride]=
  226.     src[0+2*stride]=(t1 + t2 + 1)>>1;
  227.     src[2+0*stride]=
  228.     src[1+2*stride]=(t2 + t3 + 1)>>1;
  229.     src[3+0*stride]=
  230.     src[2+2*stride]=(t3 + t4+ 1)>>1;
  231.     src[3+2*stride]=(t4 + t5+ 1)>>1;
  232.     src[0+1*stride]=(t0 + 2*t1 + t2 + 2)>>2;
  233.     src[1+1*stride]=
  234.     src[0+3*stride]=(t1 + 2*t2 + t3 + 2)>>2;
  235.     src[2+1*stride]=
  236.     src[1+3*stride]=(t2 + 2*t3 + t4 + 2)>>2;
  237.     src[3+1*stride]=
  238.     src[2+3*stride]=(t3 + 2*t4 + t5 + 2)>>2;
  239.     src[3+3*stride]=(t4 + 2*t5 + t6 + 2)>>2;
  240. }
  241. static void pred4x4_vertical_left_rv40(uint8_t *src, uint8_t *topright, int stride,
  242.                                       const int l0, const int l1, const int l2, const int l3, const int l4){
  243.     LOAD_TOP_EDGE
  244.     LOAD_TOP_RIGHT_EDGE
  245. t7=0;
  246.     src[0+0*stride]=(2*t0 + 2*t1 + l1 + 2*l2 + l3 + 4)>>3;
  247.     src[1+0*stride]=
  248.     src[0+2*stride]=(t1 + t2 + 1)>>1;
  249.     src[2+0*stride]=
  250.     src[1+2*stride]=(t2 + t3 + 1)>>1;
  251.     src[3+0*stride]=
  252.     src[2+2*stride]=(t3 + t4+ 1)>>1;
  253.     src[3+2*stride]=(t4 + t5+ 1)>>1;
  254.     src[0+1*stride]=(t0 + 2*t1 + t2 + l2 + 2*l3 + l4 + 4)>>3;
  255.     src[1+1*stride]=
  256.     src[0+3*stride]=(t1 + 2*t2 + t3 + 2)>>2;
  257.     src[2+1*stride]=
  258.     src[1+3*stride]=(t2 + 2*t3 + t4 + 2)>>2;
  259.     src[3+1*stride]=
  260.     src[2+3*stride]=(t3 + 2*t4 + t5 + 2)>>2;
  261.     src[3+3*stride]=(t4 + 2*t5 + t6 + 2)>>2;
  262. }
  263. static void pred4x4_vertical_left_rv40_c(uint8_t *src, uint8_t *topright, int stride){
  264.     LOAD_LEFT_EDGE
  265.     LOAD_DOWN_LEFT_EDGE
  266. l5=0;l6=0;l7=0;
  267.     pred4x4_vertical_left_rv40(src, topright, stride, l0, l1, l2, l3, l4);
  268. }
  269. static void pred4x4_vertical_left_rv40_nodown_c(uint8_t *src, uint8_t *topright, int stride){
  270.     LOAD_LEFT_EDGE
  271.     pred4x4_vertical_left_rv40(src, topright, stride, l0, l1, l2, l3, l3);
  272. }
  273. static void pred4x4_horizontal_up_c(uint8_t *src, uint8_t *topright, int stride){
  274.     LOAD_LEFT_EDGE
  275.     src[0+0*stride]=(l0 + l1 + 1)>>1;
  276.     src[1+0*stride]=(l0 + 2*l1 + l2 + 2)>>2;
  277.     src[2+0*stride]=
  278.     src[0+1*stride]=(l1 + l2 + 1)>>1;
  279.     src[3+0*stride]=
  280.     src[1+1*stride]=(l1 + 2*l2 + l3 + 2)>>2;
  281.     src[2+1*stride]=
  282.     src[0+2*stride]=(l2 + l3 + 1)>>1;
  283.     src[3+1*stride]=
  284.     src[1+2*stride]=(l2 + 2*l3 + l3 + 2)>>2;
  285.     src[3+2*stride]=
  286.     src[1+3*stride]=
  287.     src[0+3*stride]=
  288.     src[2+2*stride]=
  289.     src[2+3*stride]=
  290.     src[3+3*stride]=l3;
  291. }
  292. static void pred4x4_horizontal_up_rv40_c(uint8_t *src, uint8_t *topright, int stride){
  293.     LOAD_LEFT_EDGE
  294.     LOAD_DOWN_LEFT_EDGE
  295.     LOAD_TOP_EDGE
  296.     LOAD_TOP_RIGHT_EDGE
  297. l7=0; t0=0;
  298.     src[0+0*stride]=(t1 + 2*t2 + t3 + 2*l0 + 2*l1 + 4)>>3;
  299.     src[1+0*stride]=(t2 + 2*t3 + t4 + l0 + 2*l1 + l2 + 4)>>3;
  300.     src[2+0*stride]=
  301.     src[0+1*stride]=(t3 + 2*t4 + t5 + 2*l1 + 2*l2 + 4)>>3;
  302.     src[3+0*stride]=
  303.     src[1+1*stride]=(t4 + 2*t5 + t6 + l1 + 2*l2 + l3 + 4)>>3;
  304.     src[2+1*stride]=
  305.     src[0+2*stride]=(t5 + 2*t6 + t7 + 2*l2 + 2*l3 + 4)>>3;
  306.     src[3+1*stride]=
  307.     src[1+2*stride]=(t6 + 3*t7 + l2 + 3*l3 + 4)>>3;
  308.     src[3+2*stride]=
  309.     src[1+3*stride]=(l3 + 2*l4 + l5 + 2)>>2;
  310.     src[0+3*stride]=
  311.     src[2+2*stride]=(t6 + t7 + l3 + l4 + 2)>>2;
  312.     src[2+3*stride]=(l4 + l5 + 1)>>1;
  313.     src[3+3*stride]=(l4 + 2*l5 + l6 + 2)>>2;
  314. }
  315. static void pred4x4_horizontal_up_rv40_nodown_c(uint8_t *src, uint8_t *topright, int stride){
  316.     LOAD_LEFT_EDGE
  317.     LOAD_TOP_EDGE
  318.     LOAD_TOP_RIGHT_EDGE
  319. t0=0;
  320.     src[0+0*stride]=(t1 + 2*t2 + t3 + 2*l0 + 2*l1 + 4)>>3;
  321.     src[1+0*stride]=(t2 + 2*t3 + t4 + l0 + 2*l1 + l2 + 4)>>3;
  322.     src[2+0*stride]=
  323.     src[0+1*stride]=(t3 + 2*t4 + t5 + 2*l1 + 2*l2 + 4)>>3;
  324.     src[3+0*stride]=
  325.     src[1+1*stride]=(t4 + 2*t5 + t6 + l1 + 2*l2 + l3 + 4)>>3;
  326.     src[2+1*stride]=
  327.     src[0+2*stride]=(t5 + 2*t6 + t7 + 2*l2 + 2*l3 + 4)>>3;
  328.     src[3+1*stride]=
  329.     src[1+2*stride]=(t6 + 3*t7 + l2 + 3*l3 + 4)>>3;
  330.     src[3+2*stride]=
  331.     src[1+3*stride]=l3;
  332.     src[0+3*stride]=
  333.     src[2+2*stride]=(t6 + t7 + 2*l3 + 2)>>2;
  334.     src[2+3*stride]=
  335.     src[3+3*stride]=l3;
  336. }
  337. static void pred4x4_horizontal_down_c(uint8_t *src, uint8_t *topright, int stride){
  338.     const int lt= src[-1-1*stride];
  339.     LOAD_TOP_EDGE
  340.     LOAD_LEFT_EDGE
  341. t3= 0;
  342.     src[0+0*stride]=
  343.     src[2+1*stride]=(lt + l0 + 1)>>1;
  344.     src[1+0*stride]=
  345.     src[3+1*stride]=(l0 + 2*lt + t0 + 2)>>2;
  346.     src[2+0*stride]=(lt + 2*t0 + t1 + 2)>>2;
  347.     src[3+0*stride]=(t0 + 2*t1 + t2 + 2)>>2;
  348.     src[0+1*stride]=
  349.     src[2+2*stride]=(l0 + l1 + 1)>>1;
  350.     src[1+1*stride]=
  351.     src[3+2*stride]=(lt + 2*l0 + l1 + 2)>>2;
  352.     src[0+2*stride]=
  353.     src[2+3*stride]=(l1 + l2+ 1)>>1;
  354.     src[1+2*stride]=
  355.     src[3+3*stride]=(l0 + 2*l1 + l2 + 2)>>2;
  356.     src[0+3*stride]=(l2 + l3 + 1)>>1;
  357.     src[1+3*stride]=(l1 + 2*l2 + l3 + 2)>>2;
  358. }
  359. static void pred16x16_vertical_c(uint8_t *src, int stride){
  360.     int i;
  361.     const uint32_t a= ((uint32_t*)(src-stride))[0];
  362.     const uint32_t b= ((uint32_t*)(src-stride))[1];
  363.     const uint32_t c= ((uint32_t*)(src-stride))[2];
  364.     const uint32_t d= ((uint32_t*)(src-stride))[3];
  365.     for(i=0; i<16; i++){
  366.         ((uint32_t*)(src+i*stride))[0]= a;
  367.         ((uint32_t*)(src+i*stride))[1]= b;
  368.         ((uint32_t*)(src+i*stride))[2]= c;
  369.         ((uint32_t*)(src+i*stride))[3]= d;
  370.     }
  371. }
  372. static void pred16x16_horizontal_c(uint8_t *src, int stride){
  373.     int i;
  374.     for(i=0; i<16; i++){
  375.         ((uint32_t*)(src+i*stride))[0]=
  376.         ((uint32_t*)(src+i*stride))[1]=
  377.         ((uint32_t*)(src+i*stride))[2]=
  378.         ((uint32_t*)(src+i*stride))[3]= src[-1+i*stride]*0x01010101;
  379.     }
  380. }
  381. static void pred16x16_dc_c(uint8_t *src, int stride){
  382.     int i, dc=0;
  383.     for(i=0;i<16; i++){
  384.         dc+= src[-1+i*stride];
  385.     }
  386.     for(i=0;i<16; i++){
  387.         dc+= src[i-stride];
  388.     }
  389.     dc= 0x01010101*((dc + 16)>>5);
  390.     for(i=0; i<16; i++){
  391.         ((uint32_t*)(src+i*stride))[0]=
  392.         ((uint32_t*)(src+i*stride))[1]=
  393.         ((uint32_t*)(src+i*stride))[2]=
  394.         ((uint32_t*)(src+i*stride))[3]= dc;
  395.     }
  396. }
  397. static void pred16x16_left_dc_c(uint8_t *src, int stride){
  398.     int i, dc=0;
  399.     for(i=0;i<16; i++){
  400.         dc+= src[-1+i*stride];
  401.     }
  402.     dc= 0x01010101*((dc + 8)>>4);
  403.     for(i=0; i<16; i++){
  404.         ((uint32_t*)(src+i*stride))[0]=
  405.         ((uint32_t*)(src+i*stride))[1]=
  406.         ((uint32_t*)(src+i*stride))[2]=
  407.         ((uint32_t*)(src+i*stride))[3]= dc;
  408.     }
  409. }
  410. static void pred16x16_top_dc_c(uint8_t *src, int stride){
  411.     int i, dc=0;
  412.     for(i=0;i<16; i++){
  413.         dc+= src[i-stride];
  414.     }
  415.     dc= 0x01010101*((dc + 8)>>4);
  416.     for(i=0; i<16; i++){
  417.         ((uint32_t*)(src+i*stride))[0]=
  418.         ((uint32_t*)(src+i*stride))[1]=
  419.         ((uint32_t*)(src+i*stride))[2]=
  420.         ((uint32_t*)(src+i*stride))[3]= dc;
  421.     }
  422. }
  423. static void pred16x16_128_dc_c(uint8_t *src, int stride){
  424.     int i;
  425.     for(i=0; i<16; i++){
  426.         ((uint32_t*)(src+i*stride))[0]=
  427.         ((uint32_t*)(src+i*stride))[1]=
  428.         ((uint32_t*)(src+i*stride))[2]=
  429.         ((uint32_t*)(src+i*stride))[3]= 0x01010101U*128U;
  430.     }
  431. }
  432. static inline void pred16x16_plane_compat_c(uint8_t *src, int stride, const int svq3, const int rv40){
  433.   int i, j, k;
  434.   int a;
  435.   uint8_t *cm = ff_cropTbl + MAX_NEG_CROP;
  436.   const uint8_t * const src0 = src+7-stride;
  437.   const uint8_t *src1 = src+8*stride-1;
  438.   const uint8_t *src2 = src1-2*stride;      // == src+6*stride-1;
  439.   int H = src0[1] - src0[-1];
  440.   int V = src1[0] - src2[ 0];
  441.   for(k=2; k<=8; ++k) {
  442.     src1 += stride; src2 -= stride;
  443.     H += k*(src0[k] - src0[-k]);
  444.     V += k*(src1[0] - src2[ 0]);
  445.   }
  446.   if(svq3){
  447.     H = ( 5*(H/4) ) / 16;
  448.     V = ( 5*(V/4) ) / 16;
  449.     /* required for 100% accuracy */
  450.     i = H; H = V; V = i;
  451.   }else if(rv40){
  452.     H = ( H + (H>>2) ) >> 4;
  453.     V = ( V + (V>>2) ) >> 4;
  454.   }else{
  455.     H = ( 5*H+32 ) >> 6;
  456.     V = ( 5*V+32 ) >> 6;
  457.   }
  458.   a = 16*(src1[0] + src2[16] + 1) - 7*(V+H);
  459.   for(j=16; j>0; --j) {
  460.     int b = a;
  461.     a += V;
  462.     for(i=-16; i<0; i+=4) {
  463.       src[16+i] = cm[ (b    ) >> 5 ];
  464.       src[17+i] = cm[ (b+  H) >> 5 ];
  465.       src[18+i] = cm[ (b+2*H) >> 5 ];
  466.       src[19+i] = cm[ (b+3*H) >> 5 ];
  467.       b += 4*H;
  468.     }
  469.     src += stride;
  470.   }
  471. }
  472. static void pred16x16_plane_c(uint8_t *src, int stride){
  473.     pred16x16_plane_compat_c(src, stride, 0, 0);
  474. }
  475. static void pred16x16_plane_svq3_c(uint8_t *src, int stride){
  476.     pred16x16_plane_compat_c(src, stride, 1, 0);
  477. }
  478. static void pred16x16_plane_rv40_c(uint8_t *src, int stride){
  479.     pred16x16_plane_compat_c(src, stride, 0, 1);
  480. }
  481. static void pred8x8_vertical_c(uint8_t *src, int stride){
  482.     int i;
  483.     const uint32_t a= ((uint32_t*)(src-stride))[0];
  484.     const uint32_t b= ((uint32_t*)(src-stride))[1];
  485.     for(i=0; i<8; i++){
  486.         ((uint32_t*)(src+i*stride))[0]= a;
  487.         ((uint32_t*)(src+i*stride))[1]= b;
  488.     }
  489. }
  490. static void pred8x8_horizontal_c(uint8_t *src, int stride){
  491.     int i;
  492.     for(i=0; i<8; i++){
  493.         ((uint32_t*)(src+i*stride))[0]=
  494.         ((uint32_t*)(src+i*stride))[1]= src[-1+i*stride]*0x01010101;
  495.     }
  496. }
  497. static void pred8x8_128_dc_c(uint8_t *src, int stride){
  498.     int i;
  499.     for(i=0; i<8; i++){
  500.         ((uint32_t*)(src+i*stride))[0]=
  501.         ((uint32_t*)(src+i*stride))[1]= 0x01010101U*128U;
  502.     }
  503. }
  504. static void pred8x8_left_dc_c(uint8_t *src, int stride){
  505.     int i;
  506.     int dc0, dc2;
  507.     dc0=dc2=0;
  508.     for(i=0;i<4; i++){
  509.         dc0+= src[-1+i*stride];
  510.         dc2+= src[-1+(i+4)*stride];
  511.     }
  512.     dc0= 0x01010101*((dc0 + 2)>>2);
  513.     dc2= 0x01010101*((dc2 + 2)>>2);
  514.     for(i=0; i<4; i++){
  515.         ((uint32_t*)(src+i*stride))[0]=
  516.         ((uint32_t*)(src+i*stride))[1]= dc0;
  517.     }
  518.     for(i=4; i<8; i++){
  519.         ((uint32_t*)(src+i*stride))[0]=
  520.         ((uint32_t*)(src+i*stride))[1]= dc2;
  521.     }
  522. }
  523. static void pred8x8_left_dc_rv40_c(uint8_t *src, int stride){
  524.     int i;
  525.     int dc0;
  526.     dc0=0;
  527.     for(i=0;i<8; i++)
  528.         dc0+= src[-1+i*stride];
  529.     dc0= 0x01010101*((dc0 + 4)>>3);
  530.     for(i=0; i<8; i++){
  531.         ((uint32_t*)(src+i*stride))[0]=
  532.         ((uint32_t*)(src+i*stride))[1]= dc0;
  533.     }
  534. }
  535. static void pred8x8_top_dc_c(uint8_t *src, int stride){
  536.     int i;
  537.     int dc0, dc1;
  538.     dc0=dc1=0;
  539.     for(i=0;i<4; i++){
  540.         dc0+= src[i-stride];
  541.         dc1+= src[4+i-stride];
  542.     }
  543.     dc0= 0x01010101*((dc0 + 2)>>2);
  544.     dc1= 0x01010101*((dc1 + 2)>>2);
  545.     for(i=0; i<4; i++){
  546.         ((uint32_t*)(src+i*stride))[0]= dc0;
  547.         ((uint32_t*)(src+i*stride))[1]= dc1;
  548.     }
  549.     for(i=4; i<8; i++){
  550.         ((uint32_t*)(src+i*stride))[0]= dc0;
  551.         ((uint32_t*)(src+i*stride))[1]= dc1;
  552.     }
  553. }
  554. static void pred8x8_top_dc_rv40_c(uint8_t *src, int stride){
  555.     int i;
  556.     int dc0;
  557.     dc0=0;
  558.     for(i=0;i<8; i++)
  559.         dc0+= src[i-stride];
  560.     dc0= 0x01010101*((dc0 + 4)>>3);
  561.     for(i=0; i<8; i++){
  562.         ((uint32_t*)(src+i*stride))[0]=
  563.         ((uint32_t*)(src+i*stride))[1]= dc0;
  564.     }
  565. }
  566. static void pred8x8_dc_c(uint8_t *src, int stride){
  567.     int i;
  568.     int dc0, dc1, dc2, dc3;
  569.     dc0=dc1=dc2=0;
  570.     for(i=0;i<4; i++){
  571.         dc0+= src[-1+i*stride] + src[i-stride];
  572.         dc1+= src[4+i-stride];
  573.         dc2+= src[-1+(i+4)*stride];
  574.     }
  575.     dc3= 0x01010101*((dc1 + dc2 + 4)>>3);
  576.     dc0= 0x01010101*((dc0 + 4)>>3);
  577.     dc1= 0x01010101*((dc1 + 2)>>2);
  578.     dc2= 0x01010101*((dc2 + 2)>>2);
  579.     for(i=0; i<4; i++){
  580.         ((uint32_t*)(src+i*stride))[0]= dc0;
  581.         ((uint32_t*)(src+i*stride))[1]= dc1;
  582.     }
  583.     for(i=4; i<8; i++){
  584.         ((uint32_t*)(src+i*stride))[0]= dc2;
  585.         ((uint32_t*)(src+i*stride))[1]= dc3;
  586.     }
  587. }
  588. //the following 4 function should not be optimized!
  589. static void pred8x8_mad_cow_dc_l0t(uint8_t *src, int stride){
  590.     pred8x8_top_dc_c(src, stride);
  591.     pred4x4_dc_c(src, NULL, stride);
  592. }
  593. static void pred8x8_mad_cow_dc_0lt(uint8_t *src, int stride){
  594.     pred8x8_dc_c(src, stride);
  595.     pred4x4_top_dc_c(src, NULL, stride);
  596. }
  597. static void pred8x8_mad_cow_dc_l00(uint8_t *src, int stride){
  598.     pred8x8_left_dc_c(src, stride);
  599.     pred4x4_128_dc_c(src + 4*stride    , NULL, stride);
  600.     pred4x4_128_dc_c(src + 4*stride + 4, NULL, stride);
  601. }
  602. static void pred8x8_mad_cow_dc_0l0(uint8_t *src, int stride){
  603.     pred8x8_left_dc_c(src, stride);
  604.     pred4x4_128_dc_c(src    , NULL, stride);
  605.     pred4x4_128_dc_c(src + 4, NULL, stride);
  606. }
  607. static void pred8x8_dc_rv40_c(uint8_t *src, int stride){
  608.     int i;
  609.     int dc0=0;
  610.     for(i=0;i<4; i++){
  611.         dc0+= src[-1+i*stride] + src[i-stride];
  612.         dc0+= src[4+i-stride];
  613.         dc0+= src[-1+(i+4)*stride];
  614.     }
  615.     dc0= 0x01010101*((dc0 + 8)>>4);
  616.     for(i=0; i<4; i++){
  617.         ((uint32_t*)(src+i*stride))[0]= dc0;
  618.         ((uint32_t*)(src+i*stride))[1]= dc0;
  619.     }
  620.     for(i=4; i<8; i++){
  621.         ((uint32_t*)(src+i*stride))[0]= dc0;
  622.         ((uint32_t*)(src+i*stride))[1]= dc0;
  623.     }
  624. }
  625. static void pred8x8_plane_c(uint8_t *src, int stride){
  626.   int j, k;
  627.   int a;
  628.   uint8_t *cm = ff_cropTbl + MAX_NEG_CROP;
  629.   const uint8_t * const src0 = src+3-stride;
  630.   const uint8_t *src1 = src+4*stride-1;
  631.   const uint8_t *src2 = src1-2*stride;      // == src+2*stride-1;
  632.   int H = src0[1] - src0[-1];
  633.   int V = src1[0] - src2[ 0];
  634.   for(k=2; k<=4; ++k) {
  635.     src1 += stride; src2 -= stride;
  636.     H += k*(src0[k] - src0[-k]);
  637.     V += k*(src1[0] - src2[ 0]);
  638.   }
  639.   H = ( 17*H+16 ) >> 5;
  640.   V = ( 17*V+16 ) >> 5;
  641.   a = 16*(src1[0] + src2[8]+1) - 3*(V+H);
  642.   for(j=8; j>0; --j) {
  643.     int b = a;
  644.     a += V;
  645.     src[0] = cm[ (b    ) >> 5 ];
  646.     src[1] = cm[ (b+  H) >> 5 ];
  647.     src[2] = cm[ (b+2*H) >> 5 ];
  648.     src[3] = cm[ (b+3*H) >> 5 ];
  649.     src[4] = cm[ (b+4*H) >> 5 ];
  650.     src[5] = cm[ (b+5*H) >> 5 ];
  651.     src[6] = cm[ (b+6*H) >> 5 ];
  652.     src[7] = cm[ (b+7*H) >> 5 ];
  653.     src += stride;
  654.   }
  655. }
  656. #define SRC(x,y) src[(x)+(y)*stride]
  657. #define PL(y) 
  658.     const int l##y = (SRC(-1,y-1) + 2*SRC(-1,y) + SRC(-1,y+1) + 2) >> 2;
  659. #define PREDICT_8x8_LOAD_LEFT 
  660.      int l0 = ((has_topleft ? SRC(-1,-1) : SRC(-1,0)) 
  661.                      + 2*SRC(-1,0) + SRC(-1,1) + 2) >> 2; 
  662.     PL(1) PL(2) PL(3) PL(4) PL(5) PL(6) 
  663.      int l7 av_unused = (SRC(-1,6) + 3*SRC(-1,7) + 2) >> 2
  664. #define PT(x) 
  665.     const int t##x = (SRC(x-1,-1) + 2*SRC(x,-1) + SRC(x+1,-1) + 2) >> 2;
  666. #define PREDICT_8x8_LOAD_TOP 
  667.      int t0 = ((has_topleft ? SRC(-1,-1) : SRC(0,-1)) 
  668.                      + 2*SRC(0,-1) + SRC(1,-1) + 2) >> 2; 
  669.     PT(1) PT(2) PT(3) PT(4) PT(5) PT(6) 
  670.      int t7 av_unused = ((has_topright ? SRC(8,-1) : SRC(7,-1)) 
  671.                      + 2*SRC(7,-1) + SRC(6,-1) + 2) >> 2
  672. #define PTR(x) 
  673.     t##x = (SRC(x-1,-1) + 2*SRC(x,-1) + SRC(x+1,-1) + 2) >> 2;
  674. #define PREDICT_8x8_LOAD_TOPRIGHT 
  675.     int t8, t9, t10, t11, t12, t13, t14, t15; 
  676.     if(has_topright) { 
  677.         PTR(8) PTR(9) PTR(10) PTR(11) PTR(12) PTR(13) PTR(14) 
  678.         t15 = (SRC(14,-1) + 3*SRC(15,-1) + 2) >> 2; 
  679.     } else t8=t9=t10=t11=t12=t13=t14=t15= SRC(7,-1);
  680. #define PREDICT_8x8_LOAD_TOPLEFT 
  681.      int lt = (SRC(-1,0) + 2*SRC(-1,-1) + SRC(0,-1) + 2) >> 2
  682. #define PREDICT_8x8_DC(v) 
  683.     int y; 
  684.     for( y = 0; y < 8; y++ ) { 
  685.         ((uint32_t*)src)[0] = 
  686.         ((uint32_t*)src)[1] = v; 
  687.         src += stride; 
  688.     }
  689. static void pred8x8l_128_dc_c(uint8_t *src, int has_topleft, int has_topright, int stride)
  690. {
  691.     PREDICT_8x8_DC(0x80808080);
  692. }
  693. static void pred8x8l_left_dc_c(uint8_t *src, int has_topleft, int has_topright, int stride)
  694. {
  695.     PREDICT_8x8_LOAD_LEFT;
  696.     const uint32_t dc = ((l0+l1+l2+l3+l4+l5+l6+l7+4) >> 3) * 0x01010101;
  697.     PREDICT_8x8_DC(dc);
  698. }
  699. static void pred8x8l_top_dc_c(uint8_t *src, int has_topleft, int has_topright, int stride)
  700. {
  701.     PREDICT_8x8_LOAD_TOP;
  702.     const uint32_t dc = ((t0+t1+t2+t3+t4+t5+t6+t7+4) >> 3) * 0x01010101;
  703.     PREDICT_8x8_DC(dc);
  704. }
  705. static void pred8x8l_dc_c(uint8_t *src, int has_topleft, int has_topright, int stride)
  706. {
  707.     PREDICT_8x8_LOAD_LEFT;
  708.     PREDICT_8x8_LOAD_TOP;
  709.     const uint32_t dc = ((l0+l1+l2+l3+l4+l5+l6+l7
  710.                          +t0+t1+t2+t3+t4+t5+t6+t7+8) >> 4) * 0x01010101;
  711.     PREDICT_8x8_DC(dc);
  712. }
  713. static void pred8x8l_horizontal_c(uint8_t *src, int has_topleft, int has_topright, int stride)
  714. {
  715.     PREDICT_8x8_LOAD_LEFT;
  716. #define ROW(y) ((uint32_t*)(src+y*stride))[0] =
  717.                ((uint32_t*)(src+y*stride))[1] = 0x01010101 * l##y
  718.     ROW(0); ROW(1); ROW(2); ROW(3); ROW(4); ROW(5); ROW(6); ROW(7);
  719. #undef ROW
  720. }
  721. static void pred8x8l_vertical_c(uint8_t *src, int has_topleft, int has_topright, int stride)
  722. {
  723.     int y;
  724.     PREDICT_8x8_LOAD_TOP;
  725.     src[0] = t0;
  726.     src[1] = t1;
  727.     src[2] = t2;
  728.     src[3] = t3;
  729.     src[4] = t4;
  730.     src[5] = t5;
  731.     src[6] = t6;
  732.     src[7] = t7;
  733.     for( y = 1; y < 8; y++ )
  734.         *(uint64_t*)(src+y*stride) = *(uint64_t*)src;
  735. }
  736. static void pred8x8l_down_left_c(uint8_t *src, int has_topleft, int has_topright, int stride)
  737. {
  738.     PREDICT_8x8_LOAD_TOP;
  739.     PREDICT_8x8_LOAD_TOPRIGHT;
  740.     SRC(0,0)= (t0 + 2*t1 + t2 + 2) >> 2;
  741.     SRC(0,1)=SRC(1,0)= (t1 + 2*t2 + t3 + 2) >> 2;
  742.     SRC(0,2)=SRC(1,1)=SRC(2,0)= (t2 + 2*t3 + t4 + 2) >> 2;
  743.     SRC(0,3)=SRC(1,2)=SRC(2,1)=SRC(3,0)= (t3 + 2*t4 + t5 + 2) >> 2;
  744.     SRC(0,4)=SRC(1,3)=SRC(2,2)=SRC(3,1)=SRC(4,0)= (t4 + 2*t5 + t6 + 2) >> 2;
  745.     SRC(0,5)=SRC(1,4)=SRC(2,3)=SRC(3,2)=SRC(4,1)=SRC(5,0)= (t5 + 2*t6 + t7 + 2) >> 2;
  746.     SRC(0,6)=SRC(1,5)=SRC(2,4)=SRC(3,3)=SRC(4,2)=SRC(5,1)=SRC(6,0)= (t6 + 2*t7 + t8 + 2) >> 2;
  747.     SRC(0,7)=SRC(1,6)=SRC(2,5)=SRC(3,4)=SRC(4,3)=SRC(5,2)=SRC(6,1)=SRC(7,0)= (t7 + 2*t8 + t9 + 2) >> 2;
  748.     SRC(1,7)=SRC(2,6)=SRC(3,5)=SRC(4,4)=SRC(5,3)=SRC(6,2)=SRC(7,1)= (t8 + 2*t9 + t10 + 2) >> 2;
  749.     SRC(2,7)=SRC(3,6)=SRC(4,5)=SRC(5,4)=SRC(6,3)=SRC(7,2)= (t9 + 2*t10 + t11 + 2) >> 2;
  750.     SRC(3,7)=SRC(4,6)=SRC(5,5)=SRC(6,4)=SRC(7,3)= (t10 + 2*t11 + t12 + 2) >> 2;
  751.     SRC(4,7)=SRC(5,6)=SRC(6,5)=SRC(7,4)= (t11 + 2*t12 + t13 + 2) >> 2;
  752.     SRC(5,7)=SRC(6,6)=SRC(7,5)= (t12 + 2*t13 + t14 + 2) >> 2;
  753.     SRC(6,7)=SRC(7,6)= (t13 + 2*t14 + t15 + 2) >> 2;
  754.     SRC(7,7)= (t14 + 3*t15 + 2) >> 2;
  755. }
  756. static void pred8x8l_down_right_c(uint8_t *src, int has_topleft, int has_topright, int stride)
  757. {
  758.     PREDICT_8x8_LOAD_TOP;
  759.     PREDICT_8x8_LOAD_LEFT;
  760.     PREDICT_8x8_LOAD_TOPLEFT;
  761.     SRC(0,7)= (l7 + 2*l6 + l5 + 2) >> 2;
  762.     SRC(0,6)=SRC(1,7)= (l6 + 2*l5 + l4 + 2) >> 2;
  763.     SRC(0,5)=SRC(1,6)=SRC(2,7)= (l5 + 2*l4 + l3 + 2) >> 2;
  764.     SRC(0,4)=SRC(1,5)=SRC(2,6)=SRC(3,7)= (l4 + 2*l3 + l2 + 2) >> 2;
  765.     SRC(0,3)=SRC(1,4)=SRC(2,5)=SRC(3,6)=SRC(4,7)= (l3 + 2*l2 + l1 + 2) >> 2;
  766.     SRC(0,2)=SRC(1,3)=SRC(2,4)=SRC(3,5)=SRC(4,6)=SRC(5,7)= (l2 + 2*l1 + l0 + 2) >> 2;
  767.     SRC(0,1)=SRC(1,2)=SRC(2,3)=SRC(3,4)=SRC(4,5)=SRC(5,6)=SRC(6,7)= (l1 + 2*l0 + lt + 2) >> 2;
  768.     SRC(0,0)=SRC(1,1)=SRC(2,2)=SRC(3,3)=SRC(4,4)=SRC(5,5)=SRC(6,6)=SRC(7,7)= (l0 + 2*lt + t0 + 2) >> 2;
  769.     SRC(1,0)=SRC(2,1)=SRC(3,2)=SRC(4,3)=SRC(5,4)=SRC(6,5)=SRC(7,6)= (lt + 2*t0 + t1 + 2) >> 2;
  770.     SRC(2,0)=SRC(3,1)=SRC(4,2)=SRC(5,3)=SRC(6,4)=SRC(7,5)= (t0 + 2*t1 + t2 + 2) >> 2;
  771.     SRC(3,0)=SRC(4,1)=SRC(5,2)=SRC(6,3)=SRC(7,4)= (t1 + 2*t2 + t3 + 2) >> 2;
  772.     SRC(4,0)=SRC(5,1)=SRC(6,2)=SRC(7,3)= (t2 + 2*t3 + t4 + 2) >> 2;
  773.     SRC(5,0)=SRC(6,1)=SRC(7,2)= (t3 + 2*t4 + t5 + 2) >> 2;
  774.     SRC(6,0)=SRC(7,1)= (t4 + 2*t5 + t6 + 2) >> 2;
  775.     SRC(7,0)= (t5 + 2*t6 + t7 + 2) >> 2;
  776. }
  777. static void pred8x8l_vertical_right_c(uint8_t *src, int has_topleft, int has_topright, int stride)
  778. {
  779.     PREDICT_8x8_LOAD_TOP;
  780.     PREDICT_8x8_LOAD_LEFT;
  781.     PREDICT_8x8_LOAD_TOPLEFT;
  782. l7= 0;
  783.     SRC(0,6)= (l5 + 2*l4 + l3 + 2) >> 2;
  784.     SRC(0,7)= (l6 + 2*l5 + l4 + 2) >> 2;
  785.     SRC(0,4)=SRC(1,6)= (l3 + 2*l2 + l1 + 2) >> 2;
  786.     SRC(0,5)=SRC(1,7)= (l4 + 2*l3 + l2 + 2) >> 2;
  787.     SRC(0,2)=SRC(1,4)=SRC(2,6)= (l1 + 2*l0 + lt + 2) >> 2;
  788.     SRC(0,3)=SRC(1,5)=SRC(2,7)= (l2 + 2*l1 + l0 + 2) >> 2;
  789.     SRC(0,1)=SRC(1,3)=SRC(2,5)=SRC(3,7)= (l0 + 2*lt + t0 + 2) >> 2;
  790.     SRC(0,0)=SRC(1,2)=SRC(2,4)=SRC(3,6)= (lt + t0 + 1) >> 1;
  791.     SRC(1,1)=SRC(2,3)=SRC(3,5)=SRC(4,7)= (lt + 2*t0 + t1 + 2) >> 2;
  792.     SRC(1,0)=SRC(2,2)=SRC(3,4)=SRC(4,6)= (t0 + t1 + 1) >> 1;
  793.     SRC(2,1)=SRC(3,3)=SRC(4,5)=SRC(5,7)= (t0 + 2*t1 + t2 + 2) >> 2;
  794.     SRC(2,0)=SRC(3,2)=SRC(4,4)=SRC(5,6)= (t1 + t2 + 1) >> 1;
  795.     SRC(3,1)=SRC(4,3)=SRC(5,5)=SRC(6,7)= (t1 + 2*t2 + t3 + 2) >> 2;
  796.     SRC(3,0)=SRC(4,2)=SRC(5,4)=SRC(6,6)= (t2 + t3 + 1) >> 1;
  797.     SRC(4,1)=SRC(5,3)=SRC(6,5)=SRC(7,7)= (t2 + 2*t3 + t4 + 2) >> 2;
  798.     SRC(4,0)=SRC(5,2)=SRC(6,4)=SRC(7,6)= (t3 + t4 + 1) >> 1;
  799.     SRC(5,1)=SRC(6,3)=SRC(7,5)= (t3 + 2*t4 + t5 + 2) >> 2;
  800.     SRC(5,0)=SRC(6,2)=SRC(7,4)= (t4 + t5 + 1) >> 1;
  801.     SRC(6,1)=SRC(7,3)= (t4 + 2*t5 + t6 + 2) >> 2;
  802.     SRC(6,0)=SRC(7,2)= (t5 + t6 + 1) >> 1;
  803.     SRC(7,1)= (t5 + 2*t6 + t7 + 2) >> 2;
  804.     SRC(7,0)= (t6 + t7 + 1) >> 1;
  805. }
  806. static void pred8x8l_horizontal_down_c(uint8_t *src, int has_topleft, int has_topright, int stride)
  807. {
  808.     PREDICT_8x8_LOAD_TOP;
  809.     PREDICT_8x8_LOAD_LEFT;
  810.     PREDICT_8x8_LOAD_TOPLEFT;
  811. t7 = 0;
  812.     SRC(0,7)= (l6 + l7 + 1) >> 1;
  813.     SRC(1,7)= (l5 + 2*l6 + l7 + 2) >> 2;
  814.     SRC(0,6)=SRC(2,7)= (l5 + l6 + 1) >> 1;
  815.     SRC(1,6)=SRC(3,7)= (l4 + 2*l5 + l6 + 2) >> 2;
  816.     SRC(0,5)=SRC(2,6)=SRC(4,7)= (l4 + l5 + 1) >> 1;
  817.     SRC(1,5)=SRC(3,6)=SRC(5,7)= (l3 + 2*l4 + l5 + 2) >> 2;
  818.     SRC(0,4)=SRC(2,5)=SRC(4,6)=SRC(6,7)= (l3 + l4 + 1) >> 1;
  819.     SRC(1,4)=SRC(3,5)=SRC(5,6)=SRC(7,7)= (l2 + 2*l3 + l4 + 2) >> 2;
  820.     SRC(0,3)=SRC(2,4)=SRC(4,5)=SRC(6,6)= (l2 + l3 + 1) >> 1;
  821.     SRC(1,3)=SRC(3,4)=SRC(5,5)=SRC(7,6)= (l1 + 2*l2 + l3 + 2) >> 2;
  822.     SRC(0,2)=SRC(2,3)=SRC(4,4)=SRC(6,5)= (l1 + l2 + 1) >> 1;
  823.     SRC(1,2)=SRC(3,3)=SRC(5,4)=SRC(7,5)= (l0 + 2*l1 + l2 + 2) >> 2;
  824.     SRC(0,1)=SRC(2,2)=SRC(4,3)=SRC(6,4)= (l0 + l1 + 1) >> 1;
  825.     SRC(1,1)=SRC(3,2)=SRC(5,3)=SRC(7,4)= (lt + 2*l0 + l1 + 2) >> 2;
  826.     SRC(0,0)=SRC(2,1)=SRC(4,2)=SRC(6,3)= (lt + l0 + 1) >> 1;
  827.     SRC(1,0)=SRC(3,1)=SRC(5,2)=SRC(7,3)= (l0 + 2*lt + t0 + 2) >> 2;
  828.     SRC(2,0)=SRC(4,1)=SRC(6,2)= (t1 + 2*t0 + lt + 2) >> 2;
  829.     SRC(3,0)=SRC(5,1)=SRC(7,2)= (t2 + 2*t1 + t0 + 2) >> 2;
  830.     SRC(4,0)=SRC(6,1)= (t3 + 2*t2 + t1 + 2) >> 2;
  831.     SRC(5,0)=SRC(7,1)= (t4 + 2*t3 + t2 + 2) >> 2;
  832.     SRC(6,0)= (t5 + 2*t4 + t3 + 2) >> 2;
  833.     SRC(7,0)= (t6 + 2*t5 + t4 + 2) >> 2;
  834. }
  835. static void pred8x8l_vertical_left_c(uint8_t *src, int has_topleft, int has_topright, int stride)
  836. {
  837.     PREDICT_8x8_LOAD_TOP;
  838.     PREDICT_8x8_LOAD_TOPRIGHT;
  839.     SRC(0,0)= (t0 + t1 + 1) >> 1;
  840.     SRC(0,1)= (t0 + 2*t1 + t2 + 2) >> 2;
  841.     SRC(0,2)=SRC(1,0)= (t1 + t2 + 1) >> 1;
  842.     SRC(0,3)=SRC(1,1)= (t1 + 2*t2 + t3 + 2) >> 2;
  843.     SRC(0,4)=SRC(1,2)=SRC(2,0)= (t2 + t3 + 1) >> 1;
  844.     SRC(0,5)=SRC(1,3)=SRC(2,1)= (t2 + 2*t3 + t4 + 2) >> 2;
  845.     SRC(0,6)=SRC(1,4)=SRC(2,2)=SRC(3,0)= (t3 + t4 + 1) >> 1;
  846.     SRC(0,7)=SRC(1,5)=SRC(2,3)=SRC(3,1)= (t3 + 2*t4 + t5 + 2) >> 2;
  847.     SRC(1,6)=SRC(2,4)=SRC(3,2)=SRC(4,0)= (t4 + t5 + 1) >> 1;
  848.     SRC(1,7)=SRC(2,5)=SRC(3,3)=SRC(4,1)= (t4 + 2*t5 + t6 + 2) >> 2;
  849.     SRC(2,6)=SRC(3,4)=SRC(4,2)=SRC(5,0)= (t5 + t6 + 1) >> 1;
  850.     SRC(2,7)=SRC(3,5)=SRC(4,3)=SRC(5,1)= (t5 + 2*t6 + t7 + 2) >> 2;
  851.     SRC(3,6)=SRC(4,4)=SRC(5,2)=SRC(6,0)= (t6 + t7 + 1) >> 1;
  852.     SRC(3,7)=SRC(4,5)=SRC(5,3)=SRC(6,1)= (t6 + 2*t7 + t8 + 2) >> 2;
  853.     SRC(4,6)=SRC(5,4)=SRC(6,2)=SRC(7,0)= (t7 + t8 + 1) >> 1;
  854.     SRC(4,7)=SRC(5,5)=SRC(6,3)=SRC(7,1)= (t7 + 2*t8 + t9 + 2) >> 2;
  855.     SRC(5,6)=SRC(6,4)=SRC(7,2)= (t8 + t9 + 1) >> 1;
  856.     SRC(5,7)=SRC(6,5)=SRC(7,3)= (t8 + 2*t9 + t10 + 2) >> 2;
  857.     SRC(6,6)=SRC(7,4)= (t9 + t10 + 1) >> 1;
  858.     SRC(6,7)=SRC(7,5)= (t9 + 2*t10 + t11 + 2) >> 2;
  859.     SRC(7,6)= (t10 + t11 + 1) >> 1;
  860.     SRC(7,7)= (t10 + 2*t11 + t12 + 2) >> 2;
  861. }
  862. static void pred8x8l_horizontal_up_c(uint8_t *src, int has_topleft, int has_topright, int stride)
  863. {
  864.     PREDICT_8x8_LOAD_LEFT;
  865.     SRC(0,0)= (l0 + l1 + 1) >> 1;
  866.     SRC(1,0)= (l0 + 2*l1 + l2 + 2) >> 2;
  867.     SRC(0,1)=SRC(2,0)= (l1 + l2 + 1) >> 1;
  868.     SRC(1,1)=SRC(3,0)= (l1 + 2*l2 + l3 + 2) >> 2;
  869.     SRC(0,2)=SRC(2,1)=SRC(4,0)= (l2 + l3 + 1) >> 1;
  870.     SRC(1,2)=SRC(3,1)=SRC(5,0)= (l2 + 2*l3 + l4 + 2) >> 2;
  871.     SRC(0,3)=SRC(2,2)=SRC(4,1)=SRC(6,0)= (l3 + l4 + 1) >> 1;
  872.     SRC(1,3)=SRC(3,2)=SRC(5,1)=SRC(7,0)= (l3 + 2*l4 + l5 + 2) >> 2;
  873.     SRC(0,4)=SRC(2,3)=SRC(4,2)=SRC(6,1)= (l4 + l5 + 1) >> 1;
  874.     SRC(1,4)=SRC(3,3)=SRC(5,2)=SRC(7,1)= (l4 + 2*l5 + l6 + 2) >> 2;
  875.     SRC(0,5)=SRC(2,4)=SRC(4,3)=SRC(6,2)= (l5 + l6 + 1) >> 1;
  876.     SRC(1,5)=SRC(3,4)=SRC(5,3)=SRC(7,2)= (l5 + 2*l6 + l7 + 2) >> 2;
  877.     SRC(0,6)=SRC(2,5)=SRC(4,4)=SRC(6,3)= (l6 + l7 + 1) >> 1;
  878.     SRC(1,6)=SRC(3,5)=SRC(5,4)=SRC(7,3)= (l6 + 3*l7 + 2) >> 2;
  879.     SRC(0,7)=SRC(1,7)=SRC(2,6)=SRC(2,7)=SRC(3,6)=
  880.     SRC(3,7)=SRC(4,5)=SRC(4,6)=SRC(4,7)=SRC(5,5)=
  881.     SRC(5,6)=SRC(5,7)=SRC(6,4)=SRC(6,5)=SRC(6,6)=
  882.     SRC(6,7)=SRC(7,4)=SRC(7,5)=SRC(7,6)=SRC(7,7)= l7;
  883. }
  884. #undef PREDICT_8x8_LOAD_LEFT
  885. #undef PREDICT_8x8_LOAD_TOP
  886. #undef PREDICT_8x8_LOAD_TOPLEFT
  887. #undef PREDICT_8x8_LOAD_TOPRIGHT
  888. #undef PREDICT_8x8_DC
  889. #undef PTR
  890. #undef PT
  891. #undef PL
  892. #undef SRC
  893. static void pred4x4_vertical_add_c(uint8_t *pix, const DCTELEM *block, int stride){
  894.     int i;
  895.     pix -= stride;
  896.     for(i=0; i<4; i++){
  897.         uint8_t v = pix[0];
  898.         pix[1*stride]= v += block[0];
  899.         pix[2*stride]= v += block[4];
  900.         pix[3*stride]= v += block[8];
  901.         pix[4*stride]= v += block[12];
  902.         pix++;
  903.         block++;
  904.     }
  905. }
  906. static void pred4x4_horizontal_add_c(uint8_t *pix, const DCTELEM *block, int stride){
  907.     int i;
  908.     for(i=0; i<4; i++){
  909.         uint8_t v = pix[-1];
  910.         pix[0]= v += block[0];
  911.         pix[1]= v += block[1];
  912.         pix[2]= v += block[2];
  913.         pix[3]= v += block[3];
  914.         pix+= stride;
  915.         block+= 4;
  916.     }
  917. }
  918. static void pred8x8l_vertical_add_c(uint8_t *pix, const DCTELEM *block, int stride){
  919.     int i;
  920.     pix -= stride;
  921.     for(i=0; i<8; i++){
  922.         uint8_t v = pix[0];
  923.         pix[1*stride]= v += block[0];
  924.         pix[2*stride]= v += block[8];
  925.         pix[3*stride]= v += block[16];
  926.         pix[4*stride]= v += block[24];
  927.         pix[5*stride]= v += block[32];
  928.         pix[6*stride]= v += block[40];
  929.         pix[7*stride]= v += block[48];
  930.         pix[8*stride]= v += block[56];
  931.         pix++;
  932.         block++;
  933.     }
  934. }
  935. static void pred8x8l_horizontal_add_c(uint8_t *pix, const DCTELEM *block, int stride){
  936.     int i;
  937.     for(i=0; i<8; i++){
  938.         uint8_t v = pix[-1];
  939.         pix[0]= v += block[0];
  940.         pix[1]= v += block[1];
  941.         pix[2]= v += block[2];
  942.         pix[3]= v += block[3];
  943.         pix[4]= v += block[4];
  944.         pix[5]= v += block[5];
  945.         pix[6]= v += block[6];
  946.         pix[7]= v += block[7];
  947.         pix+= stride;
  948.         block+= 8;
  949.     }
  950. }
  951. static void pred16x16_vertical_add_c(uint8_t *pix, const int *block_offset, const DCTELEM *block, int stride){
  952.     int i;
  953.     for(i=0; i<16; i++)
  954.         pred4x4_vertical_add_c(pix + block_offset[i], block + i*16, stride);
  955. }
  956. static void pred16x16_horizontal_add_c(uint8_t *pix, const int *block_offset, const DCTELEM *block, int stride){
  957.     int i;
  958.     for(i=0; i<16; i++)
  959.         pred4x4_horizontal_add_c(pix + block_offset[i], block + i*16, stride);
  960. }
  961. static void pred8x8_vertical_add_c(uint8_t *pix, const int *block_offset, const DCTELEM *block, int stride){
  962.     int i;
  963.     for(i=0; i<4; i++)
  964.         pred4x4_vertical_add_c(pix + block_offset[i], block + i*16, stride);
  965. }
  966. static void pred8x8_horizontal_add_c(uint8_t *pix, const int *block_offset, const DCTELEM *block, int stride){
  967.     int i;
  968.     for(i=0; i<4; i++)
  969.         pred4x4_horizontal_add_c(pix + block_offset[i], block + i*16, stride);
  970. }
  971. /**
  972.  * Sets the intra prediction function pointers.
  973.  */
  974. void ff_h264_pred_init(H264PredContext *h, int codec_id){
  975. //    MpegEncContext * const s = &h->s;
  976.     if(codec_id != CODEC_ID_RV40){
  977.         h->pred4x4[VERT_PRED           ]= pred4x4_vertical_c;
  978.         h->pred4x4[HOR_PRED            ]= pred4x4_horizontal_c;
  979.         h->pred4x4[DC_PRED             ]= pred4x4_dc_c;
  980.         if(codec_id == CODEC_ID_SVQ3)
  981.             h->pred4x4[DIAG_DOWN_LEFT_PRED ]= pred4x4_down_left_svq3_c;
  982.         else
  983.             h->pred4x4[DIAG_DOWN_LEFT_PRED ]= pred4x4_down_left_c;
  984.         h->pred4x4[DIAG_DOWN_RIGHT_PRED]= pred4x4_down_right_c;
  985.         h->pred4x4[VERT_RIGHT_PRED     ]= pred4x4_vertical_right_c;
  986.         h->pred4x4[HOR_DOWN_PRED       ]= pred4x4_horizontal_down_c;
  987.         h->pred4x4[VERT_LEFT_PRED      ]= pred4x4_vertical_left_c;
  988.         h->pred4x4[HOR_UP_PRED         ]= pred4x4_horizontal_up_c;
  989.         h->pred4x4[LEFT_DC_PRED        ]= pred4x4_left_dc_c;
  990.         h->pred4x4[TOP_DC_PRED         ]= pred4x4_top_dc_c;
  991.         h->pred4x4[DC_128_PRED         ]= pred4x4_128_dc_c;
  992.     }else{
  993.         h->pred4x4[VERT_PRED           ]= pred4x4_vertical_c;
  994.         h->pred4x4[HOR_PRED            ]= pred4x4_horizontal_c;
  995.         h->pred4x4[DC_PRED             ]= pred4x4_dc_c;
  996.         h->pred4x4[DIAG_DOWN_LEFT_PRED ]= pred4x4_down_left_rv40_c;
  997.         h->pred4x4[DIAG_DOWN_RIGHT_PRED]= pred4x4_down_right_c;
  998.         h->pred4x4[VERT_RIGHT_PRED     ]= pred4x4_vertical_right_c;
  999.         h->pred4x4[HOR_DOWN_PRED       ]= pred4x4_horizontal_down_c;
  1000.         h->pred4x4[VERT_LEFT_PRED      ]= pred4x4_vertical_left_rv40_c;
  1001.         h->pred4x4[HOR_UP_PRED         ]= pred4x4_horizontal_up_rv40_c;
  1002.         h->pred4x4[LEFT_DC_PRED        ]= pred4x4_left_dc_c;
  1003.         h->pred4x4[TOP_DC_PRED         ]= pred4x4_top_dc_c;
  1004.         h->pred4x4[DC_128_PRED         ]= pred4x4_128_dc_c;
  1005.         h->pred4x4[DIAG_DOWN_LEFT_PRED_RV40_NODOWN]= pred4x4_down_left_rv40_nodown_c;
  1006.         h->pred4x4[HOR_UP_PRED_RV40_NODOWN]= pred4x4_horizontal_up_rv40_nodown_c;
  1007.         h->pred4x4[VERT_LEFT_PRED_RV40_NODOWN]= pred4x4_vertical_left_rv40_nodown_c;
  1008.     }
  1009.     h->pred8x8l[VERT_PRED           ]= pred8x8l_vertical_c;
  1010.     h->pred8x8l[HOR_PRED            ]= pred8x8l_horizontal_c;
  1011.     h->pred8x8l[DC_PRED             ]= pred8x8l_dc_c;
  1012.     h->pred8x8l[DIAG_DOWN_LEFT_PRED ]= pred8x8l_down_left_c;
  1013.     h->pred8x8l[DIAG_DOWN_RIGHT_PRED]= pred8x8l_down_right_c;
  1014.     h->pred8x8l[VERT_RIGHT_PRED     ]= pred8x8l_vertical_right_c;
  1015.     h->pred8x8l[HOR_DOWN_PRED       ]= pred8x8l_horizontal_down_c;
  1016.     h->pred8x8l[VERT_LEFT_PRED      ]= pred8x8l_vertical_left_c;
  1017.     h->pred8x8l[HOR_UP_PRED         ]= pred8x8l_horizontal_up_c;
  1018.     h->pred8x8l[LEFT_DC_PRED        ]= pred8x8l_left_dc_c;
  1019.     h->pred8x8l[TOP_DC_PRED         ]= pred8x8l_top_dc_c;
  1020.     h->pred8x8l[DC_128_PRED         ]= pred8x8l_128_dc_c;
  1021.     h->pred8x8[VERT_PRED8x8   ]= pred8x8_vertical_c;
  1022.     h->pred8x8[HOR_PRED8x8    ]= pred8x8_horizontal_c;
  1023.     h->pred8x8[PLANE_PRED8x8  ]= pred8x8_plane_c;
  1024.     if(codec_id != CODEC_ID_RV40){
  1025.         h->pred8x8[DC_PRED8x8     ]= pred8x8_dc_c;
  1026.         h->pred8x8[LEFT_DC_PRED8x8]= pred8x8_left_dc_c;
  1027.         h->pred8x8[TOP_DC_PRED8x8 ]= pred8x8_top_dc_c;
  1028.         h->pred8x8[ALZHEIMER_DC_L0T_PRED8x8 ]= pred8x8_mad_cow_dc_l0t;
  1029.         h->pred8x8[ALZHEIMER_DC_0LT_PRED8x8 ]= pred8x8_mad_cow_dc_0lt;
  1030.         h->pred8x8[ALZHEIMER_DC_L00_PRED8x8 ]= pred8x8_mad_cow_dc_l00;
  1031.         h->pred8x8[ALZHEIMER_DC_0L0_PRED8x8 ]= pred8x8_mad_cow_dc_0l0;
  1032.     }else{
  1033.         h->pred8x8[DC_PRED8x8     ]= pred8x8_dc_rv40_c;
  1034.         h->pred8x8[LEFT_DC_PRED8x8]= pred8x8_left_dc_rv40_c;
  1035.         h->pred8x8[TOP_DC_PRED8x8 ]= pred8x8_top_dc_rv40_c;
  1036.     }
  1037.     h->pred8x8[DC_128_PRED8x8 ]= pred8x8_128_dc_c;
  1038.     h->pred16x16[DC_PRED8x8     ]= pred16x16_dc_c;
  1039.     h->pred16x16[VERT_PRED8x8   ]= pred16x16_vertical_c;
  1040.     h->pred16x16[HOR_PRED8x8    ]= pred16x16_horizontal_c;
  1041.     h->pred16x16[PLANE_PRED8x8  ]= pred16x16_plane_c;
  1042.     switch(codec_id){
  1043.     case CODEC_ID_SVQ3:
  1044.        h->pred16x16[PLANE_PRED8x8  ]= pred16x16_plane_svq3_c;
  1045.        break;
  1046.     case CODEC_ID_RV40:
  1047.        h->pred16x16[PLANE_PRED8x8  ]= pred16x16_plane_rv40_c;
  1048.        break;
  1049.     default:
  1050.        h->pred16x16[PLANE_PRED8x8  ]= pred16x16_plane_c;
  1051.     }
  1052.     h->pred16x16[LEFT_DC_PRED8x8]= pred16x16_left_dc_c;
  1053.     h->pred16x16[TOP_DC_PRED8x8 ]= pred16x16_top_dc_c;
  1054.     h->pred16x16[DC_128_PRED8x8 ]= pred16x16_128_dc_c;
  1055.     //special lossless h/v prediction for h264
  1056.     h->pred4x4_add  [VERT_PRED   ]= pred4x4_vertical_add_c;
  1057.     h->pred4x4_add  [ HOR_PRED   ]= pred4x4_horizontal_add_c;
  1058.     h->pred8x8l_add [VERT_PRED   ]= pred8x8l_vertical_add_c;
  1059.     h->pred8x8l_add [ HOR_PRED   ]= pred8x8l_horizontal_add_c;
  1060.     h->pred8x8_add  [VERT_PRED8x8]= pred8x8_vertical_add_c;
  1061.     h->pred8x8_add  [ HOR_PRED8x8]= pred8x8_horizontal_add_c;
  1062.     h->pred16x16_add[VERT_PRED8x8]= pred16x16_vertical_add_c;
  1063.     h->pred16x16_add[ HOR_PRED8x8]= pred16x16_horizontal_add_c;
  1064. }