module Futhark.CodeGen.Backends.GenericPython.AST
  ( PyExp (..),
    PyIdx (..),
    PyArg (..),
    PyStmt (..),
    module Language.Futhark.Core,
    PyProg (..),
    PyExcept (..),
    PyFunDef (..),
    PyClassDef (..),
  )
where

import qualified Data.Text as T
import Futhark.Util.Pretty
import Language.Futhark.Core

data UnOp
  = -- | Boolean negation.
    Not
  | -- | Bitwise complement.
    Complement
  | -- | Numerical negation.
    Negate
  | -- | Absolute/numerical value.
    Abs
  deriving (UnOp -> UnOp -> Bool
forall a. (a -> a -> Bool) -> (a -> a -> Bool) -> Eq a
/= :: UnOp -> UnOp -> Bool
$c/= :: UnOp -> UnOp -> Bool
== :: UnOp -> UnOp -> Bool
$c== :: UnOp -> UnOp -> Bool
Eq, Int -> UnOp -> ShowS
[UnOp] -> ShowS
UnOp -> String
forall a.
(Int -> a -> ShowS) -> (a -> String) -> ([a] -> ShowS) -> Show a
showList :: [UnOp] -> ShowS
$cshowList :: [UnOp] -> ShowS
show :: UnOp -> String
$cshow :: UnOp -> String
showsPrec :: Int -> UnOp -> ShowS
$cshowsPrec :: Int -> UnOp -> ShowS
Show)

data PyExp
  = Integer Integer
  | Bool Bool
  | Float Double
  | String String
  | RawStringLiteral T.Text
  | Var String
  | BinOp String PyExp PyExp
  | UnOp String PyExp
  | Cond PyExp PyExp PyExp
  | Index PyExp PyIdx
  | Call PyExp [PyArg]
  | Cast PyExp String
  | Tuple [PyExp]
  | List [PyExp]
  | Field PyExp String
  | Dict [(PyExp, PyExp)]
  | Lambda String PyExp
  | None
  deriving (PyExp -> PyExp -> Bool
forall a. (a -> a -> Bool) -> (a -> a -> Bool) -> Eq a
/= :: PyExp -> PyExp -> Bool
$c/= :: PyExp -> PyExp -> Bool
== :: PyExp -> PyExp -> Bool
$c== :: PyExp -> PyExp -> Bool
Eq, Int -> PyExp -> ShowS
[PyExp] -> ShowS
PyExp -> String
forall a.
(Int -> a -> ShowS) -> (a -> String) -> ([a] -> ShowS) -> Show a
showList :: [PyExp] -> ShowS
$cshowList :: [PyExp] -> ShowS
show :: PyExp -> String
$cshow :: PyExp -> String
showsPrec :: Int -> PyExp -> ShowS
$cshowsPrec :: Int -> PyExp -> ShowS
Show)

data PyIdx
  = IdxRange PyExp PyExp
  | IdxExp PyExp
  deriving (PyIdx -> PyIdx -> Bool
forall a. (a -> a -> Bool) -> (a -> a -> Bool) -> Eq a
/= :: PyIdx -> PyIdx -> Bool
$c/= :: PyIdx -> PyIdx -> Bool
== :: PyIdx -> PyIdx -> Bool
$c== :: PyIdx -> PyIdx -> Bool
Eq, Int -> PyIdx -> ShowS
[PyIdx] -> ShowS
PyIdx -> String
forall a.
(Int -> a -> ShowS) -> (a -> String) -> ([a] -> ShowS) -> Show a
showList :: [PyIdx] -> ShowS
$cshowList :: [PyIdx] -> ShowS
show :: PyIdx -> String
$cshow :: PyIdx -> String
showsPrec :: Int -> PyIdx -> ShowS
$cshowsPrec :: Int -> PyIdx -> ShowS
Show)

