{-# OPTIONS_HADDOCK not-home #-}
{-# LANGUAGE GeneralizedNewtypeDeriving #-}
module Hedgehog.Internal.Range (
Size(..)
, Range(..)
, origin
, bounds
, lowerBound
, upperBound
, singleton
, constant
, constantFrom
, constantBounded
, linear
, linearFrom
, linearFrac
, linearFracFrom
, linearBounded
, exponential
, exponentialFrom
, exponentialBounded
, exponentialFloat
, exponentialFloatFrom
, clamp
, scaleLinear
, scaleLinearFrac
, scaleExponential
, scaleExponentialFloat
) where
import Data.Bifunctor (bimap)
import Prelude hiding (minimum, maximum)
newtype Size =
Size {
Size -> Int
unSize :: Int
} deriving (Size -> Size -> Bool
(Size -> Size -> Bool) -> (Size -> Size -> Bool) -> Eq Size
forall a. (a -> a -> Bool) -> (a -> a -> Bool) -> Eq a
$c== :: Size -> Size -> Bool
== :: Size -> Size -> Bool
$c/= :: Size -> Size -> Bool
/= :: Size -> Size -> Bool
Eq, Eq Size
Eq Size =>
(Size -> Size -> Ordering)
-> (Size -> Size -> Bool)
-> (Size -> Size -> Bool)
-> (Size -> Size -> Bool)
-> (Size -> Size -> Bool)
-> (Size -> Size -> Size)
-> (Size -> Size -> Size)
-> Ord Size
Size -> Size -> Bool
Size -> Size -> Ordering
Size -> Size -> Size
forall a.
Eq a =>
(a -> a -> Ordering)
-> (a -> a -> Bool)
-> (a -> a -> Bool)
-> (a -> a -> Bool)
-> (a -> a -> Bool)
-> (a -> a -> a)
-> (a -> a -> a)
-> Ord a
$ccompare :: Size -> Size -> Ordering
compare :: Size -> Size -> Ordering
$c< :: Size -> Size -> Bool
< :: Size -> Size -> Bool
$c<= :: Size -> Size -> Bool
<= :: Size -> Size -> Bool
$c> :: Size -> Size -> Bool
> :: Size -> Size -> Bool
$c>= :: Size -> Size -> Bool
>= :: Size -> Size -> Bool
$cmax :: Size -> Size -> Size
max :: Size -> Size -> Size
$cmin :: Size -> Size -> Size
min :: Size -> Size -> Size
Ord, Integer -> Size
Size -> Size
Size -> Size -> Size
(Size -> Size -> Size)
-> (Size -> Size -> Size)
-> (Size -> Size -> Size)
-> (Size -> Size)
-> (Size -> Size)
-> (Size -> Size)
-> (Integer -> Size)
-> Num Size
forall a.
(a -> a -> a)
-> (a -> a -> a)
-> (a -> a -> a)
-> (a -> a)
-> (a -> a)
-> (a -> a)
-> (Integer -> a)
-> Num a
$c+ :: Size -> Size -> Size
+ :: Size -> Size -> Size
$c- :: Size -> Size -> Size
- :: Size -> Size -> Size
$c* :: Size -> Size -> Size
* :: Size -> Size -> Size
$cnegate :: Size -> Size
negate :: Size -> Size
$cabs :: Size -> Size
abs :: Size -> Size
$csignum :: Size -> Size
signum :: Size -> Size
$cfromInteger :: Integer -> Size
fromInteger :: Integer -> Size
Num, Num Size
Ord Size
(Num Size, Ord Size) => (Size -> Rational) -> Real Size
Size -> Rational
forall a. (Num a, Ord a) => (a -> Rational) -> Real a
$ctoRational :: Size -> Rational
toRational :: Size -> Rational
Real, Int -> Size
Size -> Int
Size -> [Size]
Size -> Size
Size -> Size -> [Size]
Size -> Size -> Size -> [Size]
(Size -> Size)
-> (Size -> Size)
-> (Int -> Size)
-> (Size -> Int)
-> (Size -> [Size])
-> (Size -> Size -> [Size])
-> (Size -> Size -> [Size])
-> (Size -> Size -> Size -> [Size])
-> Enum Size
forall a.
(a -> a)
-> (a -> a)
-> (Int -> a)
-> (a -> Int)
-> (a -> [a])
-> (a -> a -> [a])
-> (a -> a -> [a])
-> (a -> a -> a -> [a])
-> Enum a
$csucc :: Size -> Size
succ :: Size -> Size
$cpred :: Size -> Size
pred :: Size -> Size
$ctoEnum :: Int -> Size
toEnum :: Int -> Size
$cfromEnum :: Size -> Int
fromEnum :: Size -> Int
$cenumFrom :: Size -> [Size]
enumFrom :: Size -> [Size]
$cenumFromThen :: Size -> Size -> [Size]
enumFromThen :: Size -> Size -> [Size]
$cenumFromTo :: Size -> Size -> [Size]
enumFromTo :: Size -> Size -> [Size]
$cenumFromThenTo :: Size -> Size -> Size -> [Size]
enumFromThenTo :: Size -> Size -> Size -> [Size]
Enum, Enum Size
Real Size
(Real Size, Enum Size) =>
(Size -> Size -> Size)
-> (Size -> Size -> Size)
-> (Size -> Size -> Size)
-> (Size -> Size -> Size)
-> (Size -> Size -> (Size, Size))
-> (Size -> Size -> (Size, Size))
-> (Size -> Integer)
-> Integral Size
Size -> Integer
Size -> Size -> (Size, Size)
Size -> Size -> Size
forall a.
(Real a, Enum a) =>
(a -> a -> a)
-> (a -> a -> a)
-> (a -> a -> a)
-> (a -> a -> a)
-> (a -> a -> (a, a))
-> (a -> a -> (a, a))
-> (a -> Integer)
-> Integral a
$cquot :: Size -> Size -> Size
quot :: Size -> Size -> Size
$crem :: Size -> Size -> Size
rem :: Size -> Size -> Size
$cdiv :: Size -> Size -> Size
div :: Size -> Size -> Size
$cmod :: Size -> Size -> Size
mod :: Size -> Size -> Size
$cquotRem :: Size -> Size -> (Size, Size)
quotRem :: Size -> Size -> (Size, Size)
$cdivMod :: Size -> Size -> (Size, Size)
divMod :: Size -> Size -> (Size, Size)
$ctoInteger :: Size -> Integer
toInteger :: Size -> Integer
Integral)
instance Show Size where
showsPrec :: Int -> Size -> ShowS
showsPrec Int
p (Size Int
x) =
Bool -> ShowS -> ShowS
showParen (Int
p Int -> Int -> Bool
forall a. Ord a => a -> a -> Bool
> Int
10) (ShowS -> ShowS) -> ShowS -> ShowS
forall a b. (a -> b) -> a -> b
$
String -> ShowS
showString String
"Size " ShowS -> ShowS -> ShowS
forall b c a. (b -> c) -> (a -> b) -> a -> c
.
Int -> Int -> ShowS
forall a. Show a => Int -> a -> ShowS
showsPrec Int
11 Int
x
instance Read Size where
readsPrec :: Int -> ReadS Size
readsPrec Int
p =
Bool -> ReadS Size -> ReadS Size
forall a. Bool -> ReadS a -> ReadS a
readParen (Int
p Int -> Int -> Bool
forall a. Ord a => a -> a -> Bool
> Int
10) (ReadS Size -> ReadS Size) -> ReadS Size -> ReadS Size
forall a b. (a -> b) -> a -> b
$ \String
r0 -> do
(String
"Size", String
r1) <- ReadS String
lex String
r0
(Int
s, String
r2) <- Int -> ReadS Int
forall a. Read a => Int -> ReadS a
readsPrec Int
11 String
r1
(Size, String) -> [(Size, String)]
forall a. a -> [a]
forall (f :: * -> *) a. Applicative f => a -> f a
pure (Int -> Size
Size Int
s, String
r2)
data Range a =
Range !a (Size -> (a, a))
instance Functor Range where
fmap :: forall a b. (a -> b) -> Range a -> Range b
fmap a -> b
f (Range a
z Size -> (a, a)
g) =
b -> (Size -> (b, b)) -> Range b
forall a. a -> (Size -> (a, a)) -> Range a
Range (a -> b
f a
z) ((Size -> (b, b)) -> Range b) -> (Size -> (b, b)) -> Range b
forall a b. (a -> b) -> a -> b
$ \Size
sz ->
(a -> b) -> (a -> b) -> (a, a) -> (b, b)
forall a b c d. (a -> b) -> (c -> d) -> (a, c) -> (b, d)
forall (p :: * -> * -> *) a b c d.
Bifunctor p =>
(a -> b) -> (c -> d) -> p a c -> p b d
bimap a -> b
f a -> b
f (Size -> (a, a)
g Size
sz)
origin :: Range a -> a
origin :: forall a. Range a -> a
origin (Range a
z Size -> (a, a)
_) =
a
z
bounds :: Size -> Range a -> (a, a)
bounds :: forall a. Size -> Range a -> (a, a)
bounds Size
sz (Range a
_ Size -> (a, a)
f) =
Size -> (a, a)
f Size
sz
lowerBound :: Ord a => Size -> Range a -> a
lowerBound :: forall a. Ord a => Size -> Range a -> a
lowerBound Size
sz Range a
range =
let
(a
x, a
y) =
Size -> Range a -> (a, a)
forall a. Size -> Range a -> (a, a)
bounds Size
sz Range a
range
in
a -> a -> a
forall a. Ord a => a -> a -> a
min a
x a
y
upperBound :: Ord a => Size -> Range a -> a
upperBound :: forall a. Ord a => Size -> Range a -> a
upperBound Size
sz Range a
range =
let
(a
x, a
y) =
Size -> Range a -> (a, a)
forall a. Size -> Range a -> (a, a)
bounds Size
sz Range a
range
in
a -> a -> a
forall a. Ord a => a -> a -> a
max a
x a
y
singleton :: a -> Range a
singleton :: forall a. a -> Range a
singleton a
x =
a -> (Size -> (a, a)) -> Range a
forall a. a -> (Size -> (a, a)) -> Range a
Range a
x ((Size -> (a, a)) -> Range a) -> (Size -> (a, a)) -> Range a
forall a b. (a -> b) -> a -> b
$ \Size
_ -> (a
x, a
x)
constant :: a -> a -> Range a
constant :: forall a. a -> a -> Range a
constant a
x a
y =
a -> a -> a -> Range a
forall a. a -> a -> a -> Range a
constantFrom a
x a
x a
y
constantFrom ::
a
-> a
-> a
-> Range a
constantFrom :: forall a. a -> a -> a -> Range a
constantFrom a
z a
x a
y =
a -> (Size -> (a, a)) -> Range a
forall a. a -> (Size -> (a, a)) -> Range a
Range a
z ((Size -> (a, a)) -> Range a) -> (Size -> (a, a)) -> Range a
forall a b. (a -> b) -> a -> b
$ \Size
_ -> (a
x, a
y)
constantBounded :: (Bounded a, Num a) => Range a
constantBounded :: forall a. (Bounded a, Num a) => Range a
constantBounded =
a -> a -> a -> Range a
forall a. a -> a -> a -> Range a
constantFrom a
0 a
forall a. Bounded a => a
minBound a
forall a. Bounded a => a
maxBound
linear :: Integral a => a -> a -> Range a
linear :: forall a. Integral a => a -> a -> Range a
linear a
x a
y =
a -> a -> a -> Range a
forall a. Integral a => a -> a -> a -> Range a
linearFrom a
x a
x a
y
linearFrom :: Integral a
=> a
-> a
-> a
-> Range a
linearFrom :: forall a. Integral a => a -> a -> a -> Range a
linearFrom a
z a
x a
y =
a -> (Size -> (a, a)) -> Range a
forall a. a -> (Size -> (a, a)) -> Range a
Range a
z ((Size -> (a, a)) -> Range a) -> (Size -> (a, a)) -> Range a
forall a b. (a -> b) -> a -> b
$ \Size
sz ->
let
x_sized :: a
x_sized =
a -> a -> a -> a
forall a. Ord a => a -> a -> a -> a
clamp a
x a
y (a -> a) -> a -> a
forall a b. (a -> b) -> a -> b
$ Size -> a -> a -> a
forall a. Integral a => Size -> a -> a -> a
scaleLinear Size
sz a
z a
x
y_sized :: a
y_sized =
a -> a -> a -> a
forall a. Ord a => a -> a -> a -> a
clamp a
x a
y (a -> a) -> a -> a
forall a b. (a -> b) -> a -> b
$ Size -> a -> a -> a
forall a. Integral a => Size -> a -> a -> a
scaleLinear Size
sz a
z a
y
in
(a
x_sized, a
y_sized)
linearBounded :: (Bounded a, Integral a) => Range a
linearBounded :: forall a. (Bounded a, Integral a) => Range a
linearBounded =
a -> a -> a -> Range a
forall a. Integral a => a -> a -> a -> Range a
linearFrom a
0 a
forall a. Bounded a => a
minBound a
forall a. Bounded a => a
maxBound
linearFrac :: (Fractional a, Ord a) => a -> a -> Range a
linearFrac :: forall a. (Fractional a, Ord a) => a -> a -> Range a
linearFrac a
x a
y =
a -> a -> a -> Range a
forall a. (Fractional a, Ord a) => a -> a -> a -> Range a
linearFracFrom a
x a
x a
y
linearFracFrom :: (Fractional a, Ord a) => a -> a -> a -> Range a
linearFracFrom :: forall a. (Fractional a, Ord a) => a -> a -> a -> Range a
linearFracFrom a
z a
x a
y =
a -> (Size -> (a, a)) -> Range a
forall a. a -> (Size -> (a, a)) -> Range a
Range a
z ((Size -> (a, a)) -> Range a) -> (Size -> (a, a)) -> Range a
forall a b. (a -> b) -> a -> b
$ \Size
sz ->
let
x_sized :: a
x_sized =
a -> a -> a -> a
forall a. Ord a => a -> a -> a -> a
clamp a
x a
y (a -> a) -> a -> a
forall a b. (a -> b) -> a -> b
$ Size -> a -> a -> a
forall a. Fractional a => Size -> a -> a -> a
scaleLinearFrac Size
sz a
z a
x
y_sized :: a
y_sized =
a -> a -> a -> a
forall a. Ord a => a -> a -> a -> a
clamp a
x a
y (a -> a) -> a -> a
forall a b. (a -> b) -> a -> b
$ Size -> a -> a -> a
forall a. Fractional a => Size -> a -> a -> a
scaleLinearFrac Size
sz a
z a
y
in
(a
x_sized, a
y_sized)
clamp :: Ord a => a -> a -> a -> a
clamp :: forall a. Ord a => a -> a -> a -> a
clamp a
x a
y a
n =
if a
x a -> a -> Bool
forall a. Ord a => a -> a -> Bool
> a
y then
a -> a -> a
forall a. Ord a => a -> a -> a
min a
x (a -> a -> a
forall a. Ord a => a -> a -> a
max a
y a
n)
else
a -> a -> a
forall a. Ord a => a -> a -> a
min a
y (a -> a -> a
forall a. Ord a => a -> a -> a
max a
x a
n)
scaleLinear :: Integral a => Size -> a -> a -> a
scaleLinear :: forall a. Integral a => Size -> a -> a -> a
scaleLinear Size
sz0 a
z0 a
n0 =
let
sz :: Size
sz =
Size -> Size -> Size
forall a. Ord a => a -> a -> a
max Size
0 (Size -> Size -> Size
forall a. Ord a => a -> a -> a
min Size
99 Size
sz0)
z :: Integer
z =
a -> Integer
forall a. Integral a => a -> Integer
toInteger a
z0
n :: Integer
n =
a -> Integer
forall a. Integral a => a -> Integer
toInteger a
n0
rng :: Integer
rng =
Integer
n Integer -> Integer -> Integer
forall a. Num a => a -> a -> a
- Integer
z Integer -> Integer -> Integer
forall a. Num a => a -> a -> a
+ Integer -> Integer
forall a. Num a => a -> a
signum (Integer
n Integer -> Integer -> Integer
forall a. Num a => a -> a -> a
- Integer
z)
diff :: Integer
diff =
(Integer
rng Integer -> Integer -> Integer
forall a. Num a => a -> a -> a
* Size -> Integer
forall a b. (Integral a, Num b) => a -> b
fromIntegral Size
sz) Integer -> Integer -> Integer
forall a. Integral a => a -> a -> a
`quot` Integer
100
in
Integer -> a
forall a. Num a => Integer -> a
fromInteger (Integer -> a) -> Integer -> a
forall a b. (a -> b) -> a -> b
$ Integer
z Integer -> Integer -> Integer
forall a. Num a => a -> a -> a
+ Integer
diff
scaleLinearFrac :: Fractional a => Size -> a -> a -> a
scaleLinearFrac :: forall a. Fractional a => Size -> a -> a -> a
scaleLinearFrac Size
sz0 a
z a
n =
let
sz :: Size
sz =
Size -> Size -> Size
forall a. Ord a => a -> a -> a
max Size
0 (Size -> Size -> Size
forall a. Ord a => a -> a -> a
min Size
99 Size
sz0)
diff :: a
diff =
(a
n a -> a -> a
forall a. Num a => a -> a -> a
- a
z) a -> a -> a
forall a. Num a => a -> a -> a
* (Size -> a
forall a b. (Integral a, Num b) => a -> b
fromIntegral Size
sz a -> a -> a
forall a. Fractional a => a -> a -> a
/ a
99)
in
a
z a -> a -> a
forall a. Num a => a -> a -> a
+ a
diff
exponential :: Integral a => a -> a -> Range a
exponential :: forall a. Integral a => a -> a -> Range a
exponential a
x a
y =
a -> a -> a -> Range a
forall a. Integral a => a -> a -> a -> Range a
exponentialFrom a
x a
x a
y
exponentialFrom :: Integral a
=> a
-> a
-> a
-> Range a
exponentialFrom :: forall a. Integral a => a -> a -> a -> Range a
exponentialFrom a
z a
x a
y =
a -> (Size -> (a, a)) -> Range a
forall a. a -> (Size -> (a, a)) -> Range a
Range a
z ((Size -> (a, a)) -> Range a) -> (Size -> (a, a)) -> Range a
forall a b. (a -> b) -> a -> b
$ \Size
sz ->
let
sized_x :: a
sized_x =
a -> a -> a -> a
forall a. Ord a => a -> a -> a -> a
clamp a
x a
y (a -> a) -> a -> a
forall a b. (a -> b) -> a -> b
$ Size -> a -> a -> a
forall a. Integral a => Size -> a -> a -> a
scaleExponential Size
sz a
z a
x
sized_y :: a
sized_y =
a -> a -> a -> a
forall a. Ord a => a -> a -> a -> a
clamp a
x a
y (a -> a) -> a -> a
forall a b. (a -> b) -> a -> b
$ Size -> a -> a -> a
forall a. Integral a => Size -> a -> a -> a
scaleExponential Size
sz a
z a
y
in
(a
sized_x, a
sized_y)
exponentialBounded :: (Bounded a, Integral a) => Range a
exponentialBounded :: forall a. (Bounded a, Integral a) => Range a
exponentialBounded =
a -> a -> a -> Range a
forall a. Integral a => a -> a -> a -> Range a
exponentialFrom a
0 a
forall a. Bounded a => a
minBound a
forall a. Bounded a => a
maxBound
exponentialFloat :: (Floating a, Ord a) => a -> a -> Range a
exponentialFloat :: forall a. (Floating a, Ord a) => a -> a -> Range a
exponentialFloat a
x a
y =
a -> a -> a -> Range a
forall a. (Floating a, Ord a) => a -> a -> a -> Range a
exponentialFloatFrom a
x a
x a
y
exponentialFloatFrom :: (Floating a, Ord a) => a -> a -> a -> Range a
exponentialFloatFrom :: forall a. (Floating a, Ord a) => a -> a -> a -> Range a
exponentialFloatFrom a
z a
x a
y =
a -> (Size -> (a, a)) -> Range a
forall a. a -> (Size -> (a, a)) -> Range a
Range a
z ((Size -> (a, a)) -> Range a) -> (Size -> (a, a)) -> Range a
forall a b. (a -> b) -> a -> b
$ \Size
sz ->
let
sized_x :: a
sized_x =
a -> a -> a -> a
forall a. Ord a => a -> a -> a -> a
clamp a
x a
y (a -> a) -> a -> a
forall a b. (a -> b) -> a -> b
$ Size -> a -> a -> a
forall a. Floating a => Size -> a -> a -> a
scaleExponentialFloat Size
sz a
z a
x
sized_y :: a
sized_y =
a -> a -> a -> a
forall a. Ord a => a -> a -> a -> a
clamp a
x a
y (a -> a) -> a -> a
forall a b. (a -> b) -> a -> b
$ Size -> a -> a -> a
forall a. Floating a => Size -> a -> a -> a
scaleExponentialFloat Size
sz a
z a
y
in
(a
sized_x, a
sized_y)
scaleExponential :: Integral a => Size -> a -> a -> a
scaleExponential :: forall a. Integral a => Size -> a -> a -> a
scaleExponential Size
sz a
z0 a
n0 =
let
z :: Double
z =
a -> Double
forall a b. (Integral a, Num b) => a -> b
fromIntegral a
z0
n :: Double
n =
a -> Double
forall a b. (Integral a, Num b) => a -> b
fromIntegral a
n0
in
Double -> a
forall b. Integral b => Double -> b
forall a b. (RealFrac a, Integral b) => a -> b
round (Size -> Double -> Double -> Double
forall a. Floating a => Size -> a -> a -> a
scaleExponentialFloat Size
sz Double
z Double
n :: Double)
scaleExponentialFloat :: Floating a => Size -> a -> a -> a
scaleExponentialFloat :: forall a. Floating a => Size -> a -> a -> a
scaleExponentialFloat Size
sz0 a
z a
n =
let
sz :: Size
sz =
Size -> Size -> Size -> Size
forall a. Ord a => a -> a -> a -> a
clamp Size
0 Size
99 Size
sz0
diff :: a
diff =
(((a -> a
forall a. Num a => a -> a
abs (a
n a -> a -> a
forall a. Num a => a -> a -> a
- a
z) a -> a -> a
forall a. Num a => a -> a -> a
+ a
1) a -> a -> a
forall a. Floating a => a -> a -> a
** (Size -> a
forall a b. (Real a, Fractional b) => a -> b
realToFrac Size
sz a -> a -> a
forall a. Fractional a => a -> a -> a
/ a
99)) a -> a -> a
forall a. Num a => a -> a -> a
- a
1) a -> a -> a
forall a. Num a => a -> a -> a
* a -> a
forall a. Num a => a -> a
signum (a
n a -> a -> a
forall a. Num a => a -> a -> a
- a
z)
in
a
z a -> a -> a
forall a. Num a => a -> a -> a
+ a
diff