src/HOL/Statespace/state_space.ML
author wenzelm
Wed Mar 14 20:34:20 2012 +0100 (2012-03-14)
changeset 46925 98ffc3fe31cc
parent 45741 088256c289e7
child 46949 94aa7b81bcf6
permissions -rw-r--r--
locale expressions without source positions;
haftmann@28308
     1
(*  Title:      HOL/Statespace/state_space.ML
schirmer@25171
     2
    Author:     Norbert Schirmer, TU Muenchen
schirmer@25171
     3
*)
schirmer@25171
     4
schirmer@25408
     5
signature STATE_SPACE =
wenzelm@45362
     6
sig
wenzelm@45362
     7
  val distinct_compsN : string
wenzelm@45362
     8
  val getN : string
wenzelm@45362
     9
  val putN : string
wenzelm@45362
    10
  val injectN : string
wenzelm@45362
    11
  val namespaceN : string
wenzelm@45362
    12
  val projectN : string
wenzelm@45362
    13
  val valuetypesN : string
schirmer@25408
    14
wenzelm@45362
    15
  val namespace_definition :
wenzelm@45362
    16
     bstring ->
wenzelm@45362
    17
     typ ->
wenzelm@46925
    18
     (xstring, string) Expression.expr * (binding * string option * mixfix) list ->
wenzelm@45362
    19
     string list -> string list -> theory -> theory
schirmer@25408
    20
wenzelm@45362
    21
  val define_statespace :
wenzelm@45362
    22
     string list ->
wenzelm@45362
    23
     string ->
wenzelm@45362
    24
     (string list * bstring * (string * string) list) list ->
wenzelm@45362
    25
     (string * string) list -> theory -> theory
wenzelm@45362
    26
  val define_statespace_i :
wenzelm@45362
    27
     string option ->
wenzelm@45362
    28
     string list ->
wenzelm@45362
    29
     string ->
wenzelm@45362
    30
     (typ list * bstring * (string * string) list) list ->
wenzelm@45362
    31
     (string * typ) list -> theory -> theory
schirmer@25408
    32
wenzelm@45362
    33
  val statespace_decl :
wenzelm@45362
    34
     ((string list * bstring) *
wenzelm@45362
    35
       ((string list * xstring * (bstring * bstring) list) list *
wenzelm@45362
    36
        (bstring * string) list)) parser
schirmer@25408
    37
schirmer@25408
    38
wenzelm@45362
    39
  val neq_x_y : Proof.context -> term -> term -> thm option
wenzelm@45362
    40
  val distinctNameSolver : Simplifier.solver
wenzelm@45362
    41
  val distinctTree_tac : Proof.context -> int -> tactic
wenzelm@45362
    42
  val distinct_simproc : Simplifier.simproc
schirmer@25408
    43
schirmer@25408
    44
wenzelm@45362
    45
  val get_comp : Context.generic -> string -> (typ * string) Option.option
wenzelm@45362
    46
  val get_silent : Context.generic -> bool
wenzelm@45362
    47
  val set_silent : bool -> Context.generic -> Context.generic
schirmer@25408
    48
wenzelm@45362
    49
  val gen_lookup_tr : Proof.context -> term -> string -> term
wenzelm@45362
    50
  val lookup_swap_tr : Proof.context -> term list -> term
wenzelm@45362
    51
  val lookup_tr : Proof.context -> term list -> term
wenzelm@45362
    52
  val lookup_tr' : Proof.context -> term list -> term
schirmer@25408
    53
wenzelm@45362
    54
  val gen_update_tr :
wenzelm@45362
    55
     bool -> Proof.context -> string -> term -> term -> term
wenzelm@45362
    56
  val update_tr : Proof.context -> term list -> term
wenzelm@45362
    57
  val update_tr' : Proof.context -> term list -> term
wenzelm@45362
    58
end;
schirmer@25408
    59
norbert@29247
    60
structure StateSpace : STATE_SPACE =
schirmer@25171
    61
struct
schirmer@25171
    62
wenzelm@45362
    63
(* Names *)
schirmer@25171
    64
schirmer@25171
    65
val distinct_compsN = "distinct_names"
schirmer@25171
    66
val namespaceN = "_namespace"
schirmer@25171
    67
val valuetypesN = "_valuetypes"
schirmer@25171
    68
val projectN = "project"
schirmer@25171
    69
val injectN = "inject"
schirmer@25171
    70
val getN = "get"
schirmer@25171
    71
val putN = "put"
schirmer@25171
    72
val project_injectL = "StateSpaceLocale.project_inject";
schirmer@25171
    73
schirmer@25171
    74
schirmer@25171
    75
(* Library *)
schirmer@25171
    76
schirmer@25171
    77
fun fold1 f xs = fold f (tl xs) (hd xs)
schirmer@25171
    78
fun fold1' f [] x = x
schirmer@25171
    79
  | fold1' f xs _ = fold1 f xs
wenzelm@26478
    80
wenzelm@26478
    81
fun sublist_idx eq xs ys =
schirmer@25171
    82
  let
wenzelm@26478
    83
    fun sublist n xs ys =
schirmer@25171
    84
         if is_prefix eq xs ys then SOME n
