module PprType(
pprKind, pprParendKind,
pprType, pprParendType,
- pprSourceType, pprPred, pprTheta, pprClassPred,
+ pprPred, pprTheta, pprThetaArrow, pprClassPred,
pprTyVarBndr, pprTyVarBndrs,
-- Junk
-- friends:
-- (PprType can see all the representations it's trying to print)
-import TypeRep ( Type(..), TyNote(..),
- Kind, liftedTypeKind ) -- friend
-import Type ( SourceType(..), isUTyVar, eqKind )
-import TcType ( ThetaType, PredType,
- tcSplitSigmaTy, isPredTy, isDictTy,
+import TypeRep ( Type(..), TyNote(..), PredType(..), TyThing(..), Kind, superKind ) -- friend
+import Type ( typeKind, eqKind )
+import IfaceType ( toIfaceType, toIfacePred, pprParendIfaceType,
+ toIfaceKind, pprParendIfaceKind,
+ getIfaceExt )
+
+import TcType ( ThetaType, PredType,
+ tcSplitSigmaTy, isDictTy,
tcSplitTyConApp_maybe, tcSplitFunTy_maybe
)
import Var ( TyVar, tyVarKind )
import Class ( Class )
-import TyCon ( TyCon, isPrimTyCon, isTupleTyCon, tupleTyConBoxity,
- maybeTyConSingleCon, isEnumerationTyCon,
- tyConArity, tyConName
- )
+import TyCon ( isPrimTyCon, isTupleTyCon, maybeTyConSingleCon, isEnumerationTyCon )
-- others:
-import CmdLineOpts ( opt_PprStyle_RawTypes )
import Maybes ( maybeToBool )
-import Name ( getOccString, getOccName )
-import OccName ( occNameUserString )
+import Name ( NamedThing(..), getOccString )
import Outputable
-import Unique ( Uniquable(..) )
-import Util ( lengthIs )
-import BasicTypes ( IPName(..), tupleParens, ipNameName )
+import BasicTypes ( IPName(..), ipNameName )
import PrelNames -- quite a few *Keys
\end{code}
\begin{code}
pprType, pprParendType :: Type -> SDoc
-pprType ty = ppr_ty tOP_PREC ty
-pprParendType ty = ppr_ty tYCON_PREC ty
+-- To save duplicating type-printing machinery,
+-- we print a type by converting to an IfaceType and printing that
+pprType ty = getIfaceExt $ \ ext ->
+ ppr (toIfaceType ext ty)
+pprParendType ty = getIfaceExt $ \ ext ->
+ pprParendIfaceType (toIfaceType ext ty)
pprKind, pprParendKind :: Kind -> SDoc
-pprKind = pprType
-pprParendKind = pprParendType
+pprKind k = ppr (toIfaceKind k)
+pprParendKind k = pprParendIfaceKind (toIfaceKind k)
pprPred :: PredType -> SDoc
-pprPred = pprSourceType
-
-pprSourceType :: SourceType -> SDoc
-pprSourceType (ClassP clas tys) = pprClassPred clas tys
-pprSourceType (IParam n ty) = hsep [ppr n, dcolon, ppr ty]
-pprSourceType (NType tc tys) = ppr tc <+> hsep (map pprParendType tys)
+pprPred pred = getIfaceExt $ \ ext ->
+ ppr (toIfacePred ext pred)
pprClassPred :: Class -> [Type] -> SDoc
-pprClassPred clas tys = ppr clas <+> hsep (map pprParendType tys)
+pprClassPred clas tys = ppr clas <+> sep (map pprParendType tys)
pprTheta :: ThetaType -> SDoc
-pprTheta theta = parens (hsep (punctuate comma (map pprPred theta)))
+pprTheta theta = parens (sep (punctuate comma (map pprPred theta)))
+
+pprThetaArrow :: ThetaType -> SDoc
+pprThetaArrow theta
+ | null theta = empty
+ | otherwise = parens (sep (punctuate comma (map pprPred theta))) <+> ptext SLIT("=>")
instance Outputable Type where
- ppr ty = pprType ty
+ ppr ty | typeKind ty `eqKind` superKind = pprKind ty
+ | otherwise = pprType ty
-instance Outputable SourceType where
+instance Outputable PredType where
ppr = pprPred
-instance Outputable name => Outputable (IPName name) where
- ppr (Dupable n) = char '?' <> ppr n -- Ordinary implicit parameters
- ppr (Linear n) = char '%' <> ppr n -- Splittable implicit parameters
-\end{code}
-
-
-%************************************************************************
-%* *
-\subsection{Pretty printing}
-%* *
-%************************************************************************
-
-Precedence
-~~~~~~~~~~
-@ppr_ty@ takes an @Int@ that is the precedence of the context.
-The precedence levels are:
-\begin{description}
-\item[tOP_PREC] No parens required.
-\item[fUN_PREC] Left hand argument of a function arrow.
-\item[tYCON_PREC] Argument of a type constructor.
-\end{description}
+instance Outputable name => OutputableBndr (IPName name) where
+ pprBndr _ n = ppr n -- Simple for now
+instance Outputable TyThing where
+ ppr (AnId id) = ptext SLIT("AnId") <+> ppr id
+ ppr (ATyCon tc) = ptext SLIT("ATyCon") <+> ppr tc
+ ppr (AClass cl) = ptext SLIT("AClass") <+> ppr cl
+ ppr (ADataCon dc) = ptext SLIT("ADataCon") <+> ppr dc
-\begin{code}
-tOP_PREC = (0 :: Int) -- type in ParseIface.y
-fUN_PREC = (1 :: Int) -- btype in ParseIface.y
-tYCON_PREC = (2 :: Int) -- atype in ParseIface.y
-
-maybeParen ctxt_prec inner_prec pretty
- | ctxt_prec < inner_prec = pretty
- | otherwise = parens pretty
+instance NamedThing TyThing where -- Can't put this with the type
+ getName (AnId id) = getName id -- decl, because the DataCon instance
+ getName (ATyCon tc) = getName tc -- isn't visible there
+ getName (AClass cl) = getName cl
+ getName (ADataCon dc) = getName dc
\end{code}
-\begin{code}
-ppr_ty :: Int -> Type -> SDoc
-ppr_ty ctxt_prec (TyVarTy tyvar)
- = ppr tyvar
-
-ppr_ty ctxt_prec ty@(TyConApp tycon tys)
- -- KIND CASE; it's of the form (Type x)
- | tycon `hasKey` typeConKey,
- [ty] <- tys
- = -- For kinds, print (Type x) as just x if x is a
- -- type constructor (must be Boxed, Unboxed, AnyBox)
- -- Otherwise print as (Type x)
- case ty of
- TyConApp bx [] -> ppr (getOccName bx) -- Always unqualified
- other -> maybeParen ctxt_prec tYCON_PREC
- (ppr tycon <+> ppr_ty tYCON_PREC ty)
-
- -- USAGE CASE
- | (tycon `hasKey` usOnceTyConKey || tycon `hasKey` usManyTyConKey),
- null tys
- = -- For usages (! and .), always print bare OccName, without pkg/mod/uniq
- ppr (getOccName (tyConName tycon))
-
- -- TUPLE CASE (boxed and unboxed)
- | isTupleTyCon tycon,
- tys `lengthIs` tyConArity tycon -- No magic if partially applied
- = tupleParens (tupleTyConBoxity tycon)
- (sep (punctuate comma (map (ppr_ty tOP_PREC) tys)))
-
- -- LIST CASE
- | tycon `hasKey` listTyConKey,
- [ty] <- tys
- = brackets (ppr_ty tOP_PREC ty)
-
- -- GENERAL CASE
- | otherwise
- = ppr_tc_app ctxt_prec tycon tys
-
-
-ppr_ty ctxt_prec ty@(ForAllTy _ _)
- = getPprStyle $ \ sty ->
- maybeParen ctxt_prec fUN_PREC $
- sep [ ptext SLIT("forall") <+> pp_tyvars sty <> ptext SLIT("."),
- ppr_theta theta,
- ppr_ty tOP_PREC tau
- ]
- where
- (tyvars, theta, tau) = tcSplitSigmaTy ty
-
- pp_tyvars sty = sep (map pprTyVarBndr some_tyvars)
- where
- some_tyvars | userStyle sty && not opt_PprStyle_RawTypes
- = filter (not . isUTyVar) tyvars -- hide uvars from user
- | otherwise
- = tyvars
-
- ppr_theta [] = empty
- ppr_theta theta = pprTheta theta <+> ptext SLIT("=>")
-
-
-ppr_ty ctxt_prec (FunTy ty1 ty2)
- -- we don't want to lose usage annotations or synonyms,
- -- so we mustn't use splitFunTys here.
- = maybeParen ctxt_prec fUN_PREC $
- sep [ ppr_ty fUN_PREC ty1
- , ptext arrow <+> ppr_ty tOP_PREC ty2
- ]
- where arrow | isPredTy ty1 = SLIT("=>")
- | otherwise = SLIT("->")
-
-ppr_ty ctxt_prec (AppTy ty1 ty2)
- = maybeParen ctxt_prec tYCON_PREC $
- ppr_ty fUN_PREC ty1 <+> ppr_ty tYCON_PREC ty2
-
-ppr_ty ctxt_prec (UsageTy u ty)
- = maybeParen ctxt_prec tYCON_PREC $
- ptext SLIT("__u") <+> ppr_ty tYCON_PREC u
- <+> ppr_ty tYCON_PREC ty
- -- fUN_PREC would be logical for u, but it yields a reduce/reduce conflict with AppTy
-
-ppr_ty ctxt_prec (NoteTy (SynNote ty) expansion)
- = ppr_ty ctxt_prec ty
--- = ppr_ty ctxt_prec expansion -- if we don't want to see syntys
-
-ppr_ty ctxt_prec (NoteTy (FTVNote _) ty) = ppr_ty ctxt_prec ty
-
-ppr_ty ctxt_prec (SourceTy (NType tc tys)) = ppr_tc_app ctxt_prec tc tys
-ppr_ty ctxt_prec (SourceTy pred) = braces (pprPred pred)
-
-ppr_tc_app ctxt_prec tc [] = ppr tc
-ppr_tc_app ctxt_prec tc tys = maybeParen ctxt_prec tYCON_PREC
- (sep [ppr tc, nest 4 (sep (map (ppr_ty tYCON_PREC) tys))])
-\end{code}
%************************************************************************
pprTyVarBndr :: TyVar -> SDoc
pprTyVarBndr tyvar
= getPprStyle $ \ sty ->
- if (ifaceStyle sty && not (kind `eqKind` liftedTypeKind)) || debugStyle sty then
+ if debugStyle sty then
hsep [ppr tyvar, dcolon, pprParendKind kind]
-- See comments with ppDcolon in PprCore.lhs
else
TyVarTy _ -> "*"
AppTy fun _ -> getTyDescription fun
FunTy _ res -> '-' : '>' : fun_result res
- TyConApp tycon _ -> occNameUserString (getOccName tycon)
+ NewTcApp tycon _ -> getOccString tycon
+ TyConApp tycon _ -> getOccString tycon
NoteTy (FTVNote _) ty -> getTyDescription ty
NoteTy (SynNote ty1) _ -> getTyDescription ty1
- SourceTy sty -> getSourceTyDescription sty
+ PredTy sty -> getPredTyDescription sty
ForAllTy _ ty -> getTyDescription ty
}
where
fun_result (FunTy _ res) = '>' : fun_result res
fun_result other = getTyDescription other
-getSourceTyDescription (ClassP cl tys) = getOccString cl
-getSourceTyDescription (NType tc tys) = getOccString tc
-getSourceTyDescription (IParam ip ty) = getOccString (ipNameName ip)
+getPredTyDescription (ClassP cl tys) = getOccString cl
+getPredTyDescription (IParam ip ty) = getOccString (ipNameName ip)
\end{code}