git.sesse.net Git - x264/blob - common/pixel.c

   1 /*****************************************************************************
   2  * pixel.c: h264 encoder
   3  *****************************************************************************
   4  * Copyright (C) 2003-2008 x264 project
   5  *
   6  * Authors: Loren Merritt <lorenm@u.washington.edu>
   7  *          Laurent Aimar <fenrir@via.ecp.fr>
   8  *
   9  * This program is free software; you can redistribute it and/or modify
  10  * it under the terms of the GNU General Public License as published by
  11  * the Free Software Foundation; either version 2 of the License, or
  12  * (at your option) any later version.
  13  *
  14  * This program is distributed in the hope that it will be useful,
  15  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  16  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  17  * GNU General Public License for more details.
  18  *
  19  * You should have received a copy of the GNU General Public License
  20  * along with this program; if not, write to the Free Software
  21  * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA  02111, USA.
  22  *****************************************************************************/
  23
  24 #include "common.h"
  25
  26 #ifdef HAVE_MMX
  27 #   include "x86/pixel.h"
  28 #endif
  29 #ifdef ARCH_PPC
  30 #   include "ppc/pixel.h"
  31 #endif
  32 #ifdef ARCH_ARM
  33 #   include "arm/pixel.h"
  34 #endif
  35 #ifdef ARCH_UltraSparc
  36 #   include "sparc/pixel.h"
  37 #endif
  38
  39
  40 /****************************************************************************
  41  * pixel_sad_WxH
  42  ****************************************************************************/
  43 #define PIXEL_SAD_C( name, lx, ly ) \
  44 static int name( uint8_t *pix1, int i_stride_pix1,  \
  45                  uint8_t *pix2, int i_stride_pix2 ) \
  46 {                                                   \
  47     int i_sum = 0;                                  \
  48     int x, y;                                       \
  49     for( y = 0; y < ly; y++ )                       \
  50     {                                               \
  51         for( x = 0; x < lx; x++ )                   \
  52         {                                           \
  53             i_sum += abs( pix1[x] - pix2[x] );      \
  54         }                                           \
  55         pix1 += i_stride_pix1;                      \
  56         pix2 += i_stride_pix2;                      \
  57     }                                               \
  58     return i_sum;                                   \
  59 }
  60
  61
  62 PIXEL_SAD_C( x264_pixel_sad_16x16, 16, 16 )
  63 PIXEL_SAD_C( x264_pixel_sad_16x8,  16,  8 )
  64 PIXEL_SAD_C( x264_pixel_sad_8x16,   8, 16 )
  65 PIXEL_SAD_C( x264_pixel_sad_8x8,    8,  8 )
  66 PIXEL_SAD_C( x264_pixel_sad_8x4,    8,  4 )
  67 PIXEL_SAD_C( x264_pixel_sad_4x8,    4,  8 )
  68 PIXEL_SAD_C( x264_pixel_sad_4x4,    4,  4 )
  69
  70
  71 /****************************************************************************
  72  * pixel_ssd_WxH
  73  ****************************************************************************/
  74 #define PIXEL_SSD_C( name, lx, ly ) \
  75 static int name( uint8_t *pix1, int i_stride_pix1,  \
  76                  uint8_t *pix2, int i_stride_pix2 ) \
  77 {                                                   \
  78     int i_sum = 0;                                  \
  79     int x, y;                                       \
  80     for( y = 0; y < ly; y++ )                       \
  81     {                                               \
  82         for( x = 0; x < lx; x++ )                   \
  83         {                                           \
  84             int d = pix1[x] - pix2[x];              \
  85             i_sum += d*d;                           \
  86         }                                           \
  87         pix1 += i_stride_pix1;                      \
  88         pix2 += i_stride_pix2;                      \
  89     }                                               \
  90     return i_sum;                                   \
  91 }
  92
  93 PIXEL_SSD_C( x264_pixel_ssd_16x16, 16, 16 )
  94 PIXEL_SSD_C( x264_pixel_ssd_16x8,  16,  8 )
  95 PIXEL_SSD_C( x264_pixel_ssd_8x16,   8, 16 )
  96 PIXEL_SSD_C( x264_pixel_ssd_8x8,    8,  8 )
  97 PIXEL_SSD_C( x264_pixel_ssd_8x4,    8,  4 )
  98 PIXEL_SSD_C( x264_pixel_ssd_4x8,    4,  8 )
  99 PIXEL_SSD_C( x264_pixel_ssd_4x4,    4,  4 )
 100
 101 int64_t x264_pixel_ssd_wxh( x264_pixel_function_t *pf, uint8_t *pix1, int i_pix1, uint8_t *pix2, int i_pix2, int i_width, int i_height )
 102 {
 103     int64_t i_ssd = 0;
 104     int x, y;
 105     int align = !(((intptr_t)pix1 | (intptr_t)pix2 | i_pix1 | i_pix2) & 15);
 106
 107 #define SSD(size) i_ssd += pf->ssd[size]( pix1 + y*i_pix1 + x, i_pix1, \
 108                                           pix2 + y*i_pix2 + x, i_pix2 );
 109     for( y = 0; y < i_height-15; y += 16 )
 110     {
 111         x = 0;
 112         if( align )
 113             for( ; x < i_width-15; x += 16 )
 114                 SSD(PIXEL_16x16);
 115         for( ; x < i_width-7; x += 8 )
 116             SSD(PIXEL_8x16);
 117     }
 118     if( y < i_height-7 )
 119         for( x = 0; x < i_width-7; x += 8 )
 120             SSD(PIXEL_8x8);
 121 #undef SSD
 122
 123 #define SSD1 { int d = pix1[y*i_pix1+x] - pix2[y*i_pix2+x]; i_ssd += d*d; }
 124     if( i_width % 8 != 0 )
 125     {
 126         for( y = 0; y < (i_height & ~7); y++ )
 127             for( x = i_width & ~7; x < i_width; x++ )
 128                 SSD1;
 129     }
 130     if( i_height % 8 != 0 )
 131     {
 132         for( y = i_height & ~7; y < i_height; y++ )
 133             for( x = 0; x < i_width; x++ )
 134                 SSD1;
 135     }
 136 #undef SSD1
 137
 138     return i_ssd;
 139 }
 140
 141
 142 /****************************************************************************
 143  * pixel_var_wxh
 144  ****************************************************************************/
 145 #define PIXEL_VAR_C( name, w ) \
 146 static uint64_t name( uint8_t *pix, int i_stride ) \
 147 {                                             \
 148     uint32_t sum = 0, sqr = 0;                \
 149     int x, y;                                 \
 150     for( y = 0; y < w; y++ )                  \
 151     {                                         \
 152         for( x = 0; x < w; x++ )              \
 153         {                                     \
 154             sum += pix[x];                    \
 155             sqr += pix[x] * pix[x];           \
 156         }                                     \
 157         pix += i_stride;                      \
 158     }                                         \
 159     return sum + ((uint64_t)sqr << 32);       \
 160 }
 161
 162 PIXEL_VAR_C( x264_pixel_var_16x16, 16 )
 163 PIXEL_VAR_C( x264_pixel_var_8x8,    8 )
 164
 165 /****************************************************************************
 166  * pixel_var2_wxh
 167  ****************************************************************************/
 168 static int pixel_var2_8x8( uint8_t *pix1, int i_stride1, uint8_t *pix2, int i_stride2, int *ssd )
 169 {
 170     uint32_t var = 0, sum = 0, sqr = 0;
 171     int x, y;
 172     for( y = 0; y < 8; y++ )
 173     {
 174         for( x = 0; x < 8; x++ )
 175         {
 176             int diff = pix1[x] - pix2[x];
 177             sum += diff;
 178             sqr += diff * diff;
 179         }
 180         pix1 += i_stride1;
 181         pix2 += i_stride2;
 182     }
 183     sum = abs(sum);
 184     var = sqr - (sum * sum >> 6);
 185     *ssd = sqr;
 186     return var;
 187 }
 188
 189
 190 #define HADAMARD4(d0,d1,d2,d3,s0,s1,s2,s3) {\
 191     int t0 = s0 + s1;\
 192     int t1 = s0 - s1;\
 193     int t2 = s2 + s3;\
 194     int t3 = s2 - s3;\
 195     d0 = t0 + t2;\
 196     d2 = t0 - t2;\
 197     d1 = t1 + t3;\
 198     d3 = t1 - t3;\
 199 }
 200
 201 // in: a pseudo-simd number of the form x+(y<<16)
 202 // return: abs(x)+(abs(y)<<16)
 203 static ALWAYS_INLINE uint32_t abs2( uint32_t a )
 204 {
 205     uint32_t s = ((a>>15)&0x10001)*0xffff;
 206     return (a+s)^s;
 207 }
 208
 209 /****************************************************************************
 210  * pixel_satd_WxH: sum of 4x4 Hadamard transformed differences
 211  ****************************************************************************/
 212
 213 static NOINLINE int x264_pixel_satd_4x4( uint8_t *pix1, int i_pix1, uint8_t *pix2, int i_pix2 )
 214 {
 215     uint32_t tmp[4][2];
 216     uint32_t a0,a1,a2,a3,b0,b1;
 217     int sum=0, i;
 218     for( i=0; i<4; i++, pix1+=i_pix1, pix2+=i_pix2 )
 219     {
 220         a0 = pix1[0] - pix2[0];
 221         a1 = pix1[1] - pix2[1];
 222         b0 = (a0+a1) + ((a0-a1)<<16);
 223         a2 = pix1[2] - pix2[2];
 224         a3 = pix1[3] - pix2[3];
 225         b1 = (a2+a3) + ((a2-a3)<<16);
 226         tmp[i][0] = b0 + b1;
 227         tmp[i][1] = b0 - b1;
 228     }
 229     for( i=0; i<2; i++ )
 230     {
 231         HADAMARD4( a0,a1,a2,a3, tmp[0][i], tmp[1][i], tmp[2][i], tmp[3][i] );
 232         a0 = abs2(a0) + abs2(a1) + abs2(a2) + abs2(a3);
 233         sum += ((uint16_t)a0) + (a0>>16);
 234     }
 235     return sum >> 1;
 236 }
 237
 238 static NOINLINE int x264_pixel_satd_8x4( uint8_t *pix1, int i_pix1, uint8_t *pix2, int i_pix2 )
 239 {
 240     uint32_t tmp[4][4];
 241     uint32_t a0,a1,a2,a3;
 242     int sum=0, i;
 243     for( i=0; i<4; i++, pix1+=i_pix1, pix2+=i_pix2 )
 244     {
 245         a0 = (pix1[0] - pix2[0]) + ((pix1[4] - pix2[4]) << 16);
 246         a1 = (pix1[1] - pix2[1]) + ((pix1[5] - pix2[5]) << 16);
 247         a2 = (pix1[2] - pix2[2]) + ((pix1[6] - pix2[6]) << 16);
 248         a3 = (pix1[3] - pix2[3]) + ((pix1[7] - pix2[7]) << 16);
 249         HADAMARD4( tmp[i][0], tmp[i][1], tmp[i][2], tmp[i][3], a0,a1,a2,a3 );
 250     }
 251     for( i=0; i<4; i++ )
 252     {
 253         HADAMARD4( a0,a1,a2,a3, tmp[0][i], tmp[1][i], tmp[2][i], tmp[3][i] );
 254         sum += abs2(a0) + abs2(a1) + abs2(a2) + abs2(a3);
 255     }
 256     return (((uint16_t)sum) + ((uint32_t)sum>>16)) >> 1;
 257 }
 258
 259 #define PIXEL_SATD_C( w, h, sub )\
 260 static int x264_pixel_satd_##w##x##h( uint8_t *pix1, int i_pix1, uint8_t *pix2, int i_pix2 )\
 261 {\
 262     int sum = sub( pix1, i_pix1, pix2, i_pix2 )\
 263             + sub( pix1+4*i_pix1, i_pix1, pix2+4*i_pix2, i_pix2 );\
 264     if( w==16 )\
 265         sum+= sub( pix1+8, i_pix1, pix2+8, i_pix2 )\
 266             + sub( pix1+8+4*i_pix1, i_pix1, pix2+8+4*i_pix2, i_pix2 );\
 267     if( h==16 )\
 268         sum+= sub( pix1+8*i_pix1, i_pix1, pix2+8*i_pix2, i_pix2 )\
 269             + sub( pix1+12*i_pix1, i_pix1, pix2+12*i_pix2, i_pix2 );\
 270     if( w==16 && h==16 )\
 271         sum+= sub( pix1+8+8*i_pix1, i_pix1, pix2+8+8*i_pix2, i_pix2 )\
 272             + sub( pix1+8+12*i_pix1, i_pix1, pix2+8+12*i_pix2, i_pix2 );\
 273     return sum;\
 274 }
 275 PIXEL_SATD_C( 16, 16, x264_pixel_satd_8x4 )
 276 PIXEL_SATD_C( 16, 8,  x264_pixel_satd_8x4 )
 277 PIXEL_SATD_C( 8,  16, x264_pixel_satd_8x4 )
 278 PIXEL_SATD_C( 8,  8,  x264_pixel_satd_8x4 )
 279 PIXEL_SATD_C( 4,  8,  x264_pixel_satd_4x4 )
 280
 281
 282 static NOINLINE int sa8d_8x8( uint8_t *pix1, int i_pix1, uint8_t *pix2, int i_pix2 )
 283 {
 284     uint32_t tmp[8][4];
 285     uint32_t a0,a1,a2,a3,a4,a5,a6,a7,b0,b1,b2,b3;
 286     int sum=0, i;
 287     for( i=0; i<8; i++, pix1+=i_pix1, pix2+=i_pix2 )
 288     {
 289         a0 = pix1[0] - pix2[0];
 290         a1 = pix1[1] - pix2[1];
 291         b0 = (a0+a1) + ((a0-a1)<<16);
 292         a2 = pix1[2] - pix2[2];
 293         a3 = pix1[3] - pix2[3];
 294         b1 = (a2+a3) + ((a2-a3)<<16);
 295         a4 = pix1[4] - pix2[4];
 296         a5 = pix1[5] - pix2[5];
 297         b2 = (a4+a5) + ((a4-a5)<<16);
 298         a6 = pix1[6] - pix2[6];
 299         a7 = pix1[7] - pix2[7];
 300         b3 = (a6+a7) + ((a6-a7)<<16);
 301         HADAMARD4( tmp[i][0], tmp[i][1], tmp[i][2], tmp[i][3], b0,b1,b2,b3 );
 302     }
 303     for( i=0; i<4; i++ )
 304     {
 305         HADAMARD4( a0,a1,a2,a3, tmp[0][i], tmp[1][i], tmp[2][i], tmp[3][i] );
 306         HADAMARD4( a4,a5,a6,a7, tmp[4][i], tmp[5][i], tmp[6][i], tmp[7][i] );
 307         b0  = abs2(a0+a4) + abs2(a0-a4);
 308         b0 += abs2(a1+a5) + abs2(a1-a5);
 309         b0 += abs2(a2+a6) + abs2(a2-a6);
 310         b0 += abs2(a3+a7) + abs2(a3-a7);
 311         sum += (uint16_t)b0 + (b0>>16);
 312     }
 313     return sum;
 314 }
 315
 316 static int x264_pixel_sa8d_8x8( uint8_t *pix1, int i_pix1, uint8_t *pix2, int i_pix2 )
 317 {
 318     int sum = sa8d_8x8( pix1, i_pix1, pix2, i_pix2 );
 319     return (sum+2)>>2;
 320 }
 321
 322 static int x264_pixel_sa8d_16x16( uint8_t *pix1, int i_pix1, uint8_t *pix2, int i_pix2 )
 323 {
 324     int sum = sa8d_8x8( pix1, i_pix1, pix2, i_pix2 )
 325             + sa8d_8x8( pix1+8, i_pix1, pix2+8, i_pix2 )
 326             + sa8d_8x8( pix1+8*i_pix1, i_pix1, pix2+8*i_pix2, i_pix2 )
 327             + sa8d_8x8( pix1+8+8*i_pix1, i_pix1, pix2+8+8*i_pix2, i_pix2 );
 328     return (sum+2)>>2;
 329 }
 330
 331
 332 static NOINLINE uint64_t pixel_hadamard_ac( uint8_t *pix, int stride )
 333 {
 334     uint32_t tmp[32];
 335     uint32_t a0,a1,a2,a3,dc;
 336     int sum4=0, sum8=0, i;
 337     for( i=0; i<8; i++, pix+=stride )
 338     {
 339         uint32_t *t = tmp + (i&3) + (i&4)*4;
 340         a0 = (pix[0]+pix[1]) + ((pix[0]-pix[1])<<16);
 341         a1 = (pix[2]+pix[3]) + ((pix[2]-pix[3])<<16);
 342         t[0] = a0 + a1;
 343         t[4] = a0 - a1;
 344         a2 = (pix[4]+pix[5]) + ((pix[4]-pix[5])<<16);
 345         a3 = (pix[6]+pix[7]) + ((pix[6]-pix[7])<<16);
 346         t[8] = a2 + a3;
 347         t[12] = a2 - a3;
 348     }
 349     for( i=0; i<8; i++ )
 350     {
 351         HADAMARD4( a0,a1,a2,a3, tmp[i*4+0], tmp[i*4+1], tmp[i*4+2], tmp[i*4+3] );
 352         tmp[i*4+0] = a0;
 353         tmp[i*4+1] = a1;
 354         tmp[i*4+2] = a2;
 355         tmp[i*4+3] = a3;
 356         sum4 += abs2(a0) + abs2(a1) + abs2(a2) + abs2(a3);
 357     }
 358     for( i=0; i<8; i++ )
 359     {
 360         HADAMARD4( a0,a1,a2,a3, tmp[i], tmp[8+i], tmp[16+i], tmp[24+i] );
 361         sum8 += abs2(a0) + abs2(a1) + abs2(a2) + abs2(a3);
 362     }
 363     dc = (uint16_t)(tmp[0] + tmp[8] + tmp[16] + tmp[24]);
 364     sum4 = (uint16_t)sum4 + ((uint32_t)sum4>>16) - dc;
 365     sum8 = (uint16_t)sum8 + ((uint32_t)sum8>>16) - dc;
 366     return ((uint64_t)sum8<<32) + sum4;
 367 }
 368
 369 #define HADAMARD_AC(w,h) \
 370 static uint64_t x264_pixel_hadamard_ac_##w##x##h( uint8_t *pix, int stride )\
 371 {\
 372     uint64_t sum = pixel_hadamard_ac( pix, stride );\
 373     if( w==16 )\
 374         sum += pixel_hadamard_ac( pix+8, stride );\
 375     if( h==16 )\
 376         sum += pixel_hadamard_ac( pix+8*stride, stride );\
 377     if( w==16 && h==16 )\
 378         sum += pixel_hadamard_ac( pix+8*stride+8, stride );\
 379     return ((sum>>34)<<32) + ((uint32_t)sum>>1);\
 380 }
 381 HADAMARD_AC( 16, 16 )
 382 HADAMARD_AC( 16, 8 )
 383 HADAMARD_AC( 8, 16 )
 384 HADAMARD_AC( 8, 8 )
 385
 386
 387 /****************************************************************************
 388  * pixel_sad_x4
 389  ****************************************************************************/
 390 #define SAD_X( size ) \
 391 static void x264_pixel_sad_x3_##size( uint8_t *fenc, uint8_t *pix0, uint8_t *pix1, uint8_t *pix2, int i_stride, int scores[3] )\
 392 {\
 393     scores[0] = x264_pixel_sad_##size( fenc, FENC_STRIDE, pix0, i_stride );\
 394     scores[1] = x264_pixel_sad_##size( fenc, FENC_STRIDE, pix1, i_stride );\
 395     scores[2] = x264_pixel_sad_##size( fenc, FENC_STRIDE, pix2, i_stride );\
 396 }\
 397 static void x264_pixel_sad_x4_##size( uint8_t *fenc, uint8_t *pix0, uint8_t *pix1, uint8_t *pix2, uint8_t *pix3, int i_stride, int scores[4] )\
 398 {\
 399     scores[0] = x264_pixel_sad_##size( fenc, FENC_STRIDE, pix0, i_stride );\
 400     scores[1] = x264_pixel_sad_##size( fenc, FENC_STRIDE, pix1, i_stride );\
 401     scores[2] = x264_pixel_sad_##size( fenc, FENC_STRIDE, pix2, i_stride );\
 402     scores[3] = x264_pixel_sad_##size( fenc, FENC_STRIDE, pix3, i_stride );\
 403 }
 404
 405 SAD_X( 16x16 )
 406 SAD_X( 16x8 )
 407 SAD_X( 8x16 )
 408 SAD_X( 8x8 )
 409 SAD_X( 8x4 )
 410 SAD_X( 4x8 )
 411 SAD_X( 4x4 )
 412
 413 #ifdef ARCH_UltraSparc
 414 SAD_X( 16x16_vis )
 415 SAD_X( 16x8_vis )
 416 SAD_X( 8x16_vis )
 417 SAD_X( 8x8_vis )
 418 #endif
 419
 420 /****************************************************************************
 421  * pixel_satd_x4
 422  * no faster than single satd, but needed for satd to be a drop-in replacement for sad
 423  ****************************************************************************/
 424
 425 #define SATD_X( size, cpu ) \
 426 static void x264_pixel_satd_x3_##size##cpu( uint8_t *fenc, uint8_t *pix0, uint8_t *pix1, uint8_t *pix2, int i_stride, int scores[3] )\
 427 {\
 428     scores[0] = x264_pixel_satd_##size##cpu( fenc, FENC_STRIDE, pix0, i_stride );\
 429     scores[1] = x264_pixel_satd_##size##cpu( fenc, FENC_STRIDE, pix1, i_stride );\
 430     scores[2] = x264_pixel_satd_##size##cpu( fenc, FENC_STRIDE, pix2, i_stride );\
 431 }\
 432 static void x264_pixel_satd_x4_##size##cpu( uint8_t *fenc, uint8_t *pix0, uint8_t *pix1, uint8_t *pix2, uint8_t *pix3, int i_stride, int scores[4] )\
 433 {\
 434     scores[0] = x264_pixel_satd_##size##cpu( fenc, FENC_STRIDE, pix0, i_stride );\
 435     scores[1] = x264_pixel_satd_##size##cpu( fenc, FENC_STRIDE, pix1, i_stride );\
 436     scores[2] = x264_pixel_satd_##size##cpu( fenc, FENC_STRIDE, pix2, i_stride );\
 437     scores[3] = x264_pixel_satd_##size##cpu( fenc, FENC_STRIDE, pix3, i_stride );\
 438 }
 439 #define SATD_X_DECL6( cpu )\
 440 SATD_X( 16x16, cpu )\
 441 SATD_X( 16x8, cpu )\
 442 SATD_X( 8x16, cpu )\
 443 SATD_X( 8x8, cpu )\
 444 SATD_X( 8x4, cpu )\
 445 SATD_X( 4x8, cpu )
 446 #define SATD_X_DECL7( cpu )\
 447 SATD_X_DECL6( cpu )\
 448 SATD_X( 4x4, cpu )
 449
 450 SATD_X_DECL7()
 451 #ifdef HAVE_MMX
 452 SATD_X_DECL7( _mmxext )
 453 SATD_X_DECL6( _sse2 )
 454 SATD_X_DECL7( _ssse3 )
 455 SATD_X_DECL7( _sse4 )
 456 #endif
 457
 458 #ifdef HAVE_ARMV6
 459 SATD_X_DECL7( _neon )
 460 #endif
 461
 462 /****************************************************************************
 463  * structural similarity metric
 464  ****************************************************************************/
 465 static void ssim_4x4x2_core( const uint8_t *pix1, int stride1,
 466                              const uint8_t *pix2, int stride2,
 467                              int sums[2][4])
 468 {
 469     int x, y, z;
 470     for(z=0; z<2; z++)
 471     {
 472         uint32_t s1=0, s2=0, ss=0, s12=0;
 473         for(y=0; y<4; y++)
 474             for(x=0; x<4; x++)
 475             {
 476                 int a = pix1[x+y*stride1];
 477                 int b = pix2[x+y*stride2];
 478                 s1  += a;
 479                 s2  += b;
 480                 ss  += a*a;
 481                 ss  += b*b;
 482                 s12 += a*b;
 483             }
 484         sums[z][0] = s1;
 485         sums[z][1] = s2;
 486         sums[z][2] = ss;
 487         sums[z][3] = s12;
 488         pix1 += 4;
 489         pix2 += 4;
 490     }
 491 }
 492
 493 static float ssim_end1( int s1, int s2, int ss, int s12 )
 494 {
 495     static const int ssim_c1 = (int)(.01*.01*255*255*64 + .5);
 496     static const int ssim_c2 = (int)(.03*.03*255*255*64*63 + .5);
 497     int vars = ss*64 - s1*s1 - s2*s2;
 498     int covar = s12*64 - s1*s2;
 499     return (float)(2*s1*s2 + ssim_c1) * (float)(2*covar + ssim_c2)\
 500            / ((float)(s1*s1 + s2*s2 + ssim_c1) * (float)(vars + ssim_c2));
 501 }
 502
 503 static float ssim_end4( int sum0[5][4], int sum1[5][4], int width )
 504 {
 505     int i;
 506     float ssim = 0.0;
 507     for( i = 0; i < width; i++ )
 508         ssim += ssim_end1( sum0[i][0] + sum0[i+1][0] + sum1[i][0] + sum1[i+1][0],
 509                            sum0[i][1] + sum0[i+1][1] + sum1[i][1] + sum1[i+1][1],
 510                            sum0[i][2] + sum0[i+1][2] + sum1[i][2] + sum1[i+1][2],
 511                            sum0[i][3] + sum0[i+1][3] + sum1[i][3] + sum1[i+1][3] );
 512     return ssim;
 513 }
 514
 515 float x264_pixel_ssim_wxh( x264_pixel_function_t *pf,
 516                            uint8_t *pix1, int stride1,
 517                            uint8_t *pix2, int stride2,
 518                            int width, int height, void *buf )
 519 {
 520     int x, y, z;
 521     float ssim = 0.0;
 522     int (*sum0)[4] = buf;
 523     int (*sum1)[4] = sum0 + width/4+3;
 524     width >>= 2;
 525     height >>= 2;
 526     z = 0;
 527     for( y = 1; y < height; y++ )
 528     {
 529         for( ; z <= y; z++ )
 530         {
 531             XCHG( void*, sum0, sum1 );
 532             for( x = 0; x < width; x+=2 )
 533                 pf->ssim_4x4x2_core( &pix1[4*(x+z*stride1)], stride1, &pix2[4*(x+z*stride2)], stride2, &sum0[x] );
 534         }
 535         for( x = 0; x < width-1; x += 4 )
 536             ssim += pf->ssim_end4( sum0+x, sum1+x, X264_MIN(4,width-x-1) );
 537     }
 538     return ssim;
 539 }
 540
 541
 542 /****************************************************************************
 543  * successive elimination
 544  ****************************************************************************/
 545 static int x264_pixel_ads4( int enc_dc[4], uint16_t *sums, int delta,
 546                             uint16_t *cost_mvx, int16_t *mvs, int width, int thresh )
 547 {
 548     int nmv=0, i;
 549     for( i=0; i<width; i++, sums++ )
 550     {
 551         int ads = abs( enc_dc[0] - sums[0] )
 552                 + abs( enc_dc[1] - sums[8] )
 553                 + abs( enc_dc[2] - sums[delta] )
 554                 + abs( enc_dc[3] - sums[delta+8] )
 555                 + cost_mvx[i];
 556         if( ads < thresh )
 557             mvs[nmv++] = i;
 558     }
 559     return nmv;
 560 }
 561
 562 static int x264_pixel_ads2( int enc_dc[2], uint16_t *sums, int delta,
 563                             uint16_t *cost_mvx, int16_t *mvs, int width, int thresh )
 564 {
 565     int nmv=0, i;
 566     for( i=0; i<width; i++, sums++ )
 567     {
 568         int ads = abs( enc_dc[0] - sums[0] )
 569                 + abs( enc_dc[1] - sums[delta] )
 570                 + cost_mvx[i];
 571         if( ads < thresh )
 572             mvs[nmv++] = i;
 573     }
 574     return nmv;
 575 }
 576
 577 static int x264_pixel_ads1( int enc_dc[1], uint16_t *sums, int delta,
 578                             uint16_t *cost_mvx, int16_t *mvs, int width, int thresh )
 579 {
 580     int nmv=0, i;
 581     for( i=0; i<width; i++, sums++ )
 582     {
 583         int ads = abs( enc_dc[0] - sums[0] )
 584                 + cost_mvx[i];
 585         if( ads < thresh )
 586             mvs[nmv++] = i;
 587     }
 588     return nmv;
 589 }
 590
 591
 592 /****************************************************************************
 593  * x264_pixel_init:
 594  ****************************************************************************/
 595 void x264_pixel_init( int cpu, x264_pixel_function_t *pixf )
 596 {
 597     memset( pixf, 0, sizeof(*pixf) );
 598
 599 #define INIT2_NAME( name1, name2, cpu ) \
 600     pixf->name1[PIXEL_16x16] = x264_pixel_##name2##_16x16##cpu;\
 601     pixf->name1[PIXEL_16x8]  = x264_pixel_##name2##_16x8##cpu;
 602 #define INIT4_NAME( name1, name2, cpu ) \
 603     INIT2_NAME( name1, name2, cpu ) \
 604     pixf->name1[PIXEL_8x16]  = x264_pixel_##name2##_8x16##cpu;\
 605     pixf->name1[PIXEL_8x8]   = x264_pixel_##name2##_8x8##cpu;
 606 #define INIT5_NAME( name1, name2, cpu ) \
 607     INIT4_NAME( name1, name2, cpu ) \
 608     pixf->name1[PIXEL_8x4]   = x264_pixel_##name2##_8x4##cpu;
 609 #define INIT6_NAME( name1, name2, cpu ) \
 610     INIT5_NAME( name1, name2, cpu ) \
 611     pixf->name1[PIXEL_4x8]   = x264_pixel_##name2##_4x8##cpu;
 612 #define INIT7_NAME( name1, name2, cpu ) \
 613     INIT6_NAME( name1, name2, cpu ) \
 614     pixf->name1[PIXEL_4x4]   = x264_pixel_##name2##_4x4##cpu;
 615 #define INIT2( name, cpu ) INIT2_NAME( name, name, cpu )
 616 #define INIT4( name, cpu ) INIT4_NAME( name, name, cpu )
 617 #define INIT5( name, cpu ) INIT5_NAME( name, name, cpu )
 618 #define INIT6( name, cpu ) INIT6_NAME( name, name, cpu )
 619 #define INIT7( name, cpu ) INIT7_NAME( name, name, cpu )
 620
 621 #define INIT_ADS( cpu ) \
 622     pixf->ads[PIXEL_16x16] = x264_pixel_ads4##cpu;\
 623     pixf->ads[PIXEL_16x8] = x264_pixel_ads2##cpu;\
 624     pixf->ads[PIXEL_8x8] = x264_pixel_ads1##cpu;
 625
 626     INIT7( sad, );
 627     INIT7_NAME( sad_aligned, sad, );
 628     INIT7( sad_x3, );
 629     INIT7( sad_x4, );
 630     INIT7( ssd, );
 631     INIT7( satd, );
 632     INIT7( satd_x3, );
 633     INIT7( satd_x4, );
 634     INIT4( hadamard_ac, );
 635     INIT_ADS( );
 636
 637     pixf->sa8d[PIXEL_16x16] = x264_pixel_sa8d_16x16;
 638     pixf->sa8d[PIXEL_8x8]   = x264_pixel_sa8d_8x8;
 639     pixf->var[PIXEL_16x16] = x264_pixel_var_16x16;
 640     pixf->var[PIXEL_8x8]   = x264_pixel_var_8x8;
 641
 642     pixf->ssim_4x4x2_core = ssim_4x4x2_core;
 643     pixf->ssim_end4 = ssim_end4;
 644     pixf->var2_8x8 = pixel_var2_8x8;
 645
 646 #ifdef HAVE_MMX
 647     if( cpu&X264_CPU_MMX )
 648     {
 649         INIT7( ssd, _mmx );
 650     }
 651
 652     if( cpu&X264_CPU_MMXEXT )
 653     {
 654         INIT7( sad, _mmxext );
 655         INIT7_NAME( sad_aligned, sad, _mmxext );
 656         INIT7( sad_x3, _mmxext );
 657         INIT7( sad_x4, _mmxext );
 658         INIT7( satd, _mmxext );
 659         INIT7( satd_x3, _mmxext );
 660         INIT7( satd_x4, _mmxext );
 661         INIT4( hadamard_ac, _mmxext );
 662         INIT_ADS( _mmxext );
 663         pixf->var[PIXEL_16x16] = x264_pixel_var_16x16_mmxext;
 664         pixf->var[PIXEL_8x8]   = x264_pixel_var_8x8_mmxext;
 665 #ifdef ARCH_X86
 666         pixf->sa8d[PIXEL_16x16] = x264_pixel_sa8d_16x16_mmxext;
 667         pixf->sa8d[PIXEL_8x8]   = x264_pixel_sa8d_8x8_mmxext;
 668         pixf->intra_sa8d_x3_8x8 = x264_intra_sa8d_x3_8x8_mmxext;
 669         pixf->ssim_4x4x2_core  = x264_pixel_ssim_4x4x2_core_mmxext;
 670         pixf->var2_8x8 = x264_pixel_var2_8x8_mmxext;
 671
 672         if( cpu&X264_CPU_CACHELINE_32 )
 673         {
 674             INIT5( sad, _cache32_mmxext );
 675             INIT4( sad_x3, _cache32_mmxext );
 676             INIT4( sad_x4, _cache32_mmxext );
 677         }
 678         else if( cpu&X264_CPU_CACHELINE_64 )
 679         {
 680             INIT5( sad, _cache64_mmxext );
 681             INIT4( sad_x3, _cache64_mmxext );
 682             INIT4( sad_x4, _cache64_mmxext );
 683         }
 684 #else
 685         if( cpu&X264_CPU_CACHELINE_64 )
 686         {
 687             pixf->sad[PIXEL_8x16] = x264_pixel_sad_8x16_cache64_mmxext;
 688             pixf->sad[PIXEL_8x8]  = x264_pixel_sad_8x8_cache64_mmxext;
 689             pixf->sad[PIXEL_8x4]  = x264_pixel_sad_8x4_cache64_mmxext;
 690             pixf->sad_x3[PIXEL_8x16] = x264_pixel_sad_x3_8x16_cache64_mmxext;
 691             pixf->sad_x3[PIXEL_8x8]  = x264_pixel_sad_x3_8x8_cache64_mmxext;
 692             pixf->sad_x4[PIXEL_8x16] = x264_pixel_sad_x4_8x16_cache64_mmxext;
 693             pixf->sad_x4[PIXEL_8x8]  = x264_pixel_sad_x4_8x8_cache64_mmxext;
 694         }
 695 #endif
 696         pixf->intra_satd_x3_16x16 = x264_intra_satd_x3_16x16_mmxext;
 697         pixf->intra_sad_x3_16x16  = x264_intra_sad_x3_16x16_mmxext;
 698         pixf->intra_satd_x3_8x8c  = x264_intra_satd_x3_8x8c_mmxext;
 699         pixf->intra_sad_x3_8x8c   = x264_intra_sad_x3_8x8c_mmxext;
 700         pixf->intra_sad_x3_8x8    = x264_intra_sad_x3_8x8_mmxext;
 701         pixf->intra_satd_x3_4x4   = x264_intra_satd_x3_4x4_mmxext;
 702         pixf->intra_sad_x3_4x4    = x264_intra_sad_x3_4x4_mmxext;
 703     }
 704
 705     if( cpu&X264_CPU_SSE2 )
 706     {
 707         INIT5( ssd, _sse2slow );
 708         INIT2_NAME( sad_aligned, sad, _sse2_aligned );
 709         pixf->var[PIXEL_16x16] = x264_pixel_var_16x16_sse2;
 710         pixf->ssim_4x4x2_core  = x264_pixel_ssim_4x4x2_core_sse2;
 711         pixf->ssim_end4        = x264_pixel_ssim_end4_sse2;
 712         pixf->sa8d[PIXEL_16x16] = x264_pixel_sa8d_16x16_sse2;
 713         pixf->sa8d[PIXEL_8x8]   = x264_pixel_sa8d_8x8_sse2;
 714 #ifdef ARCH_X86_64
 715         pixf->intra_sa8d_x3_8x8 = x264_intra_sa8d_x3_8x8_sse2;
 716 #endif
 717         pixf->var2_8x8 = x264_pixel_var2_8x8_sse2;
 718     }
 719
 720     if( (cpu&X264_CPU_SSE2) && !(cpu&X264_CPU_SSE2_IS_SLOW) )
 721     {
 722         INIT2( sad, _sse2 );
 723         INIT2( sad_x3, _sse2 );
 724         INIT2( sad_x4, _sse2 );
 725         INIT6( satd, _sse2 );
 726         INIT6( satd_x3, _sse2 );
 727         INIT6( satd_x4, _sse2 );
 728         if( !(cpu&X264_CPU_STACK_MOD4) )
 729         {
 730             INIT4( hadamard_ac, _sse2 );
 731         }
 732         INIT_ADS( _sse2 );
 733         pixf->var[PIXEL_8x8] = x264_pixel_var_8x8_sse2;
 734         pixf->intra_sad_x3_16x16 = x264_intra_sad_x3_16x16_sse2;
 735         if( cpu&X264_CPU_CACHELINE_64 )
 736         {
 737             INIT2( ssd, _sse2); /* faster for width 16 on p4 */
 738 #ifdef ARCH_X86
 739             INIT2( sad, _cache64_sse2 );
 740             INIT2( sad_x3, _cache64_sse2 );
 741             INIT2( sad_x4, _cache64_sse2 );
 742 #endif
 743            if( cpu&X264_CPU_SSE2_IS_FAST )
 744            {
 745                pixf->sad_x3[PIXEL_8x16] = x264_pixel_sad_x3_8x16_cache64_sse2;
 746                pixf->sad_x4[PIXEL_8x16] = x264_pixel_sad_x4_8x16_cache64_sse2;
 747            }
 748         }
 749
 750         if( cpu&X264_CPU_SSE_MISALIGN )
 751         {
 752             INIT2( sad_x3, _sse2_misalign );
 753             INIT2( sad_x4, _sse2_misalign );
 754         }
 755     }
 756
 757     if( cpu&X264_CPU_SSE2_IS_FAST && !(cpu&X264_CPU_CACHELINE_64) )
 758     {
 759         pixf->sad_aligned[PIXEL_8x16] = x264_pixel_sad_8x16_sse2;
 760         pixf->sad[PIXEL_8x16] = x264_pixel_sad_8x16_sse2;
 761         pixf->sad_x3[PIXEL_8x16] = x264_pixel_sad_x3_8x16_sse2;
 762         pixf->sad_x3[PIXEL_8x8] = x264_pixel_sad_x3_8x8_sse2;
 763         pixf->sad_x3[PIXEL_8x4] = x264_pixel_sad_x3_8x4_sse2;
 764         pixf->sad_x4[PIXEL_8x16] = x264_pixel_sad_x4_8x16_sse2;
 765         pixf->sad_x4[PIXEL_8x8] = x264_pixel_sad_x4_8x8_sse2;
 766         pixf->sad_x4[PIXEL_8x4] = x264_pixel_sad_x4_8x4_sse2;
 767     }
 768
 769     if( (cpu&X264_CPU_SSE3) && (cpu&X264_CPU_CACHELINE_64) )
 770     {
 771         INIT2( sad, _sse3 );
 772         INIT2( sad_x3, _sse3 );
 773         INIT2( sad_x4, _sse3 );
 774     }
 775
 776     if( cpu&X264_CPU_SSSE3 )
 777     {
 778         INIT7( ssd, _ssse3 );
 779         INIT7( satd, _ssse3 );
 780         INIT7( satd_x3, _ssse3 );
 781         INIT7( satd_x4, _ssse3 );
 782         if( !(cpu&X264_CPU_STACK_MOD4) )
 783         {
 784             INIT4( hadamard_ac, _ssse3 );
 785         }
 786         INIT_ADS( _ssse3 );
 787         pixf->sa8d[PIXEL_16x16]= x264_pixel_sa8d_16x16_ssse3;
 788         pixf->sa8d[PIXEL_8x8]  = x264_pixel_sa8d_8x8_ssse3;
 789         pixf->intra_satd_x3_16x16 = x264_intra_satd_x3_16x16_ssse3;
 790         pixf->intra_sad_x3_16x16  = x264_intra_sad_x3_16x16_ssse3;
 791         pixf->intra_satd_x3_8x8c  = x264_intra_satd_x3_8x8c_ssse3;
 792         pixf->intra_sad_x3_8x8c   = x264_intra_sad_x3_8x8c_ssse3;
 793         pixf->intra_satd_x3_4x4   = x264_intra_satd_x3_4x4_ssse3;
 794 #ifdef ARCH_X86_64
 795         pixf->intra_sa8d_x3_8x8 = x264_intra_sa8d_x3_8x8_ssse3;
 796 #endif
 797         pixf->var2_8x8 = x264_pixel_var2_8x8_ssse3;
 798         if( cpu&X264_CPU_CACHELINE_64 )
 799         {
 800             INIT2( sad, _cache64_ssse3 );
 801             INIT2( sad_x3, _cache64_ssse3 );
 802             INIT2( sad_x4, _cache64_ssse3 );
 803         }
 804         if( !(cpu&X264_CPU_SHUFFLE_IS_FAST) )
 805         {
 806             INIT5( ssd, _sse2 ); /* on conroe, sse2 is faster for width8/16 */
 807         }
 808     }
 809
 810     if( cpu&X264_CPU_SSE4 )
 811     {
 812         INIT7( satd, _sse4 );
 813         INIT7( satd_x3, _sse4 );
 814         INIT7( satd_x4, _sse4 );
 815         if( !(cpu&X264_CPU_STACK_MOD4) )
 816         {
 817             INIT4( hadamard_ac, _sse4 );
 818         }
 819         pixf->sa8d[PIXEL_16x16]= x264_pixel_sa8d_16x16_sse4;
 820         pixf->sa8d[PIXEL_8x8]  = x264_pixel_sa8d_8x8_sse4;
 821     }
 822 #endif //HAVE_MMX
 823
 824 #ifdef HAVE_ARMV6
 825     if( cpu&X264_CPU_ARMV6 )
 826     {
 827         pixf->sad[PIXEL_4x8] = x264_pixel_sad_4x8_armv6;
 828         pixf->sad[PIXEL_4x4] = x264_pixel_sad_4x4_armv6;
 829         pixf->sad_aligned[PIXEL_4x8] = x264_pixel_sad_4x8_armv6;
 830         pixf->sad_aligned[PIXEL_4x4] = x264_pixel_sad_4x4_armv6;
 831     }
 832     if( cpu&X264_CPU_NEON )
 833     {
 834         INIT5( sad, _neon );
 835         INIT5( sad_aligned, _neon );
 836         INIT7( sad_x3, _neon );
 837         INIT7( sad_x4, _neon );
 838         INIT7( ssd, _neon );
 839         INIT7( satd, _neon );
 840         INIT7( satd_x3, _neon );
 841         INIT7( satd_x4, _neon );
 842         INIT4( hadamard_ac, _neon );
 843         pixf->sa8d[PIXEL_8x8]   = x264_pixel_sa8d_8x8_neon;
 844         pixf->sa8d[PIXEL_16x16] = x264_pixel_sa8d_16x16_neon;
 845         pixf->var[PIXEL_8x8]    = x264_pixel_var_8x8_neon;
 846         pixf->var[PIXEL_16x16]  = x264_pixel_var_16x16_neon;
 847         pixf->var2_8x8          = x264_pixel_var2_8x8_neon;
 848
 849         pixf->ssim_4x4x2_core   = x264_pixel_ssim_4x4x2_core_neon;
 850         pixf->ssim_end4         = x264_pixel_ssim_end4_neon;
 851
 852         if( cpu&X264_CPU_FAST_NEON_MRC )
 853         {
 854             pixf->sad[PIXEL_4x8] = x264_pixel_sad_4x8_neon;
 855             pixf->sad[PIXEL_4x4] = x264_pixel_sad_4x4_neon;
 856             pixf->sad_aligned[PIXEL_4x8] = x264_pixel_sad_aligned_4x8_neon;
 857             pixf->sad_aligned[PIXEL_4x4] = x264_pixel_sad_aligned_4x4_neon;
 858         }
 859         else    // really just scheduled for dual issue / A8
 860         {
 861             INIT5( sad_aligned, _neon_dual );
 862         }
 863     }
 864 #endif
 865 #ifdef ARCH_PPC
 866     if( cpu&X264_CPU_ALTIVEC )
 867     {
 868         x264_pixel_altivec_init( pixf );
 869     }
 870 #endif
 871 #ifdef ARCH_UltraSparc
 872     INIT4( sad, _vis );
 873     INIT4( sad_x3, _vis );
 874     INIT4( sad_x4, _vis );
 875 #endif
 876
 877     pixf->ads[PIXEL_8x16] =
 878     pixf->ads[PIXEL_8x4] =
 879     pixf->ads[PIXEL_4x8] = pixf->ads[PIXEL_16x8];
 880     pixf->ads[PIXEL_4x4] = pixf->ads[PIXEL_8x8];
 881 }
 882