adamc@1: (* Copyright (c) 2008, Adam Chlipala
adamc@1:  * All rights reserved.
adamc@1:  *
adamc@1:  * Redistribution and use in source and binary forms, with or without
adamc@1:  * modification, are permitted provided that the following conditions are met:
adamc@1:  *
adamc@1:  * - Redistributions of source code must retain the above copyright notice,
adamc@1:  *   this list of conditions and the following disclaimer.
adamc@1:  * - Redistributions in binary form must reproduce the above copyright notice,
adamc@1:  *   this list of conditions and the following disclaimer in the documentation
adamc@1:  *   and/or other materials provided with the distribution.
adamc@1:  * - The names of contributors may not be used to endorse or promote products
adamc@1:  *   derived from this software without specific prior written permission.
adamc@1:  *
adamc@1:  * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS"
adamc@1:  * AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
adamc@1:  * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
adamc@1:  * ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT OWNER OR CONTRIBUTORS BE
adamc@1:  * LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR 
adamc@1:  * CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
adamc@1:  * SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
adamc@1:  * INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
adamc@1:  * CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
adamc@1:  * ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
adamc@1:  * POSSIBILITY OF SUCH DAMAGE.
adamc@1:  *)
adamc@1: 
adamc@1: (* Pretty-printing Laconic/Web *)
adamc@1: 
adamc@4: structure SourcePrint :> SOURCE_PRINT = struct
adamc@1: 
adamc@1: open Print.PD
adamc@1: open Print
adamc@1: 
adamc@4: open Source
adamc@1: 
adamc@1: fun p_kind' par (k, _) =
adamc@1:     case k of
adamc@1:         KType => string "Type"
adamc@1:       | KArrow (k1, k2) => parenIf par (box [p_kind' true k1,
adamc@1:                                              space,
adamc@1:                                              string "->",
adamc@1:                                              space,
adamc@1:                                              p_kind k2])
adamc@1:       | KName => string "Name"
adamc@1:       | KRecord k => box [string "{", p_kind k, string "}"]
adamc@82:       | KUnit => string "Unit"
adamc@18:       | KWild => string "_"
adamc@207:       | KTuple ks => box [string "(",
adamc@207:                           p_list_sep (box [space, string "*", space]) p_kind ks,
adamc@207:                           string ")"]
adamc@1: 
adamc@1: and p_kind k = p_kind' false k
adamc@1: 
adamc@1: fun p_explicitness e =
adamc@1:     case e of
adamc@1:         Explicit => string "::"
adamc@1:       | Implicit => string ":::"
adamc@1: 
adamc@1: fun p_con' par (c, _) =
adamc@1:     case c of
adamc@1:         CAnnot (c, k) => box [string "(",
adamc@1:                               p_con c,
adamc@1:                               space,
adamc@1:                               string "::",
adamc@1:                               space,
adamc@1:                               p_kind k,
adamc@1:                               string ")"]
adamc@1: 
adamc@1:       | TFun (t1, t2) => parenIf par (box [p_con' true t1,
adamc@1:                                            space,
adamc@1:                                            string "->",
adamc@1:                                            space,
adamc@1:                                            p_con t2])
adamc@1:       | TCFun (e, x, k, c) => parenIf par (box [string x,
adamc@1:                                                 space,
adamc@1:                                                 p_explicitness e,
adamc@1:                                                 space,
adamc@1:                                                 p_kind k,
adamc@1:                                                 space,
adamc@1:                                                 string "->",
adamc@1:                                                 space,
adamc@1:                                                 p_con c])
adamc@85:       | TDisjoint (c1, c2, c3) => parenIf par (box [p_con c1,
adamc@85:                                                     space,
adamc@85:                                                     string "~",
adamc@85:                                                     space,
adamc@85:                                                     p_con c2,
adamc@85:                                                     space,
adamc@85:                                                     string "->",
adamc@85:                                                     space,
adamc@85:                                                     p_con c3])
adamc@1:       | TRecord (CRecord xcs, _) => box [string "{",
adamc@1:                                          p_list (fn (x, c) =>
adamc@20:                                                     box [p_name x,
adamc@1:                                                          space,
adamc@1:                                                          string ":",
adamc@1:                                                          space,
adamc@1:                                                          p_con c]) xcs,
adamc@1:                                          string "}"]
adamc@1:       | TRecord c => box [string "$",
adamc@1:                           p_con' true c]
adamc@1: 
adamc@34:       | CVar (ss, s) => p_list_sep (string ".") string (ss @ [s])
adamc@1:       | CApp (c1, c2) => parenIf par (box [p_con c1,
adamc@1:                                            space,
adamc@1:                                            p_con' true c2])
adamc@67:       | CAbs (x, NONE, c) => parenIf par (box [string "fn",
adamc@67:                                                space,
adamc@67:                                                string x,
adamc@67:                                                space,
adamc@67:                                                string "=>",
adamc@67:                                                space,
adamc@67:                                                p_con c])
adamc@67:       | CAbs (x, SOME k, c) => parenIf par (box [string "fn",
adamc@67:                                                  space,
adamc@67:                                                  string x,
adamc@67:                                                  space,
adamc@67:                                                  string "::",
adamc@67:                                                  space,
adamc@67:                                                  p_kind k,
adamc@67:                                                  space,
adamc@67:                                                  string "=>",
adamc@67:                                                  space,
adamc@67:                                                  p_con c])
adamc@84:       | CDisjoint (c1, c2, c3) => parenIf par (box [p_con c1,
adamc@84:                                                     space,
adamc@84:                                                     string "~",
adamc@84:                                                     space,
adamc@84:                                                     p_con c2,
adamc@84:                                                     space,
adamc@84:                                                     string "=>",
adamc@84:                                                     space,
adamc@84:                                                     p_con c3])
adamc@1: 
adamc@1:       | CName s => box [string "#", string s]
adamc@1: 
adamc@1:       | CRecord xcs => box [string "[",
adamc@1:                             p_list (fn (x, c) =>
adamc@1:                                        box [p_con x,
adamc@1:                                             space,
adamc@1:                                             string "=",
adamc@1:                                             space,
adamc@1:                                             p_con c]) xcs,
adamc@1:                             string "]"]
adamc@1:       | CConcat (c1, c2) => parenIf par (box [p_con' true c1,
adamc@1:                                               space,
adamc@1:                                               string "++",
adamc@1:                                               space,
adamc@1:                                               p_con c2])
adamc@67:       | CFold => string "fold"
adamc@82: 
adamc@82:       | CUnit => string "()"
adamc@82: 
adamc@18:       | CWild k => box [string "(_",
adamc@18:                         space,
adamc@18:                         string "::",
adamc@18:                         space,
adamc@18:                         p_kind k]
adamc@207: 
adamc@207:       | CTuple cs => box [string "(",
adamc@207:                           p_list p_con cs,
adamc@207:                           string ")"]
adamc@207:       | CProj (c, n) => box [p_con c,
adamc@207:                              string ".",
adamc@207:                              string (Int.toString n)]
adamc@1:         
adamc@1: and p_con c = p_con' false c
adamc@1: 
adamc@20: and p_name (all as (c, _)) =
adamc@20:     case c of
adamc@20:         CName s => string s
adamc@20:       | _ => p_con all
adamc@20: 
adamc@170: fun p_pat' par (p, _) =
adamc@170:     case p of
adamc@170:         PWild => string "_"
adamc@170:       | PVar s => string s
adamc@173:       | PPrim p => Prim.p_t p
adamc@170:       | PCon (ms, x, NONE) => p_list_sep (string ".") string (ms @ [x])
adamc@170:       | PCon (ms, x, SOME p) => parenIf par (box [p_list_sep (string ".") string (ms @ [x]),
adamc@170:                                                   space,
adamc@170:                                                   p_pat' true p])
adamc@174:       | PRecord (xps, flex) =>
adamc@174:         let
adamc@175:             val pps = map (fn (x, p) => box [string x, space, string "=", space, p_pat p]) xps
adamc@174:         in
adamc@174:             box [string "{",
adamc@174:                  p_list_sep (box [string ",", space]) (fn x => x)
adamc@174:                  (if flex then
adamc@175:                       pps @ [string "..."]
adamc@174:                   else
adamc@175:                       pps),
adamc@174:                  string "}"]
adamc@174:         end
adamc@170: 
adamc@174: and p_pat x = p_pat' false x
adamc@170: 
adamc@8: fun p_exp' par (e, _) =
adamc@8:     case e of
adamc@8:         EAnnot (e, t) => box [string "(",
adamc@8:                               p_exp e,
adamc@8:                               space,
adamc@8:                               string ":",
adamc@8:                               space,
adamc@8:                               p_con t,
adamc@8:                               string ")"]        
adamc@8: 
adamc@14:       | EPrim p => Prim.p_t p
adamc@34:       | EVar (ss, s) => p_list_sep (string ".") string (ss @ [s])
adamc@8:       | EApp (e1, e2) => parenIf par (box [p_exp e1,
adamc@8:                                            space,
adamc@8:                                            p_exp' true e2])
adamc@8:       | EAbs (x, NONE, e) => parenIf par (box [string "fn",
adamc@8:                                                space,
adamc@8:                                                string x,
adamc@8:                                                space,
adamc@8:                                                string "=>",
adamc@8:                                                space,
adamc@8:                                                p_exp e])
adamc@8:       | EAbs (x, SOME t, e) => parenIf par (box [string "fn",
adamc@8:                                                  space,
adamc@8:                                                  string x,
adamc@8:                                                  space,
adamc@8:                                                  string ":",
adamc@8:                                                  space,
adamc@8:                                                  p_con t,
adamc@8:                                                  space,
adamc@8:                                                  string "=>",
adamc@8:                                                  space,
adamc@8:                                                  p_exp e])
adamc@8:       | ECApp (e, c) => parenIf par (box [p_exp e,
adamc@8:                                           space,
adamc@8:                                           string "[",
adamc@8:                                           p_con c,
adamc@8:                                           string "]"])
adamc@8:       | ECAbs (exp, x, k, e) => parenIf par (box [string "fn",
adamc@8:                                                   space,
adamc@8:                                                   string x,
adamc@8:                                                   space,
adamc@8:                                                   p_explicitness exp,
adamc@8:                                                   space,
adamc@8:                                                   p_kind k,
adamc@8:                                                   space,
adamc@8:                                                   string "=>",
adamc@8:                                                   space,
adamc@8:                                                   p_exp e])
adamc@85:       | EDisjoint (c1, c2, e) => parenIf par (box [p_con c1,
adamc@85:                                                    space,
adamc@85:                                                    string "~",
adamc@85:                                                    space,
adamc@85:                                                    p_con c2,
adamc@85:                                                    space,
adamc@85:                                                    string "=>",
adamc@85:                                                    space,
adamc@85:                                                    p_exp e])
adamc@8: 
adamc@12:       | ERecord xes => box [string "{",
adamc@12:                             p_list (fn (x, e) =>
adamc@21:                                        box [p_name x,
adamc@12:                                             space,
adamc@12:                                             string "=",
adamc@12:                                             space,
adamc@12:                                             p_exp e]) xes,
adamc@12:                             string "}"]
adamc@12:       | EField (e, c) => box [p_exp' true e,
adamc@12:                               string ".",
adamc@12:                               p_con' true c]
adamc@149:       | ECut (e, c) => parenIf par (box [p_exp' true e,
adamc@149:                                          space,
adamc@149:                                          string "--",
adamc@149:                                          space,
adamc@149:                                          p_con' true c])
adamc@71:       | EFold => string "fold"
adamc@12: 
adamc@170:       | ECase (e, pes) => parenIf par (box [string "case",
adamc@170:                                             space,
adamc@171:                                             p_exp e,
adamc@170:                                             space,
adamc@170:                                             string "of",
adamc@170:                                             space,
adamc@170:                                             p_list_sep (box [space, string "|", space])
adamc@170:                                             (fn (p, e) => box [p_pat p,
adamc@170:                                                                space,
adamc@170:                                                                string "=>",
adamc@170:                                                                space,
adamc@170:                                                                p_exp e]) pes])
adamc@170: 
adamc@219:       | EWild => string "_"
adamc@210: 
adamc@8: and p_exp e = p_exp' false e
adamc@8: 
adamc@191: fun p_datatype (x, xs, cons) =
adamc@156:     box [string "datatype",
adamc@156:          space,
adamc@156:          string x,
adamc@191:          p_list_sep (box []) (fn x => box [space, string x]) xs,
adamc@156:          space,
adamc@156:          string "=",
adamc@156:          space,
adamc@156:          p_list_sep (box [space, string "|", space])
adamc@156:          (fn (x, NONE) => string x
adamc@156:            | (x, SOME t) => box [string x, space, string "of", space, p_con t])
adamc@156:          cons]
adamc@156: 
adamc@30: fun p_sgn_item (sgi, _) =
adamc@30:     case sgi of
adamc@30:         SgiConAbs (x, k) => box [string "con",
adamc@30:                                  space,
adamc@30:                                  string x,
adamc@30:                                  space,
adamc@30:                                  string "::",
adamc@30:                                  space,
adamc@30:                                  p_kind k]
adamc@30:       | SgiCon (x, NONE, c) => box [string "con",
adamc@30:                                     space,
adamc@30:                                     string x,
adamc@30:                                     space,
adamc@30:                                     string "=",
adamc@30:                                     space,
adamc@30:                                     p_con c]
adamc@30:       | SgiCon (x, SOME k, c) => box [string "con",
adamc@30:                                       space,
adamc@30:                                       string x,
adamc@30:                                       space,
adamc@30:                                       string "::",
adamc@30:                                       space,
adamc@30:                                       p_kind k,
adamc@30:                                       space,
adamc@30:                                       string "=",
adamc@30:                                       space,
adamc@30:                                       p_con c]
adamc@156:       | SgiDatatype x => p_datatype x
adamc@156:       | SgiDatatypeImp (x, ms, x') =>
adamc@156:         box [string "datatype",
adamc@156:              space,
adamc@156:              string x,
adamc@156:              space,
adamc@156:              string "=",
adamc@156:              space,
adamc@156:              string "datatype",
adamc@156:              space,
adamc@156:              p_list_sep (string ".") string (ms @ [x'])]
adamc@30:       | SgiVal (x, c) => box [string "val",
adamc@30:                               space,
adamc@30:                               string x,
adamc@30:                               space,
adamc@30:                               string ":",
adamc@30:                               space,
adamc@30:                               p_con c]
adamc@30:       | SgiStr (x, sgn) => box [string "structure",
adamc@30:                                 space,
adamc@30:                                 string x,
adamc@30:                                 space,
adamc@30:                                 string ":",
adamc@30:                                 space,
adamc@30:                                 p_sgn sgn]
adamc@59:       | SgiSgn (x, sgn) => box [string "signature",
adamc@59:                                 space,
adamc@59:                                 string x,
adamc@59:                                 space,
adamc@59:                                 string "=",
adamc@59:                                 space,
adamc@59:                                 p_sgn sgn]
adamc@58:       | SgiInclude sgn => box [string "include",
adamc@58:                                space,
adamc@58:                                p_sgn sgn]
adamc@88:       | SgiConstraint (c1, c2) => box [string "constraint",
adamc@88:                                        space,
adamc@88:                                        p_con c1,
adamc@88:                                        space,
adamc@88:                                        string "~",
adamc@88:                                        space,
adamc@88:                                        p_con c2]
adamc@203:       | SgiTable (x, c) => box [string "table",
adamc@203:                                 space,
adamc@203:                                 string x,
adamc@203:                                 space,
adamc@203:                                 string ":",
adamc@203:                                 space,
adamc@203:                                 p_con c]
adamc@211:       | SgiClassAbs x => box [string "class",
adamc@211:                               space,
adamc@211:                               string x]
adamc@211:       | SgiClass (x, c) => box [string "class",
adamc@211:                                 space,
adamc@211:                                 string x,
adamc@211:                                 space,
adamc@211:                                 string "=",
adamc@211:                                 space,
adamc@211:                                 p_con c]
adamc@30: 
adamc@30: and p_sgn (sgn, _) =
adamc@30:     case sgn of
adamc@30:         SgnConst sgis => box [string "sig",
adamc@30:                               newline,
adamc@30:                               p_list_sep newline p_sgn_item sgis,
adamc@30:                               newline,
adamc@30:                               string "end"]
adamc@30:       | SgnVar x => string x
adamc@40:       | SgnFun (x, sgn, sgn') => box [string "functor",
adamc@40:                                       space,
adamc@40:                                       string "(",
adamc@40:                                       string x,
adamc@40:                                       space,
adamc@40:                                       string ":",
adamc@40:                                       p_sgn sgn,
adamc@40:                                       string ")",
adamc@40:                                       space,
adamc@40:                                       string ":",
adamc@40:                                       space,
adamc@40:                                       p_sgn sgn']
adamc@42:       | SgnWhere (sgn, x, c) => box [p_sgn sgn,
adamc@42:                                      space,
adamc@42:                                      string "where",
adamc@42:                                      space,
adamc@42:                                      string "con",
adamc@42:                                      space,
adamc@42:                                      string x,
adamc@42:                                      space,
adamc@42:                                      string "=",
adamc@42:                                      space,
adamc@42:                                      p_con c]
adamc@59:       | SgnProj (m, ms, x) => p_list_sep (string ".") string (m :: ms @ [x])
adamc@59:                                    
adamc@42: 
adamc@123: fun p_vali (x, co, e) =
adamc@123:     case co of
adamc@123:         NONE => box [string x,
adamc@123:                      space,
adamc@123:                      string "=",
adamc@123:                      space,
adamc@123:                      p_exp e]
adamc@123:       | SOME t => box [string x,
adamc@123:                        space,
adamc@123:                        string ":",
adamc@123:                        space,
adamc@123:                        p_con t,
adamc@123:                        space,
adamc@123:                        string "=",
adamc@123:                        space,
adamc@123:                        p_exp e]
adamc@123: 
adamc@1: fun p_decl ((d, _) : decl) =
adamc@1:     case d of
adamc@1:         DCon (x, NONE, c) => box [string "con",
adamc@1:                                   space,
adamc@1:                                   string x,
adamc@1:                                   space,
adamc@1:                                   string "=",
adamc@1:                                   space,
adamc@1:                                   p_con c]
adamc@1:       | DCon (x, SOME k, c) => box [string "con",
adamc@1:                                     space,
adamc@1:                                     string x,
adamc@1:                                     space,
adamc@1:                                     string "::",
adamc@1:                                     space,
adamc@1:                                     p_kind k,
adamc@1:                                     space,
adamc@1:                                     string "=",
adamc@1:                                     space,
adamc@1:                                     p_con c]
adamc@156:       | DDatatype x => p_datatype x
adamc@156:       | DDatatypeImp (x, ms, x') =>
adamc@156:         box [string "datatype",
adamc@156:              space,
adamc@156:              string x,
adamc@156:              space,
adamc@156:              string "=",
adamc@156:              space,
adamc@156:              string "datatype",
adamc@156:              space,
adamc@156:              p_list_sep (string ".") string (ms @ [x'])]
adamc@123:       | DVal vi => box [string "val",
adamc@123:                         space,
adamc@123:                         p_vali vi]
adamc@123:       | DValRec vis => box [string "val",
adamc@123:                             space,
adamc@123:                             string "rec",
adamc@123:                             space,
adamc@123:                             p_list_sep (box [newline, string "and", space]) p_vali vis]
adamc@1: 
adamc@30:       | DSgn (x, sgn) => box [string "signature",
adamc@30:                               space,
adamc@30:                               string x,
adamc@30:                               space,
adamc@30:                               string "=",
adamc@30:                               space,
adamc@30:                               p_sgn sgn]
adamc@30:       | DStr (x, NONE, str) => box [string "structure",
adamc@30:                                     space,
adamc@30:                                     string x,
adamc@30:                                     space,
adamc@30:                                     string "=",
adamc@30:                                     space,
adamc@30:                                     p_str str]
adamc@30:       | DStr (x, SOME sgn, str) => box [string "structure",
adamc@30:                                         space,
adamc@30:                                         string x,
adamc@30:                                         space,
adamc@30:                                         string ":",
adamc@30:                                         space,
adamc@30:                                         p_sgn sgn,
adamc@30:                                         space,
adamc@30:                                         string "=",
adamc@30:                                         space,
adamc@30:                                         p_str str]
adamc@48:       | DFfiStr (x, sgn) => box [string "extern",
adamc@48:                                  space,
adamc@48:                                  string "structure",
adamc@48:                                  space,
adamc@48:                                  string x,
adamc@48:                                  space,
adamc@48:                                  string ":",
adamc@48:                                  space,
adamc@48:                                  p_sgn sgn]
adamc@61:       | DOpen (m, ms) => box [string "open",
adamc@61:                               space,
adamc@61:                               p_list_sep (string ".") string (m :: ms)]
adamc@88:       | DConstraint (c1, c2) => box [string "constraint",
adamc@88:                                      space,
adamc@88:                                      p_con c1,
adamc@88:                                      space,
adamc@88:                                      string "~",
adamc@88:                                      space,
adamc@88:                                      p_con c2]
adamc@88:       | DOpenConstraints (m, ms) => box [string "open",
adamc@88:                                          space,
adamc@88:                                          string "constraints",
adamc@88:                                          space,
adamc@88:                                          p_list_sep (string ".") string (m :: ms)]
adamc@30: 
adamc@109:       | DExport str => box [string "export",
adamc@109:                             space,
adamc@109:                             p_str str]
adamc@203:       | DTable (x, c) => box [string "table",
adamc@203:                               space,
adamc@203:                               string x,
adamc@203:                               space,
adamc@203:                               string ":",
adamc@203:                               space,
adamc@203:                               p_con c]
adamc@211:       | DClass (x, c) => box [string "class",
adamc@211:                               space,
adamc@211:                               string x,
adamc@211:                               space,
adamc@211:                               string "=",
adamc@211:                               space,
adamc@211:                               p_con c]
adamc@100: 
adamc@30: and p_str (str, _) =
adamc@30:     case str of
adamc@30:         StrConst ds => box [string "struct",
adamc@30:                             newline,
adamc@30:                             p_list_sep newline p_decl ds,
adamc@30:                             newline,
adamc@30:                             string "end"]
adamc@30:       | StrVar x => string x
adamc@34:       | StrProj (str, x) => box [p_str str,
adamc@34:                                  string ".",
adamc@34:                                  string x]
adamc@40:       | StrFun (x, sgn, NONE, str) => box [string "functor",
adamc@40:                                            space,
adamc@40:                                            string "(",
adamc@40:                                            string x,
adamc@40:                                            space,
adamc@40:                                            string ":",
adamc@40:                                            p_sgn sgn,
adamc@40:                                            string ")",
adamc@40:                                            space,
adamc@40:                                            string "=>",
adamc@40:                                            space,
adamc@40:                                            p_str str]
adamc@40:       | StrFun (x, sgn, SOME sgn', str) => box [string "functor",
adamc@40:                                                 space,
adamc@40:                                                 string "(",
adamc@40:                                                 string x,
adamc@40:                                                 space,
adamc@40:                                                 string ":",
adamc@40:                                                 p_sgn sgn,
adamc@40:                                                 string ")",
adamc@40:                                                 space,
adamc@40:                                                 string ":",
adamc@40:                                                 space,
adamc@40:                                                 p_sgn sgn',
adamc@40:                                                 space,
adamc@40:                                                 string "=>",
adamc@40:                                                 space,
adamc@40:                                                 p_str str]
adamc@44:       | StrApp (str1, str2) => box [p_str str1,
adamc@44:                                     string "(",
adamc@44:                                     p_str str2,
adamc@44:                                     string ")"]
adamc@30: 
adamc@1: val p_file = p_list_sep newline p_decl
adamc@1: 
adamc@1: end