adamc@3: (* Copyright (c) 2008, Adam Chlipala
adamc@3:  * All rights reserved.
adamc@3:  *
adamc@3:  * Redistribution and use in source and binary forms, with or without
adamc@3:  * modification, are permitted provided that the following conditions are met:
adamc@3:  *
adamc@3:  * - Redistributions of source code must retain the above copyright notice,
adamc@3:  *   this list of conditions and the following disclaimer.
adamc@3:  * - Redistributions in binary form must reproduce the above copyright notice,
adamc@3:  *   this list of conditions and the following disclaimer in the documentation
adamc@3:  *   and/or other materials provided with the distribution.
adamc@3:  * - The names of contributors may not be used to endorse or promote products
adamc@3:  *   derived from this software without specific prior written permission.
adamc@3:  *
adamc@3:  * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS"
adamc@3:  * AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
adamc@3:  * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
adamc@3:  * ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT OWNER OR CONTRIBUTORS BE
adamc@3:  * LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR 
adamc@3:  * CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
adamc@3:  * SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
adamc@3:  * INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
adamc@3:  * CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
adamc@3:  * ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
adamc@3:  * POSSIBILITY OF SUCH DAMAGE.
adamc@3:  *)
adamc@3: 
adamc@3: (* Pretty-printing elaborated Laconic/Web *)
adamc@3: 
adamc@3: structure ElabPrint :> ELAB_PRINT = struct
adamc@3: 
adamc@3: open Print.PD
adamc@3: open Print
adamc@3: 
adamc@3: open Elab
adamc@3: 
adamc@3: structure E = ElabEnv
adamc@3: 
adamc@3: fun p_kind' par (k, _) =
adamc@3:     case k of
adamc@3:         KType => string "Type"
adamc@3:       | KArrow (k1, k2) => parenIf par (box [p_kind' true k1,
adamc@3:                                              space,
adamc@3:                                              string "->",
adamc@3:                                              space,
adamc@3:                                              p_kind k2])
adamc@3:       | KName => string "Name"
adamc@3:       | KRecord k => box [string "{", p_kind k, string "}"]
adamc@3: 
adamc@3:       | KError => string "<ERROR>"
adamc@3:       | KUnif (_, ref (SOME k)) => p_kind' par k
adamc@3:       | KUnif (s, _) => string ("<UNIF:" ^ s ^ ">")
adamc@3: 
adamc@3: and p_kind k = p_kind' false k
adamc@3: 
adamc@3: fun p_explicitness e =
adamc@3:     case e of
adamc@3:         Explicit => string "::"
adamc@3:       | Implicit => string ":::"
adamc@3: 
adamc@3: fun p_con' par env (c, _) =
adamc@3:     case c of
adamc@3:         TFun (t1, t2) => parenIf par (box [p_con' true env t1,
adamc@3:                                            space,
adamc@3:                                            string "->",
adamc@3:                                            space,
adamc@3:                                            p_con env t2])
adamc@3:       | TCFun (e, x, k, c) => parenIf par (box [string x,
adamc@3:                                                 space,
adamc@3:                                                 p_explicitness e,
adamc@3:                                                 space,
adamc@3:                                                 p_kind k,
adamc@3:                                                 space,
adamc@3:                                                 string "->",
adamc@3:                                                 space,
adamc@3:                                                 p_con (E.pushCRel env x k) c])
adamc@3:       | TRecord (CRecord (_, xcs), _) => box [string "{",
adamc@3:                                               p_list (fn (x, c) =>
adamc@3:                                                          box [p_con env x,
adamc@3:                                                               space,
adamc@3:                                                               string ":",
adamc@3:                                                               space,
adamc@3:                                                               p_con env c]) xcs,
adamc@3:                                               string "}"]
adamc@3:       | TRecord c => box [string "$",
adamc@3:                           p_con' true env c]
adamc@3: 
adamc@3:       | CRel n => string (#1 (E.lookupCRel env n) ^ "_" ^ Int.toString n)
adamc@3:       | CNamed n => string (#1 (E.lookupCNamed env n) ^ "__" ^ Int.toString n)
adamc@3: 
adamc@3:       | CApp (c1, c2) => parenIf par (box [p_con env c1,
adamc@3:                                            space,
adamc@3:                                            p_con' true env c2])
adamc@3:       | CAbs (e, x, k, c) => parenIf par (box [string "fn",
adamc@3:                                                space,
adamc@3:                                                string x,
adamc@3:                                                space,
adamc@3:                                                p_explicitness e,
adamc@3:                                                space,
adamc@3:                                                p_kind k,
adamc@3:                                                space,
adamc@3:                                                string "=>",
adamc@3:                                                space,
adamc@3:                                                p_con (E.pushCRel env x k) c])
adamc@3: 
adamc@3:       | CName s => box [string "#", string s]
adamc@3: 
adamc@3:       | CRecord (k, xcs) => parenIf par (box [string "[",
adamc@3:                                               p_list (fn (x, c) =>
adamc@3:                                                          box [p_con env x,
adamc@3:                                                               space,
adamc@3:                                                               string "=",
adamc@3:                                                               space,
adamc@3:                                                               p_con env c]) xcs,
adamc@3:                                               string "]::",
adamc@3:                                               p_kind k])
adamc@3:       | CConcat (c1, c2) => parenIf par (box [p_con' true env c1,
adamc@3:                                               space,
adamc@3:                                               string "++",
adamc@3:                                               space,
adamc@3:                                               p_con env c2])
adamc@3: 
adamc@3:       | CError => string "<ERROR>"
adamc@6:       | CUnif (_, _, ref (SOME c)) => p_con' par env c
adamc@6:       | CUnif (k, s, _) => box [string ("<UNIF:" ^ s ^ "::"),
adamc@6:                                p_kind k,
adamc@6:                                string ">"]
adamc@3:         
adamc@3: and p_con env = p_con' false env
adamc@3: 
adamc@3: fun p_decl env ((d, _) : decl) =
adamc@3:     case d of
adamc@5:         DCon (x, n, k, c) => box [string "con",
adamc@5:                                   space,
adamc@5:                                   string x,
adamc@5:                                   string "__",
adamc@5:                                   string (Int.toString n),
adamc@5:                                   space,
adamc@5:                                   string "::",
adamc@5:                                   space,
adamc@5:                                   p_kind k,
adamc@5:                                   space,
adamc@5:                                   string "=",
adamc@5:                                   space,
adamc@5:                                   p_con env c]
adamc@3: 
adamc@3: fun p_file env file =
adamc@3:     let
adamc@5:         val (_, pds) = ListUtil.mapfoldl (fn (d, env) =>
adamc@5:                                              (ElabUtil.declBinds env d,
adamc@5:                                               p_decl env d))
adamc@5:                              env file
adamc@3:     in
adamc@3:         p_list_sep newline (fn x => x) pds
adamc@3:     end
adamc@3: 
adamc@3: end