module Swarm.Util.UnitInterval (
UnitInterval,
getValue,
mkInterval,
safeIndex,
) where
import Data.List.NonEmpty (NonEmpty, (!!))
import Prelude hiding ((!!))
newtype UnitInterval a = UnitInterval
{ forall a. UnitInterval a -> a
getValue :: a
}
mkInterval :: (Ord a, Num a) => a -> UnitInterval a
mkInterval :: forall a. (Ord a, Num a) => a -> UnitInterval a
mkInterval = forall a. a -> UnitInterval a
UnitInterval forall b c a. (b -> c) -> (a -> b) -> a -> c
. forall a. Ord a => a -> a -> a
max a
0 forall b c a. (b -> c) -> (a -> b) -> a -> c
. forall a. Ord a => a -> a -> a
min a
1
safeIndex ::
RealFrac a =>
UnitInterval a ->
NonEmpty b ->
b
safeIndex :: forall a b. RealFrac a => UnitInterval a -> NonEmpty b -> b
safeIndex (UnitInterval a
alpha) NonEmpty b
xs =
NonEmpty b
xs forall a. NonEmpty a -> Int -> a
!! forall a b. (RealFrac a, Integral b) => a -> b
floor (a
alpha forall a. Num a => a -> a -> a
* forall a b. (Integral a, Num b) => a -> b
fromIntegral (forall (t :: * -> *) a. Foldable t => t a -> Int
length NonEmpty b
xs forall a. Num a => a -> a -> a
- Int
1))