1421 lines
53 KiB
OCaml
1421 lines
53 KiB
OCaml
(**************************************************************************)
|
|
(* *)
|
|
(* OCaml *)
|
|
(* *)
|
|
(* Thomas Gazagnaire, OCamlPro *)
|
|
(* Fabrice Le Fessant, INRIA Saclay *)
|
|
(* Hongbo Zhang, University of Pennsylvania *)
|
|
(* *)
|
|
(* Copyright 2007 Institut National de Recherche en Informatique et *)
|
|
(* en Automatique. *)
|
|
(* *)
|
|
(* All rights reserved. This file is distributed under the terms of *)
|
|
(* the GNU Lesser General Public License version 2.1, with the *)
|
|
(* special exception on linking described in the file LICENSE. *)
|
|
(* *)
|
|
(**************************************************************************)
|
|
|
|
(* Original Code from Ber-metaocaml, modified for 3.12.0 and fixed *)
|
|
(* Printing code expressions *)
|
|
(* Authors: Ed Pizzi, Fabrice Le Fessant *)
|
|
(* Extensive Rewrite: Hongbo Zhang: University of Pennsylvania *)
|
|
(* TODO more fine-grained precedence pretty-printing *)
|
|
|
|
open Asttypes
|
|
open Format
|
|
open Location
|
|
open Longident
|
|
open Parsetree
|
|
|
|
let prefix_symbols = [ '!'; '?'; '~' ] ;;
|
|
let infix_symbols = [ '='; '<'; '>'; '@'; '^'; '|'; '&'; '+'; '-'; '*'; '/';
|
|
'$'; '%' ]
|
|
|
|
let special_infix_strings =
|
|
["asr"; "land"; "lor"; "lsl"; "lsr"; "lxor"; "mod"; "or"; ":="; "!=" ]
|
|
|
|
(* determines if the string is an infix string.
|
|
checks backwards, first allowing a renaming postfix ("_102") which
|
|
may have resulted from Pexp -> Texp -> Pexp translation, then checking
|
|
if all the characters in the beginning of the string are valid infix
|
|
characters. *)
|
|
let fixity_of_string = function
|
|
| s when List.mem s special_infix_strings -> `Infix s
|
|
| s when List.mem s.[0] infix_symbols -> `Infix s
|
|
| s when List.mem s.[0] prefix_symbols -> `Prefix s
|
|
| _ -> `Normal
|
|
|
|
let view_fixity_of_exp = function
|
|
| {pexp_desc = Pexp_ident {txt=Lident l;_};_} -> fixity_of_string l
|
|
| _ -> `Normal ;;
|
|
|
|
let is_infix = function | `Infix _ -> true | _ -> false
|
|
|
|
(* which identifiers are in fact operators needing parentheses *)
|
|
let needs_parens txt =
|
|
is_infix (fixity_of_string txt)
|
|
|| List.mem txt.[0] prefix_symbols
|
|
|
|
(* some infixes need spaces around parens to avoid clashes with comment
|
|
syntax *)
|
|
let needs_spaces txt =
|
|
txt.[0]='*' || txt.[String.length txt - 1] = '*'
|
|
|
|
(* add parentheses to binders when they are in fact infix or prefix operators *)
|
|
let protect_ident ppf txt =
|
|
let format : (_, _, _) format =
|
|
if not (needs_parens txt) then "%s"
|
|
else if needs_spaces txt then "(@;%s@;)"
|
|
else "(%s)"
|
|
in fprintf ppf format txt
|
|
|
|
let protect_longident ppf print_longident longprefix txt =
|
|
let format : (_, _, _) format =
|
|
if not (needs_parens txt) then "%a.%s"
|
|
else if needs_spaces txt then "%a.(@;%s@;)"
|
|
else "%a.(%s)" in
|
|
fprintf ppf format print_longident longprefix txt
|
|
|
|
type space_formatter = (unit, Format.formatter, unit) format
|
|
|
|
let override = function
|
|
| Override -> "!"
|
|
| Fresh -> ""
|
|
|
|
(* variance encoding: need to sync up with the [parser.mly] *)
|
|
let type_variance = function
|
|
| Invariant -> ""
|
|
| Covariant -> "+"
|
|
| Contravariant -> "-"
|
|
|
|
type construct =
|
|
[ `cons of expression list
|
|
| `list of expression list
|
|
| `nil
|
|
| `normal
|
|
| `simple of Longident.t
|
|
| `tuple ]
|
|
|
|
let view_expr x =
|
|
match x.pexp_desc with
|
|
| Pexp_construct ( {txt= Lident "()"; _},_) -> `tuple
|
|
| Pexp_construct ( {txt= Lident "[]";_},_) -> `nil
|
|
| Pexp_construct ( {txt= Lident"::";_},Some _) ->
|
|
let rec loop exp acc = match exp with
|
|
| {pexp_desc=Pexp_construct ({txt=Lident "[]";_},_);_} ->
|
|
(List.rev acc,true)
|
|
| {pexp_desc=
|
|
Pexp_construct ({txt=Lident "::";_},
|
|
Some ({pexp_desc= Pexp_tuple([e1;e2]);_}));_} ->
|
|
loop e2 (e1::acc)
|
|
| e -> (List.rev (e::acc),false) in
|
|
let (ls,b) = loop x [] in
|
|
if b then
|
|
`list ls
|
|
else `cons ls
|
|
| Pexp_construct (x,None) -> `simple (x.txt)
|
|
| _ -> `normal
|
|
|
|
let is_simple_construct :construct -> bool = function
|
|
| `nil | `tuple | `list _ | `simple _ -> true
|
|
| `cons _ | `normal -> false
|
|
|
|
let pp = fprintf
|
|
|
|
class printer ()= object(self:'self)
|
|
val pipe = false
|
|
val semi = false
|
|
val ifthenelse = false
|
|
method under_pipe = {<pipe=true>}
|
|
method under_semi = {<semi=true>}
|
|
method under_ifthenelse = {<ifthenelse=true>}
|
|
method reset_semi = {<semi=false>}
|
|
method reset_ifthenelse = {<ifthenelse=false>}
|
|
method reset_pipe = {<pipe=false>}
|
|
method reset = {<pipe=false;semi=false;ifthenelse=false>}
|
|
method list : 'a . ?sep:space_formatter -> ?first:space_formatter ->
|
|
?last:space_formatter -> (Format.formatter -> 'a -> unit) ->
|
|
Format.formatter -> 'a list -> unit
|
|
= fun ?sep ?first ?last fu f xs ->
|
|
let first = match first with Some x -> x |None -> ("" : _ format6)
|
|
and last = match last with Some x -> x |None -> ("" : _ format6)
|
|
and sep = match sep with Some x -> x |None -> ("@ " : _ format6) in
|
|
let aux f = function
|
|
| [] -> ()
|
|
| [x] -> fu f x
|
|
| xs ->
|
|
let rec loop f = function
|
|
| [x] -> fu f x
|
|
| x::xs -> fu f x; pp f sep; loop f xs;
|
|
| _ -> assert false in begin
|
|
pp f first; loop f xs; pp f last;
|
|
end in
|
|
aux f xs
|
|
method option : 'a. ?first:space_formatter -> ?last:space_formatter ->
|
|
(Format.formatter -> 'a -> unit) -> Format.formatter -> 'a option -> unit =
|
|
fun ?first ?last fu f a ->
|
|
let first = match first with Some x -> x | None -> ("" : _ format6)
|
|
and last = match last with Some x -> x | None -> ("" : _ format6) in
|
|
match a with
|
|
| None -> ()
|
|
| Some x -> pp f first; fu f x; pp f last;
|
|
method paren: 'a . ?first:space_formatter -> ?last:space_formatter ->
|
|
bool -> (Format.formatter -> 'a -> unit) -> Format.formatter -> 'a -> unit =
|
|
fun ?(first=("" : _ format6)) ?(last=("" : _ format6)) b fu f x ->
|
|
if b then (pp f "("; pp f first; fu f x; pp f last; pp f ")")
|
|
else fu f x
|
|
|
|
|
|
method longident f = function
|
|
| Lident s -> protect_ident f s
|
|
| Ldot(y,s) -> protect_longident f self#longident y s
|
|
| Lapply (y,s) ->
|
|
pp f "%a(%a)" self#longident y self#longident s
|
|
method longident_loc f x = pp f "%a" self#longident x.txt
|
|
method constant f = function
|
|
| Pconst_char i -> pp f "%C" i
|
|
| Pconst_string (i, None) -> pp f "%S" i
|
|
| Pconst_string (i, Some delim) -> pp f "{%s|%s|%s}" delim i delim
|
|
| Pconst_integer (i,None) -> self#paren (i.[0]='-') (fun f -> pp f "%s") f i
|
|
| Pconst_integer (i,Some m) ->
|
|
self#paren (i.[0]='-') (fun f (i,m) -> pp f "%s%c" i m) f (i,m)
|
|
| Pconst_float (i,None) -> self#paren (i.[0]='-') (fun f -> pp f "%s") f i
|
|
| Pconst_float (i, Some m) -> self#paren (i.[0]='-') (fun f (i,m) ->
|
|
pp f "%s%c" i m) f (i,m)
|
|
|
|
(* trailing space*)
|
|
method mutable_flag f = function
|
|
| Immutable -> ()
|
|
| Mutable -> pp f "mutable@;"
|
|
method virtual_flag f = function
|
|
| Concrete -> ()
|
|
| Virtual -> pp f "virtual@;"
|
|
|
|
(* trailing space added *)
|
|
method rec_flag f rf =
|
|
match rf with
|
|
| Nonrecursive -> ()
|
|
| Recursive -> pp f "rec "
|
|
method nonrec_flag f rf =
|
|
match rf with
|
|
| Nonrecursive -> pp f "nonrec "
|
|
| Recursive -> ()
|
|
method direction_flag f = function
|
|
| Upto -> pp f "to@ "
|
|
| Downto -> pp f "downto@ "
|
|
method private_flag f = function
|
|
| Public -> ()
|
|
| Private -> pp f "private@ "
|
|
|
|
method constant_string f s = pp f "%S" s
|
|
method tyvar f str = pp f "'%s" str
|
|
method string_quot f x = pp f "`%s" x
|
|
|
|
(* c ['a,'b] *)
|
|
method class_params_def f = function
|
|
| [] -> ()
|
|
| l ->
|
|
pp f "[%a] " (* space *)
|
|
(self#list self#type_param ~sep:",") l
|
|
|
|
method type_with_label f (label,({ptyp_desc;_}as c) ) =
|
|
match label with
|
|
| Nolabel -> self#core_type1 f c (* otherwise parenthesize *)
|
|
| Labelled s -> pp f "%s:%a" s self#core_type1 c
|
|
| Optional s -> pp f "?%s:%a" s self#core_type1 c
|
|
method core_type f x =
|
|
if x.ptyp_attributes <> [] then begin
|
|
pp f "((%a)%a)" self#core_type {x with ptyp_attributes=[]}
|
|
self#attributes x.ptyp_attributes
|
|
end
|
|
else match x.ptyp_desc with
|
|
| Ptyp_arrow (l, ct1, ct2) ->
|
|
pp f "@[<2>%a@;->@;%a@]" (* FIXME remove parens later *)
|
|
self#type_with_label (l,ct1) self#core_type ct2
|
|
| Ptyp_alias (ct, s) ->
|
|
pp f "@[<2>%a@;as@;'%s@]" self#core_type1 ct s
|
|
| Ptyp_poly (sl, ct) ->
|
|
pp f "@[<2>%a%a@]"
|
|
(fun f l ->
|
|
pp f "%a"
|
|
(fun f l -> match l with
|
|
| [] -> ()
|
|
| _ ->
|
|
pp f "%a@;.@;"
|
|
(self#list self#tyvar ~sep:"@;") l)
|
|
l)
|
|
sl self#core_type ct
|
|
| _ -> pp f "@[<2>%a@]" self#core_type1 x
|
|
method core_type1 f x =
|
|
if x.ptyp_attributes <> [] then self#core_type f x
|
|
else match x.ptyp_desc with
|
|
| Ptyp_any -> pp f "_";
|
|
| Ptyp_var s -> self#tyvar f s;
|
|
| Ptyp_tuple l -> pp f "(%a)" (self#list self#core_type1 ~sep:"@;*@;") l
|
|
| Ptyp_constr (li, l) ->
|
|
pp f (* "%a%a@;" *) "%a%a"
|
|
(fun f l -> match l with
|
|
|[] -> ()
|
|
|[x]-> pp f "%a@;" self#core_type1 x
|
|
| _ -> self#list ~first:"(" ~last:")@;" self#core_type ~sep:"," f l)
|
|
l self#longident_loc li
|
|
| Ptyp_variant (l, closed, low) ->
|
|
let type_variant_helper f x =
|
|
match x with
|
|
| Rtag (l, attrs, _, ctl) -> pp f "@[<2>%a%a@;%a@]" self#string_quot l
|
|
(fun f l -> match l with
|
|
|[] -> ()
|
|
| _ -> pp f "@;of@;%a"
|
|
(self#list self#core_type ~sep:"&") ctl) ctl
|
|
self#attributes attrs
|
|
| Rinherit ct -> self#core_type f ct in
|
|
pp f "@[<2>[%a%a]@]"
|
|
(fun f l
|
|
->
|
|
match l with
|
|
| [] -> ()
|
|
| _ ->
|
|
pp f "%s@;%a"
|
|
(match (closed,low) with
|
|
| (Closed,None) -> ""
|
|
| (Closed,Some _) -> "<" (* FIXME desugar the syntax sugar*)
|
|
| (Open,_) -> ">")
|
|
(self#list type_variant_helper ~sep:"@;<1 -2>| ") l) l
|
|
(fun f low
|
|
->
|
|
match low with
|
|
|Some [] |None -> ()
|
|
|Some xs ->
|
|
pp f ">@ %a"
|
|
(self#list self#string_quot) xs) low
|
|
| Ptyp_object (l, o) ->
|
|
let core_field_type f (s, attrs, ct) =
|
|
pp f "@[<hov2>%s%a@ :%a@ @]" s
|
|
self#attributes attrs self#core_type ct
|
|
in
|
|
let field_var f = function
|
|
| Asttypes.Closed -> ()
|
|
| Asttypes.Open ->
|
|
match l with
|
|
| [] -> pp f ".."
|
|
| _ -> pp f " ;.."
|
|
in
|
|
pp f "@[<hov2><@ %a%a@ >@]" (self#list core_field_type ~sep:";") l
|
|
field_var o
|
|
| Ptyp_class (li, l) -> (*FIXME*)
|
|
pp f "@[<hov2>%a#%a@]"
|
|
(self#list self#core_type ~sep:"," ~first:"(" ~last:")") l
|
|
self#longident_loc li
|
|
| Ptyp_package (lid, cstrs) ->
|
|
let aux f (s, ct) =
|
|
pp f "type %a@ =@ %a" self#longident_loc s self#core_type ct in
|
|
(match cstrs with
|
|
|[] -> pp f "@[<hov2>(module@ %a)@]" self#longident_loc lid
|
|
|_ ->
|
|
pp f "@[<hov2>(module@ %a@ with@ %a)@]" self#longident_loc lid
|
|
(self#list aux ~sep:"@ and@ ") cstrs)
|
|
| Ptyp_extension e -> self#extension f e
|
|
| _ -> self#paren true self#core_type f x
|
|
(********************pattern********************)
|
|
(* be cautious when use [pattern], [pattern1] is preferred *)
|
|
method pattern f x =
|
|
let rec list_of_pattern acc = function (* only consider ((A|B)|C)*)
|
|
| {ppat_desc= Ppat_or (p1,p2);_} ->
|
|
list_of_pattern (p2::acc) p1
|
|
| x -> x::acc in
|
|
if x.ppat_attributes <> [] then begin
|
|
pp f "((%a)%a)" self#pattern {x with ppat_attributes=[]}
|
|
self#attributes x.ppat_attributes
|
|
end
|
|
else match x.ppat_desc with
|
|
| Ppat_alias (p, s) -> pp f "@[<2>%a@;as@;%a@]"
|
|
self#pattern p protect_ident s.txt (* RA*)
|
|
| Ppat_or (p1, p2) -> (* *)
|
|
pp f "@[<hov0>%a@]" (self#list ~sep:"@,|" self#pattern)
|
|
(list_of_pattern [] x)
|
|
| _ -> self#pattern1 f x
|
|
method pattern1 (f:Format.formatter) (x:pattern) :unit =
|
|
let rec pattern_list_helper f = function
|
|
| {ppat_desc =
|
|
Ppat_construct
|
|
({ txt = Lident("::") ;_},
|
|
Some ({ppat_desc = Ppat_tuple([pat1; pat2]);_})); _}
|
|
-> pp f "%a::%a" self#simple_pattern pat1 pattern_list_helper pat2
|
|
(*RA*)
|
|
| p -> self#pattern1 f p in
|
|
if x.ppat_attributes <> [] then self#pattern f x
|
|
else match x.ppat_desc with
|
|
| Ppat_variant (l, Some p) -> pp f "@[<2>`%s@;%a@]" l self#simple_pattern p
|
|
| Ppat_construct (({txt=Lident("()"|"[]");_}), _) -> self#simple_pattern f x
|
|
| Ppat_construct (({txt;_} as li), po) ->
|
|
(* FIXME The third field always false *)
|
|
if txt = Lident "::" then
|
|
pp f "%a" pattern_list_helper x
|
|
else
|
|
(match po with
|
|
|Some x ->
|
|
pp f "%a@;%a" self#longident_loc li self#simple_pattern x
|
|
| None -> pp f "%a@;"self#longident_loc li )
|
|
| _ -> self#simple_pattern f x
|
|
method simple_pattern (f:Format.formatter) (x:pattern) :unit =
|
|
if x.ppat_attributes <> [] then self#pattern f x
|
|
else match x.ppat_desc with
|
|
| Ppat_construct (({txt=Lident ("()"|"[]" as x);_}), _) -> pp f "%s" x
|
|
| Ppat_any -> pp f "_";
|
|
| Ppat_var ({txt = txt;_}) -> protect_ident f txt
|
|
| Ppat_array l ->
|
|
pp f "@[<2>[|%a|]@]" (self#list self#pattern1 ~sep:";") l
|
|
| Ppat_unpack (s) ->
|
|
pp f "(module@ %s)@ " s.txt
|
|
| Ppat_type li ->
|
|
pp f "#%a" self#longident_loc li
|
|
| Ppat_record (l, closed) ->
|
|
let longident_x_pattern f (li, p) =
|
|
match (li,p.ppat_desc) with
|
|
| ({txt=Lident s;_ },Ppat_var {txt;_} ) when s = txt ->
|
|
pp f "@[<2>%a@]" self#longident_loc li
|
|
| _ ->
|
|
pp f "@[<2>%a@;=@;%a@]" self#longident_loc li self#pattern1 p in
|
|
(match closed with
|
|
|Closed ->
|
|
pp f "@[<2>{@;%a@;}@]"
|
|
(self#list longident_x_pattern ~sep:";@;") l
|
|
| _ ->
|
|
pp f "@[<2>{@;%a;_}@]"
|
|
(self#list longident_x_pattern ~sep:";@;") l)
|
|
| Ppat_tuple l -> pp f "@[<1>(%a)@]" (self#list ~sep:"," self#pattern1) l
|
|
(* level1*)
|
|
| Ppat_constant (c) -> pp f "%a" self#constant c
|
|
| Ppat_interval (c1, c2) -> pp f "%a..%a" self#constant c1 self#constant c2
|
|
| Ppat_variant (l,None) -> pp f "`%s" l
|
|
| Ppat_constraint (p, ct) ->
|
|
pp f "@[<2>(%a@;:@;%a)@]" self#pattern1 p self#core_type ct
|
|
| Ppat_lazy p ->
|
|
pp f "@[<2>(lazy@;%a)@]" self#pattern1 p
|
|
| Ppat_exception p ->
|
|
pp f "@[<2>exception@;%a@]" self#pattern1 p
|
|
| Ppat_extension e -> self#extension f e
|
|
| _ -> self#paren true self#pattern f x
|
|
|
|
method label_exp f (l,opt,p) =
|
|
match l with
|
|
| Nolabel ->
|
|
pp f "%a@ " self#simple_pattern p
|
|
(*single case pattern parens needed here *)
|
|
| Optional rest ->
|
|
begin match p.ppat_desc with
|
|
| Ppat_var {txt;_} when txt = rest ->
|
|
(match opt with
|
|
| Some o -> pp f "?(%s=@;%a)@;" rest self#expression o
|
|
| None -> pp f "?%s@ " rest)
|
|
| _ ->
|
|
(match opt with
|
|
| Some o ->
|
|
pp f "?%s:(%a=@;%a)@;" rest self#pattern1 p self#expression o
|
|
| None -> pp f "?%s:%a@;" rest self#simple_pattern p)
|
|
end
|
|
| Labelled l ->
|
|
(match p.ppat_desc with
|
|
| Ppat_var {txt;_} when txt = l ->
|
|
pp f "~%s@;" l
|
|
| _ -> pp f "~%s:%a@;" l self#simple_pattern p )
|
|
method sugar_expr f e =
|
|
if e.pexp_attributes <> [] then false
|
|
else match e.pexp_desc with
|
|
| Pexp_apply ({ pexp_desc = Pexp_ident { txt = id; _ };
|
|
pexp_attributes=[]; _ }, args)
|
|
when List.for_all (fun (lab, _) -> lab = Nolabel) args -> begin
|
|
match id, List.map snd args with
|
|
| Lident "!", [e] ->
|
|
pp f "@[<hov>!%a@]" self#simple_expr e;
|
|
true
|
|
| Ldot (path, ("get"|"set" as func)), a :: other_args -> begin
|
|
let print left right print_index indexes rem_args =
|
|
match func, rem_args with
|
|
| "get", [] ->
|
|
pp f "@[%a.%s%a%s@]"
|
|
self#simple_expr a
|
|
left (self#list ~sep:"," print_index) indexes right;
|
|
true
|
|
| "set", [v] ->
|
|
pp f "@[%a.%s%a%s@ <-@;<1 2>%a@]"
|
|
self#simple_expr a
|
|
left (self#list ~sep:"," print_index) indexes right
|
|
self#simple_expr v;
|
|
true
|
|
| _ -> false
|
|
in
|
|
match path, other_args with
|
|
| Lident "Array", i :: rest ->
|
|
print "(" ")" self#expression [i] rest
|
|
| Lident "String", i :: rest ->
|
|
print "[" "]" self#expression [i] rest
|
|
| Ldot (Lident "Bigarray", "Array1"), i1 :: rest ->
|
|
print "{" "}" self#simple_expr [i1] rest
|
|
| Ldot (Lident "Bigarray", "Array2"), i1 :: i2 :: rest ->
|
|
print "{" "}" self#simple_expr [i1; i2] rest
|
|
| Ldot (Lident "Bigarray", "Array3"), i1 :: i2 :: i3 :: rest ->
|
|
print "{" "}" self#simple_expr [i1; i2; i3] rest
|
|
| Ldot (Lident "Bigarray", "Genarray"),
|
|
{pexp_desc = Pexp_array indexes; pexp_attributes = []} :: rest ->
|
|
print "{" "}" self#simple_expr indexes rest
|
|
| _ -> false
|
|
end
|
|
| _ -> false
|
|
end
|
|
| _ -> false
|
|
method expression f x =
|
|
if x.pexp_attributes <> [] then begin
|
|
pp f "((%a)@,%a)" self#expression {x with pexp_attributes=[]}
|
|
self#attributes x.pexp_attributes
|
|
end
|
|
else match x.pexp_desc with
|
|
| Pexp_function _ | Pexp_fun _ | Pexp_match _ | Pexp_try _ | Pexp_sequence _
|
|
when pipe || semi ->
|
|
self#paren true self#reset#expression f x
|
|
| Pexp_ifthenelse _ | Pexp_sequence _ when ifthenelse ->
|
|
self#paren true self#reset#expression f x
|
|
| Pexp_let _ | Pexp_letmodule _ when semi ->
|
|
self#paren true self#reset#expression f x
|
|
| Pexp_fun (l, e0, p, e) ->
|
|
pp f "@[<2>fun@;%a@;->@;%a@]"
|
|
self#label_exp (l, e0, p)
|
|
self#expression e
|
|
| Pexp_function l ->
|
|
pp f "@[<hv>function%a@]" self#case_list l
|
|
| Pexp_match (e, l) ->
|
|
pp f "@[<hv0>@[<hv0>@[<2>match %a@]@ with@]%a@]" self#reset#expression
|
|
e self#case_list l
|
|
|
|
| Pexp_try (e, l) ->
|
|
pp f "@[<0>@[<hv2>try@ %a@]@ @[<0>with%a@]@]"
|
|
(* "try@;@[<2>%a@]@\nwith@\n%a"*)
|
|
self#reset#expression e self#case_list l
|
|
| Pexp_let (rf, l, e) ->
|
|
(* pp f "@[<2>let %a%a in@;<1 -2>%a@]"
|
|
(*no identation here, a new line*) *)
|
|
(* self#rec_flag rf *)
|
|
pp f "@[<2>%a in@;<1 -2>%a@]"
|
|
self#reset#bindings (rf,l)
|
|
self#expression e
|
|
| Pexp_apply (e, l) ->
|
|
(if not (self#sugar_expr f x) then
|
|
match view_fixity_of_exp e with
|
|
| `Infix s ->
|
|
(match l with
|
|
| [ (Nolabel, _) as arg1; (Nolabel, _) as arg2 ] ->
|
|
pp f "@[<2>%a@;%s@;%a@]"
|
|
(* FIXME associativity lable_x_expression_parm*)
|
|
self#reset#label_x_expression_param arg1 s
|
|
self#label_x_expression_param arg2
|
|
| _ ->
|
|
pp f "@[<2>%a %a@]" self#simple_expr e
|
|
(self#list self#label_x_expression_param) l)
|
|
| `Prefix s ->
|
|
let s =
|
|
if List.mem s ["~+";"~-";"~+.";"~-."]
|
|
then String.sub s 1 (String.length s -1)
|
|
else s
|
|
in
|
|
(match l with
|
|
| [(Nolabel, _) as v] ->
|
|
pp f "@[<2>%s@;%a@]" s self#label_x_expression_param v
|
|
| _ ->
|
|
pp f "@[<2>%a %a@]" self#simple_expr e
|
|
(self#list self#label_x_expression_param) l
|
|
)
|
|
| _ ->
|
|
pp f "@[<hov2>%a@]" begin fun f (e,l) ->
|
|
pp f "%a@ %a" self#expression2 e
|
|
(self#list self#reset#label_x_expression_param) l
|
|
(*reset here only because [function,match,try,sequence] are
|
|
lower priority*)
|
|
end (e,l))
|
|
|
|
| Pexp_construct (li, Some eo)
|
|
when not (is_simple_construct (view_expr x))-> (* Not efficient FIXME*)
|
|
(match view_expr x with
|
|
| `cons ls -> self#list self#simple_expr f ls ~sep:"@;::@;"
|
|
| `normal ->
|
|
pp f "@[<2>%a@;%a@]" self#longident_loc li
|
|
self#simple_expr eo
|
|
| _ -> assert false)
|
|
| Pexp_setfield (e1, li, e2) ->
|
|
pp f "@[<2>%a.%a@ <-@ %a@]" self#simple_expr e1 self#longident_loc li
|
|
self#expression e2;
|
|
| Pexp_ifthenelse (e1, e2, eo) ->
|
|
(* @;@[<2>else@ %a@]@] *)
|
|
let fmt:(_,_,_)format ="@[<hv0>@[<2>if@ %a@]@;@[<2>then@ %a@]%a@]" in
|
|
pp f fmt self#under_ifthenelse#expression e1
|
|
self#under_ifthenelse#expression e2
|
|
(fun f eo -> match eo with
|
|
| Some x -> pp f "@;@[<2>else@;%a@]" self#under_semi#expression x
|
|
| None -> () (* pp f "()" *)) eo
|
|
| Pexp_sequence _ ->
|
|
let rec sequence_helper acc = function
|
|
| {pexp_desc=Pexp_sequence(e1,e2);_} ->
|
|
sequence_helper (e1::acc) e2
|
|
| v -> List.rev (v::acc) in
|
|
let lst = sequence_helper [] x in
|
|
pp f "@[<hv>%a@]"
|
|
(self#list self#under_semi#expression ~sep:";@;") lst
|
|
| Pexp_new (li) ->
|
|
pp f "@[<hov2>new@ %a@]" self#longident_loc li;
|
|
| Pexp_setinstvar (s, e) ->
|
|
pp f "@[<hov2>%s@ <-@ %a@]" s.txt self#expression e
|
|
| Pexp_override l -> (* FIXME *)
|
|
let string_x_expression f (s, e) =
|
|
pp f "@[<hov2>%s@ =@ %a@]" s.txt self#expression e in
|
|
pp f "@[<hov2>{<%a>}@]"
|
|
(self#list string_x_expression ~sep:";" ) l;
|
|
| Pexp_letmodule (s, me, e) ->
|
|
pp f "@[<hov2>let@ module@ %s@ =@ %a@ in@ %a@]" s.txt
|
|
self#reset#module_expr me self#expression e
|
|
| Pexp_assert e ->
|
|
pp f "@[<hov2>assert@ %a@]" self#simple_expr e
|
|
| Pexp_lazy (e) ->
|
|
pp f "@[<hov2>lazy@ %a@]" self#simple_expr e
|
|
(* Pexp_poly: impossible but we should print it anyway, rather
|
|
than assert false *)
|
|
| Pexp_poly (e, None) ->
|
|
pp f "@[<hov2>!poly!@ %a@]" self#simple_expr e
|
|
| Pexp_poly (e, Some ct) ->
|
|
pp f "@[<hov2>(!poly!@ %a@ : %a)@]" self#simple_expr e self#core_type ct
|
|
| Pexp_open (ovf, lid, e) ->
|
|
pp f "@[<2>let open%s %a in@;%a@]" (override ovf) self#longident_loc lid
|
|
self#expression e
|
|
| Pexp_variant (l,Some eo) ->
|
|
pp f "@[<2>`%s@;%a@]" l self#simple_expr eo
|
|
| Pexp_extension e -> self#extension f e
|
|
| Pexp_unreachable ->
|
|
pp f "."
|
|
| _ -> self#expression1 f x
|
|
method expression1 f x =
|
|
if x.pexp_attributes <> [] then self#expression f x
|
|
else match x.pexp_desc with
|
|
| Pexp_object cs -> pp f "%a" self#class_structure cs
|
|
| _ -> self#expression2 f x
|
|
(* used in [Pexp_apply] *)
|
|
method expression2 f x =
|
|
if x.pexp_attributes <> [] then self#expression f x
|
|
else match x.pexp_desc with
|
|
| Pexp_field (e, li) ->
|
|
pp f "@[<hov2>%a.%a@]" self#simple_expr e self#longident_loc li
|
|
| Pexp_send (e, s) -> pp f "@[<hov2>%a#%s@]" self#simple_expr e s
|
|
|
|
| _ -> self#simple_expr f x
|
|
method simple_expr f x =
|
|
if x.pexp_attributes <> [] then self#expression f x
|
|
else match x.pexp_desc with
|
|
| Pexp_construct _ when is_simple_construct (view_expr x) ->
|
|
(match view_expr x with
|
|
| `nil -> pp f "[]"
|
|
| `tuple -> pp f "()"
|
|
| `list xs ->
|
|
pp f "@[<hv0>[%a]@]"
|
|
(self#list self#under_semi#expression ~sep:";@;") xs
|
|
| `simple x -> self#longident f x
|
|
| _ -> assert false)
|
|
| Pexp_ident li ->
|
|
self#longident_loc f li
|
|
(* (match view_fixity_of_exp x with *)
|
|
(* |`Normal -> self#longident_loc f li *)
|
|
(* | `Prefix _ | `Infix _ -> pp f "( %a )" self#longident_loc li) *)
|
|
| Pexp_constant c -> self#constant f c;
|
|
| Pexp_pack me ->
|
|
pp f "(module@;%a)" self#module_expr me
|
|
| Pexp_newtype (lid, e) ->
|
|
pp f "fun@;(type@;%s)@;->@;%a" lid self#expression e
|
|
| Pexp_tuple l ->
|
|
pp f "@[<hov2>(%a)@]" (self#list self#simple_expr ~sep:",@;") l
|
|
| Pexp_constraint (e, ct) ->
|
|
pp f "(%a : %a)" self#expression e self#core_type ct
|
|
| Pexp_coerce (e, cto1, ct) ->
|
|
pp f "(%a%a :> %a)" self#expression e
|
|
(self#option self#core_type ~first:" : " ~last:" ")
|
|
cto1 (* no sep hint*)
|
|
self#core_type ct
|
|
| Pexp_variant (l, None) -> pp f "`%s" l
|
|
| Pexp_record (l, eo) ->
|
|
let longident_x_expression f ( li, e) =
|
|
match e.pexp_desc with
|
|
| Pexp_ident {txt;_} when li.txt = txt ->
|
|
pp f "@[<hov2>%a@]" self#longident_loc li
|
|
| _ ->
|
|
pp f "@[<hov2>%a@;=@;%a@]" self#longident_loc li self#simple_expr
|
|
e
|
|
in
|
|
pp f "@[<hv0>@[<hv2>{@;%a%a@]@;}@]"(* "@[<hov2>{%a%a}@]" *)
|
|
(self#option ~last:" with@;" self#simple_expr) eo
|
|
(self#list longident_x_expression ~sep:";@;") l
|
|
| Pexp_array (l) ->
|
|
pp f "@[<0>@[<2>[|%a|]@]@]"
|
|
(self#list self#under_semi#simple_expr ~sep:";") l
|
|
| Pexp_while (e1, e2) ->
|
|
let fmt:(_,_,_)format = "@[<2>while@;%a@;do@;%a@;done@]" in
|
|
pp f fmt self#expression e1 self#expression e2
|
|
| Pexp_for (s, e1, e2, df, e3) ->
|
|
let fmt:(_,_,_)format =
|
|
"@[<hv0>@[<hv2>@[<2>for %a =@;%a@;%a%a@;do@]@;%a@]@;done@]" in
|
|
pp f fmt self#pattern s self#expression e1 self#direction_flag df
|
|
self#expression e2 self#expression e3
|
|
| _ -> self#paren true self#expression f x
|
|
|
|
method attributes f l =
|
|
List.iter (self # attribute f) l
|
|
|
|
method item_attributes f l =
|
|
List.iter (self # item_attribute f) l
|
|
|
|
method attribute f (s, e) =
|
|
pp f "@[<2>[@@%s@ %a]@]" s.txt self#payload e
|
|
|
|
method item_attribute f (s, e) =
|
|
pp f "@[<2>[@@@@%s@ %a]@]" s.txt self#payload e
|
|
|
|
method floating_attribute f (s, e) =
|
|
pp f "@[<2>[@@@@@@%s@ %a]@]" s.txt self#payload e
|
|
|
|
method value_description f x =
|
|
(* note: value_description has an attribute field,
|
|
but they're already printed by the callers this method *)
|
|
pp f "@[<hov2>%a%a@]" self#core_type x.pval_type
|
|
(fun f x ->
|
|
if x.pval_prim<>[] then begin
|
|
pp f "@ =@ %a"
|
|
(self#list self#constant_string)
|
|
x.pval_prim ;
|
|
end) x
|
|
|
|
method extension f (s, e) =
|
|
pp f "@[<2>[%%%s@ %a]@]" s.txt self#payload e
|
|
|
|
method item_extension f (s, e) =
|
|
pp f "@[<2>[%%%%%s@ %a]@]" s.txt self#payload e
|
|
|
|
method exception_declaration f ext =
|
|
pp f "@[<hov2>exception@ %a@]" self#extension_constructor ext
|
|
|
|
method class_signature f { pcsig_self = ct; pcsig_fields = l ;_} =
|
|
let class_type_field f x =
|
|
match x.pctf_desc with
|
|
| Pctf_inherit (ct) ->
|
|
pp f "@[<2>inherit@ %a@]%a" self#class_type ct
|
|
self#item_attributes x.pctf_attributes
|
|
| Pctf_val (s, mf, vf, ct) ->
|
|
pp f "@[<2>val @ %a%a%s@ :@ %a@]%a"
|
|
self#mutable_flag mf self#virtual_flag vf s self#core_type ct
|
|
self#item_attributes x.pctf_attributes
|
|
| Pctf_method (s, pf, vf, ct) ->
|
|
pp f "@[<2>method %a %a%s :@;%a@]%a"
|
|
self#private_flag pf self#virtual_flag vf s self#core_type ct
|
|
self#item_attributes x.pctf_attributes
|
|
| Pctf_constraint (ct1, ct2) ->
|
|
pp f "@[<2>constraint@ %a@ =@ %a@]%a"
|
|
self#core_type ct1 self#core_type ct2
|
|
self#item_attributes x.pctf_attributes
|
|
| Pctf_attribute a -> self#floating_attribute f a
|
|
| Pctf_extension e ->
|
|
self#item_extension f e;
|
|
self#item_attributes f x.pctf_attributes
|
|
in
|
|
pp f "@[<hv0>@[<hv2>object@[<1>%a@]@ %a@]@ end@]"
|
|
(fun f ct -> match ct.ptyp_desc with
|
|
| Ptyp_any -> ()
|
|
| _ -> pp f " (%a)" self#core_type ct) ct
|
|
(self#list class_type_field ~sep:"@;") l ;
|
|
|
|
(* call [class_signature] called by [class_signature] *)
|
|
method class_type f x =
|
|
match x.pcty_desc with
|
|
| Pcty_signature cs ->
|
|
self#class_signature f cs;
|
|
self#attributes f x.pcty_attributes
|
|
| Pcty_constr (li, l) ->
|
|
pp f "%a%a%a"
|
|
(fun f l -> match l with
|
|
| [] -> ()
|
|
| _ -> pp f "[%a]@ " (self#list self#core_type ~sep:"," ) l) l
|
|
self#longident_loc li
|
|
self#attributes x.pcty_attributes
|
|
| Pcty_arrow (l, co, cl) ->
|
|
pp f "@[<2>%a@;->@;%a@]" (* FIXME remove parens later *)
|
|
self#type_with_label (l,co)
|
|
self#class_type cl
|
|
| Pcty_extension e ->
|
|
self#extension f e;
|
|
self#attributes f x.pcty_attributes
|
|
|
|
(* [class type a = object end] *)
|
|
method class_type_declaration_list f l =
|
|
let class_type_declaration kwd f ({pci_params=ls;pci_name={txt;_};_} as x) =
|
|
pp f "@[<2>%s %a%a%s@ =@ %a@]%a" kwd
|
|
self#virtual_flag x.pci_virt
|
|
self#class_params_def ls txt
|
|
self#class_type x.pci_expr
|
|
self#item_attributes x.pci_attributes
|
|
in
|
|
match l with
|
|
| [] -> ()
|
|
| [x] -> class_type_declaration "class type" f x
|
|
| x :: xs ->
|
|
pp f "@[<v>%a@,%a@]"
|
|
(class_type_declaration "class type") x
|
|
(self#list ~sep:"@," (class_type_declaration "and")) xs
|
|
|
|
method class_field f x =
|
|
match x.pcf_desc with
|
|
| Pcf_inherit (ovf, ce, so) ->
|
|
pp f "@[<2>inherit@ %s@ %a%a@]%a" (override ovf)
|
|
self#class_expr ce
|
|
(fun f so -> match so with
|
|
| None -> ();
|
|
| Some (s) -> pp f "@ as %s" s ) so
|
|
self#item_attributes x.pcf_attributes
|
|
| Pcf_val (s, mf, Cfk_concrete (ovf, e)) ->
|
|
pp f "@[<2>val%s %a%s =@;%a@]%a" (override ovf)
|
|
self#mutable_flag mf s.txt
|
|
self#expression e
|
|
self#item_attributes x.pcf_attributes
|
|
| Pcf_method (s, pf, Cfk_virtual ct) ->
|
|
pp f "@[<2>method virtual %a %s :@;%a@]%a"
|
|
self#private_flag pf s.txt
|
|
self#core_type ct
|
|
self#item_attributes x.pcf_attributes
|
|
| Pcf_val (s, mf, Cfk_virtual ct) ->
|
|
pp f "@[<2>val virtual %a%s :@ %a@]%a"
|
|
self#mutable_flag mf s.txt
|
|
self#core_type ct
|
|
self#item_attributes x.pcf_attributes
|
|
| Pcf_method (s, pf, Cfk_concrete (ovf, e)) ->
|
|
let bind e =
|
|
self#binding f
|
|
{pvb_pat=
|
|
{ppat_desc=Ppat_var s;ppat_loc=Location.none;ppat_attributes=[]};
|
|
pvb_expr=e;
|
|
pvb_attributes=[];
|
|
pvb_loc=Location.none;
|
|
}
|
|
in
|
|
pp f "@[<2>method%s %a%a@]%a"
|
|
(override ovf)
|
|
self#private_flag pf
|
|
(fun f e -> match e.pexp_desc with
|
|
| Pexp_poly (e, Some ct) ->
|
|
pp f "%s :@;%a=@;%a"
|
|
s.txt (self#core_type) ct self#expression e
|
|
| Pexp_poly (e,None) -> bind e
|
|
| _ -> bind e) e
|
|
self#item_attributes x.pcf_attributes
|
|
| Pcf_constraint (ct1, ct2) ->
|
|
pp f "@[<2>constraint %a =@;%a@]%a"
|
|
self#core_type ct1
|
|
self#core_type ct2
|
|
self#item_attributes x.pcf_attributes
|
|
| Pcf_initializer (e) ->
|
|
pp f "@[<2>initializer@ %a@]%a"
|
|
self#expression e
|
|
self#item_attributes x.pcf_attributes
|
|
| Pcf_attribute a -> self#floating_attribute f a
|
|
| Pcf_extension e ->
|
|
self#item_extension f e;
|
|
self#item_attributes f x.pcf_attributes
|
|
|
|
method class_structure f { pcstr_self = p; pcstr_fields = l } =
|
|
pp f "@[<hv0>@[<hv2>object%a@;%a@]@;end@]"
|
|
(fun f p -> match p.ppat_desc with
|
|
| Ppat_any -> ()
|
|
| Ppat_constraint _ -> pp f " %a" self#pattern p
|
|
| _ -> pp f " (%a)" self#pattern p) p
|
|
(self#list self#class_field ) l
|
|
|
|
method class_expr f x =
|
|
if x.pcl_attributes <> [] then begin
|
|
pp f "((%a)%a)" self#class_expr {x with pcl_attributes=[]}
|
|
self#attributes x.pcl_attributes
|
|
end else
|
|
match x.pcl_desc with
|
|
| Pcl_structure (cs) -> self#class_structure f cs
|
|
| Pcl_fun (l, eo, p, e) ->
|
|
pp f "fun@ %a@ ->@ %a"
|
|
self#label_exp (l,eo,p)
|
|
self#class_expr e
|
|
| Pcl_let (rf, l, ce) ->
|
|
pp f "%a@ in@ %a"
|
|
self#bindings (rf,l)
|
|
self#class_expr ce
|
|
| Pcl_apply (ce, l) ->
|
|
pp f "(%a@ %a)"
|
|
self#class_expr ce
|
|
(self#list self#label_x_expression_param) l
|
|
| Pcl_constr (li, l) ->
|
|
pp f "%a%a"
|
|
(fun f l-> if l <>[] then
|
|
pp f "[%a]@ "
|
|
(self#list self#core_type ~sep:"," ) l ) l
|
|
self#longident_loc li
|
|
| Pcl_constraint (ce, ct) ->
|
|
pp f "(%a@ :@ %a)"
|
|
self#class_expr ce
|
|
self#class_type ct
|
|
| Pcl_extension e -> self#extension f e
|
|
|
|
method module_type f x =
|
|
if x.pmty_attributes <> [] then begin
|
|
pp f "((%a)%a)" self#module_type {x with pmty_attributes=[]}
|
|
self#attributes x.pmty_attributes
|
|
end else
|
|
match x.pmty_desc with
|
|
| Pmty_ident li ->
|
|
pp f "%a" self#longident_loc li;
|
|
| Pmty_alias li ->
|
|
pp f "(module %a)" self#longident_loc li;
|
|
| Pmty_signature (s) ->
|
|
pp f "@[<hv0>@[<hv2>sig@ %a@]@ end@]" (* "@[<hov>sig@ %a@ end@]" *)
|
|
(self#list self#signature_item ) s (* FIXME wrong indentation*)
|
|
| Pmty_functor (_, None, mt2) ->
|
|
pp f "@[<hov2>functor () ->@ %a@]" self#module_type mt2
|
|
| Pmty_functor (s, Some mt1, mt2) ->
|
|
if s.txt = "_" then
|
|
pp f "@[<hov2>%a@ ->@ %a@]"
|
|
self#module_type mt1 self#module_type mt2
|
|
else
|
|
pp f "@[<hov2>functor@ (%s@ :@ %a)@ ->@ %a@]" s.txt
|
|
self#module_type mt1 self#module_type mt2
|
|
| Pmty_with (mt, l) ->
|
|
let with_constraint f = function
|
|
| Pwith_type (li, ({ptype_params= ls ;_} as td)) ->
|
|
let ls = List.map fst ls in
|
|
pp f "type@ %a %a =@ %a"
|
|
(self#list self#core_type ~sep:"," ~first:"(" ~last:")")
|
|
ls self#longident_loc li self#type_declaration td
|
|
| Pwith_module (li, li2) ->
|
|
pp f "module %a =@ %a" self#longident_loc li self#longident_loc
|
|
li2;
|
|
| Pwith_typesubst ({ptype_params=ls;_} as td) ->
|
|
let ls = List.map fst ls in
|
|
pp f "type@ %a %s :=@ %a"
|
|
(self#list self#core_type ~sep:"," ~first:"(" ~last:")")
|
|
ls td.ptype_name.txt
|
|
self#type_declaration td
|
|
| Pwith_modsubst (s, li2) ->
|
|
pp f "module %s :=@ %a" s.txt self#longident_loc li2 in
|
|
(match l with
|
|
| [] -> pp f "@[<hov2>%a@]" self#module_type mt
|
|
| _ -> pp f "@[<hov2>(%a@ with@ %a)@]"
|
|
self#module_type mt (self#list with_constraint ~sep:"@ and@ ") l )
|
|
| Pmty_typeof me ->
|
|
pp f "@[<hov2>module@ type@ of@ %a@]"
|
|
self#module_expr me
|
|
| Pmty_extension e -> self#extension f e
|
|
|
|
method signature f x = self#list ~sep:"@\n" self#signature_item f x
|
|
|
|
method signature_item f x :unit= begin
|
|
match x.psig_desc with
|
|
| Psig_type (rf, l) ->
|
|
self#type_def_list f (rf, l)
|
|
| Psig_value vd ->
|
|
let intro = if vd.pval_prim = [] then "val" else "external" in
|
|
pp f "@[<2>%s@ %a@ :@ %a@]%a" intro
|
|
protect_ident vd.pval_name.txt
|
|
self#value_description vd
|
|
self#item_attributes vd.pval_attributes
|
|
| Psig_typext te ->
|
|
self#type_extension f te
|
|
| Psig_exception ed ->
|
|
self#exception_declaration f ed
|
|
| Psig_class l ->
|
|
let class_description kwd f ({pci_params=ls;pci_name={txt;_};_} as x) =
|
|
pp f "@[<2>%s %a%a%s@;:@;%a@]%a" kwd
|
|
self#virtual_flag x.pci_virt
|
|
self#class_params_def ls txt
|
|
self#class_type x.pci_expr
|
|
self#item_attributes x.pci_attributes
|
|
in begin
|
|
match l with
|
|
| [] -> ()
|
|
| [x] -> class_description "class" f x
|
|
| x :: xs ->
|
|
pp f "@[<v>%a@,%a@]"
|
|
(class_description "class") x
|
|
(self#list ~sep:"@," (class_description "and")) xs
|
|
end
|
|
| Psig_module ({pmd_type={pmty_desc=Pmty_alias alias};_} as pmd) ->
|
|
pp f "@[<hov>module@ %s@ =@ %a@]%a" pmd.pmd_name.txt
|
|
self#longident_loc alias
|
|
self#item_attributes pmd.pmd_attributes
|
|
| Psig_module pmd ->
|
|
pp f "@[<hov>module@ %s@ :@ %a@]%a"
|
|
pmd.pmd_name.txt
|
|
self#module_type pmd.pmd_type
|
|
self#item_attributes pmd.pmd_attributes
|
|
| Psig_open od ->
|
|
pp f "@[<hov2>open%s@ %a@]%a"
|
|
(override od.popen_override)
|
|
self#longident_loc od.popen_lid
|
|
self#item_attributes od.popen_attributes
|
|
| Psig_include incl ->
|
|
pp f "@[<hov2>include@ %a@]%a"
|
|
self#module_type incl.pincl_mod
|
|
self#item_attributes incl.pincl_attributes
|
|
| Psig_modtype {pmtd_name=s; pmtd_type=md; pmtd_attributes=attrs} ->
|
|
pp f "@[<hov2>module@ type@ %s%a@]%a"
|
|
s.txt
|
|
(fun f md -> match md with
|
|
| None -> ()
|
|
| Some mt ->
|
|
pp_print_space f () ;
|
|
pp f "@ =@ %a" self#module_type mt
|
|
) md
|
|
self#item_attributes attrs
|
|
| Psig_class_type (l) ->
|
|
self#class_type_declaration_list f l ;
|
|
| Psig_recmodule decls ->
|
|
let rec string_x_module_type_list f ?(first=true) l =
|
|
match l with
|
|
| [] -> () ;
|
|
| pmd :: tl ->
|
|
if not first then
|
|
pp f "@ @[<hov2>and@ %s:@ %a@]%a" pmd.pmd_name.txt
|
|
self#module_type pmd.pmd_type
|
|
self#item_attributes pmd.pmd_attributes
|
|
else
|
|
pp f "@[<hov2>module@ rec@ %s:@ %a@]%a" pmd.pmd_name.txt
|
|
self#module_type pmd.pmd_type
|
|
self#item_attributes pmd.pmd_attributes;
|
|
string_x_module_type_list f ~first:false tl
|
|
in
|
|
string_x_module_type_list f decls
|
|
| Psig_attribute a -> self#floating_attribute f a
|
|
| Psig_extension(e, a) ->
|
|
self#item_extension f e;
|
|
self#item_attributes f a
|
|
end
|
|
method module_expr f x =
|
|
if x.pmod_attributes <> [] then begin
|
|
pp f "((%a)%a)" self#module_expr {x with pmod_attributes=[]}
|
|
self#attributes x.pmod_attributes
|
|
end else
|
|
match x.pmod_desc with
|
|
| Pmod_structure (s) ->
|
|
pp f "@[<hv2>struct@;@[<0>%a@]@;<1 -2>end@]"
|
|
(self#list self#structure_item ~sep:"@\n") s;
|
|
| Pmod_constraint (me, mt) ->
|
|
pp f "@[<hov2>(%a@ :@ %a)@]"
|
|
self#module_expr me
|
|
self#module_type mt
|
|
| Pmod_ident (li) ->
|
|
pp f "%a" self#longident_loc li;
|
|
| Pmod_functor (_, None, me) ->
|
|
pp f "functor ()@;->@;%a" self#module_expr me
|
|
| Pmod_functor (s, Some mt, me) ->
|
|
pp f "functor@ (%s@ :@ %a)@;->@;%a"
|
|
s.txt self#module_type mt self#module_expr me
|
|
| Pmod_apply (me1, me2) ->
|
|
pp f "%a(%a)" self#module_expr me1 self#module_expr me2
|
|
| Pmod_unpack e ->
|
|
pp f "(val@ %a)" self#expression e
|
|
| Pmod_extension e -> self#extension f e
|
|
|
|
method structure f x = self#list ~sep:"@\n" self#structure_item f x
|
|
|
|
method payload f = function
|
|
| PStr [{pstr_desc = Pstr_eval (e, attrs)}] ->
|
|
pp f "@[<2>%a@]%a"
|
|
self#expression e
|
|
self#item_attributes attrs
|
|
| PStr x -> self#structure f x
|
|
| PTyp x -> pp f ":"; self#core_type f x
|
|
| PSig x -> pp f ":"; self#signature f x
|
|
| PPat (x, None) -> pp f "?"; self#pattern f x
|
|
| PPat (x, Some e) ->
|
|
pp f "?"; self#pattern f x;
|
|
pp f " when "; self#expression f e
|
|
|
|
(* transform [f = fun g h -> ..] to [f g h = ... ] could be improved *)
|
|
method binding f {pvb_pat=p; pvb_expr=x; _} =
|
|
(* .pvb_attributes have already been printed by the caller, #bindings *)
|
|
let rec pp_print_pexp_function f x =
|
|
if x.pexp_attributes <> [] then pp f "=@;%a" self#expression x
|
|
else match x.pexp_desc with
|
|
| Pexp_fun (label, eo, p, e) ->
|
|
if label=Nolabel then
|
|
pp f "%a@ %a" self#simple_pattern p pp_print_pexp_function e
|
|
else
|
|
pp f "%a@ %a" self#label_exp (label,eo,p) pp_print_pexp_function e
|
|
| Pexp_newtype (str,e) ->
|
|
pp f "(type@ %s)@ %a" str pp_print_pexp_function e
|
|
| _ -> pp f "=@;%a" self#expression x in
|
|
if x.pexp_attributes <> [] then
|
|
pp f "%a@;=@;%a" self#pattern p self#expression x
|
|
else match (x.pexp_desc,p.ppat_desc) with
|
|
| ( _ , Ppat_constraint( p ,ty)) -> (* special case for the first*)
|
|
(match ty.ptyp_desc with
|
|
| Ptyp_poly _ ->
|
|
pp f "%a@;:@;%a=@;%a" self#simple_pattern p
|
|
self#core_type ty self#expression x
|
|
| _ ->
|
|
pp f "(%a@;:%a)=@;%a" self#simple_pattern p
|
|
self#core_type ty self#expression x)
|
|
| Pexp_constraint (e,t1),Ppat_var {txt;_} ->
|
|
pp f "%a@;:@ %a@;=@;%a" protect_ident txt self#core_type t1 self#expression e
|
|
| (_, Ppat_var _) ->
|
|
pp f "%a@ %a" self#simple_pattern p pp_print_pexp_function x
|
|
| _ ->
|
|
pp f "%a@;=@;%a" self#pattern p self#expression x
|
|
(* [in] is not printed *)
|
|
method bindings f (rf,l) =
|
|
let binding kwd rf f x =
|
|
pp f "@[<2>%s %a%a@]@ %a" kwd self#rec_flag rf
|
|
self#binding x self#item_attributes x.pvb_attributes
|
|
in
|
|
begin match l with
|
|
| [] -> ()
|
|
| [x] -> binding "let" rf f x
|
|
| x::xs ->
|
|
pp f "@[<v>%a@,%a@]"
|
|
(binding "let" rf) x
|
|
(self#list ~sep:"@," (binding "and" Nonrecursive)) xs
|
|
end
|
|
|
|
method structure_item f x = begin
|
|
match x.pstr_desc with
|
|
| Pstr_eval (e, attrs) ->
|
|
pp f "@[<hov2>;;%a@]%a"
|
|
self#expression e
|
|
self#item_attributes attrs
|
|
| Pstr_type (_, []) -> assert false
|
|
| Pstr_type (rf, l) -> self#type_def_list f (rf, l)
|
|
| Pstr_value (rf, l) ->
|
|
(* pp f "@[<hov2>let %a%a@]" self#rec_flag rf self#bindings l *)
|
|
pp f "@[<2>%a@]" self#bindings (rf,l)
|
|
| Pstr_typext te -> self#type_extension f te
|
|
| Pstr_exception ed -> self#exception_declaration f ed
|
|
| Pstr_module x ->
|
|
let rec module_helper me =
|
|
match me.pmod_desc with
|
|
| Pmod_functor(s,mt,me') when me.pmod_attributes = [] ->
|
|
if mt = None then pp f "()"
|
|
else Misc.may (pp f "(%s:%a)" s.txt self#module_type) mt;
|
|
module_helper me'
|
|
| _ -> me
|
|
in
|
|
pp f "@[<hov2>module %s%a@]%a"
|
|
x.pmb_name.txt
|
|
(fun f me ->
|
|
let me = module_helper me in
|
|
(match me.pmod_desc with
|
|
| Pmod_constraint
|
|
(me',
|
|
({pmty_desc=(Pmty_ident (_)
|
|
| Pmty_signature (_));_} as mt))
|
|
when me.pmod_attributes = [] ->
|
|
pp f " :@;%a@;=@;%a@;" self#module_type mt self#module_expr me'
|
|
| _ ->
|
|
pp f " =@ %a" self#module_expr me
|
|
)) x.pmb_expr
|
|
self#item_attributes x.pmb_attributes
|
|
| Pstr_open od ->
|
|
pp f "@[<2>open%s@;%a@]%a"
|
|
(override od.popen_override)
|
|
self#longident_loc od.popen_lid
|
|
self#item_attributes od.popen_attributes
|
|
| Pstr_modtype {pmtd_name=s; pmtd_type=md; pmtd_attributes=attrs} ->
|
|
pp f "@[<hov2>module@ type@ %s%a@]%a"
|
|
s.txt
|
|
(fun f md -> match md with
|
|
| None -> ()
|
|
| Some mt ->
|
|
pp_print_space f () ;
|
|
pp f "@ =@ %a" self#module_type mt
|
|
) md
|
|
self#item_attributes attrs
|
|
| Pstr_class l ->
|
|
let extract_class_args cl =
|
|
let rec loop acc cl =
|
|
match cl.pcl_desc with
|
|
| Pcl_fun (l, eo, p, cl') when cl.pcl_attributes = [] ->
|
|
loop ((l,eo,p) :: acc) cl'
|
|
| _ -> List.rev acc, cl
|
|
in
|
|
let args, cl = loop [] cl in
|
|
let constr, cl =
|
|
match cl.pcl_desc with
|
|
| Pcl_constraint (cl', ct) when cl.pcl_attributes = [] ->
|
|
Some ct, cl'
|
|
| _ -> None, cl
|
|
in
|
|
args, constr, cl
|
|
in
|
|
let class_constraint f ct = pp f ": @[%a@] " self#class_type ct in
|
|
let class_declaration kwd f
|
|
({pci_params=ls; pci_name={txt;_}; _} as x) =
|
|
let args, constr, cl = extract_class_args x.pci_expr in
|
|
pp f "@[<2>%s %a%a%s %a%a=@;%a@]%a" kwd
|
|
self#virtual_flag x.pci_virt
|
|
self#class_params_def ls txt
|
|
(self#list self#label_exp) args
|
|
(self#option class_constraint) constr
|
|
self#class_expr cl
|
|
self#item_attributes x.pci_attributes
|
|
in begin
|
|
match l with
|
|
| [] -> ()
|
|
| [x] -> class_declaration "class" f x
|
|
| x :: xs ->
|
|
pp f "@[<v>%a@,%a@]"
|
|
(class_declaration "class") x
|
|
(self#list ~sep:"@," (class_declaration "and")) xs
|
|
end
|
|
| Pstr_class_type (l) ->
|
|
self#class_type_declaration_list f l ;
|
|
| Pstr_primitive vd ->
|
|
pp f "@[<hov2>external@ %a@ :@ %a@]%a"
|
|
protect_ident vd.pval_name.txt
|
|
self#value_description vd
|
|
self#item_attributes vd.pval_attributes
|
|
| Pstr_include incl ->
|
|
pp f "@[<hov2>include@ %a@]%a"
|
|
self#module_expr incl.pincl_mod
|
|
self#item_attributes incl.pincl_attributes
|
|
| Pstr_recmodule decls -> (* 3.07 *)
|
|
let aux f = function
|
|
| ({pmb_expr={pmod_desc=Pmod_constraint (expr, typ)}} as pmb) ->
|
|
pp f "@[<hov2>@ and@ %s:%a@ =@ %a@]%a" pmb.pmb_name.txt
|
|
self#module_type typ
|
|
self#module_expr expr
|
|
self#item_attributes pmb.pmb_attributes
|
|
| _ -> assert false
|
|
in
|
|
begin match decls with
|
|
| ({pmb_expr={pmod_desc=Pmod_constraint (expr, typ)}} as pmb) :: l2 ->
|
|
pp f "@[<hv>@[<hov2>module@ rec@ %s:%a@ =@ %a@]%a@ %a@]"
|
|
pmb.pmb_name.txt
|
|
self#module_type typ
|
|
self#module_expr expr
|
|
self#item_attributes pmb.pmb_attributes
|
|
(fun f l2 -> List.iter (aux f) l2) l2
|
|
| _ -> assert false
|
|
end
|
|
| Pstr_attribute a -> self#floating_attribute f a
|
|
| Pstr_extension(e, a) ->
|
|
self#item_extension f e;
|
|
self#item_attributes f a
|
|
end
|
|
method type_param f (ct, a) =
|
|
pp f "%s%a" (type_variance a) self#core_type ct
|
|
method type_params f = function
|
|
[] -> ()
|
|
| l -> pp f "%a " (self#list self#type_param ~first:"(" ~last:")" ~sep:",") l
|
|
method type_def_list f (rf, l) =
|
|
let type_decl kwd rf f x =
|
|
let eq =
|
|
if (x.ptype_kind = Ptype_abstract)
|
|
&& (x.ptype_manifest = None) then ""
|
|
else " ="
|
|
in
|
|
pp f "@[<2>%s %a%a%s%s%a@]%a" kwd
|
|
self#nonrec_flag rf
|
|
self#type_params x.ptype_params
|
|
x.ptype_name.txt eq
|
|
self#type_declaration x
|
|
self#item_attributes x.ptype_attributes
|
|
in
|
|
match l with
|
|
| [] -> assert false
|
|
| [x] -> type_decl "type" rf f x
|
|
| x :: xs -> pp f "@[<v>%a@,%a@]"
|
|
(type_decl "type" rf) x
|
|
(self#list ~sep:"@," (type_decl "and" Recursive)) xs
|
|
|
|
method record_declaration f lbls =
|
|
let type_record_field f pld =
|
|
pp f "@[<2>%a%s:@;%a@;%a@]"
|
|
self#mutable_flag pld.pld_mutable
|
|
pld.pld_name.txt
|
|
self#core_type pld.pld_type
|
|
self#attributes pld.pld_attributes
|
|
in
|
|
pp f "{@\n%a}"
|
|
(self#list type_record_field ~sep:";@\n" ) lbls
|
|
|
|
method type_declaration f x =
|
|
(* type_declaration has an attribute field,
|
|
but it's been printed by the caller of this method *)
|
|
let priv f =
|
|
match x.ptype_private with
|
|
Public -> ()
|
|
| Private -> pp f "@;private"
|
|
in
|
|
let manifest f =
|
|
match x.ptype_manifest with
|
|
| None -> ()
|
|
| Some y ->
|
|
if x.ptype_kind = Ptype_abstract then
|
|
pp f "%t@;%a" priv self#core_type y
|
|
else
|
|
pp f "@;%a" self#core_type y
|
|
in
|
|
let constructor_declaration f pcd =
|
|
pp f "|@;";
|
|
self#constructor_declaration f (pcd.pcd_name.txt, pcd.pcd_args,
|
|
pcd.pcd_res, pcd.pcd_attributes)
|
|
in
|
|
let repr f =
|
|
let intro f =
|
|
if x.ptype_manifest = None then ()
|
|
else pp f "@;="
|
|
in
|
|
match x.ptype_kind with
|
|
| Ptype_variant xs ->
|
|
pp f "%t%t@\n%a" intro priv
|
|
(self#list ~sep:"@\n" constructor_declaration) xs
|
|
| Ptype_abstract -> ()
|
|
| Ptype_record l ->
|
|
pp f "%t%t@;%a" intro priv self#record_declaration l
|
|
| Ptype_open -> pp f "%t%t@;.." intro priv
|
|
in
|
|
let constraints f =
|
|
List.iter
|
|
(fun (ct1,ct2,_) ->
|
|
pp f "@[<hov2>@ constraint@ %a@ =@ %a@]"
|
|
self#core_type ct1 self#core_type ct2)
|
|
x.ptype_cstrs
|
|
in
|
|
pp f "%t%t%t" manifest repr constraints
|
|
|
|
method type_extension f x =
|
|
let extension_constructor f x =
|
|
pp f "@\n|@;%a" self#extension_constructor x
|
|
in
|
|
pp f "@[<2>type %a%a +=%a@]%a"
|
|
(fun f -> function
|
|
| [] -> ()
|
|
| l -> pp f "%a@;" (self#list self#type_param ~first:"("
|
|
~last:")" ~sep:",")
|
|
l)
|
|
x.ptyext_params
|
|
self#longident_loc x.ptyext_path
|
|
(self#list ~sep:"" extension_constructor)
|
|
x.ptyext_constructors
|
|
self#item_attributes x.ptyext_attributes
|
|
|
|
method constructor_declaration f (name, args, res, attrs) =
|
|
match res with
|
|
| None ->
|
|
pp f "%s%a@;%a" name
|
|
(fun f -> function
|
|
| Pcstr_tuple [] -> ()
|
|
| Pcstr_tuple l ->
|
|
pp f "@;of@;%a" (self#list self#core_type1 ~sep:"@;*@;") l
|
|
| Pcstr_record l -> pp f "@;of@;%a" (self#record_declaration) l
|
|
) args
|
|
self#attributes attrs
|
|
| Some r ->
|
|
pp f "%s:@;%a@;%a" name
|
|
(fun f -> function
|
|
| Pcstr_tuple [] -> self#core_type1 f r
|
|
| Pcstr_tuple l -> pp f "%a@;->@;%a"
|
|
(self#list self#core_type1 ~sep:"@;*@;") l
|
|
self#core_type1 r
|
|
| Pcstr_record l ->
|
|
pp f "%a@;->@;%a" (self#record_declaration) l self#core_type1 r
|
|
)
|
|
args
|
|
self#attributes attrs
|
|
|
|
|
|
method extension_constructor f x =
|
|
match x.pext_kind with
|
|
| Pext_decl(l, r) ->
|
|
self#constructor_declaration f (x.pext_name.txt, l, r,
|
|
x.pext_attributes)
|
|
| Pext_rebind li ->
|
|
pp f "%s%a@;=@;%a" x.pext_name.txt
|
|
self#attributes x.pext_attributes
|
|
self#longident_loc li
|
|
|
|
method case_list f l : unit =
|
|
let aux f {pc_lhs; pc_guard; pc_rhs} =
|
|
pp f "@;| @[<2>%a%a@;->@;%a@]"
|
|
self#pattern pc_lhs (self#option self#expression ~first:"@;when@;")
|
|
pc_guard self#under_pipe#expression pc_rhs
|
|
in
|
|
self#list aux f l ~sep:""
|
|
method label_x_expression_param f (l,e) =
|
|
let simple_name = match e.pexp_desc with
|
|
| Pexp_ident {txt=Lident l;_} -> Some l
|
|
| _ -> None
|
|
in match l with
|
|
| Nolabel -> self#expression2 f e ; (* level 2*)
|
|
| Optional str ->
|
|
if Some str = simple_name then
|
|
pp f "?%s" str
|
|
else
|
|
pp f "?%s:%a" str self#simple_expr e
|
|
| Labelled lbl ->
|
|
if Some lbl = simple_name then
|
|
pp f "~%s" lbl
|
|
else
|
|
pp f "~%s:%a" lbl self#simple_expr e
|
|
|
|
method directive_argument f x =
|
|
(match x with
|
|
| Pdir_none -> ()
|
|
| Pdir_string (s) -> pp f "@ %S" s
|
|
| Pdir_int (n,None) -> pp f "@ %s" n
|
|
| Pdir_int (n,Some m) -> pp f "@ %s%c" n m
|
|
| Pdir_ident (li) -> pp f "@ %a" self#longident li
|
|
| Pdir_bool (b) -> pp f "@ %s" (string_of_bool b))
|
|
|
|
method toplevel_phrase f x =
|
|
match x with
|
|
| Ptop_def (s) ->
|
|
pp_open_hvbox f 0;
|
|
self#list self#structure_item f s ;
|
|
pp_close_box f ();
|
|
| Ptop_dir (s, da) ->
|
|
pp f "@[<hov2>#%s@ %a@]" s self#directive_argument da
|
|
end;;
|
|
|
|
|
|
let default = new printer ()
|
|
|
|
|
|
let toplevel_phrase f x =
|
|
match x with
|
|
| Ptop_def (s) ->pp f "@[<hov0>%a@]" (default#list default#structure_item) s
|
|
(* pp_open_hvbox f 0; *)
|
|
(* pp_print_list structure_item f s ; *)
|
|
(* pp_close_box f (); *)
|
|
| Ptop_dir (s, da) ->
|
|
pp f "@[<hov2>#%s@ %a@]" s default#directive_argument da
|
|
(* pp f "@[<hov2>#%s@ %a@]" s directive_argument da *)
|
|
|
|
let expression f x =
|
|
pp f "@[%a@]" default#expression x
|
|
|
|
|
|
let string_of_expression x =
|
|
ignore (flush_str_formatter ()) ;
|
|
let f = str_formatter in
|
|
default#expression f x ;
|
|
flush_str_formatter () ;;
|
|
let string_of_structure x =
|
|
ignore (flush_str_formatter ());
|
|
let f = str_formatter in
|
|
default#structure f x;
|
|
flush_str_formatter ();;
|
|
|
|
let top_phrase f x =
|
|
pp_print_newline f () ;
|
|
toplevel_phrase f x;
|
|
pp f ";;" ;
|
|
pp_print_newline f ();;
|
|
|
|
let core_type=default#core_type
|
|
let pattern=default#pattern
|
|
let signature=default#signature
|
|
let structure=default#structure
|