(i1 + i2) only if it results in a valid Float.
\begin{code}
+
+{-# OPTIONS -optc-DNON_POSIX_SOURCE #-}
+
module PrelRules ( primOpRule, builtinRules ) where
#include "HsVersions.h"
import Id ( mkWildId )
import Literal ( Literal(..), isLitLitLit, mkMachInt, mkMachWord
, literalType
- , word2IntLit, int2WordLit, char2IntLit, int2CharLit
+ , word2IntLit, int2WordLit
+ , narrow8IntLit, narrow16IntLit, narrow32IntLit
+ , narrow8WordLit, narrow16WordLit, narrow32WordLit
+ , char2IntLit, int2CharLit
, float2IntLit, int2FloatLit, double2IntLit, int2DoubleLit
- , addr2IntLit, int2AddrLit, float2DoubleLit, double2FloatLit
+ , nullAddrLit, float2DoubleLit, double2FloatLit
)
import PrimOp ( PrimOp(..), primOpOcc )
import TysWiredIn ( trueDataConId, falseDataConId )
import TyCon ( tyConDataConsIfAvailable, isEnumerationTyCon, isNewTyCon )
import DataCon ( dataConTag, dataConTyCon, dataConId, fIRST_TAG )
import CoreUtils ( exprIsValue, cheapEqExpr, exprIsConApp_maybe )
-import Type ( splitTyConApp_maybe )
+import Type ( tyConAppTyCon, eqType )
import OccName ( occNameUserString)
import PrelNames ( unpackCStringFoldrName, unpackCStringFoldrIdKey, hasKey )
import Name ( Name )
import Bits ( Bits(..) )
+#if __GLASGOW_HASKELL__ >= 500
+import Word ( Word )
+#else
import Word ( Word64 )
+#endif
import Outputable
import CmdLineOpts ( opt_SimplExcessPrecision )
\end{code}
\begin{code}
-primOpRule :: PrimOp -> CoreRule
-primOpRule op
- = BuiltinRule (primop_rule op)
+primOpRule :: PrimOp -> Maybe CoreRule
+primOpRule op = fmap BuiltinRule (primop_rule op)
where
op_name = _PK_ (occNameUserString (primOpOcc op))
op_name_case = op_name _APPEND_ SLIT("->case")
-- ToDo: something for integer-shift ops?
-- NotOp
- primop_rule SeqOp = seqRule
- primop_rule TagToEnumOp = tagToEnumRule
- primop_rule DataToTagOp = dataToTagRule
+ primop_rule AddrNullOp = Just nullAddrRule
+ primop_rule SeqOp = Just seqRule
+ primop_rule TagToEnumOp = Just tagToEnumRule
+ primop_rule DataToTagOp = Just dataToTagRule
-- Int operations
- primop_rule IntAddOp = twoLits (intOp2 (+) op_name)
- primop_rule IntSubOp = twoLits (intOp2 (-) op_name)
- primop_rule IntMulOp = twoLits (intOp2 (*) op_name)
- primop_rule IntQuotOp = twoLits (intOp2Z quot op_name)
- primop_rule IntRemOp = twoLits (intOp2Z rem op_name)
- primop_rule IntNegOp = oneLit (negOp op_name)
+ primop_rule IntAddOp = Just (twoLits (intOp2 (+) op_name))
+ primop_rule IntSubOp = Just (twoLits (intOp2 (-) op_name))
+ primop_rule IntMulOp = Just (twoLits (intOp2 (*) op_name))
+ primop_rule IntQuotOp = Just (twoLits (intOp2Z quot op_name))
+ primop_rule IntRemOp = Just (twoLits (intOp2Z rem op_name))
+ primop_rule IntNegOp = Just (oneLit (negOp op_name))
-- Word operations
- primop_rule WordQuotOp = twoLits (wordOp2Z quot op_name)
- primop_rule WordRemOp = twoLits (wordOp2Z rem op_name)
+#if __GLASGOW_HASKELL__ >= 500
+ primop_rule WordAddOp = Just (twoLits (wordOp2 (+) op_name))
+ primop_rule WordSubOp = Just (twoLits (wordOp2 (-) op_name))
+ primop_rule WordMulOp = Just (twoLits (wordOp2 (*) op_name))
+#endif
+ primop_rule WordQuotOp = Just (twoLits (wordOp2Z quot op_name))
+ primop_rule WordRemOp = Just (twoLits (wordOp2Z rem op_name))
#if __GLASGOW_HASKELL__ >= 407
- primop_rule AndOp = twoLits (wordBitOp2 (.&.) op_name)
- primop_rule OrOp = twoLits (wordBitOp2 (.|.) op_name)
- primop_rule XorOp = twoLits (wordBitOp2 xor op_name)
+ primop_rule AndOp = Just (twoLits (wordBitOp2 (.&.) op_name))
+ primop_rule OrOp = Just (twoLits (wordBitOp2 (.|.) op_name))
+ primop_rule XorOp = Just (twoLits (wordBitOp2 xor op_name))
#endif
-- coercions
- primop_rule Word2IntOp = oneLit (litCoerce word2IntLit op_name)
- primop_rule Int2WordOp = oneLit (litCoerce int2WordLit op_name)
- primop_rule OrdOp = oneLit (litCoerce char2IntLit op_name)
- primop_rule ChrOp = oneLit (litCoerce int2CharLit op_name)
- primop_rule Float2IntOp = oneLit (litCoerce float2IntLit op_name)
- primop_rule Int2FloatOp = oneLit (litCoerce int2FloatLit op_name)
- primop_rule Double2IntOp = oneLit (litCoerce double2IntLit op_name)
- primop_rule Int2DoubleOp = oneLit (litCoerce int2DoubleLit op_name)
- primop_rule Addr2IntOp = oneLit (litCoerce addr2IntLit op_name)
- primop_rule Int2AddrOp = oneLit (litCoerce int2AddrLit op_name)
+ primop_rule Word2IntOp = Just (oneLit (litCoerce word2IntLit op_name))
+ primop_rule Int2WordOp = Just (oneLit (litCoerce int2WordLit op_name))
+ primop_rule Narrow8IntOp = Just (oneLit (litCoerce narrow8IntLit op_name))
+ primop_rule Narrow16IntOp = Just (oneLit (litCoerce narrow16IntLit op_name))
+ primop_rule Narrow32IntOp = Just (oneLit (litCoerce narrow32IntLit op_name))
+ primop_rule Narrow8WordOp = Just (oneLit (litCoerce narrow8WordLit op_name))
+ primop_rule Narrow16WordOp = Just (oneLit (litCoerce narrow16WordLit op_name))
+ primop_rule Narrow32WordOp = Just (oneLit (litCoerce narrow32WordLit op_name))
+ primop_rule OrdOp = Just (oneLit (litCoerce char2IntLit op_name))
+ primop_rule ChrOp = Just (oneLit (litCoerce int2CharLit op_name))
+ primop_rule Float2IntOp = Just (oneLit (litCoerce float2IntLit op_name))
+ primop_rule Int2FloatOp = Just (oneLit (litCoerce int2FloatLit op_name))
+ primop_rule Double2IntOp = Just (oneLit (litCoerce double2IntLit op_name))
+ primop_rule Int2DoubleOp = Just (oneLit (litCoerce int2DoubleLit op_name))
-- SUP: Not sure what the standard says about precision in the following 2 cases
- primop_rule Float2DoubleOp = oneLit (litCoerce float2DoubleLit op_name)
- primop_rule Double2FloatOp = oneLit (litCoerce double2FloatLit op_name)
+ primop_rule Float2DoubleOp = Just (oneLit (litCoerce float2DoubleLit op_name))
+ primop_rule Double2FloatOp = Just (oneLit (litCoerce double2FloatLit op_name))
-- Float
- primop_rule FloatAddOp = twoLits (floatOp2 (+) op_name)
- primop_rule FloatSubOp = twoLits (floatOp2 (-) op_name)
- primop_rule FloatMulOp = twoLits (floatOp2 (*) op_name)
- primop_rule FloatDivOp = twoLits (floatOp2Z (/) op_name)
- primop_rule FloatNegOp = oneLit (negOp op_name)
+ primop_rule FloatAddOp = Just (twoLits (floatOp2 (+) op_name))
+ primop_rule FloatSubOp = Just (twoLits (floatOp2 (-) op_name))
+ primop_rule FloatMulOp = Just (twoLits (floatOp2 (*) op_name))
+ primop_rule FloatDivOp = Just (twoLits (floatOp2Z (/) op_name))
+ primop_rule FloatNegOp = Just (oneLit (negOp op_name))
-- Double
- primop_rule DoubleAddOp = twoLits (doubleOp2 (+) op_name)
- primop_rule DoubleSubOp = twoLits (doubleOp2 (-) op_name)
- primop_rule DoubleMulOp = twoLits (doubleOp2 (*) op_name)
- primop_rule DoubleDivOp = twoLits (doubleOp2Z (/) op_name)
- primop_rule DoubleNegOp = oneLit (negOp op_name)
+ primop_rule DoubleAddOp = Just (twoLits (doubleOp2 (+) op_name))
+ primop_rule DoubleSubOp = Just (twoLits (doubleOp2 (-) op_name))
+ primop_rule DoubleMulOp = Just (twoLits (doubleOp2 (*) op_name))
+ primop_rule DoubleDivOp = Just (twoLits (doubleOp2Z (/) op_name))
+ primop_rule DoubleNegOp = Just (oneLit (negOp op_name))
-- Relational operators
- primop_rule IntEqOp = relop (==) `or_rule` litEq True op_name_case
- primop_rule IntNeOp = relop (/=) `or_rule` litEq False op_name_case
- primop_rule CharEqOp = relop (==) `or_rule` litEq True op_name_case
- primop_rule CharNeOp = relop (/=) `or_rule` litEq False op_name_case
-
- primop_rule IntGtOp = relop (>)
- primop_rule IntGeOp = relop (>=)
- primop_rule IntLeOp = relop (<=)
- primop_rule IntLtOp = relop (<)
-
- primop_rule CharGtOp = relop (>)
- primop_rule CharGeOp = relop (>=)
- primop_rule CharLeOp = relop (<=)
- primop_rule CharLtOp = relop (<)
-
- primop_rule FloatGtOp = relop (>)
- primop_rule FloatGeOp = relop (>=)
- primop_rule FloatLeOp = relop (<=)
- primop_rule FloatLtOp = relop (<)
- primop_rule FloatEqOp = relop (==)
- primop_rule FloatNeOp = relop (/=)
-
- primop_rule DoubleGtOp = relop (>)
- primop_rule DoubleGeOp = relop (>=)
- primop_rule DoubleLeOp = relop (<=)
- primop_rule DoubleLtOp = relop (<)
- primop_rule DoubleEqOp = relop (==)
- primop_rule DoubleNeOp = relop (/=)
-
- primop_rule WordGtOp = relop (>)
- primop_rule WordGeOp = relop (>=)
- primop_rule WordLeOp = relop (<=)
- primop_rule WordLtOp = relop (<)
- primop_rule WordEqOp = relop (==)
- primop_rule WordNeOp = relop (/=)
-
- primop_rule other = \args -> Nothing
+ primop_rule IntEqOp = Just (relop (==) `or_rule` litEq True op_name_case)
+ primop_rule IntNeOp = Just (relop (/=) `or_rule` litEq False op_name_case)
+ primop_rule CharEqOp = Just (relop (==) `or_rule` litEq True op_name_case)
+ primop_rule CharNeOp = Just (relop (/=) `or_rule` litEq False op_name_case)
+
+ primop_rule IntGtOp = Just (relop (>))
+ primop_rule IntGeOp = Just (relop (>=))
+ primop_rule IntLeOp = Just (relop (<=))
+ primop_rule IntLtOp = Just (relop (<))
+
+ primop_rule CharGtOp = Just (relop (>))
+ primop_rule CharGeOp = Just (relop (>=))
+ primop_rule CharLeOp = Just (relop (<=))
+ primop_rule CharLtOp = Just (relop (<))
+
+ primop_rule FloatGtOp = Just (relop (>))
+ primop_rule FloatGeOp = Just (relop (>=))
+ primop_rule FloatLeOp = Just (relop (<=))
+ primop_rule FloatLtOp = Just (relop (<))
+ primop_rule FloatEqOp = Just (relop (==))
+ primop_rule FloatNeOp = Just (relop (/=))
+
+ primop_rule DoubleGtOp = Just (relop (>))
+ primop_rule DoubleGeOp = Just (relop (>=))
+ primop_rule DoubleLeOp = Just (relop (<=))
+ primop_rule DoubleLtOp = Just (relop (<))
+ primop_rule DoubleEqOp = Just (relop (==))
+ primop_rule DoubleNeOp = Just (relop (/=))
+
+ primop_rule WordGtOp = Just (relop (>))
+ primop_rule WordGeOp = Just (relop (>=))
+ primop_rule WordLeOp = Just (relop (<=))
+ primop_rule WordLtOp = Just (relop (<))
+ primop_rule WordEqOp = Just (relop (==))
+ primop_rule WordNeOp = Just (relop (/=))
+
+ primop_rule other = Nothing
relop cmp = twoLits (cmpOp (\ord -> ord `cmp` EQ) op_name)
negOp name (MachFloat f) = Just (name, mkFloatVal (-f))
negOp name (MachDouble d) = Just (name, mkDoubleVal (-d))
-negOp name l@(MachInt i) = intResult name (-i)
+negOp name (MachInt i) = intResult name (-i)
negOp name l = Nothing
--------------------------
-intOp2 op name l1@(MachInt i1) l2@(MachInt i2)
+intOp2 op name (MachInt i1) (MachInt i2)
= intResult name (i1 `op` i2)
intOp2 op name l1 l2 = Nothing -- Could find LitLit
intOp2Z op name l1 l2 = Nothing -- LitLit or zero dividend
--------------------------
--- Integer is not an instance of Bits, so we operate on Word64
-wordBitOp2 op name l1@(MachWord w1) l2@(MachWord w2)
- = Just (name, mkWordVal ((fromIntegral::Word64->Integer) (fromIntegral w1 `op` fromIntegral w2)))
-wordBitOp2 op name l1 l2 = Nothing -- Could find LitLit
+#if __GLASGOW_HASKELL__ >= 500
+wordOp2 op name (MachWord w1) (MachWord w2)
+ = wordResult name (w1 `op` w2)
+wordOp2 op name l1 l2 = Nothing -- Could find LitLit
+#endif
wordOp2Z op name (MachWord w1) (MachWord w2)
| w2 /= 0 = Just (name, mkWordVal (w1 `op` w2))
wordOp2Z op name l1 l2 = Nothing -- LitLit or zero dividend
+#if __GLASGOW_HASKELL__ >= 500
+wordBitOp2 op name l1@(MachWord w1) l2@(MachWord w2)
+ = Just (name, mkWordVal (w1 `op` w2))
+#else
+-- Integer is not an instance of Bits, so we operate on Word64
+wordBitOp2 op name l1@(MachWord w1) l2@(MachWord w2)
+ = Just (name, mkWordVal ((fromIntegral::Word64->Integer) (fromIntegral w1 `op` fromIntegral w2)))
+#endif
+wordBitOp2 op name l1 l2 = Nothing -- Could find LitLit
+
--------------------------
floatOp2 op name (MachFloat f1) (MachFloat f2)
= Just (name, mkFloatVal (f1 `op` f2))
do_lit_eq is_eq name lit expr
= Just (name, Case expr (mkWildId (literalType lit))
- [(LitAlt lit, [], val_if_eq),
- (DEFAULT, [], val_if_neq)])
+ [(DEFAULT, [], val_if_neq),
+ (LitAlt lit, [], val_if_eq)])
where
val_if_eq | is_eq = trueVal
| otherwise = falseVal
-- Int range, but not in a way suitable for cross-compiling... :-(
intResult :: RuleName -> Integer -> Maybe (RuleName, CoreExpr)
intResult name result
- = Just (name, mkIntVal (toInteger ((fromInteger result)::Int)))
+ = Just (name, mkIntVal (toInteger (fromInteger result :: Int)))
+
+#if __GLASGOW_HASKELL__ >= 500
+wordResult :: RuleName -> Integer -> Maybe (RuleName, CoreExpr)
+wordResult name result
+ = Just (name, mkWordVal (toInteger (fromInteger result :: Word)))
+#endif
\end{code}
twoLits :: (Literal -> Literal -> Maybe (RuleName, CoreExpr)) -> RuleFun
twoLits rule [Lit l1, Lit l2] = rule (convFloating l1) (convFloating l2)
-twoLits rule other = Nothing
+twoLits rule _ = Nothing
oneLit :: (Literal -> Maybe (RuleName, CoreExpr)) -> RuleFun
oneLit rule [Lit l1] = rule (convFloating l1)
-oneLit rule other = Nothing
+oneLit rule _ = Nothing
-- When excess precision is not requested, cut down the precision of the
-- Rational value to that of Float/Double. We confuse host architecture
mkDoubleVal d = Lit (convFloating (MachDouble d))
\end{code}
+\begin{code}
+nullAddrRule _ = Just(SLIT("nullAddr"), Lit(nullAddrLit))
+\end{code}
+
%************************************************************************
%* *
Just (SLIT("TagToEnum"), Var (dataConId dc))
where
correct_tag dc = (dataConTag dc - fIRST_TAG) == tag
- tag = fromInteger i
- (Just (tycon,_)) = splitTyConApp_maybe ty
+ tag = fromInteger i
+ tycon = tyConAppTyCon ty
tagToEnumRule other = Nothing
\end{code}
]
--- unpackFoldrCString# "foo" c (unpackFoldrCString# "baz" c n) = unpackFoldrCString# "foobaz" c n
+-- The rule is this:
+-- unpackFoldrCString# "foo" c (unpackFoldrCString# "baz" c n) = unpackFoldrCString# "foobaz" c n
match_append_lit_str [Type ty1,
Lit (MachStr s1),
]
| unpk `hasKey` unpackCStringFoldrIdKey &&
c1 `cheapEqExpr` c2
- = ASSERT( ty1 == ty2 )
+ = ASSERT( ty1 `eqType` ty2 )
Just (SLIT("AppendLitString"),
Var unpk `App` Type ty1
`App` Lit (MachStr (s1 _APPEND_ s2))