src/Pure/Isar/expression.ML
author wenzelm
Wed Apr 27 20:58:40 2011 +0200 (2011-04-27)
changeset 42494 eef1a23c9077
parent 42482 42c7ef050bc3
child 43543 eb8b4851b039
permissions -rw-r--r--
tuned signature -- eliminated odd comment;
     1 (*  Title:      Pure/Isar/expression.ML
     2     Author:     Clemens Ballarin, TU Muenchen
     3 
     4 Locale expressions and user interface layer of locales.
     5 *)
     6 
     7 signature EXPRESSION =
     8 sig
     9   (* Locale expressions *)
    10   datatype 'term map = Positional of 'term option list | Named of (string * 'term) list
    11   type 'term expr = (string * ((string * bool) * 'term map)) list
    12   type expression_i = term expr * (binding * typ option * mixfix) list
    13   type expression = string expr * (binding * string option * mixfix) list
    14 
    15   (* Processing of context statements *)
    16   val cert_statement: Element.context_i list -> (term * term list) list list ->
    17     Proof.context -> (term * term list) list list * Proof.context
    18   val read_statement: Element.context list -> (string * string list) list list ->
    19     Proof.context -> (term * term list) list list * Proof.context
    20 
    21   (* Declaring locales *)
    22   val cert_declaration: expression_i -> (Proof.context -> Proof.context) -> Element.context_i list ->
    23     Proof.context -> (((string * typ) * mixfix) list * (string * morphism) list
    24       * Element.context_i list) * ((string * typ) list * Proof.context)
    25   val cert_read_declaration: expression_i -> (Proof.context -> Proof.context) -> Element.context list ->
    26     Proof.context -> (((string * typ) * mixfix) list * (string * morphism) list
    27       * Element.context_i list) * ((string * typ) list * Proof.context)
    28       (*FIXME*)
    29   val read_declaration: expression -> (Proof.context -> Proof.context) -> Element.context list ->
    30     Proof.context -> (((string * typ) * mixfix) list * (string * morphism) list
    31       * Element.context_i list) * ((string * typ) list * Proof.context)
    32   val add_locale: (local_theory -> local_theory) -> binding -> binding ->
    33     expression_i -> Element.context_i list -> theory -> string * local_theory
    34   val add_locale_cmd: (local_theory -> local_theory) -> binding -> binding ->
    35     expression -> Element.context list -> theory -> string * local_theory
    36 
    37   (* Interpretation *)
    38   val cert_goal_expression: expression_i -> Proof.context ->
    39     (term list list * (string * morphism) list * morphism) * Proof.context
    40   val read_goal_expression: expression -> Proof.context ->
    41     (term list list * (string * morphism) list * morphism) * Proof.context
    42   val sublocale: (local_theory -> local_theory) -> string -> expression_i ->
    43     (Attrib.binding * term) list -> theory -> Proof.state
    44   val sublocale_cmd: (local_theory -> local_theory) -> string -> expression ->
    45     (Attrib.binding * string) list -> theory -> Proof.state
    46   val interpretation: expression_i -> (Attrib.binding * term) list ->
    47     theory -> Proof.state
    48   val interpretation_cmd: expression -> (Attrib.binding * string) list ->
    49     theory -> Proof.state
    50   val interpret: expression_i -> (Attrib.binding * term) list ->
    51     bool -> Proof.state -> Proof.state
    52   val interpret_cmd: expression -> (Attrib.binding * string) list ->
    53     bool -> Proof.state -> Proof.state
    54 
    55   (* Diagnostic *)
    56   val print_dependencies: Proof.context -> bool -> expression -> unit
    57 end;
    58 
    59 structure Expression : EXPRESSION =
    60 struct
    61 
    62 datatype ctxt = datatype Element.ctxt;
    63 
    64 
    65 (*** Expressions ***)
    66 
    67 datatype 'term map =
    68   Positional of 'term option list |
    69   Named of (string * 'term) list;
    70 
    71 type 'term expr = (string * ((string * bool) * 'term map)) list;
    72 
    73 type expression = string expr * (binding * string option * mixfix) list;
    74 type expression_i = term expr * (binding * typ option * mixfix) list;
    75 
    76 
    77 (** Internalise locale names in expr **)
    78 
    79 fun intern thy instances =  map (apfst (Locale.intern thy)) instances;
    80 
    81 
    82 (** Parameters of expression **)
    83 
    84 (*Sanity check of instantiations and extraction of implicit parameters.
    85   The latter only occurs iff strict = false.
    86   Positional instantiations are extended to match full length of parameter list
    87   of instantiated locale.*)
    88 
    89 fun parameters_of thy strict (expr, fixed) =
    90   let
    91     fun reject_dups message xs =
    92       (case duplicates (op =) xs of
    93         [] => ()
    94       | dups => error (message ^ commas dups));
    95 
    96     fun parm_eq ((p1: string, mx1: mixfix), (p2, mx2)) = p1 = p2 andalso
    97       (mx1 = mx2 orelse error ("Conflicting syntax for parameter " ^ quote p1 ^ " in expression"));
    98 
    99     fun params_loc loc = Locale.params_of thy loc |> map (apfst #1);
   100     fun params_inst (loc, (prfx, Positional insts)) =
   101           let
   102             val ps = params_loc loc;
   103             val d = length ps - length insts;
   104             val insts' =
   105               if d < 0 then error ("More arguments than parameters in instantiation of locale " ^
   106                 quote (Locale.extern thy loc))
   107               else insts @ replicate d NONE;
   108             val ps' = (ps ~~ insts') |>
   109               map_filter (fn (p, NONE) => SOME p | (_, SOME _) => NONE);
   110           in (ps', (loc, (prfx, Positional insts'))) end
   111       | params_inst (loc, (prfx, Named insts)) =
   112           let
   113             val _ = reject_dups "Duplicate instantiation of the following parameter(s): "
   114               (map fst insts);
   115             val ps' = (insts, params_loc loc) |-> fold (fn (p, _) => fn ps =>
   116               if AList.defined (op =) ps p then AList.delete (op =) p ps
   117               else error (quote p ^ " not a parameter of instantiated expression"));
   118           in (ps', (loc, (prfx, Named insts))) end;
   119     fun params_expr is =
   120       let
   121         val (is', ps') = fold_map (fn i => fn ps =>
   122           let
   123             val (ps', i') = params_inst i;
   124             val ps'' = distinct parm_eq (ps @ ps');
   125           in (i', ps'') end) is []
   126       in (ps', is') end;
   127 
   128     val (implicit, expr') = params_expr expr;
   129 
   130     val implicit' = map #1 implicit;
   131     val fixed' = map (Variable.check_name o #1) fixed;
   132     val _ = reject_dups "Duplicate fixed parameter(s): " fixed';
   133     val implicit'' =
   134       if strict then []
   135       else
   136         let val _ = reject_dups
   137           "Parameter(s) declared simultaneously in expression and for clause: " (implicit' @ fixed')
   138         in map (fn (x, mx) => (Binding.name x, NONE, mx)) implicit end;
   139 
   140   in (expr', implicit'' @ fixed) end;
   141 
   142 
   143 (** Read instantiation **)
   144 
   145 (* Parse positional or named instantiation *)
   146 
   147 local
   148 
   149 fun prep_inst prep_term ctxt parms (Positional insts) =
   150       (insts ~~ parms) |> map (fn
   151         (NONE, p) => Free (p, dummyT) |
   152         (SOME t, _) => prep_term ctxt t)
   153   | prep_inst prep_term ctxt parms (Named insts) =
   154       parms |> map (fn p => case AList.lookup (op =) insts p of
   155         SOME t => prep_term ctxt t |
   156         NONE => Free (p, dummyT));
   157 
   158 in
   159 
   160 fun parse_inst x = prep_inst Syntax.parse_term x;
   161 fun make_inst x = prep_inst (K I) x;
   162 
   163 end;
   164 
   165 
   166 (* Instantiation morphism *)
   167 
   168 fun inst_morph (parm_names, parm_types) ((prfx, mandatory), insts') ctxt =
   169   let
   170     (* parameters *)
   171     val type_parm_names = fold Term.add_tfreesT parm_types [] |> map fst;
   172 
   173     (* type inference and contexts *)
   174     val parm_types' = map (Type_Infer.paramify_vars o Logic.varifyT_global) parm_types;
   175     val type_parms = fold Term.add_tvarsT parm_types' [] |> map (Logic.mk_type o TVar);
   176     val arg = type_parms @ map2 Type.constraint parm_types' insts';
   177     val res = Syntax.check_terms ctxt arg;
   178     val ctxt' = ctxt |> fold Variable.auto_fixes res;
   179 
   180     (* instantiation *)
   181     val (type_parms'', res') = chop (length type_parms) res;
   182     val insts'' = (parm_names ~~ res') |> map_filter
   183       (fn inst as (x, Free (y, _)) => if x = y then NONE else SOME inst
   184         | inst => SOME inst);
   185     val instT = Symtab.make (type_parm_names ~~ map Logic.dest_type type_parms'');
   186     val inst = Symtab.make insts'';
   187   in
   188     (Element.inst_morphism (Proof_Context.theory_of ctxt) (instT, inst) $>
   189       Morphism.binding_morphism (Binding.prefix mandatory prfx), ctxt')
   190   end;
   191 
   192 
   193 (*** Locale processing ***)
   194 
   195 (** Parsing **)
   196 
   197 fun parse_elem prep_typ prep_term ctxt =
   198   Element.map_ctxt
   199    {binding = I,
   200     typ = prep_typ ctxt,
   201     term = prep_term (Proof_Context.set_mode Proof_Context.mode_schematic ctxt),
   202     pattern = prep_term (Proof_Context.set_mode Proof_Context.mode_pattern ctxt),
   203     fact = I,
   204     attrib = I};
   205 
   206 fun parse_concl prep_term ctxt concl =
   207   (map o map) (fn (t, ps) =>
   208     (prep_term (Proof_Context.set_mode Proof_Context.mode_schematic ctxt) t,
   209       map (prep_term (Proof_Context.set_mode Proof_Context.mode_pattern ctxt)) ps)) concl;
   210 
   211 
   212 (** Simultaneous type inference: instantiations + elements + conclusion **)
   213 
   214 local
   215 
   216 fun mk_type T = (Logic.mk_type T, []);
   217 fun mk_term t = (t, []);
   218 fun mk_propp (p, pats) = (Type.constraint propT p, pats);
   219 
   220 fun dest_type (T, []) = Logic.dest_type T;
   221 fun dest_term (t, []) = t;
   222 fun dest_propp (p, pats) = (p, pats);
   223 
   224 fun extract_inst (_, (_, ts)) = map mk_term ts;
   225 fun restore_inst ((l, (p, _)), cs) = (l, (p, map dest_term cs));
   226 
   227 fun extract_elem (Fixes fixes) = map (#2 #> the_list #> map mk_type) fixes
   228   | extract_elem (Constrains csts) = map (#2 #> single #> map mk_type) csts
   229   | extract_elem (Assumes asms) = map (#2 #> map mk_propp) asms
   230   | extract_elem (Defines defs) = map (fn (_, (t, ps)) => [mk_propp (t, ps)]) defs
   231   | extract_elem (Notes _) = [];
   232 
   233 fun restore_elem (Fixes fixes, css) =
   234       (fixes ~~ css) |> map (fn ((x, _, mx), cs) =>
   235         (x, cs |> map dest_type |> try hd, mx)) |> Fixes
   236   | restore_elem (Constrains csts, css) =
   237       (csts ~~ css) |> map (fn ((x, _), cs) =>
   238         (x, cs |> map dest_type |> hd)) |> Constrains
   239   | restore_elem (Assumes asms, css) =
   240       (asms ~~ css) |> map (fn ((b, _), cs) => (b, map dest_propp cs)) |> Assumes
   241   | restore_elem (Defines defs, css) =
   242       (defs ~~ css) |> map (fn ((b, _), [c]) => (b, dest_propp c)) |> Defines
   243   | restore_elem (Notes notes, _) = Notes notes;
   244 
   245 fun check cs context =
   246   let
   247     fun prep (_, pats) (ctxt, t :: ts) =
   248       let val ctxt' = Variable.auto_fixes t ctxt
   249       in
   250         ((t, Syntax.check_props (Proof_Context.set_mode Proof_Context.mode_pattern ctxt') pats),
   251           (ctxt', ts))
   252       end;
   253     val (cs', (context', _)) = fold_map prep cs
   254       (context, Syntax.check_terms
   255         (Proof_Context.set_mode Proof_Context.mode_schematic context) (map fst cs));
   256   in (cs', context') end;
   257 
   258 in
   259 
   260 fun check_autofix insts elems concl ctxt =
   261   let
   262     val inst_cs = map extract_inst insts;
   263     val elem_css = map extract_elem elems;
   264     val concl_cs = (map o map) mk_propp concl;
   265     (* Type inference *)
   266     val (inst_cs' :: css', ctxt') =
   267       (fold_burrow o fold_burrow) check (inst_cs :: elem_css @ [concl_cs]) ctxt;
   268     val (elem_css', [concl_cs']) = chop (length elem_css) css';
   269   in
   270     (map restore_inst (insts ~~ inst_cs'),
   271       map restore_elem (elems ~~ elem_css'),
   272       concl_cs', ctxt')
   273   end;
   274 
   275 end;
   276 
   277 
   278 (** Prepare locale elements **)
   279 
   280 fun declare_elem prep_vars (Fixes fixes) ctxt =
   281       let val (vars, _) = prep_vars fixes ctxt
   282       in ctxt |> Proof_Context.add_fixes vars |> snd end
   283   | declare_elem prep_vars (Constrains csts) ctxt =
   284       ctxt |> prep_vars (map (fn (x, T) => (Binding.name x, SOME T, NoSyn)) csts) |> snd
   285   | declare_elem _ (Assumes _) ctxt = ctxt
   286   | declare_elem _ (Defines _) ctxt = ctxt
   287   | declare_elem _ (Notes _) ctxt = ctxt;
   288 
   289 
   290 (** Finish locale elements **)
   291 
   292 fun closeup _ _ false elem = elem
   293   | closeup ctxt parms true elem =
   294       let
   295         (* FIXME consider closing in syntactic phase -- before type checking *)
   296         fun close_frees t =
   297           let
   298             val rev_frees =
   299               Term.fold_aterms (fn Free (x, T) =>
   300                 if AList.defined (op =) parms x then I else insert (op =) (x, T) | _ => I) t [];
   301           in fold (Logic.all o Free) rev_frees t end;
   302 
   303         fun no_binds [] = []
   304           | no_binds _ = error "Illegal term bindings in context element";
   305       in
   306         (case elem of
   307           Assumes asms => Assumes (asms |> map (fn (a, propps) =>
   308             (a, map (fn (t, ps) => (close_frees t, no_binds ps)) propps)))
   309         | Defines defs => Defines (defs |> map (fn ((name, atts), (t, ps)) =>
   310             let val ((c, _), t') = Local_Defs.cert_def ctxt (close_frees t)
   311             in ((Thm.def_binding_optional (Binding.name c) name, atts), (t', no_binds ps)) end))
   312         | e => e)
   313       end;
   314 
   315 fun finish_primitive parms _ (Fixes fixes) = Fixes (map (fn (binding, _, mx) =>
   316       let val x = Binding.name_of binding
   317       in (binding, AList.lookup (op =) parms x, mx) end) fixes)
   318   | finish_primitive _ _ (Constrains _) = Constrains []
   319   | finish_primitive _ close (Assumes asms) = close (Assumes asms)
   320   | finish_primitive _ close (Defines defs) = close (Defines defs)
   321   | finish_primitive _ _ (Notes facts) = Notes facts;
   322 
   323 fun finish_inst ctxt (loc, (prfx, inst)) =
   324   let
   325     val thy = Proof_Context.theory_of ctxt;
   326     val (parm_names, parm_types) = Locale.params_of thy loc |> map #1 |> split_list;
   327     val (morph, _) = inst_morph (parm_names, parm_types) (prfx, inst) ctxt;
   328   in (loc, morph) end;
   329 
   330 fun finish_elem ctxt parms do_close elem =
   331   finish_primitive parms (closeup ctxt parms do_close) elem;
   332 
   333 fun finish ctxt parms do_close insts elems =
   334   let
   335     val deps = map (finish_inst ctxt) insts;
   336     val elems' = map (finish_elem ctxt parms do_close) elems;
   337   in (deps, elems') end;
   338 
   339 
   340 (** Process full context statement: instantiations + elements + conclusion **)
   341 
   342 (* Interleave incremental parsing and type inference over entire parsed stretch. *)
   343 
   344 local
   345 
   346 fun prep_full_context_statement parse_typ parse_prop prep_vars_elem prep_inst prep_vars_inst prep_expr
   347     {strict, do_close, fixed_frees} raw_import init_body raw_elems raw_concl ctxt1 =
   348   let
   349     val thy = Proof_Context.theory_of ctxt1;
   350 
   351     val (raw_insts, fixed) = parameters_of thy strict (apfst (prep_expr thy) raw_import);
   352 
   353     fun prep_insts_cumulative (loc, (prfx, inst)) (i, insts, ctxt) =
   354       let
   355         val (parm_names, parm_types) = Locale.params_of thy loc |> map #1 |> split_list;
   356         val inst' = prep_inst ctxt parm_names inst;
   357         val parm_types' = map (Type_Infer.paramify_vars o
   358           Term.map_type_tvar (fn ((x, _), S) => TVar ((x, i), S)) o Logic.varifyT_global) parm_types;
   359         val inst'' = map2 Type.constraint parm_types' inst';
   360         val insts' = insts @ [(loc, (prfx, inst''))];
   361         val (insts'', _, _, ctxt' (* FIXME not used *) ) = check_autofix insts' [] [] ctxt;
   362         val inst''' = insts'' |> List.last |> snd |> snd;
   363         val (morph, _) = inst_morph (parm_names, parm_types) (prfx, inst''') ctxt;
   364         val ctxt'' = Locale.activate_declarations (loc, morph) ctxt;
   365       in (i + 1, insts', ctxt'') end;
   366 
   367     fun prep_elem insts raw_elem (elems, ctxt) =
   368       let
   369         val ctxt' = declare_elem prep_vars_elem raw_elem ctxt;
   370         val elems' = elems @ [parse_elem parse_typ parse_prop ctxt' raw_elem];
   371         val (_, _, _, ctxt'' (* FIXME not used *) ) = check_autofix insts elems' [] ctxt';
   372       in (elems', ctxt') end;
   373 
   374     fun prep_concl raw_concl (insts, elems, ctxt) =
   375       let
   376         val concl = parse_concl parse_prop ctxt raw_concl;
   377       in check_autofix insts elems concl ctxt end;
   378 
   379     val fors = prep_vars_inst fixed ctxt1 |> fst;
   380     val ctxt2 = ctxt1 |> Proof_Context.add_fixes fors |> snd;
   381     val (_, insts', ctxt3) = fold prep_insts_cumulative raw_insts (0, [], ctxt2);
   382 
   383     val _ =
   384       if fixed_frees then ()
   385       else
   386         (case fold (fold (Variable.add_frees ctxt3) o snd o snd) insts' [] of
   387           [] => ()
   388         | frees => error ("Illegal free variables in expression: " ^
   389             commas_quote (map (Syntax.string_of_term ctxt3 o Free) (rev frees))));
   390 
   391     val ctxt4 = init_body ctxt3;
   392     val (elems, ctxt5) = fold (prep_elem insts') raw_elems ([], ctxt4);
   393     val (insts, elems', concl, ctxt6) = prep_concl raw_concl (insts', elems, ctxt5);
   394 
   395     (* Retrieve parameter types *)
   396     val xs = fold (fn Fixes fixes => (fn ps => ps @ map (Variable.check_name o #1) fixes)
   397       | _ => fn ps => ps) (Fixes fors :: elems') [];
   398     val (Ts, ctxt7) = fold_map Proof_Context.inferred_param xs ctxt6;
   399     val parms = xs ~~ Ts;  (* params from expression and elements *)
   400 
   401     val Fixes fors' = finish_primitive parms I (Fixes fors);
   402     val fixed = map (fn (b, SOME T, mx) => ((Binding.name_of b, T), mx)) fors';
   403     val (deps, elems'') = finish ctxt6 parms do_close insts elems';
   404 
   405   in ((fixed, deps, elems'', concl), (parms, ctxt7)) end
   406 
   407 in
   408 
   409 fun cert_full_context_statement x =
   410   prep_full_context_statement (K I) (K I) Proof_Context.cert_vars
   411   make_inst Proof_Context.cert_vars (K I) x;
   412 
   413 fun cert_read_full_context_statement x =
   414   prep_full_context_statement Syntax.parse_typ Syntax.parse_prop Proof_Context.read_vars
   415   make_inst Proof_Context.cert_vars (K I) x;
   416 
   417 fun read_full_context_statement x =
   418   prep_full_context_statement Syntax.parse_typ Syntax.parse_prop Proof_Context.read_vars
   419   parse_inst Proof_Context.read_vars intern x;
   420 
   421 end;
   422 
   423 
   424 (* Context statement: elements + conclusion *)
   425 
   426 local
   427 
   428 fun prep_statement prep activate raw_elems raw_concl context =
   429   let
   430      val ((_, _, elems, concl), _) =
   431        prep {strict = true, do_close = false, fixed_frees = true}
   432         ([], []) I raw_elems raw_concl context;
   433      val (_, context') = context |>
   434        Proof_Context.set_stmt true |>
   435        fold_map activate elems;
   436   in (concl, context') end;
   437 
   438 in
   439 
   440 fun cert_statement x = prep_statement cert_full_context_statement Element.activate_i x;
   441 fun read_statement x = prep_statement read_full_context_statement Element.activate x;
   442 
   443 end;
   444 
   445 
   446 (* Locale declaration: import + elements *)
   447 
   448 fun fix_params params =
   449   Proof_Context.add_fixes (map (fn ((x, T), mx) => (Binding.name x, SOME T, mx)) params) #> snd;
   450 
   451 local
   452 
   453 fun prep_declaration prep activate raw_import init_body raw_elems context =
   454   let
   455     val ((fixed, deps, elems, _), (parms, ctxt')) =
   456       prep {strict = false, do_close = true, fixed_frees = false}
   457         raw_import init_body raw_elems [] context;
   458     (* Declare parameters and imported facts *)
   459     val context' = context |>
   460       fix_params fixed |>
   461       fold (Context.proof_map o Locale.activate_facts NONE) deps;
   462     val (elems', _) = context' |>
   463       Proof_Context.set_stmt true |>
   464       fold_map activate elems;
   465   in ((fixed, deps, elems'), (parms, ctxt')) end;
   466 
   467 in
   468 
   469 fun cert_declaration x = prep_declaration cert_full_context_statement Element.activate_i x;
   470 fun cert_read_declaration x = prep_declaration cert_read_full_context_statement Element.activate x;
   471 fun read_declaration x = prep_declaration read_full_context_statement Element.activate x;
   472 
   473 end;
   474 
   475 
   476 (* Locale expression to set up a goal *)
   477 
   478 local
   479 
   480 fun props_of thy (name, morph) =
   481   let
   482     val (asm, defs) = Locale.specification_of thy name;
   483   in
   484     (case asm of NONE => defs | SOME asm => asm :: defs) |> map (Morphism.term morph)
   485   end;
   486 
   487 fun prep_goal_expression prep expression context =
   488   let
   489     val thy = Proof_Context.theory_of context;
   490 
   491     val ((fixed, deps, _, _), _) =
   492       prep {strict = true, do_close = true, fixed_frees = true} expression I [] [] context;
   493     (* proof obligations *)
   494     val propss = map (props_of thy) deps;
   495 
   496     val goal_ctxt = context |>
   497       fix_params fixed |>
   498       (fold o fold) Variable.auto_fixes propss;
   499 
   500     val export = Variable.export_morphism goal_ctxt context;
   501     val exp_fact = Drule.zero_var_indexes_list o map Thm.strip_shyps o Morphism.fact export;
   502     val exp_term = Term_Subst.zero_var_indexes o Morphism.term export;
   503     val exp_typ = Logic.type_map exp_term;
   504     val export' = Morphism.morphism {binding = I, typ = exp_typ, term = exp_term, fact = exp_fact};
   505   in ((propss, deps, export'), goal_ctxt) end;
   506 
   507 in
   508 
   509 fun cert_goal_expression x = prep_goal_expression cert_full_context_statement x;
   510 fun read_goal_expression x = prep_goal_expression read_full_context_statement x;
   511 
   512 end;
   513 
   514 
   515 (*** Locale declarations ***)
   516 
   517 (* extract specification text *)
   518 
   519 val norm_term = Envir.beta_norm oo Term.subst_atomic;
   520 
   521 fun bind_def ctxt eq (xs, env, eqs) =
   522   let
   523     val _ = Local_Defs.cert_def ctxt eq;
   524     val ((y, T), b) = Local_Defs.abs_def eq;
   525     val b' = norm_term env b;
   526     fun err msg = error (msg ^ ": " ^ quote y);
   527   in
   528     case filter (fn (Free (y', _), _) => y = y' | _ => false) env of
   529       [] => (Term.add_frees b' xs, (Free (y, T), b') :: env, eq :: eqs) |
   530       dups => if forall (fn (_, b'') => b' aconv b'') dups
   531         then (xs, env, eqs)
   532         else err "Attempt to redefine variable"
   533   end;
   534 
   535 (* text has the following structure:
   536        (((exts, exts'), (ints, ints')), (xs, env, defs))
   537    where
   538      exts: external assumptions (terms in assumes elements)
   539      exts': dito, normalised wrt. env
   540      ints: internal assumptions (terms in assumptions from insts)
   541      ints': dito, normalised wrt. env
   542      xs: the free variables in exts' and ints' and rhss of definitions,
   543        this includes parameters except defined parameters
   544      env: list of term pairs encoding substitutions, where the first term
   545        is a free variable; substitutions represent defines elements and
   546        the rhs is normalised wrt. the previous env
   547      defs: the equations from the defines elements
   548    *)
   549 
   550 fun eval_text _ _ (Fixes _) text = text
   551   | eval_text _ _ (Constrains _) text = text
   552   | eval_text _ is_ext (Assumes asms)
   553         (((exts, exts'), (ints, ints')), (xs, env, defs)) =
   554       let
   555         val ts = maps (map #1 o #2) asms;
   556         val ts' = map (norm_term env) ts;
   557         val spec' =
   558           if is_ext then ((exts @ ts, exts' @ ts'), (ints, ints'))
   559           else ((exts, exts'), (ints @ ts, ints' @ ts'));
   560       in (spec', (fold Term.add_frees ts' xs, env, defs)) end
   561   | eval_text ctxt _ (Defines defs) (spec, binds) =
   562       (spec, fold (bind_def ctxt o #1 o #2) defs binds)
   563   | eval_text _ _ (Notes _) text = text;
   564 
   565 fun eval_inst ctxt (loc, morph) text =
   566   let
   567     val thy = Proof_Context.theory_of ctxt;
   568     val (asm, defs) = Locale.specification_of thy loc;
   569     val asm' = Option.map (Morphism.term morph) asm;
   570     val defs' = map (Morphism.term morph) defs;
   571     val text' = text |>
   572       (if is_some asm
   573         then eval_text ctxt false (Assumes [(Attrib.empty_binding, [(the asm', [])])])
   574         else I) |>
   575       (if not (null defs)
   576         then eval_text ctxt false (Defines (map (fn def => (Attrib.empty_binding, (def, []))) defs'))
   577         else I)
   578 (* FIXME clone from locale.ML *)
   579   in text' end;
   580 
   581 fun eval_elem ctxt elem text =
   582   eval_text ctxt true elem text;
   583 
   584 fun eval ctxt deps elems =
   585   let
   586     val text' = fold (eval_inst ctxt) deps ((([], []), ([], [])), ([], [], []));
   587     val ((spec, (_, _, defs))) = fold (eval_elem ctxt) elems text';
   588   in (spec, defs) end;
   589 
   590 (* axiomsN: name of theorem set with destruct rules for locale predicates,
   591      also name suffix of delta predicates and assumptions. *)
   592 
   593 val axiomsN = "axioms";
   594 
   595 local
   596 
   597 (* introN: name of theorems for introduction rules of locale and
   598      delta predicates *)
   599 
   600 val introN = "intro";
   601 
   602 fun atomize_spec thy ts =
   603   let
   604     val t = Logic.mk_conjunction_balanced ts;
   605     val body = Object_Logic.atomize_term thy t;
   606     val bodyT = Term.fastype_of body;
   607   in
   608     if bodyT = propT then (t, propT, Thm.reflexive (Thm.cterm_of thy t))
   609     else (body, bodyT, Object_Logic.atomize (Thm.cterm_of thy t))
   610   end;
   611 
   612 (* achieve plain syntax for locale predicates (without "PROP") *)
   613 
   614 fun aprop_tr' n c = (Lexicon.mark_const c, fn ctxt => fn args =>
   615   if length args = n then
   616     Syntax.const "_aprop" $   (* FIXME avoid old-style early externing (!??) *)
   617       Term.list_comb (Syntax.free (Proof_Context.extern_const ctxt c), args)
   618   else raise Match);
   619 
   620 (* define one predicate including its intro rule and axioms
   621    - binding: predicate name
   622    - parms: locale parameters
   623    - defs: thms representing substitutions from defines elements
   624    - ts: terms representing locale assumptions (not normalised wrt. defs)
   625    - norm_ts: terms representing locale assumptions (normalised wrt. defs)
   626    - thy: the theory
   627 *)
   628 
   629 fun def_pred binding parms defs ts norm_ts thy =
   630   let
   631     val name = Sign.full_name thy binding;
   632 
   633     val (body, bodyT, body_eq) = atomize_spec thy norm_ts;
   634     val env = Term.add_free_names body [];
   635     val xs = filter (member (op =) env o #1) parms;
   636     val Ts = map #2 xs;
   637     val extraTs =
   638       (subtract (op =) (fold Term.add_tfreesT Ts []) (Term.add_tfrees body []))
   639       |> Library.sort_wrt #1 |> map TFree;
   640     val predT = map Term.itselfT extraTs ---> Ts ---> bodyT;
   641 
   642     val args = map Logic.mk_type extraTs @ map Free xs;
   643     val head = Term.list_comb (Const (name, predT), args);
   644     val statement = Object_Logic.ensure_propT thy head;
   645 
   646     val ([pred_def], defs_thy) =
   647       thy
   648       |> bodyT = propT ? Sign.add_advanced_trfuns ([], [], [aprop_tr' (length args) name], [])
   649       |> Sign.declare_const_global ((Binding.conceal binding, predT), NoSyn) |> snd
   650       |> Global_Theory.add_defs false
   651         [((Binding.conceal (Thm.def_binding binding), Logic.mk_equals (head, body)), [])];
   652     val defs_ctxt = Proof_Context.init_global defs_thy |> Variable.declare_term head;
   653 
   654     val cert = Thm.cterm_of defs_thy;
   655 
   656     val intro = Goal.prove_global defs_thy [] norm_ts statement (fn _ =>
   657       rewrite_goals_tac [pred_def] THEN
   658       Tactic.compose_tac (false, body_eq RS Drule.equal_elim_rule1, 1) 1 THEN
   659       Tactic.compose_tac (false,
   660         Conjunction.intr_balanced (map (Thm.assume o cert) norm_ts), 0) 1);
   661 
   662     val conjuncts =
   663       (Drule.equal_elim_rule2 OF [body_eq,
   664         Raw_Simplifier.rewrite_rule [pred_def] (Thm.assume (cert statement))])
   665       |> Conjunction.elim_balanced (length ts);
   666     val axioms = ts ~~ conjuncts |> map (fn (t, ax) =>
   667       Element.prove_witness defs_ctxt t
   668        (rewrite_goals_tac defs THEN
   669         Tactic.compose_tac (false, ax, 0) 1));
   670   in ((statement, intro, axioms), defs_thy) end;
   671 
   672 in
   673 
   674 (* main predicate definition function *)
   675 
   676 fun define_preds binding parms (((exts, exts'), (ints, ints')), defs) thy =
   677   let
   678     val defs' = map (cterm_of thy #> Assumption.assume #> Drule.abs_def) defs;
   679 
   680     val (a_pred, a_intro, a_axioms, thy'') =
   681       if null exts then (NONE, NONE, [], thy)
   682       else
   683         let
   684           val abinding = if null ints then binding else Binding.suffix_name ("_" ^ axiomsN) binding;
   685           val ((statement, intro, axioms), thy') =
   686             thy
   687             |> def_pred abinding parms defs' exts exts';
   688           val (_, thy'') =
   689             thy'
   690             |> Sign.qualified_path true abinding
   691             |> Global_Theory.note_thmss ""
   692               [((Binding.conceal (Binding.name introN), []), [([intro], [Locale.unfold_add])])]
   693             ||> Sign.restore_naming thy';
   694           in (SOME statement, SOME intro, axioms, thy'') end;
   695     val (b_pred, b_intro, b_axioms, thy'''') =
   696       if null ints then (NONE, NONE, [], thy'')
   697       else
   698         let
   699           val ((statement, intro, axioms), thy''') =
   700             thy''
   701             |> def_pred binding parms defs' (ints @ the_list a_pred) (ints' @ the_list a_pred);
   702           val (_, thy'''') =
   703             thy'''
   704             |> Sign.qualified_path true binding
   705             |> Global_Theory.note_thmss ""
   706                  [((Binding.conceal (Binding.name introN), []), [([intro], [Locale.intro_add])]),
   707                   ((Binding.conceal (Binding.name axiomsN), []),
   708                     [(map (Drule.export_without_context o Element.conclude_witness) axioms, [])])]
   709             ||> Sign.restore_naming thy''';
   710         in (SOME statement, SOME intro, axioms, thy'''') end;
   711   in ((a_pred, a_intro, a_axioms), (b_pred, b_intro, b_axioms), thy'''') end;
   712 
   713 end;
   714 
   715 
   716 local
   717 
   718 fun assumes_to_notes (Assumes asms) axms =
   719       fold_map (fn (a, spec) => fn axs =>
   720           let val (ps, qs) = chop (length spec) axs
   721           in ((a, [(ps, [])]), qs) end) asms axms
   722       |> apfst (curry Notes "")
   723   | assumes_to_notes e axms = (e, axms);
   724 
   725 fun defines_to_notes thy (Defines defs) =
   726       Notes ("", map (fn (a, (def, _)) =>
   727         (a, [([Assumption.assume (cterm_of thy def)],
   728           [(Attrib.internal o K) Locale.witness_add])])) defs)
   729   | defines_to_notes _ e = e;
   730 
   731 fun gen_add_locale prep_decl
   732     before_exit binding raw_predicate_binding raw_import raw_body thy =
   733   let
   734     val name = Sign.full_name thy binding;
   735     val _ = Locale.defined thy name andalso
   736       error ("Duplicate definition of locale " ^ quote name);
   737 
   738     val ((fixed, deps, body_elems), (parms, ctxt')) =
   739       prep_decl raw_import I raw_body (Proof_Context.init_global thy);
   740     val text as (((_, exts'), _), defs) = eval ctxt' deps body_elems;
   741 
   742     val extraTs =
   743       subtract (op =) (fold Term.add_tfreesT (map snd parms) []) (fold Term.add_tfrees exts' []);
   744     val _ =
   745       if null extraTs then ()
   746       else warning ("Additional type variable(s) in locale specification " ^
   747           Binding.print binding ^ ": " ^
   748           commas (map (Syntax.string_of_typ ctxt' o TFree) (sort_wrt #1 extraTs)));
   749 
   750     val predicate_binding =
   751       if Binding.is_empty raw_predicate_binding then binding
   752       else raw_predicate_binding;
   753     val ((a_statement, a_intro, a_axioms), (b_statement, b_intro, b_axioms), thy') =
   754       define_preds predicate_binding parms text thy;
   755 
   756     val a_satisfy = Element.satisfy_morphism a_axioms;
   757     val b_satisfy = Element.satisfy_morphism b_axioms;
   758 
   759     val params = fixed @
   760       maps (fn Fixes fixes =>
   761         map (fn (b, SOME T, mx) => ((Binding.name_of b, T), mx)) fixes | _ => []) body_elems;
   762     val asm = if is_some b_statement then b_statement else a_statement;
   763 
   764     val notes =
   765       if is_some asm then
   766         [("", [((Binding.conceal (Binding.suffix_name ("_" ^ axiomsN) binding), []),
   767           [([Assumption.assume (cterm_of thy' (the asm))],
   768             [(Attrib.internal o K) Locale.witness_add])])])]
   769       else [];
   770 
   771     val notes' = body_elems |>
   772       map (defines_to_notes thy') |>
   773       map (Element.morph_ctxt a_satisfy) |>
   774       (fn elems => fold_map assumes_to_notes elems (map Element.conclude_witness a_axioms)) |>
   775       fst |>
   776       map (Element.morph_ctxt b_satisfy) |>
   777       map_filter (fn Notes notes => SOME notes | _ => NONE);
   778 
   779     val deps' = map (fn (l, morph) => (l, morph $> b_satisfy)) deps;
   780     val axioms = map Element.conclude_witness b_axioms;
   781 
   782     val loc_ctxt = thy'
   783       |> Locale.register_locale binding (extraTs, params)
   784           (asm, rev defs) (a_intro, b_intro) axioms [] (rev notes) (rev deps')
   785       |> Named_Target.init before_exit name
   786       |> fold (fn (kind, facts) => Local_Theory.notes_kind kind facts #> snd) notes';
   787 
   788   in (name, loc_ctxt) end;
   789 
   790 in
   791 
   792 val add_locale = gen_add_locale cert_declaration;
   793 val add_locale_cmd = gen_add_locale read_declaration;
   794 
   795 end;
   796 
   797 
   798 (*** Interpretation ***)
   799 
   800 (** Interpretation in theories and proof contexts **)
   801 
   802 local
   803 
   804 fun note_eqns_register deps witss attrss eqns export export' context =
   805   let
   806     fun meta_rewrite context =
   807       map (Local_Defs.meta_rewrite_rule (Context.proof_of context) #> Drule.abs_def) o
   808         maps snd;
   809   in
   810     context
   811     |> Element.generic_note_thmss Thm.lemmaK
   812       (attrss ~~ map (fn eqn => [([Morphism.thm (export' $> export) eqn], [])]) eqns)
   813     |-> (fn facts => `(fn context => meta_rewrite context facts))
   814     |-> (fn eqns => fold (fn ((dep, morph), wits) =>
   815       fn context =>
   816         Locale.add_registration (dep, morph $> Element.satisfy_morphism
   817             (map (Element.morph_witness export') wits))
   818           (Element.eq_morphism (Context.theory_of context) eqns |>
   819             Option.map (rpair true))
   820           export context) (deps ~~ witss))
   821   end;
   822 
   823 fun gen_interpretation prep_expr parse_prop prep_attr
   824     expression equations theory =
   825   let
   826     val ((propss, deps, export), expr_ctxt) = Proof_Context.init_global theory
   827       |> prep_expr expression;
   828 
   829     val eqns = map (parse_prop expr_ctxt o snd) equations |> Syntax.check_terms expr_ctxt;
   830     val attrss = map ((apsnd o map) (prep_attr theory) o fst) equations;
   831     val goal_ctxt = fold Variable.auto_fixes eqns expr_ctxt;
   832     val export' = Variable.export_morphism goal_ctxt expr_ctxt;
   833 
   834     fun after_qed witss eqns =
   835       (Proof_Context.background_theory o Context.theory_map)
   836         (note_eqns_register deps witss attrss eqns export export');
   837 
   838   in Element.witness_proof_eqs after_qed propss eqns goal_ctxt end;
   839 
   840 fun gen_interpret prep_expr parse_prop prep_attr
   841     expression equations int state =
   842   let
   843     val _ = Proof.assert_forward_or_chain state;
   844     val ctxt = Proof.context_of state;
   845     val theory = Proof_Context.theory_of ctxt;
   846 
   847     val ((propss, deps, export), expr_ctxt) = prep_expr expression ctxt;
   848 
   849     val eqns = map (parse_prop expr_ctxt o snd) equations |> Syntax.check_terms expr_ctxt;
   850     val attrss = map ((apsnd o map) (prep_attr theory) o fst) equations;
   851     val goal_ctxt = fold Variable.auto_fixes eqns expr_ctxt;
   852     val export' = Variable.export_morphism goal_ctxt expr_ctxt;
   853 
   854     fun after_qed witss eqns = (Proof.map_context o Context.proof_map)
   855       (note_eqns_register deps witss attrss eqns export export');
   856   in
   857     state
   858     |> Element.witness_local_proof_eqs after_qed "interpret" propss eqns goal_ctxt int
   859   end;
   860 
   861 in
   862 
   863 fun interpretation x = gen_interpretation cert_goal_expression (K I) (K I) x;
   864 fun interpretation_cmd x = gen_interpretation read_goal_expression
   865   Syntax.parse_prop Attrib.intern_src x;
   866 
   867 fun interpret x = gen_interpret cert_goal_expression (K I) (K I) x;
   868 fun interpret_cmd x = gen_interpret read_goal_expression
   869   Syntax.parse_prop Attrib.intern_src x;
   870 
   871 end;
   872 
   873 
   874 (** Interpretation between locales: declaring sublocale relationships **)
   875 
   876 local
   877 
   878 fun note_eqns_dependency target deps witss attrss eqns export export' ctxt =
   879   let
   880     fun meta_rewrite ctxt =
   881       map (Local_Defs.meta_rewrite_rule ctxt #> Drule.abs_def) o maps snd;
   882     val facts =
   883       (attrss ~~ map (fn eqn => [([Morphism.thm (export' $> export) eqn], [])]) eqns);
   884     val eqns' = ctxt |> Context.Proof
   885       |> Element.generic_note_thmss Thm.lemmaK facts
   886       |> apsnd Context.the_proof  (* FIXME Context.proof_map_result *)
   887       |-> (fn facts => `(fn ctxt => meta_rewrite ctxt facts))
   888       |> fst;  (* FIXME duplication to add_thmss *)
   889   in
   890     ctxt
   891     |> Locale.add_thmss target Thm.lemmaK facts
   892     |> Proof_Context.background_theory (fold (fn ((dep, morph), wits) =>
   893       fn theory =>
   894         Locale.add_dependency target
   895           (dep, morph $> Element.satisfy_morphism
   896             (map (Element.morph_witness export') wits))
   897           (Element.eq_morphism theory eqns' |> Option.map (rpair true))
   898           export theory) (deps ~~ witss))
   899   end;
   900 
   901 fun gen_sublocale prep_expr intern parse_prop prep_attr
   902     before_exit raw_target expression equations thy =
   903   let
   904     val target = intern thy raw_target;
   905     val target_ctxt = Named_Target.init before_exit target thy;
   906     val ((propss, deps, export), expr_ctxt) = prep_expr expression target_ctxt;
   907 
   908     val eqns = map (parse_prop expr_ctxt o snd) equations |> Syntax.check_terms expr_ctxt;
   909     val attrss = map ((apsnd o map) (prep_attr thy) o fst) equations;
   910     val goal_ctxt = fold Variable.auto_fixes eqns expr_ctxt;
   911     val export' = Variable.export_morphism goal_ctxt expr_ctxt;
   912 
   913     fun after_qed witss eqns =
   914       note_eqns_dependency target deps witss attrss eqns export export';
   915 
   916   in Element.witness_proof_eqs after_qed propss eqns goal_ctxt end;
   917 in
   918 
   919 fun sublocale x = gen_sublocale cert_goal_expression (K I) (K I) (K I) x;
   920 fun sublocale_cmd x =
   921   gen_sublocale read_goal_expression Locale.intern Syntax.parse_prop Attrib.intern_src x;
   922 
   923 end;
   924 
   925 
   926 (** Print the instances that would be activated by an interpretation
   927   of the expression in the current context (clean = false) or in an
   928   empty context (clean = true). **)
   929 
   930 fun print_dependencies ctxt clean expression =
   931   let
   932     val ((_, deps, export), expr_ctxt) = read_goal_expression expression ctxt;
   933   in
   934     Locale.print_dependencies expr_ctxt clean export deps
   935   end;
   936 
   937 end;
   938