(* $Id$ *)
+module G = GrafiteAst
+
+(* from transcript *)
+
+let out_comment och s =
+ let s = if s <> "" && s.[0] = '*' then "#" ^ s else s in
+ Printf.fprintf och "%s%s%s\n\n" "(*" s "*)"
+
+let out_line_comment och s =
+ let l = 70 - String.length s in
+ let s = Printf.sprintf " %s %s" s (String.make l '*') in
+ out_comment och s
+
+let out_preamble och (path, lines) =
+ let ich = open_in path in
+ let rec print i =
+ if i > 0 then
+ let s = input_line ich in
+ begin Printf.fprintf och "%s\n" s; print (pred i) end
+ in
+ print lines;
+ out_line_comment och "This file was automatically generated: do not edit"
+
+(* from matitacLib *)
+
+let pp_ast_statement =
+ GrafiteAstPp.pp_statement ~term_pp:CicNotationPp.pp_term
+ ~lazy_term_pp:CicNotationPp.pp_term ~obj_pp:(CicNotationPp.pp_obj CicNotationPp.pp_term)
+
+(**)
+
+let dump f =
+ let och = open_out f in
+ let atexit () = close_out och in
+ let rt_base_dir = Filename.dirname Sys.argv.(0) in
+ let path = Filename.concat rt_base_dir "matita.ma.templ" in
+ let lines = 14 in
+ out_preamble och (path, lines);
+ let grafite_parser_cb ast =
+ output_string och (LexiconAstPp.pp_command ast)
+ in
+ let matita_engine_cb = function
+ | G.Executable (_, G.Macro (_, G.Inline _)) -> ()
+ | ast ->
+ output_string och (pp_ast_statement ast)
+ in
+ let matitac_lib_cb = output_string och in
+(* GrafiteParser.set_callback grafite_parser_cb; *)
+ MatitaEngine.set_callback matita_engine_cb;
+ MatitacLib.set_callback matitac_lib_cb;
+ at_exit atexit
+
let main () =
match Filename.basename Sys.argv.(0) with
|"gragrep" |"gragrep.opt" |"gragrep.opt.static" ->Gragrep.main()
(*
let _ = Paramodulation.Saturation.init () in *)
(* ALB to link paramodulation *)
+ let dump_msg = "<filename> Dump source with expanded macros to <filename>" in
+ MatitaInit.add_cmdline_spec ["-dump", Arg.String dump, dump_msg];
let _ = MatitacLib.main `COMPILER in
()
with
| GrafiteEngine.Drop
| End_of_file
- | CicNotationParser.Parse_error _ as exn -> raise exn
+ | CicNotationParser.Parse_error _
+ | GrafiteEngine.Macro _ as exn -> raise exn
| exn ->
if not matita_debug then
HLog.error (snd (MatitaExcPp.to_string exn)) ;
with GrafiteTypes.Option_error("baseuri", "not found") ->
(* no baseuri ==> nothing to clean yet *)
opt_exit n
-
+
+let get_macro_context = function
+ | Some {GrafiteTypes.proof_status = GrafiteTypes.No_proof} -> []
+ | Some status ->
+ let stack = GrafiteTypes.get_stack status in
+ let goal = Continuationals.Stack.find_goal stack in
+ GrafiteTypes.get_proof_context status goal
+ | None -> assert false
+
let rec interactive_loop () =
let str = Ulexing.from_utf8_channel stdin in
try
with
| GrafiteEngine.Drop -> pp_ocaml_mode ()
| GrafiteEngine.Macro (floc, f) ->
- begin match snd (f []) with
- | GrafiteAst.Inline (_, style, suri, prefix) ->
+ begin match f (get_macro_context !grafite_status) with
+ | _, GrafiteAst.Inline (_, style, suri, prefix) ->
let str = ApplyTransformation.txt_of_inline_macro style suri prefix in
- !out str
- | _ -> ()
- end;
- let x, y = HExtlib.loc_of_floc floc in
- HLog.error
- (sprintf "A macro has been found in a script at %d-%d" x y);
- interactive_loop ()
+ !out str;
+ interactive_loop ()
+ | _ ->
+ let x, y = HExtlib.loc_of_floc floc in
+ HLog.error (sprintf "A macro has been found in a script at %d-%d" x y);
+ interactive_loop ()
+ end
| Sys.Break -> HLog.error "user break!"; interactive_loop ()
| GrafiteTypes.Command_error _ -> interactive_loop ()
| End_of_file ->
end
;;
+let rec compiler_loop fname big_bang mode buf =
+ let include_paths =
+ Helm_registry.get_list Helm_registry.string "matita.includes" in
+ let clean_baseuri = not (Helm_registry.get_bool "matita.preserve") in
+ let matita_debug = Helm_registry.get_bool "matita.debug" in
+ let bench_mode = Helm_registry.get_bool "matita.bench" in
+ try
+ run_script buf
+ (MatitaEngine.eval_from_stream ~first_statement_only:false ~include_paths
+ ~clean_baseuri)
+ with
+ | End_of_file -> ()
+ | Sys.Break as exn ->
+ if matita_debug then raise exn;
+ HLog.error "user break!";
+ pp_times fname bench_mode false big_bang;
+ if mode = `COMPILER then
+ clean_exit (Some ~-1)
+ else
+ pp_ocaml_mode ()
+ | GrafiteEngine.Drop ->
+ if mode = `COMPILER then
+ begin
+ pp_times fname bench_mode false big_bang;
+ clean_exit (Some 1)
+ end
+ else
+ pp_ocaml_mode ()
+ | GrafiteEngine.Macro (floc, f) ->
+ begin match f (get_macro_context !grafite_status) with
+ | _, GrafiteAst.Inline (_, style, suri, prefix) ->
+ let str = ApplyTransformation.txt_of_inline_macro style suri prefix in
+ !out str;
+ compiler_loop fname big_bang mode buf
+ | _ ->
+ let x, y = HExtlib.loc_of_floc floc in
+ HLog.error (sprintf "A macro has been found in a script at %d-%d" x y);
+ if mode = `COMPILER then
+ begin
+ pp_times fname bench_mode false big_bang;
+ clean_exit (Some 1)
+ end
+ else
+ pp_ocaml_mode ()
+ end
+ | HExtlib.Localized (floc,CicNotationParser.Parse_error err) ->
+ let (x, y) = HExtlib.loc_of_floc floc in
+ HLog.error (sprintf "Parse error at %d-%d: %s" x y err);
+ if mode = `COMPILER then
+ begin
+ pp_times fname bench_mode false big_bang;
+ clean_exit (Some 1)
+ end
+ else
+ pp_ocaml_mode ()
+ | exn ->
+ if matita_debug then raise exn;
+ if mode = `COMPILER then
+ begin
+ pp_times fname bench_mode false big_bang;
+ clean_exit (Some 3)
+ end
+ else
+ pp_ocaml_mode ()
+
let main ~mode =
let big_bang = Unix.gettimeofday () in
MatitaInit.initialize_all ();
if Helm_registry.get_int "matita.verbosity" < 1 then
HLog.set_log_callback newcb;
if bench_mode then MatitaMisc.shutup ();
- let matita_debug = Helm_registry.get_bool "matita.debug" in
- try
let time = Unix.time () in
if Helm_registry.get_int "matita.verbosity" < 1 && not bench_mode then
origcb `Message ("compiling " ^ Filename.basename fname ^ "...")
else
HLog.message (sprintf "execution of %s started:" fname);
- let is =
- Ulexing.from_utf8_channel
- (match fname with
- | "stdin" -> stdin
- | fname -> open_in fname) in
- let include_paths =
- Helm_registry.get_list Helm_registry.string "matita.includes" in
- (try
- run_script is
- (MatitaEngine.eval_from_stream ~first_statement_only:false ~include_paths
- ~clean_baseuri:(not (Helm_registry.get_bool "matita.preserve")))
- with End_of_file -> ());
+ let ich = match fname with
+ | "stdin" -> stdin
+ | fname -> open_in fname
+ in
+ let buf = Ulexing.from_utf8_channel ich in
+ compiler_loop fname big_bang mode buf;
let elapsed = Unix.time () -. time in
let tm = Unix.gmtime elapsed in
let sec = string_of_int tm.Unix.tm_sec ^ "''" in
pp_times fname bench_mode true big_bang;
exit 0
end
- with
- | Sys.Break as exn ->
- if matita_debug then raise exn;
- HLog.error "user break!";
- pp_times fname bench_mode false big_bang;
- if mode = `COMPILER then
- clean_exit (Some ~-1)
- else
- pp_ocaml_mode ()
- | GrafiteEngine.Drop ->
- if mode = `COMPILER then
- begin
- pp_times fname bench_mode false big_bang;
- clean_exit (Some 1)
- end
- else
- pp_ocaml_mode ()
- | GrafiteEngine.Macro (floc,_) ->
- let x, y = HExtlib.loc_of_floc floc in
- HLog.error
- (sprintf "A macro has been found in a script at %d-%d" x y);
- if mode = `COMPILER then
- begin
- pp_times fname bench_mode false big_bang;
- clean_exit (Some 1)
- end
- else
- pp_ocaml_mode ()
- | HExtlib.Localized (floc,CicNotationParser.Parse_error err) ->
- let (x, y) = HExtlib.loc_of_floc floc in
- HLog.error (sprintf "Parse error at %d-%d: %s" x y err);
- if mode = `COMPILER then
- begin
- pp_times fname bench_mode false big_bang;
- clean_exit (Some 1)
- end
- else
- pp_ocaml_mode ()
- | exn ->
- if matita_debug then raise exn;
- if mode = `COMPILER then
- begin
- pp_times fname bench_mode false big_bang;
- clean_exit (Some 3)
- end
- else
- pp_ocaml_mode ()
-