2 ; jdclrss2.asm - colorspace conversion (SSE2)
4 ; Copyright 2009 Pierre Ossman <ossman@cendio.se> for Cendio AB
7 ; x86 SIMD extension for IJG JPEG library
8 ; Copyright (C) 1999-2006, MIYASAKA Masaru.
9 ; For conditions of distribution and use, see copyright notice in jsimdext.inc
11 ; This file should be assembled with NASM (Netwide Assembler),
12 ; can *not* be assembled with Microsoft's MASM or any compatible
13 ; assembler (including Borland's Turbo Assembler).
14 ; NASM is available from http://nasm.sourceforge.net/ or
15 ; http://sourceforge.net/project/showfiles.php?group_id=6208
19 %include "jcolsamp.inc"
21 ; --------------------------------------------------------------------------
25 ; Convert some rows of samples to the output colorspace.
28 ; jsimd_ycc_rgb_convert_sse2 (JDIMENSION out_width,
29 ; JSAMPIMAGE input_buf, JDIMENSION input_row,
30 ; JSAMPARRAY output_buf, int num_rows)
33 %define out_width
(b
) (b
)+8 ; JDIMENSION out_width
34 %define input_buf
(b
) (b
)+12 ; JSAMPIMAGE input_buf
35 %define input_row
(b
) (b
)+16 ; JDIMENSION input_row
36 %define output_buf
(b
) (b
)+20 ; JSAMPARRAY output_buf
37 %define num_rows
(b
) (b
)+24 ; int num_rows
39 %define original_ebp
ebp+0
40 %define wk
(i
) ebp-(WK_NUM
-(i
))*SIZEOF_XMMWORD
; xmmword wk[WK_NUM]
42 %define gotptr wk
(0)-SIZEOF_POINTER
; void * gotptr
45 global EXTN
(jsimd_ycc_rgb_convert_sse2
)
47 EXTN
(jsimd_ycc_rgb_convert_sse2
):
49 mov eax,esp ; eax = original ebp
51 and esp, byte (-SIZEOF_XMMWORD
) ; align to 128 bits
53 mov ebp,esp ; ebp = aligned ebp
55 pushpic
eax ; make a room for GOT address
57 ; push ecx ; need not be preserved
58 ; push edx ; need not be preserved
62 get_GOT
ebx ; get GOT address
63 movpic POINTER
[gotptr
], ebx ; save GOT address
65 mov ecx, JDIMENSION
[out_width
(eax)] ; num_cols
71 mov edi, JSAMPIMAGE
[input_buf
(eax)]
72 mov ecx, JDIMENSION
[input_row
(eax)]
73 mov esi, JSAMPARRAY
[edi+0*SIZEOF_JSAMPARRAY
]
74 mov ebx, JSAMPARRAY
[edi+1*SIZEOF_JSAMPARRAY
]
75 mov edx, JSAMPARRAY
[edi+2*SIZEOF_JSAMPARRAY
]
76 lea esi, [esi+ecx*SIZEOF_JSAMPROW
]
77 lea ebx, [ebx+ecx*SIZEOF_JSAMPROW
]
78 lea edx, [edx+ecx*SIZEOF_JSAMPROW
]
82 mov edi, JSAMPARRAY
[output_buf
(eax)]
83 mov eax, INT [num_rows
(eax)]
95 mov esi, JSAMPROW
[esi] ; inptr0
96 mov ebx, JSAMPROW
[ebx] ; inptr1
97 mov edx, JSAMPROW
[edx] ; inptr2
98 mov edi, JSAMPROW
[edi] ; outptr
99 movpic
eax, POINTER
[gotptr
] ; load GOT address (eax)
103 movdqa xmm5
, XMMWORD
[ebx] ; xmm5=Cb(0123456789ABCDEF)
104 movdqa xmm1
, XMMWORD
[edx] ; xmm1=Cr(0123456789ABCDEF)
109 psllw xmm7
,7 ; xmm7={0xFF80 0xFF80 0xFF80 0xFF80 ..}
110 movdqa xmm0
,xmm4
; xmm0=xmm4={0xFF 0x00 0xFF 0x00 ..}
112 pand xmm4
,xmm5
; xmm4=Cb(02468ACE)=CbE
113 psrlw xmm5
,BYTE_BIT
; xmm5=Cb(13579BDF)=CbO
114 pand xmm0
,xmm1
; xmm0=Cr(02468ACE)=CrE
115 psrlw xmm1
,BYTE_BIT
; xmm1=Cr(13579BDF)=CrO
123 ; R = Y + 1.40200 * Cr
124 ; G = Y - 0.34414 * Cb - 0.71414 * Cr
125 ; B = Y + 1.77200 * Cb
127 ; (This implementation)
128 ; R = Y + 0.40200 * Cr + Cr
129 ; G = Y - 0.34414 * Cb + 0.28586 * Cr - Cr
130 ; B = Y - 0.22800 * Cb + Cb + Cb
132 movdqa xmm2
,xmm4
; xmm2=CbE
133 movdqa xmm3
,xmm5
; xmm3=CbO
134 paddw xmm4
,xmm4
; xmm4=2*CbE
135 paddw xmm5
,xmm5
; xmm5=2*CbO
136 movdqa xmm6
,xmm0
; xmm6=CrE
137 movdqa xmm7
,xmm1
; xmm7=CrO
138 paddw xmm0
,xmm0
; xmm0=2*CrE
139 paddw xmm1
,xmm1
; xmm1=2*CrO
141 pmulhw xmm4
,[GOTOFF
(eax,PW_MF0228
)] ; xmm4=(2*CbE * -FIX(0.22800))
142 pmulhw xmm5
,[GOTOFF
(eax,PW_MF0228
)] ; xmm5=(2*CbO * -FIX(0.22800))
143 pmulhw xmm0
,[GOTOFF
(eax,PW_F0402
)] ; xmm0=(2*CrE * FIX(0.40200))
144 pmulhw xmm1
,[GOTOFF
(eax,PW_F0402
)] ; xmm1=(2*CrO * FIX(0.40200))
146 paddw xmm4
,[GOTOFF
(eax,PW_ONE
)]
147 paddw xmm5
,[GOTOFF
(eax,PW_ONE
)]
148 psraw xmm4
,1 ; xmm4=(CbE * -FIX(0.22800))
149 psraw xmm5
,1 ; xmm5=(CbO * -FIX(0.22800))
150 paddw xmm0
,[GOTOFF
(eax,PW_ONE
)]
151 paddw xmm1
,[GOTOFF
(eax,PW_ONE
)]
152 psraw xmm0
,1 ; xmm0=(CrE * FIX(0.40200))
153 psraw xmm1
,1 ; xmm1=(CrO * FIX(0.40200))
157 paddw xmm4
,xmm2
; xmm4=(CbE * FIX(1.77200))=(B-Y)E
158 paddw xmm5
,xmm3
; xmm5=(CbO * FIX(1.77200))=(B-Y)O
159 paddw xmm0
,xmm6
; xmm0=(CrE * FIX(1.40200))=(R-Y)E
160 paddw xmm1
,xmm7
; xmm1=(CrO * FIX(1.40200))=(R-Y)O
162 movdqa XMMWORD
[wk
(0)], xmm4
; wk(0)=(B-Y)E
163 movdqa XMMWORD
[wk
(1)], xmm5
; wk(1)=(B-Y)O
169 pmaddwd xmm2
,[GOTOFF
(eax,PW_MF0344_F0285
)]
170 pmaddwd xmm4
,[GOTOFF
(eax,PW_MF0344_F0285
)]
173 pmaddwd xmm3
,[GOTOFF
(eax,PW_MF0344_F0285
)]
174 pmaddwd xmm5
,[GOTOFF
(eax,PW_MF0344_F0285
)]
176 paddd xmm2
,[GOTOFF
(eax,PD_ONEHALF
)]
177 paddd xmm4
,[GOTOFF
(eax,PD_ONEHALF
)]
180 paddd xmm3
,[GOTOFF
(eax,PD_ONEHALF
)]
181 paddd xmm5
,[GOTOFF
(eax,PD_ONEHALF
)]
185 packssdw xmm2
,xmm4
; xmm2=CbE*-FIX(0.344)+CrE*FIX(0.285)
186 packssdw xmm3
,xmm5
; xmm3=CbO*-FIX(0.344)+CrO*FIX(0.285)
187 psubw xmm2
,xmm6
; xmm2=CbE*-FIX(0.344)+CrE*-FIX(0.714)=(G-Y)E
188 psubw xmm3
,xmm7
; xmm3=CbO*-FIX(0.344)+CrO*-FIX(0.714)=(G-Y)O
190 movdqa xmm5
, XMMWORD
[esi] ; xmm5=Y(0123456789ABCDEF)
193 psrlw xmm4
,BYTE_BIT
; xmm4={0xFF 0x00 0xFF 0x00 ..}
194 pand xmm4
,xmm5
; xmm4=Y(02468ACE)=YE
195 psrlw xmm5
,BYTE_BIT
; xmm5=Y(13579BDF)=YO
197 paddw xmm0
,xmm4
; xmm0=((R-Y)E+YE)=RE=R(02468ACE)
198 paddw xmm1
,xmm5
; xmm1=((R-Y)O+YO)=RO=R(13579BDF)
199 packuswb xmm0
,xmm0
; xmm0=R(02468ACE********)
200 packuswb xmm1
,xmm1
; xmm1=R(13579BDF********)
202 paddw xmm2
,xmm4
; xmm2=((G-Y)E+YE)=GE=G(02468ACE)
203 paddw xmm3
,xmm5
; xmm3=((G-Y)O+YO)=GO=G(13579BDF)
204 packuswb xmm2
,xmm2
; xmm2=G(02468ACE********)
205 packuswb xmm3
,xmm3
; xmm3=G(13579BDF********)
207 paddw xmm4
, XMMWORD
[wk
(0)] ; xmm4=(YE+(B-Y)E)=BE=B(02468ACE)
208 paddw xmm5
, XMMWORD
[wk
(1)] ; xmm5=(YO+(B-Y)O)=BO=B(13579BDF)
209 packuswb xmm4
,xmm4
; xmm4=B(02468ACE********)
210 packuswb xmm5
,xmm5
; xmm5=B(13579BDF********)
212 %if RGB_PIXELSIZE
== 3 ; ---------------
214 ; xmmA=(00 02 04 06 08 0A 0C 0E **), xmmB=(01 03 05 07 09 0B 0D 0F **)
215 ; xmmC=(10 12 14 16 18 1A 1C 1E **), xmmD=(11 13 15 17 19 1B 1D 1F **)
216 ; xmmE=(20 22 24 26 28 2A 2C 2E **), xmmF=(21 23 25 27 29 2B 2D 2F **)
217 ; xmmG=(** ** ** ** ** ** ** ** **), xmmH=(** ** ** ** ** ** ** ** **)
219 punpcklbw xmmA
,xmmC
; xmmA=(00 10 02 12 04 14 06 16 08 18 0A 1A 0C 1C 0E 1E)
220 punpcklbw xmmE
,xmmB
; xmmE=(20 01 22 03 24 05 26 07 28 09 2A 0B 2C 0D 2E 0F)
221 punpcklbw xmmD
,xmmF
; xmmD=(11 21 13 23 15 25 17 27 19 29 1B 2B 1D 2D 1F 2F)
225 punpcklwd xmmA
,xmmE
; xmmA=(00 10 20 01 02 12 22 03 04 14 24 05 06 16 26 07)
226 punpckhwd xmmG
,xmmE
; xmmG=(08 18 28 09 0A 1A 2A 0B 0C 1C 2C 0D 0E 1E 2E 0F)
228 psrldq xmmH
,2 ; xmmH=(02 12 04 14 06 16 08 18 0A 1A 0C 1C 0E 1E -- --)
229 psrldq xmmE
,2 ; xmmE=(22 03 24 05 26 07 28 09 2A 0B 2C 0D 2E 0F -- --)
233 punpcklwd xmmD
,xmmH
; xmmD=(11 21 02 12 13 23 04 14 15 25 06 16 17 27 08 18)
234 punpckhwd xmmC
,xmmH
; xmmC=(19 29 0A 1A 1B 2B 0C 1C 1D 2D 0E 1E 1F 2F -- --)
236 psrldq xmmB
,2 ; xmmB=(13 23 15 25 17 27 19 29 1B 2B 1D 2D 1F 2F -- --)
239 punpcklwd xmmE
,xmmB
; xmmE=(22 03 13 23 24 05 15 25 26 07 17 27 28 09 19 29)
240 punpckhwd xmmF
,xmmB
; xmmF=(2A 0B 1B 2B 2C 0D 1D 2D 2E 0F 1F 2F -- -- -- --)
242 pshufd xmmH
,xmmA
,0x4E; xmmH=(04 14 24 05 06 16 26 07 00 10 20 01 02 12 22 03)
244 punpckldq xmmA
,xmmD
; xmmA=(00 10 20 01 11 21 02 12 02 12 22 03 13 23 04 14)
245 punpckldq xmmE
,xmmH
; xmmE=(22 03 13 23 04 14 24 05 24 05 15 25 06 16 26 07)
246 punpckhdq xmmD
,xmmB
; xmmD=(15 25 06 16 26 07 17 27 17 27 08 18 28 09 19 29)
248 pshufd xmmH
,xmmG
,0x4E; xmmH=(0C 1C 2C 0D 0E 1E 2E 0F 08 18 28 09 0A 1A 2A 0B)
250 punpckldq xmmG
,xmmC
; xmmG=(08 18 28 09 19 29 0A 1A 0A 1A 2A 0B 1B 2B 0C 1C)
251 punpckldq xmmF
,xmmH
; xmmF=(2A 0B 1B 2B 0C 1C 2C 0D 2C 0D 1D 2D 0E 1E 2E 0F)
252 punpckhdq xmmC
,xmmB
; xmmC=(1D 2D 0E 1E 2E 0F 1F 2F 1F 2F -- -- -- -- -- --)
254 punpcklqdq xmmA
,xmmE
; xmmA=(00 10 20 01 11 21 02 12 22 03 13 23 04 14 24 05)
255 punpcklqdq xmmD
,xmmG
; xmmD=(15 25 06 16 26 07 17 27 08 18 28 09 19 29 0A 1A)
256 punpcklqdq xmmF
,xmmC
; xmmF=(2A 0B 1B 2B 0C 1C 2C 0D 1D 2D 0E 1E 2E 0F 1F 2F)
258 cmp ecx, byte SIZEOF_XMMWORD
259 jb short .column_st32
261 test edi, SIZEOF_XMMWORD
-1
263 ; --(aligned)-------------------
264 movntdq XMMWORD
[edi+0*SIZEOF_XMMWORD
], xmmA
265 movntdq XMMWORD
[edi+1*SIZEOF_XMMWORD
], xmmD
266 movntdq XMMWORD
[edi+2*SIZEOF_XMMWORD
], xmmF
267 add edi, byte RGB_PIXELSIZE
*SIZEOF_XMMWORD
; outptr
269 .
out1: ; --(unaligned)-----------------
270 pcmpeqb xmmH
,xmmH
; xmmH=(all 1's)
271 maskmovdqu xmmA
,xmmH
; movntdqu XMMWORD [edi], xmmA
272 add edi, byte SIZEOF_XMMWORD
; outptr
273 maskmovdqu xmmD
,xmmH
; movntdqu XMMWORD [edi], xmmD
274 add edi, byte SIZEOF_XMMWORD
; outptr
275 maskmovdqu xmmF
,xmmH
; movntdqu XMMWORD [edi], xmmF
276 add edi, byte SIZEOF_XMMWORD
; outptr
278 sub ecx, byte SIZEOF_XMMWORD
281 add esi, byte SIZEOF_XMMWORD
; inptr0
282 add ebx, byte SIZEOF_XMMWORD
; inptr1
283 add edx, byte SIZEOF_XMMWORD
; inptr2
288 pcmpeqb xmmH
,xmmH
; xmmH=(all 1's)
289 lea ecx, [ecx+ecx*2] ; imul ecx, RGB_PIXELSIZE
290 cmp ecx, byte 2*SIZEOF_XMMWORD
291 jb short .column_st16
292 maskmovdqu xmmA
,xmmH
; movntdqu XMMWORD [edi], xmmA
293 add edi, byte SIZEOF_XMMWORD
; outptr
294 maskmovdqu xmmD
,xmmH
; movntdqu XMMWORD [edi], xmmD
295 add edi, byte SIZEOF_XMMWORD
; outptr
297 sub ecx, byte 2*SIZEOF_XMMWORD
298 jmp short .column_st15
300 cmp ecx, byte SIZEOF_XMMWORD
301 jb short .column_st15
302 maskmovdqu xmmA
,xmmH
; movntdqu XMMWORD [edi], xmmA
303 add edi, byte SIZEOF_XMMWORD
; outptr
305 sub ecx, byte SIZEOF_XMMWORD
318 and ecx, byte SIZEOF_XMMWORD
-1
321 cmp eax, byte SIZEOF_XMMWORD
323 and edi, byte (-SIZEOF_XMMWORD
) ; align to 16-byte boundary
324 shl ecx, 3 ; pslldq xmmA,ecx & pslldq xmmE,ecx
327 pslldq xmmA
, SIZEOF_XMMWORD
/2
328 pslldq xmmE
, SIZEOF_XMMWORD
/2
330 sub ecx, byte (SIZEOF_XMMWORD
/2)*BYTE_BIT
344 .
adj0: ; ----------------
345 maskmovdqu xmmA
,xmmE
; movntdqu XMMWORD [edi], xmmA
347 %else
; RGB_PIXELSIZE == 4 ; -----------
349 %ifdef RGBX_FILLER_0XFF
350 pcmpeqb xmm6
,xmm6
; xmm6=XE=X(02468ACE********)
351 pcmpeqb xmm7
,xmm7
; xmm7=XO=X(13579BDF********)
353 pxor xmm6
,xmm6
; xmm6=XE=X(02468ACE********)
354 pxor xmm7
,xmm7
; xmm7=XO=X(13579BDF********)
356 ; xmmA=(00 02 04 06 08 0A 0C 0E **), xmmB=(01 03 05 07 09 0B 0D 0F **)
357 ; xmmC=(10 12 14 16 18 1A 1C 1E **), xmmD=(11 13 15 17 19 1B 1D 1F **)
358 ; xmmE=(20 22 24 26 28 2A 2C 2E **), xmmF=(21 23 25 27 29 2B 2D 2F **)
359 ; xmmG=(30 32 34 36 38 3A 3C 3E **), xmmH=(31 33 35 37 39 3B 3D 3F **)
361 punpcklbw xmmA
,xmmC
; xmmA=(00 10 02 12 04 14 06 16 08 18 0A 1A 0C 1C 0E 1E)
362 punpcklbw xmmE
,xmmG
; xmmE=(20 30 22 32 24 34 26 36 28 38 2A 3A 2C 3C 2E 3E)
363 punpcklbw xmmB
,xmmD
; xmmB=(01 11 03 13 05 15 07 17 09 19 0B 1B 0D 1D 0F 1F)
364 punpcklbw xmmF
,xmmH
; xmmF=(21 31 23 33 25 35 27 37 29 39 2B 3B 2D 3D 2F 3F)
367 punpcklwd xmmA
,xmmE
; xmmA=(00 10 20 30 02 12 22 32 04 14 24 34 06 16 26 36)
368 punpckhwd xmmC
,xmmE
; xmmC=(08 18 28 38 0A 1A 2A 3A 0C 1C 2C 3C 0E 1E 2E 3E)
370 punpcklwd xmmB
,xmmF
; xmmB=(01 11 21 31 03 13 23 33 05 15 25 35 07 17 27 37)
371 punpckhwd xmmG
,xmmF
; xmmG=(09 19 29 39 0B 1B 2B 3B 0D 1D 2D 3D 0F 1F 2F 3F)
374 punpckldq xmmA
,xmmB
; xmmA=(00 10 20 30 01 11 21 31 02 12 22 32 03 13 23 33)
375 punpckhdq xmmD
,xmmB
; xmmD=(04 14 24 34 05 15 25 35 06 16 26 36 07 17 27 37)
377 punpckldq xmmC
,xmmG
; xmmC=(08 18 28 38 09 19 29 39 0A 1A 2A 3A 0B 1B 2B 3B)
378 punpckhdq xmmH
,xmmG
; xmmH=(0C 1C 2C 3C 0D 1D 2D 3D 0E 1E 2E 3E 0F 1F 2F 3F)
380 cmp ecx, byte SIZEOF_XMMWORD
381 jb short .column_st32
383 test edi, SIZEOF_XMMWORD
-1
385 ; --(aligned)-------------------
386 movntdq XMMWORD
[edi+0*SIZEOF_XMMWORD
], xmmA
387 movntdq XMMWORD
[edi+1*SIZEOF_XMMWORD
], xmmD
388 movntdq XMMWORD
[edi+2*SIZEOF_XMMWORD
], xmmC
389 movntdq XMMWORD
[edi+3*SIZEOF_XMMWORD
], xmmH
390 add edi, byte RGB_PIXELSIZE
*SIZEOF_XMMWORD
; outptr
392 .
out1: ; --(unaligned)-----------------
393 pcmpeqb xmmE
,xmmE
; xmmE=(all 1's)
394 maskmovdqu xmmA
,xmmE
; movntdqu XMMWORD [edi], xmmA
395 add edi, byte SIZEOF_XMMWORD
; outptr
396 maskmovdqu xmmD
,xmmE
; movntdqu XMMWORD [edi], xmmD
397 add edi, byte SIZEOF_XMMWORD
; outptr
398 maskmovdqu xmmC
,xmmE
; movntdqu XMMWORD [edi], xmmC
399 add edi, byte SIZEOF_XMMWORD
; outptr
400 maskmovdqu xmmH
,xmmE
; movntdqu XMMWORD [edi], xmmH
401 add edi, byte SIZEOF_XMMWORD
; outptr
403 sub ecx, byte SIZEOF_XMMWORD
406 add esi, byte SIZEOF_XMMWORD
; inptr0
407 add ebx, byte SIZEOF_XMMWORD
; inptr1
408 add edx, byte SIZEOF_XMMWORD
; inptr2
413 pcmpeqb xmmE
,xmmE
; xmmE=(all 1's)
414 cmp ecx, byte SIZEOF_XMMWORD
/2
415 jb short .column_st16
416 maskmovdqu xmmA
,xmmE
; movntdqu XMMWORD [edi], xmmA
417 add edi, byte SIZEOF_XMMWORD
; outptr
418 maskmovdqu xmmD
,xmmE
; movntdqu XMMWORD [edi], xmmD
419 add edi, byte SIZEOF_XMMWORD
; outptr
422 sub ecx, byte SIZEOF_XMMWORD
/2
424 cmp ecx, byte SIZEOF_XMMWORD
/4
425 jb short .column_st15
426 maskmovdqu xmmA
,xmmE
; movntdqu XMMWORD [edi], xmmA
427 add edi, byte SIZEOF_XMMWORD
; outptr
429 sub ecx, byte SIZEOF_XMMWORD
/4
431 cmp ecx, byte SIZEOF_XMMWORD
/16
442 and ecx, byte SIZEOF_XMMWORD
-1
444 lea eax, [ecx+eax*4] ; RGB_PIXELSIZE
445 cmp eax, byte SIZEOF_XMMWORD
447 and edi, byte (-SIZEOF_XMMWORD
) ; align to 16-byte boundary
448 shl ecx, 3 ; pslldq xmmA,ecx & pslldq xmmE,ecx
451 pslldq xmmA
, SIZEOF_XMMWORD
/2
452 pslldq xmmE
, SIZEOF_XMMWORD
/2
454 sub ecx, byte (SIZEOF_XMMWORD
/2)*BYTE_BIT
468 .
adj0: ; ----------------
469 maskmovdqu xmmA
,xmmE
; movntdqu XMMWORD [edi], xmmA
471 %endif
; RGB_PIXELSIZE ; ---------------
483 add esi, byte SIZEOF_JSAMPROW
484 add ebx, byte SIZEOF_JSAMPROW
485 add edx, byte SIZEOF_JSAMPROW
486 add edi, byte SIZEOF_JSAMPROW
; output_buf
490 sfence
; flush the write buffer
495 ; pop edx ; need not be preserved
496 ; pop ecx ; need not be preserved
498 mov esp,ebp ; esp <- aligned ebp
499 pop esp ; esp <- original ebp
503 ; For some reason, the OS X linker does not honor the request to align the
504 ; segment unless we do this.