data PyArg
  = ArgKeyword String PyExp
  | Arg PyExp
  deriving (PyArg -> PyArg -> Bool
forall a. (a -> a -> Bool) -> (a -> a -> Bool) -> Eq a
/= :: PyArg -> PyArg -> Bool
$c/= :: PyArg -> PyArg -> Bool
== :: PyArg -> PyArg -> Bool
$c== :: PyArg -> PyArg -> Bool
Eq, Int -> PyArg -> ShowS
[PyArg] -> ShowS
PyArg -> String
forall a.
(Int -> a -> ShowS) -> (a -> String) -> ([a] -> ShowS) -> Show a
showList :: [PyArg] -> ShowS
$cshowList :: [PyArg] -> ShowS
show :: PyArg -> String
$cshow :: PyArg -> String
showsPrec :: Int -> PyArg -> ShowS
$cshowsPrec :: Int -> PyArg -> ShowS
Show)

data PyStmt
  = If PyExp [PyStmt] [PyStmt]
  | Try [PyStmt] [PyExcept]
  | While PyExp [PyStmt]
  | For String PyExp [PyStmt]
  | With PyExp [PyStmt]
  | Assign PyExp PyExp
  | AssignOp String PyExp PyExp
  | Comment String [PyStmt]
  | Assert PyExp PyExp
  | Raise PyExp
  | Exp PyExp
  | Return PyExp
  | Pass
  | -- Definition-like statements.
    Import String (Maybe String)
  | FunDef PyFunDef
  | ClassDef PyClassDef
  | -- Some arbitrary string of Python code.
    Escape T.Text
  deriving (PyStmt -> PyStmt -> Bool
forall a. (a -> a -> Bool) -> (a -> a -> Bool) -> Eq a
/= :: PyStmt -> PyStmt -> Bool
$c/= :: PyStmt -> PyStmt -> Bool
== :: PyStmt -> PyStmt -> Bool
$c== :: PyStmt -> PyStmt -> Bool
Eq, Int -> PyStmt -> ShowS
[PyStmt] -> ShowS
PyStmt -> String
forall a.
(Int -> a -> ShowS) -> (a -> String) -> ([a] -> ShowS) -> Show a
showList :: [PyStmt] -> ShowS
$cshowList :: [PyStmt] -> ShowS
show :: PyStmt -> String
$cshow :: PyStmt -> String
showsPrec :: Int -> PyStmt -> ShowS
$cshowsPrec :: Int -> PyStmt -> ShowS
Show)

data PyExcept = Catch PyExp [PyStmt]
  deriving (PyExcept -> PyExcept -> Bool
forall a. (a -> a -> Bool) -> (a -> a -> Bool) -> Eq a
/= :: PyExcept -> PyExcept -> Bool
$c/= :: PyExcept -> PyExcept -> Bool
== :: PyExcept -> PyExcept -> Bool
$c== :: PyExcept -> PyExcept -> Bool
Eq, Int -> PyExcept -> ShowS
[PyExcept] -> ShowS
PyExcept -> String
forall a.
(Int -> a -> ShowS) -> (a -> String) -> ([a] -> ShowS) -> Show a
showList :: [PyExcept] -> ShowS
$cshowList :: [PyExcept] -> ShowS
show :: PyExcept -> String
$cshow :: PyExcept -> String
showsPrec :: Int -> PyExcept -> ShowS
$cshowsPrec :: Int -> PyExcept -> ShowS
Show)

data PyFunDef = Def String [String] [PyStmt]
  deriving (PyFunDef -> PyFunDef -> Bool
forall a. (a -> a -> Bool) -> (a -> a -> Bool) -> Eq a
/= :: PyFunDef -> PyFunDef -> Bool
$c/= :: PyFunDef -> PyFunDef -> Bool
== :: PyFunDef -> PyFunDef -> Bool
$c== :: PyFunDef -> PyFunDef -> Bool
Eq, Int -> PyFunDef -> ShowS
[PyFunDef] -> ShowS
PyFunDef -> String
forall a.
(Int -> a -> ShowS) -> (a -> String) -> ([a] -> ShowS) -> Show a
showList :: [PyFunDef] -> ShowS
$cshowList :: [PyFunDef] -> ShowS
show :: PyFunDef -> String
$cshow :: PyFunDef -> String
showsPrec :: Int -> PyFunDef -> ShowS
$cshowsPrec :: Int -> PyFunDef -> ShowS
Show)

