src/Pure/Tools/codegen_package.ML
author urbanc
Fri, 17 Nov 2006 17:32:30 +0100
changeset 21405 26b51f724fe6
parent 21388 9d8344cf029f
child 21463 42dd50268c8b
permissions -rw-r--r--
added an intro lemma for freshness of products; set up the simplifier so that it can deal with the compact and long notation for freshness constraints (FIXME: it should also be able to deal with the special case of freshness of atoms)
Ignore whitespace changes - Everywhere: Within whitespace: At end of lines:
18169
45def66f86cb added modules for code generator generation two, not operational yet
haftmann
parents:
diff changeset
     1
(*  Title:      Pure/Tools/codegen_package.ML
45def66f86cb added modules for code generator generation two, not operational yet
haftmann
parents:
diff changeset
     2
    ID:         $Id$
45def66f86cb added modules for code generator generation two, not operational yet
haftmann
parents:
diff changeset
     3
    Author:     Florian Haftmann, TU Muenchen
45def66f86cb added modules for code generator generation two, not operational yet
haftmann
parents:
diff changeset
     4
20855
9f60d493c8fe clarified header comments
haftmann
parents: 20835
diff changeset
     5
Code generator extraction kernel. Code generator Isar setup.
18169
45def66f86cb added modules for code generator generation two, not operational yet
haftmann
parents:
diff changeset
     6
*)
45def66f86cb added modules for code generator generation two, not operational yet
haftmann
parents:
diff changeset
     7
45def66f86cb added modules for code generator generation two, not operational yet
haftmann
parents:
diff changeset
     8
signature CODEGEN_PACKAGE =
45def66f86cb added modules for code generator generation two, not operational yet
haftmann
parents:
diff changeset
     9
sig
20456
42be3a46dcd8 pervasive refinements
haftmann
parents: 20439
diff changeset
    10
  include BASIC_CODEGEN_THINGOL;
21121
fae2187d6e2f refined
haftmann
parents: 21081
diff changeset
    11
  val codegen_term: theory -> term -> iterm;
