jcgryext-avx2.asm 19 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436
  1. ;
  2. ; Grayscale colorspace conversion (64-bit AVX2)
  3. ;
  4. ; Copyright (C) 2011, 2016, 2024, D. R. Commander.
  5. ; Copyright (C) 2015, Intel Corporation.
  6. ; Copyright (C) 2018, Matthias Räncker.
  7. ; Copyright (C) 2023, Aliaksiej Kandracienka.
  8. ;
  9. ; Based on the x86 SIMD extension for IJG JPEG library
  10. ; Copyright (C) 1999-2006, MIYASAKA Masaru.
  11. ; For conditions of distribution and use, see copyright notice in jsimdext.inc
  12. ;
  13. ; This file should be assembled with NASM (Netwide Assembler) or Yasm.
  14. %include "jcolsamp.inc"
  15. ; --------------------------------------------------------------------------
  16. ;
  17. ; Convert some rows of samples to the output colorspace.
  18. ;
  19. ; GLOBAL(void)
  20. ; jsimd_rgb_gray_convert_avx2(JDIMENSION img_width, JSAMPARRAY input_buf,
  21. ; JSAMPIMAGE output_buf, JDIMENSION output_row,
  22. ; int num_rows);
  23. ;
  24. ; r10d = JDIMENSION img_width
  25. ; r11 = JSAMPARRAY input_buf
  26. ; r12 = JSAMPIMAGE output_buf
  27. ; r13d = JDIMENSION output_row
  28. ; r14d = int num_rows
  29. %define wk(i) r15 - (WK_NUM - (i)) * SIZEOF_YMMWORD ; ymmword wk[WK_NUM]
  30. %define WK_NUM 2
  31. align 32
  32. GLOBAL_FUNCTION(jsimd_rgb_gray_convert_avx2)
  33. EXTN(jsimd_rgb_gray_convert_avx2):
  34. ENDBR64
  35. push rbp
  36. mov rbp, rsp
  37. push r15
  38. and rsp, byte (-SIZEOF_YMMWORD) ; align to 256 bits
  39. ; Allocate stack space for wk array. r15 is used to access it.
  40. mov r15, rsp
  41. sub rsp, byte (SIZEOF_YMMWORD * WK_NUM)
  42. COLLECT_ARGS 5
  43. push rbx
  44. mov ecx, r10d
  45. test rcx, rcx
  46. jz near .return
  47. push rcx
  48. mov rsi, r12
  49. mov ecx, r13d
  50. mov rdip, JSAMPARRAY [rsi+0*SIZEOF_JSAMPARRAY]
  51. lea rdi, [rdi+rcx*SIZEOF_JSAMPROW]
  52. pop rcx
  53. mov rsi, r11
  54. mov eax, r14d
  55. test rax, rax
  56. jle near .return
  57. .rowloop:
  58. push rdi
  59. push rsi
  60. push rcx ; col
  61. mov rsip, JSAMPROW [rsi] ; inptr
  62. mov rdip, JSAMPROW [rdi] ; outptr0
  63. cmp rcx, byte SIZEOF_YMMWORD
  64. jae near .columnloop
  65. %if RGB_PIXELSIZE == 3 ; ---------------
  66. .column_ld1:
  67. push rax
  68. push rdx
  69. lea rcx, [rcx+rcx*2] ; imul ecx,RGB_PIXELSIZE
  70. test cl, SIZEOF_BYTE
  71. jz short .column_ld2
  72. sub rcx, byte SIZEOF_BYTE
  73. movzx rax, byte [rsi+rcx]
  74. .column_ld2:
  75. test cl, SIZEOF_WORD
  76. jz short .column_ld4
  77. sub rcx, byte SIZEOF_WORD
  78. movzx rdx, word [rsi+rcx]
  79. shl rax, WORD_BIT
  80. or rax, rdx
  81. .column_ld4:
  82. vmovd xmmA, eax
  83. pop rdx
  84. pop rax
  85. test cl, SIZEOF_DWORD
  86. jz short .column_ld8
  87. sub rcx, byte SIZEOF_DWORD
  88. vmovd xmmF, XMM_DWORD [rsi+rcx]
  89. vpslldq xmmA, xmmA, SIZEOF_DWORD
  90. vpor xmmA, xmmA, xmmF
  91. .column_ld8:
  92. test cl, SIZEOF_MMWORD
  93. jz short .column_ld16
  94. sub rcx, byte SIZEOF_MMWORD
  95. vmovq xmmB, XMM_MMWORD [rsi+rcx]
  96. vpslldq xmmA, xmmA, SIZEOF_MMWORD
  97. vpor xmmA, xmmA, xmmB
  98. .column_ld16:
  99. test cl, SIZEOF_XMMWORD
  100. jz short .column_ld32
  101. sub rcx, byte SIZEOF_XMMWORD
  102. vmovdqu xmmB, XMM_MMWORD [rsi+rcx]
  103. vperm2i128 ymmA, ymmA, ymmA, 1
  104. vpor ymmA, ymmB
  105. .column_ld32:
  106. test cl, SIZEOF_YMMWORD
  107. jz short .column_ld64
  108. sub rcx, byte SIZEOF_YMMWORD
  109. vmovdqa ymmF, ymmA
  110. vmovdqu ymmA, YMMWORD [rsi+0*SIZEOF_YMMWORD]
  111. .column_ld64:
  112. test cl, 2*SIZEOF_YMMWORD
  113. mov rcx, SIZEOF_YMMWORD
  114. jz short .rgb_gray_cnv
  115. vmovdqa ymmB, ymmA
  116. vmovdqu ymmA, YMMWORD [rsi+0*SIZEOF_YMMWORD]
  117. vmovdqu ymmF, YMMWORD [rsi+1*SIZEOF_YMMWORD]
  118. jmp short .rgb_gray_cnv
  119. .columnloop:
  120. vmovdqu ymmA, YMMWORD [rsi+0*SIZEOF_YMMWORD]
  121. vmovdqu ymmF, YMMWORD [rsi+1*SIZEOF_YMMWORD]
  122. vmovdqu ymmB, YMMWORD [rsi+2*SIZEOF_YMMWORD]
  123. .rgb_gray_cnv:
  124. ; ymmA=(00 10 20 01 11 21 02 12 22 03 13 23 04 14 24 05
  125. ; 15 25 06 16 26 07 17 27 08 18 28 09 19 29 0A 1A)
  126. ; ymmF=(2A 0B 1B 2B 0C 1C 2C 0D 1D 2D 0E 1E 2E 0F 1F 2F
  127. ; 0G 1G 2G 0H 1H 2H 0I 1I 2I 0J 1J 2J 0K 1K 2K 0L)
  128. ; ymmB=(1L 2L 0M 1M 2M 0N 1N 2N 0O 1O 2O 0P 1P 2P 0Q 1Q
  129. ; 2Q 0R 1R 2R 0S 1S 2S 0T 1T 2T 0U 1U 2U 0V 1V 2V)
  130. vmovdqu ymmC, ymmA
  131. vinserti128 ymmA, ymmF, xmmA, 0 ; ymmA=(00 10 20 01 11 21 02 12 22 03 13 23 04 14 24 05
  132. ; 0G 1G 2G 0H 1H 2H 0I 1I 2I 0J 1J 2J 0K 1K 2K 0L)
  133. vinserti128 ymmC, ymmC, xmmB, 0 ; ymmC=(1L 2L 0M 1M 2M 0N 1N 2N 0O 1O 2O 0P 1P 2P 0Q 1Q
  134. ; 15 25 06 16 26 07 17 27 08 18 28 09 19 29 0A 1A)
  135. vinserti128 ymmB, ymmB, xmmF, 0 ; ymmB=(2A 0B 1B 2B 0C 1C 2C 0D 1D 2D 0E 1E 2E 0F 1F 2F
  136. ; 2Q 0R 1R 2R 0S 1S 2S 0T 1T 2T 0U 1U 2U 0V 1V 2V)
  137. vperm2i128 ymmF, ymmC, ymmC, 1 ; ymmF=(15 25 06 16 26 07 17 27 08 18 28 09 19 29 0A 1A
  138. ; 1L 2L 0M 1M 2M 0N 1N 2N 0O 1O 2O 0P 1P 2P 0Q 1Q)
  139. vmovdqa ymmG, ymmA
  140. vpslldq ymmA, ymmA, 8 ; ymmA=(-- -- -- -- -- -- -- -- 00 10 20 01 11 21 02 12
  141. ; 22 03 13 23 04 14 24 05 0G 1G 2G 0H 1H 2H 0I 1I)
  142. vpsrldq ymmG, ymmG, 8 ; ymmG=(22 03 13 23 04 14 24 05 0G 1G 2G 0H 1H 2H 0I 1I
  143. ; 2I 0J 1J 2J 0K 1K 2K 0L -- -- -- -- -- -- -- --)
  144. vpunpckhbw ymmA, ymmA, ymmF ; ymmA=(00 08 10 18 20 28 01 09 11 19 21 29 02 0A 12 1A
  145. ; 0G 0O 1G 1O 2G 2O 0H 0P 1H 1P 2H 2P 0I 0Q 1I 1Q)
  146. vpslldq ymmF, ymmF, 8 ; ymmF=(-- -- -- -- -- -- -- -- 15 25 06 16 26 07 17 27
  147. ; 08 18 28 09 19 29 0A 1A 1L 2L 0M 1M 2M 0N 1N 2N)
  148. vpunpcklbw ymmG, ymmG, ymmB ; ymmG=(22 2A 03 0B 13 1B 23 2B 04 0C 14 1C 24 2C 05 0D
  149. ; 2I 2Q 0J 0R 1J 1R 2J 2R 0K 0S 1K 1S 2K 2S 0L 0T)
  150. vpunpckhbw ymmF, ymmF, ymmB ; ymmF=(15 1D 25 2D 06 0E 16 1E 26 2E 07 0F 17 1F 27 2F
  151. ; 1L 1T 2L 2T 0M 0U 1M 1U 2M 2U 0N 0V 1N 1V 2N 2V)
  152. vmovdqa ymmD, ymmA
  153. vpslldq ymmA, ymmA, 8 ; ymmA=(-- -- -- -- -- -- -- -- 00 08 10 18 20 28 01 09
  154. ; 11 19 21 29 02 0A 12 1A 0G 0O 1G 1O 2G 2O 0H 0P)
  155. vpsrldq ymmD, ymmD, 8 ; ymmD=(11 19 21 29 02 0A 12 1A 0G 0O 1G 1O 2G 2O 0H 0P
  156. ; 1H 1P 2H 2P 0I 0Q 1I 1Q -- -- -- -- -- -- -- --)
  157. vpunpckhbw ymmA, ymmA, ymmG ; ymmA=(00 04 08 0C 10 14 18 1C 20 24 28 2C 01 05 09 0D
  158. ; 0G 0K 0O 0S 1G 1K 1O 1S 2G 2K 2O 2S 0H 0L 0P 0T)
  159. vpslldq ymmG, ymmG, 8 ; ymmG=(-- -- -- -- -- -- -- -- 22 2A 03 0B 13 1B 23 2B
  160. ; 04 0C 14 1C 24 2C 05 0D 2I 2Q 0J 0R 1J 1R 2J 2R)
  161. vpunpcklbw ymmD, ymmD, ymmF ; ymmD=(11 15 19 1D 21 25 29 2D 02 06 0A 0E 12 16 1A 1E
  162. ; 1H 1L 1P 1T 2H 2L 2P 2T 0I 0M 0Q 0U 1I 1M 1Q 1U)
  163. vpunpckhbw ymmG, ymmG, ymmF ; ymmG=(22 26 2A 2E 03 07 0B 0F 13 17 1B 1F 23 27 2B 2F
  164. ; 2I 2M 2Q 2U 0J 0N 0R 0V 1J 1N 1R 1V 2J 2N 2R 2V)
  165. vmovdqa ymmE, ymmA
  166. vpslldq ymmA, ymmA, 8 ; ymmA=(-- -- -- -- -- -- -- -- 00 04 08 0C 10 14 18 1C
  167. ; 20 24 28 2C 01 05 09 0D 0G 0K 0O 0S 1G 1K 1O 1S)
  168. vpsrldq ymmE, ymmE, 8 ; ymmE=(20 24 28 2C 01 05 09 0D 0G 0K 0O 0S 1G 1K 1O 1S
  169. ; 2G 2K 2O 2S 0H 0L 0P 0T -- -- -- -- -- -- -- --)
  170. vpunpckhbw ymmA, ymmA, ymmD ; ymmA=(00 02 04 06 08 0A 0C 0E 10 12 14 16 18 1A 1C 1E
  171. ; 0G 0I 0K 0M 0O 0Q 0S 0U 1G 1I 1K 1M 1O 1Q 1S 1U)
  172. vpslldq ymmD, ymmD, 8 ; ymmD=(-- -- -- -- -- -- -- -- 11 15 19 1D 21 25 29 2D
  173. ; 02 06 0A 0E 12 16 1A 1E 1H 1L 1P 1T 2H 2L 2P 2T)
  174. vpunpcklbw ymmE, ymmE, ymmG ; ymmE=(20 22 24 26 28 2A 2C 2E 01 03 05 07 09 0B 0D 0F
  175. ; 2G 2I 2K 2M 2O 2Q 2S 2U 0H 0J 0L 0N 0P 0R 0T 0V)
  176. vpunpckhbw ymmD, ymmD, ymmG ; ymmD=(11 13 15 17 19 1B 1D 1F 21 23 25 27 29 2B 2D 2F
  177. ; 1H 1J 1L 1N 1P 1R 1T 1V 2H 2J 2L 2N 2P 2R 2T 2V)
  178. vpxor ymmH, ymmH, ymmH
  179. vmovdqa ymmC, ymmA
  180. vpunpcklbw ymmA, ymmA, ymmH ; ymmA=(00 02 04 06 08 0A 0C 0E 0G 0I 0K 0M 0O 0Q 0S 0U)
  181. vpunpckhbw ymmC, ymmC, ymmH ; ymmC=(10 12 14 16 18 1A 1C 1E 1G 1I 1K 1M 1O 1Q 1S 1U)
  182. vmovdqa ymmB, ymmE
  183. vpunpcklbw ymmE, ymmE, ymmH ; ymmE=(20 22 24 26 28 2A 2C 2E 2G 2I 2K 2M 2O 2Q 2S 2U)
  184. vpunpckhbw ymmB, ymmB, ymmH ; ymmB=(01 03 05 07 09 0B 0D 0F 0H 0J 0L 0N 0P 0R 0T 0V)
  185. vmovdqa ymmF, ymmD
  186. vpunpcklbw ymmD, ymmD, ymmH ; ymmD=(11 13 15 17 19 1B 1D 1F 1H 1J 1L 1N 1P 1R 1T 1V)
  187. vpunpckhbw ymmF, ymmF, ymmH ; ymmF=(21 23 25 27 29 2B 2D 2F 2H 2J 2L 2N 2P 2R 2T 2V)
  188. %else ; RGB_PIXELSIZE == 4 ; -----------
  189. .column_ld1:
  190. test cl, SIZEOF_XMMWORD/16
  191. jz short .column_ld2
  192. sub rcx, byte SIZEOF_XMMWORD/16
  193. vmovd xmmA, XMM_DWORD [rsi+rcx*RGB_PIXELSIZE]
  194. .column_ld2:
  195. test cl, SIZEOF_XMMWORD/8
  196. jz short .column_ld4
  197. sub rcx, byte SIZEOF_XMMWORD/8
  198. vmovq xmmF, XMM_MMWORD [rsi+rcx*RGB_PIXELSIZE]
  199. vpslldq xmmA, xmmA, SIZEOF_MMWORD
  200. vpor xmmA, xmmA, xmmF
  201. .column_ld4:
  202. test cl, SIZEOF_XMMWORD/4
  203. jz short .column_ld8
  204. sub rcx, byte SIZEOF_XMMWORD/4
  205. vmovdqa xmmF, xmmA
  206. vperm2i128 ymmF, ymmF, ymmF, 1
  207. vmovdqu xmmA, XMMWORD [rsi+rcx*RGB_PIXELSIZE]
  208. vpor ymmA, ymmA, ymmF
  209. .column_ld8:
  210. test cl, SIZEOF_XMMWORD/2
  211. jz short .column_ld16
  212. sub rcx, byte SIZEOF_XMMWORD/2
  213. vmovdqa ymmF, ymmA
  214. vmovdqu ymmA, YMMWORD [rsi+rcx*RGB_PIXELSIZE]
  215. .column_ld16:
  216. test cl, SIZEOF_XMMWORD
  217. mov rcx, SIZEOF_YMMWORD
  218. jz short .rgb_gray_cnv
  219. vmovdqa ymmE, ymmA
  220. vmovdqa ymmH, ymmF
  221. vmovdqu ymmA, YMMWORD [rsi+0*SIZEOF_YMMWORD]
  222. vmovdqu ymmF, YMMWORD [rsi+1*SIZEOF_YMMWORD]
  223. jmp short .rgb_gray_cnv
  224. .columnloop:
  225. vmovdqu ymmA, YMMWORD [rsi+0*SIZEOF_YMMWORD]
  226. vmovdqu ymmF, YMMWORD [rsi+1*SIZEOF_YMMWORD]
  227. vmovdqu ymmE, YMMWORD [rsi+2*SIZEOF_YMMWORD]
  228. vmovdqu ymmH, YMMWORD [rsi+3*SIZEOF_YMMWORD]
  229. .rgb_gray_cnv:
  230. ; ymmA=(00 10 20 30 01 11 21 31 02 12 22 32 03 13 23 33
  231. ; 04 14 24 34 05 15 25 35 06 16 26 36 07 17 27 37)
  232. ; ymmF=(08 18 28 38 09 19 29 39 0A 1A 2A 3A 0B 1B 2B 3B
  233. ; 0C 1C 2C 3C 0D 1D 2D 3D 0E 1E 2E 3E 0F 1F 2F 3F)
  234. ; ymmE=(0G 1G 2G 3G 0H 1H 2H 3H 0I 1I 2I 3I 0J 1J 2J 3J
  235. ; 0K 1K 2K 3K 0L 1L 2L 3L 0M 1M 2M 3M 0N 1N 2N 3N)
  236. ; ymmH=(0O 1O 2O 3O 0P 1P 2P 3P 0Q 1Q 2Q 3Q 0R 1R 2R 3R
  237. ; 0S 1S 2S 3S 0T 1T 2T 3T 0U 1U 2U 3U 0V 1V 2V 3V)
  238. vmovdqa ymmB, ymmA
  239. vinserti128 ymmA, ymmA, xmmE, 1 ; ymmA=(00 10 20 30 01 11 21 31 02 12 22 32 03 13 23 33
  240. ; 0G 1G 2G 3G 0H 1H 2H 3H 0I 1I 2I 3I 0J 1J 2J 3J)
  241. vperm2i128 ymmE, ymmB, ymmE, 0x31 ; ymmE=(04 14 24 34 05 15 25 35 06 16 26 36 07 17 27 37
  242. ; 0K 1K 2K 3K 0L 1L 2L 3L 0M 1M 2M 3M 0N 1N 2N 3N)
  243. vmovdqa ymmB, ymmF
  244. vinserti128 ymmF, ymmF, xmmH, 1 ; ymmF=(08 18 28 38 09 19 29 39 0A 1A 2A 3A 0B 1B 2B 3B
  245. ; 0O 1O 2O 3O 0P 1P 2P 3P 0Q 1Q 2Q 3Q 0R 1R 2R 3R)
  246. vperm2i128 ymmH, ymmB, ymmH, 0x31 ; ymmH=(0C 1C 2C 3C 0D 1D 2D 3D 0E 1E 2E 3E 0F 1F 2F 3F
  247. ; 0S 1S 2S 3S 0T 1T 2T 3T 0U 1U 2U 3U 0V 1V 2V 3V)
  248. vmovdqa ymmD, ymmA
  249. vpunpcklbw ymmA, ymmA, ymmE ; ymmA=(00 04 10 14 20 24 30 34 01 05 11 15 21 25 31 35
  250. ; 0G 0K 1G 1K 2G 2K 3G 3K 0H 0L 1H 1L 2H 2L 3H 3L)
  251. vpunpckhbw ymmD, ymmD, ymmE ; ymmD=(02 06 12 16 22 26 32 36 03 07 13 17 23 27 33 37
  252. ; 0I 0M 1I 1M 2I 2M 3I 3M 0J 0N 1J 1N 2J 2N 3J 3N)
  253. vmovdqa ymmC, ymmF
  254. vpunpcklbw ymmF, ymmF, ymmH ; ymmF=(08 0C 18 1C 28 2C 38 3C 09 0D 19 1D 29 2D 39 3D
  255. ; 0O 0S 1O 1S 2O 2S 3O 3S 0P 0T 1P 1T 2P 2T 3P 3T)
  256. vpunpckhbw ymmC, ymmC, ymmH ; ymmC=(0A 0E 1A 1E 2A 2E 3A 3E 0B 0F 1B 1F 2B 2F 3B 3F
  257. ; 0Q 0U 1Q 1U 2Q 2U 3Q 3U 0R 0V 1R 1V 2R 2V 3R 3V)
  258. vmovdqa ymmB, ymmA
  259. vpunpcklwd ymmA, ymmA, ymmF ; ymmA=(00 04 08 0C 10 14 18 1C 20 24 28 2C 30 34 38 3C
  260. ; 0G 0K 0O 0S 1G 1K 1O 1S 2G 2K 2O 2S 3G 3K 3O 3S)
  261. vpunpckhwd ymmB, ymmB, ymmF ; ymmB=(01 05 09 0D 11 15 19 1D 21 25 29 2D 31 35 39 3D
  262. ; 0H 0L 0P 0T 1H 1L 1P 1T 2H 2L 2P 2T 3H 3L 3P 3T)
  263. vmovdqa ymmG, ymmD
  264. vpunpcklwd ymmD, ymmD, ymmC ; ymmD=(02 06 0A 0E 12 16 1A 1E 22 26 2A 2E 32 36 3A 3E
  265. ; 0I 0M 0Q 0U 1I 1M 1Q 1U 2I 2M 2Q 2U 3I 3M 3Q 3U)
  266. vpunpckhwd ymmG, ymmG, ymmC ; ymmG=(03 07 0B 0F 13 17 1B 1F 23 27 2B 2F 33 37 3B 3F
  267. ; 0J 0N 0R 0V 1J 1N 1R 1V 2J 2N 2R 2V 3J 3N 3R 3V)
  268. vmovdqa ymmE, ymmA
  269. vpunpcklbw ymmA, ymmA, ymmD ; ymmA=(00 02 04 06 08 0A 0C 0E 10 12 14 16 18 1A 1C 1E
  270. ; 0G 0I 0K 0M 0O 0Q 0S 0U 1G 1I 1K 1M 1O 1Q 1S 1U)
  271. vpunpckhbw ymmE, ymmE, ymmD ; ymmE=(20 22 24 26 28 2A 2C 2E 30 32 34 36 38 3A 3C 3E
  272. ; 2G 2I 2K 2M 2O 2Q 2S 2U 3G 3I 3K 3M 3O 3Q 3S 3U)
  273. vmovdqa ymmH, ymmB
  274. vpunpcklbw ymmB, ymmB, ymmG ; ymmB=(01 03 05 07 09 0B 0D 0F 11 13 15 17 19 1B 1D 1F
  275. ; 0H 0J 0L 0N 0P 0R 0T 0V 1H 1J 1L 1N 1P 1R 1T 1V)
  276. vpunpckhbw ymmH, ymmH, ymmG ; ymmH=(21 23 25 27 29 2B 2D 2F 31 33 35 37 39 3B 3D 3F
  277. ; 2H 2J 2L 2N 2P 2R 2T 2V 3H 3J 3L 3N 3P 3R 3T 3V)
  278. vpxor ymmF, ymmF, ymmF
  279. vmovdqa ymmC, ymmA
  280. vpunpcklbw ymmA, ymmA, ymmF ; ymmA=(00 02 04 06 08 0A 0C 0E 0G 0I 0K 0M 0O 0Q 0S 0U)
  281. vpunpckhbw ymmC, ymmC, ymmF ; ymmC=(10 12 14 16 18 1A 1C 1E 1G 1I 1K 1M 1O 1Q 1S 1U)
  282. vmovdqa ymmD, ymmB
  283. vpunpcklbw ymmB, ymmB, ymmF ; ymmB=(01 03 05 07 09 0B 0D 0F 0H 0J 0L 0N 0P 0R 0T 0V)
  284. vpunpckhbw ymmD, ymmD, ymmF ; ymmD=(11 13 15 17 19 1B 1D 1F 1H 1J 1L 1N 1P 1R 1T 1V)
  285. vmovdqa ymmG, ymmE
  286. vpunpcklbw ymmE, ymmE, ymmF ; ymmE=(20 22 24 26 28 2A 2C 2E 2G 2I 2K 2M 2O 2Q 2S 2U)
  287. vpunpckhbw ymmG, ymmG, ymmF ; ymmG=(30 32 34 36 38 3A 3C 3E 3G 3I 3K 3M 3O 3Q 3S 3U)
  288. vpunpcklbw ymmF, ymmF, ymmH
  289. vpunpckhbw ymmH, ymmH, ymmH
  290. vpsrlw ymmF, ymmF, BYTE_BIT ; ymmF=(21 23 25 27 29 2B 2D 2F 2H 2J 2L 2N 2P 2R 2T 2V)
  291. vpsrlw ymmH, ymmH, BYTE_BIT ; ymmH=(31 33 35 37 39 3B 3D 3F 3H 3J 3L 3N 3P 3R 3T 3V)
  292. %endif ; RGB_PIXELSIZE ; ---------------
  293. ; ymm0=R(02468ACEGIKMOQSU)=RE, ymm2=G(02468ACEGIKMOQSU)=GE, ymm4=B(02468ACEGIKMOQSU)=BE
  294. ; ymm1=R(13579BDFHJLNPRTV)=RO, ymm3=G(13579BDFHJLNPRTV)=GO, ymm5=B(13579BDFHJLNPRTV)=BO
  295. ; (Original)
  296. ; Y = 0.29900 * R + 0.58700 * G + 0.11400 * B
  297. ;
  298. ; (This implementation)
  299. ; Y = 0.29900 * R + 0.33700 * G + 0.11400 * B + 0.25000 * G
  300. vmovdqa ymm6, ymm1
  301. vpunpcklwd ymm1, ymm1, ymm3
  302. vpunpckhwd ymm6, ymm6, ymm3
  303. vpmaddwd ymm1, ymm1, [rel PW_F0299_F0337] ; ymm1=ROL*FIX(0.299)+GOL*FIX(0.337)
  304. vpmaddwd ymm6, ymm6, [rel PW_F0299_F0337] ; ymm6=ROH*FIX(0.299)+GOH*FIX(0.337)
  305. vmovdqa ymm7, ymm6 ; ymm7=ROH*FIX(0.299)+GOH*FIX(0.337)
  306. vmovdqa ymm6, ymm0
  307. vpunpcklwd ymm0, ymm0, ymm2
  308. vpunpckhwd ymm6, ymm6, ymm2
  309. vpmaddwd ymm0, ymm0, [rel PW_F0299_F0337] ; ymm0=REL*FIX(0.299)+GEL*FIX(0.337)
  310. vpmaddwd ymm6, ymm6, [rel PW_F0299_F0337] ; ymm6=REH*FIX(0.299)+GEH*FIX(0.337)
  311. vmovdqa YMMWORD [wk(0)], ymm0 ; wk(0)=REL*FIX(0.299)+GEL*FIX(0.337)
  312. vmovdqa YMMWORD [wk(1)], ymm6 ; wk(1)=REH*FIX(0.299)+GEH*FIX(0.337)
  313. vmovdqa ymm0, ymm5 ; ymm0=BO
  314. vmovdqa ymm6, ymm4 ; ymm6=BE
  315. vmovdqa ymm4, ymm0
  316. vpunpcklwd ymm0, ymm0, ymm3
  317. vpunpckhwd ymm4, ymm4, ymm3
  318. vpmaddwd ymm0, ymm0, [rel PW_F0114_F0250] ; ymm0=BOL*FIX(0.114)+GOL*FIX(0.250)
  319. vpmaddwd ymm4, ymm4, [rel PW_F0114_F0250] ; ymm4=BOH*FIX(0.114)+GOH*FIX(0.250)
  320. vmovdqa ymm3, [rel PD_ONEHALF] ; ymm3=[PD_ONEHALF]
  321. vpaddd ymm0, ymm0, ymm1
  322. vpaddd ymm4, ymm4, ymm7
  323. vpaddd ymm0, ymm0, ymm3
  324. vpaddd ymm4, ymm4, ymm3
  325. vpsrld ymm0, ymm0, SCALEBITS ; ymm0=YOL
  326. vpsrld ymm4, ymm4, SCALEBITS ; ymm4=YOH
  327. vpackssdw ymm0, ymm0, ymm4 ; ymm0=YO
  328. vmovdqa ymm4, ymm6
  329. vpunpcklwd ymm6, ymm6, ymm2
  330. vpunpckhwd ymm4, ymm4, ymm2
  331. vpmaddwd ymm6, ymm6, [rel PW_F0114_F0250] ; ymm6=BEL*FIX(0.114)+GEL*FIX(0.250)
  332. vpmaddwd ymm4, ymm4, [rel PW_F0114_F0250] ; ymm4=BEH*FIX(0.114)+GEH*FIX(0.250)
  333. vmovdqa ymm2, [rel PD_ONEHALF] ; ymm2=[PD_ONEHALF]
  334. vpaddd ymm6, ymm6, YMMWORD [wk(0)]
  335. vpaddd ymm4, ymm4, YMMWORD [wk(1)]
  336. vpaddd ymm6, ymm6, ymm2
  337. vpaddd ymm4, ymm4, ymm2
  338. vpsrld ymm6, ymm6, SCALEBITS ; ymm6=YEL
  339. vpsrld ymm4, ymm4, SCALEBITS ; ymm4=YEH
  340. vpackssdw ymm6, ymm6, ymm4 ; ymm6=YE
  341. vpsllw ymm0, ymm0, BYTE_BIT
  342. vpor ymm6, ymm6, ymm0 ; ymm6=Y
  343. vmovdqu YMMWORD [rdi], ymm6 ; Save Y
  344. sub rcx, byte SIZEOF_YMMWORD
  345. add rsi, RGB_PIXELSIZE*SIZEOF_YMMWORD ; inptr
  346. add rdi, byte SIZEOF_YMMWORD ; outptr0
  347. cmp rcx, byte SIZEOF_YMMWORD
  348. jae near .columnloop
  349. test rcx, rcx
  350. jnz near .column_ld1
  351. pop rcx ; col
  352. pop rsi
  353. pop rdi
  354. add rsi, byte SIZEOF_JSAMPROW ; input_buf
  355. add rdi, byte SIZEOF_JSAMPROW
  356. dec rax ; num_rows
  357. jg near .rowloop
  358. .return:
  359. pop rbx
  360. vzeroupper
  361. UNCOLLECT_ARGS 5
  362. lea rsp, [rbp-8]
  363. pop r15
  364. pop rbp
  365. ret
  366. ; For some reason, the OS X linker does not honor the request to align the
  367. ; segment unless we do this.
  368. align 32