duckling/Duckling/Numeral/Types.hs
Julien Odent 54c9448fba Rename Number to Numeral
Summary: For consistency with the dimension name.

Reviewed By: JonCoens

Differential Revision: D4722216

fbshipit-source-id: 82c56d3
2017-03-16 13:49:16 -07:00

63 lines
1.6 KiB
Haskell

-- Copyright (c) 2016-present, Facebook, Inc.
-- All rights reserved.
--
-- This source code is licensed under the BSD-style license found in the
-- LICENSE file in the root directory of this source tree. An additional grant
-- of patent rights can be found in the PATENTS file in the same directory.
{-# LANGUAGE DeriveAnyClass #-}
{-# LANGUAGE DeriveGeneric #-}
{-# LANGUAGE NamedFieldPuns #-}
{-# LANGUAGE NoRebindableSyntax #-}
{-# LANGUAGE OverloadedStrings #-}
{-# LANGUAGE TypeFamilies #-}
module Duckling.Numeral.Types where
import Control.DeepSeq
import Data.Aeson
import Data.Hashable
import Data.Maybe
import Data.Text (Text)
import GHC.Generics
import Prelude
import Duckling.Resolve
data NumeralData = NumeralData
{ value :: Double
, grain :: Maybe Int
, multipliable :: Bool
}
deriving (Eq, Generic, Hashable, Ord, Show, NFData)
instance Resolve NumeralData where
type ResolvedValue NumeralData = NumeralValue
resolve _ NumeralData {value} = Just NumeralValue {vValue = value}
newtype NumeralValue = NumeralValue { vValue :: Double }
deriving (Eq, Show)
instance ToJSON NumeralValue where
toJSON (NumeralValue value) = object
[ "type" .= ("value" :: Text)
, "value" .= value
]
getIntValue :: Double -> Maybe Int
getIntValue x = if rest == 0 then Just int else Nothing
where
(int, rest) = properFraction x :: (Int, Double)
isInteger :: Double -> Bool
isInteger = isJust . getIntValue
isNatural :: Double -> Bool
isNatural x = isInteger x && x > 0
isIntegerBetween :: Double -> Int -> Int -> Bool
isIntegerBetween x low high = case getIntValue x of
Just int -> low <= int && int <= high
Nothing -> False