20600
6d75e02ed285 added codegen_data
haftmann
parents: 20485
diff changeset
    12
  val eval_term: theory -> (string (*reference name!*) * 'a option ref) * term -> 'a;
20699
0cc77abb185a refinements in codegen serializer
haftmann
parents: 20600
diff changeset
    13
  val const_of_idf: theory -> string -> string * typ;
20896
1484c7af6d68 cleaned up interfaces
haftmann
parents: 20855
diff changeset
    14
  val get_root_module: theory -> CodegenThingol.code;
18217
e0b08c9534ff added codegen package
haftmann
parents: 18169
diff changeset
    15
19884
a7be206d8655 improvements in code generator
haftmann
parents: 19816
diff changeset
    16
  type appgen;
20439
1bf42b262a38 code refinements
haftmann
parents: 20434
diff changeset
    17
  val add_appconst: string * appgen -> theory -> theory;
20600
6d75e02ed285 added codegen_data
haftmann
parents: 20485
diff changeset
    18
  val appgen_numeral: (theory -> term -> IntInf.int) -> appgen;
19607
07eeb832f28d introduced characters for code generator; some improved code lemmas for some list functions
haftmann
parents: 19597
diff changeset
    19
  val appgen_char: (term -> int option) -> appgen;
20105
454f4be984b7 adaptions in codegen
haftmann
parents: 20076
diff changeset
    20
  val appgen_case: (theory -> term
454f4be984b7 adaptions in codegen
haftmann
parents: 20076
diff changeset
    21
    -> ((string * typ) list * ((term * typ) * (term * term) list)) option)
454f4be984b7 adaptions in codegen
haftmann
parents: 20076
diff changeset
    22
    -> appgen;
454f4be984b7 adaptions in codegen
haftmann
parents: 20076
diff changeset
    23
  val appgen_let: appgen;
21012
haftmann
parents: 20976
diff changeset
    24
haftmann
parents: 20976
diff changeset
    25
  val timing: bool ref;
18169
45def66f86cb added modules for code generator generation two, not operational yet
haftmann
parents:
diff changeset
    26
end;
45def66f86cb added modules for code generator generation two, not operational yet
haftmann
parents:
diff changeset
    27
18217
e0b08c9534ff added codegen package
haftmann
parents: 18169
diff changeset
    28
structure CodegenPackage : CODEGEN_PACKAGE =
18169
45def66f86cb added modules for code generator generation two, not operational yet
haftmann
parents:
diff changeset
    29
struct
45def66f86cb added modules for code generator generation two, not operational yet
haftmann
parents:
diff changeset
    30
20896
1484c7af6d68 cleaned up interfaces
haftmann
parents: 20855
diff changeset
    31
open BasicCodegenThingol;
1484c7af6d68 cleaned up interfaces
haftmann
parents: 20855
diff changeset
    32
val tracing = CodegenThingol.tracing;
1484c7af6d68 cleaned up interfaces
haftmann
parents: 20855
diff changeset
    33
val succeed = CodegenThingol.succeed;
1484c7af6d68 cleaned up interfaces
haftmann
parents: 20855
diff changeset
    34
val fail = CodegenThingol.fail;
20699
0cc77abb185a refinements in codegen serializer
haftmann
parents: 20600
diff changeset
    35
0cc77abb185a refinements in codegen serializer
haftmann
parents: 20600
diff changeset
    36
(** code extraction **)
0cc77abb185a refinements in codegen serializer
haftmann
parents: 20600
diff changeset
    37
0cc77abb185a refinements in codegen serializer
haftmann
parents: 20600
diff changeset
    38
(* theory data *)
18217
e0b08c9534ff added codegen package
haftmann
parents: 18169
diff changeset
    39
20931
19d9b78218fd added code_abstype and code_constsubst
haftmann
parents: 20896
diff changeset
    40
structure Code = CodeDataFun
19d9b78218fd added code_abstype and code_constsubst
haftmann
parents: 20896
diff changeset
    41
(struct
19d9b78218fd added code_abstype and code_constsubst
haftmann
parents: 20896
diff changeset
    42
  val name = "Pure/code";
19d9b78218fd added code_abstype and code_constsubst
haftmann
parents: 20896
diff changeset
    43
  type T = CodegenThingol.code;
19d9b78218fd added code_abstype and code_constsubst
haftmann
parents: 20896
diff changeset
    44
  val empty = CodegenThingol.empty_code;
19d9b78218fd added code_abstype and code_constsubst
haftmann
parents: 20896
diff changeset
    45
  fun merge _ = CodegenThingol.merge_code;
19d9b78218fd added code_abstype and code_constsubst
haftmann
parents: 20896
diff changeset
    46
  fun purge _ NONE _ = CodegenThingol.empty_code
19d9b78218fd added code_abstype and code_constsubst
haftmann
parents: 20896
diff changeset
    47
    | purge NONE _ _ = CodegenThingol.empty_code
19d9b78218fd added code_abstype and code_constsubst
haftmann
parents: 20896
diff changeset
    48
    | purge (SOME thy) (SOME cs) code =
19d9b78218fd added code_abstype and code_constsubst
haftmann
parents: 20896
diff changeset
    49
        let
19d9b78218fd added code_abstype and code_constsubst
haftmann
parents: 20896
diff changeset
    50
          val cs_exisiting =
19d9b78218fd added code_abstype and code_constsubst
haftmann
parents: 20896
diff changeset
    51
            map_filter (CodegenNames.const_rev thy) (Graph.keys code);
19d9b78218fd added code_abstype and code_constsubst
haftmann
parents: 20896
diff changeset
    52
        in
19d9b78218fd added code_abstype and code_constsubst
haftmann
parents: 20896
diff changeset
    53
          Graph.del_nodes
19d9b78218fd added code_abstype and code_constsubst
haftmann
parents: 20896
diff changeset
    54
            ((Graph.all_succs code
19d9b78218fd added code_abstype and code_constsubst
haftmann
parents: 20896
diff changeset
    55
              o map (CodegenNames.const thy)
19d9b78218fd added code_abstype and code_constsubst
haftmann
parents: 20896
diff changeset
    56
              o filter (member CodegenConsts.eq_const cs_exisiting)
19d9b78218fd added code_abstype and code_constsubst
haftmann
parents: 20896
diff changeset
    57
              ) cs)
19d9b78218fd added code_abstype and code_constsubst
haftmann
parents: 20896
diff changeset
    58
            code
19d9b78218fd added code_abstype and code_constsubst
haftmann
parents: 20896
diff changeset
    59
        end;
19d9b78218fd added code_abstype and code_constsubst
haftmann
parents: 20896
diff changeset
    60
end);
19d9b78218fd added code_abstype and code_constsubst
haftmann
parents: 20896
diff changeset
    61
20600
6d75e02ed285 added codegen_data
haftmann
parents: 20485
diff changeset
    62
type appgen = theory -> ((sort -> sort) * Sorts.algebra) * Consts.T
6d75e02ed285 added codegen_data
haftmann
parents: 20485
diff changeset
    63
  -> CodegenFuncgr.T
20896
1484c7af6d68 cleaned up interfaces
haftmann
parents: 20855
diff changeset
    64
  -> bool * string list option
1484c7af6d68 cleaned up interfaces
haftmann
parents: 20855
diff changeset
    65
  -> (string * typ) * term list -> CodegenThingol.transact -> iterm * CodegenThingol.transact;
20439
1bf42b262a38 code refinements
haftmann
parents: 20434
diff changeset
    66
1bf42b262a38 code refinements
haftmann
parents: 20434
diff changeset
    67
type appgens = (int * (appgen * stamp)) Symtab.table;
20931
19d9b78218fd added code_abstype and code_constsubst
haftmann
parents: 20896
diff changeset
    68
val merge_appgens : appgens * appgens -> appgens =
20105
454f4be984b7 adaptions in codegen
haftmann
parents: 20076
diff changeset
    69
  Symtab.merge (fn ((bounds1, (_, stamp1)), (bounds2, (_, stamp2))) =>
20931
19d9b78218fd added code_abstype and code_constsubst
haftmann
parents: 20896
diff changeset
    70
    bounds1 = bounds2 andalso stamp1 = stamp2);
18217
e0b08c9534ff added codegen package
haftmann
parents: 18169
diff changeset
    71
20931
19d9b78218fd added code_abstype and code_constsubst
haftmann
parents: 20896
diff changeset
    72
structure Consttab = CodegenConsts.Consttab;
19d9b78218fd added code_abstype and code_constsubst
haftmann
parents: 20896
diff changeset
    73
type abstypes = typ Symtab.table * CodegenConsts.const Consttab.table;
19d9b78218fd added code_abstype and code_constsubst
haftmann
parents: 20896
diff changeset
    74
fun merge_abstypes ((typs1, consts1) : abstypes, (typs2, consts2) : abstypes) =
19d9b78218fd added code_abstype and code_constsubst
haftmann
parents: 20896
diff changeset
    75
  (Symtab.merge (Type.eq_type Vartab.empty) (typs1, typs2),
19d9b78218fd added code_abstype and code_constsubst
haftmann
parents: 20896
diff changeset
    76
    Consttab.merge CodegenConsts.eq_const (consts1, consts2));
20456
42be3a46dcd8 pervasive refinements
haftmann
parents: 20439
diff changeset
    77
20600
6d75e02ed285 added codegen_data
haftmann
parents: 20485
diff changeset
    78
structure CodegenPackageData = TheoryDataFun
18217
e0b08c9534ff added codegen package
haftmann
parents: 18169
diff changeset
    79
(struct
19953
2f54a51f1801 class package refinements, slight code generation refinements
haftmann
parents: 19884
diff changeset
    80
  val name = "Pure/codegen_package";
20931
19d9b78218fd added code_abstype and code_constsubst
haftmann
parents: 20896
diff changeset
    81
  type T = appgens * abstypes;
19d9b78218fd added code_abstype and code_constsubst
haftmann
parents: 20896
diff changeset
    82
  val empty = (Symtab.empty, (Symtab.empty, Consttab.empty));
18217
e0b08c9534ff added codegen package
haftmann
parents: 18169
diff changeset
    83
  val copy = I;
e0b08c9534ff added codegen package
haftmann
parents: 18169
diff changeset
    84
  val extend = I;
20931
19d9b78218fd added code_abstype and code_constsubst
haftmann
parents: 20896
diff changeset
    85
  fun merge _ ((appgens1, abstypes1), (appgens2, abstypes2)) =
19d9b78218fd added code_abstype and code_constsubst
haftmann
parents: 20896
diff changeset
    86
    (merge_appgens (appgens1, appgens2), merge_abstypes (abstypes1, abstypes2));
20456
42be3a46dcd8 pervasive refinements
haftmann
parents: 20439
diff changeset
    87
  fun print _ _ = ();
18217
e0b08c9534ff added codegen package
haftmann
parents: 18169
diff changeset
    88
end);
e0b08c9534ff added codegen package
haftmann
parents: 18169
diff changeset
    89
20600
6d75e02ed285 added codegen_data
haftmann
parents: 20485
diff changeset
    90
val _ = Context.add_setup (Code.init #> CodegenPackageData.init);
18708
4b3dadb4fe33 setup: theory -> theory;
wenzelm
parents: 18704
diff changeset
    91
18865
31aed965135c minor cleanups
haftmann
parents: 18850
diff changeset
    92
20386
d1cbe5aa6bf2 module restructuring
haftmann
parents: 20353
diff changeset
    93
(* extraction kernel *)
18865
31aed965135c minor cleanups
haftmann
parents: 18850
diff changeset
    94
20931
19d9b78218fd added code_abstype and code_constsubst
haftmann
parents: 20896
diff changeset
    95
fun check_strict (false, _) has_seri x =
19884
a7be206d8655 improvements in code generator
haftmann
parents: 19816
diff changeset
    96
      false
20931
19d9b78218fd added code_abstype and code_constsubst
haftmann
parents: 20896
diff changeset
    97
  | check_strict (_, SOME targets) has_seri x =
20699
0cc77abb185a refinements in codegen serializer
haftmann
parents: 20600
diff changeset
    98
      not (has_seri targets x)
20931
19d9b78218fd added code_abstype and code_constsubst
haftmann
parents: 20896
diff changeset
    99
  | check_strict (true, _) has_seri x =
19884
a7be206d8655 improvements in code generator
haftmann
parents: 19816
diff changeset
   100
      true;
a7be206d8655 improvements in code generator
haftmann
parents: 19816
diff changeset
   101
20931
19d9b78218fd added code_abstype and code_constsubst
haftmann
parents: 20896
diff changeset
   102
fun get_abstype thy (tyco, tys) = case Symtab.lookup ((fst o snd o CodegenPackageData.get) thy) tyco
19d9b78218fd added code_abstype and code_constsubst
haftmann
parents: 20896
diff changeset
   103
 of SOME ty => SOME ((map_atyps (fn TFree (n, _) => nth tys (the (Int.fromString n)))) ty)
19d9b78218fd added code_abstype and code_constsubst
haftmann
parents: 20896
diff changeset
   104
  | NONE => NONE;
19d9b78218fd added code_abstype and code_constsubst
haftmann
parents: 20896
diff changeset
   105
20896
1484c7af6d68 cleaned up interfaces
haftmann
parents: 20855
diff changeset
   106
fun ensure_def_class thy (algbr as ((proj_sort, _), _)) funcgr strct class trns =
20386
d1cbe5aa6bf2 module restructuring
haftmann
parents: 20353
diff changeset
   107
  let
20896
1484c7af6d68 cleaned up interfaces
haftmann
parents: 20855
diff changeset
   108
    val (v, cs) = (ClassPackage.the_consts_sign thy) class;
1484c7af6d68 cleaned up interfaces
haftmann
parents: 20855
diff changeset
   109
    val superclasses = (proj_sort o Sign.super_classes thy) class;
1484c7af6d68 cleaned up interfaces
haftmann
parents: 20855
diff changeset
   110
    val classops' = map (CodegenNames.const thy o CodegenConsts.norm_of_typ thy) cs;
1484c7af6d68 cleaned up interfaces
haftmann
parents: 20855
diff changeset
   111
    val class' = CodegenNames.class thy class;
1484c7af6d68 cleaned up interfaces
haftmann
parents: 20855
diff changeset
   112
    fun defgen_class trns =
1484c7af6d68 cleaned up interfaces
haftmann
parents: 20855
diff changeset
   113
      trns
1484c7af6d68 cleaned up interfaces
haftmann
parents: 20855
diff changeset
   114
      |> fold_map (ensure_def_class thy algbr funcgr strct) superclasses
1484c7af6d68 cleaned up interfaces
haftmann
parents: 20855
diff changeset
   115
      ||>> (fold_map (exprgen_type thy algbr funcgr strct) o map snd) cs
1484c7af6d68 cleaned up interfaces
haftmann
parents: 20855
diff changeset
   116
      |-> (fn (superclasses, classoptyps) => succeed
1484c7af6d68 cleaned up interfaces
haftmann
parents: 20855
diff changeset
   117
        (CodegenThingol.Class (superclasses,
1484c7af6d68 cleaned up interfaces
haftmann
parents: 20855
diff changeset
   118
          (unprefix "'" v, classops' ~~ classoptyps))))
18454
6720b5010a57 slight improvements in name handling
haftmann
parents: 18385
diff changeset
   119
  in
6720b5010a57 slight improvements in name handling
haftmann
parents: 18385
diff changeset
   120
    trns
20896
1484c7af6d68 cleaned up interfaces
haftmann
parents: 20855
diff changeset
   121
    |> tracing (fn _ => "generating class " ^ quote class)
1484c7af6d68 cleaned up interfaces
haftmann
parents: 20855
diff changeset
   122
    |> CodegenThingol.ensure_def defgen_class true
1484c7af6d68 cleaned up interfaces
haftmann
parents: 20855
diff changeset
   123
        ("generating class " ^ quote class) class'
1484c7af6d68 cleaned up interfaces
haftmann
parents: 20855
diff changeset
   124
    |> pair class'
18865
31aed965135c minor cleanups
haftmann
parents: 18850
diff changeset
   125
  end
20600
6d75e02ed285 added codegen_data
haftmann
parents: 20485
diff changeset
   126
and ensure_def_tyco thy algbr funcgr strct tyco trns =
18865
31aed965135c minor cleanups
haftmann
parents: 18850
diff changeset
   127
  let
20896
1484c7af6d68 cleaned up interfaces
haftmann
parents: 20855
diff changeset
   128
    fun defgen_datatype trns =
1484c7af6d68 cleaned up interfaces
haftmann
parents: 20855
diff changeset
   129
      case CodegenData.get_datatype thy tyco
1484c7af6d68 cleaned up interfaces
haftmann
parents: 20855
diff changeset
   130
       of SOME (vs, cos) =>
1484c7af6d68 cleaned up interfaces
haftmann
parents: 20855
diff changeset
   131
            trns
1484c7af6d68 cleaned up interfaces
haftmann
parents: 20855
diff changeset
   132
            |> fold_map (exprgen_tyvar_sort thy algbr funcgr strct) vs
1484c7af6d68 cleaned up interfaces
haftmann
parents: 20855
diff changeset
   133
            ||>> fold_map (fn (c, tys) =>
1484c7af6d68 cleaned up interfaces
haftmann
parents: 20855
diff changeset
   134
              fold_map (exprgen_type thy algbr funcgr strct) tys
1484c7af6d68 cleaned up interfaces
haftmann
parents: 20855
diff changeset
   135
              #-> (fn tys' =>
1484c7af6d68 cleaned up interfaces
haftmann
parents: 20855
diff changeset
   136
                pair ((CodegenNames.const thy o CodegenConsts.norm_of_typ thy)
1484c7af6d68 cleaned up interfaces
haftmann
parents: 20855
diff changeset
   137
                  (c, tys ---> Type (tyco, map TFree vs)), tys'))) cos
1484c7af6d68 cleaned up interfaces
haftmann
parents: 20855
diff changeset
   138
            |-> (fn (vs, cos) => succeed (CodegenThingol.Datatype (vs, cos)))
1484c7af6d68 cleaned up interfaces
haftmann
parents: 20855
diff changeset
   139
        | NONE =>
1484c7af6d68 cleaned up interfaces
haftmann
parents: 20855
diff changeset
   140
            trns
1484c7af6d68 cleaned up interfaces
haftmann
parents: 20855
diff changeset
   141
            |> fail ("No such datatype: " ^ quote tyco)
20699
0cc77abb185a refinements in codegen serializer
haftmann
parents: 20600
diff changeset
   142
    val tyco' = CodegenNames.tyco thy tyco;
20931
19d9b78218fd added code_abstype and code_constsubst
haftmann
parents: 20896
diff changeset
   143
    val strict = check_strict strct (CodegenSerializer.tyco_has_serialization thy) tyco';
18865
31aed965135c minor cleanups
haftmann
parents: 18850
diff changeset
   144
  in
31aed965135c minor cleanups
haftmann
parents: 18850
diff changeset
   145
    trns
20835
haftmann
parents: 20699
diff changeset
   146
    |> tracing (fn _ => "generating type constructor " ^ quote tyco)
20896
1484c7af6d68 cleaned up interfaces
haftmann
parents: 20855
diff changeset
   147
    |> CodegenThingol.ensure_def defgen_datatype strict
20389
8b6ecb22ef35 cleanup
haftmann
parents: 20386
diff changeset
   148
        ("generating type constructor " ^ quote tyco) tyco'
18865
31aed965135c minor cleanups
haftmann
parents: 18850
diff changeset
   149
    |> pair tyco'
31aed965135c minor cleanups
haftmann
parents: 18850
diff changeset
   150
  end
20600
6d75e02ed285 added codegen_data
haftmann
parents: 20485
diff changeset
   151
and exprgen_tyvar_sort thy (algbr as ((proj_sort, _), _)) funcgr strct (v, sort) trns =
18516
4424e2bce9af slightly improved serialization
haftmann
parents: 18515
diff changeset
   152
  trns
20600
6d75e02ed285 added codegen_data
haftmann
parents: 20485
diff changeset
   153
  |> fold_map (ensure_def_class thy algbr funcgr strct) (proj_sort sort)
18865
31aed965135c minor cleanups
haftmann
parents: 18850
diff changeset
   154
  |-> (fn sort => pair (unprefix "'" v, sort))
20600
6d75e02ed285 added codegen_data
haftmann
parents: 20485
diff changeset
   155
and exprgen_type thy algbr funcgr strct (TVar _) trns =
20389
8b6ecb22ef35 cleanup
haftmann
parents: 20386
diff changeset
   156
      error "TVar encountered in typ during code generation"
20600
6d75e02ed285 added codegen_data
haftmann
parents: 20485
diff changeset
   157
  | exprgen_type thy algbr funcgr strct (TFree vs) trns =
18516
4424e2bce9af slightly improved serialization
haftmann
parents: 18515
diff changeset
   158
      trns
20600
6d75e02ed285 added codegen_data
haftmann
parents: 20485
diff changeset
   159
      |> exprgen_tyvar_sort thy algbr funcgr strct vs
19167
f237c0cb3882 refined representation of codegen intermediate language
haftmann
parents: 19150
diff changeset
   160
      |-> (fn (v, sort) => pair (ITyVar v))
20600
6d75e02ed285 added codegen_data
haftmann
parents: 20485
diff changeset
   161
  | exprgen_type thy algbr funcgr strct (Type ("fun", [t1, t2])) trns =
18516
4424e2bce9af slightly improved serialization
haftmann
parents: 18515
diff changeset
   162
      trns
20600
6d75e02ed285 added codegen_data
haftmann
parents: 20485
diff changeset
   163
      |> exprgen_type thy algbr funcgr strct t1
6d75e02ed285 added codegen_data
haftmann
parents: 20485
diff changeset
   164
      ||>> exprgen_type thy algbr funcgr strct t2
18516
4424e2bce9af slightly improved serialization
haftmann
parents: 18515
diff changeset
   165
      |-> (fn (t1', t2') => pair (t1' `-> t2'))
20600
6d75e02ed285 added codegen_data
haftmann
parents: 20485
diff changeset
   166
  | exprgen_type thy algbr funcgr strct (Type (tyco, tys)) trns =
20931
19d9b78218fd added code_abstype and code_constsubst
haftmann
parents: 20896
diff changeset
   167
      case get_abstype thy (tyco, tys)
19d9b78218fd added code_abstype and code_constsubst
haftmann
parents: 20896
diff changeset
   168
       of SOME ty =>
19d9b78218fd added code_abstype and code_constsubst
haftmann
parents: 20896
diff changeset
   169
            trns
19d9b78218fd added code_abstype and code_constsubst
haftmann
parents: 20896
diff changeset
   170
            |> exprgen_type thy algbr funcgr strct ty
19d9b78218fd added code_abstype and code_constsubst
haftmann
parents: 20896
diff changeset
   171
        | NONE =>
19d9b78218fd added code_abstype and code_constsubst
haftmann
parents: 20896
diff changeset
   172
            trns
19d9b78218fd added code_abstype and code_constsubst
haftmann
parents: 20896
diff changeset
   173
            |> ensure_def_tyco thy algbr funcgr strct tyco
19d9b78218fd added code_abstype and code_constsubst
haftmann
parents: 20896
diff changeset
   174
            ||>> fold_map (exprgen_type thy algbr funcgr strct) tys
19d9b78218fd added code_abstype and code_constsubst
haftmann
parents: 20896
diff changeset
   175
            |-> (fn (tyco, tys) => pair (tyco `%% tys));
18516
4424e2bce9af slightly improved serialization
haftmann
parents: 18515
diff changeset
   176
20835
haftmann
parents: 20699
diff changeset
   177
exception CONSTRAIN of (string * typ) * typ;
21012
haftmann
parents: 20976
diff changeset
   178
val timing = ref false;
20600
6d75e02ed285 added codegen_data
haftmann
parents: 20485
diff changeset
   179
6d75e02ed285 added codegen_data
haftmann
parents: 20485
diff changeset
   180
fun exprgen_typinst thy (algbr as ((proj_sort, algebra), consts)) funcgr strct (ty_ctxt, sort_decl) trns =
20456
42be3a46dcd8 pervasive refinements
haftmann
parents: 20439
diff changeset
   181
  let
42be3a46dcd8 pervasive refinements
haftmann
parents: 20439
diff changeset
   182
    val pp = Sign.pp thy;
42be3a46dcd8 pervasive refinements
haftmann
parents: 20439
diff changeset
   183
    datatype inst =
42be3a46dcd8 pervasive refinements
haftmann
parents: 20439
diff changeset
   184
        Inst of (class * string) * inst list list
42be3a46dcd8 pervasive refinements
haftmann
parents: 20439
diff changeset
   185
      | Contxt of (string * sort) * (class list * int);
42be3a46dcd8 pervasive refinements
haftmann
parents: 20439
diff changeset
   186
    fun classrel (l as Contxt (v_sort, (classes, n)), _) class  =
42be3a46dcd8 pervasive refinements
haftmann
parents: 20439
diff changeset
   187
          Contxt (v_sort, (class :: classes, n))
42be3a46dcd8 pervasive refinements
haftmann
parents: 20439
diff changeset
   188
      | classrel (Inst ((_, tyco), lss), _) class =
42be3a46dcd8 pervasive refinements
haftmann
parents: 20439
diff changeset
   189
          Inst ((class, tyco), lss);
42be3a46dcd8 pervasive refinements
haftmann
parents: 20439
diff changeset
   190
    fun constructor tyco iss class =
42be3a46dcd8 pervasive refinements
haftmann
parents: 20439
diff changeset
   191
      Inst ((class, tyco), (map o map) fst iss);
42be3a46dcd8 pervasive refinements
haftmann
parents: 20439
diff changeset
   192
    fun variable (TFree (v, sort)) =
42be3a46dcd8 pervasive refinements
haftmann
parents: 20439
diff changeset
   193
      let
42be3a46dcd8 pervasive refinements
haftmann
parents: 20439
diff changeset
   194
        val sort' = proj_sort sort;
42be3a46dcd8 pervasive refinements
haftmann
parents: 20439
diff changeset
   195
      in map_index (fn (n, class) => (Contxt ((v, sort'), ([], n)), class)) sort' end;
42be3a46dcd8 pervasive refinements
haftmann
parents: 20439
diff changeset
   196
    val insts = Sorts.of_sort_derivation pp algebra
42be3a46dcd8 pervasive refinements
haftmann
parents: 20439
diff changeset
   197
      {classrel = classrel, constructor = constructor, variable = variable}
42be3a46dcd8 pervasive refinements
haftmann
parents: 20439
diff changeset
   198
      (ty_ctxt, proj_sort sort_decl);
42be3a46dcd8 pervasive refinements
haftmann
parents: 20439
diff changeset
   199
    fun mk_dict (Inst (inst, instss)) trns =
42be3a46dcd8 pervasive refinements
haftmann
parents: 20439
diff changeset
   200
          trns
20600
6d75e02ed285 added codegen_data
haftmann
parents: 20485
diff changeset
   201
          |> ensure_def_inst thy algbr funcgr strct inst
20456
42be3a46dcd8 pervasive refinements
haftmann
parents: 20439
diff changeset
   202
          ||>> (fold_map o fold_map) mk_dict instss
42be3a46dcd8 pervasive refinements
haftmann
parents: 20439
diff changeset
   203
          |-> (fn (inst, instss) => pair (Instance (inst, instss)))
42be3a46dcd8 pervasive refinements
haftmann
parents: 20439
diff changeset
   204
      | mk_dict (Contxt ((v, sort), (classes, k))) trns =
18517
788fa99aba33 slight improvements
haftmann
parents: 18516
diff changeset
   205
          trns
20600
6d75e02ed285 added codegen_data
haftmann
parents: 20485
diff changeset
   206
          |> fold_map (ensure_def_class thy algbr funcgr strct) classes
20456
42be3a46dcd8 pervasive refinements
haftmann
parents: 20439
diff changeset
   207
          |-> (fn classes => pair (Context (classes, (unprefix "'" v,
42be3a46dcd8 pervasive refinements
haftmann
parents: 20439
diff changeset
   208
                if length sort = 1 then ~1 else k))))
42be3a46dcd8 pervasive refinements
haftmann
parents: 20439
diff changeset
   209
  in
42be3a46dcd8 pervasive refinements
haftmann
parents: 20439
diff changeset
   210
    trns
42be3a46dcd8 pervasive refinements
haftmann
parents: 20439
diff changeset
   211
    |> fold_map mk_dict insts
42be3a46dcd8 pervasive refinements
haftmann
parents: 20439
diff changeset
   212
  end
20600
6d75e02ed285 added codegen_data
haftmann
parents: 20485
diff changeset
   213
and exprgen_typinst_const thy (algbr as (_, consts)) funcgr strct (c, ty_ctxt) trns =
6d75e02ed285 added codegen_data
haftmann
parents: 20485
diff changeset
   214
  let
6d75e02ed285 added codegen_data
haftmann
parents: 20485
diff changeset
   215
    val c' = CodegenConsts.norm_of_typ thy (c, ty_ctxt)
20699
0cc77abb185a refinements in codegen serializer
haftmann
parents: 20600
diff changeset
   216
    val idf = CodegenNames.const thy c';
20466
7c20ddbd911b explicit table with constant types
haftmann
parents: 20456
diff changeset
   217
    val ty_decl = Consts.declaration consts idf;
7c20ddbd911b explicit table with constant types
haftmann
parents: 20456
diff changeset
   218
    val insts = (op ~~ o apsnd (map (snd o dest_TVar)) oo pairself)
7c20ddbd911b explicit table with constant types
haftmann
parents: 20456
diff changeset
   219
      (curry (Consts.typargs consts) idf) (ty_ctxt, ty_decl);
20600
6d75e02ed285 added codegen_data
haftmann
parents: 20485
diff changeset
   220
    val _ = if exists not (map (Sign.of_sort thy) insts)
20835
haftmann
parents: 20699
diff changeset
   221
      then raise CONSTRAIN ((c, ty_decl), ty_ctxt) else ();
20456
42be3a46dcd8 pervasive refinements
haftmann
parents: 20439
diff changeset
   222
  in
42be3a46dcd8 pervasive refinements
haftmann
parents: 20439
diff changeset
   223
    trns
20600
6d75e02ed285 added codegen_data
haftmann
parents: 20485
diff changeset
   224
    |> fold_map (exprgen_typinst thy algbr funcgr strct) insts
20456
42be3a46dcd8 pervasive refinements
haftmann
parents: 20439
diff changeset
   225
  end
20896
1484c7af6d68 cleaned up interfaces
haftmann
parents: 20855
diff changeset
   226
and ensure_def_inst thy (algbr as ((proj_sort, _), _)) funcgr strct (class, tyco) trns =
20456
42be3a46dcd8 pervasive refinements
haftmann
parents: 20439
diff changeset
   227
  let
20896
1484c7af6d68 cleaned up interfaces
haftmann
parents: 20855
diff changeset
   228
    val (vs, classop_defs) = ((apsnd o map) Const o ClassPackage.the_inst_sign thy)
1484c7af6d68 cleaned up interfaces
haftmann
parents: 20855
diff changeset
   229
      (class, tyco);
1484c7af6d68 cleaned up interfaces
haftmann
parents: 20855
diff changeset
   230
    val classops = (map (CodegenConsts.norm_of_typ thy) o snd
1484c7af6d68 cleaned up interfaces
haftmann
parents: 20855
diff changeset
   231
      o ClassPackage.the_consts_sign thy) class;
1484c7af6d68 cleaned up interfaces
haftmann
parents: 20855
diff changeset
   232
    val arity_typ = Type (tyco, (map TFree vs));
1484c7af6d68 cleaned up interfaces
haftmann
parents: 20855
diff changeset
   233
    val superclasses = (proj_sort o Sign.super_classes thy) class
1484c7af6d68 cleaned up interfaces
haftmann
parents: 20855
diff changeset
   234
    fun gen_superarity superclass trns =
1484c7af6d68 cleaned up interfaces
haftmann
parents: 20855
diff changeset
   235
      trns
1484c7af6d68 cleaned up interfaces
haftmann
parents: 20855
diff changeset
   236
      |> ensure_def_class thy algbr funcgr strct superclass
1484c7af6d68 cleaned up interfaces
haftmann
parents: 20855
diff changeset
   237
      ||>> exprgen_typinst thy algbr funcgr strct (arity_typ, [superclass])
1484c7af6d68 cleaned up interfaces
haftmann
parents: 20855
diff changeset
   238
      |-> (fn (superclass, [Instance (inst, iss)]) => pair (superclass, (inst, iss)));
1484c7af6d68 cleaned up interfaces
haftmann
parents: 20855
diff changeset
   239
    fun gen_classop_def (classop, t) trns =
1484c7af6d68 cleaned up interfaces
haftmann
parents: 20855
diff changeset
   240
      trns
1484c7af6d68 cleaned up interfaces
haftmann
parents: 20855
diff changeset
   241
      |> ensure_def_const thy algbr funcgr strct classop
1484c7af6d68 cleaned up interfaces
haftmann
parents: 20855
diff changeset
   242
      ||>> exprgen_term thy algbr funcgr strct t;
1484c7af6d68 cleaned up interfaces
haftmann
parents: 20855
diff changeset
   243
    fun defgen_inst trns =
1484c7af6d68 cleaned up interfaces
haftmann
parents: 20855
diff changeset
   244
      trns
1484c7af6d68 cleaned up interfaces
haftmann
parents: 20855
diff changeset
   245
      |> ensure_def_class thy algbr funcgr strct class
1484c7af6d68 cleaned up interfaces
haftmann
parents: 20855
diff changeset
   246
      ||>> ensure_def_tyco thy algbr funcgr strct tyco
1484c7af6d68 cleaned up interfaces
haftmann
parents: 20855
diff changeset
   247
      ||>> fold_map (exprgen_tyvar_sort thy algbr funcgr strct) vs
1484c7af6d68 cleaned up interfaces
haftmann
parents: 20855
diff changeset
   248
      ||>> fold_map gen_superarity superclasses
1484c7af6d68 cleaned up interfaces
haftmann
parents: 20855
diff changeset
   249
      ||>> fold_map gen_classop_def (classops ~~ classop_defs)
1484c7af6d68 cleaned up interfaces
haftmann
parents: 20855
diff changeset
   250
      |-> (fn ((((class, tyco), arity), superarities), classops) =>
1484c7af6d68 cleaned up interfaces
haftmann
parents: 20855
diff changeset
   251
             succeed (CodegenThingol.Classinst ((class, (tyco, arity)), (superarities, classops))));
1484c7af6d68 cleaned up interfaces
haftmann
parents: 20855
diff changeset
   252
    val inst = CodegenNames.instance thy (class, tyco);
18865
31aed965135c minor cleanups
haftmann
parents: 18850
diff changeset
   253
  in
31aed965135c minor cleanups
haftmann
parents: 18850
diff changeset
   254
    trns
20896
1484c7af6d68 cleaned up interfaces
haftmann
parents: 20855
diff changeset
   255
    |> tracing (fn _ => "generating instance " ^ quote class ^ " / " ^ quote tyco)
1484c7af6d68 cleaned up interfaces
haftmann
parents: 20855
diff changeset
   256
    |> CodegenThingol.ensure_def defgen_inst true
1484c7af6d68 cleaned up interfaces
haftmann
parents: 20855
diff changeset
   257
         ("generating instance " ^ quote class ^ " / " ^ quote tyco) inst
18865
31aed965135c minor cleanups
haftmann
parents: 18850
diff changeset
   258
    |> pair inst
31aed965135c minor cleanups
haftmann
parents: 18850
diff changeset
   259
  end
21081
837b535137a9 dropped classop shallow namespace
haftmann
parents: 21062
diff changeset
   260
and ensure_def_const thy (algbr as (_, consts)) funcgr strct (c_tys as (c, tys)) trns =
18865
31aed965135c minor cleanups
haftmann
parents: 18850
diff changeset
   261
  let
21081
837b535137a9 dropped classop shallow namespace
haftmann
parents: 21062
diff changeset
   262
    val c' = CodegenNames.const thy c_tys;
20896
1484c7af6d68 cleaned up interfaces
haftmann
parents: 20855
diff changeset
   263
    fun defgen_datatypecons trns =
1484c7af6d68 cleaned up interfaces
haftmann
parents: 20855
diff changeset
   264
      trns
1484c7af6d68 cleaned up interfaces
haftmann
parents: 20855
diff changeset
   265
      |> ensure_def_tyco thy algbr funcgr strct
21081
837b535137a9 dropped classop shallow namespace
haftmann
parents: 21062
diff changeset
   266
          ((the o CodegenData.get_datatype_of_constr thy) c_tys)
20896
1484c7af6d68 cleaned up interfaces
haftmann
parents: 20855
diff changeset
   267
      |-> (fn _ => succeed CodegenThingol.Bot);
1484c7af6d68 cleaned up interfaces
haftmann
parents: 20855
diff changeset
   268
    fun defgen_classop trns =
1484c7af6d68 cleaned up interfaces
haftmann
parents: 20855
diff changeset
   269
      trns
1484c7af6d68 cleaned up interfaces
haftmann
parents: 20855
diff changeset
   270
      |> ensure_def_class thy algbr funcgr strct ((the o AxClass.class_of_param thy) c)
1484c7af6d68 cleaned up interfaces
haftmann
parents: 20855
diff changeset
   271
      |-> (fn _ => succeed CodegenThingol.Bot);
1484c7af6d68 cleaned up interfaces
haftmann
parents: 20855
diff changeset
   272
    fun defgen_fun trns =
21121
fae2187d6e2f refined
haftmann
parents: 21081
diff changeset
   273
      case CodegenFuncgr.funcs funcgr
21081
837b535137a9 dropped classop shallow namespace
haftmann
parents: 21062
diff changeset
   274
        (perhaps (Consttab.lookup ((snd o snd o CodegenPackageData.get) thy)) c_tys)
20456
42be3a46dcd8 pervasive refinements
haftmann
parents: 20439
diff changeset
   275
       of eq_thms as eq_thm :: _ =>
42be3a46dcd8 pervasive refinements
haftmann
parents: 20439
diff changeset
   276
            let
21012
haftmann
parents: 20976
diff changeset
   277
              val timeap = if !timing then Output.timeap_msg ("time for " ^ c')
haftmann
parents: 20976
diff changeset
   278
                else I;
20456
42be3a46dcd8 pervasive refinements
haftmann
parents: 20439
diff changeset
   279
              val msg = cat_lines ("generating code for theorems " :: map string_of_thm eq_thms);
20600
6d75e02ed285 added codegen_data
haftmann
parents: 20485
diff changeset
   280
              val ty = (Logic.unvarifyT o CodegenData.typ_func thy) eq_thm;
20466
7c20ddbd911b explicit table with constant types
haftmann
parents: 20456
diff changeset
   281
              val vs = (map dest_TFree o Consts.typargs consts) (c', ty);
20835
haftmann
parents: 20699
diff changeset
   282
              val dest_eqthm =
haftmann
parents: 20699
diff changeset
   283
                apfst (snd o strip_comb) o Logic.dest_equals o Logic.unvarify o prop_of;
20456
42be3a46dcd8 pervasive refinements
haftmann
parents: 20439
diff changeset
   284
              fun exprgen_eq (args, rhs) trns =
42be3a46dcd8 pervasive refinements
haftmann
parents: 20439
diff changeset
   285
                trns
20600
6d75e02ed285 added codegen_data
haftmann
parents: 20485
diff changeset
   286
                |> fold_map (exprgen_term thy algbr funcgr strct) args
6d75e02ed285 added codegen_data
haftmann
parents: 20485
diff changeset
   287
                ||>> exprgen_term thy algbr funcgr strct rhs;
20456
42be3a46dcd8 pervasive refinements
haftmann
parents: 20439
diff changeset
   288
            in
42be3a46dcd8 pervasive refinements
haftmann
parents: 20439
diff changeset
   289
              trns
20896
1484c7af6d68 cleaned up interfaces
haftmann
parents: 20855
diff changeset
   290
              |> CodegenThingol.message msg (fn trns => trns
21012
haftmann
parents: 20976
diff changeset
   291
              |> timeap (fold_map (exprgen_eq o dest_eqthm) eq_thms)
20600
6d75e02ed285 added codegen_data
haftmann
parents: 20485
diff changeset
   292
              ||>> fold_map (exprgen_tyvar_sort thy algbr funcgr strct) vs
6d75e02ed285 added codegen_data
haftmann
parents: 20485
diff changeset
   293
              ||>> exprgen_type thy algbr funcgr strct ty
20896
1484c7af6d68 cleaned up interfaces
haftmann
parents: 20855
diff changeset
   294
              |-> (fn ((eqs, vs), ty) => succeed (CodegenThingol.Fun (eqs, (vs, ty)))))
20456
42be3a46dcd8 pervasive refinements
haftmann
parents: 20439
diff changeset
   295
            end
42be3a46dcd8 pervasive refinements
haftmann
parents: 20439
diff changeset
   296
        | [] =>
42be3a46dcd8 pervasive refinements
haftmann
parents: 20439
diff changeset
   297
              trns
42be3a46dcd8 pervasive refinements
haftmann
parents: 20439
diff changeset
   298
              |> fail ("No defining equations found for "
21081
837b535137a9 dropped classop shallow namespace
haftmann
parents: 21062
diff changeset
   299
                   ^ (quote o CodegenConsts.string_of_const thy) c_tys);
837b535137a9 dropped classop shallow namespace
haftmann
parents: 21062
diff changeset
   300
    val defgen = if (is_some o CodegenData.get_datatype_of_constr thy) c_tys
837b535137a9 dropped classop shallow namespace
haftmann
parents: 21062
diff changeset
   301
      then defgen_datatypecons
837b535137a9 dropped classop shallow namespace
haftmann
parents: 21062
diff changeset
   302
      else if (is_some o AxClass.class_of_param thy) c andalso
837b535137a9 dropped classop shallow namespace
haftmann
parents: 21062
diff changeset
   303
        case tys of [TFree _] => true | [TVar _] => true | _ => false
20896
1484c7af6d68 cleaned up interfaces
haftmann
parents: 20855
diff changeset
   304
      then defgen_classop
21081
837b535137a9 dropped classop shallow namespace
haftmann
parents: 21062
diff changeset
   305
      else defgen_fun
20931
19d9b78218fd added code_abstype and code_constsubst
haftmann
parents: 20896
diff changeset
   306
    val strict = check_strict strct (CodegenSerializer.const_has_serialization thy) c';
18865
31aed965135c minor cleanups
haftmann
parents: 18850
diff changeset
   307
  in
31aed965135c minor cleanups
haftmann
parents: 18850
diff changeset
   308
    trns
20835
haftmann
parents: 20699
diff changeset
   309
    |> tracing (fn _ => "generating constant "
21081
837b535137a9 dropped classop shallow namespace
haftmann
parents: 21062
diff changeset
   310
        ^ (quote o CodegenConsts.string_of_const thy) c_tys)
20896
1484c7af6d68 cleaned up interfaces
haftmann
parents: 20855
diff changeset
   311
    |> CodegenThingol.ensure_def defgen strict ("generating constant "
21081
837b535137a9 dropped classop shallow namespace
haftmann
parents: 21062
diff changeset
   312
         ^ CodegenConsts.string_of_const thy c_tys) c'
20896
1484c7af6d68 cleaned up interfaces
haftmann
parents: 20855
diff changeset
   313
    |> pair c'
18865
31aed965135c minor cleanups
haftmann
parents: 18850
diff changeset
   314
  end
20896
1484c7af6d68 cleaned up interfaces
haftmann
parents: 20855
diff changeset
   315
and exprgen_term thy algbr funcgr strct (Const (c, ty)) trns =
18517
788fa99aba33 slight improvements
haftmann
parents: 18516
diff changeset
   316
      trns
20896
1484c7af6d68 cleaned up interfaces
haftmann
parents: 20855
diff changeset
   317
      |> select_appgen thy algbr funcgr strct ((c, ty), [])
20600
6d75e02ed285 added codegen_data
haftmann
parents: 20485
diff changeset
   318
  | exprgen_term thy algbr funcgr strct (Var _) trns =
20389
8b6ecb22ef35 cleanup
haftmann
parents: 20386
diff changeset
   319
      error "Var encountered in term during code generation"
20600
6d75e02ed285 added codegen_data
haftmann
parents: 20485
diff changeset
   320
  | exprgen_term thy algbr funcgr strct (Free (v, ty)) trns =
18516
4424e2bce9af slightly improved serialization
haftmann
parents: 18515
diff changeset
   321
      trns
20600
6d75e02ed285 added codegen_data
haftmann
parents: 20485
diff changeset
   322
      |> exprgen_type thy algbr funcgr strct ty
20896
1484c7af6d68 cleaned up interfaces
haftmann
parents: 20855
diff changeset
   323
      |-> (fn _ => pair (IVar v))
20600
6d75e02ed285 added codegen_data
haftmann
parents: 20485
diff changeset
   324
  | exprgen_term thy algbr funcgr strct (Abs (raw_v, ty, raw_t)) trns =
19136
00ade10f611d some refinements
haftmann
parents: 19111
diff changeset
   325
      let
20386
d1cbe5aa6bf2 module restructuring
haftmann
parents: 20353
diff changeset
   326
        val (v, t) = Syntax.variant_abs (CodegenNames.purify_var raw_v, ty, raw_t);
19136
00ade10f611d some refinements
haftmann
parents: 19111
diff changeset
   327
      in
00ade10f611d some refinements
haftmann
parents: 19111
diff changeset
   328
        trns
20600
6d75e02ed285 added codegen_data
haftmann
parents: 20485
diff changeset
   329
        |> exprgen_type thy algbr funcgr strct ty
6d75e02ed285 added codegen_data
haftmann
parents: 20485
diff changeset
   330
        ||>> exprgen_term thy algbr funcgr strct t
20896
1484c7af6d68 cleaned up interfaces
haftmann
parents: 20855
diff changeset
   331
        |-> (fn (ty, t) => pair ((v, ty) `|-> t))
19136
00ade10f611d some refinements
haftmann
parents: 19111
diff changeset
   332
      end
20896
1484c7af6d68 cleaned up interfaces
haftmann
parents: 20855
diff changeset
   333
  | exprgen_term thy algbr funcgr strct (t as _ $ _) trns =
1484c7af6d68 cleaned up interfaces
haftmann
parents: 20855
diff changeset
   334
      case strip_comb t
1484c7af6d68 cleaned up interfaces
haftmann
parents: 20855
diff changeset
   335
       of (Const (c, ty), ts) =>
18516
4424e2bce9af slightly improved serialization
haftmann
parents: 18515
diff changeset
   336
            trns
20896
1484c7af6d68 cleaned up interfaces
haftmann
parents: 20855
diff changeset
   337
            |> select_appgen thy algbr funcgr strct ((c, ty), ts)
1484c7af6d68 cleaned up interfaces
haftmann
parents: 20855
diff changeset
   338
            |-> (fn t => pair t)
1484c7af6d68 cleaned up interfaces
haftmann
parents: 20855
diff changeset
   339
        | (t', ts) =>
18516
4424e2bce9af slightly improved serialization
haftmann
parents: 18515
diff changeset
   340
            trns
20600
6d75e02ed285 added codegen_data
haftmann
parents: 20485
diff changeset
   341
            |> exprgen_term thy algbr funcgr strct t'
6d75e02ed285 added codegen_data
haftmann
parents: 20485
diff changeset
   342
            ||>> fold_map (exprgen_term thy algbr funcgr strct) ts
20896
1484c7af6d68 cleaned up interfaces
haftmann
parents: 20855
diff changeset
   343
            |-> (fn (t, ts) => pair (t `$$ ts))
20600
6d75e02ed285 added codegen_data
haftmann
parents: 20485
diff changeset
   344
and appgen_default thy algbr funcgr strct ((c, ty), ts) trns =
18865
31aed965135c minor cleanups
haftmann
parents: 18850
diff changeset
   345
  trns
20600
6d75e02ed285 added codegen_data
haftmann
parents: 20485
diff changeset
   346
  |> ensure_def_const thy algbr funcgr strct (CodegenConsts.norm_of_typ thy (c, ty))
6d75e02ed285 added codegen_data
haftmann
parents: 20485
diff changeset
   347
  ||>> exprgen_type thy algbr funcgr strct ty
6d75e02ed285 added codegen_data
haftmann
parents: 20485
diff changeset
   348
  ||>> exprgen_typinst_const thy algbr funcgr strct (c, ty)
6d75e02ed285 added codegen_data
haftmann
parents: 20485
diff changeset
   349
  ||>> fold_map (exprgen_term thy algbr funcgr strct) ts
20896
1484c7af6d68 cleaned up interfaces
haftmann
parents: 20855
diff changeset
   350
  |-> (fn (((c, ty), iss), ts) =>
1484c7af6d68 cleaned up interfaces
haftmann
parents: 20855
diff changeset
   351
         pair (IConst (c, (iss, ty)) `$$ ts))
1484c7af6d68 cleaned up interfaces
haftmann
parents: 20855
diff changeset
   352
and select_appgen thy algbr funcgr strct ((c, ty), ts) trns =
20931
19d9b78218fd added code_abstype and code_constsubst
haftmann
parents: 20896
diff changeset
   353
  case Symtab.lookup (fst (CodegenPackageData.get thy)) c
20896
1484c7af6d68 cleaned up interfaces
haftmann
parents: 20855
diff changeset
   354
   of SOME (i, (appgen, _)) =>
20105
454f4be984b7 adaptions in codegen
haftmann
parents: 20076
diff changeset
   355
        if length ts < i then
18865
31aed965135c minor cleanups
haftmann
parents: 18850
diff changeset
   356
          let
21161
ba4a40552f06 fixed problem with eta-expansion
haftmann
parents: 21129
diff changeset
   357
            val k = length ts;
ba4a40552f06 fixed problem with eta-expansion
haftmann
parents: 21129
diff changeset
   358
            val tys = (curry Library.take (i - k) o curry Library.drop k o fst o strip_type) ty;
ba4a40552f06 fixed problem with eta-expansion
haftmann
parents: 21129
diff changeset
   359
            val ctxt = (fold o fold_aterms)
ba4a40552f06 fixed problem with eta-expansion
haftmann
parents: 21129
diff changeset
   360
              (fn Free (v, _) => Name.declare v | _ => I) ts Name.context;
ba4a40552f06 fixed problem with eta-expansion
haftmann
parents: 21129
diff changeset
   361
            val vs = Name.names ctxt "a" tys;
18865
31aed965135c minor cleanups
haftmann
parents: 18850
diff changeset
   362
          in
31aed965135c minor cleanups
haftmann
parents: 18850
diff changeset
   363
            trns
20600
6d75e02ed285 added codegen_data
haftmann
parents: 20485
diff changeset
   364
            |> fold_map (exprgen_type thy algbr funcgr strct) tys
20896
1484c7af6d68 cleaned up interfaces
haftmann
parents: 20855
diff changeset
   365
            ||>> appgen thy algbr funcgr strct ((c, ty), ts @ map Free vs)
1484c7af6d68 cleaned up interfaces
haftmann
parents: 20855
diff changeset
   366
            |-> (fn (tys, t) => pair (map2 (fn (v, _) => pair v) vs tys `|--> t))
18865
31aed965135c minor cleanups
haftmann
parents: 18850
diff changeset
   367
          end
20105
454f4be984b7 adaptions in codegen
haftmann
parents: 20076
diff changeset
   368
        else if length ts > i then
18865
31aed965135c minor cleanups
haftmann
parents: 18850
diff changeset
   369
          trns
20896
1484c7af6d68 cleaned up interfaces
haftmann
parents: 20855
diff changeset
   370
          |> appgen thy algbr funcgr strct ((c, ty), Library.take (i, ts))
20600
6d75e02ed285 added codegen_data
haftmann
parents: 20485
diff changeset
   371
          ||>> fold_map (exprgen_term thy algbr funcgr strct) (Library.drop (i, ts))
20896
1484c7af6d68 cleaned up interfaces
haftmann
parents: 20855
diff changeset
   372
          |-> (fn (t, ts) => pair (t `$$ ts))
18865
31aed965135c minor cleanups
haftmann
parents: 18850
diff changeset
   373
        else
31aed965135c minor cleanups
haftmann
parents: 18850
diff changeset
   374
          trns
20896
1484c7af6d68 cleaned up interfaces
haftmann
parents: 20855
diff changeset
   375
          |> appgen thy algbr funcgr strct ((c, ty), ts)
18865
31aed965135c minor cleanups
haftmann
parents: 18850
diff changeset
   376
    | NONE =>
31aed965135c minor cleanups
haftmann
parents: 18850
diff changeset
   377
        trns
20896
1484c7af6d68 cleaned up interfaces
haftmann
parents: 20855
diff changeset
   378
        |> appgen_default thy algbr funcgr strct ((c, ty), ts);
20600
6d75e02ed285 added codegen_data
haftmann
parents: 20485
diff changeset
   379
6d75e02ed285 added codegen_data
haftmann
parents: 20485
diff changeset
   380
20835
haftmann
parents: 20699
diff changeset
   381
(* entrance points into extraction kernel *)
20600
6d75e02ed285 added codegen_data
haftmann
parents: 20485
diff changeset
   382
6d75e02ed285 added codegen_data
haftmann
parents: 20485
diff changeset
   383
fun ensure_def_const' thy algbr funcgr strct c trns =
6d75e02ed285 added codegen_data
haftmann
parents: 20485
diff changeset
   384
  ensure_def_const thy algbr funcgr strct c trns
20835
haftmann
parents: 20699
diff changeset
   385
  handle CONSTRAIN ((c, ty), ty_decl) => error (
20600
6d75e02ed285 added codegen_data
haftmann
parents: 20485
diff changeset
   386
    "Constant " ^ c ^ " with most general type\n"
6d75e02ed285 added codegen_data
haftmann
parents: 20485
diff changeset
   387
    ^ Sign.string_of_typ thy ty
6d75e02ed285 added codegen_data
haftmann
parents: 20485
diff changeset
   388
    ^ "\noccurs with type\n"
6d75e02ed285 added codegen_data
haftmann
parents: 20485
diff changeset
   389
    ^ Sign.string_of_typ thy ty_decl);
6d75e02ed285 added codegen_data
haftmann
parents: 20485
diff changeset
   390
fun exprgen_term' thy algbr funcgr strct t trns =
6d75e02ed285 added codegen_data
haftmann
parents: 20485
diff changeset
   391
  exprgen_term thy algbr funcgr strct t trns
20835
haftmann
parents: 20699
diff changeset
   392
  handle CONSTRAIN ((c, ty), ty_decl) => error ("In term " ^ (quote o Sign.string_of_term thy) t
20600
6d75e02ed285 added codegen_data
haftmann
parents: 20485
diff changeset
   393
    ^ ",\nconstant " ^ c ^ " with most general type\n"
6d75e02ed285 added codegen_data
haftmann
parents: 20485
diff changeset
   394
    ^ Sign.string_of_typ thy ty
6d75e02ed285 added codegen_data
haftmann
parents: 20485
diff changeset
   395
    ^ "\noccurs with type\n"
6d75e02ed285 added codegen_data
haftmann
parents: 20485
diff changeset
   396
    ^ Sign.string_of_typ thy ty_decl);
18516
4424e2bce9af slightly improved serialization
haftmann
parents: 18515
diff changeset
   397
18702
7dc7dcd63224 substantial improvements in code generator
haftmann
parents: 18517
diff changeset
   398
20439
1bf42b262a38 code refinements
haftmann
parents: 20434
diff changeset
   399
(* parametrized application generators, for instantiation in object logic *)
1bf42b262a38 code refinements
haftmann
parents: 20434
diff changeset
   400
(* (axiomatic extensions of extraction kernel *)
18217
e0b08c9534ff added codegen package
haftmann
parents: 18169
diff changeset
   401
20600
6d75e02ed285 added codegen_data
haftmann
parents: 20485
diff changeset
   402
fun appgen_numeral int_of_numeral thy algbr funcgr strct (app as (c, ts)) trns =
20485
3078fd2eec7b got rid of Numeral.bin type
haftmann
parents: 20466
diff changeset
   403
  case try (int_of_numeral thy) (list_comb (Const c, ts))
20835
haftmann
parents: 20699
diff changeset
   404
   of SOME i =>
20353
d73e49780ef2 code generator refinements
haftmann
parents: 20216
diff changeset
   405
        trns
21012
haftmann
parents: 20976
diff changeset
   406
        |> pair (CodegenThingol.INum i)
19884
a7be206d8655 improvements in code generator
haftmann
parents: 19816
diff changeset
   407
    | NONE =>
a7be206d8655 improvements in code generator
haftmann
parents: 19816
diff changeset
   408
        trns
20835
haftmann
parents: 20699
diff changeset
   409
        |> appgen_default thy algbr funcgr strct app;
18217
e0b08c9534ff added codegen package
haftmann
parents: 18169
diff changeset
   410
20600
6d75e02ed285 added codegen_data
haftmann
parents: 20485
diff changeset
   411
fun appgen_char char_to_index thy algbr funcgr strct (app as ((_, ty), _)) trns =
19607
07eeb832f28d introduced characters for code generator; some improved code lemmas for some list functions
haftmann
parents: 19597
diff changeset
   412
  case (char_to_index o list_comb o apfst Const) app
07eeb832f28d introduced characters for code generator; some improved code lemmas for some list functions
haftmann
parents: 19597
diff changeset
   413
   of SOME i =>
07eeb832f28d introduced characters for code generator; some improved code lemmas for some list functions
haftmann
parents: 19597
diff changeset
   414
        trns
20600
6d75e02ed285 added codegen_data
haftmann
parents: 20485
diff changeset
   415
        |> exprgen_type thy algbr funcgr strct ty
21012
haftmann
parents: 20976
diff changeset
   416
        |-> (fn _ => pair (IChar (chr i)))
19607
07eeb832f28d introduced characters for code generator; some improved code lemmas for some list functions
haftmann
parents: 19597
diff changeset
   417
    | NONE =>
07eeb832f28d introduced characters for code generator; some improved code lemmas for some list functions
haftmann
parents: 19597
diff changeset
   418
        trns
20600
6d75e02ed285 added codegen_data
haftmann
parents: 20485
diff changeset
   419
        |> appgen_default thy algbr funcgr strct app;
19607
07eeb832f28d introduced characters for code generator; some improved code lemmas for some list functions
haftmann
parents: 19597
diff changeset
   420
21012
haftmann
parents: 20976
diff changeset
   421
val debug_term = ref (Bound 0);
haftmann
parents: 20976
diff changeset
   422
20600
6d75e02ed285 added codegen_data
haftmann
parents: 20485
diff changeset
   423
fun appgen_case dest_case_expr thy algbr funcgr strct (app as (c_ty, ts)) trns =
20105
454f4be984b7 adaptions in codegen
haftmann
parents: 20076
diff changeset
   424
  let
454f4be984b7 adaptions in codegen
haftmann
parents: 20076
diff changeset
   425
    val SOME ([], ((st, sty), ds)) = dest_case_expr thy (list_comb (Const c_ty, ts));
21161
ba4a40552f06 fixed problem with eta-expansion
haftmann
parents: 21129
diff changeset
   426
    fun fold_map_aterms f (t1 $ t2) s =
20976
e324808e9f1f minor refinements in serialization
haftmann
parents: 20931
diff changeset
   427
          s
21161
ba4a40552f06 fixed problem with eta-expansion
haftmann
parents: 21129
diff changeset
   428
          |> fold_map_aterms f t1
ba4a40552f06 fixed problem with eta-expansion
haftmann
parents: 21129
diff changeset
   429
          ||>> fold_map_aterms f t2
20976
e324808e9f1f minor refinements in serialization
haftmann
parents: 20931
diff changeset
   430
          |-> (fn (t1, t2) => pair (t1 $ t2))
e324808e9f1f minor refinements in serialization
haftmann
parents: 20931
diff changeset
   431
      | fold_map_aterms f (Abs (v, ty, t)) s =
e324808e9f1f minor refinements in serialization
haftmann
parents: 20931
diff changeset
   432
          s
e324808e9f1f minor refinements in serialization
haftmann
parents: 20931
diff changeset
   433
          |> fold_map_aterms f t
e324808e9f1f minor refinements in serialization
haftmann
parents: 20931
diff changeset
   434
          |-> (fn t' => pair (Abs (v, ty, t')))
e324808e9f1f minor refinements in serialization
haftmann
parents: 20931
diff changeset
   435
      | fold_map_aterms f a s = f a s;
21012
haftmann
parents: 20976
diff changeset
   436
    fun purify_term_frees (Free (v, ty)) (renaming, ctxt) = 
20976
e324808e9f1f minor refinements in serialization
haftmann
parents: 20931
diff changeset
   437
          let
e324808e9f1f minor refinements in serialization
haftmann
parents: 20931
diff changeset
   438
            val ([v'], ctxt') = Name.variants [CodegenNames.purify_var v] ctxt;
21012
haftmann
parents: 20976
diff changeset
   439
            val renaming' = if v <> v' then Symtab.update (v, v') renaming else renaming;
haftmann
parents: 20976
diff changeset
   440
          in (Free (v', ty), (renaming', ctxt')) end
20976
e324808e9f1f minor refinements in serialization
haftmann
parents: 20931
diff changeset
   441
      | purify_term_frees t ctxt = (t, ctxt);
e324808e9f1f minor refinements in serialization
haftmann
parents: 20931
diff changeset
   442
    fun clausegen (raw_dt, raw_bt) trns =
e324808e9f1f minor refinements in serialization
haftmann
parents: 20931
diff changeset
   443
      let
21012
haftmann
parents: 20976
diff changeset
   444
        val d_vs = map fst (Term.add_frees raw_dt []);
haftmann
parents: 20976
diff changeset
   445
        val b_vs = map fst (Term.add_frees raw_bt []);
haftmann
parents: 20976
diff changeset
   446
        val (dt, (renaming, ctxt)) =
haftmann
parents: 20976
diff changeset
   447
          Name.context
haftmann
parents: 20976
diff changeset
   448
          |> fold Name.declare (subtract (op =) d_vs b_vs)
haftmann
parents: 20976
diff changeset
   449
          |> pair Symtab.empty
haftmann
parents: 20976
diff changeset
   450
          |> fold_map_aterms purify_term_frees raw_dt;
haftmann
parents: 20976
diff changeset
   451
        val bt = map_aterms (fn t as Free (v, ty) => Free (perhaps (Symtab.lookup renaming) v, ty)
haftmann
parents: 20976
diff changeset
   452
                              | t => t) raw_bt;
20976
e324808e9f1f minor refinements in serialization
haftmann
parents: 20931
diff changeset
   453
      in
e324808e9f1f minor refinements in serialization
haftmann
parents: 20931
diff changeset
   454
        trns
e324808e9f1f minor refinements in serialization
haftmann
parents: 20931
diff changeset
   455
        |> exprgen_term thy algbr funcgr strct dt
e324808e9f1f minor refinements in serialization
haftmann
parents: 20931
diff changeset
   456
        ||>> exprgen_term thy algbr funcgr strct bt
e324808e9f1f minor refinements in serialization
haftmann
parents: 20931
diff changeset
   457
      end;
20105
454f4be984b7 adaptions in codegen
haftmann
parents: 20076
diff changeset
   458
  in
454f4be984b7 adaptions in codegen
haftmann
parents: 20076
diff changeset
   459
    trns
20600
6d75e02ed285 added codegen_data
haftmann
parents: 20485
diff changeset
   460
    |> exprgen_term thy algbr funcgr strct st
6d75e02ed285 added codegen_data
haftmann
parents: 20485
diff changeset
   461
    ||>> exprgen_type thy algbr funcgr strct sty
20105
454f4be984b7 adaptions in codegen
haftmann
parents: 20076
diff changeset
   462
    ||>> fold_map clausegen ds
21012
haftmann
parents: 20976
diff changeset
   463
    |-> (fn ((se, sty), ds) => pair (ICase ((se, sty), ds)))
20105
454f4be984b7 adaptions in codegen
haftmann
parents: 20076
diff changeset
   464
  end;
454f4be984b7 adaptions in codegen
haftmann
parents: 20076
diff changeset
   465
20600
6d75e02ed285 added codegen_data
haftmann
parents: 20485
diff changeset
   466
fun appgen_let thy algbr funcgr strct (app as (_, [st, ct])) trns =
20105
454f4be984b7 adaptions in codegen
haftmann
parents: 20076
diff changeset
   467
  trns
20600
6d75e02ed285 added codegen_data
haftmann
parents: 20485
diff changeset
   468
  |> exprgen_term thy algbr funcgr strct ct
6d75e02ed285 added codegen_data
haftmann
parents: 20485
diff changeset
   469
  ||>> exprgen_term thy algbr funcgr strct st
21012
haftmann
parents: 20976
diff changeset
   470
  |-> (fn ((v, ty) `|-> be, se) =>
haftmann
parents: 20976
diff changeset
   471
            pair (ICase ((se, ty), case be
haftmann
parents: 20976
diff changeset
   472
              of ICase ((IVar w, _), ds) => if v = w then ds else [(IVar v, be)]
20105
454f4be984b7 adaptions in codegen
haftmann
parents: 20076
diff changeset
   473
               | _ => [(IVar v, be)]
21012
haftmann
parents: 20976
diff changeset
   474
            ))
haftmann
parents: 20976
diff changeset
   475
        | _ => appgen_default thy algbr funcgr strct app);
20105
454f4be984b7 adaptions in codegen
haftmann
parents: 20076
diff changeset
   476
20439
1bf42b262a38 code refinements
haftmann
parents: 20434
diff changeset
   477
fun add_appconst (c, appgen) thy =
1bf42b262a38 code refinements
haftmann
parents: 20434
diff changeset
   478
  let
20931
19d9b78218fd added code_abstype and code_constsubst
haftmann
parents: 20896
diff changeset
   479
    val i = (length o fst o strip_type o Sign.the_const_type thy) c;
19d9b78218fd added code_abstype and code_constsubst
haftmann
parents: 20896
diff changeset
   480
    val _ = Code.change thy (K CodegenThingol.empty_code);
19d9b78218fd added code_abstype and code_constsubst
haftmann
parents: 20896
diff changeset
   481
  in
19d9b78218fd added code_abstype and code_constsubst
haftmann
parents: 20896
diff changeset
   482
    (CodegenPackageData.map o apfst)
19d9b78218fd added code_abstype and code_constsubst
haftmann
parents: 20896
diff changeset
   483
      (Symtab.update (c, (i, (appgen, stamp ())))) thy
19d9b78218fd added code_abstype and code_constsubst
haftmann
parents: 20896
diff changeset
   484
  end;
19d9b78218fd added code_abstype and code_constsubst
haftmann
parents: 20896
diff changeset
   485
19d9b78218fd added code_abstype and code_constsubst
haftmann
parents: 20896
diff changeset
   486
19d9b78218fd added code_abstype and code_constsubst
haftmann
parents: 20896
diff changeset
   487
19d9b78218fd added code_abstype and code_constsubst
haftmann
parents: 20896
diff changeset
   488
(** abstype and constsubst interface **)
19d9b78218fd added code_abstype and code_constsubst
haftmann
parents: 20896
diff changeset
   489
19d9b78218fd added code_abstype and code_constsubst
haftmann
parents: 20896
diff changeset
   490
fun gen_abstyp prep_const prep_typ (raw_abstyp, raw_substtyp) raw_absconsts thy =
19d9b78218fd added code_abstype and code_constsubst
haftmann
parents: 20896
diff changeset
   491
  let
19d9b78218fd added code_abstype and code_constsubst
haftmann
parents: 20896
diff changeset
   492
    val abstyp = Type.no_tvars (prep_typ thy raw_abstyp);
19d9b78218fd added code_abstype and code_constsubst
haftmann
parents: 20896
diff changeset
   493
    val substtyp = Type.no_tvars (prep_typ thy raw_substtyp);
19d9b78218fd added code_abstype and code_constsubst
haftmann
parents: 20896
diff changeset
   494
    val absconsts = (map o pairself) (prep_const thy) raw_absconsts;
19d9b78218fd added code_abstype and code_constsubst
haftmann
parents: 20896
diff changeset
   495
    val Type (abstyco, tys) = abstyp handle BIND => error ("bad type: " ^ Sign.string_of_typ thy abstyp);
19d9b78218fd added code_abstype and code_constsubst
haftmann
parents: 20896
diff changeset
   496
    val typarms = map (fst o dest_TFree) tys handle MATCH => error ("bad type: " ^ Sign.string_of_typ thy abstyp);
19d9b78218fd added code_abstype and code_constsubst
haftmann
parents: 20896
diff changeset
   497
    fun mk_index v = 
19d9b78218fd added code_abstype and code_constsubst
haftmann
parents: 20896
diff changeset
   498
      let
19d9b78218fd added code_abstype and code_constsubst
haftmann
parents: 20896
diff changeset
   499
        val k = find_index (fn w => v = w) typarms;
19d9b78218fd added code_abstype and code_constsubst
haftmann
parents: 20896
diff changeset
   500
      in if k = ~1
19d9b78218fd added code_abstype and code_constsubst
haftmann
parents: 20896
diff changeset
   501
        then error ("free type variable: " ^ quote v)
19d9b78218fd added code_abstype and code_constsubst
haftmann
parents: 20896
diff changeset
   502
        else TFree (string_of_int k, [])
19d9b78218fd added code_abstype and code_constsubst
haftmann
parents: 20896
diff changeset
   503
      end;
19d9b78218fd added code_abstype and code_constsubst
haftmann
parents: 20896
diff changeset
   504
    val typpat = map_atyps (fn TFree (v, _) => mk_index v) substtyp;
19d9b78218fd added code_abstype and code_constsubst
haftmann
parents: 20896
diff changeset
   505
    fun apply_typpat (Type (tyco, tys)) =
19d9b78218fd added code_abstype and code_constsubst
haftmann
parents: 20896
diff changeset
   506
          let
19d9b78218fd added code_abstype and code_constsubst
haftmann
parents: 20896
diff changeset
   507
            val tys' = map apply_typpat tys;
19d9b78218fd added code_abstype and code_constsubst
haftmann
parents: 20896
diff changeset
   508
          in if tyco = abstyco then
19d9b78218fd added code_abstype and code_constsubst
haftmann
parents: 20896
diff changeset
   509
            (map_atyps (fn TFree (n, _) => nth tys' (the (Int.fromString n)))) typpat
19d9b78218fd added code_abstype and code_constsubst
haftmann
parents: 20896
diff changeset
   510
          else
19d9b78218fd added code_abstype and code_constsubst
haftmann
parents: 20896
diff changeset
   511
            Type (tyco, tys')
19d9b78218fd added code_abstype and code_constsubst
haftmann
parents: 20896
diff changeset
   512
          end
19d9b78218fd added code_abstype and code_constsubst
haftmann
parents: 20896
diff changeset
   513
      | apply_typpat ty = ty;
19d9b78218fd added code_abstype and code_constsubst
haftmann
parents: 20896
diff changeset
   514
    val string_of_typ = setmp show_sorts true (Sign.string_of_typ thy);
19d9b78218fd added code_abstype and code_constsubst
haftmann
parents: 20896
diff changeset
   515
    fun add_consts (c1, c2) =
19d9b78218fd added code_abstype and code_constsubst
haftmann
parents: 20896
diff changeset
   516
      let
19d9b78218fd added code_abstype and code_constsubst
haftmann
parents: 20896
diff changeset
   517
        val _ = if CodegenNames.has_nsp CodegenNames.nsp_fun (CodegenNames.const thy c2)
19d9b78218fd added code_abstype and code_constsubst
haftmann
parents: 20896
diff changeset
   518
          then ()
19d9b78218fd added code_abstype and code_constsubst
haftmann
parents: 20896
diff changeset
   519
          else error ("not a function: " ^ CodegenConsts.string_of_const thy c2);
21121
fae2187d6e2f refined
haftmann
parents: 21081
diff changeset
   520
        val funcgr = CodegenFuncgr.make thy [c1, c2];
fae2187d6e2f refined
haftmann
parents: 21081
diff changeset
   521
        val ty1 = (apply_typpat o CodegenFuncgr.typ funcgr) c1;
fae2187d6e2f refined
haftmann
parents: 21081
diff changeset
   522
        val ty2 = CodegenFuncgr.typ funcgr c2;
20931
19d9b78218fd added code_abstype and code_constsubst
haftmann
parents: 20896
diff changeset
   523
        val _ = if Sign.typ_equiv thy (ty1, ty2) then () else
19d9b78218fd added code_abstype and code_constsubst
haftmann
parents: 20896
diff changeset
   524
          error ("Incompatiable type signatures of " ^ CodegenConsts.string_of_const thy c1
19d9b78218fd added code_abstype and code_constsubst
haftmann
parents: 20896
diff changeset
   525
            ^ " and " ^ CodegenConsts.string_of_const thy c2 ^ ":\n"
19d9b78218fd added code_abstype and code_constsubst
haftmann
parents: 20896
diff changeset
   526
            ^ string_of_typ ty1 ^ "\n" ^ string_of_typ ty2);
19d9b78218fd added code_abstype and code_constsubst
haftmann
parents: 20896
diff changeset
   527
      in Consttab.update (c1, c2) end;
19d9b78218fd added code_abstype and code_constsubst
haftmann
parents: 20896
diff changeset
   528
    val _ = Code.change thy (K CodegenThingol.empty_code);
19d9b78218fd added code_abstype and code_constsubst
haftmann
parents: 20896
diff changeset
   529
  in
19d9b78218fd added code_abstype and code_constsubst
haftmann
parents: 20896
diff changeset
   530
    thy
19d9b78218fd added code_abstype and code_constsubst
haftmann
parents: 20896
diff changeset
   531
    |> (CodegenPackageData.map o apsnd) (fn (abstypes, abscs) =>
19d9b78218fd added code_abstype and code_constsubst
haftmann
parents: 20896
diff changeset
   532
          (abstypes
19d9b78218fd added code_abstype and code_constsubst
haftmann
parents: 20896
diff changeset
   533
          |> Symtab.update (abstyco, typpat),
19d9b78218fd added code_abstype and code_constsubst
haftmann
parents: 20896
diff changeset
   534
          abscs
19d9b78218fd added code_abstype and code_constsubst
haftmann
parents: 20896
diff changeset
   535
          |> fold add_consts absconsts)
19d9b78218fd added code_abstype and code_constsubst
haftmann
parents: 20896
diff changeset
   536
       )
19d9b78218fd added code_abstype and code_constsubst
haftmann
parents: 20896
diff changeset
   537
  end;
19d9b78218fd added code_abstype and code_constsubst
haftmann
parents: 20896
diff changeset
   538
19d9b78218fd added code_abstype and code_constsubst
haftmann
parents: 20896
diff changeset
   539
fun gen_constsubst prep_const raw_constsubsts thy =
19d9b78218fd added code_abstype and code_constsubst
haftmann
parents: 20896
diff changeset
   540
  let
19d9b78218fd added code_abstype and code_constsubst
haftmann
parents: 20896
diff changeset
   541
    val constsubsts = (map o pairself) (prep_const thy) raw_constsubsts;
19d9b78218fd added code_abstype and code_constsubst
haftmann
parents: 20896
diff changeset
   542
    val string_of_typ = setmp show_sorts true (Sign.string_of_typ thy);
19d9b78218fd added code_abstype and code_constsubst
haftmann
parents: 20896
diff changeset
   543
    fun add_consts (c1, c2) =
19d9b78218fd added code_abstype and code_constsubst
haftmann
parents: 20896
diff changeset
   544
      let
19d9b78218fd added code_abstype and code_constsubst
haftmann
parents: 20896
diff changeset
   545
        val _ = if CodegenNames.has_nsp CodegenNames.nsp_fun (CodegenNames.const thy c2)
19d9b78218fd added code_abstype and code_constsubst
haftmann
parents: 20896
diff changeset
   546
          then ()
19d9b78218fd added code_abstype and code_constsubst
haftmann
parents: 20896
diff changeset
   547
          else error ("not a function: " ^ CodegenConsts.string_of_const thy c2);
21121
fae2187d6e2f refined
haftmann
parents: 21081
diff changeset
   548
        val funcgr = CodegenFuncgr.make thy [c1, c2];
fae2187d6e2f refined
haftmann
parents: 21081
diff changeset
   549
        val ty1 = CodegenFuncgr.typ funcgr c1;
fae2187d6e2f refined
haftmann
parents: 21081
diff changeset
   550
        val ty2 = CodegenFuncgr.typ funcgr c2;
20931
19d9b78218fd added code_abstype and code_constsubst
haftmann
parents: 20896
diff changeset
   551
        val _ = if Sign.typ_equiv thy (ty1, ty2) then () else
19d9b78218fd added code_abstype and code_constsubst
haftmann
parents: 20896
diff changeset
   552
          error ("Incompatiable type signatures of " ^ CodegenConsts.string_of_const thy c1
19d9b78218fd added code_abstype and code_constsubst
haftmann
parents: 20896
diff changeset
   553
            ^ " and " ^ CodegenConsts.string_of_const thy c2 ^ ":\n"
19d9b78218fd added code_abstype and code_constsubst
haftmann
parents: 20896
diff changeset
   554
            ^ string_of_typ ty1 ^ "\n" ^ string_of_typ ty2);
19d9b78218fd added code_abstype and code_constsubst
haftmann
parents: 20896
diff changeset
   555
      in Consttab.update (c1, c2) end;
19d9b78218fd added code_abstype and code_constsubst
haftmann
parents: 20896
diff changeset
   556
    val _ = Code.change thy (K CodegenThingol.empty_code);
19d9b78218fd added code_abstype and code_constsubst
haftmann
parents: 20896
diff changeset
   557
  in
19d9b78218fd added code_abstype and code_constsubst
haftmann
parents: 20896
diff changeset
   558
    thy
19d9b78218fd added code_abstype and code_constsubst
haftmann
parents: 20896
diff changeset
   559
    |> (CodegenPackageData.map o apsnd o apsnd) (fold add_consts constsubsts)
19d9b78218fd added code_abstype and code_constsubst
haftmann
parents: 20896
diff changeset
   560
  end;
19d9b78218fd added code_abstype and code_constsubst
haftmann
parents: 20896
diff changeset
   561
19d9b78218fd added code_abstype and code_constsubst
haftmann
parents: 20896
diff changeset
   562
val abstyp = gen_abstyp CodegenConsts.norm Sign.certify_typ;
19d9b78218fd added code_abstype and code_constsubst
haftmann
parents: 20896
diff changeset
   563
val abstyp_e = gen_abstyp CodegenConsts.read_const (fn thy => Sign.read_typ (thy, K NONE));
19d9b78218fd added code_abstype and code_constsubst
haftmann
parents: 20896
diff changeset
   564
19d9b78218fd added code_abstype and code_constsubst
haftmann
parents: 20896
diff changeset
   565
val constsubst = gen_constsubst CodegenConsts.norm;
19d9b78218fd added code_abstype and code_constsubst
haftmann
parents: 20896
diff changeset
   566
val constsubst_e = gen_constsubst CodegenConsts.read_const;
20439
1bf42b262a38 code refinements
haftmann
parents: 20434
diff changeset
   567
18217
e0b08c9534ff added codegen package
haftmann
parents: 18169
diff changeset
   568
18516
4424e2bce9af slightly improved serialization
haftmann
parents: 18515
diff changeset
   569
20439
1bf42b262a38 code refinements
haftmann
parents: 20434
diff changeset
   570
(** code generation interfaces **)
18516
4424e2bce9af slightly improved serialization
haftmann
parents: 18515
diff changeset
   571
21121
fae2187d6e2f refined
haftmann
parents: 21081
diff changeset
   572
fun generate thy funcgr targets init gen it =
20466
7c20ddbd911b explicit table with constant types
haftmann
parents: 20456
diff changeset
   573
  let
21121
fae2187d6e2f refined
haftmann
parents: 21081
diff changeset
   574
    val cs = map_filter (Consttab.lookup ((snd o snd o CodegenPackageData.get) thy))
fae2187d6e2f refined
haftmann
parents: 21081
diff changeset
   575
      (CodegenFuncgr.all funcgr);
fae2187d6e2f refined
haftmann
parents: 21081
diff changeset
   576
    val funcgr' = CodegenFuncgr.make thy cs;
20931
19d9b78218fd added code_abstype and code_constsubst
haftmann
parents: 20896
diff changeset
   577
    val qnaming = NameSpace.qualified_names NameSpace.default_naming;
20466
7c20ddbd911b explicit table with constant types
haftmann
parents: 20456
diff changeset
   578
    val algebr = ClassPackage.operational_algebra thy;
7c20ddbd911b explicit table with constant types
haftmann
parents: 20456
diff changeset
   579
    val consttab = Consts.empty
21121
fae2187d6e2f refined
haftmann
parents: 21081
diff changeset
   580
      |> fold (fn c => Consts.declare qnaming
fae2187d6e2f refined
haftmann
parents: 21081
diff changeset
   581
           ((CodegenNames.const thy c, CodegenFuncgr.typ funcgr' c), true))
fae2187d6e2f refined
haftmann
parents: 21081
diff changeset
   582
           (CodegenFuncgr.all funcgr');
20466
7c20ddbd911b explicit table with constant types
haftmann
parents: 20456
diff changeset
   583
    val algbr = (algebr, consttab);
7c20ddbd911b explicit table with constant types
haftmann
parents: 20456
diff changeset
   584
  in   
21121
fae2187d6e2f refined
haftmann
parents: 21081
diff changeset
   585
    Code.change_yield thy (CodegenThingol.start_transact init (gen thy algbr funcgr'
20600
6d75e02ed285 added codegen_data
haftmann
parents: 20485
diff changeset
   586
        (true, targets) it))
21121
fae2187d6e2f refined
haftmann
parents: 21081
diff changeset
   587
    |> fst
20466
7c20ddbd911b explicit table with constant types
haftmann
parents: 20456
diff changeset
   588
  end;
18516
4424e2bce9af slightly improved serialization
haftmann
parents: 18515
diff changeset
   589
20600
6d75e02ed285 added codegen_data
haftmann
parents: 20485
diff changeset
   590
fun codegen_term thy t =
20353
d73e49780ef2 code generator refinements
haftmann
parents: 20216
diff changeset
   591
  let
20600
6d75e02ed285 added codegen_data
haftmann
parents: 20485
diff changeset
   592
    val ct = Thm.cterm_of thy t;
21129
8e621232a865 clarified make_term interface
haftmann
parents: 21121
diff changeset
   593
    val (ct', funcgr) = CodegenFuncgr.make_term thy (K K) ct;
20835
haftmann
parents: 20699
diff changeset
   594
    val t' = Thm.term_of ct';
21121
fae2187d6e2f refined
haftmann
parents: 21081
diff changeset
   595
  in generate thy funcgr (SOME []) NONE exprgen_term' t' end;
19136
00ade10f611d some refinements
haftmann
parents: 19111
diff changeset
   596
20699
0cc77abb185a refinements in codegen serializer
haftmann
parents: 20600
diff changeset
   597
fun const_of_idf thy =
0cc77abb185a refinements in codegen serializer
haftmann
parents: 20600
diff changeset
   598
  CodegenConsts.typ_of_inst thy o the o CodegenNames.const_rev thy;
19150
1457d810b408 class package and codegen refinements
haftmann
parents: 19136
diff changeset
   599
19967
33da452f0abe slight refinements
haftmann
parents: 19956
diff changeset
   600
fun get_root_module thy =
20600
6d75e02ed285 added codegen_data
haftmann
parents: 20485
diff changeset
   601
  Code.get thy;
19042
630b8dd0b31a exported some interfaces useful for other code generator approaches
haftmann
parents: 19038
diff changeset
   602
20600
6d75e02ed285 added codegen_data
haftmann
parents: 20485
diff changeset
   603
fun eval_term thy (ref_spec, t) =
20213
2b319e945905 added eval_term
haftmann
parents: 20194
diff changeset
   604
  let
21388
9d8344cf029f corrected polymorphism check
haftmann
parents: 21161
diff changeset
   605
    val _ = (Term.fold_types o Term.fold_atyps) (fn _ =>
20401
f01ae74f29f2 more concise string serialization
haftmann
parents: 20389
diff changeset
   606
      error ("Term" ^ Sign.string_of_term thy t ^ "is polymorhpic"))
21388
9d8344cf029f corrected polymorphism check
haftmann
parents: 21161
diff changeset
   607
      t;
21121
fae2187d6e2f refined
haftmann
parents: 21081
diff changeset
   608
    val t' = codegen_term thy t;
fae2187d6e2f refined
haftmann
parents: 21081
diff changeset
   609
  in CodegenSerializer.eval_term thy (ref_spec, t') (Code.get thy) end;
18217
e0b08c9534ff added codegen package
haftmann
parents: 18169
diff changeset
   610
e0b08c9534ff added codegen package
haftmann
parents: 18169
diff changeset
   611
18516
4424e2bce9af slightly improved serialization
haftmann
parents: 18515
diff changeset
   612
20439
1bf42b262a38 code refinements
haftmann
parents: 20434
diff changeset
   613
(** toplevel interface and setup **)
18756
5eb3df798405 more general serializer
haftmann
parents: 18708
diff changeset
   614
5eb3df798405 more general serializer
haftmann
parents: 18708
diff changeset
   615
local
19150
1457d810b408 class package and codegen refinements
haftmann
parents: 19136
diff changeset
   616
20699
0cc77abb185a refinements in codegen serializer
haftmann
parents: 20600
diff changeset
   617
structure P = OuterParse
0cc77abb185a refinements in codegen serializer
haftmann
parents: 20600
diff changeset
   618
and K = OuterKeyword
0cc77abb185a refinements in codegen serializer
haftmann
parents: 20600
diff changeset
   619
20439
1bf42b262a38 code refinements
haftmann
parents: 20434
diff changeset
   620
fun code raw_cs seris thy =
18217
e0b08c9534ff added codegen package
haftmann
parents: 18169
diff changeset
   621
  let
20600
6d75e02ed285 added codegen_data
haftmann
parents: 20485
diff changeset
   622
    val cs = map (CodegenConsts.read_const thy) raw_cs;
21081
837b535137a9 dropped classop shallow namespace
haftmann
parents: 21062
diff changeset
   623
    val seris' = map (fn (target, args as _ :: _) =>
837b535137a9 dropped classop shallow namespace
haftmann
parents: 21062
diff changeset
   624
          (target, SOME (CodegenSerializer.get_serializer thy target args))
837b535137a9 dropped classop shallow namespace
haftmann
parents: 21062
diff changeset
   625
      | (target, []) => (CodegenSerializer.assert_serializer thy target, NONE)) seris;
20439
1bf42b262a38 code refinements
haftmann
parents: 20434
diff changeset
   626
    fun generate' thy = case cs
20600
6d75e02ed285 added codegen_data
haftmann
parents: 20485
diff changeset
   627
     of [] => []
20439
1bf42b262a38 code refinements
haftmann
parents: 20434
diff changeset
   628
      | _ =>
21121
fae2187d6e2f refined
haftmann
parents: 21081
diff changeset
   629
          generate thy (CodegenFuncgr.make thy cs) (case map fst seris' of [] => NONE | xs => SOME xs)
21081
837b535137a9 dropped classop shallow namespace
haftmann
parents: 21062
diff changeset
   630
            NONE (fold_map oooo ensure_def_const') cs;
20896
1484c7af6d68 cleaned up interfaces
haftmann
parents: 20855
diff changeset
   631
    fun serialize' [] code seri =
1484c7af6d68 cleaned up interfaces
haftmann
parents: 20855
diff changeset
   632
          seri NONE code 
1484c7af6d68 cleaned up interfaces
haftmann
parents: 20855
diff changeset
   633
      | serialize' cs code seri =
1484c7af6d68 cleaned up interfaces
haftmann
parents: 20855
diff changeset
   634
          seri (SOME cs) code;
20600
6d75e02ed285 added codegen_data
haftmann
parents: 20485
diff changeset
   635
    val cs = generate' thy;
20699
0cc77abb185a refinements in codegen serializer
haftmann
parents: 20600
diff changeset
   636
    val code = Code.get thy;
18217
e0b08c9534ff added codegen package
haftmann
parents: 18169
diff changeset
   637
  in
21081
837b535137a9 dropped classop shallow namespace
haftmann
parents: 21062
diff changeset
   638
    (map (serialize' cs code) (map_filter snd seris'); ())
18217
e0b08c9534ff added codegen package
haftmann
parents: 18169
diff changeset
   639
  end;
e0b08c9534ff added codegen package
haftmann
parents: 18169
diff changeset
   640
21062
876dd2695423 slight adaption
haftmann
parents: 21012
diff changeset
   641
val (codeK, code_abstypeK, code_axiomsK) =
876dd2695423 slight adaption
haftmann
parents: 21012
diff changeset
   642
  ("code_gen", "code_abstype", "code_axioms");
20699
0cc77abb185a refinements in codegen serializer
haftmann
parents: 20600
diff changeset
   643
18217
e0b08c9534ff added codegen package
haftmann
parents: 18169
diff changeset
   644
in
e0b08c9534ff added codegen package
haftmann
parents: 18169
diff changeset
   645
20439
1bf42b262a38 code refinements
haftmann
parents: 20434
diff changeset
   646
val codeP =
20835
haftmann
parents: 20699
diff changeset
   647
  OuterSyntax.improper_command codeK "generate and serialize executable code for constants" K.diag (
20439
1bf42b262a38 code refinements
haftmann
parents: 20434
diff changeset
   648
    Scan.repeat P.term
1bf42b262a38 code refinements
haftmann
parents: 20434
diff changeset
   649
    -- Scan.repeat (P.$$$ "(" |--
20896
1484c7af6d68 cleaned up interfaces
haftmann
parents: 20855
diff changeset
   650
        P.name -- P.arguments
20439
1bf42b262a38 code refinements
haftmann
parents: 20434
diff changeset
   651
        --| P.$$$ ")")
20600
6d75e02ed285 added codegen_data
haftmann
parents: 20485
diff changeset
   652
    >> (fn (raw_cs, seris) => Toplevel.keep (code raw_cs seris o Toplevel.theory_of))
20439
1bf42b262a38 code refinements
haftmann
parents: 20434
diff changeset
   653
  );
1bf42b262a38 code refinements
haftmann
parents: 20434
diff changeset
   654
20931
19d9b78218fd added code_abstype and code_constsubst
haftmann
parents: 20896
diff changeset
   655
val code_abstypeP =
19d9b78218fd added code_abstype and code_constsubst
haftmann
parents: 20896
diff changeset
   656
  OuterSyntax.command code_abstypeK "axiomatic abstypes for code generation" K.thy_decl (
19d9b78218fd added code_abstype and code_constsubst
haftmann
parents: 20896
diff changeset
   657
    (P.typ -- P.typ -- Scan.optional (P.$$$ "where" |-- Scan.repeat1
19d9b78218fd added code_abstype and code_constsubst
haftmann
parents: 20896
diff changeset
   658
        (P.term --| (P.$$$ "\\<equiv>" || P.$$$ "==") -- P.term)) [])
19d9b78218fd added code_abstype and code_constsubst
haftmann
parents: 20896
diff changeset
   659
    >> (Toplevel.theory o uncurry abstyp_e)
20428
67fa1c6ba89e refinements
haftmann
parents: 20401
diff changeset
   660
  );
67fa1c6ba89e refinements
haftmann
parents: 20401
diff changeset
   661
21062
876dd2695423 slight adaption
haftmann
parents: 21012
diff changeset
   662
val code_axiomsP =
876dd2695423 slight adaption
haftmann
parents: 21012
diff changeset
   663
  OuterSyntax.command code_axiomsK "axiomatic constant equalities for code generation" K.thy_decl (
20931
19d9b78218fd added code_abstype and code_constsubst
haftmann
parents: 20896
diff changeset
   664
    Scan.repeat1 (P.term --| (P.$$$ "\\<equiv>" || P.$$$ "==") -- P.term)
19d9b78218fd added code_abstype and code_constsubst
haftmann
parents: 20896
diff changeset
   665
    >> (Toplevel.theory o constsubst_e)
18217
e0b08c9534ff added codegen package
haftmann
parents: 18169
diff changeset
   666
  );
e0b08c9534ff added codegen package
haftmann
parents: 18169
diff changeset
   667
21062
876dd2695423 slight adaption
haftmann
parents: 21012
diff changeset
   668
val _ = OuterSyntax.add_parsers [codeP, code_abstypeP, code_axiomsP];
18217
e0b08c9534ff added codegen package
haftmann
parents: 18169
diff changeset
   669
e0b08c9534ff added codegen package
haftmann
parents: 18169
diff changeset
   670
end; (* local *)
e0b08c9534ff added codegen package
haftmann
parents: 18169
diff changeset
   671
e0b08c9534ff added codegen package
haftmann
parents: 18169
diff changeset
   672
end; (* struct *)