|
1 ; |
|
2 ; jdclrmmx.asm - colorspace conversion (MMX) |
|
3 ; |
|
4 ; Copyright 2009 Pierre Ossman <ossman@cendio.se> for Cendio AB |
|
5 ; |
|
6 ; Based on |
|
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 |
|
10 ; |
|
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 |
|
16 ; |
|
17 ; [TAB8] |
|
18 |
|
19 %include "jcolsamp.inc" |
|
20 |
|
21 ; -------------------------------------------------------------------------- |
|
22 ; |
|
23 ; Convert some rows of samples to the output colorspace. |
|
24 ; |
|
25 ; GLOBAL(void) |
|
26 ; jsimd_ycc_rgb_convert_mmx (JDIMENSION out_width, |
|
27 ; JSAMPIMAGE input_buf, JDIMENSION input_row, |
|
28 ; JSAMPARRAY output_buf, int num_rows) |
|
29 ; |
|
30 |
|
31 %define out_width(b) (b)+8 ; JDIMENSION out_width |
|
32 %define input_buf(b) (b)+12 ; JSAMPIMAGE input_buf |
|
33 %define input_row(b) (b)+16 ; JDIMENSION input_row |
|
34 %define output_buf(b) (b)+20 ; JSAMPARRAY output_buf |
|
35 %define num_rows(b) (b)+24 ; int num_rows |
|
36 |
|
37 %define original_ebp ebp+0 |
|
38 %define wk(i) ebp-(WK_NUM-(i))*SIZEOF_MMWORD ; mmword wk[WK_NUM] |
|
39 %define WK_NUM 2 |
|
40 %define gotptr wk(0)-SIZEOF_POINTER ; void * gotptr |
|
41 |
|
42 align 16 |
|
43 global EXTN(jsimd_ycc_rgb_convert_mmx) |
|
44 |
|
45 EXTN(jsimd_ycc_rgb_convert_mmx): |
|
46 push ebp |
|
47 mov eax,esp ; eax = original ebp |
|
48 sub esp, byte 4 |
|
49 and esp, byte (-SIZEOF_MMWORD) ; align to 64 bits |
|
50 mov [esp],eax |
|
51 mov ebp,esp ; ebp = aligned ebp |
|
52 lea esp, [wk(0)] |
|
53 pushpic eax ; make a room for GOT address |
|
54 push ebx |
|
55 ; push ecx ; need not be preserved |
|
56 ; push edx ; need not be preserved |
|
57 push esi |
|
58 push edi |
|
59 |
|
60 get_GOT ebx ; get GOT address |
|
61 movpic POINTER [gotptr], ebx ; save GOT address |
|
62 |
|
63 mov ecx, JDIMENSION [out_width(eax)] ; num_cols |
|
64 test ecx,ecx |
|
65 jz near .return |
|
66 |
|
67 push ecx |
|
68 |
|
69 mov edi, JSAMPIMAGE [input_buf(eax)] |
|
70 mov ecx, JDIMENSION [input_row(eax)] |
|
71 mov esi, JSAMPARRAY [edi+0*SIZEOF_JSAMPARRAY] |
|
72 mov ebx, JSAMPARRAY [edi+1*SIZEOF_JSAMPARRAY] |
|
73 mov edx, JSAMPARRAY [edi+2*SIZEOF_JSAMPARRAY] |
|
74 lea esi, [esi+ecx*SIZEOF_JSAMPROW] |
|
75 lea ebx, [ebx+ecx*SIZEOF_JSAMPROW] |
|
76 lea edx, [edx+ecx*SIZEOF_JSAMPROW] |
|
77 |
|
78 pop ecx |
|
79 |
|
80 mov edi, JSAMPARRAY [output_buf(eax)] |
|
81 mov eax, INT [num_rows(eax)] |
|
82 test eax,eax |
|
83 jle near .return |
|
84 alignx 16,7 |
|
85 .rowloop: |
|
86 push eax |
|
87 push edi |
|
88 push edx |
|
89 push ebx |
|
90 push esi |
|
91 push ecx ; col |
|
92 |
|
93 mov esi, JSAMPROW [esi] ; inptr0 |
|
94 mov ebx, JSAMPROW [ebx] ; inptr1 |
|
95 mov edx, JSAMPROW [edx] ; inptr2 |
|
96 mov edi, JSAMPROW [edi] ; outptr |
|
97 movpic eax, POINTER [gotptr] ; load GOT address (eax) |
|
98 alignx 16,7 |
|
99 .columnloop: |
|
100 |
|
101 movq mm5, MMWORD [ebx] ; mm5=Cb(01234567) |
|
102 movq mm1, MMWORD [edx] ; mm1=Cr(01234567) |
|
103 |
|
104 pcmpeqw mm4,mm4 |
|
105 pcmpeqw mm7,mm7 |
|
106 psrlw mm4,BYTE_BIT |
|
107 psllw mm7,7 ; mm7={0xFF80 0xFF80 0xFF80 0xFF80} |
|
108 movq mm0,mm4 ; mm0=mm4={0xFF 0x00 0xFF 0x00 ..} |
|
109 |
|
110 pand mm4,mm5 ; mm4=Cb(0246)=CbE |
|
111 psrlw mm5,BYTE_BIT ; mm5=Cb(1357)=CbO |
|
112 pand mm0,mm1 ; mm0=Cr(0246)=CrE |
|
113 psrlw mm1,BYTE_BIT ; mm1=Cr(1357)=CrO |
|
114 |
|
115 paddw mm4,mm7 |
|
116 paddw mm5,mm7 |
|
117 paddw mm0,mm7 |
|
118 paddw mm1,mm7 |
|
119 |
|
120 ; (Original) |
|
121 ; R = Y + 1.40200 * Cr |
|
122 ; G = Y - 0.34414 * Cb - 0.71414 * Cr |
|
123 ; B = Y + 1.77200 * Cb |
|
124 ; |
|
125 ; (This implementation) |
|
126 ; R = Y + 0.40200 * Cr + Cr |
|
127 ; G = Y - 0.34414 * Cb + 0.28586 * Cr - Cr |
|
128 ; B = Y - 0.22800 * Cb + Cb + Cb |
|
129 |
|
130 movq mm2,mm4 ; mm2=CbE |
|
131 movq mm3,mm5 ; mm3=CbO |
|
132 paddw mm4,mm4 ; mm4=2*CbE |
|
133 paddw mm5,mm5 ; mm5=2*CbO |
|
134 movq mm6,mm0 ; mm6=CrE |
|
135 movq mm7,mm1 ; mm7=CrO |
|
136 paddw mm0,mm0 ; mm0=2*CrE |
|
137 paddw mm1,mm1 ; mm1=2*CrO |
|
138 |
|
139 pmulhw mm4,[GOTOFF(eax,PW_MF0228)] ; mm4=(2*CbE * -FIX(0.22800)) |
|
140 pmulhw mm5,[GOTOFF(eax,PW_MF0228)] ; mm5=(2*CbO * -FIX(0.22800)) |
|
141 pmulhw mm0,[GOTOFF(eax,PW_F0402)] ; mm0=(2*CrE * FIX(0.40200)) |
|
142 pmulhw mm1,[GOTOFF(eax,PW_F0402)] ; mm1=(2*CrO * FIX(0.40200)) |
|
143 |
|
144 paddw mm4,[GOTOFF(eax,PW_ONE)] |
|
145 paddw mm5,[GOTOFF(eax,PW_ONE)] |
|
146 psraw mm4,1 ; mm4=(CbE * -FIX(0.22800)) |
|
147 psraw mm5,1 ; mm5=(CbO * -FIX(0.22800)) |
|
148 paddw mm0,[GOTOFF(eax,PW_ONE)] |
|
149 paddw mm1,[GOTOFF(eax,PW_ONE)] |
|
150 psraw mm0,1 ; mm0=(CrE * FIX(0.40200)) |
|
151 psraw mm1,1 ; mm1=(CrO * FIX(0.40200)) |
|
152 |
|
153 paddw mm4,mm2 |
|
154 paddw mm5,mm3 |
|
155 paddw mm4,mm2 ; mm4=(CbE * FIX(1.77200))=(B-Y)E |
|
156 paddw mm5,mm3 ; mm5=(CbO * FIX(1.77200))=(B-Y)O |
|
157 paddw mm0,mm6 ; mm0=(CrE * FIX(1.40200))=(R-Y)E |
|
158 paddw mm1,mm7 ; mm1=(CrO * FIX(1.40200))=(R-Y)O |
|
159 |
|
160 movq MMWORD [wk(0)], mm4 ; wk(0)=(B-Y)E |
|
161 movq MMWORD [wk(1)], mm5 ; wk(1)=(B-Y)O |
|
162 |
|
163 movq mm4,mm2 |
|
164 movq mm5,mm3 |
|
165 punpcklwd mm2,mm6 |
|
166 punpckhwd mm4,mm6 |
|
167 pmaddwd mm2,[GOTOFF(eax,PW_MF0344_F0285)] |
|
168 pmaddwd mm4,[GOTOFF(eax,PW_MF0344_F0285)] |
|
169 punpcklwd mm3,mm7 |
|
170 punpckhwd mm5,mm7 |
|
171 pmaddwd mm3,[GOTOFF(eax,PW_MF0344_F0285)] |
|
172 pmaddwd mm5,[GOTOFF(eax,PW_MF0344_F0285)] |
|
173 |
|
174 paddd mm2,[GOTOFF(eax,PD_ONEHALF)] |
|
175 paddd mm4,[GOTOFF(eax,PD_ONEHALF)] |
|
176 psrad mm2,SCALEBITS |
|
177 psrad mm4,SCALEBITS |
|
178 paddd mm3,[GOTOFF(eax,PD_ONEHALF)] |
|
179 paddd mm5,[GOTOFF(eax,PD_ONEHALF)] |
|
180 psrad mm3,SCALEBITS |
|
181 psrad mm5,SCALEBITS |
|
182 |
|
183 packssdw mm2,mm4 ; mm2=CbE*-FIX(0.344)+CrE*FIX(0.285) |
|
184 packssdw mm3,mm5 ; mm3=CbO*-FIX(0.344)+CrO*FIX(0.285) |
|
185 psubw mm2,mm6 ; mm2=CbE*-FIX(0.344)+CrE*-FIX(0.714)=(G-Y)E |
|
186 psubw mm3,mm7 ; mm3=CbO*-FIX(0.344)+CrO*-FIX(0.714)=(G-Y)O |
|
187 |
|
188 movq mm5, MMWORD [esi] ; mm5=Y(01234567) |
|
189 |
|
190 pcmpeqw mm4,mm4 |
|
191 psrlw mm4,BYTE_BIT ; mm4={0xFF 0x00 0xFF 0x00 ..} |
|
192 pand mm4,mm5 ; mm4=Y(0246)=YE |
|
193 psrlw mm5,BYTE_BIT ; mm5=Y(1357)=YO |
|
194 |
|
195 paddw mm0,mm4 ; mm0=((R-Y)E+YE)=RE=(R0 R2 R4 R6) |
|
196 paddw mm1,mm5 ; mm1=((R-Y)O+YO)=RO=(R1 R3 R5 R7) |
|
197 packuswb mm0,mm0 ; mm0=(R0 R2 R4 R6 ** ** ** **) |
|
198 packuswb mm1,mm1 ; mm1=(R1 R3 R5 R7 ** ** ** **) |
|
199 |
|
200 paddw mm2,mm4 ; mm2=((G-Y)E+YE)=GE=(G0 G2 G4 G6) |
|
201 paddw mm3,mm5 ; mm3=((G-Y)O+YO)=GO=(G1 G3 G5 G7) |
|
202 packuswb mm2,mm2 ; mm2=(G0 G2 G4 G6 ** ** ** **) |
|
203 packuswb mm3,mm3 ; mm3=(G1 G3 G5 G7 ** ** ** **) |
|
204 |
|
205 paddw mm4, MMWORD [wk(0)] ; mm4=(YE+(B-Y)E)=BE=(B0 B2 B4 B6) |
|
206 paddw mm5, MMWORD [wk(1)] ; mm5=(YO+(B-Y)O)=BO=(B1 B3 B5 B7) |
|
207 packuswb mm4,mm4 ; mm4=(B0 B2 B4 B6 ** ** ** **) |
|
208 packuswb mm5,mm5 ; mm5=(B1 B3 B5 B7 ** ** ** **) |
|
209 |
|
210 %if RGB_PIXELSIZE == 3 ; --------------- |
|
211 |
|
212 ; mmA=(00 02 04 06 ** ** ** **), mmB=(01 03 05 07 ** ** ** **) |
|
213 ; mmC=(10 12 14 16 ** ** ** **), mmD=(11 13 15 17 ** ** ** **) |
|
214 ; mmE=(20 22 24 26 ** ** ** **), mmF=(21 23 25 27 ** ** ** **) |
|
215 ; mmG=(** ** ** ** ** ** ** **), mmH=(** ** ** ** ** ** ** **) |
|
216 |
|
217 punpcklbw mmA,mmC ; mmA=(00 10 02 12 04 14 06 16) |
|
218 punpcklbw mmE,mmB ; mmE=(20 01 22 03 24 05 26 07) |
|
219 punpcklbw mmD,mmF ; mmD=(11 21 13 23 15 25 17 27) |
|
220 |
|
221 movq mmG,mmA |
|
222 movq mmH,mmA |
|
223 punpcklwd mmA,mmE ; mmA=(00 10 20 01 02 12 22 03) |
|
224 punpckhwd mmG,mmE ; mmG=(04 14 24 05 06 16 26 07) |
|
225 |
|
226 psrlq mmH,2*BYTE_BIT ; mmH=(02 12 04 14 06 16 -- --) |
|
227 psrlq mmE,2*BYTE_BIT ; mmE=(22 03 24 05 26 07 -- --) |
|
228 |
|
229 movq mmC,mmD |
|
230 movq mmB,mmD |
|
231 punpcklwd mmD,mmH ; mmD=(11 21 02 12 13 23 04 14) |
|
232 punpckhwd mmC,mmH ; mmC=(15 25 06 16 17 27 -- --) |
|
233 |
|
234 psrlq mmB,2*BYTE_BIT ; mmB=(13 23 15 25 17 27 -- --) |
|
235 |
|
236 movq mmF,mmE |
|
237 punpcklwd mmE,mmB ; mmE=(22 03 13 23 24 05 15 25) |
|
238 punpckhwd mmF,mmB ; mmF=(26 07 17 27 -- -- -- --) |
|
239 |
|
240 punpckldq mmA,mmD ; mmA=(00 10 20 01 11 21 02 12) |
|
241 punpckldq mmE,mmG ; mmE=(22 03 13 23 04 14 24 05) |
|
242 punpckldq mmC,mmF ; mmC=(15 25 06 16 26 07 17 27) |
|
243 |
|
244 cmp ecx, byte SIZEOF_MMWORD |
|
245 jb short .column_st16 |
|
246 |
|
247 movq MMWORD [edi+0*SIZEOF_MMWORD], mmA |
|
248 movq MMWORD [edi+1*SIZEOF_MMWORD], mmE |
|
249 movq MMWORD [edi+2*SIZEOF_MMWORD], mmC |
|
250 |
|
251 sub ecx, byte SIZEOF_MMWORD |
|
252 jz short .nextrow |
|
253 |
|
254 add esi, byte SIZEOF_MMWORD ; inptr0 |
|
255 add ebx, byte SIZEOF_MMWORD ; inptr1 |
|
256 add edx, byte SIZEOF_MMWORD ; inptr2 |
|
257 add edi, byte RGB_PIXELSIZE*SIZEOF_MMWORD ; outptr |
|
258 jmp near .columnloop |
|
259 alignx 16,7 |
|
260 |
|
261 .column_st16: |
|
262 lea ecx, [ecx+ecx*2] ; imul ecx, RGB_PIXELSIZE |
|
263 cmp ecx, byte 2*SIZEOF_MMWORD |
|
264 jb short .column_st8 |
|
265 movq MMWORD [edi+0*SIZEOF_MMWORD], mmA |
|
266 movq MMWORD [edi+1*SIZEOF_MMWORD], mmE |
|
267 movq mmA,mmC |
|
268 sub ecx, byte 2*SIZEOF_MMWORD |
|
269 add edi, byte 2*SIZEOF_MMWORD |
|
270 jmp short .column_st4 |
|
271 .column_st8: |
|
272 cmp ecx, byte SIZEOF_MMWORD |
|
273 jb short .column_st4 |
|
274 movq MMWORD [edi+0*SIZEOF_MMWORD], mmA |
|
275 movq mmA,mmE |
|
276 sub ecx, byte SIZEOF_MMWORD |
|
277 add edi, byte SIZEOF_MMWORD |
|
278 .column_st4: |
|
279 movd eax,mmA |
|
280 cmp ecx, byte SIZEOF_DWORD |
|
281 jb short .column_st2 |
|
282 mov DWORD [edi+0*SIZEOF_DWORD], eax |
|
283 psrlq mmA,DWORD_BIT |
|
284 movd eax,mmA |
|
285 sub ecx, byte SIZEOF_DWORD |
|
286 add edi, byte SIZEOF_DWORD |
|
287 .column_st2: |
|
288 cmp ecx, byte SIZEOF_WORD |
|
289 jb short .column_st1 |
|
290 mov WORD [edi+0*SIZEOF_WORD], ax |
|
291 shr eax,WORD_BIT |
|
292 sub ecx, byte SIZEOF_WORD |
|
293 add edi, byte SIZEOF_WORD |
|
294 .column_st1: |
|
295 cmp ecx, byte SIZEOF_BYTE |
|
296 jb short .nextrow |
|
297 mov BYTE [edi+0*SIZEOF_BYTE], al |
|
298 |
|
299 %else ; RGB_PIXELSIZE == 4 ; ----------- |
|
300 |
|
301 %ifdef RGBX_FILLER_0XFF |
|
302 pcmpeqb mm6,mm6 ; mm6=(X0 X2 X4 X6 ** ** ** **) |
|
303 pcmpeqb mm7,mm7 ; mm7=(X1 X3 X5 X7 ** ** ** **) |
|
304 %else |
|
305 pxor mm6,mm6 ; mm6=(X0 X2 X4 X6 ** ** ** **) |
|
306 pxor mm7,mm7 ; mm7=(X1 X3 X5 X7 ** ** ** **) |
|
307 %endif |
|
308 ; mmA=(00 02 04 06 ** ** ** **), mmB=(01 03 05 07 ** ** ** **) |
|
309 ; mmC=(10 12 14 16 ** ** ** **), mmD=(11 13 15 17 ** ** ** **) |
|
310 ; mmE=(20 22 24 26 ** ** ** **), mmF=(21 23 25 27 ** ** ** **) |
|
311 ; mmG=(30 32 34 36 ** ** ** **), mmH=(31 33 35 37 ** ** ** **) |
|
312 |
|
313 punpcklbw mmA,mmC ; mmA=(00 10 02 12 04 14 06 16) |
|
314 punpcklbw mmE,mmG ; mmE=(20 30 22 32 24 34 26 36) |
|
315 punpcklbw mmB,mmD ; mmB=(01 11 03 13 05 15 07 17) |
|
316 punpcklbw mmF,mmH ; mmF=(21 31 23 33 25 35 27 37) |
|
317 |
|
318 movq mmC,mmA |
|
319 punpcklwd mmA,mmE ; mmA=(00 10 20 30 02 12 22 32) |
|
320 punpckhwd mmC,mmE ; mmC=(04 14 24 34 06 16 26 36) |
|
321 movq mmG,mmB |
|
322 punpcklwd mmB,mmF ; mmB=(01 11 21 31 03 13 23 33) |
|
323 punpckhwd mmG,mmF ; mmG=(05 15 25 35 07 17 27 37) |
|
324 |
|
325 movq mmD,mmA |
|
326 punpckldq mmA,mmB ; mmA=(00 10 20 30 01 11 21 31) |
|
327 punpckhdq mmD,mmB ; mmD=(02 12 22 32 03 13 23 33) |
|
328 movq mmH,mmC |
|
329 punpckldq mmC,mmG ; mmC=(04 14 24 34 05 15 25 35) |
|
330 punpckhdq mmH,mmG ; mmH=(06 16 26 36 07 17 27 37) |
|
331 |
|
332 cmp ecx, byte SIZEOF_MMWORD |
|
333 jb short .column_st16 |
|
334 |
|
335 movq MMWORD [edi+0*SIZEOF_MMWORD], mmA |
|
336 movq MMWORD [edi+1*SIZEOF_MMWORD], mmD |
|
337 movq MMWORD [edi+2*SIZEOF_MMWORD], mmC |
|
338 movq MMWORD [edi+3*SIZEOF_MMWORD], mmH |
|
339 |
|
340 sub ecx, byte SIZEOF_MMWORD |
|
341 jz short .nextrow |
|
342 |
|
343 add esi, byte SIZEOF_MMWORD ; inptr0 |
|
344 add ebx, byte SIZEOF_MMWORD ; inptr1 |
|
345 add edx, byte SIZEOF_MMWORD ; inptr2 |
|
346 add edi, byte RGB_PIXELSIZE*SIZEOF_MMWORD ; outptr |
|
347 jmp near .columnloop |
|
348 alignx 16,7 |
|
349 |
|
350 .column_st16: |
|
351 cmp ecx, byte SIZEOF_MMWORD/2 |
|
352 jb short .column_st8 |
|
353 movq MMWORD [edi+0*SIZEOF_MMWORD], mmA |
|
354 movq MMWORD [edi+1*SIZEOF_MMWORD], mmD |
|
355 movq mmA,mmC |
|
356 movq mmD,mmH |
|
357 sub ecx, byte SIZEOF_MMWORD/2 |
|
358 add edi, byte 2*SIZEOF_MMWORD |
|
359 .column_st8: |
|
360 cmp ecx, byte SIZEOF_MMWORD/4 |
|
361 jb short .column_st4 |
|
362 movq MMWORD [edi+0*SIZEOF_MMWORD], mmA |
|
363 movq mmA,mmD |
|
364 sub ecx, byte SIZEOF_MMWORD/4 |
|
365 add edi, byte 1*SIZEOF_MMWORD |
|
366 .column_st4: |
|
367 cmp ecx, byte SIZEOF_MMWORD/8 |
|
368 jb short .nextrow |
|
369 movd DWORD [edi+0*SIZEOF_DWORD], mmA |
|
370 |
|
371 %endif ; RGB_PIXELSIZE ; --------------- |
|
372 |
|
373 alignx 16,7 |
|
374 |
|
375 .nextrow: |
|
376 pop ecx |
|
377 pop esi |
|
378 pop ebx |
|
379 pop edx |
|
380 pop edi |
|
381 pop eax |
|
382 |
|
383 add esi, byte SIZEOF_JSAMPROW |
|
384 add ebx, byte SIZEOF_JSAMPROW |
|
385 add edx, byte SIZEOF_JSAMPROW |
|
386 add edi, byte SIZEOF_JSAMPROW ; output_buf |
|
387 dec eax ; num_rows |
|
388 jg near .rowloop |
|
389 |
|
390 emms ; empty MMX state |
|
391 |
|
392 .return: |
|
393 pop edi |
|
394 pop esi |
|
395 ; pop edx ; need not be preserved |
|
396 ; pop ecx ; need not be preserved |
|
397 pop ebx |
|
398 mov esp,ebp ; esp <- aligned ebp |
|
399 pop esp ; esp <- original ebp |
|
400 pop ebp |
|
401 ret |
|
402 |
|
403 ; For some reason, the OS X linker does not honor the request to align the |
|
404 ; segment unless we do this. |
|
405 align 16 |