schirmer@25171
    85
         else (case ys of [] => NONE
schirmer@25171
    86
               | (y::ys') => sublist (n+1) xs ys')
schirmer@25171
    87
  in sublist 0 xs ys end;
schirmer@25171
    88
schirmer@25171
    89
fun is_sublist eq xs ys = is_some (sublist_idx eq xs ys);
schirmer@25171
    90
schirmer@25171
    91
fun sorted_subset eq [] ys = true
schirmer@25171
    92
  | sorted_subset eq (x::xs) [] = false
schirmer@25171
    93
  | sorted_subset eq (x::xs) (y::ys) = if eq (x,y) then sorted_subset eq xs ys
schirmer@25171
    94
                                       else sorted_subset eq (x::xs) ys;
schirmer@25171
    95
schirmer@25171
    96
schirmer@25171
    97
schirmer@25171
    98
type namespace_info =
schirmer@25171
    99
 {declinfo: (typ*string) Termtab.table, (* type, name of statespace *)
schirmer@25171
   100
  distinctthm: thm Symtab.table,
schirmer@25171
   101
  silent: bool
schirmer@25171
   102
 };
wenzelm@26478
   103
wenzelm@33519
   104
structure NameSpaceData = Generic_Data
wenzelm@33519
   105
(
wenzelm@26478
   106
  type T = namespace_info;
schirmer@25171
   107
  val empty = {declinfo = Termtab.empty, distinctthm = Symtab.empty, silent = false};
schirmer@25171
   108
  val extend = I;
wenzelm@33519
   109
  fun merge
wenzelm@33519
   110
    ({declinfo=declinfo1, distinctthm=distinctthm1, silent=silent1},
wenzelm@33519
   111
      {declinfo=declinfo2, distinctthm=distinctthm2, silent=silent2}) : T =
wenzelm@33519
   112
    {declinfo = Termtab.merge (K true) (declinfo1, declinfo2),
wenzelm@33519
   113
     distinctthm = Symtab.merge (K true) (distinctthm1, distinctthm2),
wenzelm@41472
   114
     silent = silent1 andalso silent2 (* FIXME odd merge *)}
wenzelm@33519
   115
);
schirmer@25171
   116
schirmer@25171
   117
fun make_namespace_data declinfo distinctthm silent =
schirmer@25171
   118
     {declinfo=declinfo,distinctthm=distinctthm,silent=silent};
schirmer@25171
   119
schirmer@25171
   120
schirmer@25171
   121
fun delete_declinfo n ctxt =
schirmer@25171
   122
  let val {declinfo,distinctthm,silent} = NameSpaceData.get ctxt;
wenzelm@26478
   123
  in NameSpaceData.put
schirmer@25171
   124
       (make_namespace_data (Termtab.delete_safe n declinfo) distinctthm silent) ctxt
schirmer@25171
   125
  end;
schirmer@25171
   126
schirmer@25171
   127
schirmer@25171
   128
fun update_declinfo (n,v) ctxt =
schirmer@25171
   129
  let val {declinfo,distinctthm,silent} = NameSpaceData.get ctxt;
wenzelm@26478
   130
  in NameSpaceData.put
schirmer@25171
   131
      (make_namespace_data (Termtab.update (n,v) declinfo) distinctthm silent) ctxt
schirmer@25171
   132
  end;
schirmer@25171
   133
schirmer@25171
   134
fun set_silent silent ctxt =
schirmer@25171
   135
  let val {declinfo,distinctthm,...} = NameSpaceData.get ctxt;
wenzelm@26478
   136
  in NameSpaceData.put
schirmer@25171
   137
      (make_namespace_data declinfo distinctthm silent) ctxt
schirmer@25171
   138
  end;
schirmer@25171
   139
schirmer@25171
   140
val get_silent = #silent o NameSpaceData.get;
wenzelm@26478
   141
wenzelm@46925
   142
fun expression_no_pos (expr, fixes) : Expression.expression =
wenzelm@46925
   143
  (map (fn (name, inst) => ((name, Position.none), inst)) expr, fixes);
wenzelm@46925
   144
schirmer@25171
   145
fun prove_interpretation_in ctxt_tac (name, expr) thy =
schirmer@25171
   146
   thy
wenzelm@46925
   147
   |> Expression.sublocale_cmd I (name, Position.none) (expression_no_pos expr) []
wenzelm@26478
   148
   |> Proof.global_terminal_proof
wenzelm@32194
   149
         (Method.Basic (fn ctxt => SIMPLE_METHOD (ctxt_tac ctxt)), NONE)
wenzelm@42361
   150
   |> Proof_Context.theory_of
schirmer@25171
   151
norbert@29247
   152
fun add_locale name expr elems thy =
wenzelm@45362
   153
  thy
wenzelm@41585
   154
  |> Expression.add_locale I (Binding.name name) (Binding.name name) expr elems
haftmann@29362
   155
  |> snd
wenzelm@33671
   156
  |> Local_Theory.exit;
norbert@29247
   157
norbert@29247
   158
fun add_locale_cmd name expr elems thy =
wenzelm@45362
   159
  thy
wenzelm@46925
   160
  |> Expression.add_locale_cmd I (Binding.name name) Binding.empty (expression_no_pos expr) elems
haftmann@29362
   161
  |> snd
wenzelm@33671
   162
  |> Local_Theory.exit;
norbert@29247
   163
schirmer@25171
   164
type statespace_info =
schirmer@25171
   165
 {args: (string * sort) list, (* type arguments *)
schirmer@25171
   166
  parents: (typ list * string * string option list) list,
schirmer@25171
   167
             (* type instantiation, state-space name, component renamings *)
schirmer@25171
   168
  components: (string * typ) list,
schirmer@25171
   169
  types: typ list (* range types of state space *)
schirmer@25171
   170
 };
schirmer@25171
   171
wenzelm@33519
   172
structure StateSpaceData = Generic_Data
wenzelm@33519
   173
(
schirmer@25171
   174
  type T = statespace_info Symtab.table;
schirmer@25171
   175
  val empty = Symtab.empty;
schirmer@25171
   176
  val extend = I;
wenzelm@33519
   177
  fun merge data : T = Symtab.merge (K true) data;
wenzelm@33519
   178
);
schirmer@25171
   179
schirmer@25171
   180
fun add_statespace name args parents components types ctxt =
wenzelm@26478
   181
     StateSpaceData.put
schirmer@25171
   182
      (Symtab.update_new (name, {args=args,parents=parents,
schirmer@25171
   183
                                components=components,types=types}) (StateSpaceData.get ctxt))
wenzelm@26478
   184
      ctxt;
schirmer@25171
   185
schirmer@25171
   186
fun get_statespace ctxt name =
schirmer@25171
   187
      Symtab.lookup (StateSpaceData.get ctxt) name;
schirmer@25171
   188
schirmer@25171
   189
schirmer@25171
   190
fun mk_free ctxt name =
wenzelm@26478
   191
  if Variable.is_fixed ctxt name orelse Variable.is_declared ctxt name
schirmer@25171
   192
  then
wenzelm@42488
   193
    let val n' = Variable.intern_fixed ctxt name
wenzelm@42488
   194
    in SOME (Free (n', Proof_Context.infer_type ctxt (n', dummyT))) end
