-- needs -package text

{- HetCATS/PrettyPrint.hs

$Id$

Author: Klaus L�ttich

Year: 2002

This class needs to be instantiated for every datastructure in AS_*

for LaTeX and isolatin-1 pretty printing. It is only neccessary to

provide one isolatin-1 printing method for prototypes, but for real

nice output you need to implement every method.

todo:

-}

module PrettyPrint

( renderLatex

, debugRenderLatex

, renderText

, PrettyPrint(..)

, renderLatexVerb

, startTab, endTab, setTab

, printToken_latex

, setTabWithSpaces

--, module PrettyPrint

)

where

import Data.Char (isSpace,isAlphaNum,isDigit)

import Control.Monad.State (State(..),evalState,get,put)

import Data.List (isPrefixOf)

-- debugging

-- import IOExts (trace)

import Id

import Common.Lib.Pretty

import GlobalAnnotations

import LaTeX_funs

-- This type class allows pretty printing of instantiating Datatypes

-- for infomation on the predefined functions see above

class Show a => PrettyPrint a where

printLatex, printLatex0, printText, printText0 :: GlobalAnnos -> a -> Doc

printLatex0 ga a = printText0 ga a

printLatex ga a = printLatex0 ga a

printText ga a = printText0 ga a

{-instance (PrettyPrint a) => PrettyPrint [a] where

printText0 ga l = cat $ map (printText0 ga) l

printLatex0 ga l = cat $ map (printLatex0 ga) l

printText ga l = cat $ map (printText ga) l

printLatex ga l = cat $ map (printLatex ga) l

-}

----------------------------------------------------------------------

-- two Styles for Formatting (Standard is Style PageMode 100 1.5)

latexStyle, textStyle :: Style

textStyle = style {lineLength=80, ribbonsPerLine= 1.19}

-- maximum line length 80 with 67 printable chars (up to 13 indentation chars)

latexStyle = textStyle { ribbonsPerLine = 1.1

, lineLength = calc_line_length "345.0pt"}

-- a LatexRenderingState

-- field indentTabs : for the number of tab

-- stops set those need to be rendererd after every newline.

-- field recentlySet : number of setTab makros indentTabs is only

-- increased if recentlySet is 0

data LRState = LRS { indentTabs :: ![Int]

, recentlySet

, totalTabStops

, setTabsThisLine

, indentTabsWritten :: !Int

, onlyTabs :: !Bool

, isSetLine :: !Bool

, collSpaceIndents :: ![Int]

}

deriving (Show)

-- the initial state for using the state based rendering via LRState

initialLRState :: LRState

initialLRState = LRS { indentTabs = []

, recentlySet = 0

, totalTabStops = 0

, setTabsThisLine = 0

, indentTabsWritten = 0

, onlyTabs = False

, isSetLine = False

, collSpaceIndents = []

}

-- a function that knows how to print LaTeX TextDetails

latex_txt :: TextDetails -> State LRState ShowS -> State LRState ShowS

latex_txt (Chr c) cont

| c == '\n' = do endOfLine

indent <- getIndent

s <- cont

return (showString "\\\\". showChar c . indent . s)

| otherwise = do s <- cont

return (showChar c . s)

latex_txt (Str s1) cont

| null s1 = cont

| all isSpace s1 = do s2 <- cont

return (showChar ' ' . s2)

| otherwise = do setOnlyTabs False

s2 <- cont

return (showString s1 . s2)

latex_txt (PStr s1) cont

| s1 == startTab = do indent <- addTabStop

s2 <- cont

return (indent . s2)

| s1 == endTab = do subTabStop

cont

| s1 == setTab = do setTabStop

s2 <- cont

return (showString s1 . s2)

| setTabWSp

`isPrefixOf`

s1 = do addTabWithSpaces s1

cont

| s1 == "\n" = do endOfLine

indent <- getIndent

s2 <- cont

return (showString "\\\\\n" . indent . s2)

| otherwise = do setOnlyTabs False

s2 <- cont

return (showString s1 . s2)

-- a function that knows how to print LaTeX TextDetails

debug_latex_txt :: TextDetails -> State LRState String -> State LRState String

debug_latex_txt (Chr c) cont

| c == '\n' = do state <- get

endOfLine

indent <- getIndent

