[project @ 2001-05-01 09:16:55 by qrczak]
[ghc-hetmet.git] / ghc / compiler / rename / RnExpr.lhs
index 3cf439d..eb92cd3 100644 (file)
@@ -11,40 +11,42 @@ free variables.
 
 \begin{code}
 module RnExpr (
-       rnMatch, rnGRHSs, rnPat, rnExpr, rnExprs,
+       rnMatch, rnGRHSs, rnPat, rnExpr, rnExprs, rnStmt,
        checkPrecMatch
    ) where
 
 #include "HsVersions.h"
 
 import {-# SOURCE #-} RnBinds  ( rnBinds ) 
-import {-# SOURCE #-} RnSource ( rnHsSigType, rnHsType )
+import {-# SOURCE #-} RnSource ( rnHsTypeFVs )
 
 import HsSyn
 import RdrHsSyn
 import RnHsSyn
 import RnMonad
 import RnEnv
-import RnIfaces                ( lookupFixityRn )
+import RnHiFiles       ( lookupFixityRn )
 import CmdLineOpts     ( DynFlag(..), opt_IgnoreAsserts )
-import Literal         ( inIntRange )
+import Literal         ( inIntRange, inCharRange )
 import BasicTypes      ( Fixity(..), FixityDirection(..), defaultFixity, negateFixity )
-import PrelNames       ( hasKey, assertIdKey,
+import PrelNames       ( hasKey, assertIdKey, minusName, negateName, fromIntegerName,
                          eqClass_RDR, foldr_RDR, build_RDR, eqString_RDR,
                          cCallableClass_RDR, cReturnableClass_RDR, 
                          monadClass_RDR, enumClass_RDR, ordClass_RDR,
-                         ratioDataCon_RDR, negate_RDR, assertErr_RDR,
-                         ioDataCon_RDR, plusInteger_RDR, timesInteger_RDR
+                         ratioDataCon_RDR, assertErr_RDR,
+                         ioDataCon_RDR, plusInteger_RDR, timesInteger_RDR,
+                         fromInteger_RDR, fromRational_RDR,
                        )
 import TysPrim         ( charPrimTyCon, addrPrimTyCon, intPrimTyCon, 
                          floatPrimTyCon, doublePrimTyCon
                        )
-import TysWiredIn      ( intTyCon, integerTyCon )
+import TysWiredIn      ( intTyCon )
 import Name            ( NamedThing(..), mkSysLocalName, nameSrcLoc )
 import NameSet
 import UniqFM          ( isNullUFM )
 import FiniteMap       ( elemFM )
 import UniqSet         ( emptyUniqSet )
+import List            ( intersectBy )
 import ListSetOps      ( unionLists, removeDups )
 import Maybes          ( maybeToBool )
 import Outputable
@@ -71,7 +73,7 @@ rnPat (SigPatIn pat ty)
     
     if glaExts
     then rnPat pat             `thenRn` \ (pat', fvs1) ->
-         rnHsType doc ty       `thenRn` \ (ty',  fvs2) ->
+         rnHsTypeFVs doc ty    `thenRn` \ (ty',  fvs2) ->
          returnRn (SigPatIn pat' ty', fvs1 `plusFV` fvs2)
 
     else addErrRn (patSigErr ty)       `thenRn_`
@@ -92,12 +94,11 @@ rnPat (NPatIn lit)
     lookupOrigName eqClass_RDR         `thenRn` \ eq   ->      -- Needed to find equality on pattern
     returnRn (NPatIn lit', fvs1 `addOneFV` eq)
 
-rnPat (NPlusKPatIn name lit minus)
+rnPat (NPlusKPatIn name lit)
   = rnOverLit lit                      `thenRn` \ (lit', fvs) ->
     lookupOrigName ordClass_RDR                `thenRn` \ ord ->
     lookupBndrRn name                  `thenRn` \ name' ->
-    lookupOccRn minus                  `thenRn` \ minus' ->
-    returnRn (NPlusKPatIn name' lit' minus', fvs `addOneFV` ord `addOneFV` minus')
+    returnRn (NPlusKPatIn name' lit', fvs `addOneFV` ord `addOneFV` minusName)
 
 rnPat (LazyPatIn pat)
   = rnPat pat          `thenRn` \ (pat', fvs) ->
@@ -120,10 +121,10 @@ rnPat (ConOpPatIn pat1 con _ pat2)
 
     getModeRn          `thenRn` \ mode ->
        -- See comments with rnExpr (OpApp ...)
-    (case mode of
-       InterfaceMode -> returnRn (ConOpPatIn pat1' con' defaultFixity pat2')
-       SourceMode    -> lookupFixityRn con'    `thenRn` \ fixity ->
-                        mkConOpPatRn pat1' con' fixity pat2'
+    (if isInterfaceMode mode
+       then returnRn (ConOpPatIn pat1' con' defaultFixity pat2')
+       else lookupFixityRn con'        `thenRn` \ fixity ->
+            mkConOpPatRn pat1' con' fixity pat2'
     )                                                          `thenRn` \ pat' ->
     returnRn (pat', fvs1 `plusFV` fvs2 `addOneFV` con')
 
@@ -146,7 +147,7 @@ rnPat (RecPatIn con rpats)
     rnRpats rpats      `thenRn` \ (rpats', fvs) ->
     returnRn (RecPatIn con' rpats', fvs `addOneFV` con')
 rnPat (TypePatIn name) =
-    (rnHsType (text "type pattern") name) `thenRn` \ (name', fvs) ->
+    (rnHsTypeFVs (text "type pattern") name) `thenRn` \ (name', fvs) ->
     returnRn (TypePatIn name', fvs)
 \end{code}
 
@@ -173,7 +174,7 @@ rnMatch match@(Match _ pats maybe_rhs_sig grhss)
        tyvars_in_pats = extractPatsTyVars pats
        forall_tyvars  = filter (not . (`elemFM` name_env)) tyvars_in_sigs
        doc_sig        = text "a pattern type-signature"
-       doc_pats       = text "in a pattern match"
+       doc_pats       = text "a pattern match"
     in
     bindNakedTyVarsFVRn doc_sig forall_tyvars  $ \ sig_tyvars ->
 
@@ -187,7 +188,7 @@ rnMatch match@(Match _ pats maybe_rhs_sig grhss)
     doptRn Opt_GlasgowExts             `thenRn` \ opt_GlasgowExts ->
     (case maybe_rhs_sig of
        Nothing -> returnRn (Nothing, emptyFVs)
-       Just ty | opt_GlasgowExts -> rnHsType doc_sig ty        `thenRn` \ (ty', ty_fvs) ->
+       Just ty | opt_GlasgowExts -> rnHsTypeFVs doc_sig ty     `thenRn` \ (ty', ty_fvs) ->
                                     returnRn (Just ty', ty_fvs)
                | otherwise       -> addErrRn (patSigErr ty)    `thenRn_`
                                     returnRn (Nothing, emptyFVs)
@@ -228,15 +229,15 @@ rnGRHS (GRHS guarded locn)
                returnRn ()
     )          `thenRn_`
 
-    rnStmts rnExpr guarded     `thenRn` \ (guarded', fvs) ->
+    rnStmts guarded    `thenRn` \ ((_, guarded'), fvs) ->
     returnRn (GRHS guarded' locn, fvs)
   where
        -- Standard Haskell 1.4 guards are just a single boolean
        -- expression, rather than a list of qualifiers as in the
        -- Glasgow extension
-    is_standard_guard [ExprStmt _ _]                = True
-    is_standard_guard [GuardStmt _ _, ExprStmt _ _] = True
-    is_standard_guard other                        = False
+    is_standard_guard [ExprStmt _ _]               = True
+    is_standard_guard [ExprStmt _ _, ExprStmt _ _] = True
+    is_standard_guard other                       = False
 \end{code}
 
 %************************************************************************
@@ -312,20 +313,19 @@ rnExpr (OpApp e1 op _ e2)
        -- that the deriving code generator got the association correct
        -- Don't even look up the fixity when in interface mode
     getModeRn                          `thenRn` \ mode -> 
-    (case mode of
-       SourceMode    -> lookupFixityRn op_name         `thenRn` \ fixity ->
-                        mkOpAppRn e1' op' fixity e2'
-       InterfaceMode -> returnRn (OpApp e1' op' defaultFixity e2')
+    (if isInterfaceMode mode
+       then returnRn (OpApp e1' op' defaultFixity e2')
+       else lookupFixityRn op_name             `thenRn` \ fixity ->
+            mkOpAppRn e1' op' fixity e2'
     )                                  `thenRn` \ final_e -> 
 
     returnRn (final_e,
              fv_e1 `plusFV` fv_op `plusFV` fv_e2)
 
-rnExpr (NegApp e n)
+rnExpr (NegApp e)
   = rnExpr e                   `thenRn` \ (e', fv_e) ->
-    lookupOrigName negate_RDR  `thenRn` \ neg ->
-    mkNegAppRn e' neg          `thenRn` \ final_e ->
-    returnRn (final_e, fv_e `addOneFV` neg)
+    mkNegAppRn e'              `thenRn` \ final_e ->
+    returnRn (final_e, fv_e `addOneFV` negateName)
 
 rnExpr (HsPar e)
   = rnExpr e           `thenRn` \ (e', fvs_e) ->
@@ -375,13 +375,12 @@ rnExpr (HsWith expr binds)
 rnExpr e@(HsDo do_or_lc stmts src_loc)
   = pushSrcLocRn src_loc $
     lookupOrigNames implicit_rdr_names `thenRn` \ implicit_fvs ->
-    rnStmts rnExpr stmts                       `thenRn` \ (stmts', fvs) ->
+    rnStmts stmts                      `thenRn` \ ((_, stmts'), fvs) ->
        -- check the statement list ends in an expression
     case last stmts' of {
        ExprStmt _ _ -> returnRn () ;
-       ReturnStmt _ -> returnRn () ;   -- for list comprehensions
        _            -> addErrRn (doStmtListErr e)
-    }                                          `thenRn_`
+    }                                  `thenRn_`
     returnRn (HsDo do_or_lc stmts' src_loc, fvs `plusFV` implicit_fvs)
   where
     implicit_rdr_names = [foldr_RDR, build_RDR, monadClass_RDR]
@@ -411,8 +410,8 @@ rnExpr (RecordUpd expr rbinds)
     returnRn (RecordUpd expr' rbinds', fvExpr `plusFV` fvRbinds)
 
 rnExpr (ExprWithTySig expr pty)
-  = rnExpr expr                                        `thenRn` \ (expr', fvExpr) ->
-    rnHsSigType (text "an expression") pty     `thenRn` \ (pty', fvTy) ->
+  = rnExpr expr                                                   `thenRn` \ (expr', fvExpr) ->
+    rnHsTypeFVs (text "an expression type signature") pty  `thenRn` \ (pty', fvTy) ->
     returnRn (ExprWithTySig expr' pty', fvExpr `plusFV` fvTy)
 
 rnExpr (HsIf p b1 b2 src_loc)
@@ -422,10 +421,11 @@ rnExpr (HsIf p b1 b2 src_loc)
     rnExpr b2          `thenRn` \ (b2', fvB2) ->
     returnRn (HsIf p' b1' b2' src_loc, plusFVs [fvP, fvB1, fvB2])
 
-rnExpr (HsType a) = 
-    (rnHsType doc a) `thenRn` \ (t, fvT) -> returnRn (HsType t, fvT)
-       where doc = text "renaming a type pattern"
-                   
+rnExpr (HsType a)
+  = rnHsTypeFVs doc a  `thenRn` \ (t, fvT) -> 
+    returnRn (HsType t, fvT)
+  where 
+    doc = text "renaming a type pattern"
 
 rnExpr (ArithSeqIn seq)
   = lookupOrigName enumClass_RDR       `thenRn` \ enum ->
@@ -538,56 +538,64 @@ be @{r}@, and the free var set for the entire Quals will be @{r}@. This
 Quals.
 
 \begin{code}
-type RnExprTy = RdrNameHsExpr -> RnMS (RenamedHsExpr, FreeVars)
+rnStmts :: [RdrNameStmt]
+       -> RnMS (([Name], [RenamedStmt]), FreeVars)
 
-rnStmts :: RnExprTy
-       -> [RdrNameStmt] 
-       -> RnMS ([RenamedStmt], FreeVars)
+rnStmts []
+  = returnRn (([], []), emptyFVs)
 
-rnStmts rn_expr []
-  = returnRn ([], emptyFVs)
+rnStmts (stmt:stmts)
+  = getLocalNameEnv            `thenRn` \ name_env ->
+    rnStmt stmt                                $ \ stmt' ->
+    rnStmts stmts                      `thenRn` \ ((binders, stmts'), fvs) ->
+    returnRn ((binders, stmt' : stmts'), fvs)
 
-rnStmts rn_expr (stmt:stmts)
-  = rnStmt rn_expr stmt                                $ \ stmt' ->
-    rnStmts rn_expr stmts                      `thenRn` \ (stmts', fvs) ->
-    returnRn (stmt' : stmts', fvs)
+rnStmt :: RdrNameStmt
+       -> (RenamedStmt -> RnMS (([Name], a), FreeVars))
+       -> RnMS (([Name], a), FreeVars)
+-- The thing list of names returned is the list returned by the
+-- thing_inside, plus the binders of the arguments stmt
 
-rnStmt :: RnExprTy -> RdrNameStmt
-       -> (RenamedStmt -> RnMS (a, FreeVars))
-       -> RnMS (a, FreeVars)
 -- Because of mutual recursion we have to pass in rnExpr.
 
-rnStmt rn_expr (BindStmt pat expr src_loc) thing_inside
+rnStmt (ParStmt stmtss) thing_inside
+  = mapFvRn rnStmts stmtss             `thenRn` \ (bndrstmtss, fv_stmtss) ->
+    let binderss = map fst bndrstmtss
+       checkBndrs all_bndrs bndrs
+         = checkRn (null (intersectBy eqOcc all_bndrs bndrs)) err `thenRn_`
+           returnRn (bndrs ++ all_bndrs)
+       eqOcc n1 n2 = nameOccName n1 == nameOccName n2
+       err = text "duplicate binding in parallel list comprehension"
+    in
+    foldlRn checkBndrs [] binderss     `thenRn` \ new_binders ->
+    bindLocalNamesFV new_binders       $
+    thing_inside (ParStmtOut bndrstmtss)`thenRn` \ ((rest_bndrs, result), fv_rest) ->
+    returnRn ((new_binders ++ rest_bndrs, result), fv_stmtss `plusFV` fv_rest)
+
+rnStmt (BindStmt pat expr src_loc) thing_inside
   = pushSrcLocRn src_loc $
-    rn_expr expr                                       `thenRn` \ (expr', fv_expr) ->
-    bindLocalsFVRn doc binders                         $ \ new_binders ->
-    rnPat pat                                          `thenRn` \ (pat', fv_pat) ->
-    thing_inside (BindStmt pat' expr' src_loc)         `thenRn` \ (result, fvs) -> 
-    returnRn (result, fv_expr `plusFV` fvs `plusFV` fv_pat)
+    rnExpr expr                                        `thenRn` \ (expr', fv_expr) ->
+    bindLocalsFVRn doc binders                 $ \ new_binders ->
+    rnPat pat                                  `thenRn` \ (pat', fv_pat) ->
+    thing_inside (BindStmt pat' expr' src_loc) `thenRn` \ ((rest_binders, result), fvs) ->
+    -- ZZ is shadowing handled correctly?
+    returnRn ((new_binders ++ rest_binders, result),
+             fv_expr `plusFV` fvs `plusFV` fv_pat)
   where
     binders = collectPatBinders pat
     doc = text "a pattern in do binding" 
 
-rnStmt rn_expr (ExprStmt expr src_loc) thing_inside
+rnStmt (ExprStmt expr src_loc) thing_inside
   = pushSrcLocRn src_loc $
-    rn_expr expr                               `thenRn` \ (expr', fv_expr) ->
+    rnExpr expr                                `thenRn` \ (expr', fv_expr) ->
     thing_inside (ExprStmt expr' src_loc)      `thenRn` \ (result, fvs) ->
     returnRn (result, fv_expr `plusFV` fvs)
 
-rnStmt rn_expr (GuardStmt expr src_loc) thing_inside
-  = pushSrcLocRn src_loc $
-    rn_expr expr                               `thenRn` \ (expr', fv_expr) ->
-    thing_inside (GuardStmt expr' src_loc)     `thenRn` \ (result, fvs) ->
-    returnRn (result, fv_expr `plusFV` fvs)
-
-rnStmt rn_expr (ReturnStmt expr) thing_inside
-  = rn_expr expr                               `thenRn` \ (expr', fv_expr) ->
-    thing_inside (ReturnStmt expr')            `thenRn` \ (result, fvs) ->
-    returnRn (result, fv_expr `plusFV` fvs)
-
-rnStmt rn_expr (LetStmt binds) thing_inside
-  = rnBinds binds              $ \ binds' ->
-    thing_inside (LetStmt binds')
+rnStmt (LetStmt binds) thing_inside
+  = rnBinds binds                              $ \ binds' ->
+    let new_binders = collectHsBinders binds' in
+    thing_inside (LetStmt binds')    `thenRn` \ ((rest_binders, result), fvs) ->
+    returnRn ((new_binders ++ rest_binders, result), fvs )
 \end{code}
 
 %************************************************************************
@@ -628,20 +636,20 @@ mkOpAppRn e1@(OpApp e11 op1 fix1 e12) op2 fix2 e2
 
 ---------------------------
 --     (- neg_arg) `op` e2
-mkOpAppRn e1@(NegApp neg_arg neg_op) op2 fix2 e2
+mkOpAppRn e1@(NegApp neg_arg) op2 fix2 e2
   | nofix_error
   = addErrRn (precParseErr (pp_prefix_minus,negateFixity) (ppr_op op2,fix2))   `thenRn_`
     returnRn (OpApp e1 op2 fix2 e2)
 
   | associate_right
   = mkOpAppRn neg_arg op2 fix2 e2      `thenRn` \ new_e ->
-    returnRn (NegApp new_e neg_op)
+    returnRn (NegApp new_e)
   where
     (nofix_error, associate_right) = compareFixity negateFixity fix2
 
 ---------------------------
 --     e1 `op` - neg_arg
-mkOpAppRn e1 op1 fix1 e2@(NegApp neg_arg neg_op)       -- NegApp can occur on the right
+mkOpAppRn e1 op1 fix1 e2@(NegApp neg_arg)      -- NegApp can occur on the right
   | not associate_right                                        -- We *want* right association
   = addErrRn (precParseErr (ppr_op op1, fix1) (pp_prefix_minus, negateFixity)) `thenRn_`
     returnRn (OpApp e1 op1 fix1 e2)
@@ -667,13 +675,13 @@ right_op_ok fix1 other
   = True
 
 -- Parser initially makes negation bind more tightly than any other operator
-mkNegAppRn neg_arg neg_op
+mkNegAppRn neg_arg
   = 
 #ifdef DEBUG
     getModeRn                  `thenRn` \ mode ->
     ASSERT( not_op_app mode neg_arg )
 #endif
-    returnRn (NegApp neg_arg neg_op)
+    returnRn (NegApp neg_arg)
 
 not_op_app SourceMode (OpApp _ _ _ _) = False
 not_op_app mode other                = True
@@ -714,10 +722,10 @@ checkPrecMatch True op (Match _ (p1:p2:_) _ _)
        -- True indicates an infix lhs
   = getModeRn          `thenRn` \ mode ->
        -- See comments with rnExpr (OpApp ...)
-    case mode of
-       InterfaceMode -> returnRn ()
-       SourceMode    -> checkPrec op p1 False  `thenRn_`
-                        checkPrec op p2 True
+    if isInterfaceMode mode
+       then returnRn ()
+       else checkPrec op p1 False      `thenRn_`
+            checkPrec op p2 True
 
 checkPrecMatch True op _ = panic "checkPrecMatch"
 
@@ -745,7 +753,7 @@ checkPrec op pat right
 checkSectionPrec left_or_right section op arg
   = case arg of
        OpApp _ op fix _ -> go_for_it (ppr_op op)     fix
-       NegApp _ _       -> go_for_it pp_prefix_minus negateFixity
+       NegApp _         -> go_for_it pp_prefix_minus negateFixity
        other            -> returnRn ()
   where
     HsVar op_name = op
@@ -791,7 +799,10 @@ that the types and classes they involve
 are made available.
 
 \begin{code}
-litFVs (HsChar c)             = returnRn (unitFV charTyCon_name)
+litFVs (HsChar c)
+   = checkRn (inCharRange c) (bogusCharError c) `thenRn_`
+     returnRn (unitFV charTyCon_name)
+
 litFVs (HsCharPrim c)         = returnRn (unitFV (getName charPrimTyCon))
 litFVs (HsString s)           = returnRn (mkFVs [listTyCon_name, charTyCon_name])
 litFVs (HsStringPrim s)       = returnRn (unitFV (getName addrPrimTyCon))
@@ -804,18 +815,20 @@ litFVs (HsLitLit l bogus_ty)  = lookupOrigName cCallableClass_RDR `thenRn` \ cc
 litFVs lit                   = pprPanic "RnExpr.litFVs" (ppr lit)      -- HsInteger and HsRat only appear 
                                                                        -- in post-typechecker translations
 
-rnOverLit (HsIntegral i from_integer)
-  = lookupOccRn from_integer           `thenRn` \ from_integer' ->
-    (if inIntRange i then
-       returnRn emptyFVs
-     else
-       lookupOrigNames [plusInteger_RDR, timesInteger_RDR]
-    )                                  `thenRn` \ ns ->
-    returnRn (HsIntegral i from_integer', ns `addOneFV` from_integer')
-
-rnOverLit (HsFractional i n)
-  = lookupOccRn n                                                         `thenRn` \ n' ->
-    lookupOrigNames [ratioDataCon_RDR, plusInteger_RDR, timesInteger_RDR]  `thenRn` \ ns' ->
+rnOverLit (HsIntegral i)
+  | inIntRange i
+  = returnRn (HsIntegral i, unitFV fromIntegerName)
+  | otherwise
+  = lookupOrigNames [fromInteger_RDR, plusInteger_RDR, timesInteger_RDR]       `thenRn` \ ns ->
+       -- Big integers are built, using + and *, out of small integers
+       -- [No particular reason why we use fromIntegerName in one case can 
+       --  fromInteger_RDR in the other; but plusInteger_RDR means we 
+       --  can get away without plusIntegerName altogether.]
+    returnRn (HsIntegral i, ns)
+
+rnOverLit (HsFractional i)
+  = lookupOrigNames [fromRational_RDR, ratioDataCon_RDR, 
+                    plusInteger_RDR, timesInteger_RDR]  `thenRn` \ ns ->
        -- We have to make sure that the Ratio type is imported with
        -- its constructor, because literals of type Ratio t are
        -- built with that constructor.
@@ -823,7 +836,7 @@ rnOverLit (HsFractional i n)
        -- when fractionalClass does.
        -- The plus/times integer operations may be needed to construct the numerator
        -- and denominator (see DsUtils.mkIntegerLit)
-    returnRn (HsFractional i n', ns' `addOneFV` n')
+    returnRn (HsFractional i, ns)
 \end{code}
 
 %************************************************************************
@@ -906,4 +919,7 @@ patSynErr e
 doStmtListErr e
   = sep [ptext SLIT("`do' statements must end in expression:"),
         nest 4 (ppr e)]
+
+bogusCharError c
+  = ptext SLIT("character literal out of range: '\\") <> int c <> char '\''
 \end{code}