composewell/streamly
1
1
Fork 0
mirror of https://github.com/composewell/streamly.git synced 2024-08-16 23:10:36 +03:00
Code Issues Packages Projects Releases Wiki Activity

mid xor

Browse Source
This commit is contained in:
Adithya Kumar 2023-08-23 04:37:43 +05:30
parent 2c7d6ebd9d
commit a8e5a4e100
1 changed files with 43 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

50
core/src/Streamly/Internal/Data/Serialize/RecordTH.hs
View File

@ -286,19 +286,54 @@ memcmpCStr ptr0 arr off len = go ptr0 off
w8To16 :: Word8 -> Word16
w8To16 = fromIntegral
{-# INLINE xorCull #-}
xorCull :: (Integral a, Bits a) => a -> a -> Word8
xorCull a b = fromIntegral (xor a b)
{-# INLINE xorFill #-}
xorFill :: (Integral a, Bits a) => a -> a -> Word64
xorFill a b = fromIntegral (xor a b)
shiftAdd conv xs =
foldl' (.|.) zeroBits $
map (\(j, x) -> shiftL x (j * 8)) $
zip [(length xs - 1),(length xs - 2) .. 0] $ map conv xs
-- XXX Single byte comparision seems to be the fastest!
-- XXX This is architecture dependent?
-- XXX Little endian did not work?
xorCmp :: [Word8] -> Name -> Name -> Q Exp
{-
xorCmp tag arr off
| length tag > 8 = [|$(go 0) == zeroBits|]
where
tagLen = length tag
last8Off = tagLen - 8
go i
| i >= tagLen = [|zeroBits|]
go i
| i > last8Off = go last8Off
go i = do
let ty = [t|Word64|]
offInc = 8
wIntegral =
litIntegral
(shiftAdd
w8To16
[ tag !! i
, tag !! (i + 1)
, tag !! (i + 2)
, tag !! (i + 3)
, tag !! (i + 4)
, tag !! (i + 5)
, tag !! (i + 6)
, tag !! (i + 7)
])
[|xor (unsafeInlineIO
(Unbox.peekByteIndex
($(varE off) + $(litIntegral i))
$(varE arr)))
($(wIntegral) :: $(ty)) .|.
$(go (i + offInc))|]
-}
xorCmp tag arr off = [|$(go 0) == zeroBits|]
where
shiftAdd conv xs =
foldl' (.|.) zeroBits $
map (\(j, x) -> shiftL x (j * 8)) $
zip [(length xs - 1),(length xs - 2) .. 0] $ map conv xs
tagLen = length tag
go i
| i >= tagLen = [|zeroBits|]
@ -347,6 +382,7 @@ xorCmp tag arr off = [|$(go 0) == zeroBits|]
, tag !! (i + 6)
, tag !! (i + 7)
])
[|xor
(unsafeInlineIO
(Unbox.peekByteIndex