s <- cont

return ("\\\\%"++show state++c:(indent s))

| otherwise = do s <- cont

return (c:s)

debug_latex_txt (Str s1) cont

| null s1 = cont

| all isSpace s1 = do s2 <- cont

return ( ' ':s2)

| otherwise = do setOnlyTabs False

s2 <- cont

return (s1 ++ s2)

debug_latex_txt (PStr s1) cont

| s1 == startTab = do indent <- addTabStop

s2 <- cont

return (s1 ++ indent s2)

| s1 == endTab = do subTabStop

s2 <- cont

return (s1 ++ s2)

| s1 == setTab = do setTabStop

s2 <- cont

return (s1 ++ s2)

| setTabWSp

`isPrefixOf`

s1 = do addTabWithSpaces s1

s2 <- cont

return (s1 ++ s2)

| s1 == "\n" = do state <- get

endOfLine

indent <- getIndent

s2 <- cont

return ("\\\\%"++show state++s1++indent s2)

| otherwise = do setOnlyTabs False

s2 <- cont

return (s1 ++ s2)

setOnlyTabs :: Bool -> State LRState ()

setOnlyTabs b = do state <- get

put $ state {onlyTabs = b}

-- a function to produce a String containing the actual tab stops in use

getIndent :: State LRState ShowS

getIndent = do state <- get

let indentTabsSum = foldl (+) 0 (indentTabs state)

put $ state { indentTabsWritten = indentTabsSum

, collSpaceIndents = []

, onlyTabs = True

, totalTabStops = max (totalTabStops state)

(indentTabsSum +

length

(collSpaceIndents state))

}

let indent_fun = foldl (.) id (replicate indentTabsSum

(showString "\\>"))

new_tab_line = foldl space_format id

(collSpaceIndents state)

. showString "\\kill\n"

return $ if null (collSpaceIndents state) then

indent_fun

else

indent_fun . new_tab_line . indent_fun

where space_format :: (ShowS) -> Int -> ShowS

space_format sf1 i = sf1 . showString (replicate i '~')

. showString "\\="

endOfLine :: State LRState ()

endOfLine = do state <- get

put $ state { isSetLine = False

, setTabsThisLine = 0

}

setTabStop :: State LRState ()

setTabStop = State (\state -> ( ()

, let new_setTabsThisLine =

succ $ setTabsThisLine state

in state

{

recentlySet = succ $ recentlySet state

, setTabsThisLine = new_setTabsThisLine

, totalTabStops =

max (totalTabStops state)

(new_setTabsThisLine +

indentTabsWritten state)

, isSetLine = True}

))

addTabWithSpaces :: String -> State LRState ()

addTabWithSpaces s = let delayed_indent :: Int

delayed_indent =

(read . reverse . fst

. span isDigit . tail . reverse) s

in State (\state -> ( ()

, state { collSpaceIndents =

collSpaceIndents state

++ [delayed_indent]

}

))

-- increase the indentTabs in the state by 1

addTabStop :: State LRState ShowS

addTabStop = State (\state -> let (new_indentTabs,newTabs) =

let nT = if isSetLine state

then recentlySet state

else 1

in (condAdd_indentTabs nT, nT)

indentTabsSum = foldl (+) 0

condAdd_indentTabs i =

if i + indentTabsSum

(indentTabs state)

>

totalTabStops state

then indentTabs state

else indentTabs state ++ [i]

new_recentlySet =

if isSetLine state

then 0

else recentlySet state

inTabs nT = foldl (.) id

(replicate nT $ showString "\\>")

(indent_fun,

new_indentTabsWritten) =

if indentTabsSum

new_indentTabs

>

indentTabsWritten state

&& not (isSetLine state)

&& onlyTabs state

then (inTabs newTabs,

newTabs +

indentTabsWritten state)

else (id,

indentTabsWritten state)

in (indent_fun

,state { recentlySet = new_recentlySet

, indentTabs = new_indentTabs

, indentTabsWritten =

new_indentTabsWritten

})

)

-- decrease the indentTabs in the state by 1

subTabStop :: State LRState ()

subTabStop = State (\state -> ( ()

, state {indentTabs =

if null $ indentTabs state

then []

else init $ indentTabs state

}

))

-- |