data PyClassDef = Class String [PyStmt]
  deriving (PyClassDef -> PyClassDef -> Bool
forall a. (a -> a -> Bool) -> (a -> a -> Bool) -> Eq a
/= :: PyClassDef -> PyClassDef -> Bool
$c/= :: PyClassDef -> PyClassDef -> Bool
== :: PyClassDef -> PyClassDef -> Bool
$c== :: PyClassDef -> PyClassDef -> Bool
Eq, Int -> PyClassDef -> ShowS
[PyClassDef] -> ShowS
PyClassDef -> String
forall a.
(Int -> a -> ShowS) -> (a -> String) -> ([a] -> ShowS) -> Show a
showList :: [PyClassDef] -> ShowS
$cshowList :: [PyClassDef] -> ShowS
show :: PyClassDef -> String
$cshow :: PyClassDef -> String
showsPrec :: Int -> PyClassDef -> ShowS
$cshowsPrec :: Int -> PyClassDef -> ShowS
Show)

newtype PyProg = PyProg [PyStmt]
  deriving (PyProg -> PyProg -> Bool
forall a. (a -> a -> Bool) -> (a -> a -> Bool) -> Eq a
/= :: PyProg -> PyProg -> Bool
$c/= :: PyProg -> PyProg -> Bool
== :: PyProg -> PyProg -> Bool
$c== :: PyProg -> PyProg -> Bool
Eq, Int -> PyProg -> ShowS
[PyProg] -> ShowS
PyProg -> String
forall a.
(Int -> a -> ShowS) -> (a -> String) -> ([a] -> ShowS) -> Show a
showList :: [PyProg] -> ShowS
$cshowList :: [PyProg] -> ShowS
show :: PyProg -> String
$cshow :: PyProg -> String
showsPrec :: Int -> PyProg -> ShowS
$cshowsPrec :: Int -> PyProg -> ShowS
Show)

instance Pretty PyIdx where
  ppr :: PyIdx -> Doc
ppr (IdxExp PyExp
e) = forall a. Pretty a => a -> Doc
ppr PyExp
e
  ppr (IdxRange PyExp
from PyExp
to) = forall a. Pretty a => a -> Doc
ppr PyExp
from forall a. Semigroup a => a -> a -> a
<> String -> Doc
text String
":" forall a. Semigroup a => a -> a -> a
<> forall a. Pretty a => a -> Doc
ppr PyExp
to

instance Pretty PyArg where
  ppr :: PyArg -> Doc
ppr (ArgKeyword String
k PyExp
e) = String -> Doc
text String
k forall a. Semigroup a => a -> a -> a
<> Doc
equals forall a. Semigroup a => a -> a -> a
<> forall a. Pretty a => a -> Doc
ppr PyExp
e
  ppr (Arg PyExp
e) = forall a. Pretty a => a -> Doc
ppr PyExp
e

instance Pretty PyExp where
  ppr :: PyExp -> Doc
ppr (Integer Integer
x) = forall a. Pretty a => a -> Doc
ppr Integer
x
  ppr (Bool Bool
x) = forall a. Pretty a => a -> Doc
ppr Bool
x
  ppr (Float Double
x)
    | forall a. RealFloat a => a -> Bool
isInfinite Double
x = String -> Doc
text forall a b. (a -> b) -> a -> b
$ if Double
x forall a. Ord a => a -> a -> Bool
> Double
0 then String
"float('inf')" else String
"float('-inf')"
    | Bool
