mirror of
https://github.com/facebook/duckling.git
synced 2024-10-27 14:26:54 +03:00
bfb6ba0387
Summary: Today things like `at single`, `at a few`, `at a couple of` would return a `Time`. Discussed with blandinw to do this very explicit hack right now until other use cases show up. Reviewed By: niteria Differential Revision: D5325369 fbshipit-source-id: aec0402
68 lines
1.9 KiB
Haskell
68 lines
1.9 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
|
|
-- Hack until other use cases pop up,
|
|
-- at which point we'll craft a generic solution.
|
|
-- This allows to explicitly flag Numerals that don't work well with Time.
|
|
-- See `ruleTODLatent`. Prevents things like "at single", "dozens o'clock".
|
|
, okForAnyTime :: 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
|