helm/ocaml/cic_notation/cicNotation.ml

   1 (* Copyright (C) 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 GrafiteAst
  27
  28 type notation_id =
  29   | RuleId of CicNotationParser.rule_id
  30   | InterpretationId of CicNotationRew.interpretation_id
  31   | PrettyPrinterId of CicNotationRew.pretty_printer_id
  32
  33 let process_notation st =
  34   match st with
  35   | Notation (loc, dir, l1, associativity, precedence, l2) ->
  36       let rule_id =
  37         if dir <> Some `RightToLeft then
  38           [ RuleId (CicNotationParser.extend l1 ?precedence ?associativity
  39               (fun env loc -> CicNotationFwd.instantiate_level2 env l2)) ]
  40         else
  41           []
  42       in
  43       let pp_id =
  44         if dir <> Some `LeftToRight then
  45           [ PrettyPrinterId
  46               (CicNotationRew.add_pretty_printer ?precedence ?associativity
  47                 l2 l1) ]
  48         else
  49           []
  50       in
  51       st, rule_id @ pp_id
  52   | Interpretation (loc, dsc, l2, l3) ->
  53       let interp_id = CicNotationRew.add_interpretation dsc l2 l3 in
  54       st, [ InterpretationId interp_id ]
  55   | st -> st, []
  56
  57 let remove_notation = function
  58   | RuleId id -> CicNotationParser.delete id
  59   | PrettyPrinterId id -> CicNotationRew.remove_pretty_printer id
  60   | InterpretationId id -> CicNotationRew.remove_interpretation id
  61
  62 let load_notation fname =
  63   let ic = open_in fname in
  64   let lexbuf = Ulexing.from_utf8_channel ic in
  65   try
  66     while true do
  67       match GrafiteParser.parse_statement lexbuf with
  68       | Executable (_, Command (_, cmd)) -> ignore (process_notation cmd)
  69       | _ -> ()
  70     done
  71   with End_of_file -> close_in ic
  72
  73 let get_all_notations () =
  74   List.map
  75     (fun (interp_id, dsc) ->
  76       InterpretationId interp_id, "interpretation: " ^ dsc)
  77     (CicNotationRew.get_all_interpretations ())
  78
  79 let get_active_notations () =
  80   List.map (fun id -> InterpretationId id)
  81     (CicNotationRew.get_active_interpretations ())
  82
  83 let set_active_notations ids =
  84   let interp_ids =
  85     HExtlib.filter_map
  86       (function InterpretationId interp_id -> Some interp_id | _ -> None)
  87       ids
  88   in
  89   CicNotationRew.set_active_interpretations interp_ids
  90