otherwise = forall a. Pretty a => a -> Doc
ppr Double
x
  ppr (String String
x) = String -> Doc
text forall a b. (a -> b) -> a -> b
$ forall a. Show a => a -> String
show String
x
  ppr (RawStringLiteral Text
s) = String -> Doc
text String
"\"\"\"" forall a. Semigroup a => a -> a -> a
<> Text -> Doc
strictText Text
s forall a. Semigroup a => a -> a -> a
<> String -> Doc
text String
"\"\"\""
  ppr (Var String
n) = String -> Doc
text forall a b. (a -> b) -> a -> b
$ forall a b. (a -> b) -> [a] -> [b]
map (\Char
x -> if Char
x forall a. Eq a => a -> a -> Bool
== Char
'\'' then Char
'm' else Char
x) String
n
  ppr (Field PyExp
e String
s) = forall a. Pretty a => a -> Doc
ppr PyExp
e forall a. Semigroup a => a -> a -> a
<> String -> Doc
text String
"." forall a. Semigroup a => a -> a -> a
<> String -> Doc
text String
s
  ppr (BinOp String
s PyExp
e1 PyExp
e2) = Doc -> Doc
parens (forall a. Pretty a => a -> Doc
ppr PyExp
e1 Doc -> Doc -> Doc
<+> String -> Doc
text String
s Doc -> Doc -> Doc
<+> forall a. Pretty a => a -> Doc
ppr PyExp
e2)
  ppr (UnOp String
s PyExp
e) = String -> Doc
text String
s forall a. Semigroup a => a -> a -> a
<> Doc -> Doc
parens (forall a. Pretty a => a -> Doc
ppr PyExp
e)
  ppr (Cond PyExp
e1 PyExp
e2 PyExp
e3) = forall a. Pretty a => a -> Doc
ppr PyExp
e2 Doc -> Doc -> Doc
<+> String -> Doc
text String
"if" Doc -> Doc -> Doc
<+> forall a. Pretty a => a -> Doc
ppr PyExp
e1 Doc -> Doc -> Doc
<+> String -> Doc
text String
"else" Doc -> Doc -> Doc
<+> forall a. Pretty a => a -> Doc
ppr PyExp
e3
  ppr (Cast PyExp
src String
bt) =
    String -> Doc
text String
"ct.cast"
      forall a. Semigroup a => a -> a -> a
<> Doc -> Doc
parens
        ( forall a. Pretty a => a -> Doc
ppr PyExp
src forall a. Semigroup a => a -> a -> a
<> String -> Doc
text String
","
            Doc -> Doc -> Doc
<+> String -> Doc
text String
"ct.POINTER" forall a. Semigroup a => a -> a -> a
<> Doc -> Doc
parens (String -> Doc
text String
bt)
        )
  ppr (Index PyExp
src PyIdx
idx) = forall a. Pretty a => a -> Doc
ppr PyExp
src forall a. Semigroup a => a -> a -> a
<> Doc -> Doc
brackets (forall a. Pretty a => a -> Doc
ppr PyIdx
idx)
  ppr (Call PyExp
fun [PyArg]
exps) = forall a. Pretty a => a -> Doc
ppr PyExp
fun forall a. Semigroup a => a -> a -> a
<> Doc -> Doc
parens ([Doc] -> Doc
commasep forall a b. (a -> b) -> a -> b
$ forall a b. (a -> b) -> [a] -> [b]
map forall a. Pretty a => a -> Doc
ppr [PyArg]
exps)
  ppr (Tuple [PyExp
dim]) = Doc -> Doc
parens (forall a. Pretty a => a -> Doc
ppr PyExp
dim forall a. Semigroup a => a -> a -> a
<> String -> Doc
text String
",")
  ppr (Tuple [PyExp]
dims) = Doc -> Doc
parens ([Doc] -> Doc
commasep forall a b. (a -> b) -> a -> b
$ forall a b. (a -> b) -> [a] -> [b]
map forall a. Pretty a => a -> Doc
ppr [PyExp]
dims)
  ppr (List [PyExp]
es) = Doc -> Doc
brackets forall a b. (a -> b) -> a -> b
$ [Doc] -> Doc
commasep forall a b. (a -> b) -> a -> b
$ forall a b. (a -> b) -> [a] -> [b]
map forall a. Pretty a => a -> Doc
ppr [PyExp]
es
  ppr (Dict [(PyExp, PyExp)]
kvs) = Doc -> Doc
braces forall a b. (a -> b) -> a -> b
$ [Doc] -> Doc
commasep forall a b. (a -> b) -> a -> b
$ forall a b. (a -> b) -> [a] -> [b]
map forall {a} {a}. (Pretty a, Pretty a) => (a, a) -> Doc
ppElem [(PyExp, PyExp)]
kvs
    where
      ppElem :: (a, a) -> Doc
