X-Git-Url: http://matita.cs.unibo.it/gitweb/?a=blobdiff_plain;f=helm%2Focaml%2Fcic_notation%2FcicNotationRew.ml;h=15947f3f238b00d0f9ad875c221f04e3a2e0127a;hb=34113d572c334c351ba66f4b05db503eed4d48f2;hp=151ec8a08516cf8074fe6aad4483a662be46d0ff;hpb=ec54d490477ece51c19d79750dda9805ffda663c;p=helm.git diff --git a/helm/ocaml/cic_notation/cicNotationRew.ml b/helm/ocaml/cic_notation/cicNotationRew.ml index 151ec8a08..15947f3f2 100644 --- a/helm/ocaml/cic_notation/cicNotationRew.ml +++ b/helm/ocaml/cic_notation/cicNotationRew.ml @@ -29,13 +29,14 @@ type pattern_id = int type interpretation_id = pattern_id type pretty_printer_id = pattern_id +let default_prec = 50 +let default_assoc = Gramext.NonA + type term_info = { sort: (Cic.id, CicNotationPt.sort_kind) Hashtbl.t; uri: (Cic.id, string) Hashtbl.t; } -let warning s = prerr_endline ("CicNotation WARNING: " ^ s) - let get_types uri = let o,_ = CicEnvironment.get_obj CicUniv.empty_ugraph uri in match o with @@ -62,6 +63,7 @@ let constructor_of_inductive_type uri i j = with Not_found -> assert false) module Ast = CicNotationPt +module Parser = CicNotationParser let string_of_name = function | Cic.Name s -> s @@ -71,17 +73,60 @@ let ident_of_name n = Ast.Ident (string_of_name n, None) let idref id t = Ast.AttributedTerm (`IdRef id, t) +let resolve_binder = function + | `Lambda -> "\\lambda" + | `Pi -> "\\Pi" + | `Forall -> "\\forall" + | `Exists -> "\\exists" + +let binder_attributes = [None, "mathcolor", "blue"] +let atop_attributes = [None, "linethickness", "0pt"] +let indent_attributes = [None, "indent", "1em"] +let keyword_attributes = [None, "mathcolor", "blue"] + let pp_ast0 t k = - prerr_endline "pp_ast0"; - let rec aux t = CicNotationUtil.visit_ast ~special_k k t + let reset_href t = Ast.AttributedTerm (`Href [], t) in + let builtin_symbol s = reset_href (Ast.Literal (`Symbol s)) in + let binder_symbol s = + Ast.AttributedTerm (`XmlAttrs binder_attributes, builtin_symbol s) + in + let rec aux = function + | Ast.Appl ts -> + Ast.AttributedTerm (`Level (Parser.apply_prec, Parser.apply_assoc), + Ast.Layout + (Ast.Box ((Ast.HOV, true, true), + (CicNotationUtil.dress + (Ast.Layout Ast.Break) + (List.map k ts))))) + | Ast.Binder (`Forall, (Ast.Ident ("_", _), ty), body) + | Ast.Binder (`Pi, (Ast.Ident ("_", _), ty), body) -> + Ast.AttributedTerm (`Level (Parser.binder_prec, Parser.binder_assoc), + Ast.Layout (Ast.Box ((Ast.HV, false, true), [ + aux_ty ty; + Ast.Layout Ast.Break; + binder_symbol "\\to"; + k body]))) + | Ast.Binder (binder_kind, (id, ty), body) -> + Ast.AttributedTerm (`Level (Parser.binder_prec, Parser.binder_assoc), + Ast.Layout (Ast.Box ((Ast.HV, false, true), [ + binder_symbol (resolve_binder binder_kind); + k id; + builtin_symbol ":"; + aux_ty ty; + Ast.Layout Ast.Break; + builtin_symbol "."; + k body ]))) + | t -> CicNotationUtil.visit_ast ~special_k k t + and aux_ty = function + | None -> builtin_symbol "?" + | Some ty -> k ty and special_k = function - | Ast.AttributedTerm (attrs, t) -> Ast.AttributedTerm (attrs, aux t) + | Ast.AttributedTerm (attrs, t) -> Ast.AttributedTerm (attrs, k t) | _ -> assert false in aux t let ast_of_acic0 term_info acic k = -(* prerr_endline "ast_of_acic0"; *) let k = k term_info in let register_uri id uri = Hashtbl.add term_info.uri id uri in let sort_of_id id = @@ -199,12 +244,8 @@ let ast_of_acic0 term_info acic k = let level1_patterns21 = Hashtbl.create 211 let level2_patterns32 = Hashtbl.create 211 -let (compiled21: (CicNotationPt.term -> (CicNotationEnv.t * int) option) -option ref) = - ref None -let (compiled32: (Cic.annterm -> ((string * Cic.annterm) list * int) option) -option ref) = - ref None +let compiled21 = ref None +let compiled32 = ref None let pattern21_matrix = ref [] let pattern32_matrix = ref [] @@ -212,19 +253,18 @@ let pattern32_matrix = ref [] let get_compiled21 () = match !compiled21 with | None -> assert false - | Some f -> f + | Some f -> Lazy.force f let get_compiled32 () = match !compiled32 with | None -> assert false - | Some f -> f + | Some f -> Lazy.force f let set_compiled21 f = compiled21 := Some f let set_compiled32 f = compiled32 := Some f -let instantiate21 env precedence associativity l1 = - prerr_endline "instantiate21"; +let instantiate21 env (* precedence associativity *) l1 = let rec subst_singleton env t = - CicNotationUtil.boxify (subst env t) + CicNotationUtil.group (subst env t) and subst env = function | Ast.AttributedTerm (_, t) -> subst env t | Ast.Variable var -> @@ -240,6 +280,8 @@ let instantiate21 env precedence associativity l1 = assert (CicNotationEnv.well_typed expected_ty value); [ CicNotationEnv.term_of_value value ] | Ast.Magic m -> subst_magic env m + | Ast.Literal (`Keyword k) as t -> + [ Ast.AttributedTerm (`XmlAttrs keyword_attributes, t) ] | Ast.Literal _ as t -> [ t ] | Ast.Layout l -> [ Ast.Layout (subst_layout env l) ] | t -> [ CicNotationUtil.visit_ast (subst_singleton env) t ] @@ -253,19 +295,17 @@ let instantiate21 env precedence associativity l1 = let values = CicNotationUtil.ncombine rec_values in let sep = match sep_opt with - | None -> [] - | Some l -> [ CicNotationPt.Literal l ] - in + | None -> [] + | Some l -> [ CicNotationPt.Literal l ] + in let rec instantiate_list acc = function | [] -> List.rev acc | value_set :: [] -> let env = CicNotationEnv.combine rec_decls value_set in - instantiate_list - ((CicNotationUtil.boxify (subst env p)) :: acc) [] + instantiate_list (CicNotationUtil.group (subst env p) :: acc) [] | value_set :: tl -> let env = CicNotationEnv.combine rec_decls value_set in - instantiate_list - ((CicNotationUtil.boxify (subst env p @ sep)) :: acc) tl + instantiate_list (CicNotationUtil.group ((subst env p) @ sep) :: acc) tl in instantiate_list [] values | Ast.Opt p -> @@ -288,7 +328,8 @@ let instantiate21 env precedence associativity l1 = end | _ -> assert false (* impossible *) and subst_layout env = function - | Ast.Box (kind, tl) -> Ast.Box (kind, List.concat (List.map (subst env) tl)) + | Ast.Box (kind, tl) -> + Ast.Box (kind, List.concat (List.map (subst env) tl)) | l -> CicNotationUtil.visit_layout (subst_singleton env) l in subst_singleton env l1 @@ -297,6 +338,7 @@ let rec pp_ast1 term = let rec pp_value = function | CicNotationEnv.NumValue _ as v -> v | CicNotationEnv.StringValue _ as v -> v +(* | CicNotationEnv.TermValue t when t == term -> CicNotationEnv.TermValue (pp_ast0 t pp_ast1) *) | CicNotationEnv.TermValue t -> CicNotationEnv.TermValue (pp_ast1 t) | CicNotationEnv.OptValue None as v -> v | CicNotationEnv.OptValue (Some v) -> @@ -307,19 +349,25 @@ let rec pp_ast1 term = let ast_env_of_env env = List.map (fun (var, (ty, value)) -> (var, (ty, pp_value value))) env in - match (get_compiled21 ()) term with - | None -> pp_ast0 term pp_ast1 - | Some (env, pid) -> - let precedence, associativity, l1 = - try - Hashtbl.find level1_patterns21 pid - with Not_found -> assert false - in - instantiate21 (ast_env_of_env env) precedence associativity l1 + match term with + | Ast.AttributedTerm (attrs, t) -> Ast.AttributedTerm (attrs, pp_ast1 t) + | _ -> + begin + match (get_compiled21 ()) term with + | None -> pp_ast0 term pp_ast1 + | Some (env, pid) -> + let precedence, associativity, l1 = + try + Hashtbl.find level1_patterns21 pid + with Not_found -> assert false + in + Ast.AttributedTerm (`Level (precedence, associativity), + (instantiate21 (ast_env_of_env env) l1)) + end let instantiate32 term_info env symbol args = let rec instantiate_arg = function - | Ast.IdentArg (n, name) -> + | GrafiteAst.IdentArg (n, name) -> let t = (try List.assoc name env with Not_found -> assert false) in let rec count_lambda = function | Ast.Binder (`Lambda, _, body) -> 1 + count_lambda body @@ -345,18 +393,21 @@ let rec ast_of_acic1 term_info annterm = let env' = List.map (fun (name, term) -> (name, ast_of_acic1 term_info term)) env in - let symbol, args = + let symbol, args, uris = try Hashtbl.find level2_patterns32 pid with Not_found -> assert false in - instantiate32 term_info env' symbol args + let ast = instantiate32 term_info env' symbol args in + match uris with + | [] -> ast + | _ -> Ast.AttributedTerm (`Href uris, ast) let load_patterns32 t = - set_compiled32 (CicNotationMatcher.Matcher32.compiler t) + set_compiled32 (lazy (CicNotationMatcher.Matcher32.compiler t)) let load_patterns21 t = - set_compiled21 (CicNotationMatcher.Matcher21.compiler t) + set_compiled21 (lazy (CicNotationMatcher.Matcher21.compiler t)) let ast_of_acic id_to_sort annterm = let term_info = { sort = id_to_sort; uri = Hashtbl.create 211 } in @@ -373,12 +424,15 @@ let fresh_id = let add_interpretation (symbol, args) appl_pattern = let id = fresh_id () in - Hashtbl.add level2_patterns32 id (symbol, args); + let uris = CicNotationUtil.find_appl_pattern_uris appl_pattern in + Hashtbl.add level2_patterns32 id (symbol, args, uris); pattern32_matrix := (appl_pattern, id) :: !pattern32_matrix; load_patterns32 !pattern32_matrix; id -let add_pretty_printer ?precedence ?associativity l2 l1 = +let add_pretty_printer + ?(precedence = default_prec) ?(associativity = default_assoc) l2 l1 += let id = fresh_id () in let l2' = CicNotationUtil.strip_attributes l2 in Hashtbl.add level1_patterns21 id (precedence, associativity, l1);