pixel-32.asm
上传用户:hjq518
上传日期:2021-12-09
资源大小:5084k
文件大小:11k
源码类别:

Audio

开发平台:

Visual C++

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