ppElem (a
k, a
v) = forall a. Pretty a => a -> Doc
ppr a
k forall a. Semigroup a => a -> a -> a
<> Doc
colon Doc -> Doc -> Doc
<+> forall a. Pretty a => a -> Doc
ppr a
v
  ppr (Lambda String
p PyExp
e) = String -> Doc
text String
"lambda" Doc -> Doc -> Doc
<+> String -> Doc
text String
p forall a. Semigroup a => a -> a -> a
<> String -> Doc
text String
":" Doc -> Doc -> Doc
<+> forall a. Pretty a => a -> Doc
ppr PyExp
e
  ppr PyExp
None = String -> Doc
text String
"None"

instance Pretty PyStmt where
  ppr :: PyStmt -> Doc
ppr (If PyExp
cond [] []) =
    String -> Doc
text String
"if"
      Doc -> Doc -> Doc
<+> forall a. Pretty a => a -> Doc
ppr PyExp
cond forall a. Semigroup a => a -> a -> a
<> String -> Doc
text String
":"
      Doc -> Doc -> Doc
</> Int -> Doc -> Doc
indent Int
2 (String -> Doc
text String
"pass")
  ppr (If PyExp
cond [] [PyStmt]
fbranch) =
    String -> Doc
text String
"if"
      Doc -> Doc -> Doc
<+> forall a. Pretty a => a -> Doc
ppr PyExp
cond forall a. Semigroup a => a -> a -> a
<> String -> Doc
text String
":"
      Doc -> Doc -> Doc
</> Int -> Doc -> Doc
indent Int
2 (String -> Doc
text String
"pass")
      Doc -> Doc -> Doc
</> String -> Doc
text String
"else:"
      Doc -> Doc -> Doc
</> Int -> Doc -> Doc
indent Int
2 ([Doc] -> Doc
stack forall a b. (a -> b) -> a -> b
$ forall a b. (a -> b) -> [a] -> [b]
map forall a. Pretty a => a -> Doc
ppr [PyStmt]
fbranch)
  ppr (If PyExp
cond [PyStmt]
tbranch []) =
    String -> Doc
text String
"if"
      Doc -> Doc -> Doc
<+> forall a. Pretty a => a -> Doc
ppr PyExp
cond forall a. Semigroup a => a -> a -> a
<> String -> Doc
text String
":"
      Doc -> Doc -> Doc
</> Int -> Doc -> Doc
indent Int
2 ([Doc] -> Doc
stack forall a b. (a -> b) -> a -> b
$ forall a b. (a -> b) -> [a] -> [b]
map forall a. Pretty a => a -> Doc
ppr [PyStmt]
tbranch)
  ppr (If PyExp
cond [PyStmt]
tbranch [PyStmt]
fbranch) =
    String -> Doc
text String
"if"
      Doc -> Doc -> Doc
<+> forall a. Pretty a => a -> Doc
ppr PyExp
cond forall a. Semigroup a => a -> a -> a
<> String -> Doc
text String
":"
      Doc -> Doc -> Doc
</> Int -> Doc -> Doc
indent Int
2 ([Doc] -> Doc
stack forall a b. (a -> b) -> a -> b
$ forall a b. (a -> b) -> [a] -> [b]
map forall a. Pretty a => a -> Doc
ppr [PyStmt]
tbranch)
      Doc -> Doc -> Doc
</> String -> Doc
text String
"else:"
      Doc -> Doc -> Doc
</> Int -> Doc -> Doc
indent Int
2 ([Doc] -> Doc
stack forall a b. (a -> b) -> a -> b
$ forall a b. (a -> b) -> [a] -> [b]
map forall a. Pretty a => a -> Doc
ppr [PyStmt]
fbranch)
  ppr (Try [PyStmt]
pystms [PyExcept]
pyexcepts) =
    String -> Doc
