helm/matita/matitacLib.ml

   1 (* Copyright (C) 2004-2005, HELM Team.
   2  *
   3  * This file is part of HELM, an Hypertextual, Electronic
   4  * Library of Mathematics, developed at the Computer Science
   5  * Department, University of Bologna, Italy.
   6  *
   7  * HELM is free software; you can redistribute it and/or
   8  * modify it under the terms of the GNU General Public License
   9  * as published by the Free Software Foundation; either version 2
  10  * of the License, or (at your option) any later version.
  11  *
  12  * HELM is distributed in the hope that it will be useful,
  13  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  14  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  15  * GNU General Public License for more details.
  16  *
  17  * You should have received a copy of the GNU General Public License
  18  * along with HELM; if not, write to the Free Software
  19  * Foundation, Inc., 59 Temple Place - Suite 330, Boston,
  20  * MA  02111-1307, USA.
  21  *
  22  * For details, see the HELM World-Wide-Web page,
  23  * http://helm.cs.unibo.it/
  24  *)
  25
  26 open Printf
  27
  28 open GrafiteTypes
  29
  30 (** {2 Initialization} *)
  31
  32 let status = ref None
  33
  34 let run_script is eval_function  =
  35   let status =
  36     match !status with
  37     | None -> assert false
  38     | Some s -> s
  39   in
  40   let slash_n_RE = Pcre.regexp "\\n" in
  41   let cb =
  42     if Helm_registry.get_bool "matita.quiet" then
  43       (fun _ _ -> ())
  44     else
  45       (fun status stm ->
  46         (* dump_status status; *)
  47         let stm = GrafiteAstPp.pp_statement stm in
  48         let stm = Pcre.replace ~rex:slash_n_RE stm in
  49         let stm =
  50           if String.length stm > 50 then
  51             String.sub stm 0 50 ^ " ..."
  52           else
  53             stm
  54         in
  55         HLog.debug ("Executing: ``" ^ stm ^ "''"))
  56   in
  57   try
  58     eval_function status is cb
  59   with
  60   | GrafiteEngine.Drop
  61   | End_of_file
  62   | CicNotationParser.Parse_error _ as exn -> raise exn
  63   | exn ->
  64       HLog.error (snd (MatitaExcPp.to_string exn));
  65       raise exn
  66
  67 let fname () =
  68   match Helm_registry.get_list Helm_registry.string "matita.args" with
  69   | [x] -> x
  70   | _ -> MatitaInit.die_usage ()
  71
  72 let pp_ocaml_mode () =
  73   HLog.message "";
  74   HLog.message "                      ** Entering Ocaml mode ** ";
  75   HLog.message "";
  76   HLog.message "Type 'go ();;' to enter an interactive matitac";
  77   HLog.message ""
  78
  79 let clean_exit n =
  80  let opt_exit =
  81   function
  82      None -> ()
  83    | Some n -> exit n
  84  in
  85   match !status with
  86      None -> opt_exit n
  87    | Some status ->
  88       try
  89        let baseuri = GrafiteTypes.get_string_option !status "baseuri" in
  90        let basedir = Helm_registry.get "matita.basedir" in
  91        LibraryClean.clean_baseuris ~basedir ~verbose:false [baseuri];
  92        opt_exit n
  93       with GrafiteTypes.Option_error("baseuri", "not found") ->
  94        (* no baseuri ==> nothing to clean yet *)
  95        opt_exit n
  96
  97 let rec interactive_loop () =
  98   let str = Ulexing.from_utf8_channel stdin in
  99   try
 100     run_script str
 101       (MatitaEngine.eval_from_stream_greedy
 102       ~include_paths:(Helm_registry.get_list Helm_registry.string
 103         "matita.includes"))
 104   with
 105   | GrafiteEngine.Drop -> pp_ocaml_mode ()
 106   | Sys.Break -> HLog.error "user break!"; interactive_loop ()
 107   | GrafiteTypes.Command_error _ -> interactive_loop ()
 108   | End_of_file ->
 109      print_newline ();
 110      clean_exit (Some 0)
 111   | HExtlib.Localized (floc,CicNotationParser.Parse_error err) ->
 112      let (x, y) = HExtlib.loc_of_floc floc in
 113      HLog.error (sprintf "Parse error at %d-%d: %s" x y err);
 114      interactive_loop ()
 115   | exn -> HLog.error (Printexc.to_string exn); interactive_loop ()
 116
 117 let go () =
 118   Helm_registry.load_from BuildTimeConf.matita_conf;
 119   CicNotation.load_notation BuildTimeConf.core_notation_script;
 120   Http_getter.init ();
 121   MetadataTypes.ownerize_tables (Helm_registry.get "matita.owner");
 122   LibraryDb.create_owner_environment ();
 123   CicEnvironment.set_trust (* environment trust *)
 124     (let trust = Helm_registry.get_bool "matita.environment_trust" in
 125      fun _ -> trust);
 126   status := Some (ref (Lazy.force MatitaEngine.initial_status));
 127   Sys.catch_break true;
 128   interactive_loop ()
 129
 130 let main ~mode =
 131   MatitaInit.initialize_all ();
 132   (* must be called after init since args are set by cmdline parsing *)
 133   let fname = fname () in
 134   status := Some (ref (Lazy.force MatitaEngine.initial_status));
 135   Sys.catch_break true;
 136   let origcb = HLog.get_log_callback () in
 137   let newcb tag s =
 138     match tag with
 139     | `Debug | `Message -> ()
 140     | `Warning | `Error -> origcb tag s
 141   in
 142   if Helm_registry.get_bool "matita.quiet" then
 143     HLog.set_log_callback newcb;
 144   let matita_debug = Helm_registry.get_bool "matita.debug" in
 145   try
 146     let time = Unix.time () in
 147     if Helm_registry.get_bool "matita.quiet" then
 148       origcb `Message ("compiling " ^ Filename.basename fname ^ "...")
 149     else
 150       HLog.message (sprintf "execution of %s started:" fname);
 151     let is =
 152       Ulexing.from_utf8_channel
 153         (match fname with
 154         | "stdin" -> stdin
 155         | fname -> open_in fname)
 156     in
 157     run_script is
 158       (MatitaEngine.eval_from_stream
 159         ~include_paths:(Helm_registry.get_list Helm_registry.string
 160           "matita.includes")
 161         ~clean_baseuri:(not (Helm_registry.get_bool "matita.preserve")));
 162     let elapsed = Unix.time () -. time in
 163     let tm = Unix.gmtime elapsed in
 164     let sec =
 165       if tm.Unix.tm_sec > 0 then (string_of_int tm.Unix.tm_sec ^ "''") else ""
 166     in
 167     let min =
 168       if tm.Unix.tm_min > 0 then (string_of_int tm.Unix.tm_min ^ "' ") else ""
 169     in
 170     let hou =
 171       if tm.Unix.tm_hour > 0 then (string_of_int tm.Unix.tm_hour ^ "h ") else ""
 172     in
 173     let proof_status,moo_content_rev,metadata,status =
 174       match !status with
 175       | Some s -> !s.proof_status, !s.moo_content_rev, !s.metadata, !s
 176       | None -> assert false
 177     in
 178     if proof_status <> GrafiteTypes.No_proof then
 179      begin
 180       HLog.error
 181        "there are still incomplete proofs at the end of the script";
 182       clean_exit (Some 2)
 183      end
 184     else
 185      begin
 186        let basedir = Helm_registry.get "matita.basedir" in
 187        let moo_fname = GrafiteMisc.obj_file_of_script ~basedir fname in
 188        let metadata_fname = GrafiteMisc.metadata_file_of_script ~basedir fname in
 189        GrafiteMarshal.save_moo moo_fname moo_content_rev;
 190        LibraryNoDb.save_metadata metadata_fname metadata;
 191        HLog.message
 192          (sprintf "execution of %s completed in %s." fname (hou^min^sec));
 193        exit 0
 194      end
 195   with
 196   | Sys.Break ->
 197       HLog.error "user break!";
 198       if mode = `COMPILER then
 199         clean_exit (Some ~-1)
 200       else
 201         pp_ocaml_mode ()
 202   | GrafiteEngine.Drop ->
 203       if mode = `COMPILER then
 204         clean_exit (Some 1)
 205       else
 206         pp_ocaml_mode ()
 207   | HExtlib.Localized (floc,CicNotationParser.Parse_error err) ->
 208      let (x, y) = HExtlib.loc_of_floc floc in
 209      HLog.error (sprintf "Parse error at %d-%d: %s" x y err);
 210      if mode = `COMPILER then
 211        clean_exit (Some 1)
 212      else
 213        pp_ocaml_mode ()
 214   | exn ->
 215       if matita_debug then raise exn;
 216       if mode = `COMPILER then
 217         clean_exit (Some 3)
 218       else
 219         pp_ocaml_mode ()
 220