ghc/compiler/deSugar/DsUtils.lhs

   1 %
   2 % (c) The GRASP/AQUA Project, Glasgow University, 1992-1998
   3 %
   4 \section[DsUtils]{Utilities for desugaring}
   5
   6 This module exports some utility functions of no great interest.
   7
   8 \begin{code}
   9 module DsUtils (
  10         CanItFail(..), EquationInfo(..), MatchResult(..),
  11         EqnNo, EqnSet,
  12
  13         tidyLitPat, tidyNPat,
  14
  15         mkDsLet, mkDsLets,
  16
  17         cantFailMatchResult, extractMatchResult,
  18         combineMatchResults,
  19         adjustMatchResult, adjustMatchResultDs,
  20         mkCoLetsMatchResult, mkGuardedMatchResult,
  21         mkCoPrimCaseMatchResult, mkCoAlgCaseMatchResult,
  22
  23         mkErrorAppDs, mkNilExpr, mkConsExpr, mkListExpr,
  24         mkIntExpr, mkCharExpr,
  25         mkStringLit, mkStringLitFS, mkIntegerExpr,
  26
  27         mkSelectorBinds, mkTupleExpr, mkTupleSelector, mkCoreTup,
  28
  29         selectMatchVar
  30     ) where
  31
  32 #include "HsVersions.h"
  33
  34 import {-# SOURCE #-} Match ( matchSimply )
  35
  36 import HsSyn
  37 import TcHsSyn          ( TypecheckedPat, hsPatType )
  38 import CoreSyn
  39
  40 import DsMonad
  41
  42 import CoreUtils        ( exprType, mkIfThenElse, mkCoerce )
  43 import MkId             ( iRREFUT_PAT_ERROR_ID, mkReboxingAlt, mkNewTypeBody )
  44 import Id               ( idType, Id, mkWildId )
  45 import Literal          ( Literal(..), inIntRange, tARGET_MAX_INT )
  46 import TyCon            ( isNewTyCon, tyConDataCons )
  47 import DataCon          ( DataCon, dataConSourceArity )
  48 import Type             ( mkFunTy, isUnLiftedType, Type, splitTyConApp )
  49 import TcType           ( tcTyConAppTyCon, isIntTy, isFloatTy, isDoubleTy )
  50 import TysPrim          ( intPrimTy )
  51 import TysWiredIn       ( nilDataCon, consDataCon,
  52                           tupleCon,
  53                           unitDataConId, unitTy,
  54                           charTy, charDataCon,
  55                           intTy, intDataCon, smallIntegerDataCon,
  56                           floatDataCon,
  57                           doubleDataCon,
  58                           stringTy, isPArrFakeCon )
  59 import BasicTypes       ( Boxity(..) )
  60 import UniqSet          ( mkUniqSet, minusUniqSet, isEmptyUniqSet, UniqSet )
  61 import PrelNames        ( unpackCStringName, unpackCStringUtf8Name,
  62                           plusIntegerName, timesIntegerName,
  63                           lengthPName, indexPName )
  64 import Outputable
  65 import UnicodeUtil      ( intsToUtf8, stringToUtf8 )
  66 import Util             ( isSingleton, notNull )
  67 import FastString
  68 \end{code}
  69
  70
  71
  72 %************************************************************************
  73 %*                                                                      *
  74 \subsection{Tidying lit pats}
  75 %*                                                                      *
  76 %************************************************************************
  77
  78 \begin{code}
  79 tidyLitPat :: HsLit -> TypecheckedPat -> TypecheckedPat
  80 tidyLitPat (HsChar c) pat = mkCharLitPat c
  81 tidyLitPat lit        pat = pat
  82
  83 tidyNPat :: HsLit -> Type -> TypecheckedPat -> TypecheckedPat
  84 tidyNPat (HsString s) _ pat
  85   | lengthFS s <= 1     -- Short string literals only
  86   = foldr (\c pat -> mkPrefixConPat consDataCon [mkCharLitPat c,pat] stringTy)
  87           (mkNilPat stringTy) (unpackIntFS s)
  88         -- The stringTy is the type of the whole pattern, not
  89         -- the type to instantiate (:) or [] with!
  90   where
  91
  92 tidyNPat lit lit_ty default_pat
  93   | isIntTy lit_ty      = mkPrefixConPat intDataCon    [LitPat (mk_int lit)]    lit_ty
  94   | isFloatTy lit_ty    = mkPrefixConPat floatDataCon  [LitPat (mk_float lit)]  lit_ty
  95   | isDoubleTy lit_ty   = mkPrefixConPat doubleDataCon [LitPat (mk_double lit)] lit_ty
  96   | otherwise           = default_pat
  97
  98   where
  99     mk_int    (HsInteger i) = HsIntPrim i
 100
 101     mk_float  (HsInteger i) = HsFloatPrim (fromInteger i)
 102     mk_float  (HsRat f _)   = HsFloatPrim f
 103
 104     mk_double (HsInteger i) = HsDoublePrim (fromInteger i)
 105     mk_double (HsRat f _)   = HsDoublePrim f
 106 \end{code}
 107
 108
 109 %************************************************************************
 110 %*                                                                      *
 111 \subsection{Building lets}
 112 %*                                                                      *
 113 %************************************************************************
 114
 115 Use case, not let for unlifted types.  The simplifier will turn some
 116 back again.
 117
 118 \begin{code}
 119 mkDsLet :: CoreBind -> CoreExpr -> CoreExpr
 120 mkDsLet (NonRec bndr rhs) body
 121   | isUnLiftedType (idType bndr) = Case rhs bndr [(DEFAULT,[],body)]
 122 mkDsLet bind body
 123   = Let bind body
 124
 125 mkDsLets :: [CoreBind] -> CoreExpr -> CoreExpr
 126 mkDsLets binds body = foldr mkDsLet body binds
 127 \end{code}
 128
 129
 130 %************************************************************************
 131 %*                                                                      *
 132 \subsection{ Selecting match variables}
 133 %*                                                                      *
 134 %************************************************************************
 135
 136 We're about to match against some patterns.  We want to make some
 137 @Ids@ to use as match variables.  If a pattern has an @Id@ readily at
 138 hand, which should indeed be bound to the pattern as a whole, then use it;
 139 otherwise, make one up.
 140
 141 \begin{code}
 142 selectMatchVar :: TypecheckedPat -> DsM Id
 143 selectMatchVar (VarPat var)     = returnDs var
 144 selectMatchVar (AsPat var pat)  = returnDs var
 145 selectMatchVar (LazyPat pat)    = selectMatchVar pat
 146 selectMatchVar other_pat        = newSysLocalDs (hsPatType other_pat) -- OK, better make up one...
 147 \end{code}
 148
 149
 150 %************************************************************************
 151 %*                                                                      *
 152 %* type synonym EquationInfo and access functions for its pieces        *
 153 %*                                                                      *
 154 %************************************************************************
 155 \subsection[EquationInfo-synonym]{@EquationInfo@: a useful synonym}
 156
 157 The ``equation info'' used by @match@ is relatively complicated and
 158 worthy of a type synonym and a few handy functions.
 159
 160 \begin{code}
 161
 162 type EqnNo   = Int
 163 type EqnSet  = UniqSet EqnNo
 164
 165 data EquationInfo
 166   = EqnInfo
 167         EqnNo           -- The number of the equation
 168
 169         DsMatchContext  -- The context info is used when producing warnings
 170                         -- about shadowed patterns.  It's the context
 171                         -- of the *first* thing matched in this group.
 172                         -- Should perhaps be a list of them all!
 173
 174         [TypecheckedPat]    -- The patterns for an eqn
 175
 176         MatchResult         -- Encapsulates the guards and bindings
 177 \end{code}
 178
 179 \begin{code}
 180 data MatchResult
 181   = MatchResult
 182         CanItFail       -- Tells whether the failure expression is used
 183         (CoreExpr -> DsM CoreExpr)
 184                         -- Takes a expression to plug in at the
 185                         -- failure point(s). The expression should
 186                         -- be duplicatable!
 187
 188 data CanItFail = CanFail | CantFail
 189
 190 orFail CantFail CantFail = CantFail
 191 orFail _        _        = CanFail
 192 \end{code}
 193
 194 Functions on MatchResults
 195
 196 \begin{code}
 197 cantFailMatchResult :: CoreExpr -> MatchResult
 198 cantFailMatchResult expr = MatchResult CantFail (\ ignore -> returnDs expr)
 199
 200 extractMatchResult :: MatchResult -> CoreExpr -> DsM CoreExpr
 201 extractMatchResult (MatchResult CantFail match_fn) fail_expr
 202   = match_fn (error "It can't fail!")
 203
 204 extractMatchResult (MatchResult CanFail match_fn) fail_expr
 205   = mkFailurePair fail_expr             `thenDs` \ (fail_bind, if_it_fails) ->
 206     match_fn if_it_fails                `thenDs` \ body ->
 207     returnDs (mkDsLet fail_bind body)
 208
 209
 210 combineMatchResults :: MatchResult -> MatchResult -> MatchResult
 211 combineMatchResults (MatchResult CanFail      body_fn1)
 212                     (MatchResult can_it_fail2 body_fn2)
 213   = MatchResult can_it_fail2 body_fn
 214   where
 215     body_fn fail = body_fn2 fail                        `thenDs` \ body2 ->
 216                    mkFailurePair body2                  `thenDs` \ (fail_bind, duplicatable_expr) ->
 217                    body_fn1 duplicatable_expr           `thenDs` \ body1 ->
 218                    returnDs (Let fail_bind body1)
 219
 220 combineMatchResults match_result1@(MatchResult CantFail body_fn1) match_result2
 221   = match_result1
 222
 223
 224 adjustMatchResult :: (CoreExpr -> CoreExpr) -> MatchResult -> MatchResult
 225 adjustMatchResult encl_fn (MatchResult can_it_fail body_fn)
 226   = MatchResult can_it_fail (\fail -> body_fn fail      `thenDs` \ body ->
 227                                       returnDs (encl_fn body))
 228
 229 adjustMatchResultDs :: (CoreExpr -> DsM CoreExpr) -> MatchResult -> MatchResult
 230 adjustMatchResultDs encl_fn (MatchResult can_it_fail body_fn)
 231   = MatchResult can_it_fail (\fail -> body_fn fail      `thenDs` \ body ->
 232                                       encl_fn body)
 233
 234
 235 mkCoLetsMatchResult :: [CoreBind] -> MatchResult -> MatchResult
 236 mkCoLetsMatchResult binds match_result
 237   = adjustMatchResult (mkDsLets binds) match_result
 238
 239
 240 mkGuardedMatchResult :: CoreExpr -> MatchResult -> MatchResult
 241 mkGuardedMatchResult pred_expr (MatchResult can_it_fail body_fn)
 242   = MatchResult CanFail (\fail -> body_fn fail  `thenDs` \ body ->
 243                                   returnDs (mkIfThenElse pred_expr body fail))
 244
 245 mkCoPrimCaseMatchResult :: Id                           -- Scrutinee
 246                     -> [(Literal, MatchResult)]         -- Alternatives
 247                     -> MatchResult
 248 mkCoPrimCaseMatchResult var match_alts
 249   = MatchResult CanFail mk_case
 250   where
 251     mk_case fail
 252       = mapDs (mk_alt fail) match_alts          `thenDs` \ alts ->
 253         returnDs (Case (Var var) var ((DEFAULT, [], fail) : alts))
 254
 255     mk_alt fail (lit, MatchResult _ body_fn) = body_fn fail     `thenDs` \ body ->
 256                                                returnDs (LitAlt lit, [], body)
 257
 258
 259 mkCoAlgCaseMatchResult :: Id                                    -- Scrutinee
 260                     -> [(DataCon, [CoreBndr], MatchResult)]     -- Alternatives
 261                     -> MatchResult
 262
 263 mkCoAlgCaseMatchResult var match_alts
 264   | isNewTyCon tycon            -- Newtype case; use a let
 265   = ASSERT( null (tail match_alts) && null (tail arg_ids) )
 266     mkCoLetsMatchResult [NonRec arg_id newtype_rhs] match_result
 267
 268   | isPArrFakeAlts match_alts   -- Sugared parallel array; use a literal case
 269   = MatchResult CanFail mk_parrCase
 270
 271   | otherwise                   -- Datatype case; use a case
 272   = MatchResult fail_flag mk_case
 273   where
 274         -- Common stuff
 275     scrut_ty = idType var
 276     tycon    = tcTyConAppTyCon scrut_ty         -- Newtypes must be opaque here
 277
 278         -- Stuff for newtype
 279     (_, arg_ids, match_result) = head match_alts
 280     arg_id                     = head arg_ids
 281     newtype_rhs                = mkNewTypeBody tycon (idType arg_id) (Var var)
 282
 283         -- Stuff for data types
 284     data_cons      = tyConDataCons tycon
 285     match_results  = [match_result | (_,_,match_result) <- match_alts]
 286
 287     fail_flag | exhaustive_case
 288               = foldr1 orFail [can_it_fail | MatchResult can_it_fail _ <- match_results]
 289               | otherwise
 290               = CanFail
 291
 292     wild_var = mkWildId (idType var)
 293     mk_case fail = mapDs (mk_alt fail) match_alts       `thenDs` \ alts ->
 294                    returnDs (Case (Var var) wild_var (mk_default fail ++ alts))
 295
 296     mk_alt fail (con, args, MatchResult _ body_fn)
 297         = body_fn fail                          `thenDs` \ body ->
 298           getUniquesDs                          `thenDs` \ us ->
 299           returnDs (mkReboxingAlt us con args body)
 300
 301     mk_default fail | exhaustive_case = []
 302                     | otherwise       = [(DEFAULT, [], fail)]
 303
 304     un_mentioned_constructors
 305         = mkUniqSet data_cons `minusUniqSet` mkUniqSet [ con | (con, _, _) <- match_alts]
 306     exhaustive_case = isEmptyUniqSet un_mentioned_constructors
 307
 308         -- Stuff for parallel arrays
 309         --
 310         -- * the following is to desugar cases over fake constructors for
 311         --   parallel arrays, which are introduced by `tidy1' in the `PArrPat'
 312         --   case
 313         --
 314         -- Concerning `isPArrFakeAlts':
 315         --
 316         -- * it is *not* sufficient to just check the type of the type
 317         --   constructor, as we have to be careful not to confuse the real
 318         --   representation of parallel arrays with the fake constructors;
 319         --   moreover, a list of alternatives must not mix fake and real
 320         --   constructors (this is checked earlier on)
 321         --
 322         -- FIXME: We actually go through the whole list and make sure that
 323         --        either all or none of the constructors are fake parallel
 324         --        array constructors.  This is to spot equations that mix fake
 325         --        constructors with the real representation defined in
 326         --        `PrelPArr'.  It would be nicer to spot this situation
 327         --        earlier and raise a proper error message, but it can really
 328         --        only happen in `PrelPArr' anyway.
 329         --
 330     isPArrFakeAlts [(dcon, _, _)]      = isPArrFakeCon dcon
 331     isPArrFakeAlts ((dcon, _, _):alts) =
 332       case (isPArrFakeCon dcon, isPArrFakeAlts alts) of
 333         (True , True ) -> True
 334         (False, False) -> False
 335         _              ->
 336           panic "DsUtils: You may not mix `[:...:]' with `PArr' patterns"
 337     --
 338     mk_parrCase fail =
 339       dsLookupGlobalId lengthPName                      `thenDs` \lengthP  ->
 340       unboxAlt                                          `thenDs` \alt      ->
 341       returnDs (Case (len lengthP) (mkWildId intTy) [alt])
 342       where
 343         elemTy      = case splitTyConApp (idType var) of
 344                         (_, [elemTy]) -> elemTy
 345                         _               -> panic panicMsg
 346         panicMsg    = "DsUtils.mkCoAlgCaseMatchResult: not a parallel array?"
 347         len lengthP = mkApps (Var lengthP) [Type elemTy, Var var]
 348         --
 349         unboxAlt =
 350           newSysLocalDs intPrimTy                       `thenDs` \l        ->
 351           dsLookupGlobalId indexPName           `thenDs` \indexP   ->
 352           mapDs (mkAlt indexP) match_alts               `thenDs` \alts     ->
 353           returnDs (DataAlt intDataCon, [l], (Case (Var l) wild (dft : alts)))
 354           where
 355             wild = mkWildId intPrimTy
 356             dft  = (DEFAULT, [], fail)
 357         --
 358         -- each alternative matches one array length (corresponding to one
 359         -- fake array constructor), so the match is on a literal; each
 360         -- alternative's body is extended by a local binding for each
 361         -- constructor argument, which are bound to array elements starting
 362         -- with the first
 363         --
 364         mkAlt indexP (con, args, MatchResult _ bodyFun) =
 365           bodyFun fail                                  `thenDs` \body     ->
 366           returnDs (LitAlt lit, [], mkDsLets binds body)
 367           where
 368             lit   = MachInt $ toInteger (dataConSourceArity con)
 369             binds = [NonRec arg (indexExpr i) | (i, arg) <- zip [1..] args]
 370             --
 371             indexExpr i = mkApps (Var indexP) [Type elemTy, Var var, mkIntExpr i]
 372 \end{code}
 373
 374
 375 %************************************************************************
 376 %*                                                                      *
 377 \subsection{Desugarer's versions of some Core functions}
 378 %*                                                                      *
 379 %************************************************************************
 380
 381 \begin{code}
 382 mkErrorAppDs :: Id              -- The error function
 383              -> Type            -- Type to which it should be applied
 384              -> String          -- The error message string to pass
 385              -> DsM CoreExpr
 386
 387 mkErrorAppDs err_id ty msg
 388   = getSrcLocDs                 `thenDs` \ src_loc ->
 389     let
 390         full_msg = showSDoc (hcat [ppr src_loc, text "|", text msg])
 391         core_msg = Lit (MachStr (mkFastString (stringToUtf8 full_msg)))
 392     in
 393     returnDs (mkApps (Var err_id) [Type ty, core_msg])
 394 \end{code}
 395
 396
 397 *************************************************************
 398 %*                                                                      *
 399 \subsection{Making literals}
 400 %*                                                                      *
 401 %************************************************************************
 402
 403 \begin{code}
 404 mkCharExpr    :: Int        -> CoreExpr      -- Returns C# c :: Int
 405 mkIntExpr     :: Integer    -> CoreExpr      -- Returns I# i :: Int
 406 mkIntegerExpr :: Integer    -> DsM CoreExpr  -- Result :: Integer
 407 mkStringLit   :: String     -> DsM CoreExpr  -- Result :: String
 408 mkStringLitFS :: FastString -> DsM CoreExpr  -- Result :: String
 409
 410 mkIntExpr  i = mkConApp intDataCon  [mkIntLit i]
 411 mkCharExpr c = mkConApp charDataCon [mkLit (MachChar c)]
 412
 413 mkIntegerExpr i
 414   | inIntRange i        -- Small enough, so start from an Int
 415   = returnDs (mkSmallIntegerLit i)
 416
 417 -- Special case for integral literals with a large magnitude:
 418 -- They are transformed into an expression involving only smaller
 419 -- integral literals. This improves constant folding.
 420
 421   | otherwise           -- Big, so start from a string
 422   = dsLookupGlobalId plusIntegerName            `thenDs` \ plus_id ->
 423     dsLookupGlobalId timesIntegerName   `thenDs` \ times_id ->
 424     let
 425         plus a b  = Var plus_id  `App` a `App` b
 426         times a b = Var times_id `App` a `App` b
 427
 428         -- Transform i into (x1 + (x2 + (x3 + (...) * b) * b) * b) with abs xi <= b
 429         horner :: Integer -> Integer -> CoreExpr
 430         horner b i | abs q <= 1 = if r == 0 || r == i
 431                                   then mkSmallIntegerLit i
 432                                   else mkSmallIntegerLit r `plus` mkSmallIntegerLit (i-r)
 433                    | r == 0     =                             horner b q `times` mkSmallIntegerLit b
 434                    | otherwise  = mkSmallIntegerLit r `plus` (horner b q `times` mkSmallIntegerLit b)
 435                    where
 436                      (q,r) = i `quotRem` b
 437
 438     in
 439     returnDs (horner tARGET_MAX_INT i)
 440
 441 mkSmallIntegerLit i = mkConApp smallIntegerDataCon [mkIntLit i]
 442
 443 mkStringLit str = mkStringLitFS (mkFastString str)
 444
 445 mkStringLitFS str
 446   | nullFastString str
 447   = returnDs (mkNilExpr charTy)
 448
 449   | lengthFS str == 1
 450   = let
 451         the_char = mkCharExpr (headIntFS str)
 452     in
 453     returnDs (mkConsExpr charTy the_char (mkNilExpr charTy))
 454
 455   | all safeChar int_chars
 456   = dsLookupGlobalId unpackCStringName  `thenDs` \ unpack_id ->
 457     returnDs (App (Var unpack_id) (Lit (MachStr str)))
 458
 459   | otherwise
 460   = dsLookupGlobalId unpackCStringUtf8Name      `thenDs` \ unpack_id ->
 461     returnDs (App (Var unpack_id) (Lit (MachStr (mkFastString (intsToUtf8 int_chars)))))
 462
 463   where
 464     int_chars = unpackIntFS str
 465     safeChar c = c >= 1 && c <= 0xFF
 466 \end{code}
 467
 468
 469 %************************************************************************
 470 %*                                                                      *
 471 \subsection[mkSelectorBind]{Make a selector bind}
 472 %*                                                                      *
 473 %************************************************************************
 474
 475 This is used in various places to do with lazy patterns.
 476 For each binder $b$ in the pattern, we create a binding:
 477 \begin{verbatim}
 478     b = case v of pat' -> b'
 479 \end{verbatim}
 480 where @pat'@ is @pat@ with each binder @b@ cloned into @b'@.
 481
 482 ToDo: making these bindings should really depend on whether there's
 483 much work to be done per binding.  If the pattern is complex, it
 484 should be de-mangled once, into a tuple (and then selected from).
 485 Otherwise the demangling can be in-line in the bindings (as here).
 486
 487 Boring!  Boring!  One error message per binder.  The above ToDo is
 488 even more helpful.  Something very similar happens for pattern-bound
 489 expressions.
 490
 491 \begin{code}
 492 mkSelectorBinds :: TypecheckedPat       -- The pattern
 493                 -> CoreExpr             -- Expression to which the pattern is bound
 494                 -> DsM [(Id,CoreExpr)]
 495
 496 mkSelectorBinds (VarPat v) val_expr
 497   = returnDs [(v, val_expr)]
 498
 499 mkSelectorBinds pat val_expr
 500   | isSingleton binders || is_simple_pat pat
 501   = newSysLocalDs (exprType val_expr)   `thenDs` \ val_var ->
 502
 503         -- For the error message we make one error-app, to avoid duplication.
 504         -- But we need it at different types... so we use coerce for that
 505     mkErrorAppDs iRREFUT_PAT_ERROR_ID
 506                  unitTy (showSDoc (ppr pat))    `thenDs` \ err_expr ->
 507     newSysLocalDs unitTy                        `thenDs` \ err_var ->
 508     mapDs (mk_bind val_var err_var) binders     `thenDs` \ binds ->
 509     returnDs ( (val_var, val_expr) :
 510                (err_var, err_expr) :
 511                binds )
 512
 513
 514   | otherwise
 515   = mkErrorAppDs iRREFUT_PAT_ERROR_ID
 516                  tuple_ty (showSDoc (ppr pat))                  `thenDs` \ error_expr ->
 517     matchSimply val_expr PatBindRhs pat local_tuple error_expr  `thenDs` \ tuple_expr ->
 518     newSysLocalDs tuple_ty                                      `thenDs` \ tuple_var ->
 519     let
 520         mk_tup_bind binder
 521           = (binder, mkTupleSelector binders binder tuple_var (Var tuple_var))
 522     in
 523     returnDs ( (tuple_var, tuple_expr) : map mk_tup_bind binders )
 524   where
 525     binders     = collectPatBinders pat
 526     local_tuple = mkTupleExpr binders
 527     tuple_ty    = exprType local_tuple
 528
 529     mk_bind scrut_var err_var bndr_var
 530     -- (mk_bind sv err_var) generates
 531     --          bv = case sv of { pat -> bv; other -> coerce (type-of-bv) err_var }
 532     -- Remember, pat binds bv
 533       = matchSimply (Var scrut_var) PatBindRhs pat
 534                     (Var bndr_var) error_expr                   `thenDs` \ rhs_expr ->
 535         returnDs (bndr_var, rhs_expr)
 536       where
 537         error_expr = mkCoerce (idType bndr_var) (Var err_var)
 538
 539     is_simple_pat (TuplePat ps Boxed)    = all is_triv_pat ps
 540     is_simple_pat (ConPatOut _ ps _ _ _) = all is_triv_pat (hsConArgs ps)
 541     is_simple_pat (VarPat _)             = True
 542     is_simple_pat (ParPat p)             = is_simple_pat p
 543     is_simple_pat other                  = False
 544
 545     is_triv_pat (VarPat v)  = True
 546     is_triv_pat (WildPat _) = True
 547     is_triv_pat (ParPat p)  = is_triv_pat p
 548     is_triv_pat other       = False
 549 \end{code}
 550
 551
 552 @mkTupleExpr@ builds a tuple; the inverse to @mkTupleSelector@.  If it
 553 has only one element, it is the identity function.
 554
 555 \begin{code}
 556 mkTupleExpr :: [Id] -> CoreExpr
 557
 558 {- This code has been replaced by mkCoreTup below
 559 mkTupleExpr []   = Var unitDataConId
 560 mkTupleExpr [id] = Var id
 561 mkTupleExpr ids  = mkConApp (tupleCon Boxed (length ids))
 562                             (map (Type . idType) ids ++ [ Var i | i <-ids])
 563 -}
 564
 565 mkTupleExpr ids = mkCoreTup(map Var ids)
 566
 567 mkCoreTup :: [CoreExpr] -> CoreExpr
 568 mkCoreTup []   = Var unitDataConId
 569 mkCoreTup [c]  = c
 570 mkCoreTup cs  = mkConApp (tupleCon Boxed (length cs))
 571                          (map (Type . exprType) cs ++ cs)
 572
 573 \end{code}
 574
 575
 576 @mkTupleSelector@ builds a selector which scrutises the given
 577 expression and extracts the one name from the list given.
 578 If you want the no-shadowing rule to apply, the caller
 579 is responsible for making sure that none of these names
 580 are in scope.
 581
 582 If there is just one id in the ``tuple'', then the selector is
 583 just the identity.
 584
 585 \begin{code}
 586 mkTupleSelector :: [Id]         -- The tuple args
 587                 -> Id           -- The selected one
 588                 -> Id           -- A variable of the same type as the scrutinee
 589                 -> CoreExpr     -- Scrutinee
 590                 -> CoreExpr
 591
 592 mkTupleSelector [var] should_be_the_same_var scrut_var scrut
 593   = ASSERT(var == should_be_the_same_var)
 594     scrut
 595
 596 mkTupleSelector vars the_var scrut_var scrut
 597   = ASSERT( notNull vars )
 598     Case scrut scrut_var [(DataAlt (tupleCon Boxed (length vars)), vars, Var the_var)]
 599 \end{code}
 600
 601
 602 %************************************************************************
 603 %*                                                                      *
 604 \subsection[mkFailurePair]{Code for pattern-matching and other failures}
 605 %*                                                                      *
 606 %************************************************************************
 607
 608 Call the constructor Ids when building explicit lists, so that they
 609 interact well with rules.
 610
 611 \begin{code}
 612 mkNilExpr :: Type -> CoreExpr
 613 mkNilExpr ty = mkConApp nilDataCon [Type ty]
 614
 615 mkConsExpr :: Type -> CoreExpr -> CoreExpr -> CoreExpr
 616 mkConsExpr ty hd tl = mkConApp consDataCon [Type ty, hd, tl]
 617
 618 mkListExpr :: Type -> [CoreExpr] -> CoreExpr
 619 mkListExpr ty xs = foldr (mkConsExpr ty) (mkNilExpr ty) xs
 620
 621 \end{code}
 622
 623
 624 %************************************************************************
 625 %*                                                                      *
 626 \subsection[mkFailurePair]{Code for pattern-matching and other failures}
 627 %*                                                                      *
 628 %************************************************************************
 629
 630 Generally, we handle pattern matching failure like this: let-bind a
 631 fail-variable, and use that variable if the thing fails:
 632 \begin{verbatim}
 633         let fail.33 = error "Help"
 634         in
 635         case x of
 636                 p1 -> ...
 637                 p2 -> fail.33
 638                 p3 -> fail.33
 639                 p4 -> ...
 640 \end{verbatim}
 641 Then
 642 \begin{itemize}
 643 \item
 644 If the case can't fail, then there'll be no mention of @fail.33@, and the
 645 simplifier will later discard it.
 646
 647 \item
 648 If it can fail in only one way, then the simplifier will inline it.
 649
 650 \item
 651 Only if it is used more than once will the let-binding remain.
 652 \end{itemize}
 653
 654 There's a problem when the result of the case expression is of
 655 unboxed type.  Then the type of @fail.33@ is unboxed too, and
 656 there is every chance that someone will change the let into a case:
 657 \begin{verbatim}
 658         case error "Help" of
 659           fail.33 -> case ....
 660 \end{verbatim}
 661
 662 which is of course utterly wrong.  Rather than drop the condition that
 663 only boxed types can be let-bound, we just turn the fail into a function
 664 for the primitive case:
 665 \begin{verbatim}
 666         let fail.33 :: Void -> Int#
 667             fail.33 = \_ -> error "Help"
 668         in
 669         case x of
 670                 p1 -> ...
 671                 p2 -> fail.33 void
 672                 p3 -> fail.33 void
 673                 p4 -> ...
 674 \end{verbatim}
 675
 676 Now @fail.33@ is a function, so it can be let-bound.
 677
 678 \begin{code}
 679 mkFailurePair :: CoreExpr       -- Result type of the whole case expression
 680               -> DsM (CoreBind, -- Binds the newly-created fail variable
 681                                 -- to either the expression or \ _ -> expression
 682                       CoreExpr) -- Either the fail variable, or fail variable
 683                                 -- applied to unit tuple
 684 mkFailurePair expr
 685   | isUnLiftedType ty
 686   = newFailLocalDs (unitTy `mkFunTy` ty)        `thenDs` \ fail_fun_var ->
 687     newSysLocalDs unitTy                        `thenDs` \ fail_fun_arg ->
 688     returnDs (NonRec fail_fun_var (Lam fail_fun_arg expr),
 689               App (Var fail_fun_var) (Var unitDataConId))
 690
 691   | otherwise
 692   = newFailLocalDs ty           `thenDs` \ fail_var ->
 693     returnDs (NonRec fail_var expr, Var fail_var)
 694   where
 695     ty = exprType expr
 696 \end{code}
 697
 698