MyouProject/pixie
2
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Perf improvement for isOneColor.

Browse source
This commit is contained in:
treeform 2021-08-17 20:45:42 -07:00
parent ec8f812e66
commit fdef525352
1 changed files with 7 additions and 5 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

12
src/pixie/images.nim
View file

@ -135,13 +135,15 @@ proc isOneColor*(image: Image): bool =
var i: int var i: int
when defined(amd64) and not defined(pixieNoSimd): when defined(amd64) and not defined(pixieNoSimd):
let colorVec = mm_set1_epi32(cast[int32](color)) let colorVec = mm_set1_epi32(cast[int32](color))
for j in countup(0, image.data.len - 4, 4): for j in countup(0, image.data.len - 8, 8):
let let
values = mm_loadu_si128(image.data[j].addr) values0 = mm_loadu_si128(image.data[j].addr)
mask = mm_movemask_epi8(mm_cmpeq_epi8(values, colorVec)) values1 = mm_loadu_si128(image.data[j + 4].addr)
if mask != uint16.high.int: mask0 = mm_movemask_epi8(mm_cmpeq_epi8(values0, colorVec))
mask1 = mm_movemask_epi8(mm_cmpeq_epi8(values1, colorVec))
if mask0 != uint16.high.int or mask1 != uint16.high.int:
return false return false
i += 4 i += 8
for j in i ..< image.data.len: for j in i ..< image.data.len:
if image.data[j] != color: if image.data[j] != color: