haftmann@37743: (* Title: Tools/Code/code_printer.ML haftmann@28060: Author: Florian Haftmann, TU Muenchen haftmann@28060: haftmann@28060: Generic operations for pretty printing of target language code. haftmann@28060: *) haftmann@28060: haftmann@28060: signature CODE_PRINTER = haftmann@28060: sig haftmann@32908: type itype = Code_Thingol.itype haftmann@32908: type iterm = Code_Thingol.iterm haftmann@32908: type const = Code_Thingol.const haftmann@32908: type dict = Code_Thingol.dict haftmann@32908: haftmann@35228: val eqn_error: thm option -> string -> 'a haftmann@28060: haftmann@28060: val @@ : 'a * 'a -> 'a list haftmann@28060: val @| : 'a list * 'a -> 'a list haftmann@28060: val str: string -> Pretty.T haftmann@28060: val concat: Pretty.T list -> Pretty.T haftmann@28060: val brackets: Pretty.T list -> Pretty.T haftmann@34178: val enclose: string -> string -> Pretty.T list -> Pretty.T haftmann@39011: val commas: Pretty.T list -> Pretty.T list haftmann@34178: val enum: string -> string -> string -> Pretty.T list -> Pretty.T haftmann@34178: val enum_default: string -> string -> string -> string -> Pretty.T list -> Pretty.T haftmann@28060: val semicolon: Pretty.T list -> Pretty.T haftmann@33986: val doublesemicolon: Pretty.T list -> Pretty.T haftmann@34178: val indent: int -> Pretty.T -> Pretty.T haftmann@39290: val markup_stmt: string -> Pretty.T -> Pretty.T haftmann@39290: val format: string list -> int -> Pretty.T -> string haftmann@28060: haftmann@30636: val first_upper: string -> string haftmann@30636: val first_lower: string -> string haftmann@30636: type var_ctxt haftmann@30636: val make_vars: string list -> var_ctxt haftmann@30636: val intro_vars: string list -> var_ctxt -> var_ctxt haftmann@30636: val lookup_var: var_ctxt -> string -> string haftmann@32913: val intro_base_names: (string -> bool) -> (string -> string) haftmann@32913: -> string list -> var_ctxt -> var_ctxt haftmann@32908: val aux_params: var_ctxt -> iterm list list -> string list haftmann@30636: haftmann@31056: type literals haftmann@31056: val Literals: { literal_char: string -> string, literal_string: string -> string, haftmann@38195: literal_numeral: int -> string, haftmann@38195: literal_positive_numeral: int -> string, haftmann@38195: literal_alternative_numeral: int -> string, haftmann@34931: literal_naive_numeral: int -> string, haftmann@31056: literal_list: Pretty.T list -> Pretty.T, infix_cons: int * string } haftmann@31056: -> literals haftmann@31056: val literal_char: literals -> string -> string haftmann@31056: val literal_string: literals -> string -> string haftmann@34931: val literal_numeral: literals -> int -> string haftmann@34931: val literal_positive_numeral: literals -> int -> string haftmann@38195: val literal_alternative_numeral: literals -> int -> string haftmann@34931: val literal_naive_numeral: literals -> int -> string haftmann@31056: val literal_list: literals -> Pretty.T list -> Pretty.T haftmann@31056: val infix_cons: literals -> int * string haftmann@31056: haftmann@28060: type lrx haftmann@28060: val L: lrx haftmann@28060: val R: lrx haftmann@28060: val X: lrx haftmann@28060: type fixity haftmann@28060: val BR: fixity haftmann@28060: val NOBR: fixity haftmann@28060: val INFX: int * lrx -> fixity haftmann@28060: val APP: fixity haftmann@28060: val brackify: fixity -> Pretty.T list -> Pretty.T haftmann@37239: val brackify_infix: int * lrx -> fixity -> Pretty.T * Pretty.T * Pretty.T -> Pretty.T haftmann@31665: val brackify_block: fixity -> Pretty.T -> Pretty.T list -> Pretty.T -> Pretty.T haftmann@37638: val applify: string -> string -> ('a -> Pretty.T) -> fixity -> Pretty.T -> 'a list -> Pretty.T haftmann@39148: val tuplify: (fixity -> 'a -> Pretty.T) -> fixity -> 'a list -> Pretty.T option haftmann@39148: haftmann@28060: type tyco_syntax haftmann@34141: type simple_const_syntax haftmann@37854: type complex_const_syntax haftmann@28060: type const_syntax haftmann@37854: type activated_complex_const_syntax haftmann@37854: datatype activated_const_syntax = Plain_const_syntax of int * string haftmann@37854: | Complex_const_syntax of activated_complex_const_syntax haftmann@37854: val requires_args: const_syntax -> int haftmann@37854: val parse_const_syntax: Token.T list -> const_syntax option * Token.T list haftmann@37854: val parse_tyco_syntax: Token.T list -> tyco_syntax option * Token.T list haftmann@37854: val plain_const_syntax: string -> const_syntax haftmann@37849: val simple_const_syntax: simple_const_syntax -> const_syntax haftmann@37854: val complex_const_syntax: complex_const_syntax -> const_syntax haftmann@31056: val activate_const_syntax: theory -> literals haftmann@37854: -> string -> const_syntax -> Code_Thingol.naming -> activated_const_syntax * Code_Thingol.naming haftmann@35228: val gen_print_app: (thm option -> var_ctxt -> const * iterm list -> Pretty.T list) haftmann@35228: -> (thm option -> var_ctxt -> fixity -> iterm -> Pretty.T) haftmann@37849: -> (string -> activated_const_syntax option) haftmann@35228: -> thm option -> var_ctxt -> fixity -> const * iterm list -> Pretty.T haftmann@35228: val gen_print_bind: (thm option -> var_ctxt -> fixity -> iterm -> Pretty.T) haftmann@35228: -> thm option -> fixity haftmann@31889: -> iterm -> var_ctxt -> Pretty.T * var_ctxt haftmann@28060: end; haftmann@28060: haftmann@28060: structure Code_Printer : CODE_PRINTER = haftmann@28060: struct haftmann@28060: haftmann@28060: open Code_Thingol; haftmann@28060: haftmann@39278: (** generic nonsense *) haftmann@39278: wenzelm@44818: fun eqn_error (SOME thm) s = wenzelm@44818: error (s ^ ",\nin equation " ^ Display.string_of_thm_without_context thm) haftmann@35228: | eqn_error NONE s = error s; haftmann@28060: haftmann@39278: val code_presentationN = "code_presentation"; haftmann@39291: val stmt_nameN = "stmt_name"; haftmann@39278: val _ = Markup.add_mode code_presentationN YXML.output_markup; haftmann@39278: haftmann@39278: haftmann@34071: (** assembling and printing text pieces **) haftmann@28060: haftmann@28060: infixr 5 @@; haftmann@28060: infixr 5 @|; haftmann@28060: fun x @@ y = [x, y]; haftmann@28060: fun xs @| y = xs @ [y]; haftmann@39884: val str = Print_Mode.setmp [] Pretty.str; haftmann@28060: val concat = Pretty.block o Pretty.breaks; haftmann@39884: val commas = Print_Mode.setmp [] Pretty.commas; haftmann@39884: fun enclose l r = Print_Mode.setmp [] (Pretty.enclose l r); haftmann@34178: val brackets = enclose "(" ")" o Pretty.breaks; haftmann@39884: fun enum sep l r = Print_Mode.setmp [] (Pretty.enum sep l r); haftmann@34178: fun enum_default default sep l r [] = str default haftmann@34178: | enum_default default sep l r xs = enum sep l r xs; haftmann@28060: fun semicolon ps = Pretty.block [concat ps, str ";"]; haftmann@33986: fun doublesemicolon ps = Pretty.block [concat ps, str ";;"]; haftmann@39884: fun indent i = Print_Mode.setmp [] (Pretty.indent i); haftmann@28060: haftmann@39884: fun markup_stmt name = Print_Mode.setmp [code_presentationN] haftmann@39303: (Pretty.mark (code_presentationN, [(stmt_nameN, name)])); haftmann@39296: haftmann@39801: fun filter_presentation [] tree = haftmann@39801: Buffer.empty wenzelm@44818: |> fold XML.add_content tree haftmann@39801: | filter_presentation presentation_names tree = haftmann@39291: let wenzelm@44818: fun is_selected (name, attrs) = haftmann@39801: name = code_presentationN haftmann@39801: andalso member (op =) presentation_names (the (Properties.get attrs stmt_nameN)); haftmann@39801: fun add_content_with_space tree (is_first, buf) = haftmann@39801: buf haftmann@39801: |> not is_first ? Buffer.add "\n\n" haftmann@39801: |> XML.add_content tree haftmann@39801: |> pair false; haftmann@39801: fun filter (XML.Elem (name_attrs, xs)) = haftmann@39801: fold (if is_selected name_attrs then add_content_with_space else filter) xs haftmann@48447: | filter (XML.Text _) = I; haftmann@39801: in snd (fold filter tree (true, Buffer.empty)) end; haftmann@39296: haftmann@39296: fun format presentation_names width = haftmann@39884: Print_Mode.setmp [code_presentationN] (Pretty.string_of_margin width) haftmann@39296: #> YXML.parse_body haftmann@39801: #> filter_presentation presentation_names haftmann@39902: #> Buffer.add "\n" haftmann@39801: #> Buffer.content; haftmann@34071: haftmann@28060: haftmann@30636: (** names and variable name contexts **) haftmann@30636: haftmann@30636: type var_ctxt = string Symtab.table * Name.context; haftmann@30636: haftmann@30636: fun make_vars names = (fold (fn name => Symtab.update_new (name, name)) names Symtab.empty, haftmann@30636: Name.make_context names); haftmann@30636: haftmann@30636: fun intro_vars names (namemap, namectxt) = haftmann@30636: let wenzelm@44208: val (names', namectxt') = fold_map Name.variant names namectxt; haftmann@30636: val namemap' = fold2 (curry Symtab.update) names names' namemap; haftmann@30636: in (namemap', namectxt') end; haftmann@30636: wenzelm@44818: fun lookup_var (namemap, _) name = wenzelm@44818: case Symtab.lookup namemap name of wenzelm@44818: SOME name' => name' haftmann@30636: | NONE => error ("Invalid name in context: " ^ quote name); haftmann@30636: wenzelm@40875: val first_upper = implode o nth_map 0 Symbol.to_ascii_upper o raw_explode; wenzelm@40875: val first_lower = implode o nth_map 0 Symbol.to_ascii_lower o raw_explode; haftmann@30636: haftmann@32908: fun aux_params vars lhss = haftmann@32908: let haftmann@32908: fun fish_param _ (w as SOME _) = w haftmann@32908: | fish_param (IVar (SOME v)) NONE = SOME v haftmann@32908: | fish_param _ NONE = NONE; haftmann@32908: fun fillup_param _ (_, SOME v) = v haftmann@32908: | fillup_param x (i, NONE) = x ^ string_of_int i; haftmann@32908: val fished1 = fold (map2 fish_param) lhss (replicate (length (hd lhss)) NONE); wenzelm@44206: val x = singleton (Name.variant_list (map_filter I fished1)) "x"; haftmann@32908: val fished2 = map_index (fillup_param x) fished1; wenzelm@44208: val (fished3, _) = fold_map Name.variant fished2 Name.context; haftmann@32908: val vars' = intro_vars fished3 vars; haftmann@32908: in map (lookup_var vars') fished3 end; haftmann@32908: wenzelm@44818: fun intro_base_names no_syntax deresolve names = names haftmann@32913: |> map_filter (fn name => if no_syntax name then haftmann@32913: let val name' = deresolve name in haftmann@32913: if Long_Name.is_qualified name' then NONE else SOME name' haftmann@32913: end else NONE) haftmann@32913: |> intro_vars; haftmann@32913: haftmann@30636: haftmann@31056: (** pretty literals **) haftmann@31056: haftmann@31056: datatype literals = Literals of { haftmann@31056: literal_char: string -> string, haftmann@31056: literal_string: string -> string, haftmann@34931: literal_numeral: int -> string, haftmann@34931: literal_positive_numeral: int -> string, haftmann@38195: literal_alternative_numeral: int -> string, haftmann@34931: literal_naive_numeral: int -> string, haftmann@31056: literal_list: Pretty.T list -> Pretty.T, haftmann@31056: infix_cons: int * string haftmann@31056: }; haftmann@31056: haftmann@31056: fun dest_Literals (Literals lits) = lits; haftmann@31056: haftmann@31056: val literal_char = #literal_char o dest_Literals; haftmann@31056: val literal_string = #literal_string o dest_Literals; haftmann@31056: val literal_numeral = #literal_numeral o dest_Literals; haftmann@34931: val literal_positive_numeral = #literal_positive_numeral o dest_Literals; haftmann@38195: val literal_alternative_numeral = #literal_alternative_numeral o dest_Literals; haftmann@34931: val literal_naive_numeral = #literal_naive_numeral o dest_Literals; haftmann@31056: val literal_list = #literal_list o dest_Literals; haftmann@31056: val infix_cons = #infix_cons o dest_Literals; haftmann@31056: haftmann@31056: haftmann@28060: (** syntax printer **) haftmann@28060: haftmann@28060: (* binding priorities *) haftmann@28060: haftmann@28060: datatype lrx = L | R | X; haftmann@28060: haftmann@28060: datatype fixity = haftmann@28060: BR haftmann@28060: | NOBR haftmann@28060: | INFX of (int * lrx); haftmann@28060: haftmann@28060: val APP = INFX (~1, L); haftmann@28060: haftmann@28060: fun fixity_lrx L L = false haftmann@28060: | fixity_lrx R R = false haftmann@28060: | fixity_lrx _ _ = true; haftmann@28060: haftmann@28060: fun fixity NOBR _ = false haftmann@28060: | fixity _ NOBR = false haftmann@33986: | fixity (INFX (pr, lr)) (INFX (print_ctxt, lr_ctxt)) = haftmann@33986: pr < print_ctxt haftmann@33986: orelse pr = print_ctxt haftmann@28060: andalso fixity_lrx lr lr_ctxt haftmann@33986: orelse print_ctxt = ~1 haftmann@28060: | fixity BR (INFX _) = false haftmann@28060: | fixity _ _ = true; haftmann@28060: haftmann@28060: fun gen_brackify _ [p] = p haftmann@34178: | gen_brackify true (ps as _::_) = enclose "(" ")" ps haftmann@28060: | gen_brackify false (ps as _::_) = Pretty.block ps; haftmann@28060: haftmann@28060: fun brackify fxy_ctxt = haftmann@28060: gen_brackify (fixity BR fxy_ctxt) o Pretty.breaks; haftmann@28060: haftmann@37239: fun brackify_infix infx fxy_ctxt (l, m, r) = haftmann@37239: (if fixity (INFX infx) fxy_ctxt then enclose "(" ")" else Pretty.block) haftmann@37239: ([l, str " ", m, Pretty.brk 1, r]); haftmann@28060: haftmann@31665: fun brackify_block fxy_ctxt p1 ps p2 = haftmann@31665: let val p = Pretty.block_enclose (p1, p2) ps haftmann@31665: in if fixity BR fxy_ctxt haftmann@34178: then enclose "(" ")" [p] haftmann@31665: else p haftmann@31665: end; haftmann@31665: haftmann@37638: fun applify opn cls f fxy_ctxt p [] = p haftmann@37638: | applify opn cls f fxy_ctxt p ps = haftmann@34242: (if (fixity BR fxy_ctxt) then enclose "(" ")" else Pretty.block) haftmann@37638: (p @@ enum "," opn cls (map f ps)); haftmann@34242: haftmann@39148: fun tuplify _ _ [] = NONE haftmann@39148: | tuplify print fxy [x] = SOME (print fxy x) haftmann@39148: | tuplify print _ xs = SOME (enum "," "(" ")" (map (print NOBR) xs)); haftmann@39148: haftmann@28060: haftmann@28060: (* generic syntax *) haftmann@28060: haftmann@28060: type tyco_syntax = int * ((fixity -> itype -> Pretty.T) haftmann@28060: -> fixity -> itype list -> Pretty.T); haftmann@34141: haftmann@34141: type simple_const_syntax = int * ((fixity -> iterm -> Pretty.T) haftmann@34141: -> fixity -> (iterm * itype) list -> Pretty.T); haftmann@37854: haftmann@37854: type complex_const_syntax = int * (string list * (literals -> string list haftmann@31056: -> (var_ctxt -> fixity -> iterm -> Pretty.T) haftmann@35228: -> thm option -> var_ctxt -> fixity -> (iterm * itype) list -> Pretty.T)); haftmann@37854: haftmann@37854: datatype const_syntax = plain_const_syntax of string haftmann@37854: | complex_const_syntax of complex_const_syntax; haftmann@37854: haftmann@37854: fun requires_args (plain_const_syntax _) = 0 haftmann@37854: | requires_args (complex_const_syntax (k, _)) = k; haftmann@28060: haftmann@34163: fun simple_const_syntax syn = wenzelm@44818: complex_const_syntax wenzelm@44818: (apsnd (fn f => ([], (fn _ => fn _ => fn print => fn _ => fn vars => f (print vars)))) syn); haftmann@28060: haftmann@37854: type activated_complex_const_syntax = int * ((var_ctxt -> fixity -> iterm -> Pretty.T) haftmann@37854: -> thm option -> var_ctxt -> fixity -> (iterm * itype) list -> Pretty.T) haftmann@31056: haftmann@37854: datatype activated_const_syntax = Plain_const_syntax of int * string haftmann@37854: | Complex_const_syntax of activated_complex_const_syntax; haftmann@37854: haftmann@37854: fun activate_const_syntax thy literals c (plain_const_syntax s) naming = haftmann@37854: (Plain_const_syntax (Code.args_number thy c, s), naming) haftmann@37854: | activate_const_syntax thy literals c (complex_const_syntax (n, (cs, f))) naming = haftmann@37854: fold_map (Code_Thingol.ensure_declared_const thy) cs naming haftmann@37854: |-> (fn cs' => pair (Complex_const_syntax (n, f literals cs'))); haftmann@37854: wenzelm@44818: fun gen_print_app print_app_expr print_term const_syntax some_thm vars fxy bulwahn@45880: (app as ((c, ((_, (arg_tys, _)), _)), ts)) = wenzelm@44818: case const_syntax c of wenzelm@44818: NONE => brackify fxy (print_app_expr some_thm vars app) wenzelm@44818: | SOME (Plain_const_syntax (_, s)) => wenzelm@44818: brackify fxy (str s :: map (print_term some_thm vars BR) ts) wenzelm@44818: | SOME (Complex_const_syntax (k, print)) => wenzelm@44818: let wenzelm@44818: fun print' fxy ts = bulwahn@45880: print (print_term some_thm) some_thm vars fxy (ts ~~ take k arg_tys); wenzelm@44818: in wenzelm@44818: if k = length ts wenzelm@44818: then print' fxy ts haftmann@28060: else if k < length ts wenzelm@44818: then case chop k ts of (ts1, ts2) => wenzelm@44818: brackify fxy (print' APP ts1 :: map (print_term some_thm vars BR) ts2) wenzelm@44818: else print_term some_thm vars fxy (Code_Thingol.eta_expand k app) wenzelm@44818: end; haftmann@28060: haftmann@33986: fun gen_print_bind print_term thm (fxy : fixity) pat vars = haftmann@28060: let haftmann@31889: val vs = Code_Thingol.fold_varnames (insert (op =)) pat []; haftmann@31873: val vars' = intro_vars vs vars; haftmann@33986: in (print_term thm vars' fxy pat, vars') end; haftmann@28060: haftmann@28060: haftmann@28060: (* mixfix syntax *) haftmann@28060: haftmann@28060: datatype 'a mixfix = haftmann@28060: Arg of fixity haftmann@37854: | String of string haftmann@37854: | Break; haftmann@28060: haftmann@28060: fun mk_mixfix prep_arg (fixity_this, mfx) = haftmann@28060: let haftmann@28060: fun is_arg (Arg _) = true haftmann@28060: | is_arg _ = false; haftmann@28060: val i = (length o filter is_arg) mfx; haftmann@28060: fun fillin _ [] [] = haftmann@28060: [] haftmann@33986: | fillin print (Arg fxy :: mfx) (a :: args) = haftmann@33986: (print fxy o prep_arg) a :: fillin print mfx args haftmann@37854: | fillin print (String s :: mfx) args = haftmann@37854: str s :: fillin print mfx args haftmann@37854: | fillin print (Break :: mfx) args = haftmann@37854: Pretty.brk 1 :: fillin print mfx args; haftmann@28060: in haftmann@33986: (i, fn print => fn fixity_ctxt => fn args => haftmann@33986: gen_brackify (fixity fixity_this fixity_ctxt) (fillin print mfx args)) haftmann@28060: end; haftmann@28060: haftmann@28060: fun parse_infix prep_arg (x, i) s = haftmann@28060: let haftmann@28060: val l = case x of L => INFX (i, L) | _ => INFX (i, X); haftmann@28060: val r = case x of R => INFX (i, R) | _ => INFX (i, X); haftmann@28060: in haftmann@37854: mk_mixfix prep_arg (INFX (i, x), [Arg l, String " ", String s, Break, Arg r]) haftmann@28060: end; haftmann@28060: haftmann@37854: fun parse_mixfix mk_plain mk_complex prep_arg s = haftmann@28060: let haftmann@28060: val sym_any = Scan.one Symbol.is_regular; haftmann@28060: val parse = Scan.optional ($$ "!" >> K true) false -- Scan.repeat ( haftmann@28060: ($$ "(" -- $$ "_" -- $$ ")" >> K (Arg NOBR)) haftmann@28060: || ($$ "_" >> K (Arg BR)) haftmann@37854: || ($$ "/" |-- Scan.repeat ($$ " ") >> (K Break)) haftmann@28060: || (Scan.repeat1 haftmann@28060: ( $$ "'" |-- sym_any haftmann@28060: || Scan.unless ($$ "_" || $$ "/" || $$ "(" |-- $$ "_" |-- $$ ")") haftmann@37854: sym_any) >> (String o implode))); wenzelm@44818: fun err s (_, NONE) = (fn () => "malformed mixfix annotation: " ^ quote s) wenzelm@44818: | err _ (_, SOME msg) = msg; wenzelm@44818: in wenzelm@44818: case Scan.finite Symbol.stopper parse (Symbol.explode s) of wenzelm@44818: ((false, [String s]), []) => mk_plain s haftmann@37854: | ((_, p as [_]), []) => mk_complex (mk_mixfix prep_arg (NOBR, p)) haftmann@37854: | ((b, p as _ :: _ :: _), []) => mk_complex (mk_mixfix prep_arg (if b then NOBR else BR, p)) wenzelm@44818: | _ => Scan.!! (err s) Scan.fail () haftmann@28060: end; haftmann@28060: haftmann@37854: fun parse_syntax mk_plain mk_complex prep_arg = haftmann@37854: Scan.option ( wenzelm@47823: ((@{keyword "infix"} >> K X) wenzelm@47823: || (@{keyword "infixl"} >> K L) wenzelm@47823: || (@{keyword "infixr"} >> K R)) haftmann@37854: -- Parse.nat -- Parse.string haftmann@37854: >> (fn ((x, i), s) => mk_complex (parse_infix prep_arg (x, i) s)) haftmann@37854: || Parse.string >> (fn s => (parse_mixfix mk_plain mk_complex prep_arg s))); haftmann@28060: wenzelm@38145: fun parse_tyco_syntax x = parse_syntax (fn s => (0, (K o K o K o str) s)) I I x; haftmann@37854: haftmann@37854: val parse_const_syntax = parse_syntax plain_const_syntax simple_const_syntax fst; haftmann@37854: haftmann@28060: end; (*struct*)