{-# LANGUAGE DefaultSignatures, ScopedTypeVariables #-}

-- From http://stackoverflow.com/a/15911213

module Data.Text.ICU.BitMask
       (
    -- * Bit mask twiddling API
    -- $api
    -- * Types
      ToBitMask(..)
    -- * Functions
    , fromBitMask
    , highestValueInBitMask
    ) where

import Data.Bits ((.&.), (.|.))
import Data.Maybe (listToMaybe)

-- $api
-- Conversion to and from enumerated types representable as
-- a compact bitmask.

class ToBitMask a where
  toBitMask :: a -> Int

instance (ToBitMask a) => ToBitMask [a] where
  toBitMask :: [a] -> Int
toBitMask = forall (t :: * -> *) a b.
Foldable t =>
(a -> b -> b) -> b -> t a -> b
foldr (forall a. Bits a => a -> a -> a
(.|.) forall b c a. (b -> c) -> (a -> b) -> a -> c
. forall a. ToBitMask a => a -> Int
toBitMask) Int
0

fromBitMask :: (Enum a, Bounded a, ToBitMask a) => Int -> [a]
fromBitMask :: forall a. (Enum a, Bounded a, ToBitMask a) => Int -> [a]
fromBitMask Int
bm = forall a. (a -> Bool) -> [a] -> [a]
filter forall {a}. ToBitMask a => a -> Bool
inBitMask forall a b. (a -> b) -> a -> b
$ forall a. Enum a => a -> [a]
enumFrom forall a. Bounded a => a
minBound
  where inBitMask :: a -> Bool
inBitMask a
val = (Int
bm forall a. Bits a => a -> a -> a
.&. forall a. ToBitMask a => a -> Int
toBitMask a
val) forall a. Eq a => a -> a -> Bool
== forall a. ToBitMask a => a -> Int
toBitMask a
val

highestValueInBitMask :: (Enum a, Bounded a, ToBitMask a) => Int -> Maybe a
highestValueInBitMask :: forall a. (Enum a, Bounded a, ToBitMask a) => Int -> Maybe a
highestValueInBitMask = forall a. [a] -> Maybe a
listToMaybe forall b c a. (b -> c) -> (a -> b) -> a -> c
. forall a. [a] -> [a]
reverse forall b c a. (b -> c) -> (a -> b) -> a -> c
. forall a. (Enum a, Bounded a, ToBitMask a) => Int -> [a]
fromBitMask