Pretty.hs

{-# OPTIONS_GHC -fno-warn-orphans #-}

-- | Compatibility shims for mainland-pretty; the prettyprinting
-- library used by language-c-quote.
module Futhark.CodeGen.Backends.GenericC.Pretty
  ( expText,
    definitionsText,
    typeText,
    idText,
    funcText,
    funcsText,
  )
where

import Data.Text qualified as T
import Language.C.Pretty ()
import Language.C.Syntax qualified as C
import Text.PrettyPrint.Mainland qualified as MPP
import Text.PrettyPrint.Mainland.Class qualified as MPP

render :: MPP.Doc -> String
render = MPP.pretty 8000

-- | Prettyprint a C expression.
expText :: C.Exp -> T.Text
expText = T.pack . render . MPP.ppr

-- | Prettyprint a list of C definitions.
definitionsText :: [C.Definition] -> T.Text
definitionsText = T.unlines . map (T.pack . render . MPP.ppr)

-- | Prettyprint a single C type.
typeText :: C.Type -> T.Text
typeText = T.pack . render . MPP.ppr

-- | Prettyprint a single identifier.
idText :: C.Id -> T.Text
idText = T.pack . render . MPP.ppr

-- | Prettyprint a single function.
funcText :: C.Func -> T.Text
funcText = T.pack . render . MPP.ppr

-- | Prettyprint a list of functions.
funcsText :: [C.Func] -> T.Text
funcsText = T.unlines . map funcText