[project @ 2000-04-03 16:46:41 by simonpj]
[ghc-hetmet.git] / ghc / compiler / utils / Outputable.lhs
index c34404b..19ad666 100644 (file)
@@ -1,5 +1,5 @@
 %
-% (c) The GRASP Project, Glasgow University, 1992-1996
+% (c) The GRASP Project, Glasgow University, 1992-1998
 %
 \section[Outputable]{Classes for pretty-printing}
 
@@ -7,22 +7,26 @@ Defines classes for pretty-printing and forcing, both forms of
 ``output.''
 
 \begin{code}
+{-# OPTIONS -fno-prune-tydecls #-}
+-- Hopefully temporary; 3.02 complained about not being able
+-- to see the consructors for ForeignObj
+
 module Outputable (
        Outputable(..),                 -- Class
 
-       PprStyle, 
+       PprStyle, CodeStyle(..), 
        getPprStyle, withPprStyle, pprDeeper,
        codeStyle, ifaceStyle, userStyle, debugStyle, asmStyle,
        ifPprDebug, ifNotPprForUser,
 
        SDoc,           -- Abstract
-       interppSP, interpp'SP, pprQuotedList,
+       interppSP, interpp'SP, pprQuotedList, pprWithCommas,
        empty, nest,
        text, char, ptext,
        int, integer, float, double, rational,
-       parens, brackets, braces, quotes, doubleQuotes,
-       semi, comma, colon, space, equals,
-       lparen, rparen, lbrack, rbrack, lbrace, rbrace,
+       parens, brackets, braces, quotes, doubleQuotes, angleBrackets,
+       semi, comma, colon, dcolon, space, equals, dot,
+       lparen, rparen, lbrack, rbrack, lbrace, rbrace, underscore,
        (<>), (<+>), hcat, hsep, 
        ($$), ($+$), vcat, 
        sep, cat, 
@@ -30,24 +34,29 @@ module Outputable (
        hang, punctuate,
        speakNth, speakNTimes,
 
-       showSDoc, printSDoc, printErrs, printDump, 
-       printForC, printForAsm, printForIface,
-       pprCols,
+       printSDoc, printErrs, printDump, 
+       printForC, printForAsm, printForIface, printForUser,
+       pprCode, pprCols,
+       showSDoc, showSDocDebug, showSDocIface, showsPrecSDoc, 
+       pprFSAsString,
+
 
        -- error handling
-       pprPanic, pprPanic#, pprError, pprTrace, assertPprPanic,
-       panic, panic#, assertPanic
+       pprPanic, pprPanic#, pprError, pprTrace, assertPprPanic, warnPprTrace,
+       trace, panic, panic#, assertPanic
     ) where
 
 #include "HsVersions.h"
 
+
 import IO              ( Handle, hPutChar, hPutStr, stderr, stdout )
-import CmdLineOpts     ( opt_PprStyle_All, opt_PprStyle_Debug, opt_PprUserLength )
+import CmdLineOpts     ( opt_PprStyle_Debug, opt_PprUserLength )
 import FastString
 import qualified Pretty
 import Pretty          ( Doc, Mode(..), TextDetails(..), fullRender )
-import Util            ( panic, assertPanic, panic# )
-import GlaExts         ( trace )
+import Panic
+import ST              ( runST )
+import Foreign
 \end{code}
 
 
@@ -148,16 +157,19 @@ printSDoc d sty = printDoc PageMode stdout (d sty)
 printErrs :: SDoc -> IO ()
 printErrs doc = printDoc PageMode stderr (final_doc user_style)
              where
-               final_doc = doc $$ text ""
+               final_doc = doc         -- $$ text ""
                user_style = mkUserStyle (PartWay opt_PprUserLength)
 
 printDump :: SDoc -> IO ()
-printDump doc = printDoc PageMode stderr (final_doc PprDebug)
-             where
-               final_doc = doc $$ text ""
+printDump doc = printForUser stdout (doc $$ text "")
+               -- We used to always print in debug style, but I want
+               -- to try the effect of a more user-ish style (unless you
+               -- say -dppr-debug
 
+printForUser :: Handle -> SDoc -> IO ()
+printForUser handle doc = printDoc PageMode handle (doc (mkUserStyle AllTheWay))
 
--- printForC, printForAsm doe what they sound like
+-- printForC, printForAsm do what they sound like
 printForC :: Handle -> SDoc -> IO ()
 printForC handle doc = printDoc LeftMode handle (doc (PprCode CStyle))
 
@@ -169,14 +181,26 @@ printForAsm handle doc = printDoc LeftMode handle (doc (PprCode AsmStyle))
 printForIface :: Handle -> SDoc -> IO ()
 printForIface handle doc = printDoc OneLineMode handle (doc PprInterface)
 
+pprCode :: CodeStyle -> SDoc -> SDoc
+pprCode cs d = withPprStyle (PprCode cs) d
 
+-- Can't make SDoc an instance of Show because SDoc is just a function type
+-- However, Doc *is* an instance of Show
 -- showSDoc just blasts it out as a string
 showSDoc :: SDoc -> String
 showSDoc d = show (d (mkUserStyle AllTheWay))
 
-mkUserStyle depth |  opt_PprStyle_Debug 
-                 || opt_PprStyle_All = PprDebug
-                 |  otherwise        = PprUser depth
+showSDocIface :: SDoc -> String
+showSDocIface doc = showDocWith OneLineMode (doc PprInterface)
+
+showSDocDebug :: SDoc -> String
+showSDocDebug d = show (d PprDebug)
+
+showsPrecSDoc :: Int -> SDoc -> ShowS
+showsPrecSDoc p d = showsPrec p (d (mkUserStyle AllTheWay))
+
+mkUserStyle depth |  opt_PprStyle_Debug = PprDebug
+                 |  otherwise          = PprUser depth
 \end{code}
 
 \begin{code}
@@ -193,8 +217,17 @@ rational n sty = Pretty.rational n
 parens d sty       = Pretty.parens (d sty)
 braces d sty       = Pretty.braces (d sty)
 brackets d sty     = Pretty.brackets (d sty)
-quotes d sty       = Pretty.quotes (d sty)
 doubleQuotes d sty = Pretty.doubleQuotes (d sty)
+angleBrackets d    = char '<' <> d <> char '>'
+
+-- quotes encloses something in single quotes...
+-- but it omits them if the thing ends in a single quote
+-- so that we don't get `foo''.  Instead we just have foo'.
+quotes d sty = case show pp_d of
+                ('\'' : _) -> pp_d
+                other      -> Pretty.quotes pp_d
+            where
+              pp_d = d sty
 
 semi sty   = Pretty.semi
 comma sty  = Pretty.comma
@@ -207,6 +240,9 @@ lbrack sty = Pretty.lbrack
 rbrack sty = Pretty.rbrack
 lbrace sty = Pretty.lbrace
 rbrace sty = Pretty.rbrace
+dcolon sty = Pretty.ptext SLIT("::")
+underscore = char '_'
+dot       = char '.'
 
 nest n d sty    = Pretty.nest n (d sty)
 (<>) d1 d2 sty  = (Pretty.<>)  (d1 sty) (d2 sty)
@@ -256,15 +292,36 @@ instance (Outputable a) => Outputable [a] where
     ppr xs = brackets (fsep (punctuate comma (map ppr xs)))
 
 instance (Outputable a, Outputable b) => Outputable (a, b) where
-    ppr (x,y) =
-      hang (hcat [lparen, ppr x, comma]) 4 ((<>) (ppr y) rparen)
+    ppr (x,y) = parens (sep [ppr x <> comma, ppr y])
+
+instance Outputable a => Outputable (Maybe a) where
+  ppr Nothing = text "Nothing"
+  ppr (Just x) = text "Just" <+> ppr x
 
 -- ToDo: may not be used
 instance (Outputable a, Outputable b, Outputable c) => Outputable (a, b, c) where
     ppr (x,y,z) =
-      parens (sep [ (<>) (ppr x) comma,
-                     (<>) (ppr y) comma,
-                     ppr z ])
+      parens (sep [ppr x <> comma,
+                  ppr y <> comma,
+                  ppr z ])
+
+instance (Outputable a, Outputable b, Outputable c, Outputable d) =>
+        Outputable (a, b, c, d) where
+    ppr (x,y,z,w) =
+      parens (sep [ppr x <> comma,
+                  ppr y <> comma,
+                  ppr z <> comma,
+                  ppr w])
+
+instance Outputable FastString where
+    ppr fs = text (unpackFS fs)                -- Prints an unadorned string,
+                                       -- no double quotes or anything
+
+pprFSAsString :: FastString -> SDoc                    -- The Char instance of Show prints
+pprFSAsString fs = text (showList (unpackFS fs) "")    -- strings with double quotes and escapes
+
+instance Show FastString  where
+    showsPrec p fs = showsPrecSDoc p (ppr fs)
 \end{code}
 
 
@@ -286,10 +343,21 @@ printDoc mode hdl doc
     put (PStr s) next = hPutFS   hdl s >> next 
 
     done = hPutChar hdl '\n'
+
+showDocWith :: Mode -> Doc -> String
+showDocWith mode doc
+  = fullRender mode 100 1.5 put "" doc
+  where
+    put (Chr c)   s  = c:s
+    put (Str s1)  s2 = s1 ++ s2
+    put (PStr s1) s2 = _UNPK_ s1 ++ s2
 \end{code}
 
 
 \begin{code}
+pprWithCommas :: (a -> SDoc) -> [a] -> SDoc
+pprWithCommas pp xs = hsep (punctuate comma (map pp xs))
+
 interppSP  :: Outputable a => [a] -> SDoc
 interppSP  xs = hsep (map ppr xs)
 
@@ -302,8 +370,6 @@ pprQuotedList xs = hsep (punctuate comma (map (quotes . ppr) xs))
 \end{code}
 
 
-
-
 %************************************************************************
 %*                                                                     *
 \subsection{Printing numbers verbally}
@@ -339,27 +405,30 @@ speakNTimes t | t == 1       = ptext SLIT("once")
               | otherwise  = int t <+> ptext SLIT("times")
 \end{code}
 
+
 %************************************************************************
 %*                                                                     *
-\subsection[Utils-errors]{Error handling}
+\subsection{Error handling}
 %*                                                                     *
 %************************************************************************
 
 \begin{code}
-pprPanic heading pretty_msg = panic (show (doc PprDebug))
-                           where
-                             doc = text heading <+> pretty_msg
-
-pprError heading pretty_msg = error (heading++ " " ++ (show pretty_msg))
-
-pprTrace heading pretty_msg = trace (show (doc PprDebug))
-                           where
-                             doc = text heading <+> pretty_msg
+pprPanic :: String -> SDoc -> a
+pprError :: String -> SDoc -> a
+pprTrace :: String -> SDoc -> a -> a
+pprPanic  = pprAndThen panic
+pprError  = pprAndThen error
+pprTrace  = pprAndThen trace
 
 pprPanic# heading pretty_msg = panic# (show (doc PprDebug))
                             where
                               doc = text heading <+> pretty_msg
 
+pprAndThen :: (String -> a) -> String -> SDoc -> a
+pprAndThen cont heading pretty_msg = cont (show (doc PprDebug))
+    where
+     doc = sep [text heading, nest 4 pretty_msg]
+
 assertPprPanic :: String -> Int -> SDoc -> a
 assertPprPanic file line msg
   = panic (show (doc PprDebug))
@@ -368,4 +437,12 @@ assertPprPanic file line msg
                           text file, 
                           text "line", int line], 
                    msg]
+
+warnPprTrace :: Bool -> String -> Int -> SDoc -> a -> a
+warnPprTrace False file line msg x = x
+warnPprTrace True  file line msg x
+  = trace (show (doc PprDebug)) x
+  where
+    doc = sep [hsep [text "WARNING: file", text file, text "line", int line],
+              msg]
 \end{code}