Skip to content
This repository has been archived by the owner on Nov 25, 2021. It is now read-only.

Commit

Permalink
Add validating serialization instances for uniform distribution
Browse files Browse the repository at this point in the history
  • Loading branch information
Shimuuar committed Jul 11, 2016
1 parent c28df73 commit abac74c
Showing 1 changed file with 40 additions and 15 deletions.
55 changes: 40 additions & 15 deletions Statistics/Distribution/Uniform.hs
Original file line number Diff line number Diff line change
@@ -1,3 +1,4 @@
{-# LANGUAGE OverloadedStrings #-}
{-# LANGUAGE DeriveDataTypeable, DeriveGeneric #-}
-- |
-- Module : Statistics.Distribution.Uniform
Expand All @@ -14,42 +15,66 @@ module Statistics.Distribution.Uniform
UniformDistribution
-- * Constructors
, uniformDistr
, uniformDistrE
-- ** Accessors
, uniformA
, uniformB
) where

import Data.Aeson (FromJSON, ToJSON)
import Data.Binary (Binary)
import Data.Data (Data, Typeable)
import GHC.Generics (Generic)
import qualified Statistics.Distribution as D
import Control.Applicative
import Data.Aeson (FromJSON(..), ToJSON, Value(..), (.:))
import Data.Binary (Binary(..))
import Data.Data (Data, Typeable)
import GHC.Generics (Generic)
import qualified System.Random.MWC as MWC
import Data.Binary (put, get)
import Control.Applicative ((<$>), (<*>))

import qualified Statistics.Distribution as D
import Statistics.Internal



-- | Uniform distribution from A to B
data UniformDistribution = UniformDistribution {
uniformA :: {-# UNPACK #-} !Double -- ^ Low boundary of distribution
, uniformB :: {-# UNPACK #-} !Double -- ^ Upper boundary of distribution
} deriving (Eq, Read, Show, Typeable, Data, Generic)
} deriving (Eq, Typeable, Data, Generic)

instance Show UniformDistribution where
showsPrec i (UniformDistribution a b) = defaultShow2 "uniformDistr" a b i
instance Read UniformDistribution where
readPrec = defaultReadPrecM2 "uniformDistr" uniformDistrE

instance FromJSON UniformDistribution
instance ToJSON UniformDistribution
instance FromJSON UniformDistribution where
parseJSON (Object v) = do
a <- v .: "uniformA"
b <- v .: "uniformB"
maybe (fail errMsg) return $ uniformDistrE a b
parseJSON _ = empty

instance Binary UniformDistribution where
put (UniformDistribution x y) = put x >> put y
get = UniformDistribution <$> get <*> get
put (UniformDistribution x y) = put x >> put y
get = do
a <- get
b <- get
maybe (fail errMsg) return $ uniformDistrE a b

-- | Create uniform distribution.
uniformDistr :: Double -> Double -> UniformDistribution
uniformDistr a b
| b < a = uniformDistr b a
| a < b = UniformDistribution a b
| otherwise = error "Statistics.Distribution.Uniform.uniform: wrong parameters"
uniformDistr a b = maybe (error errMsg) id $ uniformDistrE a b

-- | Create uniform distribution.
uniformDistrE :: Double -> Double -> Maybe UniformDistribution
uniformDistrE a b
| b < a = Just $ UniformDistribution b a
| a < b = Just $ UniformDistribution a b
| otherwise = Nothing
-- NOTE: failure is in default branch to guard againist NaNs.

errMsg :: String
errMsg = "Statistics.Distribution.Uniform.uniform: wrong parameters"


instance D.Distribution UniformDistribution where
cumulative (UniformDistribution a b) x
| x < a = 0
Expand Down

0 comments on commit abac74c

Please sign in to comment.