X-Git-Url: http://git.megacz.com/?a=blobdiff_plain;f=ghc%2Fcompiler%2Ftypecheck%2FTcGenDeriv.lhs;h=758dbaa6bf635647faef9ede0e487487178fbcef;hb=f43ebad1020dccdf6fde2fddc91994b27d0f565e;hp=743851770d5793b8d22e286f4b6030f5197dcec5;hpb=e7498a3ee1d0484d02a9e86633cc179c76ebf36e;p=ghc-hetmet.git diff --git a/ghc/compiler/typecheck/TcGenDeriv.lhs b/ghc/compiler/typecheck/TcGenDeriv.lhs index 7438517..758dbaa 100644 --- a/ghc/compiler/typecheck/TcGenDeriv.lhs +++ b/ghc/compiler/typecheck/TcGenDeriv.lhs @@ -1,5 +1,5 @@ % -% (c) The GRASP/AQUA Project, Glasgow University, 1992-1996 +% (c) The GRASP/AQUA Project, Glasgow University, 1992-1998 % \section[TcGenDeriv]{Generating derived instance declarations} @@ -9,87 +9,65 @@ This module is nominally ``subordinate'' to @TcDeriv@, which is the This is where we do all the grimy bindings' generation. \begin{code} -#include "HsVersions.h" - module TcGenDeriv ( - a_Expr, - a_PN, - a_Pat, - ah_PN, - b_Expr, - b_PN, - b_Pat, - bh_PN, - c_Expr, - c_PN, - c_Pat, - ch_PN, - cmp_eq_PN, - d_Expr, - d_PN, - d_Pat, - dh_PN, - eqH_Int_PN, - eqTag_Expr, - eq_PN, - error_PN, - false_Expr, - false_PN, - geH_PN, gen_Bounded_binds, gen_Enum_binds, - gen_Eval_binds, gen_Eq_binds, gen_Ix_binds, gen_Ord_binds, gen_Read_binds, gen_Show_binds, gen_tag_n_con_monobind, - gtTag_Expr, - gt_PN, - leH_PN, - ltH_Int_PN, - ltTag_Expr, - lt_PN, - minusH_PN, - mkInt_PN, - rangeSize_PN, - true_Expr, - true_PN, - - con2tag_PN, tag2con_PN, maxtag_PN, + + con2tag_RDR, tag2con_RDR, maxtag_RDR, TagThingWanted(..) ) where -IMP_Ubiq() - -import HsSyn ( HsBinds(..), Bind(..), MonoBinds(..), Match(..), GRHSsAndBinds(..), - GRHS(..), HsExpr(..), HsLit(..), InPat(..), Qual(..), Stmt, - ArithSeqInfo, Sig, PolyType, FixityDecl, Fake ) -import RdrHsSyn ( RdrNameMonoBinds(..), RdrNameHsExpr(..), RdrNamePat(..) ) -import RnHsSyn ( RenamedFixityDecl(..) ) ---import RnUtils - -import Id ( GenId, dataConArity, isNullaryDataCon, dataConTag, - dataConRawArgTys, fIRST_TAG, - isDataCon, DataCon(..), ConTag(..) ) -import IdUtils ( primOpId ) -import Maybes ( maybeToBool ) -import Name ( moduleNamePair, origName, RdrName(..) ) -import PrelMods ( fromPrelude, pRELUDE, pRELUDE_BUILTIN, pRELUDE_LIST, pRELUDE_TEXT ) -import PrelVals ( eRROR_ID ) - -import PrimOp ( PrimOp(..) ) -import SrcLoc ( mkGeneratedSrcLoc ) -import TyCon ( TyCon, tyConDataCons, isEnumerationTyCon, maybeTyConSingleCon ) -import Type ( eqTy, isPrimType ) +#include "HsVersions.h" + +import HsSyn ( InPat(..), HsExpr(..), MonoBinds(..), + Match(..), GRHSs(..), Stmt(..), HsLit(..), + HsBinds(..), StmtCtxt(..), HsType(..), + unguardedRHS, mkSimpleMatch, mkMonoBind, andMonoBindList + ) +import RdrHsSyn ( mkHsOpApp, RdrNameMonoBinds, RdrNameHsExpr, RdrNamePat ) +import RdrName ( RdrName, mkUnqual ) +import BasicTypes ( RecFlag(..), Fixity(..), FixityDirection(..) + , maxPrecedence, defaultFixity + , Boxity(..) + ) +import FieldLabel ( fieldLabelName ) +import DataCon ( isNullaryDataCon, dataConTag, + dataConOrigArgTys, dataConSourceArity, fIRST_TAG, + DataCon, + dataConFieldLabels ) +import Name ( getOccString, getOccName, getSrcLoc, occNameString, + occNameUserString, nameRdrName, varName, + Name, NamedThing(..), + isDataSymOcc, isSymOcc + ) + +import PrelInfo -- Lots of RdrNames +import SrcLoc ( generatedSrcLoc, SrcLoc ) +import TyCon ( TyCon, isNewTyCon, tyConDataCons, isEnumerationTyCon, + maybeTyConSingleCon, tyConFamilySize + ) +import Type ( isUnLiftedType, isUnboxedType, Type ) import TysPrim ( charPrimTy, intPrimTy, wordPrimTy, addrPrimTy, floatPrimTy, doublePrimTy ) -import TysWiredIn ( falseDataCon, trueDataCon, intDataCon ) ---import Unique -import Util ( mapAccumL, zipEqual, zipWith3Equal, nOfThem, panic, assertPanic ) +import Util ( mapAccumL, zipEqual, zipWithEqual, + zipWith3Equal, nOfThem ) +import Panic ( panic, assertPanic ) +import Maybes ( maybeToBool, orElse ) +import Constants +import List ( partition, intersperse ) +import Outputable ( pprPanic, ppr, pprTrace ) + +#if __GLASGOW_HASKELL__ >= 404 +import GlaExts ( fromInt ) +#endif \end{code} %************************************************************************ @@ -171,60 +149,68 @@ instance ... Eq (Foo ...) where produced don't get through the typechecker. \end{itemize} + +deriveEq :: RdrName -- Class + -> RdrName -- Type constructor + -> [ (RdrName, [RdrType]) ] -- Constructors + -> (RdrContext, -- Context for the inst decl + [RdrBind], -- Binds in the inst decl + [RdrBind]) -- Extra value bindings outside + +deriveEq clas tycon constrs + = (context, [eq_bind, ne_bind], []) + where + context = [(clas, [ty]) | (_, tys) <- constrs, ty <- tys] + + ne_bind = mkBind + (nullary_cons, non_nullary_cons) = partition is_nullary constrs + is_nullary (_, args) = null args + \begin{code} gen_Eq_binds :: TyCon -> RdrNameMonoBinds gen_Eq_binds tycon = let - (nullary_cons, nonnullary_cons) - = partition isNullaryDataCon (tyConDataCons tycon) + tycon_loc = getSrcLoc tycon + (nullary_cons, nonnullary_cons) + | isNewTyCon tycon = ([], tyConDataCons tycon) + | otherwise = partition isNullaryDataCon (tyConDataCons tycon) rest = if (null nullary_cons) then case maybeTyConSingleCon tycon of Just _ -> [] Nothing -> -- if cons don't match, then False - [([a_Pat, b_Pat], false_Expr)] + [([wildPat, wildPat], false_Expr)] else -- calc. and compare the tags [([a_Pat, b_Pat], - untag_Expr tycon [(a_PN,ah_PN), (b_PN,bh_PN)] - (cmp_tags_Expr eqH_Int_PN ah_PN bh_PN true_Expr false_Expr))] + untag_Expr tycon [(a_RDR,ah_RDR), (b_RDR,bh_RDR)] + (cmp_tags_Expr eqH_Int_RDR ah_RDR bh_RDR true_Expr false_Expr))] in - mk_FunMonoBind eq_PN ((map pats_etc nonnullary_cons) ++ rest) - `AndMonoBinds` boring_ne_method + mk_FunMonoBind tycon_loc eq_RDR ((map pats_etc nonnullary_cons) ++ rest) + `AndMonoBinds` + mk_easy_FunMonoBind tycon_loc ne_RDR [a_Pat, b_Pat] [] ( + HsApp (HsVar not_RDR) (HsPar (mk_easy_App eq_RDR [a_RDR, b_RDR]))) where ------------------------------------------------------------------ pats_etc data_con = let - con1_pat = ConPatIn data_con_PN (map VarPatIn as_needed) - con2_pat = ConPatIn data_con_PN (map VarPatIn bs_needed) - - data_con_PN = origName data_con - con_arity = dataConArity data_con - as_needed = take con_arity as_PNs - bs_needed = take con_arity bs_PNs - tys_needed = dataConRawArgTys data_con + con1_pat = ConPatIn data_con_RDR (map VarPatIn as_needed) + con2_pat = ConPatIn data_con_RDR (map VarPatIn bs_needed) + + data_con_RDR = qual_orig_name data_con + con_arity = length tys_needed + as_needed = take con_arity as_RDRs + bs_needed = take con_arity bs_RDRs + tys_needed = dataConOrigArgTys data_con in ([con1_pat, con2_pat], nested_eq_expr tys_needed as_needed bs_needed) where nested_eq_expr [] [] [] = true_Expr nested_eq_expr tys as bs - = foldr1 and_Expr (zipWith3Equal "nested_eq" nested_eq tys as bs) + = foldl1 and_Expr (zipWith3Equal "nested_eq" nested_eq tys as bs) where nested_eq ty a b = HsPar (eq_Expr ty (HsVar a) (HsVar b)) -{-OLD: - nested_eq_expr [] [] [] = true_Expr - nested_eq_expr [ty] [a] [b] = - nested_eq_expr (t:ts) (a:as) (b:bs) - = let - rest_expr = nested_eq_expr ts as bs - in - and_Expr (eq_Expr t (HsVar a) (HsVar b)) rest_expr --} - -boring_ne_method - = mk_easy_FunMonoBind ne_PN [a_Pat, b_Pat] [] $ - HsApp (HsVar not_PN) (HsPar (mk_easy_App eq_PN [a_PN, b_PN])) \end{code} %************************************************************************ @@ -305,6 +291,7 @@ cmp_eq (O3 a1 b1 c1) (O3 a2 b2 c2) Again, we must be careful about unboxed comparisons. For example, if \tr{a1} and \tr{a2} were \tr{Int#}s in the 2nd example above, we'd need to generate: + \begin{verbatim} cmp_eq lt eq gt (O2 a1) (O2 a2) = compareInt# a1 a2 @@ -319,51 +306,81 @@ cmp_eq _ _ = EQ \end{verbatim} \end{itemize} +If there is only one constructor in the Data Type we don't need the WildCard Pattern. +JJQC-30-Nov-1997 + \begin{code} gen_Ord_binds :: TyCon -> RdrNameMonoBinds gen_Ord_binds tycon - = defaulted `AndMonoBinds` compare + = compare -- `AndMonoBinds` compare + -- The default declaration in PrelBase handles this where + tycon_loc = getSrcLoc tycon -------------------------------------------------------------------- - compare = mk_easy_FunMonoBind compare_PN + compare = mk_easy_FunMonoBind tycon_loc compare_RDR [a_Pat, b_Pat] [cmp_eq] (if maybeToBool (maybeTyConSingleCon tycon) then - cmp_eq_Expr ltTag_Expr eqTag_Expr gtTag_Expr a_Expr b_Expr + +-- cmp_eq_Expr ltTag_Expr eqTag_Expr gtTag_Expr a_Expr b_Expr +-- Wierd. Was: case (cmp a b) of { LT -> LT; EQ -> EQ; GT -> GT } + + cmp_eq_Expr a_Expr b_Expr else - untag_Expr tycon [(a_PN, ah_PN), (b_PN, bh_PN)] - (cmp_tags_Expr eqH_Int_PN ah_PN bh_PN + untag_Expr tycon [(a_RDR, ah_RDR), (b_RDR, bh_RDR)] + (cmp_tags_Expr eqH_Int_RDR ah_RDR bh_RDR -- True case; they are equal -- If an enumeration type we are done; else -- recursively compare their components (if isEnumerationTyCon tycon then eqTag_Expr else - cmp_eq_Expr ltTag_Expr eqTag_Expr gtTag_Expr a_Expr b_Expr +-- cmp_eq_Expr ltTag_Expr eqTag_Expr gtTag_Expr a_Expr b_Expr +-- Ditto + cmp_eq_Expr a_Expr b_Expr ) -- False case; they aren't equal -- So we need to do a less-than comparison on the tags - (cmp_tags_Expr ltH_Int_PN ah_PN bh_PN ltTag_Expr gtTag_Expr))) + (cmp_tags_Expr ltH_Int_RDR ah_RDR bh_RDR ltTag_Expr gtTag_Expr))) + tycon_data_cons = tyConDataCons tycon (nullary_cons, nonnullary_cons) - = partition (\ con -> dataConArity con == 0) (tyConDataCons tycon) - - cmp_eq - = mk_FunMonoBind cmp_eq_PN (map pats_etc nonnullary_cons ++ deflt_pats_etc) + | isNewTyCon tycon = ([], tyConDataCons tycon) + | otherwise = partition isNullaryDataCon tycon_data_cons + + cmp_eq = + mk_FunMonoBind tycon_loc + cmp_eq_RDR + (if null nonnullary_cons && (length nullary_cons == 1) then + -- catch this specially to avoid warnings + -- about overlapping patterns from the desugarer. + let + data_con = head nullary_cons + data_con_RDR = qual_orig_name data_con + pat = ConPatIn data_con_RDR [] + in + [([pat,pat], eqTag_Expr)] + else + map pats_etc nonnullary_cons ++ + -- leave out wildcards to silence desugarer. + (if length tycon_data_cons == 1 then + [] + else + [([WildPatIn, WildPatIn], default_rhs)])) where pats_etc data_con = ([con1_pat, con2_pat], nested_compare_expr tys_needed as_needed bs_needed) where - con1_pat = ConPatIn data_con_PN (map VarPatIn as_needed) - con2_pat = ConPatIn data_con_PN (map VarPatIn bs_needed) + con1_pat = ConPatIn data_con_RDR (map VarPatIn as_needed) + con2_pat = ConPatIn data_con_RDR (map VarPatIn bs_needed) - data_con_PN = origName data_con - con_arity = dataConArity data_con - as_needed = take con_arity as_PNs - bs_needed = take con_arity bs_PNs - tys_needed = dataConRawArgTys data_con + data_con_RDR = qual_orig_name data_con + con_arity = length tys_needed + as_needed = take con_arity as_RDRs + bs_needed = take con_arity bs_RDRs + tys_needed = dataConOrigArgTys data_con nested_compare_expr [ty] [a] [b] = careful_compare_Case ty ltTag_Expr eqTag_Expr gtTag_Expr (HsVar a) (HsVar b) @@ -372,27 +389,30 @@ gen_Ord_binds tycon = let eq_expr = nested_compare_expr tys as bs in careful_compare_Case ty ltTag_Expr eq_expr gtTag_Expr (HsVar a) (HsVar b) - deflt_pats_etc - = if null nullary_cons - then [] - else [([a_Pat, b_Pat], eqTag_Expr)] + default_rhs | null nullary_cons = impossible_Expr -- Keep desugarer from complaining about + -- inexhaustive patterns + | otherwise = eqTag_Expr -- Some nullary constructors; + -- Tags are equal, no args => return EQ -------------------------------------------------------------------- +{- Not necessary: the default decls in PrelBase handle these + defaulted = foldr1 AndMonoBinds [lt, le, ge, gt, max_, min_] -lt = mk_easy_FunMonoBind lt_PN [a_Pat, b_Pat] [] ( +lt = mk_easy_FunMonoBind generatedSrcLoc lt_RDR [a_Pat, b_Pat] [] ( compare_Case true_Expr false_Expr false_Expr a_Expr b_Expr) -le = mk_easy_FunMonoBind le_PN [a_Pat, b_Pat] [] ( +le = mk_easy_FunMonoBind generatedSrcLoc le_RDR [a_Pat, b_Pat] [] ( compare_Case true_Expr true_Expr false_Expr a_Expr b_Expr) -ge = mk_easy_FunMonoBind ge_PN [a_Pat, b_Pat] [] ( +ge = mk_easy_FunMonoBind generatedSrcLoc ge_RDR [a_Pat, b_Pat] [] ( compare_Case false_Expr true_Expr true_Expr a_Expr b_Expr) -gt = mk_easy_FunMonoBind gt_PN [a_Pat, b_Pat] [] ( +gt = mk_easy_FunMonoBind generatedSrcLoc gt_RDR [a_Pat, b_Pat] [] ( compare_Case false_Expr false_Expr true_Expr a_Expr b_Expr) -max_ = mk_easy_FunMonoBind max_PN [a_Pat, b_Pat] [] ( +max_ = mk_easy_FunMonoBind generatedSrcLoc max_RDR [a_Pat, b_Pat] [] ( compare_Case b_Expr a_Expr a_Expr a_Expr b_Expr) -min_ = mk_easy_FunMonoBind min_PN [a_Pat, b_Pat] [] ( +min_ = mk_easy_FunMonoBind generatedSrcLoc min_RDR [a_Pat, b_Pat] [] ( compare_Case a_Expr b_Expr b_Expr a_Expr b_Expr) +-} \end{code} %************************************************************************ @@ -411,6 +431,11 @@ we use both @con2tag_Foo@ and @tag2con_Foo@ functions, as well as a \begin{verbatim} instance ... Enum (Foo ...) where + succ x = toEnum (1 + fromEnum x) + pred x = toEnum (fromEnum x - 1) + + toEnum i = tag2con_Foo i + enumFrom a = map tag2con_Foo [con2tag_Foo a .. maxtag_Foo] -- or, really... @@ -435,34 +460,81 @@ For @enumFromTo@ and @enumFromThenTo@, we use the default methods. gen_Enum_binds :: TyCon -> RdrNameMonoBinds gen_Enum_binds tycon - = enum_from `AndMonoBinds` enum_from_then + = succ_enum `AndMonoBinds` + pred_enum `AndMonoBinds` + to_enum `AndMonoBinds` + enum_from `AndMonoBinds` + enum_from_then `AndMonoBinds` + from_enum where + tycon_loc = getSrcLoc tycon + occ_nm = getOccString tycon + + succ_enum + = mk_easy_FunMonoBind tycon_loc succ_RDR [a_Pat] [] $ + untag_Expr tycon [(a_RDR, ah_RDR)] $ + HsIf (HsApp (HsApp (HsVar eq_RDR) + (HsVar (maxtag_RDR tycon))) + (mk_easy_App mkInt_RDR [ah_RDR])) + (illegal_Expr "succ" occ_nm "tried to take `succ' of last tag in enumeration") + (HsApp (HsVar (tag2con_RDR tycon)) + (HsApp (HsApp (HsVar plus_RDR) + (mk_easy_App mkInt_RDR [ah_RDR])) + (HsLit (HsInt 1)))) + tycon_loc + + pred_enum + = mk_easy_FunMonoBind tycon_loc pred_RDR [a_Pat] [] $ + untag_Expr tycon [(a_RDR, ah_RDR)] $ + HsIf (HsApp (HsApp (HsVar eq_RDR) (HsLit (HsInt 0))) + (mk_easy_App mkInt_RDR [ah_RDR])) + (illegal_Expr "pred" occ_nm "tried to take `pred' of first tag in enumeration") + (HsApp (HsVar (tag2con_RDR tycon)) + (HsApp (HsApp (HsVar plus_RDR) + (mk_easy_App mkInt_RDR [ah_RDR])) + (HsLit (HsInt (-1))))) + tycon_loc + + to_enum + = mk_easy_FunMonoBind tycon_loc toEnum_RDR [a_Pat] [] $ + HsIf (HsApp (HsApp + (HsVar and_RDR) + (HsApp (HsApp (HsVar ge_RDR) + (HsVar a_RDR)) + (HsLit (HsInt 0)))) + (HsApp (HsApp (HsVar le_RDR) + (HsVar a_RDR)) + (HsVar (maxtag_RDR tycon)))) + (mk_easy_App (tag2con_RDR tycon) [a_RDR]) + (illegal_toEnum_tag occ_nm (maxtag_RDR tycon)) + tycon_loc + enum_from - = mk_easy_FunMonoBind enumFrom_PN [a_Pat] [] $ - untag_Expr tycon [(a_PN, ah_PN)] $ - HsApp (mk_easy_App map_PN [tag2con_PN tycon]) $ + = mk_easy_FunMonoBind tycon_loc enumFrom_RDR [a_Pat] [] $ + untag_Expr tycon [(a_RDR, ah_RDR)] $ + HsApp (mk_easy_App map_RDR [tag2con_RDR tycon]) $ HsPar (enum_from_to_Expr - (mk_easy_App mkInt_PN [ah_PN]) - (HsVar (maxtag_PN tycon))) + (mk_easy_App mkInt_RDR [ah_RDR]) + (HsVar (maxtag_RDR tycon))) enum_from_then - = mk_easy_FunMonoBind enumFromThen_PN [a_Pat, b_Pat] [] $ - untag_Expr tycon [(a_PN, ah_PN), (b_PN, bh_PN)] $ - HsApp (mk_easy_App map_PN [tag2con_PN tycon]) $ + = mk_easy_FunMonoBind tycon_loc enumFromThen_RDR [a_Pat, b_Pat] [] $ + untag_Expr tycon [(a_RDR, ah_RDR), (b_RDR, bh_RDR)] $ + HsApp (mk_easy_App map_RDR [tag2con_RDR tycon]) $ HsPar (enum_from_then_to_Expr - (mk_easy_App mkInt_PN [ah_PN]) - (mk_easy_App mkInt_PN [bh_PN]) - (HsVar (maxtag_PN tycon))) -\end{code} - -%************************************************************************ -%* * -\subsubsection{Generating @Eval@ instance declarations} -%* * -%************************************************************************ - -\begin{code} -gen_Eval_binds tycon = EmptyMonoBinds + (mk_easy_App mkInt_RDR [ah_RDR]) + (mk_easy_App mkInt_RDR [bh_RDR]) + (HsIf (HsApp (HsApp (HsVar gt_RDR) + (mk_easy_App mkInt_RDR [ah_RDR])) + (mk_easy_App mkInt_RDR [bh_RDR])) + (HsLit (HsInt 0)) + (HsVar (maxtag_RDR tycon)) + tycon_loc)) + + from_enum + = mk_easy_FunMonoBind tycon_loc fromEnum_RDR [a_Pat] [] $ + untag_Expr tycon [(a_RDR, ah_RDR)] $ + (mk_easy_App mkInt_RDR [ah_RDR]) \end{code} %************************************************************************ @@ -479,24 +551,25 @@ gen_Bounded_binds tycon ASSERT(length data_cons == 1) min_bound_1con `AndMonoBinds` max_bound_1con where - data_cons = tyConDataCons tycon + data_cons = tyConDataCons tycon + tycon_loc = getSrcLoc tycon ----- enum-flavored: --------------------------- - min_bound_enum = mk_easy_FunMonoBind minBound_PN [] [] (HsVar data_con_1_PN) - max_bound_enum = mk_easy_FunMonoBind maxBound_PN [] [] (HsVar data_con_N_PN) + min_bound_enum = mk_easy_FunMonoBind tycon_loc minBound_RDR [] [] (HsVar data_con_1_RDR) + max_bound_enum = mk_easy_FunMonoBind tycon_loc maxBound_RDR [] [] (HsVar data_con_N_RDR) data_con_1 = head data_cons data_con_N = last data_cons - data_con_1_PN = origName data_con_1 - data_con_N_PN = origName data_con_N + data_con_1_RDR = qual_orig_name data_con_1 + data_con_N_RDR = qual_orig_name data_con_N ----- single-constructor-flavored: ------------- - arity = dataConArity data_con_1 + arity = dataConSourceArity data_con_1 - min_bound_1con = mk_easy_FunMonoBind minBound_PN [] [] $ - mk_easy_App data_con_1_PN (nOfThem arity minBound_PN) - max_bound_1con = mk_easy_FunMonoBind maxBound_PN [] [] $ - mk_easy_App data_con_1_PN (nOfThem arity maxBound_PN) + min_bound_1con = mk_easy_FunMonoBind tycon_loc minBound_RDR [] [] $ + mk_easy_App data_con_1_RDR (nOfThem arity minBound_RDR) + max_bound_1con = mk_easy_FunMonoBind tycon_loc maxBound_RDR [] [] $ + mk_easy_App data_con_1_RDR (nOfThem arity maxBound_RDR) \end{code} %************************************************************************ @@ -565,107 +638,131 @@ gen_Ix_binds tycon then enum_ixes else single_con_ixes where - tycon_str = _UNPK_ (snd (moduleNamePair tycon)) + tycon_str = getOccString tycon + tycon_loc = getSrcLoc tycon -------------------------------------------------------------- enum_ixes = enum_range `AndMonoBinds` enum_index `AndMonoBinds` enum_inRange enum_range - = mk_easy_FunMonoBind range_PN [TuplePatIn [a_Pat, b_Pat]] [] $ - untag_Expr tycon [(a_PN, ah_PN)] $ - untag_Expr tycon [(b_PN, bh_PN)] $ - HsApp (mk_easy_App map_PN [tag2con_PN tycon]) $ + = mk_easy_FunMonoBind tycon_loc range_RDR + [TuplePatIn [a_Pat, b_Pat] Boxed] [] $ + untag_Expr tycon [(a_RDR, ah_RDR)] $ + untag_Expr tycon [(b_RDR, bh_RDR)] $ + HsApp (mk_easy_App map_RDR [tag2con_RDR tycon]) $ HsPar (enum_from_to_Expr - (mk_easy_App mkInt_PN [ah_PN]) - (mk_easy_App mkInt_PN [bh_PN])) + (mk_easy_App mkInt_RDR [ah_RDR]) + (mk_easy_App mkInt_RDR [bh_RDR])) enum_index - = mk_easy_FunMonoBind index_PN [AsPatIn c_PN (TuplePatIn [a_Pat, b_Pat]), d_Pat] [] ( - HsIf (HsPar (mk_easy_App inRange_PN [c_PN, d_PN])) ( - untag_Expr tycon [(a_PN, ah_PN)] ( - untag_Expr tycon [(d_PN, dh_PN)] ( + = mk_easy_FunMonoBind tycon_loc index_RDR + [AsPatIn c_RDR (TuplePatIn [a_Pat, wildPat] Boxed), + d_Pat] [] ( + HsIf (HsPar (mk_easy_App inRange_RDR [c_RDR, d_RDR])) ( + untag_Expr tycon [(a_RDR, ah_RDR)] ( + untag_Expr tycon [(d_RDR, dh_RDR)] ( let - grhs = [OtherwiseGRHS (mk_easy_App mkInt_PN [c_PN]) mkGeneratedSrcLoc] + rhs = mk_easy_App mkInt_RDR [c_RDR] in HsCase - (HsPar (OpApp (HsVar dh_PN) (HsVar minusH_PN) (HsVar ah_PN))) - [PatMatch (VarPatIn c_PN) - (GRHSMatch (GRHSsAndBindsIn grhs EmptyBinds))] - mkGeneratedSrcLoc + (genOpApp (HsVar dh_RDR) minusH_RDR (HsVar ah_RDR)) + [mkSimpleMatch [VarPatIn c_RDR] rhs Nothing tycon_loc] + tycon_loc )) ) {-else-} ( - HsApp (HsVar error_PN) (HsLit (HsString (_PK_ ("Ix."++tycon_str++".index: out of range\n")))) + HsApp (HsVar error_RDR) (HsLit (HsString (_PK_ ("Ix."++tycon_str++".index: out of range\n")))) ) - mkGeneratedSrcLoc) + tycon_loc) enum_inRange - = mk_easy_FunMonoBind inRange_PN [TuplePatIn [a_Pat, b_Pat], c_Pat] [] ( - untag_Expr tycon [(a_PN, ah_PN)] ( - untag_Expr tycon [(b_PN, bh_PN)] ( - untag_Expr tycon [(c_PN, ch_PN)] ( - HsIf (HsPar (OpApp (HsVar ch_PN) (HsVar geH_PN) (HsVar ah_PN))) ( - (OpApp (HsVar ch_PN) (HsVar leH_PN) (HsVar bh_PN)) + = mk_easy_FunMonoBind tycon_loc inRange_RDR + [TuplePatIn [a_Pat, b_Pat] Boxed, c_Pat] [] ( + untag_Expr tycon [(a_RDR, ah_RDR)] ( + untag_Expr tycon [(b_RDR, bh_RDR)] ( + untag_Expr tycon [(c_RDR, ch_RDR)] ( + HsIf (genOpApp (HsVar ch_RDR) geH_RDR (HsVar ah_RDR)) ( + (genOpApp (HsVar ch_RDR) leH_RDR (HsVar bh_RDR)) ) {-else-} ( false_Expr - ) mkGeneratedSrcLoc)))) + ) tycon_loc)))) -------------------------------------------------------------- - single_con_ixes = single_con_range `AndMonoBinds` - single_con_index `AndMonoBinds` single_con_inRange + single_con_ixes + = single_con_range `AndMonoBinds` + single_con_index `AndMonoBinds` + single_con_inRange data_con = case maybeTyConSingleCon tycon of -- just checking... Nothing -> panic "get_Ix_binds" - Just dc -> if (any isPrimType (dataConRawArgTys dc)) then + Just dc -> if (any isUnLiftedType (dataConOrigArgTys dc)) then error ("ERROR: Can't derive Ix for a single-constructor type with primitive argument types: "++tycon_str) else dc - con_arity = dataConArity data_con - data_con_PN = origName data_con - con_pat xs = ConPatIn data_con_PN (map VarPatIn xs) - con_expr xs = mk_easy_App data_con_PN xs + con_arity = dataConSourceArity data_con + data_con_RDR = qual_orig_name data_con + + as_needed = take con_arity as_RDRs + bs_needed = take con_arity bs_RDRs + cs_needed = take con_arity cs_RDRs - as_needed = take con_arity as_PNs - bs_needed = take con_arity bs_PNs - cs_needed = take con_arity cs_PNs + con_pat xs = ConPatIn data_con_RDR (map VarPatIn xs) + con_expr = mk_easy_App data_con_RDR cs_needed -------------------------------------------------------------- single_con_range - = mk_easy_FunMonoBind range_PN [TuplePatIn [con_pat as_needed, con_pat bs_needed]] [] ( - ListComp (con_expr cs_needed) (zipWith3Equal "single_con_range" mk_qual as_needed bs_needed cs_needed) - ) + = mk_easy_FunMonoBind tycon_loc range_RDR + [TuplePatIn [con_pat as_needed, con_pat bs_needed] Boxed] [] $ + HsDo ListComp stmts tycon_loc where - mk_qual a b c = GeneratorQual (VarPatIn c) - (HsApp (HsVar range_PN) (ExplicitTuple [HsVar a, HsVar b])) + stmts = zipWith3Equal "single_con_range" mk_qual as_needed bs_needed cs_needed + ++ + [ReturnStmt con_expr] + + mk_qual a b c = BindStmt (VarPatIn c) + (HsApp (HsVar range_RDR) + (ExplicitTuple [HsVar a, HsVar b] Boxed)) + tycon_loc ---------------- single_con_index - = mk_easy_FunMonoBind index_PN [TuplePatIn [con_pat as_needed, con_pat bs_needed], con_pat cs_needed] [range_size] ( + = mk_easy_FunMonoBind tycon_loc index_RDR + [TuplePatIn [con_pat as_needed, con_pat bs_needed] Boxed, + con_pat cs_needed] [range_size] ( foldl mk_index (HsLit (HsInt 0)) (zip3 as_needed bs_needed cs_needed)) where mk_index multiply_by (l, u, i) - =OpApp ( - (HsApp (HsApp (HsVar index_PN) (ExplicitTuple [HsVar l, HsVar u])) (HsVar i)) - ) (HsVar plus_PN) ( - OpApp ( - (HsApp (HsVar rangeSize_PN) (ExplicitTuple [HsVar l, HsVar u])) - ) (HsVar times_PN) multiply_by + = genOpApp ( + (HsApp (HsApp (HsVar index_RDR) + (ExplicitTuple [HsVar l, HsVar u] Boxed)) (HsVar i)) + ) plus_RDR ( + genOpApp ( + (HsApp (HsVar rangeSize_RDR) + (ExplicitTuple [HsVar l, HsVar u] Boxed)) + ) times_RDR multiply_by ) range_size - = mk_easy_FunMonoBind rangeSize_PN [TuplePatIn [a_Pat, b_Pat]] [] ( - OpApp ( - (HsApp (HsApp (HsVar index_PN) (ExplicitTuple [a_Expr, b_Expr])) b_Expr) - ) (HsVar plus_PN) (HsLit (HsInt 1))) + = mk_easy_FunMonoBind tycon_loc rangeSize_RDR + [TuplePatIn [a_Pat, b_Pat] Boxed] [] ( + genOpApp ( + (HsApp (HsApp (HsVar index_RDR) + (ExplicitTuple [a_Expr, b_Expr] Boxed)) b_Expr) + ) plus_RDR (HsLit (HsInt 1))) ------------------ single_con_inRange - = mk_easy_FunMonoBind inRange_PN [TuplePatIn [con_pat as_needed, con_pat bs_needed], con_pat cs_needed] [] ( + = mk_easy_FunMonoBind tycon_loc inRange_RDR + [TuplePatIn [con_pat as_needed, con_pat bs_needed] Boxed, + con_pat cs_needed] + [] ( foldl1 and_Expr (zipWith3Equal "single_con_inRange" in_range as_needed bs_needed cs_needed)) where - in_range a b c = HsApp (HsApp (HsVar inRange_PN) (ExplicitTuple [HsVar a, HsVar b])) (HsVar c) + in_range a b c = HsApp (HsApp (HsVar inRange_RDR) + (ExplicitTuple [HsVar a, HsVar b] Boxed)) + (HsVar c) \end{code} %************************************************************************ @@ -674,63 +771,155 @@ gen_Ix_binds tycon %* * %************************************************************************ -Ignoring all the infix-ery mumbo jumbo (ToDo) - \begin{code} -gen_Read_binds :: [RenamedFixityDecl] -> TyCon -> RdrNameMonoBinds +gen_Read_binds :: (Name -> Maybe Fixity) -> TyCon -> RdrNameMonoBinds -gen_Read_binds fixities tycon +gen_Read_binds get_fixity tycon = reads_prec `AndMonoBinds` read_list where + tycon_loc = getSrcLoc tycon ----------------------------------------------------------------------- - read_list = mk_easy_FunMonoBind readList_PN [] [] - (HsApp (HsVar _readList_PN) (HsPar (HsApp (HsVar readsPrec_PN) (HsLit (HsInt 0))))) + read_list = mk_easy_FunMonoBind tycon_loc readList_RDR [] [] + (HsApp (HsVar readList___RDR) (HsPar (HsApp (HsVar readsPrec_RDR) (HsLit (HsInt 0))))) ----------------------------------------------------------------------- reads_prec = let read_con_comprehensions = map read_con (tyConDataCons tycon) in - mk_easy_FunMonoBind readsPrec_PN [a_Pat, b_Pat] [] ( - foldl1 append_Expr read_con_comprehensions + mk_easy_FunMonoBind tycon_loc readsPrec_RDR [zz_a_Pat, b_Pat] [] ( + foldr1 append_Expr read_con_comprehensions ) where read_con data_con -- note: "b" is the string being "read" - = let - data_con_PN = origName data_con - data_con_str= snd (moduleNamePair data_con) - con_arity = dataConArity data_con - as_needed = take con_arity as_PNs - bs_needed = take con_arity bs_PNs - con_expr = mk_easy_App data_con_PN as_needed - nullary_con = isNullaryDataCon data_con - - con_qual - = GeneratorQual - (TuplePatIn [LitPatIn (HsString data_con_str), d_Pat]) - (HsApp (HsVar lex_PN) c_Expr) - - field_quals = snd (mapAccumL mk_qual d_Expr (zipEqual "as_needed" as_needed bs_needed)) - - read_paren_arg - = if nullary_con then -- must be False (parens are surely optional) - false_Expr - else -- parens depend on precedence... - HsPar (OpApp a_Expr (HsVar gt_PN) (HsLit (HsInt 9))) - in - HsApp ( + = HsApp ( readParen_Expr read_paren_arg $ HsPar $ - HsLam (mk_easy_Match [c_Pat] [] ( - ListComp (ExplicitTuple [con_expr, - if null bs_needed then d_Expr else HsVar (last bs_needed)]) - (con_qual : field_quals))) - ) (HsVar b_PN) - where - mk_qual draw_from (con_field, str_left) - = (HsVar str_left, -- what to draw from down the line... - GeneratorQual - (TuplePatIn [VarPatIn con_field, VarPatIn str_left]) - (HsApp (HsApp (HsVar readsPrec_PN) (HsLit (HsInt 10))) draw_from)) + HsLam (mk_easy_Match tycon_loc [c_Pat] [] $ + HsDo ListComp stmts tycon_loc) + ) (HsVar b_RDR) + where + data_con_RDR = qual_orig_name data_con + data_con_str = occNameUserString (getOccName data_con) + con_arity = dataConSourceArity data_con + con_expr = mk_easy_App data_con_RDR as_needed + nullary_con = con_arity == 0 + labels = dataConFieldLabels data_con + lab_fields = length labels + dc_nm = getName data_con + is_infix = isDataSymOcc (getOccName dc_nm) + + as_needed = take con_arity as_RDRs + bs_needed + | is_infix = take (1 + con_arity) bs_RDRs + | lab_fields == 0 = take con_arity bs_RDRs + | otherwise = take (4*lab_fields + 1) bs_RDRs + -- (label, '=' and field)*n, (n-1)*',' + '{' + '}' + + (as1:as2:_) = as_needed + (bs1:bs2:bs3:_) = bs_needed + + con_qual + | not is_infix = + BindStmt + (TuplePatIn [LitPatIn (mkHsString data_con_str), d_Pat] Boxed) + (HsApp (HsVar lex_RDR) c_Expr) + tycon_loc + | otherwise = + BindStmt + (TuplePatIn [LitPatIn (mkHsString data_con_str), VarPatIn bs2] Boxed) + (HsApp (HsVar lex_RDR) (HsVar bs1)) + tycon_loc + + + str_qual str res draw_from = + BindStmt + (TuplePatIn [LitPatIn (mkHsString str), VarPatIn res] Boxed) + (HsApp (HsVar lex_RDR) draw_from) + tycon_loc + + str_qual_paren str res draw_from = + BindStmt + (TuplePatIn [LitPatIn (mkHsString str), VarPatIn res] Boxed) + (HsApp (readParen_Expr true_Expr (HsVar lex_RDR)) draw_from) + tycon_loc + + read_label f = [rd_lab, str_qual "="] + -- There might be spaces between the label and '=' + where + rd_lab + | is_op = str_qual_paren nm + | otherwise = str_qual nm + + occ_nm = getOccName (fieldLabelName f) + is_op = isSymOcc occ_nm + nm = occNameUserString occ_nm + + field_quals + | is_infix = + snd (mapAccumL mk_qual_infix + c_Expr + [ (mk_read_qual lp as1, bs1, bs2) + , (mk_read_qual rp as2, bs3, bs3) + ]) + | lab_fields == 0 = -- common case. + snd (mapAccumL mk_qual + d_Expr + (zipWithEqual "as_needed" + (\ con_field draw_from -> (mk_read_qual 10 con_field, + draw_from)) + as_needed bs_needed)) + | otherwise = + snd $ + mapAccumL mk_qual d_Expr + (zipEqual "bs_needed" + ((str_qual "{": + concat ( + intersperse [str_qual ","] $ + zipWithEqual + "field_quals" + (\ as b -> as ++ [b]) + -- The labels + (map read_label labels) + -- The fields + (map (mk_read_qual 10) as_needed))) ++ [str_qual "}"]) + bs_needed) + + mk_qual_infix draw_from (f, str_left, str_left2) = + (HsVar str_left2, -- what to draw from down the line... + f str_left draw_from) + + mk_qual draw_from (f, str_left) = + (HsVar str_left, -- what to draw from down the line... + f str_left draw_from) + + mk_read_qual p con_field res draw_from = + BindStmt + (TuplePatIn [VarPatIn con_field, VarPatIn res] Boxed) + (HsApp (HsApp (HsVar readsPrec_RDR) (HsLit (HsInt p))) draw_from) + tycon_loc + + result_expr = ExplicitTuple [con_expr, if null bs_needed + then d_Expr + else HsVar (last bs_needed)] Boxed + + [lp,rp] = getLRPrecs is_infix get_fixity dc_nm + + quals + | is_infix = let (h:t) = field_quals in (h:con_qual:t) + | otherwise = con_qual:field_quals + + stmts = quals ++ [ReturnStmt result_expr] + + {- + c.f. Figure 18 in Haskell 1.1 report. + -} + paren_prec_limit + | not is_infix = defaultPrecedence + | otherwise = getPrecedence get_fixity dc_nm + + read_paren_arg -- parens depend on precedence... + | nullary_con = false_Expr -- it's optional. + | otherwise = HsPar (genOpApp zz_a_Expr gt_RDR (HsLit (HsInt paren_prec_limit))) \end{code} %************************************************************************ @@ -739,54 +928,155 @@ gen_Read_binds fixities tycon %* * %************************************************************************ -Ignoring all the infix-ery mumbo jumbo (ToDo) - \begin{code} -gen_Show_binds :: [RenamedFixityDecl] -> TyCon -> RdrNameMonoBinds +gen_Show_binds :: (Name -> Maybe Fixity) -> TyCon -> RdrNameMonoBinds -gen_Show_binds fixities tycon +gen_Show_binds get_fixity tycon = shows_prec `AndMonoBinds` show_list where + tycon_loc = getSrcLoc tycon ----------------------------------------------------------------------- - show_list = mk_easy_FunMonoBind showList_PN [] [] - (HsApp (HsVar _showList_PN) (HsPar (HsApp (HsVar showsPrec_PN) (HsLit (HsInt 0))))) + show_list = mk_easy_FunMonoBind tycon_loc showList_RDR [] [] + (HsApp (HsVar showList___RDR) (HsPar (HsApp (HsVar showsPrec_RDR) (HsLit (HsInt 0))))) ----------------------------------------------------------------------- - shows_prec - = mk_FunMonoBind showsPrec_PN (map pats_etc (tyConDataCons tycon)) + shows_prec = mk_FunMonoBind tycon_loc showsPrec_RDR (map pats_etc (tyConDataCons tycon)) where pats_etc data_con - = let - data_con_PN = origName data_con - con_arity = dataConArity data_con - bs_needed = take con_arity bs_PNs - con_pat = ConPatIn data_con_PN (map VarPatIn bs_needed) - nullary_con = isNullaryDataCon data_con - - show_con - = let (mod, nm) = moduleNamePair data_con - space_maybe = if nullary_con then _NIL_ else SLIT(" ") - in - HsApp (HsVar showString_PN) (HsLit (HsString (nm _APPEND_ space_maybe))) - - show_thingies = show_con : (spacified real_show_thingies) - - real_show_thingies - = [ HsApp (HsApp (HsVar showsPrec_PN) (HsLit (HsInt 10))) (HsVar b) - | b <- bs_needed ] - in - if nullary_con then -- skip the showParen junk... - ASSERT(null bs_needed) - ([a_Pat, con_pat], show_con) - else - ([a_Pat, con_pat], - showParen_Expr (HsPar (OpApp a_Expr (HsVar ge_PN) (HsLit (HsInt 10)))) - (HsPar (nested_compose_Expr show_thingies))) - where - spacified [] = [] - spacified [x] = [x] - spacified (x:xs) = (x : (HsVar showSpace_PN) : spacified xs) + | nullary_con = -- skip the showParen junk... + ASSERT(null bs_needed) + ([wildPat, con_pat], show_con) + | otherwise = + ([a_Pat, con_pat], + showParen_Expr (HsPar (genOpApp a_Expr ge_RDR (HsLit (HsInt paren_prec_limit)))) + (HsPar (nested_compose_Expr show_thingies))) + where + data_con_RDR = qual_orig_name data_con + con_arity = dataConSourceArity data_con + bs_needed = take con_arity bs_RDRs + con_pat = ConPatIn data_con_RDR (map VarPatIn bs_needed) + nullary_con = con_arity == 0 + labels = dataConFieldLabels data_con + lab_fields = length labels + + dc_nm = getName data_con + dc_occ_nm = getOccName data_con + dc_occ_nm_str = occNameUserString dc_occ_nm + + is_infix = isDataSymOcc dc_occ_nm + + + show_con + | is_infix = mk_showString_app (' ':dc_occ_nm_str) + | otherwise = mk_showString_app (dc_occ_nm_str ++ space_ocurly_maybe) + where + space_ocurly_maybe + | nullary_con = "" + | lab_fields == 0 = " " + | otherwise = "{" + + + show_all con fs@(x:xs) + | is_infix = x:con:xs + | otherwise = + let + ccurly_maybe + | lab_fields > 0 = [mk_showString_app "}"] + | otherwise = [] + in + con:fs ++ ccurly_maybe + + show_thingies = show_all show_con real_show_thingies_with_labs + + show_label l = mk_showString_app (the_name ++ "=") + where + occ_nm = getOccName (fieldLabelName l) + -- legal, but rare. + is_op = isSymOcc occ_nm + the_name + | is_op = '(':nm ++ ")" + | otherwise = nm + + nm = occNameUserString occ_nm + + + mk_showString_app str = HsApp (HsVar showString_RDR) + (HsLit (mkHsString str)) + + prec_cons = getLRPrecs is_infix get_fixity dc_nm + + real_show_thingies + | is_infix = + [ HsApp (HsApp (HsVar showsPrec_RDR) (HsLit (HsInt p))) (HsVar b) + | (p,b) <- zip prec_cons bs_needed ] + | otherwise = + [ HsApp (HsApp (HsVar showsPrec_RDR) (HsLit (HsInt 10))) (HsVar b) + | b <- bs_needed ] + + real_show_thingies_with_labs + | lab_fields == 0 = intersperse (HsVar showSpace_RDR) real_show_thingies + | otherwise = --Assumption: no of fields == no of labelled fields + -- (and in same order) + concat $ + intersperse ([mk_showString_app ","]) $ -- Using SLIT()s containing ,s spells trouble. + zipWithEqual "gen_Show_binds" + (\ a b -> [a,b]) + (map show_label labels) + real_show_thingies + + {- + c.f. Figure 16 and 17 in Haskell 1.1 report + -} + paren_prec_limit + | not is_infix = defaultPrecedence + 1 + | otherwise = getPrecedence get_fixity dc_nm + 1 + \end{code} +\begin{code} +getLRPrecs :: Bool -> (Name -> Maybe Fixity) -> Name -> [Integer] +getLRPrecs is_infix get_fixity nm = [lp, rp] + where + {- + Figuring out the fixities of the arguments to a constructor, + cf. Figures 16-18 in Haskell 1.1 report. + -} + (con_left_assoc, con_right_assoc) = isLRAssoc get_fixity nm + paren_con_prec = getPrecedence get_fixity nm + + lp + | not is_infix = defaultPrecedence + 1 + | con_left_assoc = paren_con_prec + | otherwise = paren_con_prec + 1 + + rp + | not is_infix = defaultPrecedence + 1 + | con_right_assoc = paren_con_prec + | otherwise = paren_con_prec + 1 + +defaultPrecedence :: Integer +defaultPrecedence = fromInt maxPrecedence + +getPrecedence :: (Name -> Maybe Fixity) -> Name -> Integer +getPrecedence get_fixity nm + = case get_fixity nm of + Just (Fixity x _) -> fromInt x + other -> defaultPrecedence + +isLRAssoc :: (Name -> Maybe Fixity) -> Name -> (Bool, Bool) +isLRAssoc get_fixity nm = + case get_fixity nm `orElse` defaultFixity of + Fixity _ InfixN -> (False, False) + Fixity _ InfixR -> (False, True) + Fixity _ InfixL -> (True, False) + +isInfixOccName :: String -> Bool +isInfixOccName str = + case str of + (':':_) -> True + _ -> False +\end{code} + + %************************************************************************ %* * \subsection{Generating extra binds (@con2tag@ and @tag2con@)} @@ -814,35 +1104,37 @@ gen_tag_n_con_monobind TagThingWanted) -> RdrNameMonoBinds -gen_tag_n_con_monobind (pn, tycon, GenCon2Tag) - = mk_FunMonoBind pn (map mk_stuff (tyConDataCons tycon)) - where - mk_stuff :: DataCon -> ([RdrNamePat], RdrNameHsExpr) +gen_tag_n_con_monobind (rdr_name, tycon, GenCon2Tag) + | lots_of_constructors + = mk_FunMonoBind (getSrcLoc tycon) rdr_name + [([VarPatIn a_RDR], HsApp getTag_Expr a_Expr)] - mk_stuff var - = ASSERT(isDataCon var) - ([pat], HsLit (HsIntPrim (toInteger ((dataConTag var) - fIRST_TAG)))) - where - pat = ConPatIn var_PN (nOfThem (dataConArity var) WildPatIn) - var_PN = origName var + | otherwise + = mk_FunMonoBind (getSrcLoc tycon) rdr_name (map mk_stuff (tyConDataCons tycon)) -gen_tag_n_con_monobind (pn, tycon, GenTag2Con) - = mk_FunMonoBind pn (map mk_stuff (tyConDataCons tycon)) where - mk_stuff :: DataCon -> ([RdrNamePat], RdrNameHsExpr) + lots_of_constructors = tyConFamilySize tycon > mAX_FAMILY_SIZE_FOR_VEC_RETURNS + mk_stuff :: DataCon -> ([RdrNamePat], RdrNameHsExpr) mk_stuff var - = ASSERT(isDataCon var) - ([lit_pat], HsVar var_PN) + = ([pat], HsLit (HsIntPrim (toInteger ((dataConTag var) - fIRST_TAG)))) where - lit_pat = ConPatIn mkInt_PN [LitPatIn (HsIntPrim (toInteger ((dataConTag var) - fIRST_TAG)))] - var_PN = origName var - -gen_tag_n_con_monobind (pn, tycon, GenMaxTag) - = mk_easy_FunMonoBind pn [] [] (HsApp (HsVar mkInt_PN) (HsLit (HsIntPrim max_tag))) + pat = ConPatIn var_RDR (nOfThem (dataConSourceArity var) WildPatIn) + var_RDR = qual_orig_name var + +gen_tag_n_con_monobind (rdr_name, tycon, GenTag2Con) + = mk_FunMonoBind (getSrcLoc tycon) rdr_name + [([ConPatIn mkInt_RDR [VarPatIn a_RDR]], + ExprWithTySig (HsApp tagToEnum_Expr a_Expr) + (HsTyVar (qual_orig_name tycon)))] + +gen_tag_n_con_monobind (rdr_name, tycon, GenMaxTag) + = mk_easy_FunMonoBind (getSrcLoc tycon) + rdr_name [] [] (HsApp (HsVar mkInt_RDR) (HsLit (HsIntPrim max_tag))) where max_tag = case (tyConDataCons tycon) of data_cons -> toInteger ((length data_cons) - fIRST_TAG) + \end{code} %************************************************************************ @@ -866,36 +1158,32 @@ multi-clause definitions; it generates: \end{verbatim} \begin{code} -mk_easy_FunMonoBind :: RdrName -> [RdrNamePat] +mk_easy_FunMonoBind :: SrcLoc -> RdrName -> [RdrNamePat] -> [RdrNameMonoBinds] -> RdrNameHsExpr -> RdrNameMonoBinds -mk_easy_FunMonoBind fun pats binds expr - = FunMonoBind fun False{-not infix-} [mk_easy_Match pats binds expr] mkGeneratedSrcLoc +mk_easy_FunMonoBind loc fun pats binds expr + = FunMonoBind fun False{-not infix-} [mk_easy_Match loc pats binds expr] loc -mk_easy_Match pats binds expr - = mk_match pats expr (mkbind binds) - where - mkbind [] = EmptyBinds - mkbind bs = SingleBind (RecBind (foldr1 AndMonoBinds bs)) +mk_easy_Match loc pats binds expr + = mk_match loc pats expr (mkMonoBind (andMonoBindList binds) [] Recursive) -- The renamer expects everything in its input to be a -- "recursive" MonoBinds, and it is its job to sort things out -- from there. -mk_FunMonoBind :: RdrName +mk_FunMonoBind :: SrcLoc -> RdrName -> [([RdrNamePat], RdrNameHsExpr)] -> RdrNameMonoBinds -mk_FunMonoBind fun [] = panic "TcGenDeriv:mk_FunMonoBind" -mk_FunMonoBind fun pats_and_exprs +mk_FunMonoBind loc fun [] = panic "TcGenDeriv:mk_FunMonoBind" +mk_FunMonoBind loc fun pats_and_exprs = FunMonoBind fun False{-not infix-} - [ mk_match p e EmptyBinds | (p,e) <-pats_and_exprs ] - mkGeneratedSrcLoc + [ mk_match loc p e EmptyBinds | (p,e) <-pats_and_exprs ] + loc -mk_match pats expr binds - = foldr PatMatch - (GRHSMatch (GRHSsAndBindsIn [OtherwiseGRHS expr mkGeneratedSrcLoc] binds)) - (map paren pats) +mk_match loc pats expr binds + = Match [] (map paren pats) Nothing + (GRHSs (unguardedRHS expr loc) binds Nothing) where paren p@(VarPatIn _) = p paren other_p = ParPatIn other_p @@ -905,8 +1193,10 @@ mk_match pats expr binds mk_easy_App f xs = foldl HsApp (HsVar f) (map HsVar xs) \end{code} +ToDo: Better SrcLocs. + \begin{code} -compare_Case, cmp_eq_Expr :: +compare_Case :: RdrNameHsExpr -> RdrNameHsExpr -> RdrNameHsExpr -> RdrNameHsExpr -> RdrNameHsExpr -> RdrNameHsExpr @@ -921,30 +1211,26 @@ careful_compare_Case :: -- checks for primitive types... -> RdrNameHsExpr -> RdrNameHsExpr -> RdrNameHsExpr -compare_Case = compare_gen_Case compare_PN -cmp_eq_Expr = compare_gen_Case cmp_eq_PN +compare_Case = compare_gen_Case compare_RDR +cmp_eq_Expr a b = HsApp (HsApp (HsVar cmp_eq_RDR) a) b + -- Was: compare_gen_Case cmp_eq_RDR compare_gen_Case fun lt eq gt a b = HsCase (HsPar (HsApp (HsApp (HsVar fun) a) b)) {-of-} - [PatMatch (ConPatIn ltTag_PN []) - (GRHSMatch (GRHSsAndBindsIn [OtherwiseGRHS lt mkGeneratedSrcLoc] EmptyBinds)), - - PatMatch (ConPatIn eqTag_PN []) - (GRHSMatch (GRHSsAndBindsIn [OtherwiseGRHS eq mkGeneratedSrcLoc] EmptyBinds)), - - PatMatch (ConPatIn gtTag_PN []) - (GRHSMatch (GRHSsAndBindsIn [OtherwiseGRHS gt mkGeneratedSrcLoc] EmptyBinds))] - mkGeneratedSrcLoc + [mkSimpleMatch [ConPatIn ltTag_RDR []] lt Nothing generatedSrcLoc, + mkSimpleMatch [ConPatIn eqTag_RDR []] eq Nothing generatedSrcLoc, + mkSimpleMatch [ConPatIn gtTag_RDR []] gt Nothing generatedSrcLoc] + generatedSrcLoc careful_compare_Case ty lt eq gt a b - = if not (isPrimType ty) then - compare_gen_Case compare_PN lt eq gt a b + = if not (isUnboxedType ty) then + compare_gen_Case compare_RDR lt eq gt a b else -- we have to do something special for primitive things... - HsIf (HsPar (OpApp a (HsVar relevant_eq_op) b)) + HsIf (genOpApp a relevant_eq_op b) eq - (HsIf (HsPar (OpApp a (HsVar relevant_lt_op) b)) lt gt mkGeneratedSrcLoc) - mkGeneratedSrcLoc + (HsIf (genOpApp a relevant_lt_op b) lt gt generatedSrcLoc) + generatedSrcLoc where relevant_eq_op = assoc_ty_id eq_op_tbl ty relevant_lt_op = assoc_ty_id lt_op_tbl ty @@ -953,41 +1239,41 @@ assoc_ty_id tyids ty = if null res then panic "assoc_ty" else head res where - res = [id | (ty',id) <- tyids, eqTy ty ty'] + res = [id | (ty',id) <- tyids, ty == ty'] eq_op_tbl = - [(charPrimTy, eqH_Char_PN) - ,(intPrimTy, eqH_Int_PN) - ,(wordPrimTy, eqH_Word_PN) - ,(addrPrimTy, eqH_Addr_PN) - ,(floatPrimTy, eqH_Float_PN) - ,(doublePrimTy, eqH_Double_PN) + [(charPrimTy, eqH_Char_RDR) + ,(intPrimTy, eqH_Int_RDR) + ,(wordPrimTy, eqH_Word_RDR) + ,(addrPrimTy, eqH_Addr_RDR) + ,(floatPrimTy, eqH_Float_RDR) + ,(doublePrimTy, eqH_Double_RDR) ] lt_op_tbl = - [(charPrimTy, ltH_Char_PN) - ,(intPrimTy, ltH_Int_PN) - ,(wordPrimTy, ltH_Word_PN) - ,(addrPrimTy, ltH_Addr_PN) - ,(floatPrimTy, ltH_Float_PN) - ,(doublePrimTy, ltH_Double_PN) + [(charPrimTy, ltH_Char_RDR) + ,(intPrimTy, ltH_Int_RDR) + ,(wordPrimTy, ltH_Word_RDR) + ,(addrPrimTy, ltH_Addr_RDR) + ,(floatPrimTy, ltH_Float_RDR) + ,(doublePrimTy, ltH_Double_RDR) ] ----------------------------------------------------------------------- and_Expr, append_Expr :: RdrNameHsExpr -> RdrNameHsExpr -> RdrNameHsExpr -and_Expr a b = OpApp a (HsVar and_PN) b -append_Expr a b = OpApp a (HsVar append_PN) b +and_Expr a b = genOpApp a and_RDR b +append_Expr a b = genOpApp a append_RDR b ----------------------------------------------------------------------- eq_Expr :: Type -> RdrNameHsExpr -> RdrNameHsExpr -> RdrNameHsExpr eq_Expr ty a b - = if not (isPrimType ty) then - OpApp a (HsVar eq_PN) b + = if not (isUnboxedType ty) then + genOpApp a eq_RDR b else -- we have to do something special for primitive things... - OpApp a (HsVar relevant_eq_op) b + genOpApp a relevant_eq_op b where relevant_eq_op = assoc_ty_id eq_op_tbl ty \end{code} @@ -997,11 +1283,8 @@ untag_Expr :: TyCon -> [(RdrName, RdrName)] -> RdrNameHsExpr -> RdrNameHsExpr untag_Expr tycon [] expr = expr untag_Expr tycon ((untag_this, put_tag_here) : more) expr = HsCase (HsPar (HsApp (con2tag_Expr tycon) (HsVar untag_this))) {-of-} - [PatMatch (VarPatIn put_tag_here) - (GRHSMatch (GRHSsAndBindsIn grhs EmptyBinds))] - mkGeneratedSrcLoc - where - grhs = [OtherwiseGRHS (untag_Expr tycon more expr) mkGeneratedSrcLoc] + [mkSimpleMatch [VarPatIn put_tag_here] (untag_Expr tycon more expr) Nothing generatedSrcLoc] + generatedSrcLoc cmp_tags_Expr :: RdrName -- Comparison op -> RdrName -> RdrName -- Things to compare @@ -1010,7 +1293,7 @@ cmp_tags_Expr :: RdrName -- Comparison op -> RdrNameHsExpr cmp_tags_Expr op a b true_case false_case - = HsIf (HsPar (OpApp (HsVar a) (HsVar op) (HsVar b))) true_case false_case mkGeneratedSrcLoc + = HsIf (genOpApp (HsVar a) op (HsVar b)) true_case false_case generatedSrcLoc enum_from_to_Expr :: RdrNameHsExpr -> RdrNameHsExpr @@ -1019,140 +1302,108 @@ enum_from_then_to_Expr :: RdrNameHsExpr -> RdrNameHsExpr -> RdrNameHsExpr -> RdrNameHsExpr -enum_from_to_Expr f t2 = HsApp (HsApp (HsVar enumFromTo_PN) f) t2 -enum_from_then_to_Expr f t t2 = HsApp (HsApp (HsApp (HsVar enumFromThenTo_PN) f) t) t2 +enum_from_to_Expr f t2 = HsApp (HsApp (HsVar enumFromTo_RDR) f) t2 +enum_from_then_to_Expr f t t2 = HsApp (HsApp (HsApp (HsVar enumFromThenTo_RDR) f) t) t2 showParen_Expr, readParen_Expr :: RdrNameHsExpr -> RdrNameHsExpr -> RdrNameHsExpr -showParen_Expr e1 e2 = HsApp (HsApp (HsVar showParen_PN) e1) e2 -readParen_Expr e1 e2 = HsApp (HsApp (HsVar readParen_PN) e1) e2 +showParen_Expr e1 e2 = HsApp (HsApp (HsVar showParen_RDR) e1) e2 +readParen_Expr e1 e2 = HsApp (HsApp (HsVar readParen_RDR) e1) e2 nested_compose_Expr :: [RdrNameHsExpr] -> RdrNameHsExpr nested_compose_Expr [e] = parenify e nested_compose_Expr (e:es) - = HsApp (HsApp (HsVar compose_PN) (parenify e)) (nested_compose_Expr es) + = HsApp (HsApp (HsVar compose_RDR) (parenify e)) (nested_compose_Expr es) + +-- impossible_Expr is used in case RHSs that should never happen. +-- We generate these to keep the desugarer from complaining that they *might* happen! +impossible_Expr = HsApp (HsVar error_RDR) (HsLit (HsString (_PK_ "Urk! in TcGenDeriv"))) + +-- illegal_Expr is used when signalling error conditions in the RHS of a derived +-- method. It is currently only used by Enum.{succ,pred} +illegal_Expr meth tp msg = + HsApp (HsVar error_RDR) (HsLit (HsString (_PK_ (meth ++ '{':tp ++ "}: " ++ msg)))) + +-- illegal_toEnum_tag is an extended version of illegal_Expr, which also allows you +-- to include the value of a_RDR in the error string. +illegal_toEnum_tag tp maxtag = + HsApp (HsVar error_RDR) + (HsApp (HsApp (HsVar append_RDR) + (HsLit (HsString (_PK_ ("toEnum{" ++ tp ++ "}: tag ("))))) + (HsApp (HsApp (HsApp + (HsVar showsPrec_RDR) + (HsLit (HsInt 0))) + (HsVar a_RDR)) + (HsApp (HsApp + (HsVar append_RDR) + (HsLit (HsString (_PK_ ") is outside of enumeration's range (0,")))) + (HsApp (HsApp (HsApp + (HsVar showsPrec_RDR) + (HsLit (HsInt 0))) + (HsVar maxtag)) + (HsLit (HsString (_PK_ ")"))))))) parenify e@(HsVar _) = e parenify e = HsPar e -\end{code} -\begin{code} -a_PN = Unqual SLIT("a") -b_PN = Unqual SLIT("b") -c_PN = Unqual SLIT("c") -d_PN = Unqual SLIT("d") -ah_PN = Unqual SLIT("a#") -bh_PN = Unqual SLIT("b#") -ch_PN = Unqual SLIT("c#") -dh_PN = Unqual SLIT("d#") -cmp_eq_PN = Unqual SLIT("cmp_eq") -rangeSize_PN = Unqual SLIT("rangeSize") - -as_PNs = [ Unqual (_PK_ ("a"++show i)) | i <- [(1::Int) .. ] ] -bs_PNs = [ Unqual (_PK_ ("b"++show i)) | i <- [(1::Int) .. ] ] -cs_PNs = [ Unqual (_PK_ ("c"++show i)) | i <- [(1::Int) .. ] ] - -eq_PN = prelude_method SLIT("Eq") SLIT("==") -ne_PN = prelude_method SLIT("Eq") SLIT("/=") -le_PN = prelude_method SLIT("Ord") SLIT("<=") -lt_PN = prelude_method SLIT("Ord") SLIT("<") -ge_PN = prelude_method SLIT("Ord") SLIT(">=") -gt_PN = prelude_method SLIT("Ord") SLIT(">") -max_PN = prelude_method SLIT("Ord") SLIT("max") -min_PN = prelude_method SLIT("Ord") SLIT("min") -compare_PN = prelude_method SLIT("Ord") SLIT("compare") -minBound_PN = prelude_method SLIT("Bounded") SLIT("minBound") -maxBound_PN = prelude_method SLIT("Bounded") SLIT("maxBound") -ltTag_PN = Unqual SLIT("LT") -eqTag_PN = Unqual SLIT("EQ") -gtTag_PN = Unqual SLIT("GT") -enumFrom_PN = prelude_method SLIT("Enum") SLIT("enumFrom") -enumFromTo_PN = prelude_method SLIT("Enum") SLIT("enumFromTo") -enumFromThen_PN = prelude_method SLIT("Enum") SLIT("enumFromThen") -enumFromThenTo_PN= prelude_method SLIT("Enum") SLIT("enumFromThenTo") -range_PN = prelude_method SLIT("Ix") SLIT("range") -index_PN = prelude_method SLIT("Ix") SLIT("index") -inRange_PN = prelude_method SLIT("Ix") SLIT("inRange") -readsPrec_PN = prelude_method SLIT("Read") SLIT("readsPrec") -readList_PN = prelude_method SLIT("Read") SLIT("readList") -showsPrec_PN = prelude_method SLIT("Show") SLIT("showsPrec") -showList_PN = prelude_method SLIT("Show") SLIT("showList") -plus_PN = prelude_method SLIT("Num") SLIT("+") -times_PN = prelude_method SLIT("Num") SLIT("*") - -false_PN = prelude_val pRELUDE SLIT("False") -true_PN = prelude_val pRELUDE SLIT("True") -eqH_Char_PN = prelude_primop CharEqOp -ltH_Char_PN = prelude_primop CharLtOp -eqH_Word_PN = prelude_primop WordEqOp -ltH_Word_PN = prelude_primop WordLtOp -eqH_Addr_PN = prelude_primop AddrEqOp -ltH_Addr_PN = prelude_primop AddrLtOp -eqH_Float_PN = prelude_primop FloatEqOp -ltH_Float_PN = prelude_primop FloatLtOp -eqH_Double_PN = prelude_primop DoubleEqOp -ltH_Double_PN = prelude_primop DoubleLtOp -eqH_Int_PN = prelude_primop IntEqOp -ltH_Int_PN = prelude_primop IntLtOp -geH_PN = prelude_primop IntGeOp -leH_PN = prelude_primop IntLeOp -minusH_PN = prelude_primop IntSubOp -and_PN = prelude_val pRELUDE SLIT("&&") -not_PN = prelude_val pRELUDE SLIT("not") -append_PN = prelude_val pRELUDE_LIST SLIT("++") -map_PN = prelude_val pRELUDE_LIST SLIT("map") -compose_PN = prelude_val pRELUDE SLIT(".") -mkInt_PN = prelude_val pRELUDE_BUILTIN SLIT("I#") -error_PN = prelude_val pRELUDE SLIT("error") -showString_PN = prelude_val pRELUDE_TEXT SLIT("showString") -showParen_PN = prelude_val pRELUDE_TEXT SLIT("showParen") -readParen_PN = prelude_val pRELUDE_TEXT SLIT("readParen") -lex_PN = prelude_val pRELUDE_TEXT SLIT("lex") -showSpace_PN = prelude_val pRELUDE_TEXT SLIT("__showSpace") -_showList_PN = prelude_val pRELUDE SLIT("__showList") -_readList_PN = prelude_val pRELUDE SLIT("__readList") - -prelude_val m s = Unqual s -prelude_method c o = Unqual o -prelude_primop o = origName (primOpId o) - -a_Expr = HsVar a_PN -b_Expr = HsVar b_PN -c_Expr = HsVar c_PN -d_Expr = HsVar d_PN -ltTag_Expr = HsVar ltTag_PN -eqTag_Expr = HsVar eqTag_PN -gtTag_Expr = HsVar gtTag_PN -false_Expr = HsVar false_PN -true_Expr = HsVar true_PN - -con2tag_Expr tycon = HsVar (con2tag_PN tycon) - -a_Pat = VarPatIn a_PN -b_Pat = VarPatIn b_PN -c_Pat = VarPatIn c_PN -d_Pat = VarPatIn d_PN - -con2tag_PN, tag2con_PN, maxtag_PN :: TyCon -> RdrName - -con2tag_PN tycon - = let (mod, nm) = moduleNamePair tycon - con2tag = SLIT("con2tag_") _APPEND_ nm _APPEND_ SLIT("#") - in - (if fromPrelude mod then Unqual else Qual mod) con2tag +-- genOpApp wraps brackets round the operator application, so that the +-- renamer won't subsequently try to re-associate it. +-- For some reason the renamer doesn't reassociate it right, and I can't +-- be bothered to find out why just now. -tag2con_PN tycon - = let (mod, nm) = moduleNamePair tycon - tag2con = SLIT("tag2con_") _APPEND_ nm _APPEND_ SLIT("#") - in - (if fromPrelude mod then Unqual else Qual mod) tag2con +genOpApp e1 op e2 = mkHsOpApp e1 op e2 +\end{code} -maxtag_PN tycon - = let (mod, nm) = moduleNamePair tycon - maxtag = SLIT("maxtag_") _APPEND_ nm _APPEND_ SLIT("#") - in - (if fromPrelude mod then Unqual else Qual mod) maxtag +\begin{code} +qual_orig_name n = nameRdrName (getName n) +varUnqual n = mkUnqual varName n + +zz_a_RDR = varUnqual SLIT("_a") +a_RDR = varUnqual SLIT("a") +b_RDR = varUnqual SLIT("b") +c_RDR = varUnqual SLIT("c") +d_RDR = varUnqual SLIT("d") +ah_RDR = varUnqual SLIT("a#") +bh_RDR = varUnqual SLIT("b#") +ch_RDR = varUnqual SLIT("c#") +dh_RDR = varUnqual SLIT("d#") +cmp_eq_RDR = varUnqual SLIT("cmp_eq") +rangeSize_RDR = varUnqual SLIT("rangeSize") + +as_RDRs = [ varUnqual (_PK_ ("a"++show i)) | i <- [(1::Int) .. ] ] +bs_RDRs = [ varUnqual (_PK_ ("b"++show i)) | i <- [(1::Int) .. ] ] +cs_RDRs = [ varUnqual (_PK_ ("c"++show i)) | i <- [(1::Int) .. ] ] + +mkHsString s = HsString (_PK_ s) + +zz_a_Expr = HsVar zz_a_RDR +a_Expr = HsVar a_RDR +b_Expr = HsVar b_RDR +c_Expr = HsVar c_RDR +d_Expr = HsVar d_RDR +ltTag_Expr = HsVar ltTag_RDR +eqTag_Expr = HsVar eqTag_RDR +gtTag_Expr = HsVar gtTag_RDR +false_Expr = HsVar false_RDR +true_Expr = HsVar true_RDR + +getTag_Expr = HsVar getTag_RDR +tagToEnum_Expr = HsVar tagToEnumH_RDR +con2tag_Expr tycon = HsVar (con2tag_RDR tycon) + +wildPat = WildPatIn +zz_a_Pat = VarPatIn zz_a_RDR +a_Pat = VarPatIn a_RDR +b_Pat = VarPatIn b_RDR +c_Pat = VarPatIn c_RDR +d_Pat = VarPatIn d_RDR + +con2tag_RDR, tag2con_RDR, maxtag_RDR :: TyCon -> RdrName + +con2tag_RDR tycon = varUnqual (_PK_ ("con2tag_" ++ occNameString (getOccName tycon) ++ "#")) +tag2con_RDR tycon = varUnqual (_PK_ ("tag2con_" ++ occNameString (getOccName tycon) ++ "#")) +maxtag_RDR tycon = varUnqual (_PK_ ("maxtag_" ++ occNameString (getOccName tycon) ++ "#")) \end{code}