orga2-simd/filtros/diff_asm.asm

; asmsyntax=nasm

default rel
global _diff_asm
global diff_asm

; Alias para no volverme loco con los registros
%define src rdi
%define src2 rsi
%define dst rdx
%define cont rcx
%define size r8

; Algunos defines ultiles
%define PIXEL_SIZE  4

; Mascaras y esas cosas
section .data
	BLUE_MASK:	db 0,0,0,255,4,4,4,255,8,8,8,255,12,12,12,255 ; TODO: Ver que hacer con el alpha
	;TODO: Si el alpha viene en 255 entonces puedo poner el mismo byte que vino!!!!
	GREEN_MASK:	db 1,1,1,255,5,5,5,255,9,9,9,255,13,13,13,255
	RED_MASK:	db 2,2,2,255,6,6,6,255,10,10,10,255,14,14,14,255

section .text
;void diff_asm    (
    ;unsigned char *src, unsigned char *src2, unsigned char *dst, int filas, int cols)

_diff_asm:
diff_asm:
    push rbp
    mov rbp, rsp                ; Stack frame

.calc_size:                     ; Calculo el tamaño de la imagen en bytes
    mov r9, dst                 ; Muevo el destino un rato para hacer el calculo del tamaño de la imagen
    mov rax, cont               ; Cargo filas
    mul size                    ; Multiplico por las columnas
    xor dst, dst                ; Limpio temportal
    mov dst, PIXEL_SIZE         ; Multiplico temporal
    mul dst                     ; Multiplico por el tamaño del pixel
    mov size, rax               ; Guardo el tamaño
    mov dst, r9
    xor cont, cont              ; Limpio el contador
;    movdqu xmm3, [DIVIDIR]      ; Cargo el valor para las divisiones para el promedio

.loop:
    cmp cont, size              ; Comparo tamaño con el contador
    jge .fin                    ; Si ya recorri la imagen voy al final
    pxor xmm1, xmm1
    movdqu xmm0, [src]          ; Cargo la imagen 1 => xmm0 = |B(P1)|G(P1)|R(P1)|A(P1)|B(P2)|G(P2)|......|
    movdqu xmm1, [src2]         ; Cargo la imagen 2 => xmm1 = |B(P1)|G(P1)|R(P1)|A(P1)|B(P2)|G(P2)|......|

.cuentas:                       ; Resto componente a componente de 4 pixeles de cada imagen
    psubb xmm1, xmm0            ; xmm1 = |B(img1_P1) - R(img2_P1)|G(Img1_P1) - Img2_P2|....|
    pabsb xmm0, xmm1            ; Almaceno en xmm0 el valor absoluto de la resta anterior

.max:
	; TODO: Mover afuera del ciclo
	movdqu xmm15, [BLUE_MASK]
	movdqu xmm14, [GREEN_MASK]
	movdqu xmm13, [RED_MASK]

	movdqu xmm2, xmm0
	movdqu xmm3, xmm0

	pshufb xmm0, xmm15			; Pongo en la misma posición cada elemento maximo
	pshufb xmm2, xmm14
	pshufb xmm3, xmm13

	pmaxub xmm2, xmm3
	pmaxub xmm0, xmm2

.write:
    movdqu [dst], xmm0          ; Escribo en el destino
.endLoop:
    add src, PIXEL_SIZE * 4     ; Incremento 4 pixeles
    add src2, PIXEL_SIZE * 4
    add dst, PIXEL_SIZE * 4
    add cont, PIXEL_SIZE * 4
    jmp .loop

.fin:
    pop rbp
    ret