From 0a32a06b9c3206484c34860148dfbe23935b8e3b Mon Sep 17 00:00:00 2001
From: Tom Smeding <tom@tomsmeding.com>
Date: Tue, 29 Apr 2025 17:52:32 +0200
Subject: pretty: Print types of monoid ops

---
 src/AST/Pretty.hs | 10 +++++-----
 1 file changed, 5 insertions(+), 5 deletions(-)

(limited to 'src/AST/Pretty.hs')

diff --git a/src/AST/Pretty.hs b/src/AST/Pretty.hs
index b6ad7d2..e09f3ae 100644
--- a/src/AST/Pretty.hs
+++ b/src/AST/Pretty.hs
@@ -297,24 +297,24 @@ ppExpr' d val expr = case expr of
     e2' <- ppExpr' 11 val e2
     e3' <- ppExpr' 11 val e3
     return $ ppParen (d > 10) $
-      ppApp (annotate AMonoid (ppString "accum") <> ppX expr) [ppString (ppAcPrj t prj), e1', e2', e3']
+      ppApp (annotate AMonoid (ppString "accum") <> ppX expr <+> ppString "@" <> ppSMTy' 11 (acPrjTy prj t)) [ppString (ppAcPrj t prj), e1', e2', e3']
 
   EZero _ t e1 -> do
     e1' <- ppExpr' 11 val e1
     return $ ppParen (d > 0) $
       annotate AMonoid (ppString "zero") <> ppX expr <+> ppString "@" <> ppSMTy' 11 t <+> e1'
 
-  EPlus _ _ a b -> do
+  EPlus _ t a b -> do
     a' <- ppExpr' 11 val a
     b' <- ppExpr' 11 val b
     return $ ppParen (d > 10) $
-      ppApp (annotate AMonoid (ppString "plus") <> ppX expr) [a', b']
+      ppApp (annotate AMonoid (ppString "plus") <> ppX expr <+> ppString "@" <> ppSMTy' 11 t) [a', b']
 
   EOneHot _ t prj a b -> do
     a' <- ppExpr' 11 val a
     b' <- ppExpr' 11 val b
     return $ ppParen (d > 10) $
-      ppApp (annotate AMonoid (ppString "onehot") <> ppX expr) [ppString (ppAcPrj t prj), a', b']
+      ppApp (annotate AMonoid (ppString "onehot") <> ppX expr <+> ppString "@" <> ppSMTy' 11 (acPrjTy prj t)) [ppString (ppAcPrj t prj), a', b']
 
   EError _ _ s -> return $ ppParen (d > 10) $ ppString "error" <> ppX expr <+> ppString (show s)
 
@@ -348,7 +348,7 @@ ppLam args body = ppString "(" <> hang 2 (ppString "\\" <> sep (args ++ [ppStrin
                                           <> softline <> body <> ppString ")")
 
 ppAcPrj :: SMTy a -> SAcPrj p a b -> String
-ppAcPrj _ SAPHere = "@"
+ppAcPrj _ SAPHere = "."
 ppAcPrj (SMTPair t _) (SAPFst prj) = "(" ++ ppAcPrj t prj ++ ",)"
 ppAcPrj (SMTPair _ t) (SAPSnd prj) = "(," ++ ppAcPrj t prj ++ ")"
 ppAcPrj (SMTLEither t _) (SAPLeft prj) = "(" ++ ppAcPrj t prj ++ "|)"
-- 
cgit v1.2.3-70-g09d2