text String
"try:"
      Doc -> Doc -> Doc
</> Int -> Doc -> Doc
indent Int
2 ([Doc] -> Doc
stack forall a b. (a -> b) -> a -> b
$ forall a b. (a -> b) -> [a] -> [b]
map forall a. Pretty a => a -> Doc
ppr [PyStmt]
pystms)
      Doc -> Doc -> Doc
</> [Doc] -> Doc
stack (forall a b. (a -> b) -> [a] -> [b]
map forall a. Pretty a => a -> Doc
ppr [PyExcept]
pyexcepts)
  ppr (While PyExp
cond [PyStmt]
body) =
    String -> Doc
text String
"while"
      Doc -> Doc -> Doc
<+> forall a. Pretty a => a -> Doc
ppr PyExp
cond forall a. Semigroup a => a -> a -> a
<> String -> Doc
text String
":"
      Doc -> Doc -> Doc
</> Int -> Doc -> Doc
indent Int
2 ([Doc] -> Doc
stack forall a b. (a -> b) -> a -> b
$ forall a b. (a -> b) -> [a] -> [b]
map forall a. Pretty a => a -> Doc
ppr [PyStmt]
body)
  ppr (For String
i PyExp
what [PyStmt]
body) =
    String -> Doc
text String
"for"
      Doc -> Doc -> Doc
<+> forall a. Pretty a => a -> Doc
ppr String
i
      Doc -> Doc -> Doc
<+> String -> Doc
text String
"in"
      Doc -> Doc -> Doc
<+> forall a. Pretty a => a -> Doc
ppr PyExp
what forall a. Semigroup a => a -> a -> a
<> String -> Doc
text String
":"
      Doc -> Doc -> Doc
</> Int -> Doc -> Doc
indent Int
2 ([Doc] -> Doc
stack forall a b. (a -> b) -> a -> b
$ forall a b. (a -> b) -> [a] -> [b]
map forall a. Pretty a => a -> Doc
ppr [PyStmt]
body)
  ppr (With PyExp
what [PyStmt]
body) =
    String -> Doc
text String
"with"
      Doc -> Doc -> Doc
<+> forall a. Pretty a => a -> Doc
ppr PyExp
what forall a. Semigroup a => a -> a -> a
<> String -> Doc
text String
":"
      Doc -> Doc -> Doc
</> Int -> Doc -> Doc
indent Int
2 ([Doc] -> Doc
stack forall a b. (a -> b) -> a -> b
$ forall a b. (a -> b) -> [a] -> [b]
map forall a. Pretty a => a -> Doc
ppr [PyStmt]
body)
  ppr (Assign PyExp
e1 PyExp
e2) = forall a. Pretty a => a -> Doc
ppr PyExp
e1 Doc -> Doc -> Doc
<+> String -> Doc
text String
"=" Doc -> Doc -> Doc
<+> forall a. Pretty a => a -> Doc
ppr PyExp
e2
  ppr (AssignOp String
op PyExp
e1 PyExp
e2) = forall a. Pretty a => a -> Doc
ppr PyExp
e1 Doc -> Doc -> Doc
<+> String -> Doc
text (String
op forall a. [a] -> [a] -> [a]
++ String
"=") Doc -> Doc -> Doc
<+> forall a. Pretty a => a -> Doc
ppr PyExp
e2
  ppr (Comment String
s [PyStmt]
body) = String -> Doc
text String
"#" forall a. Semigroup a => a -> a -> a
<> String -> Doc
text String
s Doc -> Doc -> Doc
</> [Doc] -> Doc
stack (forall a b. (a -> b) -> [a] -> [b]
map forall a. Pretty a => a -> Doc
ppr [PyStmt]
body)
  ppr (Assert PyExp
e1 PyExp
e2) = String -> Doc
text String
"assert" Doc -> Doc -> Doc
<+> forall a. Pretty a => a -> Doc
ppr PyExp
e1 forall a. Semigroup a => a -> a -> a
<> String -> Doc
text String
"," Doc -> Doc -> Doc
<+> forall a. Pretty a => a -> Doc
ppr PyExp
e2
  ppr (Raise PyExp
e) = String -> Doc
text String
"raise" Doc -> Doc -> Doc
<+> forall a. Pretty a => a -> Doc
ppr PyExp
e
  ppr (Exp PyExp
c) = forall a. Pretty a => a -> Doc
ppr PyExp
c
  ppr (Return PyExp
e) = String -> Doc
text String
"return" Doc -> Doc -> Doc
<+> forall a. Pretty a => a -> Doc
ppr PyExp
e
  ppr PyStmt
