-(* Copyright (C) 2005, HELM Team.
- *
- * This file is part of HELM, an Hypertextual, Electronic
- * Library of Mathematics, developed at the Computer Science
- * Department, University of Bologna, Italy.
- *
- * HELM is free software; you can redistribute it and/or
- * modify it under the terms of the GNU General Public License
- * as published by the Free Software Foundation; either version 2
- * of the License, or (at your option) any later version.
- *
- * HELM is distributed in the hope that it will be useful,
- * but WITHOUT ANY WARRANTY; without even the implied warranty of
- * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
- * GNU General Public License for more details.
- *
- * You should have received a copy of the GNU General Public License
- * along with HELM; if not, write to the Free Software
- * Foundation, Inc., 59 Temple Place - Suite 330, Boston,
- * MA 02111-1307, USA.
- *
- * For details, see the HELM World-Wide-Web page,
- * http://helm.cs.unibo.it/
- *)
-
-(* $Id$ *)
-
-module S = Set.Make (String)
-
-module GA = GrafiteAst
-module HR = Helm_registry
-
-let print_times msg =
- let times = Unix.times () in
- let stamp = times.Unix.tms_utime +. times.Unix.tms_utime in
- Printf.printf "TIME STAMP: %s: %f\n" msg stamp; flush stdout; stamp
-
-let fix_name f =
- let f =
- if Pcre.pmatch ~pat:"^\\./" f then String.sub f 2 (String.length f - 2)
- else f
- in
- HExtlib.normalize_path f
-
-(* FG: old function left for reference *)
-let exclude excluded_files files =
- let map file = not (List.mem (fix_name file) excluded_files) in
- List.filter map files
-
-let generate_theory theory_file deps =
- if theory_file = "" then deps else
- let map (files, deps) (t, d) =
- if t = theory_file then files, deps else
- S.add t files, List.fold_left (fun deps dep -> S.add dep deps) deps d
- in
- let out_include och dep =
- Printf.fprintf och "include \"%s\".\n\n" dep
- in
- let fileset, depset = List.fold_left map (S.empty, S.empty) deps in
- let top_depset = S.diff fileset depset in
- let och = open_out theory_file in
- begin
- MatitaMisc.out_preamble och;
- S.iter (out_include och) top_depset;
- close_out och;
- (theory_file, S.elements top_depset) :: deps
- end
-
-let main () =
- let _ = print_times "inizio" in
- let include_paths = ref [] in
- let use_stdout = ref false in
- let theory_file = ref "" in
- (* all are maps from "file" to "something" *)
- let include_deps = Hashtbl.create 13 in
- let baseuri_of = Hashtbl.create 13 in
- let baseuri_of_inv = Hashtbl.create 13 in
- let dot_name = "depends" in
- let dot_file = ref "" in
- let set_dot_file () = dot_file := dot_name^".dot" in
- let set_theory_file s = theory_file := s ^ ".ma" in
- (* helpers *)
- let rec baseuri_of_script s =
- try Hashtbl.find baseuri_of s
- with Not_found ->
- let _,b,_,_ =
- Librarian.baseuri_of_script ~include_paths:!include_paths s
- in
- Hashtbl.add baseuri_of s b;
- Hashtbl.add baseuri_of_inv b s;
- let _ =
- if Filename.check_suffix s ".mma" then
- let generated = Filename.chop_suffix s ".mma" ^ ".ma" in
- ignore (baseuri_of_script generated)
- in
- b
- in
- let script_of_baseuri ma b =
- try Some (Hashtbl.find baseuri_of_inv b)
- with Not_found ->
- HLog.error ("Skipping dependency of '"^ma^"' over '"^b^"'");
- HLog.error ("Please include the file defining such baseuri, or fix");
- HLog.error ("possibly incorrect verbatim URIs in the .ma file.");
- None
- in
- let buri alias = NUri.baseuri_of_uri (NUri.uri_of_string alias) in
- let resolve alias current_buri =
- let buri = buri alias in
- if buri <> current_buri then Some buri else None
- in
- (* initialization *)
- MatitaInit.add_cmdline_spec
- ["-dot", Arg.Unit set_dot_file,
- "Save dependency graph in dot format and generate a png";
- "-stdout", Arg.Set use_stdout,
- "Print dependences on stdout";
- "-theory <name>", Arg.String set_theory_file,
- "generate a theory file <name>.ma (it includes all other files)"
- ];
- MatitaInit.parse_cmdline_and_configuration_file ();
- MatitaInit.initialize_environment ();
- if not (Helm_registry.get_bool "matita.verbose") then MatitaMisc.shutup ();
- let cmdline_args = HR.get_list HR.string "matita.args" in
- let test x =
- Filename.check_suffix x ".ma" || Filename.check_suffix x ".mma"
- in
- let files = fun () -> match cmdline_args with
- | [] -> HExtlib.find ~test "."
- | _ -> cmdline_args
- in
- let args =
- let roots = Librarian.find_roots_in_dir (Sys.getcwd ()) in
- match roots with
- | [] ->
- prerr_endline ("No roots found in " ^ Sys.getcwd ());
- exit 1
- | [x] ->
- Sys.chdir (Filename.dirname x);
- let opts = Librarian.load_root_file "root" in
- include_paths :=
- (try Str.split (Str.regexp " ") (List.assoc "include_paths" opts)
- with Not_found -> []) @
- (HR.get_list HR.string "matita.includes");
- files ()
- | _ ->
- let roots = List.map (HExtlib.chop_prefix (Sys.getcwd()^"/")) roots in
- prerr_endline ("Too many roots found:\n\t"^String.concat "\n\t" roots);
- prerr_endline ("\nEnter one of these directories and retry");
- exit 1
- in
- let ma_files = args in
- (* here we go *)
- (* fills:
- Hashtbl.add include_deps ma_file ma_file
- Hashtbl.add include_deps_dot ma_file baseuri
- *)
- let _ = print_times "prima di iter1" in
- List.iter (fun ma_file -> ignore (baseuri_of_script ma_file)) ma_files;
- let _ = print_times "in mezzo alle due iter" in
- let map s _ l = s :: l in
- let ma_files = Hashtbl.fold map baseuri_of [] in
- List.iter
- (fun ma_file ->
- let _ = if Filename.check_suffix ma_file ".mma" then
- let generated = Filename.chop_suffix ma_file ".mma" ^ ".ma" in
- Hashtbl.add include_deps generated ma_file;
- in
- let ma_baseuri = baseuri_of_script ma_file in
- let dependencies =
- let _ = print_times "prima deps_of_iter" in
- try DependenciesParser.deps_of_file ma_file
- with Sys_error _ -> []
- in
- let _ = print_times "dopo deps_of_iter" in
- let handle_uri uri =
- if not (Http_getter_storage.is_legacy uri) then
- let dep = resolve uri ma_baseuri in
- match dep with
- | None -> ()
- | Some u ->
- match script_of_baseuri ma_file u with
- | Some d -> Hashtbl.add include_deps ma_file d
- | None -> ()
- in
- let handle_script path =
- ignore (baseuri_of_script path);
- Hashtbl.add include_deps ma_file path
- in
- List.iter
- (function
- | DependenciesParser.UriDep uri ->
- let uri = NUri.string_of_uri uri in
- handle_uri uri
- | DependenciesParser.InlineDep path ->
- if Librarian.is_uri path
- then handle_uri path else handle_script path
- | DependenciesParser.IncludeDep path ->
- handle_script path)
- dependencies)
- ma_files;
- (* generate regular depend output *)
- let _ = print_times "dopo di iter2" in
- let deps =
- List.fold_left
- (fun acc ma_file ->
- let deps = Hashtbl.find_all include_deps ma_file in
- let deps = List.fast_sort Pervasives.compare deps in
- let deps = HExtlib.list_uniq deps in
- let deps = List.map fix_name deps in
- (fix_name ma_file, deps) :: acc)
- [] ma_files
- in
- let extern =
- List.fold_left
- (fun acc (_,d) ->
- List.fold_left
- (fun a x ->
- if List.exists (fun (t,_) -> x=t) deps then a
- else x::a)
- acc d)
- [] deps
- in
- let where = if !use_stdout then None else Some (Sys.getcwd()) in
- let all_deps =
- deps @
- HExtlib.list_uniq (List.sort Pervasives.compare (List.map (fun x -> x,[]) extern))
- in
- (* theory generation *)
- let all_deps_and_theory = generate_theory !theory_file all_deps in
- (* matita depend file generation *)
- Librarian.write_deps_file where all_deps_and_theory;
- (* dot generation *)
- if !dot_file <> "" then
- begin
- let oc = open_out !dot_file in
- let fmt = Format.formatter_of_out_channel oc in
- GraphvizPp.Dot.header fmt;
- List.iter
- (fun (ma_file,deps) ->
- GraphvizPp.Dot.node ma_file fmt;
- List.iter (fun dep -> GraphvizPp.Dot.edge ma_file dep fmt) deps)
- deps;
- List.iter
- (fun x -> GraphvizPp.Dot.node ~attrs:["style","dashed"] x fmt)
- extern;
- GraphvizPp.Dot.trailer fmt;
- close_out oc;
- ignore(Sys.command ("tred "^ !dot_file^"| dot -Tpng -o"^dot_name^".png"));
- HLog.message ("Type 'eog "^dot_name^".png' to view the graph");
- end;
- let _ = print_times "fine" in ()