all initialization code is now in the new matitaInit.ml module.

[helm.git] / helm / matita / matitadep.ml

diff --git a/helm/matita/matitadep.ml b/helm/matita/matitadep.ml
index 88cac82807ff10953bb6549372b70c5a49b94de2..70240068fe71624063bac38e4ffdaf33bf652d0d 100644 (file)
--- a/helm/matita/matitadep.ml
+++ b/helm/matita/matitadep.ml
@@ -35,7 +35,7 @@ let usage =
   Printf.sprintf "MatitaDep v%s\nUsage: matitadep [options] file...\nOptions:"
     BuildTimeConf.version
 
-module TA = GrafiteAst 
+module GA = GrafiteAst 
 module U = UriManager
 
 let deps = Hashtbl.create (Array.length Sys.argv)
@@ -75,27 +75,67 @@ let find path =
     raise exc
 ;;
 
+(* note, this function works cause moo files are transitively closed *)
+let process_notation_of_file file =
+    let ic = open_in file in
+    let istream = Stream.of_channel ic in
+    let notation_ids = ref [] in
+    try
+      while true do
+        match GrafiteParser.parse_statement istream with
+        | GA.Executable (_, GA.Command (_, notation)) ->
+          let stm, ids = CicNotation.process_notation notation in
+          notation_ids := ids @ !notation_ids
+        | _ -> ()
+      done; []
+    with
+    | End_of_file -> close_in ic; !notation_ids
+    | CicNotationParser.Parse_error _ as exn ->
+        prerr_endline ("Unable to parse: " ^ file);
+        prerr_endline (MatitaExcPp.to_string exn);
+        raise exn
+
 let main () =
-  Helm_registry.load_from BuildTimeConf.matita_conf;
-  CicNotation.load_notation BuildTimeConf.core_notation_script;
+  MatitaInit.initialize_notation ();
   let files = ref [] in
   Arg.parse arg_spec (add_l files) usage;
   List.iter
    (fun file -> 
     let ic = open_in file in
     let istream = Stream.of_channel ic in
+    let notation_ids = ref [] in
     (try
       while true do
         try 
-          match GrafiteParser.parse_statement istream with
-          | TA.Executable (_, TA.Command (_, TA.Set (_, "baseuri", uri))) ->
+          let stm = GrafiteParser.parse_statement istream in
+          (match stm with 
+          | GA.Executable (_, GA.Command (_, notation)) ->
+            let stm, ids = CicNotation.process_notation notation in
+            notation_ids := ids @ !notation_ids
+          | _ -> ());
+          match stm with
+          | GA.Executable (_, GA.Command (_, GA.Set (_, "baseuri", uri))) ->
               let uri = MatitaMisc.strip_trailing_slash uri in
               baseuris := uri :: !baseuris
-          | TA.Executable (_, TA.Command 
-              (_, TA.Alias (_, TA.Ident_alias(_, uri)))) -> 
+          | GA.Executable (_, GA.Command 
+              (_, GA.Alias (_, GA.Ident_alias(_, uri)))) -> 
                 Hashtbl.add aliases file uri
-          | TA.Executable (_, TA.Command (_, TA.Include (_, path))) ->
-              Hashtbl.add deps file (MatitaMisc.obj_file_of_script (find path))
+          | GA.Executable (_, GA.Command (_, GA.Include (_, path))) ->
+              (* maybe to process coq.ma is useles since it contains only
+               * interpretations... no idea if notations will be added later *)
+              let moo_file = 
+                MatitaMisc.obj_file_of_script (
+                  if path <> "coq.ma" then find path else path)
+              in
+              let ids = process_notation_of_file moo_file in
+              notation_ids := ids @ !notation_ids;
+              (try
+                Hashtbl.add deps file moo_file
+               with
+                Sys_error _ ->
+                 MatitaLog.error
+                  ("In file " ^ file ^ " unable to include " ^ path)
+              )
           | _ -> ()
         with 
           CicNotationParser.Parse_error _ as exn ->
@@ -111,7 +151,9 @@ let main () =
         match dep with 
         | None -> ()
         | Some u -> Hashtbl.add deps file (MatitaMisc.obj_file_of_baseuri u))
-      aliases
+      aliases;
+    (* remove notation ids *)
+    List.iter CicNotation.remove_notation !notation_ids;
    ) !files;
   List.iter
    (fun file ->