pixel-32.asm
上传用户:lctgjx
上传日期:2022-06-04
资源大小:8887k
文件大小:10k
源码类别:

流媒体/Mpeg4/MP4

开发平台:

Visual C++

  1. ;*****************************************************************************
  2. ;* pixel-32.asm: h264 encoder library
  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. %include "x86inc.asm"
  24. %include "x86util.asm"
  25. SECTION .text
  26. INIT_MMX
  27. %macro LOAD_DIFF_4x8P 1 ; dx
  28.     LOAD_DIFF  m0, m7, none, [r0+%1],      [r2+%1]
  29.     LOAD_DIFF  m1, m6, none, [r0+%1+r1],   [r2+%1+r3]
  30.     LOAD_DIFF  m2, m7, none, [r0+%1+r1*2], [r2+%1+r3*2]
  31.     LOAD_DIFF  m3, m6, none, [r0+%1+r4],   [r2+%1+r5]
  32.     lea  r0, [r0+4*r1]
  33.     lea  r2, [r2+4*r3]
  34.     LOAD_DIFF  m4, m7, none, [r0+%1],      [r2+%1]
  35.     LOAD_DIFF  m5, m6, none, [r0+%1+r1],   [r2+%1+r3]
  36.     LOAD_DIFF  m6, m7, none, [r0+%1+r1*2], [r2+%1+r3*2]
  37.     movq [spill], m5
  38.     LOAD_DIFF  m7, m5, none, [r0+%1+r4],   [r2+%1+r5]
  39.     movq m5, [spill]
  40. %endmacro
  41. %macro SUM4x8_MM 0
  42.     movq [spill],   m6
  43.     movq [spill+8], m7
  44.     ABS2     m0, m1, m6, m7
  45.     ABS2     m2, m3, m6, m7
  46.     paddw    m0, m2
  47.     paddw    m1, m3
  48.     movq     m6, [spill]
  49.     movq     m7, [spill+8]
  50.     ABS2     m4, m5, m2, m3
  51.     ABS2     m6, m7, m2, m3
  52.     paddw    m4, m6
  53.     paddw    m5, m7
  54.     paddw    m0, m4
  55.     paddw    m1, m5
  56.     paddw    m0, m1
  57. %endmacro
  58. ;-----------------------------------------------------------------------------
  59. ; int x264_pixel_sa8d_8x8_mmxext( uint8_t *, int, uint8_t *, int )
  60. ;-----------------------------------------------------------------------------
  61. cglobal x264_pixel_sa8d_8x8_internal_mmxext
  62.     push   r0
  63.     push   r2
  64.     sub    esp, 0x74
  65. %define args  esp+0x74
  66. %define spill esp+0x60 ; +16
  67. %define trans esp+0    ; +96
  68.     LOAD_DIFF_4x8P 0
  69.     HADAMARD8_V m0, m1, m2, m3, m4, m5, m6, m7
  70.     movq   [spill], m1
  71.     TRANSPOSE4x4W 4, 5, 6, 7, 1
  72.     movq   [trans+0x00], m4
  73.     movq   [trans+0x08], m5
  74.     movq   [trans+0x10], m6
  75.     movq   [trans+0x18], m7
  76.     movq   m1, [spill]
  77.     TRANSPOSE4x4W 0, 1, 2, 3, 4
  78.     movq   [trans+0x20], m0
  79.     movq   [trans+0x28], m1
  80.     movq   [trans+0x30], m2
  81.     movq   [trans+0x38], m3
  82.     mov    r0, [args+4]
  83.     mov    r2, [args]
  84.     LOAD_DIFF_4x8P 4
  85.     HADAMARD8_V m0, m1, m2, m3, m4, m5, m6, m7
  86.     movq   [spill], m7
  87.     TRANSPOSE4x4W 0, 1, 2, 3, 7
  88.     movq   [trans+0x40], m0
  89.     movq   [trans+0x48], m1
  90.     movq   [trans+0x50], m2
  91.     movq   [trans+0x58], m3
  92.     movq   m7, [spill]
  93.     TRANSPOSE4x4W 4, 5, 6, 7, 1
  94.     movq   m0, [trans+0x00]
  95.     movq   m1, [trans+0x08]
  96.     movq   m2, [trans+0x10]
  97.     movq   m3, [trans+0x18]
  98.     HADAMARD8_V m0, m1, m2, m3, m4, m5, m6, m7
  99.     SUM4x8_MM
  100.     movq   [trans], m0
  101.     movq   m0, [trans+0x20]
  102.     movq   m1, [trans+0x28]
  103.     movq   m2, [trans+0x30]
  104.     movq   m3, [trans+0x38]
  105.     movq   m4, [trans+0x40]
  106.     movq   m5, [trans+0x48]
  107.     movq   m6, [trans+0x50]
  108.     movq   m7, [trans+0x58]
  109.     HADAMARD8_V m0, m1, m2, m3, m4, m5, m6, m7
  110.     SUM4x8_MM
  111.     pavgw  m0, [trans]
  112.     add   esp, 0x7c
  113.     ret
  114. %undef args
  115. %undef spill
  116. %undef trans
  117. %macro SUM_MM_X3 8 ; 3x sum, 4x tmp, op
  118.     pxor        %7, %7
  119.     pshufw      %4, %1, 01001110b
  120.     pshufw      %5, %2, 01001110b
  121.     pshufw      %6, %3, 01001110b
  122.     paddusw     %1, %4
  123.     paddusw     %2, %5
  124.     paddusw     %3, %6
  125.     punpcklwd   %1, %7
  126.     punpcklwd   %2, %7
  127.     punpcklwd   %3, %7
  128.     pshufw      %4, %1, 01001110b
  129.     pshufw      %5, %2, 01001110b
  130.     pshufw      %6, %3, 01001110b
  131.     %8          %1, %4
  132.     %8          %2, %5
  133.     %8          %3, %6
  134. %endmacro
  135. %macro LOAD_4x8P 1 ; dx
  136.     pxor        m7, m7
  137.     movd        m6, [eax+%1+7*FENC_STRIDE]
  138.     movd        m0, [eax+%1+0*FENC_STRIDE]
  139.     movd        m1, [eax+%1+1*FENC_STRIDE]
  140.     movd        m2, [eax+%1+2*FENC_STRIDE]
  141.     movd        m3, [eax+%1+3*FENC_STRIDE]
  142.     movd        m4, [eax+%1+4*FENC_STRIDE]
  143.     movd        m5, [eax+%1+5*FENC_STRIDE]
  144.     punpcklbw   m6, m7
  145.     punpcklbw   m0, m7
  146.     punpcklbw   m1, m7
  147.     movq   [spill], m6
  148.     punpcklbw   m2, m7
  149.     punpcklbw   m3, m7
  150.     movd        m6, [eax+%1+6*FENC_STRIDE]
  151.     punpcklbw   m4, m7
  152.     punpcklbw   m5, m7
  153.     punpcklbw   m6, m7
  154.     movq        m7, [spill]
  155. %endmacro
  156. ;-----------------------------------------------------------------------------
  157. ; void x264_intra_sa8d_x3_8x8_core_mmxext( uint8_t *fenc, int16_t edges[2][8], int *res )
  158. ;-----------------------------------------------------------------------------
  159. cglobal x264_intra_sa8d_x3_8x8_core_mmxext
  160.     mov    eax, [esp+4]
  161.     mov    ecx, [esp+8]
  162.     sub    esp, 0x70
  163. %define args  esp+0x74
  164. %define spill esp+0x60 ; +16
  165. %define trans esp+0    ; +96
  166. %define sum   esp+0    ; +32
  167.     LOAD_4x8P 0
  168.     HADAMARD8_V m0, m1, m2, m3, m4, m5, m6, m7
  169.     movq   [spill], m0
  170.     TRANSPOSE4x4W 4, 5, 6, 7, 0
  171.     movq   [trans+0x00], m4
  172.     movq   [trans+0x08], m5
  173.     movq   [trans+0x10], m6
  174.     movq   [trans+0x18], m7
  175.     movq   m0, [spill]
  176.     TRANSPOSE4x4W 0, 1, 2, 3, 4
  177.     movq   [trans+0x20], m0
  178.     movq   [trans+0x28], m1
  179.     movq   [trans+0x30], m2
  180.     movq   [trans+0x38], m3
  181.     LOAD_4x8P 4
  182.     HADAMARD8_V m0, m1, m2, m3, m4, m5, m6, m7
  183.     movq   [spill], m7
  184.     TRANSPOSE4x4W 0, 1, 2, 3, 7
  185.     movq   [trans+0x40], m0
  186.     movq   [trans+0x48], m1
  187.     movq   [trans+0x50], m2
  188.     movq   [trans+0x58], m3
  189.     movq   m7, [spill]
  190.     TRANSPOSE4x4W 4, 5, 6, 7, 0
  191.     movq   m0, [trans+0x00]
  192.     movq   m1, [trans+0x08]
  193.     movq   m2, [trans+0x10]
  194.     movq   m3, [trans+0x18]
  195.     HADAMARD8_V m0, m1, m2, m3, m4, m5, m6, m7
  196.     movq [spill+0], m0
  197.     movq [spill+8], m1
  198.     ABS2     m2, m3, m0, m1
  199.     ABS2     m4, m5, m0, m1
  200.     paddw    m2, m4
  201.     paddw    m3, m5
  202.     ABS2     m6, m7, m4, m5
  203.     movq     m0, [spill+0]
  204.     movq     m1, [spill+8]
  205.     paddw    m2, m6
  206.     paddw    m3, m7
  207.     paddw    m2, m3
  208.     ABS1     m1, m4
  209.     paddw    m2, m1 ; 7x4 sum
  210.     movq     m7, m0
  211.     movq     m1, [ecx+8] ; left bottom
  212.     psllw    m1, 3
  213.     psubw    m7, m1
  214.     ABS2     m0, m7, m5, m3
  215.     paddw    m0, m2
  216.     paddw    m7, m2
  217.     movq [sum+0], m0 ; dc
  218.     movq [sum+8], m7 ; left
  219.     movq   m0, [trans+0x20]
  220.     movq   m1, [trans+0x28]
  221.     movq   m2, [trans+0x30]
  222.     movq   m3, [trans+0x38]
  223.     movq   m4, [trans+0x40]
  224.     movq   m5, [trans+0x48]
  225.     movq   m6, [trans+0x50]
  226.     movq   m7, [trans+0x58]
  227.     HADAMARD8_V m0, m1, m2, m3, m4, m5, m6, m7
  228.     movd   [sum+0x10], m0
  229.     movd   [sum+0x12], m1
  230.     movd   [sum+0x14], m2
  231.     movd   [sum+0x16], m3
  232.     movd   [sum+0x18], m4
  233.     movd   [sum+0x1a], m5
  234.     movd   [sum+0x1c], m6
  235.     movd   [sum+0x1e], m7
  236.     movq [spill],   m0
  237.     movq [spill+8], m1
  238.     ABS2     m2, m3, m0, m1
  239.     ABS2     m4, m5, m0, m1
  240.     paddw    m2, m4
  241.     paddw    m3, m5
  242.     paddw    m2, m3
  243.     movq     m0, [spill]
  244.     movq     m1, [spill+8]
  245.     ABS2     m6, m7, m4, m5
  246.     ABS1     m1, m3
  247.     paddw    m2, m7
  248.     paddw    m1, m6
  249.     paddw    m2, m1 ; 7x4 sum
  250.     movq     m1, m0
  251.     movq     m7, [ecx+0]
  252.     psllw    m7, 3   ; left top
  253.     movzx   edx, word [ecx+0]
  254.     add      dx,  [ecx+16]
  255.     lea     edx, [4*edx+32]
  256.     and     edx, -64
  257.     movd     m6, edx ; dc
  258.     psubw    m1, m7
  259.     psubw    m0, m6
  260.     ABS2     m0, m1, m5, m6
  261.     movq     m3, [sum+0] ; dc
  262.     paddw    m0, m2
  263.     paddw    m1, m2
  264.     movq     m2, m0
  265.     paddw    m0, m3
  266.     paddw    m1, [sum+8] ; h
  267.     psrlq    m2, 16
  268.     paddw    m2, m3
  269.     movq     m3, [ecx+16] ; top left
  270.     movq     m4, [ecx+24] ; top right
  271.     psllw    m3, 3
  272.     psllw    m4, 3
  273.     psubw    m3, [sum+16]
  274.     psubw    m4, [sum+24]
  275.     ABS2     m3, m4, m5, m6
  276.     paddw    m2, m3
  277.     paddw    m2, m4 ; v
  278.     SUM_MM_X3 m0, m1, m2, m3, m4, m5, m6, paddd
  279.     mov     eax, [args+8]
  280.     movd    ecx, m2
  281.     movd    edx, m1
  282.     add     ecx, 2
  283.     add     edx, 2
  284.     shr     ecx, 2
  285.     shr     edx, 2
  286.     mov [eax+0], ecx ; i8x8_v satd
  287.     mov [eax+4], edx ; i8x8_h satd
  288.     movd    ecx, m0
  289.     add     ecx, 2
  290.     shr     ecx, 2
  291.     mov [eax+8], ecx ; i8x8_dc satd
  292.     add     esp, 0x70
  293.     ret
  294. %undef args
  295. %undef spill
  296. %undef trans
  297. %undef sum
  298. ;-----------------------------------------------------------------------------
  299. ; void x264_pixel_ssim_4x4x2_core_mmxext( const uint8_t *pix1, int stride1,
  300. ;                                         const uint8_t *pix2, int stride2, int sums[2][4] )
  301. ;-----------------------------------------------------------------------------
  302. cglobal x264_pixel_ssim_4x4x2_core_mmxext
  303.     push     ebx
  304.     push     edi
  305.     mov      ebx, [esp+16]
  306.     mov      edx, [esp+24]
  307.     mov      edi, 4
  308.     pxor      m0, m0
  309. .loop:
  310.     mov      eax, [esp+12]
  311.     mov      ecx, [esp+20]
  312.     add      eax, edi
  313.     add      ecx, edi
  314.     pxor      m1, m1
  315.     pxor      m2, m2
  316.     pxor      m3, m3
  317.     pxor      m4, m4
  318. %rep 4
  319.     movd      m5, [eax]
  320.     movd      m6, [ecx]
  321.     punpcklbw m5, m0
  322.     punpcklbw m6, m0
  323.     paddw     m1, m5
  324.     paddw     m2, m6
  325.     movq      m7, m5
  326.     pmaddwd   m5, m5
  327.     pmaddwd   m7, m6
  328.     pmaddwd   m6, m6
  329.     paddd     m3, m5
  330.     paddd     m4, m7
  331.     paddd     m3, m6
  332.     add      eax, ebx
  333.     add      ecx, edx
  334. %endrep
  335.     mov      eax, [esp+28]
  336.     lea      eax, [eax+edi*4]
  337.     pshufw    m5, m1, 0xE
  338.     pshufw    m6, m2, 0xE
  339.     paddusw   m1, m5
  340.     paddusw   m2, m6
  341.     punpcklwd m1, m2
  342.     pshufw    m2, m1, 0xE
  343.     pshufw    m5, m3, 0xE
  344.     pshufw    m6, m4, 0xE
  345.     paddusw   m1, m2
  346.     paddd     m3, m5
  347.     paddd     m4, m6
  348.     punpcklwd m1, m0
  349.     punpckldq m3, m4
  350.     movq [eax+0], m1
  351.     movq [eax+8], m3
  352.     sub      edi, 4
  353.     jge .loop
  354.     pop       edi
  355.     pop       ebx
  356.     emms
  357.     ret