ocaml/asmcomp/printcmm.ml

308 lines
10 KiB
OCaml

(**************************************************************************)
(* *)
(* OCaml *)
(* *)
(* Xavier Leroy, projet Cristal, INRIA Rocquencourt *)
(* *)
(* Copyright 1996 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. *)
(* *)
(**************************************************************************)
(* Pretty-printing of C-- code *)
open Format
open Cmm
module V = Backend_var
module VP = Backend_var.With_provenance
let rec_flag ppf = function
| Nonrecursive -> ()
| Recursive -> fprintf ppf " rec"
let machtype_component ppf = function
| Val -> fprintf ppf "val"
| Addr -> fprintf ppf "addr"
| Int -> fprintf ppf "int"
| Float -> fprintf ppf "float"
let machtype ppf mty =
match Array.length mty with
| 0 -> fprintf ppf "unit"
| n -> machtype_component ppf mty.(0);
for i = 1 to n-1 do
fprintf ppf "*%a" machtype_component mty.(i)
done
let exttype ppf = function
| XInt -> fprintf ppf "int"
| XInt32 -> fprintf ppf "int32"
| XInt64 -> fprintf ppf "int64"
| XFloat -> fprintf ppf "float"
let extcall_signature ppf (ty_res, ty_args) =
begin match ty_args with
| [] -> ()
| ty_arg1 :: ty_args ->
exttype ppf ty_arg1;
List.iter (fun ty -> fprintf ppf ",%a" exttype ty) ty_args
end;
fprintf ppf "->%a" machtype ty_res
let integer_comparison = function
| Ceq -> "=="
| Cne -> "!="
| Clt -> "<"
| Cle -> "<="
| Cgt -> ">"
| Cge -> ">="
let float_comparison = function
| CFeq -> "=="
| CFneq -> "!="
| CFlt -> "<"
| CFnlt -> "!<"
| CFle -> "<="
| CFnle -> "!<="
| CFgt -> ">"
| CFngt -> "!>"
| CFge -> ">="
| CFnge -> "!>="
let chunk = function
| Byte_unsigned -> "unsigned int8"
| Byte_signed -> "signed int8"
| Sixteen_unsigned -> "unsigned int16"
| Sixteen_signed -> "signed int16"
| Thirtytwo_unsigned -> "unsigned int32"
| Thirtytwo_signed -> "signed int32"
| Word_int -> "int"
| Word_val -> "val"
| Single -> "float32"
| Double -> "float64"
| Double_u -> "float64u"
let phantom_defining_expr ppf defining_expr =
match defining_expr with
| Cphantom_const_int i -> Targetint.print ppf i
| Cphantom_const_symbol sym -> Format.pp_print_string ppf sym
| Cphantom_var var -> V.print ppf var
| Cphantom_offset_var { var; offset_in_words; } ->
Format.fprintf ppf "%a+(%d)" V.print var offset_in_words
| Cphantom_read_field { var; field; } ->
Format.fprintf ppf "%a[%d]" V.print var field
| Cphantom_read_symbol_field { sym; field; } ->
Format.fprintf ppf "%s[%d]" sym field
| Cphantom_block { tag; fields; } ->
Format.fprintf ppf "[%d: " tag;
List.iter (fun field ->
Format.fprintf ppf "%a; " V.print field)
fields;
Format.fprintf ppf "]"
let phantom_defining_expr_opt ppf defining_expr =
match defining_expr with
| None -> Format.pp_print_string ppf "()"
| Some defining_expr -> phantom_defining_expr ppf defining_expr
let location d =
if not !Clflags.locations then ""
else Debuginfo.to_string d
let operation d = function
| Capply _ty -> "app" ^ location d
| Cextcall(lbl, _ty_res, _ty_args, _alloc) ->
Printf.sprintf "extcall \"%s\"%s" lbl (location d)
| Cload (c, Asttypes.Immutable) -> Printf.sprintf "load %s" (chunk c)
| Cload (c, Asttypes.Mutable) -> Printf.sprintf "load_mut %s" (chunk c)
| Calloc -> "alloc" ^ location d
| Cstore (c, init) ->
let init =
match init with
| Lambda.Heap_initialization -> "(heap-init)"
| Lambda.Root_initialization -> "(root-init)"
| Lambda.Assignment -> ""
in
Printf.sprintf "store %s%s" (chunk c) init
| Caddi -> "+"
| Csubi -> "-"
| Cmuli -> "*"
| Cmulhi -> "*h"
| Cdivi -> "/"
| Cmodi -> "mod"
| Cand -> "and"
| Cor -> "or"
| Cxor -> "xor"
| Clsl -> "<<"
| Clsr -> ">>u"
| Casr -> ">>s"
| Ccmpi c -> integer_comparison c
| Caddv -> "+v"
| Cadda -> "+a"
| Ccmpa c -> Printf.sprintf "%sa" (integer_comparison c)
| Cnegf -> "~f"
| Cabsf -> "absf"
| Caddf -> "+f"
| Csubf -> "-f"
| Cmulf -> "*f"
| Cdivf -> "/f"
| Cfloatofint -> "floatofint"
| Cintoffloat -> "intoffloat"
| Ccmpf c -> Printf.sprintf "%sf" (float_comparison c)
| Craise k -> Lambda.raise_kind k ^ location d
| Ccheckbound -> "checkbound" ^ location d
let rec expr ppf = function
| Cconst_int (n, _dbg) -> fprintf ppf "%i" n
| Cconst_natint (n, _dbg) ->
fprintf ppf "%s" (Nativeint.to_string n)
| Cconst_float (n, _dbg) -> fprintf ppf "%F" n
| Cconst_symbol (s, _dbg) -> fprintf ppf "\"%s\"" s
| Cvar id -> V.print ppf id
| Clet(id, def, (Clet(_, _, _) as body)) ->
let print_binding id ppf def =
fprintf ppf "@[<2>%a@ %a@]"
VP.print id expr def in
let rec in_part ppf = function
| Clet(id, def, body) ->
fprintf ppf "@ %a" (print_binding id) def;
in_part ppf body
| exp -> exp in
fprintf ppf "@[<2>(let@ @[<1>(%a" (print_binding id) def;
let exp = in_part ppf body in
fprintf ppf ")@]@ %a)@]" sequence exp
| Clet(id, def, body) ->
fprintf ppf
"@[<2>(let@ @[<2>%a@ %a@]@ %a)@]"
VP.print id expr def sequence body
| Clet_mut(id, kind, def, body) ->
fprintf ppf
"@[<2>(let_mut@ @[<2>%a: %a@ %a@]@ %a)@]"
VP.print id machtype kind expr def sequence body
| Cphantom_let(var, def, (Cphantom_let(_, _, _) as body)) ->
let print_binding var ppf def =
fprintf ppf "@[<2>%a@ %a@]" VP.print var
phantom_defining_expr_opt def
in
let rec in_part ppf = function
| Cphantom_let(var, def, body) ->
fprintf ppf "@ %a" (print_binding var) def;
in_part ppf body
| exp -> exp in
fprintf ppf "@[<2>(let?@ @[<1>(%a" (print_binding var) def;
let exp = in_part ppf body in
fprintf ppf ")@]@ %a)@]" sequence exp
| Cphantom_let(var, def, body) ->
fprintf ppf
"@[<2>(let?@ @[<2>%a@ %a@]@ %a)@]"
VP.print var
phantom_defining_expr_opt def
sequence body
| Cassign(id, exp) ->
fprintf ppf "@[<2>(assign @[<2>%a@ %a@])@]" V.print id expr exp
| Ctuple el ->
let tuple ppf el =
let first = ref true in
List.iter
(fun e ->
if !first then first := false else fprintf ppf "@ ";
expr ppf e)
el in
fprintf ppf "@[<1>[%a]@]" tuple el
| Cop(op, el, dbg) ->
fprintf ppf "@[<2>(%s" (operation dbg op);
List.iter (fun e -> fprintf ppf "@ %a" expr e) el;
begin match op with
| Capply mty -> fprintf ppf "@ %a" machtype mty
| Cextcall(_, ty_res, ty_args, _) ->
fprintf ppf "@ %a" extcall_signature (ty_res, ty_args)
| _ -> ()
end;
fprintf ppf ")@]"
| Csequence(e1, e2) ->
fprintf ppf "@[<2>(seq@ %a@ %a)@]" sequence e1 sequence e2
| Cifthenelse(e1, _e2_dbg, e2, _e3_dbg, e3, _dbg) ->
fprintf ppf "@[<2>(if@ %a@ %a@ %a)@]" expr e1 expr e2 expr e3
| Cswitch(e1, index, cases, _dbg) ->
let print_case i ppf =
for j = 0 to Array.length index - 1 do
if index.(j) = i then fprintf ppf "case %i:" j
done in
let print_cases ppf =
for i = 0 to Array.length cases - 1 do
fprintf ppf "@ @[<2>%t@ %a@]" (print_case i) sequence (fst cases.(i))
done in
fprintf ppf "@[<v 0>@[<2>(switch@ %a@ @]%t)@]" expr e1 print_cases
| Ccatch(flag, handlers, e1) ->
let print_handler ppf (i, ids, e2, _dbg) =
fprintf ppf "(%d%a)@ %a"
i
(fun ppf ids ->
List.iter
(fun (id, ty) ->
fprintf ppf "@ %a: %a"
VP.print id machtype ty)
ids) ids
sequence e2
in
let print_handlers ppf l =
List.iter (print_handler ppf) l
in
fprintf ppf
"@[<2>(catch%a@ %a@;<1 -2>with%a)@]"
rec_flag flag
sequence e1
print_handlers handlers
| Cexit (i, el) ->
fprintf ppf "@[<2>(exit %d" i;
List.iter (fun e -> fprintf ppf "@ %a" expr e) el;
fprintf ppf ")@]"
| Ctrywith(e1, id, e2, _dbg) ->
fprintf ppf "@[<2>(try@ %a@;<1 -2>with@ %a@ %a)@]"
sequence e1 VP.print id sequence e2
and sequence ppf = function
| Csequence(e1, e2) -> fprintf ppf "%a@ %a" sequence e1 sequence e2
| e -> expression ppf e
and expression ppf e = fprintf ppf "%a" expr e
let fundecl ppf f =
let print_cases ppf cases =
let first = ref true in
List.iter
(fun (id, ty) ->
if !first then first := false else fprintf ppf "@ ";
fprintf ppf "%a: %a" VP.print id machtype ty)
cases in
fprintf ppf "@[<1>(function%s %s@;<1 4>@[<1>(%a)@]@ @[%a@])@]@."
(location f.fun_dbg) f.fun_name
print_cases f.fun_args sequence f.fun_body
let data_item ppf = function
| Cdefine_symbol s -> fprintf ppf "\"%s\":" s
| Cglobal_symbol s -> fprintf ppf "global \"%s\"" s
| Cint8 n -> fprintf ppf "byte %i" n
| Cint16 n -> fprintf ppf "int16 %i" n
| Cint32 n -> fprintf ppf "int32 %s" (Nativeint.to_string n)
| Cint n -> fprintf ppf "int %s" (Nativeint.to_string n)
| Csingle f -> fprintf ppf "single %F" f
| Cdouble f -> fprintf ppf "double %F" f
| Csymbol_address s -> fprintf ppf "addr \"%s\"" s
| Cstring s -> fprintf ppf "string \"%s\"" s
| Cskip n -> fprintf ppf "skip %i" n
| Calign n -> fprintf ppf "align %i" n
let data ppf dl =
let items ppf = List.iter (fun d -> fprintf ppf "@ %a" data_item d) dl in
fprintf ppf "@[<hv 1>(data%t)@]" items
let phrase ppf = function
| Cfunction f -> fundecl ppf f
| Cdata dl -> data ppf dl