schirmer@25171
   195
  else NONE
wenzelm@26478
   196
wenzelm@26478
   197
schirmer@25171
   198
fun get_dist_thm ctxt name = Symtab.lookup (#distinctthm (NameSpaceData.get ctxt)) name;
wenzelm@26478
   199
fun get_comp ctxt name =
wenzelm@26478
   200
     Option.mapPartial
wenzelm@26478
   201
       (Termtab.lookup (#declinfo (NameSpaceData.get ctxt)))
schirmer@25171
   202
       (mk_free (Context.proof_of ctxt) name);
schirmer@25171
   203
schirmer@25171
   204
schirmer@25171
   205
(*** Tactics ***)
schirmer@25171
   206
schirmer@25171
   207
fun neq_x_y ctxt x y =
schirmer@25171
   208
  (let
wenzelm@26478
   209
    val dist_thm = the (get_dist_thm (Context.Proof ctxt) (#1 (dest_Free x)));
schirmer@25171
   210
    val ctree = cprop_of dist_thm |> Thm.dest_comb |> #2 |> Thm.dest_comb |> #2;
schirmer@25171
   211
    val tree = term_of ctree;
schirmer@25171
   212
    val x_path = the (DistinctTreeProver.find_tree x tree);
schirmer@25171
   213
    val y_path = the (DistinctTreeProver.find_tree y tree);
schirmer@25171
   214
    val thm = DistinctTreeProver.distinctTreeProver dist_thm x_path y_path;
wenzelm@26478
   215
  in SOME thm
wenzelm@45361
   216
  end handle Option.Option => NONE)
schirmer@25171
   217
wenzelm@42368
   218
fun distinctTree_tac ctxt = SUBGOAL (fn (goal, i) =>
wenzelm@42368
   219
  (case goal of
wenzelm@42368
   220
    Const (@{const_name Trueprop}, _) $
wenzelm@42368
   221
      (Const (@{const_name Not}, _) $
wenzelm@42368
   222
        (Const (@{const_name HOL.eq}, _) $ (x as Free _) $ (y as Free _))) =>
wenzelm@42368
   223
      (case neq_x_y ctxt x y of
wenzelm@42368
   224
        SOME neq => rtac neq i
wenzelm@42368
   225
      | NONE => no_tac)
wenzelm@42368
   226
  | _ => no_tac));
schirmer@25171
   227
wenzelm@45362
   228
val distinctNameSolver =
wenzelm@45362
   229
  mk_solver "distinctNameSolver" (distinctTree_tac o Simplifier.the_context);
schirmer@25171
   230
schirmer@25171
   231
val distinct_simproc =
wenzelm@38715
   232
  Simplifier.simproc_global @{theory HOL} "StateSpace.distinct_simproc" ["x = y"]
haftmann@38864
   233
    (fn thy => fn ss => (fn (Const (@{const_name HOL.eq},_)$(x as Free _)$(y as Free _)) =>
wenzelm@30289
   234
        (case try Simplifier.the_context ss of
wenzelm@45362
   235
          SOME ctxt =>
wenzelm@45362
   236
            Option.map (fn neq => DistinctTreeProver.neq_to_eq_False OF [neq])
wenzelm@45362
   237
              (neq_x_y ctxt x y)
schirmer@25171
   238
        | NONE => NONE)
wenzelm@45362
   239
      | _ => NONE));
schirmer@25171
   240
schirmer@25171
   241
local
wenzelm@26478
   242
  val ss = HOL_basic_ss
wenzelm@26478
   243
in
wenzelm@26478
   244
fun interprete_parent name dist_thm_name parent_expr thy =
schirmer@25171
   245
  let
wenzelm@42368
   246
    fun solve_tac ctxt = CSUBGOAL (fn (goal, i) =>
schirmer@25171
   247
      let
wenzelm@42361
   248
        val distinct_thm = Proof_Context.get_thm ctxt dist_thm_name;
schirmer@25171
   249
        val rule = DistinctTreeProver.distinct_implProver distinct_thm goal;
wenzelm@42368
   250
      in rtac rule i end);
wenzelm@26478
   251
wenzelm@42368
   252
    fun tac ctxt =
wenzelm@42368
   253
      Locale.intro_locales_tac true ctxt [] THEN ALLGOALS (solve_tac ctxt);
wenzelm@26478
   254
wenzelm@45362
   255
  in
wenzelm@45362
   256
    thy |> prove_interpretation_in tac (name, parent_expr)
wenzelm@26478
   257
  end;
schirmer@25171
   258
schirmer@25171
   259
end;
schirmer@25171
   260
wenzelm@26478
   261
fun namespace_definition name nameT parent_expr parent_comps new_comps thy =
schirmer@25171
   262
  let
schirmer@25171
   263
    val all_comps = parent_comps @ new_comps;
norbert@29247
   264
    val vars = (map (fn n => (Binding.name n, NONE, NoSyn)) all_comps);
haftmann@28965
   265
    val full_name = Sign.full_bname thy name;
schirmer@25171
   266
    val dist_thm_name = distinct_compsN;
schirmer@25171
   267
norbert@29247
   268
    val dist_thm_full_name = dist_thm_name;
wenzelm@26478
   269
    fun comps_of_thm thm = prop_of thm
schirmer@25171
   270
             |> (fn (_$(_$t)) => DistinctTreeProver.dest_tree t) |> map (fst o dest_Free);
wenzelm@26478
   271
wenzelm@38835
   272
    fun type_attr phi = Thm.declaration_attribute (fn thm => fn context =>
wenzelm@38835
   273
      (case context of
wenzelm@38835
   274
        Context.Theory _ => context
wenzelm@38835
   275
      | Context.Proof ctxt =>
wenzelm@26478
   276
        let
wenzelm@38835
   277
          val {declinfo,distinctthm=tt,silent} = NameSpaceData.get context;
wenzelm@32960
   278
          val all_names = comps_of_thm thm;
schirmer@25171
   279
          fun upd name tt =
wenzelm@38835
   280
               (case Symtab.lookup tt name of
schirmer@25171
   281
                 SOME dthm => if sorted_subset (op =) (comps_of_thm dthm) all_names
schirmer@25171
   282
                              then Symtab.update (name,thm) tt else tt
wenzelm@38835
   283
               | NONE => Symtab.update (name,thm) tt)
schirmer@25171
   284
schirmer@25171
   285
          val tt' = tt |> fold upd all_names;
schirmer@25171
   286
          val activate_simproc =
wenzelm@38835
   287
            Simplifier.map_ss
wenzelm@38835
   288
              (Simplifier.with_context (Context_Position.set_visible false ctxt)
wenzelm@38835
   289
                (fn ss => ss addsimprocs [distinct_simproc]));
wenzelm@38835
   290
          val context' =
wenzelm@38835
   291
              context
schirmer@25171
   292
              |> NameSpaceData.put {declinfo=declinfo,distinctthm=tt',silent=silent}
wenzelm@38835
   293
              |> activate_simproc;
wenzelm@38835
   294
        in context' end));
wenzelm@26478
   295
schirmer@25171
   296
    val attr = Attrib.internal type_attr;
schirmer@25171
   297
wenzelm@45362
   298
    val assume =
wenzelm@45362
   299
      ((Binding.name dist_thm_name, [attr]),
wenzelm@45362
   300
        [(HOLogic.Trueprop $
wenzelm@45362
   301
          (Const (@{const_name all_distinct}, Type (@{type_name tree}, [nameT]) --> HOLogic.boolT) $
wenzelm@45362
   302
            DistinctTreeProver.mk_tree (fn n => Free (n, nameT)) nameT
wenzelm@45362
   303
              (sort fast_string_ord all_comps)), [])]);
wenzelm@45362
   304
  in
wenzelm@45362
   305
    thy
wenzelm@45362
   306
    |> add_locale name ([], vars) [Element.Assumes [assume]]
wenzelm@45362
   307
    |> Proof_Context.theory_of
wenzelm@45362
   308
    |> interprete_parent name dist_thm_full_name parent_expr
schirmer@25171
   309
  end;
schirmer@25171
   310
hoelzl@32651
   311
fun encode_dot x = if x= #"." then #"_" else x;
schirmer@25171
   312
schirmer@25171
   313
fun encode_type (TFree (s, _)) = s
schirmer@25171
   314
  | encode_type (TVar ((s,i),_)) = "?" ^ s ^ string_of_int i
wenzelm@26478
   315
  | encode_type (Type (n,Ts)) =
schirmer@25171
   316
      let
schirmer@25171
   317
        val Ts' = fold1' (fn x => fn y => x ^ "_" ^ y) (map encode_type Ts) "";
hoelzl@32651
   318
        val n' = String.map encode_dot n;
schirmer@25171
   319
      in if Ts'="" then n' else Ts' ^ "_" ^ n' end;
schirmer@25171
   320
schirmer@25171
   321
fun project_name T = projectN ^"_"^encode_type T;
schirmer@25171
   322
fun inject_name T = injectN ^"_"^encode_type T;
schirmer@25171
   323
schirmer@25171
   324
fun project_free T pT V = Free (project_name T, V --> pT);
schirmer@25171
   325
fun inject_free T pT V = Free (inject_name T, pT --> V);
schirmer@25171
   326
schirmer@25171
   327
fun get_name n = getN ^ "_" ^ n;
schirmer@25171
   328
fun put_name n = putN ^ "_" ^ n;
schirmer@25171
   329
fun get_const n T nT V = Free (get_name n, (nT --> V) --> T);
schirmer@25171
   330
fun put_const n T nT V = Free (put_name n, T --> (nT --> V) --> (nT --> V));
schirmer@25171
   331
wenzelm@45362
   332
fun lookup_const T nT V =
wenzelm@45362
   333
  Const (@{const_name StateFun.lookup}, (V --> T) --> nT --> (nT --> V) --> T);
wenzelm@45362
   334
wenzelm@26478
   335
fun update_const T nT V =
wenzelm@45362
   336
  Const (@{const_name StateFun.update},
wenzelm@45362
   337
    (V --> T) --> (T --> V) --> nT --> (T --> T) --> (nT --> V) --> (nT --> V));
schirmer@25171
   338
wenzelm@45362
   339
fun K_const T = Const (@{const_name K_statefun}, T --> T --> T);
schirmer@25171
   340
schirmer@25171
   341
schirmer@25171
   342
fun add_declaration name decl thy =
schirmer@25171
   343
  thy
wenzelm@41585
   344
  |> Named_Target.init I name
wenzelm@45291
   345
  |> (fn lthy => Local_Theory.declaration {syntax = false, pervasive = false} (decl lthy) lthy)
wenzelm@33671
   346
  |> Local_Theory.exit_global;
schirmer@25171
   347
schirmer@25171
   348
fun parent_components thy (Ts, pname, renaming) =
schirmer@25171
   349
  let
schirmer@25171
   350
    val ctxt = Context.Theory thy;
schirmer@25171
   351
    fun rename [] xs = xs
schirmer@25171
   352
      | rename (NONE::rs)  (x::xs) = x::rename rs xs
schirmer@25171
   353
      | rename (SOME r::rs) ((x,T)::xs) = (r,T)::rename rs xs;
wenzelm@45362
   354
    val {args, parents, components, ...} = the (Symtab.lookup (StateSpaceData.get ctxt) pname);
schirmer@25171
   355
    val inst = map fst args ~~ Ts;
schirmer@25171
   356
    val subst = Term.map_type_tfree (the o AList.lookup (op =) inst o fst);
wenzelm@26478
   357
    val parent_comps =
wenzelm@32952
   358
      maps (fn (Ts',n,rs) => parent_components thy (map subst Ts',n,rs)) parents;
schirmer@25171
   359
    val all_comps = rename renaming (parent_comps @ map (apsnd subst) components);
schirmer@25171
   360
  in all_comps end;
schirmer@25171
   361
wenzelm@43278
   362
fun take_upto i xs = List.take(xs,i) handle General.Subscript => xs;
schirmer@25171
   363
schirmer@25171
   364
fun statespace_definition state_type args name parents parent_comps components thy =
wenzelm@26478
   365
  let
haftmann@28965
   366
    val full_name = Sign.full_bname thy name;
schirmer@25171
   367
    val all_comps = parent_comps @ components;
schirmer@25171
   368
schirmer@25171
   369
    val components' = map (fn (n,T) => (n,(T,full_name))) components;
wenzelm@26478
   370
    val all_comps' = map (fn (n,T) => (n,(T,full_name))) all_comps;
schirmer@25171
   371
norbert@29247
   372
    fun parent_expr (_,n,rs) = (suffix namespaceN n,((n,false),Expression.Positional rs));
norbert@29247
   373
        (* FIXME: a more specific renaming-prefix (including parameter names) may be nicer *)
norbert@29247
   374
    val parents_expr = map parent_expr parents;
schirmer@25171
   375
    fun distinct_types Ts =
wenzelm@27276
   376
      let val tab = fold (fn T => fn tab => Typtab.update (T,()) tab) Ts Typtab.empty;
wenzelm@27276
   377
      in map fst (Typtab.dest tab) end;
schirmer@25171
   378
schirmer@25171
   379
    val Ts = distinct_types (map snd all_comps);
schirmer@25171
   380
    val arg_names = map fst args;
wenzelm@43324
   381
    val valueN = singleton (Name.variant_list arg_names) "'value";
wenzelm@43324
   382
    val nameN = singleton (Name.variant_list (valueN :: arg_names)) "'name";
schirmer@25171
   383
    val valueT = TFree (valueN, Sign.defaultS thy);
schirmer@25171
   384
    val nameT = TFree (nameN, Sign.defaultS thy);
schirmer@25171
   385
    val stateT = nameT --> valueT;
schirmer@25171
   386
    fun projectT T = valueT --> T;
wenzelm@26478
   387
    fun injectT T = T --> valueT;
norbert@29247
   388
    val locinsts = map (fn T => (project_injectL,
wenzelm@45362
   389
                    (("",false),Expression.Positional
wenzelm@45362
   390
                             [SOME (Free (project_name T,projectT T)),
norbert@29247
   391
                              SOME (Free ((inject_name T,injectT T)))]))) Ts;
wenzelm@32952
   392
    val locs = maps (fn T => [(Binding.name (project_name T),NONE,NoSyn),
wenzelm@32952
   393
                                     (Binding.name (inject_name T),NONE,NoSyn)]) Ts;
wenzelm@32952
   394
    val constrains = maps (fn T => [(project_name T,projectT T),(inject_name T,injectT T)]) Ts;
schirmer@25171
   395
norbert@29247
   396
    fun interprete_parent_valuetypes (Ts, pname, _) thy =
schirmer@25171
   397
      let
wenzelm@26478
   398
        val {args,types,...} =
schirmer@25171
   399
             the (Symtab.lookup (StateSpaceData.get (Context.Theory thy)) pname);
schirmer@25171
   400
        val inst = map fst args ~~ Ts;
schirmer@25171
   401
        val subst = Term.map_type_tfree (the o AList.lookup (op =) inst o fst);
wenzelm@32952
   402
        val pars = maps ((fn T => [project_name T,inject_name T]) o subst) types;
norbert@29247
   403
wenzelm@45362
   404
        val expr = ([(suffix valuetypesN name,
norbert@29247
   405
                     (("",false),Expression.Positional (map SOME pars)))],[]);
wenzelm@29291
   406
      in
wenzelm@30473
   407
        prove_interpretation_in (ALLGOALS o solve_tac o Assumption.all_prems_of)
wenzelm@29291
   408
          (suffix valuetypesN name, expr) thy
wenzelm@29291
   409
      end;
schirmer@25171
   410
schirmer@25171
   411
    fun interprete_parent (_, pname, rs) =
schirmer@25171
   412
      let
norbert@29247
   413
        val expr = ([(pname, (("",false), Expression.Positional rs))],[])
wenzelm@26478
   414
      in prove_interpretation_in
haftmann@29360
   415
           (fn ctxt => Locale.intro_locales_tac false ctxt [])
schirmer@25171
   416
           (full_name, expr) end;
schirmer@25171
   417
schirmer@25171
   418
    fun declare_declinfo updates lthy phi ctxt =
schirmer@25171
   419
      let
wenzelm@26478
   420
        fun upd_prf ctxt =
schirmer@25171
   421
          let
schirmer@25171
   422
            fun upd (n,v) =
schirmer@25171
   423
              let
wenzelm@42361
   424
                val nT = Proof_Context.infer_type (Local_Theory.target_of lthy) (n, dummyT)
wenzelm@26478
   425
              in Context.proof_map
wenzelm@26478
   426
                  (update_declinfo (Morphism.term phi (Free (n,nT)),v))
schirmer@25171
   427
              end;
schirmer@25171
   428
          in ctxt |> fold upd updates end;
schirmer@25171
   429
schirmer@25171
   430
      in Context.mapping I upd_prf ctxt end;
schirmer@25171
   431
wenzelm@26478
   432
   fun string_of_typ T =
wenzelm@39134
   433
      Print_Mode.setmp []
wenzelm@39134
   434
        (Syntax.string_of_typ (Config.put show_sorts true (Syntax.init_pretty_global thy))) T;
schirmer@25171
   435
   val fixestate = (case state_type of
schirmer@25171
   436
         NONE => []
wenzelm@26478
   437
       | SOME s =>
wenzelm@26478
   438
          let
wenzelm@32960
   439
            val fx = Element.Fixes [(Binding.name s,SOME (string_of_typ stateT),NoSyn)];
wenzelm@26478
   440
            val cs = Element.Constrains
schirmer@25171
   441
                       (map (fn (n,T) =>  (n,string_of_typ T))
schirmer@25171
   442
                         ((map (fn (n,_) => (n,nameT)) all_comps) @
schirmer@25171
   443
                          constrains))
schirmer@25171
   444
          in [fx,cs] end
schirmer@25171
   445
       )
schirmer@25171
   446
wenzelm@26478
   447
wenzelm@26478
   448
  in thy
wenzelm@26478
   449
     |> namespace_definition
norbert@29247
   450
           (suffix namespaceN name) nameT (parents_expr,[])
schirmer@25171
   451
           (map fst parent_comps) (map fst components)
schirmer@25171
   452
     |> Context.theory_map (add_statespace full_name args parents components [])
norbert@29247
   453
     |> add_locale (suffix valuetypesN name) (locinsts,locs) []
wenzelm@45362
   454
     |> Proof_Context.theory_of
wenzelm@26478
   455
     |> fold interprete_parent_valuetypes parents
norbert@29247
   456
     |> add_locale_cmd name
norbert@29247
   457
              ([(suffix namespaceN full_name ,(("",false),Expression.Named [])),
norbert@29247
   458
                (suffix valuetypesN full_name,(("",false),Expression.Named  []))],[]) fixestate
wenzelm@45362
   459
     |> Proof_Context.theory_of
schirmer@25171
   460
     |> fold interprete_parent parents
haftmann@38389
   461
     |> add_declaration full_name (declare_declinfo components')
schirmer@25171
   462
  end;
schirmer@25171
   463
schirmer@25171
   464
schirmer@25171
   465
(* prepare arguments *)
schirmer@25171
   466
wenzelm@27283
   467
fun read_raw_parent ctxt raw_T =
wenzelm@42361
   468
  (case Proof_Context.read_typ_abbrev ctxt raw_T of
schirmer@25171
   469
    Type (name, Ts) => (Ts, name)
wenzelm@27283
   470
  | T => error ("Bad parent statespace specification: " ^ Syntax.string_of_typ ctxt T));
schirmer@25171
   471
wenzelm@36149
   472
fun read_typ ctxt raw_T env =
wenzelm@36149
   473
  let
wenzelm@36149
   474
    val ctxt' = fold (Variable.declare_typ o TFree) env ctxt;
wenzelm@36149
   475
    val T = Syntax.read_typ ctxt' raw_T;
wenzelm@45741
   476
    val env' = Term.add_tfreesT T env;
wenzelm@36149
   477
  in (T, env') end;
wenzelm@36149
   478
wenzelm@36149
   479
fun cert_typ ctxt raw_T env =
wenzelm@36149
   480
  let
wenzelm@42361
   481
    val thy = Proof_Context.theory_of ctxt;
wenzelm@36149
   482
    val T = Type.no_tvars (Sign.certify_typ thy raw_T)
wenzelm@36149
   483
      handle TYPE (msg, _, _) => error msg;
wenzelm@45741
   484
    val env' = Term.add_tfreesT T env;
wenzelm@36149
   485
  in (T, env') end;
wenzelm@36149
   486
schirmer@25171
   487
fun gen_define_statespace prep_typ state_space args name parents comps thy =
schirmer@25171
   488
  let (* - args distinct
schirmer@25171
   489
         - only args may occur in comps and parent-instantiations
wenzelm@26478
   490
         - number of insts must match parent args
schirmer@25171
   491
         - no duplicate renamings
schirmer@25171
   492
         - renaming should occur in namespace
schirmer@25171
   493
      *)
wenzelm@26478
   494
    val _ = writeln ("Defining statespace " ^ quote name ^ " ...");
schirmer@25171
   495
wenzelm@42361
   496
    val ctxt = Proof_Context.init_global thy;
wenzelm@27283
   497
schirmer@25171
   498
    fun add_parent (Ts,pname,rs) env =
schirmer@25171
   499
      let
haftmann@28965
   500
        val full_pname = Sign.full_bname thy pname;
wenzelm@26478
   501
        val {args,components,...} =
schirmer@25171
   502
              (case get_statespace (Context.Theory thy) full_pname of
schirmer@25171
   503
                SOME r => r
schirmer@25171
   504
               | NONE => error ("Undefined statespace " ^ quote pname));
schirmer@25171
   505
schirmer@25171
   506
wenzelm@27283
   507
        val (Ts',env') = fold_map (prep_typ ctxt) Ts env
wenzelm@26478
   508
            handle ERROR msg => cat_error msg
wenzelm@36149
   509
                    ("The error(s) above occurred in parent statespace specification "
schirmer@25171
   510
                    ^ quote pname);
schirmer@25171
   511
        val err_insts = if length args <> length Ts' then
schirmer@25171
   512
            ["number of type instantiation(s) does not match arguments of parent statespace "
schirmer@25171
   513
              ^ quote pname]
schirmer@25171
   514
            else [];
wenzelm@26478
   515
schirmer@25171
   516
        val rnames = map fst rs
schirmer@25171
   517
        val err_dup_renamings = (case duplicates (op =) rnames of
schirmer@25171
   518
             [] => []
schirmer@25171
   519
            | dups => ["Duplicate renaming(s) for " ^ commas dups])
schirmer@25171
   520
schirmer@25171
   521
        val cnames = map fst components;
haftmann@36692
   522
        val err_rename_unknowns = (case subtract (op =) cnames rnames of
schirmer@25171
   523
              [] => []
schirmer@25171
   524
             | rs => ["Unknown components " ^ commas rs]);
wenzelm@26478
   525
schirmer@25171
   526
schirmer@25171
   527
        val rs' = map (AList.lookup (op =) rs o fst) components;
schirmer@25171
   528
        val errs =err_insts @ err_dup_renamings @ err_rename_unknowns
schirmer@25171
   529
      in if null errs then ((Ts',full_pname,rs'),env')
schirmer@25171
   530
         else error (cat_lines (errs @ ["in parent statespace " ^ quote pname]))
schirmer@25171
   531
      end;
wenzelm@26478
   532
schirmer@25171
   533
    val (parents',env) = fold_map add_parent parents [];
schirmer@25171
   534
schirmer@25171
   535
    val err_dup_args =
schirmer@25171
   536
         (case duplicates (op =) args of
schirmer@25171
   537
            [] => []
schirmer@25171
   538
          | dups => ["Duplicate type argument(s) " ^ commas dups]);
wenzelm@26478
   539
schirmer@25171
   540
wenzelm@26478
   541
    val err_dup_components =
schirmer@25171
   542
         (case duplicates (op =) (map fst comps) of
schirmer@25171
   543
           [] => []
schirmer@25171
   544
          | dups => ["Duplicate state-space components " ^ commas dups]);
schirmer@25171
   545
schirmer@25171
   546
    fun prep_comp (n,T) env =
wenzelm@27283
   547
      let val (T', env') = prep_typ ctxt T env handle ERROR msg =>
wenzelm@36149
   548
       cat_error msg ("The error(s) above occurred in component " ^ quote n)
schirmer@25171
   549
      in ((n,T'), env') end;
schirmer@25171
   550
schirmer@25171
   551
    val (comps',env') = fold_map prep_comp comps env;
schirmer@25171
   552
schirmer@25171
   553
    val err_extra_frees =
schirmer@25171
   554
      (case subtract (op =) args (map fst env') of
schirmer@25171
   555
        [] => []
schirmer@25171
   556
      | extras => ["Extra free type variable(s) " ^ commas extras]);
schirmer@25171
   557
schirmer@25171
   558
    val defaultS = Sign.defaultS thy;
schirmer@25171
   559
    val args' = map (fn x => (x, AList.lookup (op =) env x |> the_default defaultS)) args;
schirmer@25171
   560
schirmer@25171
   561
schirmer@25171
   562
    fun fst_eq ((x:string,_),(y,_)) = x = y;
wenzelm@26478
   563
    fun snd_eq ((_,t:typ),(_,u)) = t = u;
schirmer@25171
   564
wenzelm@32952
   565
    val raw_parent_comps = maps (parent_components thy) parents';
wenzelm@26478
   566
    fun check_type (n,T) =
schirmer@25171
   567
          (case distinct (snd_eq) (filter (curry fst_eq (n,T)) raw_parent_comps) of
schirmer@25171
   568
             []  => []
schirmer@25171
   569
           | [_] => []
wenzelm@45660
   570
           | rs  => ["Different types for component " ^ quote n ^ ": " ^
wenzelm@32432
   571
                commas (map (Syntax.string_of_typ ctxt o snd) rs)])
wenzelm@26478
   572
wenzelm@32952
   573
    val err_dup_types = maps check_type (duplicates fst_eq raw_parent_comps)
schirmer@25171
   574
schirmer@25171
   575
    val parent_comps = distinct (fst_eq) raw_parent_comps;
schirmer@25171
   576
    val all_comps = parent_comps @ comps';
schirmer@25171
   577
    val err_comp_in_parent = (case duplicates (op =) (map fst all_comps) of
schirmer@25171
   578
               [] => []
wenzelm@45660
   579
             | xs => ["Components already defined in parents: " ^ commas_quote xs]);
schirmer@25171
   580
    val errs = err_dup_args @ err_dup_components @ err_extra_frees @
schirmer@25171
   581
               err_dup_types @ err_comp_in_parent;
wenzelm@26478
   582
  in if null errs
schirmer@25171
   583
     then thy |> statespace_definition state_space args' name parents' parent_comps comps'
wenzelm@26478
   584
     else error (cat_lines errs)
schirmer@25171
   585
  end
schirmer@25171
   586
  handle ERROR msg => cat_error msg ("Failed to define statespace " ^ quote name);
wenzelm@26478
   587
wenzelm@36149
   588
val define_statespace = gen_define_statespace read_typ NONE;
wenzelm@36149
   589
val define_statespace_i = gen_define_statespace cert_typ;
wenzelm@26478
   590
schirmer@25171
   591
wenzelm@45362
   592
schirmer@25171
   593
(*** parse/print - translations ***)
schirmer@25171
   594
schirmer@25171
   595
local
wenzelm@26478
   596
fun map_get_comp f ctxt (Free (name,_)) =
wenzelm@26478
   597
      (case (get_comp ctxt name) of
wenzelm@26478
   598
        SOME (T,_) => f T T dummyT
schirmer@25171
   599
      | NONE => (Syntax.free "arbitrary"(*; error "context not ready"*)))
schirmer@25171
   600
  | map_get_comp _ _ _ = Syntax.free "arbitrary";
schirmer@25171
   601
schirmer@25171
   602
val get_comp_projection = map_get_comp project_free;
schirmer@25171
   603
val get_comp_injection  = map_get_comp inject_free;
schirmer@25171
   604
schirmer@25171
   605
fun name_of (Free (n,_)) = n;
schirmer@25171
   606
in
schirmer@25171
   607
wenzelm@26478
   608
fun gen_lookup_tr ctxt s n =
wenzelm@45362
   609
  (case get_comp (Context.Proof ctxt) n of
wenzelm@45362
   610
    SOME (T, _) =>
wenzelm@45362
   611
      Syntax.const @{const_name StateFun.lookup} $
wenzelm@45362
   612
        Syntax.free (project_name T) $ Syntax.free n $ s
wenzelm@45362
   613
  | NONE =>
wenzelm@45362
   614
      if get_silent (Context.Proof ctxt)
wenzelm@45362
   615
      then Syntax.const @{const_name StateFun.lookup} $
wenzelm@45362
   616
        Syntax.const @{const_syntax undefined} $ Syntax.free n $ s
wenzelm@45660
   617
      else raise TERM ("StateSpace.gen_lookup_tr: component " ^ quote n ^ " not defined", []));
schirmer@25171
   618
wenzelm@42052
   619
fun lookup_tr ctxt [s, x] =
wenzelm@42264
   620
  (case Term_Position.strip_positions x of
wenzelm@42052
   621
    Free (n,_) => gen_lookup_tr ctxt s n
wenzelm@42052
   622
  | _ => raise Match);
wenzelm@42052
   623
schirmer@25171
   624
fun lookup_swap_tr ctxt [Free (n,_),s] = gen_lookup_tr ctxt s n;
schirmer@25171
   625
wenzelm@45362
   626
fun lookup_tr' ctxt [_ $ Free (prj, _), n as (_ $ Free (name, _)), s] =
wenzelm@45362
   627
      (case get_comp (Context.Proof ctxt) name of
wenzelm@45362
   628
        SOME (T, _) =>
wenzelm@45362
   629
          if prj = project_name T
wenzelm@45362
   630
          then Syntax.const "_statespace_lookup" $ s $ n
wenzelm@45362
   631
          else raise Match
schirmer@25171
   632
      | NONE => raise Match)
schirmer@25171
   633
  | lookup_tr' _ ts = raise Match;
schirmer@25171
   634
wenzelm@26478
   635
fun gen_update_tr id ctxt n v s =
schirmer@25171
   636
  let
wenzelm@45362
   637
    fun pname T = if id then @{const_name Fun.id} else project_name T;
wenzelm@45362
   638
    fun iname T = if id then @{const_name Fun.id} else inject_name T;
wenzelm@45362
   639
  in
wenzelm@45362
   640
    (case get_comp (Context.Proof ctxt) n of
wenzelm@45362
   641
      SOME (T, _) =>
wenzelm@45362
   642
        Syntax.const @{const_name StateFun.update} $
wenzelm@45362
   643
          Syntax.free (pname T) $ Syntax.free (iname T) $
wenzelm@45362
   644
          Syntax.free n $ (Syntax.const @{const_name K_statefun} $ v) $ s
wenzelm@45362
   645
    | NONE =>
wenzelm@45362
   646
        if get_silent (Context.Proof ctxt) then
wenzelm@45362
   647
          Syntax.const @{const_name StateFun.update} $
wenzelm@45362
   648
            Syntax.const @{const_syntax undefined} $ Syntax.const @{const_syntax undefined} $
wenzelm@45362
   649
            Syntax.free n $ (Syntax.const @{const_name K_statefun} $ v) $ s
wenzelm@45362
   650
       else raise TERM ("StateSpace.gen_update_tr: component " ^ n ^ " not defined", []))
schirmer@25171
   651
   end;
schirmer@25171
   652
wenzelm@42052
   653
fun update_tr ctxt [s, x, v] =
wenzelm@42264
   654
  (case Term_Position.strip_positions x of
wenzelm@42052
   655
    Free (n, _) => gen_update_tr false ctxt n v s
wenzelm@42052
   656
  | _ => raise Match);
schirmer@25171
   657
wenzelm@45362
   658
fun update_tr' ctxt
wenzelm@45362
   659
        [_ $ Free (prj, _), _ $ Free (inj, _), n as (_ $ Free (name, _)), (Const (k, _) $ v), s] =
wenzelm@45362
   660
      if Long_Name.base_name k = Long_Name.base_name @{const_name K_statefun} then
schirmer@25171
   661
        (case get_comp (Context.Proof ctxt) name of
wenzelm@45362
   662
          SOME (T, _) =>
wenzelm@45362
   663
            if inj = inject_name T andalso prj = project_name T then
wenzelm@45362
   664
              Syntax.const "_statespace_update" $ s $ n $ v
wenzelm@45362
   665
            else raise Match
wenzelm@45362
   666
        | NONE => raise Match)
schirmer@25171
   667
     else raise Match
schirmer@25171
   668
  | update_tr' _ _ = raise Match;
schirmer@25171
   669
schirmer@25171
   670
end;
schirmer@25171
   671
schirmer@25171
   672
schirmer@25171
   673
(*** outer syntax *)
schirmer@25171
   674
wenzelm@45362
   675
local
wenzelm@45362
   676
schirmer@25171
   677
val type_insts =
wenzelm@36960
   678
  Parse.typ >> single ||
wenzelm@36960
   679
  Parse.$$$ "(" |-- Parse.!!! (Parse.list1 Parse.typ --| Parse.$$$ ")")
schirmer@25171
   680
wenzelm@36960
   681
val comp = Parse.name -- (Parse.$$$ "::" |-- Parse.!!! Parse.typ);
schirmer@25171
   682
fun plus1_unless test scan =
wenzelm@36960
   683
  scan ::: Scan.repeat (Parse.$$$ "+" |-- Scan.unless test (Parse.!!! scan));
schirmer@25171
   684
wenzelm@36960
   685
val mapsto = Parse.$$$ "=";
wenzelm@36960
   686
val rename = Parse.name -- (mapsto |-- Parse.name);
wenzelm@36960
   687
val renames = Scan.optional (Parse.$$$ "[" |-- Parse.!!! (Parse.list1 rename --| Parse.$$$ "]")) [];
schirmer@25171
   688
wenzelm@45362
   689
val parent =
wenzelm@45362
   690
  ((type_insts -- Parse.xname) || (Parse.xname >> pair [])) -- renames
wenzelm@45362
   691
    >> (fn ((insts, name), renames) => (insts,name, renames));
schirmer@25171
   692
wenzelm@45362
   693
in
schirmer@25171
   694
schirmer@25171
   695
val statespace_decl =
wenzelm@45362
   696
  Parse.type_args -- Parse.name --
wenzelm@36960
   697
    (Parse.$$$ "=" |--
wenzelm@45362
   698
      ((Scan.repeat1 comp >> pair []) ||
wenzelm@45362
   699
        (plus1_unless comp parent --
wenzelm@45362
   700
          Scan.optional (Parse.$$$ "+" |-- Parse.!!! (Scan.repeat1 comp)) [])));
wenzelm@45362
   701
val _ =
wenzelm@45362
   702
  Outer_Syntax.command "statespace" "define state space" Keyword.thy_decl
wenzelm@45362
   703
    (statespace_decl >> (fn ((args, name), (parents, comps)) =>
wenzelm@45362
   704
      Toplevel.theory (define_statespace args name parents comps)));
schirmer@25171
   705
wenzelm@45362
   706
end;
schirmer@25171
   707
wenzelm@45362
   708
end;