quant-a.S
上传用户:lctgjx
上传日期:2022-06-04
资源大小:8887k
文件大小:9k
源码类别:

流媒体/Mpeg4/MP4

开发平台:

Visual C++

  1. /*****************************************************************************
  2.  * quant.S: h264 encoder
  3.  *****************************************************************************
  4.  * Copyright (C) 2009 x264 project
  5.  *
  6.  * Authors: David Conrad <lessen42@gmail.com>
  7.  *
  8.  * This program is free software; you can redistribute it and/or modify
  9.  * it under the terms of the GNU General Public License as published by
  10.  * the Free Software Foundation; either version 2 of the License, or
  11.  * (at your option) any later version.
  12.  *
  13.  * This program is distributed in the hope that it will be useful,
  14.  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  15.  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  16.  * GNU General Public License for more details.
  17.  *
  18.  * You should have received a copy of the GNU General Public License
  19.  * along with this program; if not, write to the Free Software
  20.  * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA  02111, USA.
  21.  *****************************************************************************/
  22. #include "asm.S"
  23. .fpu neon
  24. .section .rodata
  25. .align 4
  26. pmovmskb_byte:
  27. .byte 1,2,4,8,16,32,64,128
  28. .byte 1,2,4,8,16,32,64,128
  29. .text
  30. .macro QUANT_TWO bias0 bias1 mf0 mf1 mf2 mf3 load_mf=no
  31.     vadd.u16    q8,  q8,  bias0
  32.     vadd.u16    q9,  q9,  bias1
  33. .ifc load_mf, yes
  34.     vld1.64     {mf0-mf3}, [r1,:128]!
  35. .endif
  36.     vmull.u16   q10, d16, mf0
  37.     vmull.u16   q11, d17, mf1
  38.     vmull.u16   q12, d18, mf2
  39.     vmull.u16   q13, d19, mf3
  40.     vshr.s16    q14, q14, #15
  41.     vshr.s16    q15, q15, #15
  42.     vshrn.u32   d16, q10, #16
  43.     vshrn.u32   d17, q11, #16
  44.     vshrn.u32   d18, q12, #16
  45.     vshrn.u32   d19, q13, #16
  46.     veor        q8,  q8,  q14
  47.     veor        q9,  q9,  q15
  48.     vsub.s16    q8,  q8,  q14
  49.     vsub.s16    q9,  q9,  q15
  50.     vorr        bias0, q8,  q9
  51.     vst1.64     {d16-d19}, [r0,:128]!
  52. .endm
  53. .macro QUANT_END d
  54.     vmov        r2,  r3,  d
  55.     orrs        r0,  r2,  r3
  56.     movne       r0,  #1
  57.     bx          lr
  58. .endm
  59. // quant_2x2_dc( int16_t dct[2][2], int mf, int bias )
  60. function x264_quant_2x2_dc_neon, export=1
  61.     vld1.64     {d0}, [r0,:64]
  62.     vabs.s16    d3,  d0
  63.     vdup.16     d2,  r2
  64.     vdup.16     d1,  r1
  65.     vadd.u16    d3,  d3,  d2
  66.     vmull.u16   q3,  d3,  d1
  67.     vshr.s16    d0,  d0,  #15
  68.     vshrn.u32   d3,  q3,  #16
  69.     veor        d3,  d3,  d0
  70.     vsub.s16    d3,  d3,  d0
  71.     vst1.64     {d3}, [r0,:64]
  72.     QUANT_END   d3
  73. .endfunc
  74. // quant_4x4_dc( int16_t dct[4][4], int mf, int bias )
  75. function x264_quant_4x4_dc_neon, export=1
  76.     vld1.64     {d28-d31}, [r0,:128]
  77.     vabs.s16    q8,  q14
  78.     vabs.s16    q9,  q15
  79.     vdup.16     q0,  r2
  80.     vdup.16     q2,  r1
  81.     QUANT_TWO   q0,  q0,  d4,  d5,  d4,  d5
  82.     vorr        d0,  d0,  d1
  83.     QUANT_END   d0
  84. .endfunc
  85. // quant_4x4( int16_t dct[4][4], uint16_t mf[16], uint16_t bias[16] )
  86. function x264_quant_4x4_neon, export=1
  87.     vld1.64     {d28-d31}, [r0,:128]
  88.     vabs.s16    q8,  q14
  89.     vabs.s16    q9,  q15
  90.     vld1.64     {d0-d3}, [r2,:128]
  91.     vld1.64     {d4-d7}, [r1,:128]
  92.     QUANT_TWO   q0,  q1,  d4,  d5,  d6,  d7
  93.     vorr        d0,  d0,  d1
  94.     QUANT_END   d0
  95. .endfunc
  96. // quant_8x8( int16_t dct[8][8], uint16_t mf[64], uint16_t bias[64] )
  97. function x264_quant_8x8_neon, export=1
  98.     vld1.64     {d28-d31}, [r0,:128]
  99.     vabs.s16    q8,  q14
  100.     vabs.s16    q9,  q15
  101.     vld1.64     {d0-d3},   [r2,:128]!
  102.     vld1.64     {d4-d7},   [r1,:128]!
  103.     QUANT_TWO   q0,  q1,  d4,  d5,  d6,  d7
  104. .rept 3
  105.     vld1.64     {d28-d31}, [r0,:128]
  106.     vabs.s16    q8,  q14
  107.     vabs.s16    q9,  q15
  108.     vld1.64     {d2-d5},   [r2,:128]!
  109.     QUANT_TWO   q1,  q2,  d4,  d5,  d6,  d7, yes
  110.     vorr        q0,  q0,  q1
  111. .endr
  112.     vorr        d0,  d0,  d1
  113.     QUANT_END   d0
  114. .endfunc
  115. .macro DEQUANT_START mf_size offset dc=no
  116.     mov         r3,  #0x2b
  117.     mul         r3,  r3,  r2
  118.     lsr         r3,  r3,  #8            // i_qbits = i_qp / 6
  119.     add         ip,  r3,  r3,  lsl #1
  120.     sub         r2,  r2,  ip,  lsl #1   // i_mf = i_qp % 6
  121. .ifc dc,no
  122.     add         r1,  r1,  r2, lsl #mf_size  // dequant_mf[i_mf]
  123. .else
  124.     ldr         r1, [r1,  r2, lsl #mf_size] // dequant_mf[i_mf][0][0]
  125. .endif
  126.     subs        r3,  r3,  #offset      // 6 for 8x8
  127. .endm
  128. // dequant_4x4( int16_t dct[4][4], int dequant_mf[6][4][4], int i_qp )
  129. .macro DEQUANT size bits
  130. function x264_dequant_size()_neon, export=1
  131.     DEQUANT_START bits+2, bits
  132. .ifc size, 8x8
  133.     mov         r2,  #4
  134. .endif
  135.     blt         dequant_size()_rshift
  136.     vdup.16     q15, r3
  137. dequant_size()_lshift_loop:
  138. .ifc size, 8x8
  139.     subs        r2,  r2,  #1
  140. .endif
  141.     vld1.32     {d16-d17}, [r1,:128]!
  142.     vld1.32     {d18-d19}, [r1,:128]!
  143.     vmovn.s32   d4,  q8
  144.     vld1.32     {d20-d21}, [r1,:128]!
  145.     vmovn.s32   d5,  q9
  146.     vld1.32     {d22-d23}, [r1,:128]!
  147.     vmovn.s32   d6,  q10
  148.     vld1.16     {d0-d3},   [r0,:128]
  149.     vmovn.s32   d7,  q11
  150.     vmul.s16    q0,  q0,  q2
  151.     vmul.s16    q1,  q1,  q3
  152.     vshl.s16    q0,  q0,  q15
  153.     vshl.s16    q1,  q1,  q15
  154.     vst1.16     {d0-d3},   [r0,:128]!
  155. .ifc size, 8x8
  156.     bgt         dequant_size()_lshift_loop
  157. .endif
  158.     bx          lr
  159. dequant_size()_rshift:
  160.     vdup.32     q15, r3
  161.     rsb         r3,  r3,  #0
  162.     mov         ip,  #1
  163.     sub         r3,  r3,  #1
  164.     lsl         ip,  ip,  r3
  165. .ifc size, 8x8
  166. dequant_size()_rshift_loop:
  167.     subs        r2,  r2,  #1
  168. .endif
  169.     vdup.32     q10, ip
  170.     vld1.32     {d16-d17}, [r1,:128]!
  171.     vdup.32     q11, ip
  172.     vld1.32     {d18-d19}, [r1,:128]!
  173.     vmovn.s32   d4,  q8
  174.     vld1.32     {d16-d17}, [r1,:128]!
  175.     vmovn.s32   d5,  q9
  176.     vld1.32     {d18-d19}, [r1,:128]!
  177.     vmovn.s32   d6,  q8
  178.     vld1.16     {d0-d3},   [r0,:128]
  179.     vmovn.s32   d7,  q9
  180.     vdup.32     q12, ip
  181.     vdup.32     q13, ip
  182.     vmlal.s16   q10, d0,  d4
  183.     vmlal.s16   q11, d1,  d5
  184.     vmlal.s16   q12, d2,  d6
  185.     vmlal.s16   q13, d3,  d7
  186.     vshl.s32    q10, q10, q15
  187.     vshl.s32    q11, q11, q15
  188.     vshl.s32    q12, q12, q15
  189.     vshl.s32    q13, q13, q15
  190.     vmovn.s32   d0,  q10
  191.     vmovn.s32   d1,  q11
  192.     vmovn.s32   d2,  q12
  193.     vmovn.s32   d3,  q13
  194.     vst1.16     {d0-d3},   [r0,:128]!
  195. .ifc size, 8x8
  196.     bgt         dequant_size()_rshift_loop
  197. .endif
  198.     bx          lr
  199. .endfunc
  200. .endm
  201. DEQUANT 4x4, 4
  202. DEQUANT 8x8, 6
  203. // dequant_4x4_dc( int16_t dct[4][4], int dequant_mf[6][4][4], int i_qp )
  204. function x264_dequant_4x4_dc_neon, export=1
  205.     DEQUANT_START 6, 6, yes
  206.     blt         dequant_4x4_dc_rshift
  207.     lsl         r1,  r1,  r3
  208.     vdup.16     q2,  r1
  209.     vld1.16     {d0-d3},   [r0,:128]
  210.     vdup.16     q15, r3
  211.     vmul.s16    q0,  q0,  q2
  212.     vmul.s16    q1,  q1,  q2
  213.     vst1.16     {d0-d3},   [r0,:128]
  214.     bx          lr
  215. dequant_4x4_dc_rshift:
  216.     vdup.16     d4,  r1
  217.     vdup.32     q15, r3
  218.     rsb         r3,  r3,  #0
  219.     mov         ip,  #1
  220.     sub         r3,  r3,  #1
  221.     lsl         ip,  ip,  r3
  222.     vdup.32     q10, ip
  223.     vdup.32     q11, ip
  224.     vld1.16     {d0-d3},   [r0,:128]
  225.     vdup.32     q12, ip
  226.     vdup.32     q13, ip
  227.     vmlal.s16   q10, d0,  d4
  228.     vmlal.s16   q11, d1,  d4
  229.     vmlal.s16   q12, d2,  d4
  230.     vmlal.s16   q13, d3,  d4
  231.     vshl.s32    q10, q10, q15
  232.     vshl.s32    q11, q11, q15
  233.     vshl.s32    q12, q12, q15
  234.     vshl.s32    q13, q13, q15
  235.     vmovn.s32   d0,  q10
  236.     vmovn.s32   d1,  q11
  237.     vmovn.s32   d2,  q12
  238.     vmovn.s32   d3,  q13
  239.     vst1.16     {d0-d3},   [r0,:128]
  240.     bx          lr
  241. .endfunc
  242. // int coeff_last( int16_t *l )
  243. function x264_coeff_last4_arm, export=1
  244.     ldrd        r2,  [r0]
  245.     subs        r0,  r3,  #0
  246.     movne       r0,  #2
  247.     movne       r2,  r3
  248.     lsrs        r2,  r2,  #16
  249.     addne       r0,  r0,  #1
  250.     bx          lr
  251. .endfunc
  252. .macro COEFF_LAST_1x size
  253. function x264_coeff_lastsize()_neon, export=1
  254. .if size == 15
  255.     sub         r0,  r0,  #2
  256.     vld1.64     {d0-d3}, [r0]
  257. .else
  258.     vld1.64     {d0-d3}, [r0,:128]
  259. .endif
  260.     vtst.16     q0,  q0
  261.     vtst.16     q1,  q1
  262.     vshrn.u16   d0,  q0,  #8
  263.     vshrn.u16   d1,  q1,  #8
  264.     vshrn.u16   d0,  q0,  #4
  265.     vclz.i32    d0,  d0
  266.     mov         ip,  #7
  267.     mov         r3,  #size - 9
  268.     vmov        r0,  r1,  d0
  269.     subs        r1,  ip,  r1,  lsr #2
  270.     addge       r0,  r1,  #size - 8
  271.     sublts      r0,  r3,  r0,  lsr #2
  272.     movlt       r0,  #0
  273.     bx          lr
  274. .endfunc
  275. .endm
  276. COEFF_LAST_1x 15
  277. COEFF_LAST_1x 16
  278. function x264_coeff_last64_neon, export=1
  279.     vld1.64     {d16-d19}, [r0,:128]!
  280.     vqmovn.u16  d16, q8
  281.     vqmovn.u16  d17, q9
  282.     vld1.64     {d20-d23}, [r0,:128]!
  283.     vqmovn.u16  d18, q10
  284.     vqmovn.u16  d19, q11
  285.     vld1.64     {d24-d27}, [r0,:128]!
  286.     vqmovn.u16  d20, q12
  287.     vqmovn.u16  d21, q13
  288.     vld1.64     {d28-d31}, [r0,:128]!
  289.     vqmovn.u16  d22, q14
  290.     vqmovn.u16  d23, q15
  291.     movrel      r1, pmovmskb_byte
  292.     vld1.64     {d0-d1}, [r1,:128]
  293.     vtst.8      q8,  q8
  294.     vtst.8      q9,  q9
  295.     vtst.8      q10, q10
  296.     vtst.8      q11, q11
  297.     vand        q8,  q8,  q0
  298.     vand        q9,  q9,  q0
  299.     vand        q10, q10, q0
  300.     vand        q11, q11, q0
  301.     vpadd.u8    d0,  d16, d17
  302.     vpadd.u8    d1,  d18, d19
  303.     vpadd.u8    d2,  d20, d21
  304.     vpadd.u8    d3,  d22, d23
  305.     vpadd.u8    d0,  d0,  d1
  306.     vpadd.u8    d1,  d2,  d3
  307.     vpadd.u8    d0,  d0,  d1
  308.     vclz.i32    d0,  d0
  309.     mov         ip,  #31
  310.     vmov        r0,  r1,  d0
  311.     subs        r1,  ip,  r1
  312.     addge       r0,  r1,  #32
  313.     sublts      r0,  ip,  r0
  314.     movlt       r0,  #0
  315.     bx          lr
  316. .endfunc