move mask invert out

This commit is contained in:
Ryan Oldenburg 2022-06-29 00:37:51 -05:00
parent cf81556100
commit 8a07e3cf9b
2 changed files with 20 additions and 10 deletions

View file

@ -213,17 +213,15 @@ proc getValueSmooth*(mask: Mask, x, y: float32): uint8 {.raises: [].} =
proc invert*(mask: Mask) {.raises: [].} =
## Inverts all of the values - creates a negative of the mask.
var i: int
when defined(amd64) and allowSimd:
let vec255 = mm_set1_epi8(255)
for _ in 0 ..< mask.data.len div 16:
var values = mm_loadu_si128(mask.data[i].addr)
values = mm_sub_epi8(vec255, values)
mm_storeu_si128(mask.data[i].addr, values)
i += 16
when allowSimd and compiles(invertImageSimd):
invertMaskSimd(
cast[ptr UncheckedArray[uint8]](mask.data[0].addr),
mask.data.len
)
return
for j in i ..< mask.data.len:
mask.data[j] = 255 - mask.data[j]
for i in 0 ..< mask.data.len:
mask.data[i] = 255 - mask.data[i]
proc spread*(mask: Mask, spread: float32) {.raises: [PixieError].} =
## Grows the mask by spread.

View file

@ -255,3 +255,15 @@ when defined(amd64):
data[i] = rgbx
toPremultipliedAlphaSimd(cast[ptr UncheckedArray[uint32]](data), len)
proc invertMaskSimd*(data: ptr UncheckedArray[uint8], len: int) =
var i: int
let vec255 = mm_set1_epi8(255)
for _ in 0 ..< len div 16:
var values = mm_loadu_si128(data[i].addr)
values = mm_sub_epi8(vec255, values)
mm_storeu_si128(data[i].addr, values)
i += 16
for j in i ..< len:
data[j] = 255 - data[j]