(* 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 GA = GrafiteAst module U = UriManager 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 let exclude excluded_files files = let map file = not (List.mem (fix_name file) excluded_files) in List.filter map files let main () = (* let _ = print_times "inizio" in *) let include_paths = ref [] in let excluded_files = ref [] in let use_stdout = ref false 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_excluded_file name = excluded_files := name :: !excluded_files in (* helpers *) let 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; 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 = U.buri_of_uri (U.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"; "-exclude", Arg.String set_excluded_file, "Exclude a file from the dependences"; "-stdout", Arg.Set use_stdout, "Print dependences on stdout" ]; 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 files = fun () -> match cmdline_args with | [] -> HExtlib.find ~test:(fun x -> Filename.check_suffix x ".ma") "." | _ -> 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 args = exclude !excluded_files args 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 *) List.iter (fun ma_file -> let ma_baseuri = baseuri_of_script ma_file in let dependencies = DependenciesParser.deps_of_file ma_file in List.iter (function | DependenciesParser.UriDep uri -> let uri = UriManager.string_of_uri uri in 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 -> ()) | DependenciesParser.IncludeDep path -> ignore (baseuri_of_script path); Hashtbl.add include_deps ma_file 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 Librarian.write_deps_file where (deps@HExtlib.list_uniq (List.sort Pervasives.compare (List.map (fun x -> x,[]) extern))); (* 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 () *)