diff options
author | Ian Lynagh <igloo@earth.li> | 2011-11-06 20:36:52 +0000 |
---|---|---|
committer | Ian Lynagh <igloo@earth.li> | 2011-11-06 20:36:52 +0000 |
commit | e369e412098733f1503177b69d02a56e76ef7a1c (patch) | |
tree | b622cb97eefc4853a6bce0c89e33366b35a80e7a /compiler/deSugar/DsListComp.lhs | |
parent | 2a62d60f765fc0d28dc0e904e0c886a21010c188 (diff) | |
download | haskell-e369e412098733f1503177b69d02a56e76ef7a1c.tar.gz |
Whitespace in deSugar/DsListComp.lhs
Diffstat (limited to 'compiler/deSugar/DsListComp.lhs')
-rw-r--r-- | compiler/deSugar/DsListComp.lhs | 321 |
1 files changed, 157 insertions, 164 deletions
diff --git a/compiler/deSugar/DsListComp.lhs b/compiler/deSugar/DsListComp.lhs index 99dae74989..1daade77b8 100644 --- a/compiler/deSugar/DsListComp.lhs +++ b/compiler/deSugar/DsListComp.lhs @@ -6,13 +6,6 @@ Desugaring list comprehensions, monad comprehensions and array comprehensions \begin{code} -{-# OPTIONS -fno-warn-tabs #-} --- The above warning supression flag is a temporary kludge. --- While working on this module you are encouraged to remove it and --- detab the module (please do the detabbing in a separate patch). See --- http://hackage.haskell.org/trac/ghc/wiki/Commentary/CodingStyle#TabsvsSpaces --- for details - {-# LANGUAGE NamedFieldPuns #-} {-# OPTIONS -fno-warn-incomplete-patterns #-} -- The above warning supression flag is a temporary kludge. @@ -32,7 +25,7 @@ import TcHsSyn import CoreSyn import MkCore -import DsMonad -- the monadery used in the desugarer +import DsMonad -- the monadery used in the desugarer import DsUtils import DynFlags @@ -55,16 +48,16 @@ turned on'' (if you read Gill {\em et al.}'s paper on the subject). There will be at least one ``qualifier'' in the input. \begin{code} -dsListComp :: [LStmt Id] - -> Type -- Type of entire list - -> DsM CoreExpr -dsListComp lquals res_ty = do +dsListComp :: [LStmt Id] + -> Type -- Type of entire list + -> DsM CoreExpr +dsListComp lquals res_ty = do dflags <- getDOptsDs let quals = map unLoc lquals elt_ty = case tcTyConAppArgs res_ty of [elt_ty] -> elt_ty _ -> pprPanic "dsListComp" (ppr res_ty $$ ppr lquals) - + if not (dopt Opt_EnableRewriteRules dflags) || dopt Opt_IgnoreInterfacePragmas dflags -- Either rules are switched off, or we are ignoring what there are; -- Either way foldr/build won't happen, so use the more efficient @@ -72,32 +65,32 @@ dsListComp lquals res_ty = do || isParallelComp quals -- Foldr-style desugaring can't handle parallel list comprehensions then deListComp quals (mkNilExpr elt_ty) - else mkBuildExpr elt_ty (\(c, _) (n, _) -> dfListComp c n quals) - -- Foldr/build should be enabled, so desugar + else mkBuildExpr elt_ty (\(c, _) (n, _) -> dfListComp c n quals) + -- Foldr/build should be enabled, so desugar -- into foldrs and builds - where + where -- We must test for ParStmt anywhere, not just at the head, because an extension -- to list comprehensions would be to add brackets to specify the associativity -- of qualifier lists. This is really easy to do by adding extra ParStmts into the -- mix of possibly a single element in length, so we do this to leave the possibility open isParallelComp = any isParallelStmt - + isParallelStmt (ParStmt _ _ _ _) = True isParallelStmt _ = False - - + + -- This function lets you desugar a inner list comprehension and a list of the binders -- of that comprehension that we need in the outer comprehension into such an expression -- and the type of the elements that it outputs (tuples of binders) dsInnerListComp :: ([LStmt Id], [Id]) -> DsM (CoreExpr, Type) dsInnerListComp (stmts, bndrs) - = do { expr <- dsListComp (stmts ++ [noLoc $ mkLastStmt (mkBigLHsVarTup bndrs)]) + = do { expr <- dsListComp (stmts ++ [noLoc $ mkLastStmt (mkBigLHsVarTup bndrs)]) (mkListTy bndrs_tuple_type) ; return (expr, bndrs_tuple_type) } where bndrs_tuple_type = mkBigCoreVarTupTy bndrs - + -- This function factors out commonality between the desugaring strategies for GroupStmt. -- Given such a statement it gives you back an expression representing how to compute the transformed -- list and the tuple that you need to bind from that list in order to proceed with your desugaring @@ -108,19 +101,19 @@ dsTransStmt (TransStmt { trS_form = form, trS_stmts = stmts, trS_bndrs = binderM from_bndrs_tys = map idType from_bndrs to_bndrs_tys = map idType to_bndrs to_bndrs_tup_ty = mkBigCoreTupTy to_bndrs_tys - + -- Desugar an inner comprehension which outputs a list of tuples of the "from" binders (expr, from_tup_ty) <- dsInnerListComp (stmts, from_bndrs) - + -- Work out what arguments should be supplied to that expression: i.e. is an extraction -- function required? If so, create that desugared function and add to arguments usingExpr' <- dsLExpr using usingArgs <- case by of Nothing -> return [expr] - Just by_e -> do { by_e' <- dsLExpr by_e + Just by_e -> do { by_e' <- dsLExpr by_e ; lam <- matchTuple from_bndrs by_e' ; return [lam, expr] } - + -- Create an unzip function for the appropriate arity and element types and find "map" unzip_stuff <- mkUnzipBind form from_bndrs_tys map_id <- dsLookupGlobalId mapName @@ -132,7 +125,7 @@ dsTransStmt (TransStmt { trS_form = form, trS_stmts = stmts, trS_bndrs = binderM -- We make sure we instantiate the type variable "a" to be a list of "from" tuples and -- the "b" to be a tuple of "to" lists! -- Then finally we bind the unzip function around that expression - bound_unzipped_inner_list_expr + bound_unzipped_inner_list_expr = case unzip_stuff of Nothing -> inner_list_expr Just (unzip_fn, unzip_rhs) -> Let (Rec [(unzip_fn, unzip_rhs)]) $ @@ -142,16 +135,16 @@ dsTransStmt (TransStmt { trS_form = form, trS_stmts = stmts, trS_bndrs = binderM , Var unzip_fn , inner_list_expr] - -- Build a pattern that ensures the consumer binds into the NEW binders, + -- Build a pattern that ensures the consumer binds into the NEW binders, -- which hold lists rather than single values let pat = mkBigLHsVarPatTup to_bndrs return (bound_unzipped_inner_list_expr, pat) \end{code} %************************************************************************ -%* * +%* * \subsection[DsListComp-ordinary]{Ordinary desugaring of list comprehensions} -%* * +%* * %************************************************************************ Just as in Phil's chapter~7 in SLPJ, using the rules for @@ -176,11 +169,11 @@ TQ << [ e | b , qs ] ++ L >> = TQ << [ e | p <- L1, qs ] ++ L2 >> = letrec h = \ u1 -> - case u1 of - [] -> TE << L2 >> - (u2 : u3) -> - (( \ TE << p >> -> ( TQ << [e | qs] ++ (h u3) >> )) u2) - [] (h u3) + case u1 of + [] -> TE << L2 >> + (u2 : u3) -> + (( \ TE << p >> -> ( TQ << [e | qs] ++ (h u3) >> )) u2) + [] (h u3) in h ( TE << L1 >> ) @@ -218,13 +211,13 @@ deListComp :: [Stmt Id] -> CoreExpr -> DsM CoreExpr deListComp [] _ = panic "deListComp" -deListComp (LastStmt body _ : quals) list +deListComp (LastStmt body _ : quals) list = -- Figure 7.4, SLPJ, p 135, rule C above ASSERT( null quals ) do { core_body <- dsLExpr body ; return (mkConsExpr (exprType core_body) core_body list) } - -- Non-last: must be a guard + -- Non-last: must be a guard deListComp (ExprStmt guard _ _ _ : quals) list = do -- rule B above core_guard <- dsLExpr guard core_rest <- deListComp quals list @@ -246,18 +239,18 @@ deListComp (BindStmt pat list1 _ _ : quals) core_list2 = do -- rule A' above deListComp (ParStmt stmtss_w_bndrs _ _ _ : quals) list = do { exps_and_qual_tys <- mapM dsInnerListComp stmtss_w_bndrs ; let (exps, qual_tys) = unzip exps_and_qual_tys - + ; (zip_fn, zip_rhs) <- mkZipBind qual_tys - -- Deal with [e | pat <- zip l1 .. ln] in example above - ; deBindComp pat (Let (Rec [(zip_fn, zip_rhs)]) (mkApps (Var zip_fn) exps)) - quals list } - where - bndrs_s = map snd stmtss_w_bndrs + -- Deal with [e | pat <- zip l1 .. ln] in example above + ; deBindComp pat (Let (Rec [(zip_fn, zip_rhs)]) (mkApps (Var zip_fn) exps)) + quals list } + where + bndrs_s = map snd stmtss_w_bndrs - -- pat is the pattern ((x1,..,xn), (y1,..,ym)) in the example above - pat = mkBigLHsPatTup pats - pats = map mkBigLHsVarPatTup bndrs_s + -- pat is the pattern ((x1,..,xn), (y1,..,ym)) in the example above + pat = mkBigLHsPatTup pats + pats = map mkBigLHsVarPatTup bndrs_s \end{code} @@ -269,38 +262,38 @@ deBindComp :: OutPat Id -> DsM (Expr Id) deBindComp pat core_list1 quals core_list2 = do let - u3_ty@u1_ty = exprType core_list1 -- two names, same thing + u3_ty@u1_ty = exprType core_list1 -- two names, same thing -- u1_ty is a [alpha] type, and u2_ty = alpha u2_ty = hsLPatType pat res_ty = exprType core_list2 h_ty = u1_ty `mkFunTy` res_ty - + [h, u1, u2, u3] <- newSysLocalsDs [h_ty, u1_ty, u2_ty, u3_ty] -- the "fail" value ... let core_fail = App (Var h) (Var u3) letrec_body = App (Var h) core_list1 - + rest_expr <- deListComp quals core_fail - core_match <- matchSimply (Var u2) (StmtCtxt ListComp) pat rest_expr core_fail - + core_match <- matchSimply (Var u2) (StmtCtxt ListComp) pat rest_expr core_fail + let rhs = Lam u1 $ - Case (Var u1) u1 res_ty - [(DataAlt nilDataCon, [], core_list2), - (DataAlt consDataCon, [u2, u3], core_match)] - -- Increasing order of tag - + Case (Var u1) u1 res_ty + [(DataAlt nilDataCon, [], core_list2), + (DataAlt consDataCon, [u2, u3], core_match)] + -- Increasing order of tag + return (Let (Rec [(h, rhs)]) letrec_body) \end{code} %************************************************************************ -%* * +%* * \subsection[DsListComp-foldr-build]{Foldr/Build desugaring of list comprehensions} -%* * +%* * %************************************************************************ @dfListComp@ are the rules used with foldr/build turned on: @@ -308,12 +301,12 @@ deBindComp pat core_list1 quals core_list2 = do \begin{verbatim} TE[ e | ] c n = c e n TE[ e | b , q ] c n = if b then TE[ e | q ] c n else n -TE[ e | p <- l , q ] c n = let - f = \ x b -> case x of - p -> TE[ e | q ] c b - _ -> b - in - foldr f n l +TE[ e | p <- l , q ] c n = let + f = \ x b -> case x of + p -> TE[ e | q ] c b + _ -> b + in + foldr f n l \end{verbatim} \begin{code} @@ -323,12 +316,12 @@ dfListComp :: Id -> Id -- 'c' and 'n' dfListComp _ _ [] = panic "dfListComp" -dfListComp c_id n_id (LastStmt body _ : quals) +dfListComp c_id n_id (LastStmt body _ : quals) = ASSERT( null quals ) do { core_body <- dsLExpr body ; return (mkApps (Var c_id) [core_body, Var n_id]) } - -- Non-last: must be a guard + -- Non-last: must be a guard dfListComp c_id n_id (ExprStmt guard _ _ _ : quals) = do core_guard <- dsLExpr guard core_rest <- dfListComp c_id n_id quals @@ -342,19 +335,19 @@ dfListComp c_id n_id (LetStmt binds : quals) = do dfListComp c_id n_id (stmt@(TransStmt {}) : quals) = do (inner_list_expr, pat) <- dsTransStmt stmt -- Anyway, we bind the newly grouped list via the generic binding function - dfBindComp c_id n_id (pat, inner_list_expr) quals - + dfBindComp c_id n_id (pat, inner_list_expr) quals + dfListComp c_id n_id (BindStmt pat list1 _ _ : quals) = do -- evaluate the two lists core_list1 <- dsLExpr list1 - + -- Do the rest of the work in the generic binding builder dfBindComp c_id n_id (pat, core_list1) quals - -dfBindComp :: Id -> Id -- 'c' and 'n' + +dfBindComp :: Id -> Id -- 'c' and 'n' -> (LPat Id, CoreExpr) - -> [Stmt Id] -- the rest of the qual's - -> DsM CoreExpr + -> [Stmt Id] -- the rest of the qual's + -> DsM CoreExpr dfBindComp c_id n_id (pat, core_list1) quals = do -- find the required type let x_ty = hsLPatType pat @@ -368,87 +361,87 @@ dfBindComp c_id n_id (pat, core_list1) quals = do -- build the pattern match core_expr <- matchSimply (Var x) (StmtCtxt ListComp) - pat core_rest (Var b) + pat core_rest (Var b) -- now build the outermost foldr, and return mkFoldrExpr x_ty b_ty (mkLams [x, b] core_expr) (Var n_id) core_list1 \end{code} %************************************************************************ -%* * +%* * \subsection[DsFunGeneration]{Generation of zip/unzip functions for use in desugaring} -%* * +%* * %************************************************************************ \begin{code} mkZipBind :: [Type] -> DsM (Id, CoreExpr) --- mkZipBind [t1, t2] --- = (zip, \as1:[t1] as2:[t2] --- -> case as1 of --- [] -> [] --- (a1:as'1) -> case as2 of --- [] -> [] --- (a2:as'2) -> (a1, a2) : zip as'1 as'2)] +-- mkZipBind [t1, t2] +-- = (zip, \as1:[t1] as2:[t2] +-- -> case as1 of +-- [] -> [] +-- (a1:as'1) -> case as2 of +-- [] -> [] +-- (a2:as'2) -> (a1, a2) : zip as'1 as'2)] mkZipBind elt_tys = do ass <- mapM newSysLocalDs elt_list_tys as' <- mapM newSysLocalDs elt_tys as's <- mapM newSysLocalDs elt_list_tys - + zip_fn <- newSysLocalDs zip_fn_ty - - let inner_rhs = mkConsExpr elt_tuple_ty - (mkBigCoreVarTup as') - (mkVarApps (Var zip_fn) as's) + + let inner_rhs = mkConsExpr elt_tuple_ty + (mkBigCoreVarTup as') + (mkVarApps (Var zip_fn) as's) zip_body = foldr mk_case inner_rhs (zip3 ass as' as's) - + return (zip_fn, mkLams ass zip_body) where elt_list_tys = map mkListTy elt_tys elt_tuple_ty = mkBigCoreTupTy elt_tys elt_tuple_list_ty = mkListTy elt_tuple_ty - + zip_fn_ty = mkFunTys elt_list_tys elt_tuple_list_ty mk_case (as, a', as') rest - = Case (Var as) as elt_tuple_list_ty - [(DataAlt nilDataCon, [], mkNilExpr elt_tuple_ty), - (DataAlt consDataCon, [a', as'], rest)] - -- Increasing order of tag - - + = Case (Var as) as elt_tuple_list_ty + [(DataAlt nilDataCon, [], mkNilExpr elt_tuple_ty), + (DataAlt consDataCon, [a', as'], rest)] + -- Increasing order of tag + + mkUnzipBind :: TransForm -> [Type] -> DsM (Maybe (Id, CoreExpr)) --- mkUnzipBind [t1, t2] +-- mkUnzipBind [t1, t2] -- = (unzip, \ys :: [(t1, t2)] -> foldr (\ax :: (t1, t2) axs :: ([t1], [t2]) -- -> case ax of -- (x1, x2) -> case axs of -- (xs1, xs2) -> (x1 : xs1, x2 : xs2)) -- ([], []) -- ys) --- +-- -- We use foldr here in all cases, even if rules are turned off, because we may as well! mkUnzipBind ThenForm _ = return Nothing -- No unzipping for ThenForm -mkUnzipBind _ elt_tys +mkUnzipBind _ elt_tys = do { ax <- newSysLocalDs elt_tuple_ty ; axs <- newSysLocalDs elt_list_tuple_ty ; ys <- newSysLocalDs elt_tuple_list_ty ; xs <- mapM newSysLocalDs elt_tys ; xss <- mapM newSysLocalDs elt_list_tys - + ; unzip_fn <- newSysLocalDs unzip_fn_ty ; [us1, us2] <- sequence [newUniqueSupply, newUniqueSupply] ; let nil_tuple = mkBigCoreTup (map mkNilExpr elt_tys) - concat_expressions = map mkConcatExpression (zip3 elt_tys (map Var xs) (map Var xss)) - tupled_concat_expression = mkBigCoreTup concat_expressions - - folder_body_inner_case = mkTupleCase us1 xss tupled_concat_expression axs (Var axs) - folder_body_outer_case = mkTupleCase us2 xs folder_body_inner_case ax (Var ax) - folder_body = mkLams [ax, axs] folder_body_outer_case - + concat_expressions = map mkConcatExpression (zip3 elt_tys (map Var xs) (map Var xss)) + tupled_concat_expression = mkBigCoreTup concat_expressions + + folder_body_inner_case = mkTupleCase us1 xss tupled_concat_expression axs (Var axs) + folder_body_outer_case = mkTupleCase us2 xs folder_body_inner_case ax (Var ax) + folder_body = mkLams [ax, axs] folder_body_outer_case + ; unzip_body <- mkFoldrExpr elt_tuple_ty elt_list_tuple_ty folder_body nil_tuple (Var ys) ; return (Just (unzip_fn, mkLams [ys] unzip_body)) } where @@ -456,16 +449,16 @@ mkUnzipBind _ elt_tys elt_tuple_list_ty = mkListTy elt_tuple_ty elt_list_tys = map mkListTy elt_tys elt_list_tuple_ty = mkBigCoreTupTy elt_list_tys - + unzip_fn_ty = elt_tuple_list_ty `mkFunTy` elt_list_tuple_ty - + mkConcatExpression (list_element_ty, head, tail) = mkConsExpr list_element_ty head tail \end{code} %************************************************************************ -%* * +%* * \subsection[DsPArrComp]{Desugaring of array comprehensions} -%* * +%* * %************************************************************************ \begin{code} @@ -474,7 +467,7 @@ mkUnzipBind _ elt_tys -- -- [:e | qss:] = <<[:e | qss:]>> () [:():] -- -dsPArrComp :: [Stmt Id] +dsPArrComp :: [Stmt Id] -> DsM CoreExpr -- Special case for parallel comprehension @@ -486,7 +479,7 @@ dsPArrComp (ParStmt qss _ _ _ : quals) = dePArrParComp qss quals -- -- if matching again p cannot fail, or else -- --- <<[:e' | p <- e, qs:]>> = +-- <<[:e' | p <- e, qs:]>> = -- <<[:e' | qs:]>> p (filterP (\x -> case x of {p -> True; _ -> False}) e) -- dsPArrComp (BindStmt p e _ _ : qs) = do @@ -510,10 +503,10 @@ dsPArrComp qs = do -- no ParStmt in `qs' -- the work horse -- -dePArrComp :: [Stmt Id] - -> LPat Id -- the current generator pattern - -> CoreExpr -- the current generator expression - -> DsM CoreExpr +dePArrComp :: [Stmt Id] + -> LPat Id -- the current generator pattern + -> CoreExpr -- the current generator expression + -> DsM CoreExpr dePArrComp [] _ _ = panic "dePArrComp" @@ -543,7 +536,7 @@ dePArrComp (ExprStmt b _ _ _ : qs) pa cea = do -- -- if matching again p cannot fail, or else -- --- <<[:e' | p <- e, qs:]>> pa ea = +-- <<[:e' | p <- e, qs:]>> pa ea = -- let ef = \pa -> filterP (\x -> case x of {p -> True; _ -> False}) e -- in -- <<[:e' | qs:]>> (pa, p) (crossMapP ea ef) @@ -561,17 +554,17 @@ dePArrComp (BindStmt p e _ _ : qs) pa cea = do let cef | isIrrefutableHsPat p = ce | otherwise = mkApps (Var filterP) [Type ety'ce, mkLams [v] pred, ce] (clam, _) <- mkLambda ety'cea pa cef - let ety'cef = ety'ce -- filter doesn't change the element type + let ety'cef = ety'ce -- filter doesn't change the element type pa' = mkLHsPatTup [pa, p] - dePArrComp qs pa' (mkApps (Var crossMapP) + dePArrComp qs pa' (mkApps (Var crossMapP) [Type ety'cea, Type ety'cef, cea, clam]) -- --- <<[:e' | let ds, qs:]>> pa ea = --- <<[:e' | qs:]>> (pa, (x_1, ..., x_n)) --- (mapP (\v@pa -> let ds in (v, (x_1, ..., x_n))) ea) +-- <<[:e' | let ds, qs:]>> pa ea = +-- <<[:e' | qs:]>> (pa, (x_1, ..., x_n)) +-- (mapP (\v@pa -> let ds in (v, (x_1, ..., x_n))) ea) -- where --- {x_1, ..., x_n} = DV (ds) -- Defined Variables +-- {x_1, ..., x_n} = DV (ds) -- Defined Variables -- dePArrComp (LetStmt ds : qs) pa cea = do mapP <- dsLookupDPHId mapPName @@ -580,7 +573,7 @@ dePArrComp (LetStmt ds : qs) pa cea = do v <- newSysLocalDs ty'cea clet <- dsLocalBinds ds (mkCoreTup (map Var xs)) let'v <- newSysLocalDs (exprType clet) - let projBody = mkCoreLet (NonRec let'v clet) $ + let projBody = mkCoreLet (NonRec let'v clet) $ mkCoreTup [Var v, Var let'v] errTy = exprType projBody errMsg = ptext (sLit "DsListComp.dePArrComp: internal error!") @@ -588,19 +581,19 @@ dePArrComp (LetStmt ds : qs) pa cea = do ccase <- matchSimply (Var v) (StmtCtxt PArrComp) pa projBody cerr let pa' = mkLHsPatTup [pa, mkLHsPatTup (map nlVarPat xs)] proj = mkLams [v] ccase - dePArrComp qs pa' (mkApps (Var mapP) + dePArrComp qs pa' (mkApps (Var mapP) [Type ty'cea, Type errTy, proj, cea]) -- -- The parser guarantees that parallel comprehensions can only appear as -- singeltons qualifier lists, which we already special case in the caller. -- So, encountering one here is a bug. -- -dePArrComp (ParStmt _ _ _ _ : _) _ _ = +dePArrComp (ParStmt _ _ _ _ : _) _ _ = panic "DsListComp.dePArrComp: malformed comprehension AST" --- <<[:e' | qs | qss:]>> pa ea = --- <<[:e' | qss:]>> (pa, (x_1, ..., x_n)) --- (zipP ea <<[:(x_1, ..., x_n) | qs:]>>) +-- <<[:e' | qs | qss:]>> pa ea = +-- <<[:e' | qss:]>> (pa, (x_1, ..., x_n)) +-- (zipP ea <<[:(x_1, ..., x_n) | qs:]>>) -- where -- {x_1, ..., x_n} = DV (qs) -- @@ -630,19 +623,19 @@ dePArrParComp qss quals = do -- generate Core corresponding to `\p -> e' -- -deLambda :: Type -- type of the argument - -> LPat Id -- argument pattern - -> LHsExpr Id -- body - -> DsM (CoreExpr, Type) +deLambda :: Type -- type of the argument + -> LPat Id -- argument pattern + -> LHsExpr Id -- body + -> DsM (CoreExpr, Type) deLambda ty p e = mkLambda ty p =<< dsLExpr e -- generate Core for a lambda pattern match, where the body is already in Core -- -mkLambda :: Type -- type of the argument - -> LPat Id -- argument pattern - -> CoreExpr -- desugared body - -> DsM (CoreExpr, Type) +mkLambda :: Type -- type of the argument + -> LPat Id -- argument pattern + -> CoreExpr -- desugared body + -> DsM (CoreExpr, Type) mkLambda ty p ce = do v <- newSysLocalDs ty let errMsg = ptext (sLit "DsListComp.deLambda: internal error!") @@ -655,10 +648,10 @@ mkLambda ty p ce = do -- expression -- parrElemType :: CoreExpr -> Type -parrElemType e = +parrElemType e = case splitTyConApp_maybe (exprType e) of Just (tycon, [ty]) | tycon == parrTyCon -> ty - _ -> panic + _ -> panic "DsListComp.parrElemType: not a parallel array type" \end{code} @@ -683,7 +676,7 @@ dsMcStmt (LastStmt body ret_op) stmts ; return (App ret_op' body') } -- [ .. | let binds, stmts ] -dsMcStmt (LetStmt binds) stmts +dsMcStmt (LetStmt binds) stmts = do { rest <- dsMcStmts stmts ; dsLocalBinds binds rest } @@ -696,7 +689,7 @@ dsMcStmt (BindStmt pat rhs bind_op fail_op) stmts -- -- [ .. | exp, stmts ] -- -dsMcStmt (ExprStmt exp then_exp guard_exp _) stmts +dsMcStmt (ExprStmt exp then_exp guard_exp _) stmts = do { exp' <- dsLExpr exp ; guard_exp' <- dsExpr guard_exp ; then_exp' <- dsExpr then_exp @@ -707,7 +700,7 @@ dsMcStmt (ExprStmt exp then_exp guard_exp _) stmts -- Group statements desugar like this: -- -- [| (q, then group by e using f); rest |] --- ---> f {qt} (\qv -> e) [| q; return qv |] >>= \ n_tup -> +-- ---> f {qt} (\qv -> e) [| q; return qv |] >>= \ n_tup -> -- case unzip n_tup of qv' -> [| rest |] -- -- where variables (v1:t1, ..., vk:tk) are bound by q @@ -723,7 +716,7 @@ dsMcStmt (TransStmt { trS_stmts = stmts, trS_bndrs = bndrs , trS_ret = return_op, trS_bind = bind_op , trS_fmap = fmap_op, trS_form = form }) stmts_rest = do { let (from_bndrs, to_bndrs) = unzip bndrs - from_bndr_tys = map idType from_bndrs -- Types ty + from_bndr_tys = map idType from_bndrs -- Types ty -- Desugar an inner comprehension which outputs a list of tuples of the "from" binders ; expr <- dsInnerMonadComp stmts from_bndrs return_op @@ -738,7 +731,7 @@ dsMcStmt (TransStmt { trS_stmts = stmts, trS_bndrs = bndrs ; return [lam, expr] } -- Generate the expressions to build the grouped list - -- Build a pattern that ensures the consumer binds into the NEW binders, + -- Build a pattern that ensures the consumer binds into the NEW binders, -- which hold monads rather than single values ; bind_op' <- dsExpr bind_op ; let bind_ty = exprType bind_op' -- m2 (n (a,b,c)) -> (n (a,b,c) -> r1) -> r2 @@ -752,14 +745,14 @@ dsMcStmt (TransStmt { trS_stmts = stmts, trS_bndrs = bndrs ; us <- newUniqueSupply ; let rhs' = mkApps usingExpr' usingArgs body' = mkTupleCase us to_bndrs body tup_n_var tup_n_expr - + ; return (mkApps bind_op' [rhs', Lam n_tup_var body']) } -- Parallel statements. Use `Control.Monad.Zip.mzip` to zip parallel -- statements, for example: -- -- [ body | qs1 | qs2 | qs3 ] --- -> [ body | (bndrs1, (bndrs2, bndrs3)) +-- -> [ body | (bndrs1, (bndrs2, bndrs3)) -- <- [bndrs1 | qs1] `mzip` ([bndrs2 | qs2] `mzip` [bndrs3 | qs3]) ] -- -- where `mzip` has type @@ -775,16 +768,16 @@ dsMcStmt (ParStmt pairs mzip_op bind_op return_op) stmts_rest -- Pattern with tuples of variables -- [v1,v2,v3] => (v1, (v2, v3)) pat = foldr1 (\p1 p2 -> mkLHsPatTup [p1, p2]) pats - (rhs, _) = foldr1 (\(e1,t1) (e2,t2) -> + (rhs, _) = foldr1 (\(e1,t1) (e2,t2) -> (mkApps mzip_op' [Type t1, Type t2, e1, e2], - mkBoxedTupleTy [t1,t2])) + mkBoxedTupleTy [t1,t2])) exps_w_tys ; dsMcBindStmt pat rhs bind_op noSyntaxExpr stmts_rest } where ds_inner (stmts, bndrs) = do { exp <- dsInnerMonadComp stmts bndrs mono_ret_op ; return (exp, tup_ty) } - where + where mono_ret_op = HsWrap (WpTyApp tup_ty) return_op tup_ty = mkBigCoreVarTupTy bndrs @@ -794,7 +787,7 @@ dsMcStmt stmt _ = pprPanic "dsMcStmt: unexpected stmt" (ppr stmt) matchTuple :: [Id] -> CoreExpr -> DsM CoreExpr -- (matchTuple [a,b,c] body) -- returns the Core term --- \x. case x of (a,b,c) -> body +-- \x. case x of (a,b,c) -> body matchTuple ids body = do { us <- newUniqueSupply ; tup_id <- newSysLocalDs (mkBigCoreVarTupTy ids) @@ -809,10 +802,10 @@ dsMcBindStmt :: LPat Id -> [LStmt Id] -> DsM CoreExpr dsMcBindStmt pat rhs' bind_op fail_op stmts - = do { body <- dsMcStmts stmts + = do { body <- dsMcStmts stmts ; bind_op' <- dsExpr bind_op ; var <- selectSimpleMatchVarL pat - ; let bind_ty = exprType bind_op' -- rhs -> (pat -> res1) -> res2 + ; let bind_ty = exprType bind_op' -- rhs -> (pat -> res1) -> res2 res1_ty = funResultTy (funArgTy (funResultTy bind_ty)) ; match <- matchSinglePat (Var var) (StmtCtxt DoExpr) pat res1_ty (cantFailMatchResult body) @@ -828,20 +821,20 @@ dsMcBindStmt pat rhs' bind_op fail_op stmts ; fail_msg <- mkStringExpr (mk_fail_msg pat) ; extractMatchResult match (App fail_op' fail_msg) } | otherwise - = extractMatchResult match (error "It can't fail") + = extractMatchResult match (error "It can't fail") mk_fail_msg :: Located e -> String - mk_fail_msg pat = "Pattern match failure in monad comprehension at " ++ + mk_fail_msg pat = "Pattern match failure in monad comprehension at " ++ showSDoc (ppr (getLoc pat)) -- Desugar nested monad comprehensions, for example in `then..` constructs -- dsInnerMonadComp quals [a,b,c] ret_op --- returns the desugaring of +-- returns the desugaring of -- [ (a,b,c) | quals ] dsInnerMonadComp :: [LStmt Id] - -> [Id] -- Return a tuple of these variables - -> HsExpr Id -- The monomorphic "return" operator + -> [Id] -- Return a tuple of these variables + -> HsExpr Id -- The monomorphic "return" operator -> DsM CoreExpr dsInnerMonadComp stmts bndrs ret_op = dsMcStmts (stmts ++ [noLoc (LastStmt (mkBigLHsVarTup bndrs) ret_op)]) @@ -858,11 +851,11 @@ dsInnerMonadComp stmts bndrs ret_op -- , fmap (selN2 :: (t1, t2) -> t2) ys ) mkMcUnzipM :: TransForm - -> SyntaxExpr TcId -- fmap - -> Id -- Of type n (a,b,c) - -> [Type] -- [a,b,c] - -> DsM CoreExpr -- Of type (n a, n b, n c) -mkMcUnzipM ThenForm _ ys _ + -> SyntaxExpr TcId -- fmap + -> Id -- Of type n (a,b,c) + -> [Type] -- [a,b,c] + -> DsM CoreExpr -- Of type (n a, n b, n c) +mkMcUnzipM ThenForm _ ys _ = return (Var ys) -- No unzipping to do mkMcUnzipM _ fmap_op ys elt_tys @@ -870,12 +863,12 @@ mkMcUnzipM _ fmap_op ys elt_tys ; xs <- mapM newSysLocalDs elt_tys ; let tup_ty = mkBigCoreTupTy elt_tys ; tup_xs <- newSysLocalDs tup_ty - + ; let mk_elt i = mkApps fmap_op' -- fmap :: forall a b. (a -> b) -> n a -> n b [ Type tup_ty, Type (elt_tys !! i) , mk_sel i, Var ys] - mk_sel n = Lam tup_xs $ + mk_sel n = Lam tup_xs $ mkTupleSelector xs (xs !! n) tup_xs (Var tup_xs) ; return (mkBigCoreTup (map mk_elt [0..length elt_tys - 1])) } |