--------------------------------------------------------------------------------
-- Copyright © 2011 National Institute of Aerospace / Galois, Inc.
--------------------------------------------------------------------------------

-- | A pretty printer for Copilot specifications as GraphViz/dot graphs.

{-# LANGUAGE Safe #-}
{-# LANGUAGE GADTs #-}

module Copilot.Core.PrettyDot
  ( prettyPrintDot
  , prettyPrintExprDot
  ) where

import Copilot.Core
import Copilot.Core.Type.Show (showWithType, ShowType(..), showType)
import Prelude hiding (id, (<>))
import Text.PrettyPrint.HughesPJ
import Data.List (intersperse)
import Text.Printf

-- | Create a temporary/internal name from an extern variable name.
mkExtTmpVar :: String -> String
mkExtTmpVar :: String -> String
mkExtTmpVar = (String
"ext_" String -> String -> String
forall a. [a] -> [a] -> [a]
++)

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

-- | Pretty print a Copilot Expression as a GraphViz graph part.
--
-- See the
-- <https://github.com/Copilot-Language/copilot-discussion/tree/master/TutorialAndDevGuide/DevGuide development guide> for details.
ppExprDot :: Int       -- ^ Index or ID of the next node in the graph.
          -> Int       -- ^ Index or ID of the parent node in the graph.
          -> Bool      -- ^ Mark externs with the prefix @externV:@.
          -> Expr a    -- ^ The expression to pretty print.
          -> (Doc,Int)
ppExprDot :: Int -> Int -> Bool -> Expr a -> (Doc, Int)
ppExprDot Int
ii Int
pere Bool
bb Expr a
e0 = case Expr a
e0 of
  Const Type a
t a
x                  -> (String -> Doc
text (String -> String -> String -> String
forall r. PrintfType r => String -> r
printf String
"%s [label=\"const: %s\",color=red1, style=filled]\n" (Int -> String
forall a. Show a => a -> String
show Int
ii::String) ((ShowType -> Type a -> a -> String
forall a. ShowType -> Type a -> a -> String
showWithType ShowType
Haskell Type a
t a
x)::String) )
                  Doc -> Doc -> Doc
<> String -> Doc
text (String -> String -> String -> String
forall r. PrintfType r => String -> r
printf String
"%s -> %s\n" (Int -> String
forall a. Show a => a -> String
show Int
pere::String) (Int -> String
forall a. Show a => a -> String
show Int
ii::String)),Int
iiInt -> Int -> Int
forall a. Num a => a -> a -> a
+Int
1)
  Drop Type a
_ DropIdx
0 Int
id                -> (String -> Doc
text (String -> String -> String -> String
forall r. PrintfType r => String -> r
printf String
"%s [label=\"stream: %s\",color=crimson, style=filled]\n" (Int -> String
forall a. Show a => a -> String
show Int
ii::String) (Int -> String
forall a. Show a => a -> String
show Int
id::String) )
                  Doc -> Doc -> Doc
<> String -> Doc
text (String -> String -> String -> String
forall r. PrintfType r => String -> r
printf String
"%s -> %s\n" (Int -> String
forall a. Show a => a -> String
show Int
pere::String) (Int -> String
forall a. Show a => a -> String
show Int
ii::String)),Int
iiInt -> Int -> Int
forall a. Num a => a -> a -> a
+Int
1)
  Drop Type a
_ DropIdx
i Int
id                ->  (String -> Doc
text (String -> String -> String -> String -> String
forall r. PrintfType r => String -> r
printf String
"%s [label=\"drop %s: \nstream: %s\",color=crimson, style=filled]\n" (Int -> String
forall a. Show a => a -> String
show Int
ii::String) (DropIdx -> String
forall a. Show a => a -> String
show DropIdx
i::String) (Int -> String
forall a. Show a => a -> String
show Int
id::String) )
                  Doc -> Doc -> Doc
<> String -> Doc
text (String -> String -> String -> String
forall r. PrintfType r => String -> r
printf String
"%s -> %s\n" (Int -> String
forall a. Show a => a -> String
show Int
pere::String) (Int -> String
forall a. Show a => a -> String
show Int
ii::String)),Int
iiInt -> Int -> Int
forall a. Num a => a -> a -> a
+Int
1)
  ExternVar Type a
_ String
name Maybe [a]
_         -> (if Bool
bb
                                   then (String -> Doc
text (String -> String -> String -> String
forall r. PrintfType r => String -> r
printf String
"%s [label=\"externV: %s\",color=cyan1, style=filled]\n" (Int -> String
forall a. Show a => a -> String
show Int
ii::String) (String
name::String)) Doc -> Doc -> Doc
<> String -> Doc
text (String -> String -> String -> String
forall r. PrintfType r => String -> r
printf String
"%s -> %s\n" (Int -> String
forall a. Show a => a -> String
show Int
pere::String) (Int -> String
forall a. Show a => a -> String
show Int
ii::String)))
                                   else (String -> Doc
text (String -> String -> String -> String
forall r. PrintfType r => String -> r
printf String
"%s [label=\"%s\",color=cyan1, style=filled]\n" (Int -> String
forall a. Show a => a -> String
show Int
ii::String) (String -> String
mkExtTmpVar String
name))      Doc -> Doc -> Doc
<> String -> Doc
text (String -> String -> String -> String
forall r. PrintfType r => String -> r
printf String
"%s -> %s\n" (Int -> String
forall a. Show a => a -> String
show Int
pere::String) (Int -> String
forall a. Show a => a -> String
show Int
ii::String)))
                  ,Int
iiInt -> Int -> Int
forall a. Num a => a -> a -> a
+Int
1)
  Local Type a
_ Type a
_ String
name Expr a
e1 Expr a
e2       -> let (Doc
r1, Int
i1) = Int -> Int -> Bool -> Expr a -> (Doc, Int)
forall a. Int -> Int -> Bool -> Expr a -> (Doc, Int)
ppExprDot (Int
iiInt -> Int -> Int
forall a. Num a => a -> a -> a
+Int
2) (Int
iiInt -> Int -> Int
forall a. Num a => a -> a -> a
+Int
1) Bool
bb Expr a
e1
                                in let (Doc
r2, Int
i2) = Int -> Int -> Bool -> Expr a -> (Doc, Int)
forall a. Int -> Int -> Bool -> Expr a -> (Doc, Int)
ppExprDot (Int
i1) Int
ii Bool
bb Expr a
e2
                                in (String -> Doc
text (String -> String -> String
forall r. PrintfType r => String -> r
printf String
"%s [label=\"local:\",color=blue, style=filled]\n" (Int -> String
forall a. Show a => a -> String
show Int
ii::String) )
                  Doc -> Doc -> Doc
<> String -> Doc
text (String -> String -> String -> String
forall r. PrintfType r => String -> r
printf String
"%s -> %s\n" (Int -> String
forall a. Show a => a -> String
show Int
pere::String) (Int -> String
forall a. Show a => a -> String
show Int
ii::String))
                  Doc -> Doc -> Doc
<> String -> Doc
text (String -> String -> String -> String
forall r. PrintfType r => String -> r
printf String
"%s [label=\"def: %s\",color=blue, style=filled]\n" ((Int -> String
forall a. Show a => a -> String
show (Int -> String) -> Int -> String
forall a b. (a -> b) -> a -> b
$ Int
iiInt -> Int -> Int
forall a. Num a => a -> a -> a
+Int
1)::String) (String
name::String) )
                  Doc -> Doc -> Doc
<> String -> Doc
text (String -> String -> String -> String
forall r. PrintfType r => String -> r
printf String
"%s -> %s\n" (Int -> String
forall a. Show a => a -> String
show Int
ii::String) (Int -> String
forall a. Show a => a -> String
show (Int -> String) -> Int -> String
forall a b. (a -> b) -> a -> b
$ Int
iiInt -> Int -> Int
forall a. Num a => a -> a -> a
+Int
1::String))
                  Doc -> Doc -> Doc
<> Doc
r1
                  Doc -> Doc -> Doc
<> Doc
r2 ,Int
i2)

  Var Type a
_ String
name                 -> (String -> Doc
text (String -> String -> String -> String
forall r. PrintfType r => String -> r
printf String
"%s [label=\"var: %s\",color=blue, style=filled]\n" (Int -> String
forall a. Show a => a -> String
show Int
ii::String) (String
name::String) )
                  Doc -> Doc -> Doc
<> String -> Doc
text (String -> String -> String -> String
forall r. PrintfType r => String -> r
printf String
"%s -> %s\n" (Int -> String
forall a. Show a => a -> String
show Int
pere::String) (Int -> String
forall a. Show a => a -> String
show Int
ii::String)),Int
iiInt -> Int -> Int
forall a. Num a => a -> a -> a
+Int
1)

  Op1 Op1 a a
op Expr a
e                   -> let (Doc
r1,Int
i1) = Int -> Int -> Bool -> Expr a -> (Doc, Int)
forall a. Int -> Int -> Bool -> Expr a -> (Doc, Int)
ppExprDot (Int
iiInt -> Int -> Int
forall a. Num a => a -> a -> a
+Int
1) Int
ii Bool
bb Expr a
e
           in (String -> Doc
text (String -> String -> String -> String
forall r. PrintfType r => String -> r
printf String
"%s [label=\"op1: %s\",color=green4, style=filled]\n" (Int -> String
forall a. Show a => a -> String
show Int
ii::String) (Op1 a a -> String
forall a b. Op1 a b -> String
ppOp1 Op1 a a
op::String))
                  Doc -> Doc -> Doc
<> String -> Doc
text (String -> String -> String -> String
forall r. PrintfType r => String -> r
printf String
"%s -> %s\n" (Int -> String
forall a. Show a => a -> String
show Int
pere::String) (Int -> String
forall a. Show a => a -> String
show Int
ii::String))
                  Doc -> Doc -> Doc
<> Doc
r1,Int
i1)

  Op2 Op2 a b a
op Expr a
e1 Expr b
e2               -> let (Doc
r1,Int
i1) = Int -> Int -> Bool -> Expr a -> (Doc, Int)
forall a. Int -> Int -> Bool -> Expr a -> (Doc, Int)
ppExprDot (Int
iiInt -> Int -> Int
forall a. Num a => a -> a -> a
+Int
1) Int
ii Bool
bb Expr a
e1
                                in let (Doc
r2,Int
i2) = Int -> Int -> Bool -> Expr b -> (Doc, Int)
forall a. Int -> Int -> Bool -> Expr a -> (Doc, Int)
ppExprDot Int
i1 Int
ii Bool
bb Expr b
e2
           in (String -> Doc
text (String -> String -> String -> String
forall r. PrintfType r => String -> r
printf String
"%s [label=\"op2: %s\",color=green4, style=filled]\n" (Int -> String
forall a. Show a => a -> String
show Int
ii::String) (Op2 a b a -> String
forall a b c. Op2 a b c -> String
ppOp2 Op2 a b a
op::String))
                  Doc -> Doc -> Doc
<> String -> Doc
text (String -> String -> String -> String
forall r. PrintfType r => String -> r
printf String
"%s -> %s\n" (Int -> String
forall a. Show a => a -> String
show Int
pere::String) (Int -> String
forall a. Show a => a -> String
show Int
ii::String))
                  Doc -> Doc -> Doc
<> Doc
r1
                  Doc -> Doc -> Doc
<> Doc
r2 ,Int
i2)
  Op3 Op3 a b c a
op Expr a
e1 Expr b
e2 Expr c
e3            -> let (Doc
r1,Int
i1) = Int -> Int -> Bool -> Expr a -> (Doc, Int)
forall a. Int -> Int -> Bool -> Expr a -> (Doc, Int)
ppExprDot (Int
iiInt -> Int -> Int
forall a. Num a => a -> a -> a
+Int
1) Int
ii Bool
bb Expr a
e1
                                in let (Doc
r2,Int
i2) = Int -> Int -> Bool -> Expr b -> (Doc, Int)
forall a. Int -> Int -> Bool -> Expr a -> (Doc, Int)
ppExprDot Int
i1 Int
ii Bool
bb Expr b
e2
                                in let (Doc
r3,Int
i3) = Int -> Int -> Bool -> Expr c -> (Doc, Int)
forall a. Int -> Int -> Bool -> Expr a -> (Doc, Int)
ppExprDot Int
i2 Int
ii Bool
bb Expr c
e3
           in (String -> Doc
text (String -> String -> String -> String
forall r. PrintfType r => String -> r
printf String
"%s [label=\"op3: %s\",color=green4, style=filled]\n" (Int -> String
forall a. Show a => a -> String
show Int
ii::String) (Op3 a b c a -> String
forall a b c d. Op3 a b c d -> String
ppOp3 Op3 a b c a
op::String))
                  Doc -> Doc -> Doc
<> String -> Doc
text (String -> String -> String -> String
forall r. PrintfType r => String -> r
printf String
"%s -> %s\n" (Int -> String
forall a. Show a => a -> String
show Int
pere::String) (Int -> String
forall a. Show a => a -> String
show Int
ii::String))
                  Doc -> Doc -> Doc
<> Doc
r1
                  Doc -> Doc -> Doc
<> Doc
r2
                  Doc -> Doc -> Doc
<> Doc
r3 ,Int
i3)

  Label Type a
_ String
s Expr a
e                -> let (Doc
r1,Int
i1) = Int -> Int -> Bool -> Expr a -> (Doc, Int)
forall a. Int -> Int -> Bool -> Expr a -> (Doc, Int)
ppExprDot (Int
iiInt -> Int -> Int
forall a. Num a => a -> a -> a
+Int
1) Int
ii Bool
bb Expr a
e
           in (String -> Doc
text (String -> String -> String -> String
forall r. PrintfType r => String -> r
printf String
"%s [label=\"label: %s\",color=plum, style=filled]\n" (Int -> String
forall a. Show a => a -> String
show Int
ii::String) (String
s::String))
                  Doc -> Doc -> Doc
<> String -> Doc
text (String -> String -> String -> String
forall r. PrintfType r => String -> r
printf String
"%s -> %s\n" (Int -> String
forall a. Show a => a -> String
show Int
pere::String) (Int -> String
forall a. Show a => a -> String
show Int
ii::String))
                  Doc -> Doc -> Doc
<> Doc
r1,Int
i1)

-- | Pretty print an untyped Copilot Expression as a graphiViz graph part.
--
-- See the
-- <https://github.com/Copilot-Language/copilot-discussion/tree/master/TutorialAndDevGuide/DevGuide development guide> for details.
ppUExpr :: Int        -- ^ Index or ID of the next node in the graph.
        -> Int        -- ^ Index or ID of the parent node in the graph.
        -> Bool       -- ^ Mark externs with the prefix @externV:@.
        -> UExpr      -- ^ The expression to pretty print.
        -> (Doc, Int)
ppUExpr :: Int -> Int -> Bool -> UExpr -> (Doc, Int)
ppUExpr Int
i Int
pere Bool
bb UExpr { uExprExpr :: ()
uExprExpr = Expr a
e0 } = Int -> Int -> Bool -> Expr a -> (Doc, Int)
forall a. Int -> Int -> Bool -> Expr a -> (Doc, Int)
ppExprDot Int
i Int
pere Bool
bb Expr a
e0

-- | Pretty print a unary operator.
ppOp1 :: Op1 a b -> String
ppOp1 :: Op1 a b -> String
ppOp1 Op1 a b
op = case Op1 a b
op of
  Op1 a b
Not      -> String
"not"
  Abs Type a
_    -> String
"abs"
  Sign Type a
_   -> String
"signum"
  Recip Type a
_  -> String
"recip"
  Exp Type a
_    -> String
"exp"
  Sqrt Type a
_   -> String
"sqrt"
  Log Type a
_    -> String
"log"
  Sin Type a
_    -> String
"sin"
  Tan Type a
_    -> String
"tan"
  Cos Type a
_    -> String
"cos"
  Asin Type a
_   -> String
"asin"
  Atan Type a
_   -> String
"atan"
  Acos Type a
_   -> String
"acos"
  Sinh Type a
_   -> String
"sinh"
  Tanh Type a
_   -> String
"tanh"
  Cosh Type a
_   -> String
"cosh"
  Asinh Type a
_  -> String
"asinh"
  Atanh Type a
_  -> String
"atanh"
  Acosh Type a
_  -> String
"acosh"
  BwNot Type a
_  -> String
"~"
  Cast Type a
_ Type b
_ -> String
"(cast)"

-- | Pretty print a binary operator.
ppOp2 :: Op2 a b c -> String
ppOp2 :: Op2 a b c -> String
ppOp2 Op2 a b c
op = case Op2 a b c
op of
  Op2 a b c
And          -> String
"&&"
  Op2 a b c
Or           -> String
"||"
  Add      Type a
_   -> String
"+"
  Sub      Type a
_   -> String
"-"
  Mul      Type a
_   -> String
"*"
  Div      Type a
_   -> String
"div"
  Mod      Type a
_   -> String
"mod"
  Fdiv     Type a
_   -> String
"/"
  Pow      Type a
_   -> String
"**"
  Logb     Type a
_   -> String
"logBase"
  Eq       Type a
_   -> String
"=="
  Ne       Type a
_   -> String
"/="
  Le       Type a
_   -> String
"<="
  Ge       Type a
_   -> String
">="
  Lt       Type a
_   -> String
"<"
  Gt       Type a
_   -> String
">"
  BwAnd    Type a
_   -> String
"&"
  BwOr     Type a
_   -> String
"|"
  BwXor    Type a
_   -> String
"^"
  BwShiftL Type a
_ Type b
_ -> String
"<<"
  BwShiftR Type a
_ Type b
_ -> String
">>"

-- | Pretty print a ternary operator.
ppOp3 :: Op3 a b c d -> String
ppOp3 :: Op3 a b c d -> String
ppOp3 Op3 a b c d
op = case Op3 a b c d
op of
  Mux Type b
_    -> String
"mux"

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

-- | Pretty print a stream as a GraphViz graph part.
ppStream :: Int        -- ^ Index or ID of the next node in the graph.
         -> Stream     -- ^ Stream to pretty print
         -> (Doc, Int)
ppStream :: Int -> Stream -> (Doc, Int)
ppStream Int
i
  Stream
    { streamId :: Stream -> Int
streamId       = Int
id
    , streamBuffer :: ()
streamBuffer   = [a]
buffer
    , streamExpr :: ()
streamExpr     = Expr a
e
    , streamExprType :: ()
streamExprType = Type a
t
    }
      =
  (String -> Doc
text (String -> String -> String -> String -> String
forall r. PrintfType r => String -> r
printf String
"%s [label=\"stream: %s\ntype: %s\",color=mediumblue, style=filled]\n" (Int -> String
forall a. Show a => a -> String
show Int
i::String) (Int -> String
forall a. Show a => a -> String
show Int
id::String) (Type a -> String
forall a. Type a -> String
showType Type a
t::String))
    Doc -> Doc -> Doc
<> String -> Doc
text (String -> String -> String
forall r. PrintfType r => String -> r
printf String
"%s [label=\"++\",color=yellow, style=filled]\n" ((Int -> String
forall a. Show a => a -> String
show (Int -> String) -> Int -> String
forall a b. (a -> b) -> a -> b
$ Int
iInt -> Int -> Int
forall a. Num a => a -> a -> a
+Int
1)::String))
    Doc -> Doc -> Doc
<> String -> Doc
text (String -> String -> String -> String
forall r. PrintfType r => String -> r
printf String
"%s -> %s\n" (Int -> String
forall a. Show a => a -> String
show Int
i::String) ((Int -> String
forall a. Show a => a -> String
show (Int -> String) -> Int -> String
forall a b. (a -> b) -> a -> b
$ Int
iInt -> Int -> Int
forall a. Num a => a -> a -> a
+Int
1)::String))
    Doc -> Doc -> Doc
<> String -> Doc
text (String -> String -> String -> String
forall r. PrintfType r => String -> r
printf String
"%s [label=\"[%s]\",color=green, style=filled]\n" ((Int -> String
forall a. Show a => a -> String
show (Int -> String) -> Int -> String
forall a b. (a -> b) -> a -> b
$ Int
iInt -> Int -> Int
forall a. Num a => a -> a -> a
+Int
2)::String) (([String] -> String
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
concat ([String] -> String) -> [String] -> String
forall a b. (a -> b) -> a -> b
$ String -> [String] -> [String]
forall a. a -> [a] -> [a]
intersperse String
"," ([String] -> [String]) -> [String] -> [String]
forall a b. (a -> b) -> a -> b
$ (a -> String) -> [a] -> [String]
forall a b. (a -> b) -> [a] -> [b]
map (ShowType -> Type a -> a -> String
forall a. ShowType -> Type a -> a -> String
showWithType ShowType
Haskell Type a
t) [a]
buffer )) ::String)
    Doc -> Doc -> Doc
<> String -> Doc
text (String -> String -> String -> String
forall r. PrintfType r => String -> r
printf String
"%s -> %s\n" (Int -> String
forall a. Show a => a -> String
show (Int
iInt -> Int -> Int
forall a. Num a => a -> a -> a
+Int
1)::String) ((Int -> String
forall a. Show a => a -> String
show (Int -> String) -> Int -> String
forall a b. (a -> b) -> a -> b
$ Int
iInt -> Int -> Int
forall a. Num a => a -> a -> a
+Int
2)::String))
    Doc -> Doc -> Doc
<> Doc
r1, Int
i1)
    where (Doc
r1, Int
i1) = Int -> Int -> Bool -> Expr a -> (Doc, Int)
forall a. Int -> Int -> Bool -> Expr a -> (Doc, Int)
ppExprDot (Int
iInt -> Int -> Int
forall a. Num a => a -> a -> a
+Int
3) (Int
iInt -> Int -> Int
forall a. Num a => a -> a -> a
+Int
1) Bool
True Expr a
e
--------------------------------------------------------------------------------

-- | Pretty print a trigger as a GraphViz graph part.
ppTrigger :: Int        -- ^ Index or ID of the next node in the graph.
          -> Trigger    -- ^ Trigger to pretty print
          -> (Doc, Int)
ppTrigger :: Int -> Trigger -> (Doc, Int)
ppTrigger Int
i
  Trigger
    { triggerName :: Trigger -> String
triggerName  = String
name
    , triggerGuard :: Trigger -> Expr Bool
triggerGuard = Expr Bool
e
    , triggerArgs :: Trigger -> [UExpr]
triggerArgs  = [UExpr]
args }
  =  ( String -> Doc
text (String -> String -> String -> String
forall r. PrintfType r => String -> r
printf String
"%s [label=\"trigger: %s\",color=mediumblue, style=filled]\n" (Int -> String
forall a. Show a => a -> String
show Int
i::String) (String
name::String) )
  Doc -> Doc -> Doc
<> String -> Doc
text (String -> String -> String
forall r. PrintfType r => String -> r
printf String
"%s [label=\"guard\",color=yellow, style=filled]\n" ((Int -> String
forall a. Show a => a -> String
show (Int -> String) -> Int -> String
forall a b. (a -> b) -> a -> b
$ Int
iInt -> Int -> Int
forall a. Num a => a -> a -> a
+Int
1)::String))
  Doc -> Doc -> Doc
<> String -> Doc
text (String -> String -> String -> String
forall r. PrintfType r => String -> r
printf String
"%s -> %s\n" (Int -> String
forall a. Show a => a -> String
show Int
i::String) ((Int -> String
forall a. Show a => a -> String
show (Int -> String) -> Int -> String
forall a b. (a -> b) -> a -> b
$ Int
iInt -> Int -> Int
forall a. Num a => a -> a -> a
+Int
1)::String))
  Doc -> Doc -> Doc
<> Doc
r1
  Doc -> Doc -> Doc
<> String -> Doc
text (String -> String -> String
forall r. PrintfType r => String -> r
printf String
"%s [label=\"args\",color=yellow, style=filled]\n" (Int -> String
forall a. Show a => a -> String
show Int
i1::String))
  Doc -> Doc -> Doc
<> String -> Doc
text (String -> String -> String -> String
forall r. PrintfType r => String -> r
printf String
"%s -> %s\n" (Int -> String
forall a. Show a => a -> String
show Int
i::String) (Int -> String
forall a. Show a => a -> String
show Int
i1::String))
  Doc -> Doc -> Doc
<>  ([Doc] -> Doc
vcat ([Doc]
r2))
  ,Int
i2)
  where
    (Doc
r1, Int
i1) = Int -> Int -> Bool -> Expr Bool -> (Doc, Int)
forall a. Int -> Int -> Bool -> Expr a -> (Doc, Int)
ppExprDot (Int
iInt -> Int -> Int
forall a. Num a => a -> a -> a
+Int
2) (Int
iInt -> Int -> Int
forall a. Num a => a -> a -> a
+Int
1) Bool
True Expr Bool
e
    ([Doc]
r2, Int
i2) = Int -> Int -> Bool -> [UExpr] -> ([Doc], Int)
ppUExprL (Int
i1Int -> Int -> Int
forall a. Num a => a -> a -> a
+Int
1) (Int
i1) Bool
True [UExpr]
args

-- | Pretty print a list of untyped Copilot Expressions as a GraphViz graph
-- part.
ppUExprL :: Int           -- ^ Index or ID of the next node in the graph.
         -> Int           -- ^ Index or ID of the parent node in the graph.
         -> Bool          -- ^ Mark externs with the prefix @externV:@.
         -> [UExpr]       -- ^ The list of expressions to pretty print.
         -> ([Doc], Int)
ppUExprL :: Int -> Int -> Bool -> [UExpr] -> ([Doc], Int)
ppUExprL Int
i Int
_ Bool
_ [] = ([], Int
i)

ppUExprL Int
i Int
pere Bool
bb (UExpr
a:[UExpr]
b) = ((Doc
r1Doc -> [Doc] -> [Doc]
forall a. a -> [a] -> [a]
:[Doc]
r2), Int
i2)
  where
    (Doc
r1, Int
i1) = Int -> Int -> Bool -> UExpr -> (Doc, Int)
ppUExpr Int
i Int
pere Bool
bb UExpr
a
    ([Doc]
r2, Int
i2) = Int -> Int -> Bool -> [UExpr] -> ([Doc], Int)
ppUExprL Int
i1 Int
pere Bool
bb [UExpr]
b

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

-- | Pretty print an observer as a GraphViz graph part.
ppObserver :: Int         -- ^ Index or ID of the next node in the graph.
           -> Observer    -- ^ Observer to pretty print
           -> (Doc, Int)
ppObserver :: Int -> Observer -> (Doc, Int)
ppObserver Int
i
  Observer
    { observerName :: Observer -> String
observerName     = String
name
    , observerExpr :: ()
observerExpr     = Expr a
e }
  =
  (String -> Doc
text (String -> String -> String -> String
forall r. PrintfType r => String -> r
printf String
"%s [label=\"observer: \n%s\",color=mediumblue, style=filled]\n" (Int -> String
forall a. Show a => a -> String
show Int
i::String) String
name::String)
  Doc -> Doc -> Doc
<> Doc
r1, Int
i1)
  where (Doc
r1, Int
i1) = Int -> Int -> Bool -> Expr a -> (Doc, Int)
forall a. Int -> Int -> Bool -> Expr a -> (Doc, Int)
ppExprDot (Int
iInt -> Int -> Int
forall a. Num a => a -> a -> a
+Int
1) Int
i Bool
True Expr a
e
--------------------------------------------------------------------------------

-- | Pretty print a property as a GraphViz graph part.
ppProperty :: Int         -- ^ Index or ID of the next node in the graph.
           -> Property    -- ^ Property to pretty print
           -> (Doc, Int)
ppProperty :: Int -> Property -> (Doc, Int)
ppProperty Int
i
  Property
    { propertyName :: Property -> String
propertyName     = String
name
    , propertyExpr :: Property -> Expr Bool
propertyExpr     = Expr Bool
e }
  =
  (String -> Doc
text (String -> String -> String -> String
forall r. PrintfType r => String -> r
printf String
"%s [label=\"property: \n%s\",color=mediumblue, style=filled]\n" (Int -> String
forall a. Show a => a -> String
show Int
i::String) String
name::String)
  Doc -> Doc -> Doc
<> Doc
r1, Int
i1)
  where (Doc
r1, Int
i1) = Int -> Int -> Bool -> Expr Bool -> (Doc, Int)
forall a. Int -> Int -> Bool -> Expr a -> (Doc, Int)
ppExprDot (Int
iInt -> Int -> Int
forall a. Num a => a -> a -> a
+Int
1) Int
i Bool
True Expr Bool
e

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


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

-- | Pretty print a list of streams as a GraphViz graph part.
ppStreamL :: Int        -- ^ Index or ID of the next node in the graph.
          -> [Stream]   -- ^ List of streams to pretty print
          -> (Doc, Int)
ppStreamL :: Int -> [Stream] -> (Doc, Int)
ppStreamL Int
i [] = (Doc
empty,Int
i)

ppStreamL Int
i (Stream
a:[Stream]
b) = ((Doc
s1Doc -> Doc -> Doc
$$Doc
s2),(Int
i2))
  where
    (Doc
s1,Int
i1) = Int -> Stream -> (Doc, Int)
ppStream Int
i Stream
a
    (Doc
s2,Int
i2) = Int -> [Stream] -> (Doc, Int)
ppStreamL Int
i1 [Stream]
b

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

-- | Pretty print a list of triggers as a GraphViz graph part.
ppTriggerL :: Int        -- ^ Index or ID of the next node in the graph.
           -> [Trigger]  -- ^ List of triggers to pretty print
           -> (Doc, Int)
ppTriggerL :: Int -> [Trigger] -> (Doc, Int)
ppTriggerL Int
i [] = (Doc
empty,Int
i)

ppTriggerL Int
i (Trigger
a:[Trigger]
b) = ((Doc
s1Doc -> Doc -> Doc
$$Doc
s2),(Int
i2))
  where
    (Doc
s1,Int
i1) = Int -> Trigger -> (Doc, Int)
ppTrigger Int
i Trigger
a
    (Doc
s2,Int
i2) = Int -> [Trigger] -> (Doc, Int)
ppTriggerL Int
i1 [Trigger]
b

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

-- | Pretty print a list of observers as a GraphViz graph part.
ppObserverL :: Int         -- ^ Index or ID of the next node in the graph.
            -> [Observer]  -- ^ List of observers to pretty print
            -> (Doc, Int)
ppObserverL :: Int -> [Observer] -> (Doc, Int)
ppObserverL Int
i [] = (Doc
empty,Int
i)

ppObserverL Int
i (Observer
a:[Observer]
b) = ((Doc
s1Doc -> Doc -> Doc
$$Doc
s2),(Int
i2))
  where
    (Doc
s1,Int
i1) = Int -> Observer -> (Doc, Int)
ppObserver Int
i Observer
a
    (Doc
s2,Int
i2) = Int -> [Observer] -> (Doc, Int)
ppObserverL Int
i1 [Observer]
b


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

-- | Pretty print a list of properties as a GraphViz graph part.
ppPropertyL :: Int         -- ^ Index or ID of the next node in the graph.
            -> [Property]  -- ^ List of properties to pretty print
            -> (Doc, Int)
ppPropertyL :: Int -> [Property] -> (Doc, Int)
ppPropertyL Int
i [] = (Doc
empty,Int
i)

ppPropertyL Int
i (Property
a:[Property]
b) = ((Doc
s1Doc -> Doc -> Doc
$$Doc
s2),(Int
i2))
  where
    (Doc
s1,Int
i1) = Int -> Property -> (Doc, Int)
ppProperty Int
i Property
a
    (Doc
s2,Int
i2) = Int -> [Property] -> (Doc, Int)
ppPropertyL Int
i1 [Property]
b

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

-- | Pretty-print a Copilot specification as a GraphViz/dot graph.
ppSpecDot :: Int        -- ^ Index or ID of the next node in the graph.
          -> Spec       -- ^ Spec to pretty print.
          -> (Doc, Int)
ppSpecDot :: Int -> Spec -> (Doc, Int)
ppSpecDot Int
i Spec
spec =
  ((Doc
aa Doc -> Doc -> Doc
$$ Doc
cs Doc -> Doc -> Doc
$$ Doc
ds Doc -> Doc -> Doc
$$ Doc
es Doc -> Doc -> Doc
$$ Doc
fs Doc -> Doc -> Doc
$$ Doc
bb),Int
i4)
  where
    aa :: Doc
aa = String -> Doc
text String
"digraph G {\nnode [shape=box]\n"
    (Doc
cs, Int
i1) = Int -> [Stream] -> (Doc, Int)
ppStreamL Int
i    (Spec -> [Stream]
specStreams    Spec
spec)
    (Doc
ds, Int
i2) = Int -> [Trigger] -> (Doc, Int)
ppTriggerL Int
i1  (Spec -> [Trigger]
specTriggers   Spec
spec)
    (Doc
es, Int
i3) = Int -> [Observer] -> (Doc, Int)
ppObserverL Int
i2 (Spec -> [Observer]
specObservers  Spec
spec)
    (Doc
fs, Int
i4) = Int -> [Property] -> (Doc, Int)
ppPropertyL Int
i3 (Spec -> [Property]
specProperties Spec
spec)
    bb :: Doc
bb = String -> Doc
text String
"\n}\n"

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

-- | Pretty-print a Copilot expression as a GraphViz/dot graph.
prettyPrintExprDot :: Bool     -- ^ Mark externs with the prefix @externV:@.
                   -> Expr a   -- ^ The expression to pretty print.
                   -> String
prettyPrintExprDot :: Bool -> Expr a -> String
prettyPrintExprDot Bool
bb Expr a
s = Doc -> String
render Doc
rr
  where
    (Doc
r1, Int
_) = Int -> Int -> Bool -> Expr a -> (Doc, Int)
forall a. Int -> Int -> Bool -> Expr a -> (Doc, Int)
ppExprDot Int
1 Int
0 Bool
bb Expr a
s
    rr :: Doc
rr = String -> Doc
text String
"digraph G {\nnode [shape=box]\n" Doc -> Doc -> Doc
$$ (String -> Doc
text String
"0 [label=\"file: \n?????\",color=red, style=filled]\n") Doc -> Doc -> Doc
<> Doc
r1 Doc -> Doc -> Doc
$$ String -> Doc
text String
"\n}\n"

-- | Pretty-print a Copilot specification as a GraphViz/dot graph.
prettyPrintDot :: Spec -> String
prettyPrintDot :: Spec -> String
prettyPrintDot Spec
s = Doc -> String
render Doc
r1
  where (Doc
r1, Int
_) = Int -> Spec -> (Doc, Int)
ppSpecDot Int
0 Spec
s

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