Pass = String -> Doc
text String
"pass"
  ppr (Import String
from (Just String
as)) =
    String -> Doc
text String
"import" Doc -> Doc -> Doc
<+> String -> Doc
text String
from Doc -> Doc -> Doc
<+> String -> Doc
text String
"as" Doc -> Doc -> Doc
<+> String -> Doc
text String
as
  ppr (Import String
from Maybe String
Nothing) =
    String -> Doc
text String
"import" Doc -> Doc -> Doc
<+> String -> Doc
text String
from
  ppr (FunDef PyFunDef
d) = forall a. Pretty a => a -> Doc
ppr PyFunDef
d
  ppr (ClassDef PyClassDef
d) = forall a. Pretty a => a -> Doc
ppr PyClassDef
d
  ppr (Escape Text
s) = [Doc] -> Doc
stack forall a b. (a -> b) -> a -> b
$ forall a b. (a -> b) -> [a] -> [b]
map Text -> Doc
strictText forall a b. (a -> b) -> a -> b
$ Text -> [Text]
T.lines Text
s

instance Pretty PyFunDef where
  ppr :: PyFunDef -> Doc
ppr (Def String
fname [String]
params [PyStmt]
body) =
    String -> Doc
text String
"def"
      Doc -> Doc -> Doc
<+> String -> Doc
text String
fname forall a. Semigroup a => a -> a -> a
<> Doc -> Doc
parens ([Doc] -> Doc
commasep forall a b. (a -> b) -> a -> b
$ forall a b. (a -> b) -> [a] -> [b]
map forall a. Pretty a => a -> Doc
ppr [String]
params) forall a. Semigroup a => a -> a -> a
<> String -> Doc
text String
":"
      Doc -> Doc -> Doc
</> Int -> Doc -> Doc
indent Int
2 ([Doc] -> Doc
stack (forall a b. (a -> b) -> [a] -> [b]
map forall a. Pretty a => a -> Doc
ppr [PyStmt]
body))

instance Pretty PyClassDef where
  ppr :: PyClassDef -> Doc
ppr (Class String
cname [PyStmt]
body) =
    String -> Doc
text String
"class"
      Doc -> Doc -> Doc
<+> String -> Doc
text String
cname forall a. Semigroup a => a -> a -> a
<> String -> Doc
text String
":"
      Doc -> Doc -> Doc
</> Int -> Doc -> Doc
indent Int
2 ([Doc] -> Doc
stack (forall a b. (a -> b) -> [a] -> [b]
map forall a. Pretty a => a -> Doc
ppr [PyStmt]
body))

instance Pretty PyExcept where
  ppr :: PyExcept -> Doc
ppr (Catch PyExp
pyexp [PyStmt]
stms) =
    String -> Doc
text String
"except"
      Doc -> Doc -> Doc
<+> forall a. Pretty a => a -> Doc
ppr PyExp
pyexp
      Doc -> Doc -> Doc
<+> String -> Doc
text String
"as e:"
      Doc -> Doc -> Doc
</> Int -> Doc -> Doc
indent Int
2 ([Doc] -> Doc
stack forall a b. (a -> b) -> a -> b
$ forall a b. (a -> b) -> [a] -> [b]
map forall a. Pretty a => a -> Doc
ppr [PyStmt]
stms)

instance Pretty PyProg where
  ppr :: PyProg -> Doc
ppr (PyProg [PyStmt]
stms) = [Doc] -> Doc
stack (forall a b. (a -> b) -> [a] -> [b]
map forall a. Pretty a => a -> Doc
ppr [PyStmt]
stms)