253 lines
9.1 KiB
Haskell
253 lines
9.1 KiB
Haskell
{-
|
|
Tock: a compiler for parallel languages
|
|
Copyright (C) 2007 University of Kent
|
|
|
|
This program is free software; you can redistribute it and/or modify it
|
|
under the terms of the GNU General Public License as published by the
|
|
Free Software Foundation, either version 2 of the License, or (at your
|
|
option) any later version.
|
|
|
|
This program is distributed in the hope that it will be useful, but
|
|
WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
|
|
General Public License for more details.
|
|
|
|
You should have received a copy of the GNU General Public License along
|
|
with this program. If not, see <http://www.gnu.org/licenses/>.
|
|
-}
|
|
|
|
-- | Definitions of intrinsic FUNCTIONs and PROCs.
|
|
module Intrinsics where
|
|
|
|
import Data.Char
|
|
import Data.List
|
|
|
|
import qualified AST as A
|
|
|
|
intrinsicFunctions :: [(String, ([A.Type], [(A.Type, String)]))]
|
|
intrinsicFunctions =
|
|
[ -- Multiple length arithmetic functions
|
|
-- Appendix L of the occam 2 manual (and section J.1)
|
|
("ASHIFTLEFT", ([A.Int], [(A.Int, "argument"), (A.Int, "places")]))
|
|
, ("ASHIFTRIGHT", ([A.Int], [(A.Int, "argument"), (A.Int, "places")]))
|
|
, ("LONGADD", ([A.Int], [(A.Int, "left"), (A.Int, "right"), (A.Int, "carry.in")]))
|
|
, ("LONGDIFF", ([A.Int,A.Int], [(A.Int, "left"), (A.Int, "right"), (A.Int, "borrow.in")]))
|
|
, ("LONGDIV", ([A.Int,A.Int], [(A.Int, "dividend.hi"), (A.Int, "dividend.lo"), (A.Int, "divisor")]))
|
|
, ("LONGPROD", ([A.Int,A.Int], [(A.Int, "left"), (A.Int, "right"), (A.Int, "carry.in")]))
|
|
, ("LONGSUB", ([A.Int], [(A.Int, "left"), (A.Int, "right"), (A.Int, "borrow.in")]))
|
|
, ("LONGSUM", ([A.Int,A.Int], [(A.Int, "left"), (A.Int, "right"), (A.Int, "carry.in")]))
|
|
, ("NORMALISE", ([A.Int, A.Int, A.Int], [(A.Int, "hi.in"), (A.Int, "lo.in")]))
|
|
, ("ROTATELEFT", ([A.Int], [(A.Int, "argument"), (A.Int, "places")]))
|
|
, ("ROTATERIGHT", ([A.Int], [(A.Int, "argument"), (A.Int, "places")]))
|
|
, ("SHIFTLEFT", ([A.Int, A.Int], [(A.Int, "hi.in"), (A.Int, "lo.in"), (A.Int, "places")]))
|
|
, ("SHIFTRIGHT", ([A.Int, A.Int], [(A.Int, "hi.in"), (A.Int, "lo.in"), (A.Int, "places")]))
|
|
|
|
-- IEEE floating point arithmetic
|
|
-- Appendix M of the occam 2 manual (and section J.3)
|
|
] ++ concatMap doubleD [
|
|
("IEEECOMPARE", ([A.Int], [(A.Real32, "X"), (A.Real32, "Y")]))
|
|
] ++ concatMap doubleNum [
|
|
("IEEE32OP", ([A.Bool, A.Real32], [(A.Real32, "X"), (A.Int, "Rm"), (A.Int, "Op"), (A.Real32, "Y")]))
|
|
, ("IEEE32REM", ([A.Bool, A.Real32], [(A.Real32, "X"), (A.Real32, "Y")]))
|
|
, ("REAL32EQ", ([A.Bool], [(A.Real32, "X"), (A.Real32, "Y")]))
|
|
, ("REAL32GT", ([A.Bool], [(A.Real32, "X"), (A.Real32, "Y")]))
|
|
, ("REAL32OP", ([A.Real32], [(A.Real32, "X"), (A.Int, "Op"), (A.Real32, "Y")]))
|
|
, ("REAL32REM", ([A.Real32], [(A.Real32, "X"), (A.Real32, "Y")]))
|
|
|
|
|
|
-- Floating point functions
|
|
-- Appendix K of the occam 2 manual (and section J.2)
|
|
] ++ [
|
|
("ARGUMENT.REDUCE", ([A.Bool, A.Int32, A.Real32], [(A.Real32, "X"), (A.Real32, "Y"), (A.Real32, "Y.err")]))
|
|
, ("DARGUMENT.REDUCE", ([A.Bool, A.Int32, A.Real64], [(A.Real64, "X"), (A.Real64, "Y"), (A.Real64, "Y.err")]))
|
|
] ++ concatMap doubleD [
|
|
simple "ABS"
|
|
, ("COPYSIGN", ([A.Real32], [(A.Real32, "X"), (A.Real32, "Y")]))
|
|
, simple "DIVBY2"
|
|
, ("FLOATING.UNPACK", ([A.Int, A.Real32], [(A.Real32, "X")]))
|
|
, simple "FPINT"
|
|
, query "ISNAN"
|
|
, simple "LOGB"
|
|
, simple "MINUSX"
|
|
, simple "MULBY2"
|
|
, ("NEXTAFTER", ([A.Real32], [(A.Real32, "X"), (A.Real32, "Y")]))
|
|
, query "NOTFINITE"
|
|
, ("ORDERED", ([A.Bool], [(A.Real32, "X"), (A.Real32, "Y")]))
|
|
, ("SCALEB", ([A.Real32], [(A.Real32, "X"), (A.Int, "n")]))
|
|
, simple "SQRT"
|
|
]
|
|
|
|
-- Elementary floating point functions
|
|
-- Appendix N of the occam 2 manual (and section J.4)
|
|
++ [(n, ts) | (n, (ts, _)) <- simpleFloatIntrinsics]
|
|
++ concatMap doubleD [("RAN", ([A.Real32, A.Int32], [(A.Int32, "N")]))]
|
|
where
|
|
query n = (n, ([A.Bool], [(A.Real32, "X")]))
|
|
simple n = (n, ([A.Real32], [(A.Real32, "X")]))
|
|
|
|
doubleNum orig@(n, (rs, ps)) = [orig, (map rep n, (map dt rs, zip (map (dt . fst) ps) (map snd ps)))]
|
|
where
|
|
rep '3' = '6'
|
|
rep '2' = '4'
|
|
rep c = c
|
|
|
|
doubleD orig@(n, (rs, ps)) = [orig, ("D"++n, (map dt rs, zip (map (dt . fst) ps) (map snd ps)))]
|
|
|
|
dt :: A.Type -> A.Type
|
|
dt A.Real32 = A.Real64
|
|
dt A.Int32 = A.Int64
|
|
dt t = t
|
|
|
|
simpleFloatIntrinsics :: [(String, (([A.Type], [(A.Type, String)]), String))]
|
|
simpleFloatIntrinsics = concatMap double $
|
|
-- Same order as occam manual:
|
|
[("ALOG", ([A.Real32], [(A.Real32, "X")]), "log")
|
|
,("ALOG10", ([A.Real32], [(A.Real32, "X")]), "log10")
|
|
] ++ map s [
|
|
"EXP",
|
|
"TAN",
|
|
"SIN",
|
|
"ASIN",
|
|
"COS",
|
|
"ACOS",
|
|
"SINH",
|
|
"COSH",
|
|
"TANH",
|
|
"ATAN",
|
|
"ATAN2"
|
|
]
|
|
++ [("POWER", ([A.Real32], [(A.Real32, "X"), (A.Real32, "Y")]), "pow")]
|
|
where
|
|
s n = (n, ([A.Real32], [(A.Real32, "X")]), map toLower n)
|
|
|
|
double (occn, ts@(rts, pts), cn) = [(occn, (ts, cn++"f")),
|
|
("D"++occn, ((map dt rts, zip (map (dt . fst) pts) (map snd pts)), cn))]
|
|
dt A.Real32 = A.Real64
|
|
dt A.Int32 = A.Int64
|
|
dt t = t
|
|
|
|
intrinsicProcs :: [(String, [(A.AbbrevMode, A.Type, String)])]
|
|
intrinsicProcs =
|
|
[ ("ASSERT", [(A.ValAbbrev, A.Bool, "value")])
|
|
, ("CAUSEERROR", [])
|
|
, ("RESCHEDULE", [])
|
|
] ++ concat [
|
|
(zip ["INT" ++ suffix ++ "TOSTRING", "HEX" ++ suffix ++ "TOSTRING"] $ repeat
|
|
[ (A.Abbrev, A.Int, "len")
|
|
, (A.Abbrev, A.Array [A.UnknownDimension] A.Byte, "string")
|
|
, (A.ValAbbrev, t, "n")
|
|
])
|
|
++ (zip ["STRINGTOINT" ++ suffix, "STRINGTOHEX" ++ suffix] $ repeat
|
|
[ (A.Abbrev, A.Bool, "error")
|
|
, (A.Abbrev, t, "n")
|
|
, (A.ValAbbrev, A.Array [A.UnknownDimension] A.Byte, "string")
|
|
])
|
|
| (t, suffix) <- [(A.Int, ""),(A.Int16, "16"),(A.Int32, "32"),(A.Int64, "64")]
|
|
] ++ [
|
|
("BOOLTOSTRING",
|
|
[ (A.Abbrev, A.Int, "len")
|
|
, (A.Abbrev, A.Array [A.UnknownDimension] A.Byte, "string")
|
|
, (A.ValAbbrev, A.Bool, "b")
|
|
])
|
|
, ("STRINGTOBOOL",
|
|
[ (A.Abbrev, A.Bool, "error")
|
|
, (A.Abbrev, A.Bool, "b")
|
|
, (A.ValAbbrev, A.Array [A.UnknownDimension] A.Byte, "string")
|
|
])
|
|
]
|
|
++ [("RESIZE.MOBILE.ARRAY.1D", [(A.Abbrev, A.Mobile A.Infer, "mobile")
|
|
,(A.ValAbbrev, A.Int, "count")
|
|
])]
|
|
|
|
rainIntrinsicFunctions :: [(String, ([A.Type], [(A.Type, String)]))]
|
|
rainIntrinsicFunctions =
|
|
-- Time functions:
|
|
[ ("toSeconds", ([A.Real64], [(A.Time, "time")]))
|
|
, ("toMillis", ([A.Int64], [(A.Time, "time")]))
|
|
, ("toMicros", ([A.Int64], [(A.Time, "time")]))
|
|
, ("toNanos", ([A.Int64], [(A.Time, "time")]))
|
|
, ("fromSeconds", ([A.Time], [(A.Real64, "value")]))
|
|
, ("fromMillis", ([A.Time], [(A.Int64, "value")]))
|
|
, ("fromMicros", ([A.Time], [(A.Int64, "value")]))
|
|
, ("fromNanos", ([A.Time], [(A.Int64, "value")]))
|
|
]
|
|
|
|
-- I think several of these operators are defined where they shouldn't be..
|
|
occamIntrinsicOperators :: [(String, A.Type, [A.Type])]
|
|
occamIntrinsicOperators = concat
|
|
[comparison ">"
|
|
,comparison ">="
|
|
,comparison "<"
|
|
,comparison "<="
|
|
,comparison "AFTER"
|
|
,arithmetic "+"
|
|
,arithmetic "-"
|
|
,arithmetic "*"
|
|
,arithmetic "/"
|
|
,arithmetic "\\"
|
|
,arithmetic "REM"
|
|
,arithmetic "PLUS"
|
|
,arithmetic "TIMES"
|
|
,arithmetic "MINUS"
|
|
,equalityOp "="
|
|
,equalityOp "<>"
|
|
,unaryArith "-" -- \\ [("-",A.Byte,[A.Byte])]
|
|
,bitwiseOpr "/\\"
|
|
,bitwiseOpr "\\/"
|
|
,bitwiseOpr "><"
|
|
,shiftingOp ">>"
|
|
,shiftingOp "<<"
|
|
,booleanOpr "AND"
|
|
,booleanOpr "OR"
|
|
,[("NOT", A.Bool, [A.Bool])]
|
|
,[("~", t, [t]) | t <- allIntegerTypes]
|
|
,[("MINUS", t, [t]) | t <- allNumericTypes]
|
|
]
|
|
where
|
|
booleanOpr :: String -> [(String, A.Type, [A.Type])]
|
|
booleanOpr op = [(op, A.Bool, [A.Bool, A.Bool])]
|
|
|
|
comparison :: String -> [(String, A.Type, [A.Type])]
|
|
comparison op = [(op, A.Bool, [t, t])
|
|
| t <- allNumericTypes]
|
|
|
|
equalityOp :: String -> [(String, A.Type, [A.Type])]
|
|
equalityOp op = [(op, A.Bool, [t, t])
|
|
| t <- allTypes]
|
|
|
|
arithmetic :: String -> [(String, A.Type, [A.Type])]
|
|
arithmetic op = [(op, t, [t, t])
|
|
| t <- allNumericTypes]
|
|
|
|
bitwiseOpr :: String -> [(String, A.Type, [A.Type])]
|
|
bitwiseOpr op = [(op, t, [t, t])
|
|
| t <- allIntegerTypes]
|
|
|
|
shiftingOp :: String -> [(String, A.Type, [A.Type])]
|
|
shiftingOp op = [(op, t, [t, A.Int])
|
|
| t <- allIntegerTypes]
|
|
|
|
unaryArith :: String -> [(String, A.Type, [A.Type])]
|
|
unaryArith op = [(op, t, [t])
|
|
| t <- allNumericTypes]
|
|
|
|
allNumericTypes :: [A.Type]
|
|
allNumericTypes = allIntegerTypes ++ [A.Real32, A.Real64]
|
|
|
|
allIntegerTypes :: [A.Type]
|
|
allIntegerTypes
|
|
= [A.Byte
|
|
-- ,A.UInt16
|
|
-- ,A.UInt32
|
|
-- ,A.UInt64
|
|
-- ,A.Int8
|
|
,A.Int16
|
|
,A.Int32
|
|
,A.Int64
|
|
,A.Int
|
|
]
|
|
|
|
allTypes :: [A.Type]
|
|
allTypes = A.Bool : allNumericTypes
|
|
|