ron-0.13: lib/Data/ZigZag.hs

-- Copyright (c) 2016, Pasqualino `Titto` Assini

module Data.ZigZag (
    zzEncode,
    zzDecode64,
) where

import RON.Prelude

import Data.Bits (
    Bits,
    FiniteBits,
    finiteBitSize,
    shiftL,
    shiftR,
    xor,
    (.&.),
 )

{-# SPECIALIZE INLINE zzEncode :: Int8 -> Word8 #-}
{-# SPECIALIZE INLINE zzEncode :: Int16 -> Word16 #-}
{-# SPECIALIZE INLINE zzEncode :: Int32 -> Word32 #-}
{-# SPECIALIZE INLINE zzEncode :: Int64 -> Word64 #-}
zzEncode :: (FiniteBits a, Integral a, Num b) => a -> b
zzEncode w =
    fromIntegral ((w `shiftL` 1) `xor` (w `shiftR` (finiteBitSize w - 1)))

{-# INLINE zzDecode #-}
zzDecode :: (Bits a1, Integral a1, Num a) => a1 -> a
zzDecode w = fromIntegral ((w `shiftR` 1) `xor` negate (w .&. 1))

{-# INLINE zzDecode64 #-}
zzDecode64 :: Word64 -> Int64
zzDecode64 = zzDecode