-- a constant String for starting a LaTeX indentation with tab stop

startTab :: String

startTab = "\\@begT@"

-- |

-- a constant String for releasing a LaTeX indentation with tab stop

endTab :: String

endTab = "\\@endT@"

-- |

-- a constant String to set a tab stop and enable it

setTab :: String

setTab = "\\="

-- | a constant String indicating the start of a space based indentation

setTabWSp :: String

setTabWSp = "\\@setTS@{"

-- | function to set up a space based indentation macro

setTabWithSpaces :: Int -> String

setTabWithSpaces i = (showString setTabWSp . shows i) "}"

-- functions for producing IO printable Strings

renderLatex,renderText,renderLatexVerb :: Maybe Int -> Doc -> String

renderLatex mi d = ((showString "\\begin{hetcasl}\n") .

(evalState (fullRender (mode latexStyle')

(lineLength latexStyle')

(ribbonsPerLine latexStyle')

latex_txt

(return id)

d)

initialLRState)) "\n\\end{hetcasl}\n"

where -- d' = ptext "\\begin{hetcasl}" $+$ d $+$ ptext "\\end{hetcasl}"

latexStyle' = latexStyle {lineLength =

(case mi of

Nothing -> lineLength latexStyle

Just l -> l) }

debugRenderLatex :: Maybe Int -> Doc -> String

debugRenderLatex mi d = evalState (fullRender (mode latexStyle')

(lineLength latexStyle')

(ribbonsPerLine latexStyle')

debug_latex_txt

(return "")

d')

initialLRState

where d' = ptext "\\begin{hetcasl}" $+$ d $+$ ptext "\\end{hetcasl}"

latexStyle' = latexStyle {lineLength =

(case mi of

Nothing -> lineLength latexStyle

Just l -> l) }

-- this lacks some environment starting and closing LaTeX commands

renderLatexVerb mi d = renderStyle textStyle' d'

where d' = text "\\begin{verbatim}" $+$ d $+$ text "\\end{verbatim}"

textStyle' = textStyle {lineLength =

(case mi of

Nothing -> lineLength textStyle

Just l -> l) }

renderText mi d = fullRender (mode textStyle')

(lineLength textStyle')

(ribbonsPerLine textStyle')

string_txt_comp

""

d

where textStyle' = textStyle {lineLength =

maybe (lineLength textStyle) id mi}

string_txt_comp td = case td of

Chr c -> showChar c

Str s -> showString s

PStr s -> showString s

-- moved instance from Id.hs (to avoid cyclic imports via GlobalAnnotations)

instance PrettyPrint Token where

printText0 _ = ptext . tokStr

printLatex0 _ = printToken_latex casl_axiom_latex

printToken_latex :: (String -> Doc) -> TokenOrPlace -> Doc

printToken_latex strConvDoc_fun t =

let s = tokStr t

printToken =

if isPlace t

then strConvDoc_fun . escape_latex

else

if all isAlphaNum s

|| '\\' `elem` escape_latex s

|| head s == '\''

then (\x -> latex_macro "\\Id{"<>x<>latex_macro "}")

. strConvDoc_fun . escape_latex

else (\x -> latex_macro "\\Ax{"<>x<>latex_macro "}")

. strConvDoc_fun

in printToken s

instance PrettyPrint Id where

printText0 ga (Id tops cs _) =

printId printText0 comma brackets ga tops cs

printLatex0 ga (Id tops cs _) =

printId printLatex0 comma_latex brackets_latex ga tops cs

printId :: (forall a . PrettyPrint a => GlobalAnnos -> a -> Doc)

-> Doc -- ^ a comma seperator

-> (Doc -> Doc) -- ^ a function that surrounds the given Doc

-- with brackets

-> GlobalAnnos -> [TokenOrPlace] -> [Id] -> Doc

printId pf comma_fun brackets_fun ga tops ids =

let (toks,places) = splitMixToken tops

glue_tok = fcat . (map (pf ga))

comp = if null ids then empty

else brackets_fun $ fcat $

punctuate comma_fun $ map (pf ga) ids

in fcat [glue_tok toks, comp, glue_tok places]

-- some useful instances ---------------------------------------------

instance PrettyPrint String where

printText0 _ = ptext

printLatex0 _ = error "use a function from module LaTeX_funs"