helm/software/lambda-delta/toplevel/top.ml

   1 (*
   2     ||M||  This file is part of HELM, an Hypertextual, Electronic
   3     ||A||  Library of Mathematics, developed at the Computer Science
   4     ||T||  Department, University of Bologna, Italy.
   5     ||I||
   6     ||T||  HELM is free software; you can redistribute it and/or
   7     ||A||  modify it under the terms of the GNU General Public License
   8     \   /  version 2 or (at your option) any later version.
   9      \ /   This software is distributed as is, NO WARRANTY.
  10       V_______________________________________________________________ *)
  11
  12 module P    = Printf
  13 module U    = NUri
  14 module C    = Cps
  15 module L    = Log
  16 module H    = Hierarchy
  17 module AO   = AutOutput
  18 module MA   = MetaAut
  19 module MO   = MetaOutput
  20 module MBag = MetaBag
  21 module BagO = BagOutput
  22 module BagR = BagReduction
  23 module BagU = BagUntrusted
  24
  25 type status = {
  26    mst : MA.status;
  27    ac  : AO.counters;
  28    mc  : MO.counters;
  29    bagc: BagO.counters
  30 }
  31
  32 let initial_status = {
  33    ac   = AO.initial_counters;
  34    mc   = MO.initial_counters;
  35    bagc = BagO.initial_counters;
  36    mst  = MA.initial_status
  37 }
  38
  39 let count count_fun c item =
  40    if !L.level > 2 then count_fun C.start c item else c
  41
  42 let flush () = L.flush 0; L.flush_err ()
  43
  44 let bag_error s msg =
  45    L.error BagO.specs (L.Warn s :: L.Loc :: msg); flush ()
  46
  47 let main =
  48 try
  49    let version_string = "Helena Checker 0.8.0 M - December 2008" in
  50    let stage = ref 3 in
  51    let meta_file = ref None in
  52    let set_hierarchy s =
  53       let f = function
  54          | Some g -> H.graph := g
  55          | None   -> L.warn (P.sprintf "Unknown type hierarchy: %s" s)
  56       in
  57       H.graph_of_string f s
  58    in
  59    let set_summary i = L.level := i in
  60    let print_version () = L.warn version_string; exit 0 in
  61    let set_stage i = stage := i in
  62    let close = function
  63       | None          -> ()
  64       | Some (och, _) -> close_out och
  65    in
  66    let set_meta_file name =
  67       close !meta_file;
  68       let och = open_out name in
  69       let frm = Format.formatter_of_out_channel och in
  70       Format.pp_set_margin frm max_int;
  71       meta_file := Some (och, frm)
  72    in
  73    let read_file name =
  74       if !L.level > 0 then Time.gmtime version_string;
  75       if !L.level > 1 then
  76          L.warn (P.sprintf "Processing file: %s" name);
  77       if !L.level > 0 then Time.utime_stamp "started";
  78       let ich = open_in name in
  79       let lexbuf = Lexing.from_channel ich in
  80       let book = AutParser.book AutLexer.token lexbuf in
  81       close_in ich;
  82       if !L.level > 0 then Time.utime_stamp "parsed";
  83       let rec aux st = function
  84          | []         -> st
  85          | item :: tl ->
  86 (* stage 3 *)
  87             let f st _ = function
  88                | None           -> st
  89                | Some (i, u, _) ->
  90                   Log.warn (P.sprintf "[%u] %s" i (U.string_of_uri u)); st
  91             in
  92 (* stage 2 *)
  93             let f st item =
  94                let st = {st with bagc = count BagO.count_item st.bagc item} in
  95                if !stage > 2 then BagU.type_check (f st) !H.graph item else st
  96             in
  97 (* stage 1 *)
  98             let f mst item =
  99                let st = {st with
 100                   mst = mst; mc = count MO.count_item st.mc item
 101                } in
 102                begin match !meta_file with
 103                   | None          -> ()
 104                   | Some (_, frm) -> MO.pp_item C.start frm item
 105                end;
 106                if !stage > 1 then MBag.bag_of_meta (f st) item else st
 107             in
 108 (* stage 0 *)
 109             let st = {st with ac = count AO.count_item st.ac item} in
 110             let st =
 111                if !stage > 0 then MA.meta_of_aut f st.mst item else st
 112             in
 113             aux st tl
 114       in
 115       let st = aux initial_status book in
 116       if !L.level > 0 then Time.utime_stamp "processed";
 117       if !L.level > 2 then AO.print_counters C.start st.ac;
 118       if !L.level > 2 && !stage > 0 then MO.print_counters C.start st.mc;
 119       if !L.level > 2 && !stage > 1 then BagO.print_counters C.start st.bagc;
 120    in
 121    let help =
 122       "Usage: helena [ -Vin | -Ss <number> | -m <file> | -h <string> ] <file> ...\n\n" ^
 123       "Summary levels: 0 just errors, 1 time stamps, 2 processed file names, \
 124        3 data information, 4 typing information, 5 reduction information\n\n" ^
 125        "Stages: 0 parsing, 1 to intermediate, 2 to untrusted, 3 to trusted\n"
 126    in
 127    let help_S = "<number>  set summary level" in
 128    let help_V = " show version information" in
 129    let help_h = "<string>  set type hierarchy" in
 130    let help_i = " show local references by index" in
 131    let help_m = "<file>  output intermediate representation" in
 132    let help_n = " activate naive sort inclusion" in
 133    let help_s = "<number>  Set translation stage" in
 134    L.box 0; L.box_err ();
 135    H.set_new_sorts ignore ["Set"; "Prop"];
 136    Arg.parse [
 137       ("-S", Arg.Int set_summary, help_S);
 138       ("-V", Arg.Unit print_version, help_V);
 139       ("-h", Arg.String set_hierarchy, help_h);
 140       ("-i", Arg.Set BagO.indexes, help_i);
 141       ("-m", Arg.String set_meta_file, help_m);
 142       ("-n", Arg.Set BagR.nsi, help_n);
 143       ("-s", Arg.Int set_stage, help_s)
 144    ] read_file help;
 145    if !L.level > 0 then Time.utime_stamp "at exit";
 146    flush ()
 147 with BagType.TypeError msg -> bag_error "Type Error" msg