[project @ 1999-11-29 17:34:14 by simonpj]
[ghc-hetmet.git] / ghc / compiler / hsSyn / HsBinds.lhs
index d020b76..c09ccc3 100644 (file)
@@ -1,5 +1,5 @@
 %
-% (c) The GRASP/AQUA Project, Glasgow University, 1992-1996
+% (c) The GRASP/AQUA Project, Glasgow University, 1992-1998
 %
 \section[HsBinds]{Abstract syntax: top-level bindings and signatures}
 
@@ -11,23 +11,21 @@ module HsBinds where
 #include "HsVersions.h"
 
 import {-# SOURCE #-} HsExpr    ( pprExpr, HsExpr )
-import {-# SOURCE #-} HsMatches ( pprMatches, Match, pprGRHSsAndBinds, GRHSsAndBinds )
+import {-# SOURCE #-} HsMatches ( pprMatches, Match, pprGRHSs, GRHSs )
 
 -- friends:
-import HsPragmas       ( GenPragmas, ClassOpPragmas )
 import HsTypes         ( HsType )
 import CoreSyn         ( CoreExpr )
 import PprCore         ()         -- Instances for Outputable
 
 --others:
-import Id              ( DictVar, Id, GenId )
-import Name            ( OccName, NamedThing(..) )
-import BasicTypes      ( RecFlag(..) )
+import Id              ( Id )
+import NameSet         ( NameSet, nameSetToList )
+import BasicTypes      ( RecFlag(..), Fixity )
 import Outputable      
 import Bag
 import SrcLoc          ( SrcLoc )
-import Type            ( GenType )
-import TyVar           ( GenTyVar )
+import Var             ( TyVar )
 \end{code}
 
 %************************************************************************
@@ -45,19 +43,19 @@ grammar.
 Collections of bindings, created by dependency analysis and translation:
 
 \begin{code}
-data HsBinds flexi id pat              -- binders and bindees
+data HsBinds id pat            -- binders and bindees
   = EmptyBinds
 
-  | ThenBinds  (HsBinds flexi id pat)
-               (HsBinds flexi id pat)
+  | ThenBinds  (HsBinds id pat)
+               (HsBinds id pat)
 
-  | MonoBind   (MonoBinds flexi id pat)
+  | MonoBind   (MonoBinds id pat)
                [Sig id]                -- Empty on typechecker output
                RecFlag
 \end{code}
 
 \begin{code}
-nullBinds :: HsBinds flexi id pat -> Bool
+nullBinds :: HsBinds id pat -> Bool
 
 nullBinds EmptyBinds           = True
 nullBinds (ThenBinds b1 b2)    = nullBinds b1 && nullBinds b2
@@ -65,8 +63,8 @@ nullBinds (MonoBind b _ _)    = nullMonoBinds b
 \end{code}
 
 \begin{code}
-instance (Outputable pat, NamedThing id, Outputable id) =>
-               Outputable (HsBinds flexi id pat) where
+instance (Outputable pat, Outputable id) =>
+               Outputable (HsBinds id pat) where
     ppr binds = ppr_binds binds
 
 ppr_binds EmptyBinds = empty
@@ -92,32 +90,33 @@ ppr_binds (MonoBind bind sigs is_rec)
 Global bindings (where clauses)
 
 \begin{code}
-data MonoBinds flexi id pat
+data MonoBinds id pat
   = EmptyMonoBinds
 
-  | AndMonoBinds    (MonoBinds flexi id pat)
-                   (MonoBinds flexi id pat)
+  | AndMonoBinds    (MonoBinds id pat)
+                   (MonoBinds id pat)
 
   | PatMonoBind     pat
-                   (GRHSsAndBinds flexi id pat)
+                   (GRHSs id pat)
                    SrcLoc
 
   | FunMonoBind     id
-                   Bool                        -- True => infix declaration
-                   [Match flexi id pat]        -- must have at least one Match
+                   Bool                -- True => infix declaration
+                   [Match id pat]
                    SrcLoc
 
   | VarMonoBind            id                  -- TRANSLATION
-                   (HsExpr flexi id pat)
+                   (HsExpr id pat)
 
   | CoreMonoBind    id                 -- TRANSLATION
                    CoreExpr            -- No zonking; this is a final CoreExpr with Ids and Types!
 
-  | AbsBinds                   -- Binds abstraction; TRANSLATION
-               [GenTyVar flexi]          -- Type variables
-               [id]                      -- Dicts
-               [([GenTyVar flexi], id, id)]  -- (type variables, polymorphic, momonmorphic) triples
-               (MonoBinds flexi id pat)      -- The "business end"
+  | AbsBinds                           -- Binds abstraction; TRANSLATION
+               [TyVar]                 -- Type variables
+               [id]                    -- Dicts
+               [([TyVar], id, id)]     -- (type variables, polymorphic, momonmorphic) triples
+               NameSet                 -- Set of *polymorphic* variables that have an INLINE pragma
+               (MonoBinds id pat)      -- The "business end"
 
        -- Creates bindings for *new* (polymorphic, overloaded) locals
        -- in terms of *old* (monomorphic, non-overloaded) ones.
@@ -152,28 +151,34 @@ So the desugarer tries to do a better job:
                                      in (fm,gm)
 
 \begin{code}
-nullMonoBinds :: MonoBinds flexi id pat -> Bool
+nullMonoBinds :: MonoBinds id pat -> Bool
 
 nullMonoBinds EmptyMonoBinds        = True
 nullMonoBinds (AndMonoBinds bs1 bs2) = nullMonoBinds bs1 && nullMonoBinds bs2
 nullMonoBinds other_monobind        = False
 
-andMonoBinds :: [MonoBinds flexi id pat] -> MonoBinds flexi id pat
-andMonoBinds binds = foldr AndMonoBinds EmptyMonoBinds binds
+andMonoBinds :: MonoBinds id pat -> MonoBinds id pat -> MonoBinds id pat
+andMonoBinds EmptyMonoBinds mb = mb
+andMonoBinds mb EmptyMonoBinds = mb
+andMonoBinds mb1 mb2 = AndMonoBinds mb1 mb2
+
+andMonoBindList :: [MonoBinds id pat] -> MonoBinds id pat
+andMonoBindList binds = foldr AndMonoBinds EmptyMonoBinds binds
 \end{code}
 
 \begin{code}
-instance (NamedThing id, Outputable id, Outputable pat) =>
-               Outputable (MonoBinds flexi id pat) where
+instance (Outputable id, Outputable pat) =>
+               Outputable (MonoBinds id pat) where
     ppr mbind = ppr_monobind mbind
 
 
+ppr_monobind :: (Outputable id, Outputable pat) => MonoBinds id pat -> SDoc
 ppr_monobind EmptyMonoBinds = empty
 ppr_monobind (AndMonoBinds binds1 binds2)
-      = ($$) (ppr_monobind binds1) (ppr_monobind binds2)
+      = ppr_monobind binds1 $$ ppr_monobind binds2
 
-ppr_monobind (PatMonoBind pat grhss_n_binds locn)
-      = sep [ppr pat, nest 4 (pprGRHSsAndBinds False grhss_n_binds)]
+ppr_monobind (PatMonoBind pat grhss locn)
+      = sep [ppr pat, nest 4 (pprGRHSs False grhss)]
 
 ppr_monobind (FunMonoBind fun inf matches locn)
       = pprMatches (False, ppr fun) matches
@@ -185,12 +190,14 @@ ppr_monobind (VarMonoBind name expr)
 ppr_monobind (CoreMonoBind name expr)
       = sep [ppr name <+> equals, nest 4 (ppr expr)]
 
-ppr_monobind (AbsBinds tyvars dictvars exports val_binds)
-     = ($$) (sep [ptext SLIT("AbsBinds"),
-                 brackets (interpp'SP tyvars),
-                 brackets (interpp'SP dictvars),
-                 brackets (interpp'SP exports)])
-              (nest 4 (ppr val_binds))
+ppr_monobind (AbsBinds tyvars dictvars exports inlines val_binds)
+     = sep [ptext SLIT("AbsBinds"),
+           brackets (interpp'SP tyvars),
+           brackets (interpp'SP dictvars),
+           brackets (sep (punctuate comma (map ppr exports))),
+           brackets (interpp'SP (nameSetToList inlines))]
+       $$
+       nest 4 (ppr val_binds)
 \end{code}
 
 %************************************************************************
@@ -210,50 +217,96 @@ data Sig name
                (HsType name)
                SrcLoc
 
-  | ClassOpSig name                    -- Selector name
-               (Maybe name)            -- Default-method name (if any)
+  | ClassOpSig name            -- Selector name
+               name            -- Default-method name (if any)
+               Bool            -- True <=> there is an explicit, programmer-supplied
+                               -- default declaration in the class decl
                (HsType name)
                SrcLoc
 
   | SpecSig    name            -- specialise a function or datatype ...
-               (HsType name) -- ... to these types
-               (Maybe name)    -- ... maybe using this as the code for it
+               (HsType name)   -- ... to these types
+               SrcLoc
+
+  | InlineSig  name            -- INLINE f
+               (Maybe Int)     -- phase
                SrcLoc
 
-  | InlineSig  name              -- INLINE f
+  | NoInlineSig        name            -- NOINLINE f
+               (Maybe Int)     -- phase
                SrcLoc
 
-  | MagicUnfoldingSig
-               name            -- Associate the "name"d function with
-               FAST_STRING     -- the compiler-builtin unfolding (known
-               SrcLoc          -- by the String name)
+  | SpecInstSig (HsType name)  -- (Class tys); should be a specialisation of the 
+                               -- current instance decl
+               SrcLoc
+
+  | FixSig     (FixitySig name)                -- Fixity declaration
+
+
+data FixitySig name  = FixitySig name Fixity SrcLoc
 \end{code}
 
 \begin{code}
-instance (NamedThing name, Outputable name) => Outputable (Sig name) where
+sigsForMe :: (name -> Bool) -> [Sig name] -> [Sig name]
+sigsForMe f sigs
+  = filter sig_for_me sigs
+  where
+    sig_for_me (Sig         n _ _)       = f n
+    sig_for_me (ClassOpSig  n _ _ _ _)           = f n
+    sig_for_me (SpecSig     n _ _)       = f n
+    sig_for_me (InlineSig   n _   _)     = f n  
+    sig_for_me (NoInlineSig n _   _)     = f n  
+    sig_for_me (SpecInstSig _ _)         = False
+    sig_for_me (FixSig (FixitySig n _ _)) = f n
+
+isFixitySig :: Sig name -> Bool
+isFixitySig (FixSig _) = True
+isFixitySig _         = False
+
+isClassOpSig :: Sig name -> Bool
+isClassOpSig (ClassOpSig _ _ _ _ _) = True
+isClassOpSig _                     = False
+
+isPragSig :: Sig name -> Bool
+       -- Identifies pragmas 
+isPragSig (SpecSig _ _ _)     = True
+isPragSig (InlineSig   _ _ _) = True
+isPragSig (NoInlineSig _ _ _) = True
+isPragSig (SpecInstSig _ _)   = True
+isPragSig other                      = False
+\end{code}
+
+\begin{code}
+instance (Outputable name) => Outputable (Sig name) where
     ppr sig = ppr_sig sig
 
+instance Outputable name => Outputable (FixitySig name) where
+  ppr (FixitySig name fixity loc) = sep [ppr fixity, ppr name]
+
 
 ppr_sig (Sig var ty _)
-      = sep [ppr var <+> ptext SLIT("::"),
-            nest 4 (ppr ty)]
+      = sep [ppr var <+> dcolon, nest 4 (ppr ty)]
 
-ppr_sig (ClassOpSig var _ ty _)
-      = sep [ppr (getOccName var) <+> ptext SLIT("::"),
-            nest 4 (ppr ty)]
+ppr_sig (ClassOpSig var _ _ ty _)
+      = sep [ppr var <+> dcolon, nest 4 (ppr ty)]
 
-ppr_sig (SpecSig var ty using _)
-      = sep [ hsep [text "{-# SPECIALIZE", ppr var, ptext SLIT("::")],
-             nest 4 (hsep [ppr ty, pp_using using, text "#-}"])
+ppr_sig (SpecSig var ty _)
+      = sep [ hsep [text "{-# SPECIALIZE", ppr var, dcolon],
+             nest 4 (ppr ty <+> text "#-}")
        ]
-      where
-       pp_using Nothing   = empty
-       pp_using (Just me) = hsep [char '=', ppr me]
 
-ppr_sig (InlineSig var _)
-        = hsep [text "{-# INLINE", ppr var, text "#-}"]
+ppr_sig (InlineSig var phase _)
+        = hsep [text "{-# INLINE", ppr_phase phase, ppr var, text "#-}"]
+
+ppr_sig (NoInlineSig var phase _)
+        = hsep [text "{-# NOINLINE", ppr_phase phase, ppr var, text "#-}"]
+
+ppr_sig (SpecInstSig ty _)
+      = hsep [text "{-# SPECIALIZE instance", ppr ty, text "#-}"]
+
+ppr_sig (FixSig fix_sig) = ppr fix_sig
 
-ppr_sig (MagicUnfoldingSig var str _)
-      = hsep [text "{-# MAGIC_UNFOLDING", ppr var, ptext str, text "#-}"]
+ppr_phase Nothing = empty
+ppr_phase (Just n) = int n
 \end{code}