开通博客赚积分
发布资源赚积分
分类

源码开发语言/平台
当前位置：首页 > 源码/资料 > 多媒体 > 流媒体/Mpeg4/MP4 > 查看源码
h264pred.c
资源名称：X264CODEC.rar　[点击查看]
上传用户：lctgjx
上传日期：2022-06-04
资源大小：8887k
文件大小：37k
源码类别：

流媒体/Mpeg4/MP4

开发平台：

Visual C++

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