blob: 9eaec38ba6e0a0fc6f64a146ed6eee63419fc84d (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
|
module Operators (
primitives
) where
import Control.Monad.Except
import Error (LispError (..), LispResult (..))
import Parser
primitives :: [(String, [Expr] -> LispResult Expr)]
primitives =
[
("+", arithmetic (+))
, ("-", arithmetic (-))
, ("*", arithmetic (*))
, ("/", arithmetic (/))
]
data LispNumber = I Integer
| F Double
instance Num LispNumber where
(I a) + (I b) = I $ a + b
(F a) + (F b) = F $ a + b
(I a) - (I b) = I $ a - b
(F a) - (F b) = F $ a - b
(I a) * (I b) = I $ a * b
(F a) * (F b) = F $ a * b
instance Fractional LispNumber where
(I a) / (I b) = I $ a `div` b
(F a) / (F b) = F $ a / b
arithmetic :: (LispNumber -> LispNumber -> LispNumber) -> [Expr] -> LispResult Expr
arithmetic op args
| length args < 2 = throwError $ ArgCount 2 args
| otherwise = do
as <- mapM unwrapNum args
return . wrapNum $ foldl1 op as
unwrapNum :: Expr -> LispResult LispNumber
unwrapNum (IntLiteral n) = return $ I n
unwrapNum (FloatLiteral n) = return $ F n
unwrapNum x = throwError $ TypeMismatch "number" x
wrapNum :: LispNumber -> Expr
wrapNum (I n) = IntLiteral n
wrapNum (F n) = FloatLiteral n
|