balsoft/ligo
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

Merge branch 'print-optimized-mini-c' into 'dev'

Browse Source 
Add `print-mini-c --optimize=entrypoint` to print optimized mini_c

See merge request ligolang/ligo!515
This commit is contained in:
Rémi Lesenechal 2020-04-21 13:28:39 +00:00
commit 16ecd532b9
2 changed files with 28 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

18
src/bin/cli.ml
View File

@ -224,15 +224,27 @@ let print_ast_typed =
let doc = "Subcommand: Print the typed AST.\n Warning: Intended for development of LIGO and can break at any time." in let doc = "Subcommand: Print the typed AST.\n Warning: Intended for development of LIGO and can break at any time." in
(Term.ret term, Term.info ~doc cmdname) (Term.ret term, Term.info ~doc cmdname)
let optimize =
let open Arg in
let docv = "ENTRY_POINT" in
let doc = "Apply Mini-C optimizations as if compiling $(docv)" in
let info =
info ~docv ~doc ["optimize"] in
value @@ opt (some string) None info
let print_mini_c = let print_mini_c =
let f source_file syntax display_format = ( let f source_file syntax display_format optimize = (
toplevel ~display_format @@ toplevel ~display_format @@
let%bind typed,_ = Compile.Utils.type_file source_file syntax Env in let%bind typed,_ = Compile.Utils.type_file source_file syntax Env in
let%bind mini_c = Compile.Of_typed.compile typed in let%bind mini_c = Compile.Of_typed.compile typed in
ok @@ Format.asprintf "%a\n" Compile.Of_mini_c.pretty_print mini_c match optimize with
| None -> ok @@ Format.asprintf "%a\n" Compile.Of_mini_c.pretty_print mini_c
| Some entry_point ->
let%bind mini_c = Compile.Of_mini_c.aggregate_contract mini_c entry_point in
ok @@ Format.asprintf "%a\n" Mini_c.PP.expression mini_c
) )
in in
let term = Term.(const f $ source_file 0 $ syntax $ display_format) in let term = Term.(const f $ source_file 0 $ syntax $ display_format $ optimize) in
let cmdname = "print-mini-c" in let cmdname = "print-mini-c" in
let doc = "Subcommand: Print Mini-C. Warning: Intended for development of LIGO and can break at any time." in let doc = "Subcommand: Print Mini-C. Warning: Intended for development of LIGO and can break at any time." in
(Term.ret term, Term.info ~doc cmdname) (Term.ret term, Term.info ~doc cmdname)

12
src/main/compile/of_mini_c.ml
View File

@ -34,3 +34,15 @@ let aggregate_and_compile_expression = fun program exp ->
let pretty_print program = let pretty_print program =
Mini_c.PP.program program Mini_c.PP.program program
(* TODO refactor? *)
let aggregate = fun program form ->
let%bind aggregated = aggregate_entry program form in
ok @@ Self_mini_c.all_expression aggregated
let aggregate_contract = fun (program : Types.program) name ->
let%bind (exp, idx) = get_entry program name in
let program' = List.take idx program in
aggregate program' (ContractForm exp)