src/Tools/Code/code_target.ML
author haftmann
Thu Jan 30 16:09:04 2014 +0100 (2014-01-30)
changeset 55188 7ca0204ece66
parent 55153 eedd549de3ef
child 55291 82780e5f7622
child 55304 55ac31bc08a4
permissions -rw-r--r--
reduced prominence of "permissive code generation"
haftmann@37745
     1
(*  Title:      Tools/Code/code_target.ML
haftmann@24219
     2
    Author:     Florian Haftmann, TU Muenchen
haftmann@24219
     3
haftmann@38910
     4
Generic infrastructure for target language data.
haftmann@24219
     5
*)
haftmann@24219
     6
haftmann@24219
     7
signature CODE_TARGET =
haftmann@24219
     8
sig
haftmann@36471
     9
  val cert_tyco: theory -> string -> string
haftmann@36471
    10
  val read_tyco: theory -> string -> string
haftmann@36471
    11
haftmann@38933
    12
  val export_code_for: theory -> Path.T option -> string -> int option -> string -> Token.T list
haftmann@55150
    13
    -> Code_Thingol.program -> Code_Symbol.T list -> unit
haftmann@38933
    14
  val produce_code_for: theory -> string -> int option -> string -> Token.T list
haftmann@55150
    15
    -> Code_Thingol.program -> Code_Symbol.T list -> (string * string) list * string option list
haftmann@38933
    16
  val present_code_for: theory -> string -> int option -> string -> Token.T list
haftmann@55150
    17
    -> Code_Thingol.program -> Code_Symbol.T list * Code_Symbol.T list -> string
haftmann@38918
    18
  val check_code_for: theory -> string -> bool -> Token.T list
haftmann@55150
    19
    -> Code_Thingol.program -> Code_Symbol.T list -> unit
haftmann@38918
    20
haftmann@38918
    21
  val export_code: theory -> string list
haftmann@38933
    22
    -> (((string * string) * Path.T option) * Token.T list) list -> unit
haftmann@38929
    23
  val produce_code: theory -> string list
haftmann@48568
    24
    -> string -> int option -> string -> Token.T list -> (string * string) list * string option list
haftmann@55150
    25
  val present_code: theory -> string list -> Code_Symbol.T list
haftmann@38933
    26
    -> string -> int option -> string -> Token.T list -> string
haftmann@38918
    27
  val check_code: theory -> string list
haftmann@38918
    28
    -> ((string * bool) * Token.T list) list -> unit
haftmann@38918
    29
haftmann@48568
    30
  val generatedN: string
haftmann@55147
    31
  val evaluator: theory -> string -> Code_Thingol.program
haftmann@55150
    32
    -> Code_Symbol.T list -> ((string * class list) list * Code_Thingol.itype) * Code_Thingol.iterm
haftmann@48568
    33
    -> (string * string) list * string
haftmann@41344
    34
haftmann@28054
    35
  type serializer
haftmann@34152
    36
  type literals = Code_Printer.literals
haftmann@37822
    37
  val add_target: string * { serializer: serializer, literals: literals,
wenzelm@41940
    38
    check: { env_var: string, make_destination: Path.T -> Path.T, make_command: string -> string } }
wenzelm@41940
    39
    -> theory -> theory
haftmann@28663
    40
  val extend_target: string *
haftmann@55147
    41
      (string * (Code_Thingol.program -> Code_Thingol.program))
haftmann@28054
    42
    -> theory -> theory
haftmann@28054
    43
  val assert_target: theory -> string -> string
haftmann@38918
    44
  val the_literals: theory -> string -> literals
haftmann@28054
    45
  type serialization
wenzelm@36959
    46
  val parse_args: 'a parser -> Token.T list -> 'a
haftmann@38916
    47
  val serialization: (int -> Path.T option -> 'a -> unit)
haftmann@55150
    48
    -> (Code_Symbol.T list -> int -> 'a -> (string * string) list * (Code_Symbol.T -> string option))
haftmann@38925
    49
    -> 'a -> serialization
haftmann@38918
    50
  val set_default_code_width: int -> theory -> theory
haftmann@38917
    51
haftmann@52137
    52
  type ('a, 'b, 'c, 'd, 'e, 'f) symbol_attr_decl
haftmann@52138
    53
  type identifier_data
haftmann@52138
    54
  val set_identifiers: (string, string, string, string, string, string) symbol_attr_decl
haftmann@52138
    55
    -> theory -> theory
haftmann@55153
    56
  type raw_const_syntax = Code_Printer.raw_const_syntax
haftmann@34152
    57
  type tyco_syntax = Code_Printer.tyco_syntax
haftmann@55153
    58
  val set_printings: (raw_const_syntax, tyco_syntax, string, unit, unit, (string * string list)) symbol_attr_decl
haftmann@52137
    59
    -> theory -> theory
haftmann@55153
    60
  val add_const_syntax: string -> string -> raw_const_syntax option -> theory -> theory
haftmann@52137
    61
  val add_tyco_syntax: string -> string -> tyco_syntax option -> theory -> theory
haftmann@38923
    62
  val add_class_syntax: string -> class -> string option -> theory -> theory
haftmann@38923
    63
  val add_instance_syntax: string -> class * string -> unit option -> theory -> theory
haftmann@28054
    64
  val add_reserved: string -> string -> theory -> theory
haftmann@33969
    65
  val add_include: string -> string * (string * string list) option -> theory -> theory
haftmann@39646
    66
haftmann@39750
    67
  val codegen_tool: string (*theory name*) -> string (*export_code expr*) -> unit
wenzelm@43564
    68
wenzelm@43564
    69
  val setup: theory -> theory
haftmann@24219
    70
end;
haftmann@24219
    71
haftmann@28054
    72
structure Code_Target : CODE_TARGET =
haftmann@24219
    73
struct
haftmann@24219
    74
haftmann@55150
    75
open Basic_Code_Symbol;
haftmann@28054
    76
open Basic_Code_Thingol;
haftmann@34152
    77
haftmann@34152
    78
type literals = Code_Printer.literals;
haftmann@52137
    79
type ('a, 'b, 'c, 'd, 'e, 'f) symbol_attr_decl =
haftmann@52137
    80
  (string * (string * 'a option) list, string * (string * 'b option) list,
haftmann@52137
    81
    class * (string * 'c option) list, (class * class) * (string * 'd option) list,
haftmann@52137
    82
    (class * string) * (string * 'e option) list,
haftmann@52137
    83
    string * (string * 'f option) list) Code_Symbol.attr;
haftmann@52138
    84
type identifier_data = (string, string, string, string, string, string) Code_Symbol.data;
haftmann@52137
    85
haftmann@34152
    86
type tyco_syntax = Code_Printer.tyco_syntax;
haftmann@55153
    87
type raw_const_syntax = Code_Printer.raw_const_syntax;
haftmann@34152
    88
haftmann@24219
    89
haftmann@52377
    90
(** checking and parsing of symbols **)
haftmann@52377
    91
haftmann@52377
    92
fun cert_const thy const =
haftmann@52377
    93
  let
haftmann@52377
    94
    val _ = if Sign.declared_const thy const then ()
haftmann@52377
    95
      else error ("No such constant: " ^ quote const);
haftmann@52377
    96
  in const end;
haftmann@52377
    97
haftmann@52377
    98
fun cert_tyco thy tyco =
haftmann@52377
    99
  let
haftmann@52377
   100
    val _ = if Sign.declared_tyname thy tyco then ()
haftmann@52377
   101
      else error ("No such type constructor: " ^ quote tyco);
haftmann@52377
   102
  in tyco end;
haftmann@52377
   103
haftmann@52377
   104
fun read_tyco thy = #1 o dest_Type
haftmann@52377
   105
  o Proof_Context.read_type_name_proper (Proof_Context.init_global thy) true;
haftmann@52377
   106
haftmann@52377
   107
fun cert_class thy class =
haftmann@52377
   108
  let
haftmann@52377
   109
    val _ = Axclass.get_info thy class;
haftmann@52377
   110
  in class end;
haftmann@52377
   111
haftmann@52377
   112
fun read_class thy = Proof_Context.read_class (Proof_Context.init_global thy);
haftmann@52377
   113
haftmann@52377
   114
val parse_classrel_ident = Parse.class --| @{keyword "<"} -- Parse.class;
haftmann@52377
   115
haftmann@52377
   116
fun cert_inst thy (class, tyco) =
haftmann@52377
   117
  (cert_class thy class, cert_tyco thy tyco);
haftmann@52377
   118
haftmann@52377
   119
fun read_inst thy (raw_tyco, raw_class) =
haftmann@55147
   120
  (read_tyco thy raw_tyco, read_class thy raw_class);
haftmann@52377
   121
haftmann@52377
   122
val parse_inst_ident = Parse.xname --| @{keyword "::"} -- Parse.class;
haftmann@52377
   123
haftmann@52377
   124
fun cert_syms thy =
haftmann@52377
   125
  Code_Symbol.map_attr (apfst (cert_const thy)) (apfst (cert_tyco thy))
haftmann@52377
   126
    (apfst (cert_class thy)) ((apfst o pairself) (cert_class thy)) (apfst (cert_inst thy)) I;
haftmann@52377
   127
haftmann@52377
   128
fun read_syms thy =
haftmann@52377
   129
  Code_Symbol.map_attr (apfst (Code.read_const thy)) (apfst (read_tyco thy))
haftmann@52377
   130
    (apfst (read_class thy)) ((apfst o pairself) (read_class thy)) (apfst (read_inst thy)) I;
haftmann@52377
   131
haftmann@52377
   132
fun check_name is_module s =
haftmann@52377
   133
  let
haftmann@52377
   134
    val _ = if s = "" then error "Bad empty code name" else ();
haftmann@52377
   135
    val xs = Long_Name.explode s;
haftmann@52377
   136
    val xs' = if is_module
haftmann@52377
   137
        then map (Name.desymbolize true) xs
haftmann@52377
   138
      else if length xs < 2
haftmann@52377
   139
        then error ("Bad code name without module component: " ^ quote s)
haftmann@52377
   140
      else
haftmann@52377
   141
        let
haftmann@52377
   142
          val (ys, y) = split_last xs;
haftmann@52377
   143
          val ys' = map (Name.desymbolize true) ys;
haftmann@52377
   144
          val y' = Name.desymbolize false y;
haftmann@52377
   145
        in ys' @ [y'] end;
haftmann@52377
   146
  in if xs' = xs
haftmann@52377
   147
    then s
haftmann@52377
   148
    else error ("Invalid code name: " ^ quote s ^ "\n"
haftmann@52377
   149
      ^ "better try " ^ quote (Long_Name.implode xs'))
haftmann@52377
   150
  end;
haftmann@52377
   151
haftmann@52377
   152
haftmann@52137
   153
(** serializations and serializer **)
haftmann@52137
   154
haftmann@52137
   155
(* serialization: abstract nonsense to cover different destinies for generated code *)
haftmann@24219
   156
haftmann@55150
   157
datatype destination = Export of Path.T option | Produce | Present of Code_Symbol.T list;
haftmann@55150
   158
type serialization = int -> destination -> ((string * string) list * (Code_Symbol.T -> string option)) option;
haftmann@27014
   159
haftmann@39659
   160
fun serialization output _ content width (Export some_path) =
haftmann@39661
   161
      (output width some_path content; NONE)
haftmann@39661
   162
  | serialization _ string content width Produce =
haftmann@39661
   163
      string [] width content |> SOME
haftmann@55147
   164
  | serialization _ string content width (Present syms) =
haftmann@55147
   165
     string syms width content
haftmann@48568
   166
     |> (apfst o map o apsnd) (Pretty.output (SOME width) o Pretty.str)
haftmann@39661
   167
     |> SOME;
haftmann@39659
   168
haftmann@39661
   169
fun export some_path f = (f (Export some_path); ());
haftmann@39661
   170
fun produce f = the (f Produce);
haftmann@55147
   171
fun present syms f = space_implode "\n\n" (map snd (fst (the (f (Present syms)))));
haftmann@38917
   172
haftmann@24219
   173
haftmann@52137
   174
(* serializers: functions producing serializations *)
haftmann@28054
   175
haftmann@39142
   176
type serializer = Token.T list
haftmann@52138
   177
  -> Proof.context
haftmann@39142
   178
  -> {
haftmann@52138
   179
    module_name: string,
haftmann@38926
   180
    reserved_syms: string list,
haftmann@52138
   181
    identifiers: identifier_data,
haftmann@38926
   182
    includes: (string * Pretty.T) list,
haftmann@38926
   183
    class_syntax: string -> string option,
haftmann@38926
   184
    tyco_syntax: string -> Code_Printer.tyco_syntax option,
haftmann@55153
   185
    const_syntax: string -> Code_Printer.const_syntax option }
haftmann@41342
   186
  -> Code_Thingol.program
haftmann@28054
   187
  -> serialization;
haftmann@27000
   188
haftmann@52137
   189
datatype description =
haftmann@52137
   190
    Fundamental of { serializer: serializer,
haftmann@38910
   191
      literals: literals,
haftmann@37822
   192
      check: { env_var: string, make_destination: Path.T -> Path.T,
wenzelm@41940
   193
        make_command: string -> string } }
haftmann@38909
   194
  | Extension of string *
haftmann@55147
   195
      (Code_Thingol.program -> Code_Thingol.program);
haftmann@28054
   196
haftmann@52137
   197
haftmann@52137
   198
(** theory data **)
haftmann@52137
   199
haftmann@28054
   200
datatype target = Target of {
haftmann@28054
   201
  serial: serial,
haftmann@37821
   202
  description: description,
haftmann@28054
   203
  reserved: string list,
haftmann@52138
   204
  identifiers: identifier_data,
haftmann@55153
   205
  printings: (Code_Printer.const_syntax, Code_Printer.tyco_syntax, string, unit, unit,
haftmann@52137
   206
    (Pretty.T * string list)) Code_Symbol.data
haftmann@28054
   207
};
haftmann@27103
   208
haftmann@52138
   209
fun make_target ((serial, description), (reserved, (identifiers, printings))) =
wenzelm@43564
   210
  Target { serial = serial, description = description, reserved = reserved,
haftmann@52138
   211
    identifiers = identifiers, printings = printings };
haftmann@52138
   212
fun map_target f (Target { serial, description, reserved, identifiers, printings }) =
haftmann@52138
   213
  make_target (f ((serial, description), (reserved, (identifiers, printings))));
haftmann@37821
   214
fun merge_target strict target (Target { serial = serial1, description = description,
haftmann@52138
   215
  reserved = reserved1, identifiers = identifiers1, printings = printings1 },
haftmann@37821
   216
    Target { serial = serial2, description = _,
haftmann@52138
   217
      reserved = reserved2, identifiers = identifiers2, printings = printings2 }) =
haftmann@28054
   218
  if serial1 = serial2 orelse not strict then
haftmann@52138
   219
    make_target ((serial1, description), (merge (op =) (reserved1, reserved2),
haftmann@52138
   220
      (Code_Symbol.merge_data (identifiers1, identifiers2),
haftmann@52138
   221
        Code_Symbol.merge_data (printings1, printings2))))
haftmann@28054
   222
  else
haftmann@37821
   223
    error ("Incompatible targets: " ^ quote target);
haftmann@27103
   224
haftmann@37821
   225
fun the_description (Target { description, ... }) = description;
haftmann@28054
   226
fun the_reserved (Target { reserved, ... }) = reserved;
haftmann@52138
   227
fun the_identifiers (Target { identifiers , ... }) = identifiers;
haftmann@52137
   228
fun the_printings (Target { printings, ... }) = printings;
haftmann@27437
   229
haftmann@34248
   230
structure Targets = Theory_Data
haftmann@34071
   231
(
haftmann@54890
   232
  type T = target Symtab.table * int;
haftmann@54890
   233
  val empty = (Symtab.empty, 80);
haftmann@34071
   234
  val extend = I;
haftmann@54890
   235
  fun merge ((target1, width1), (target2, width2)) : T =
haftmann@54890
   236
    (Symtab.join (merge_target true) (target1, target2), Int.max (width1, width2));
haftmann@34071
   237
);
haftmann@27436
   238
haftmann@54890
   239
fun assert_target thy target = if Symtab.defined (fst (Targets.get thy)) target
haftmann@33969
   240
  then target
haftmann@33969
   241
  else error ("Unknown code target language: " ^ quote target);
haftmann@28054
   242
haftmann@28054
   243
fun put_target (target, seri) thy =
haftmann@27304
   244
  let
haftmann@54890
   245
    val lookup_target = Symtab.lookup (fst (Targets.get thy));
haftmann@28054
   246
    val _ = case seri
haftmann@37821
   247
     of Extension (super, _) => if is_some (lookup_target super) then ()
haftmann@28054
   248
          else error ("Unknown code target language: " ^ quote super)
haftmann@28054
   249
      | _ => ();
haftmann@37821
   250
    val overwriting = case (Option.map the_description o lookup_target) target
haftmann@28663
   251
     of NONE => false
haftmann@37821
   252
      | SOME (Extension _) => true
haftmann@37821
   253
      | SOME (Fundamental _) => (case seri
haftmann@37821
   254
         of Extension _ => error ("Will not overwrite existing target " ^ quote target)
haftmann@28663
   255
          | _ => true);
haftmann@28663
   256
    val _ = if overwriting
haftmann@28054
   257
      then warning ("Overwriting existing target " ^ quote target)
wenzelm@43564
   258
      else ();
haftmann@28054
   259
  in
haftmann@28054
   260
    thy
haftmann@54890
   261
    |> (Targets.map o apfst o Symtab.update)
haftmann@52138
   262
        (target, make_target ((serial (), seri),
haftmann@52138
   263
          ([], (Code_Symbol.empty_data, Code_Symbol.empty_data))))
haftmann@28054
   264
  end;
haftmann@27436
   265
haftmann@37821
   266
fun add_target (target, seri) = put_target (target, Fundamental seri);
haftmann@28054
   267
fun extend_target (target, (super, modify)) =
haftmann@37821
   268
  put_target (target, Extension (super, modify));
haftmann@27436
   269
haftmann@28054
   270
fun map_target_data target f thy =
haftmann@27436
   271
  let
haftmann@28054
   272
    val _ = assert_target thy target;
haftmann@28054
   273
  in
haftmann@28054
   274
    thy
haftmann@54890
   275
    |> (Targets.map o apfst o Symtab.map_entry target o map_target o apsnd) f
haftmann@28054
   276
  end;
haftmann@27304
   277
haftmann@28054
   278
fun map_reserved target =
haftmann@52138
   279
  map_target_data target o apfst;
haftmann@52138
   280
fun map_identifiers target =
haftmann@52138
   281
  map_target_data target o apsnd o apfst;
haftmann@52137
   282
fun map_printings target =
haftmann@52137
   283
  map_target_data target o apsnd o apsnd;
haftmann@27000
   284
haftmann@34248
   285
fun set_default_code_width k = (Targets.map o apsnd) (K k);
haftmann@34071
   286
haftmann@27000
   287
haftmann@34021
   288
(** serializer usage **)
haftmann@34021
   289
haftmann@34021
   290
(* montage *)
haftmann@34021
   291
haftmann@37822
   292
fun the_fundamental thy =
haftmann@34021
   293
  let
haftmann@54890
   294
    val (targets, _) = Targets.get thy;
haftmann@37822
   295
    fun fundamental target = case Symtab.lookup targets target
haftmann@37821
   296
     of SOME data => (case the_description data
haftmann@37822
   297
         of Fundamental data => data
haftmann@37822
   298
          | Extension (super, _) => fundamental super)
haftmann@34021
   299
      | NONE => error ("Unknown code target language: " ^ quote target);
haftmann@37822
   300
  in fundamental end;
haftmann@37822
   301
haftmann@37822
   302
fun the_literals thy = #literals o the_fundamental thy;
haftmann@34021
   303
haftmann@39817
   304
fun collapse_hierarchy thy =
haftmann@38927
   305
  let
haftmann@54890
   306
    val (targets, _) = Targets.get thy;
wenzelm@43564
   307
    fun collapse target =
haftmann@38927
   308
      let
haftmann@38927
   309
        val data = case Symtab.lookup targets target
haftmann@38927
   310
         of SOME data => data
haftmann@38927
   311
          | NONE => error ("Unknown code target language: " ^ quote target);
haftmann@38927
   312
      in case the_description data
haftmann@55147
   313
       of Fundamental _ => (I, data)
haftmann@38927
   314
        | Extension (super, modify) => let
haftmann@39817
   315
            val (modify', data') = collapse super
haftmann@55147
   316
          in (modify' #> modify, merge_target false target (data', data)) end
haftmann@38927
   317
      end;
haftmann@39817
   318
  in collapse end;
haftmann@39817
   319
haftmann@39817
   320
local
haftmann@39817
   321
haftmann@39817
   322
fun activate_target thy target =
haftmann@39817
   323
  let
haftmann@54890
   324
    val (_, default_width) = Targets.get thy;
haftmann@39817
   325
    val (modify, data) = collapse_hierarchy thy target;
haftmann@54890
   326
  in (default_width, data, modify) end;
haftmann@38927
   327
haftmann@55147
   328
fun project_program thy syms_hidden syms1 program2 =
haftmann@38927
   329
  let
wenzelm@42361
   330
    val ctxt = Proof_Context.init_global thy;
haftmann@55147
   331
    val syms2 = subtract (op =) syms_hidden syms1;
haftmann@55147
   332
    val program3 = Code_Symbol.Graph.restrict (not o member (op =) syms_hidden) program2;
haftmann@55147
   333
    val syms4 = Code_Symbol.Graph.all_succs program3 syms2;
haftmann@54890
   334
    val unimplemented = Code_Thingol.unimplemented program3;
wenzelm@42359
   335
    val _ =
haftmann@54889
   336
      if null unimplemented then ()
wenzelm@42359
   337
      else error ("No code equations for " ^
haftmann@54889
   338
        commas (map (Proof_Context.extern_const ctxt) unimplemented));
haftmann@55147
   339
    val program4 = Code_Symbol.Graph.restrict (member (op =) syms4) program3;
haftmann@55147
   340
  in (syms4, program4) end;
haftmann@38927
   341
haftmann@55149
   342
fun prepare_serializer thy (serializer : serializer) reserved identifiers
haftmann@55147
   343
    printings module_name args proto_program syms =
haftmann@38927
   344
  let
haftmann@55149
   345
    val syms_hidden = Code_Symbol.symbols_of printings;
haftmann@55147
   346
    val (syms_all, program) = project_program thy syms_hidden syms proto_program;
haftmann@39484
   347
    fun select_include (name, (content, cs)) =
haftmann@55150
   348
      if null cs orelse exists (fn c => member (op =) syms_all (Constant c)) cs
haftmann@39484
   349
      then SOME (name, content) else NONE;
haftmann@52137
   350
    val includes = map_filter select_include (Code_Symbol.dest_module_data printings);
haftmann@34021
   351
  in
haftmann@52138
   352
    (serializer args (Proof_Context.init_global thy) {
haftmann@52138
   353
      module_name = module_name,
haftmann@38926
   354
      reserved_syms = reserved,
haftmann@52138
   355
      identifiers = identifiers,
haftmann@38926
   356
      includes = includes,
haftmann@55149
   357
      const_syntax = Code_Symbol.lookup_constant_data printings,
haftmann@55149
   358
      tyco_syntax = Code_Symbol.lookup_type_constructor_data printings,
haftmann@55149
   359
      class_syntax = Code_Symbol.lookup_type_class_data printings },
haftmann@41342
   360
      program)
haftmann@34021
   361
  end;
haftmann@34021
   362
haftmann@55147
   363
fun mount_serializer thy target some_width module_name args program syms =
haftmann@34021
   364
  let
haftmann@54890
   365
    val (default_width, data, modify) = activate_target thy target;
haftmann@38921
   366
    val serializer = case the_description data
haftmann@38927
   367
     of Fundamental seri => #serializer seri;
haftmann@52138
   368
    val (prepared_serializer, prepared_program) =
haftmann@55149
   369
      prepare_serializer thy serializer
haftmann@52138
   370
        (the_reserved data) (the_identifiers data) (the_printings data)
haftmann@55147
   371
        module_name args (modify program) syms
haftmann@34071
   372
    val width = the_default default_width some_width;
haftmann@41344
   373
  in (fn program => prepared_serializer program width, prepared_program) end;
haftmann@41344
   374
haftmann@55147
   375
fun invoke_serializer thy target some_width module_name args program syms =
haftmann@41344
   376
  let
haftmann@53413
   377
    val check = if module_name = "" then I else check_name true;
haftmann@41344
   378
    val (mounted_serializer, prepared_program) = mount_serializer thy
haftmann@55147
   379
      target some_width (check module_name) args program syms;
haftmann@41344
   380
  in mounted_serializer prepared_program end;
haftmann@34021
   381
haftmann@52138
   382
fun assert_module_name "" = error "Empty module name not allowed here"
haftmann@38933
   383
  | assert_module_name module_name = module_name;
haftmann@38933
   384
haftmann@48426
   385
fun using_master_directory thy =
haftmann@48426
   386
  Option.map (Path.append (File.pwd ()) o Path.append (Thy_Load.master_directory thy));
haftmann@48371
   387
haftmann@34021
   388
in
haftmann@34021
   389
haftmann@48568
   390
val generatedN = "Generated_Code";
haftmann@48568
   391
haftmann@39484
   392
fun export_code_for thy some_path target some_width module_name args =
haftmann@48371
   393
  export (using_master_directory thy some_path)
haftmann@55147
   394
  oo invoke_serializer thy target some_width module_name args;
haftmann@38918
   395
haftmann@39484
   396
fun produce_code_for thy target some_width module_name args =
haftmann@39102
   397
  let
haftmann@41344
   398
    val serializer = invoke_serializer thy target some_width (assert_module_name module_name) args;
haftmann@55147
   399
  in fn program => fn syms =>
haftmann@55147
   400
    produce (serializer program syms) |> apsnd (fn deresolve => map deresolve syms)
haftmann@39484
   401
  end;
haftmann@38929
   402
haftmann@39484
   403
fun present_code_for thy target some_width module_name args =
haftmann@39484
   404
  let
haftmann@41344
   405
    val serializer = invoke_serializer thy target some_width (assert_module_name module_name) args;
haftmann@55147
   406
  in fn program => fn (syms, selects) =>
haftmann@55147
   407
    present selects (serializer program syms)
haftmann@39484
   408
  end;
haftmann@38918
   409
haftmann@55147
   410
fun check_code_for thy target strict args program syms =
haftmann@37824
   411
  let
haftmann@37824
   412
    val { env_var, make_destination, make_command } =
haftmann@37824
   413
      (#check o the_fundamental thy) target;
wenzelm@41939
   414
    fun ext_check p =
wenzelm@43564
   415
      let
haftmann@37824
   416
        val destination = make_destination p;
haftmann@41344
   417
        val _ = export (SOME destination) (invoke_serializer thy target (SOME 80)
haftmann@55147
   418
          generatedN args program syms);
haftmann@48568
   419
        val cmd = make_command generatedN;
wenzelm@43850
   420
      in
wenzelm@43850
   421
        if Isabelle_System.bash ("cd " ^ File.shell_path p ^ " && " ^ cmd ^ " 2>&1") <> 0
haftmann@37824
   422
        then error ("Code check failed for " ^ target ^ ": " ^ cmd)
haftmann@37824
   423
        else ()
haftmann@37824
   424
      end;
wenzelm@43850
   425
  in
wenzelm@43850
   426
    if getenv env_var = ""
haftmann@37825
   427
    then if strict
haftmann@37825
   428
      then error (env_var ^ " not set; cannot check code for " ^ target)
haftmann@37825
   429
      else warning (env_var ^ " not set; skipped checking code for " ^ target)
wenzelm@41939
   430
    else Isabelle_System.with_tmp_dir "Code_Test" ext_check
haftmann@37824
   431
  end;
haftmann@37824
   432
haftmann@55147
   433
fun evaluation mounted_serializer prepared_program syms ((vs, ty), t) =
haftmann@41344
   434
  let
haftmann@41344
   435
    val _ = if Code_Thingol.contains_dict_var t then
haftmann@41344
   436
      error "Term to be evaluated contains free dictionaries" else ();
wenzelm@43324
   437
    val v' = singleton (Name.variant_list (map fst vs)) "a";
haftmann@41344
   438
    val vs' = (v', []) :: vs;
haftmann@55147
   439
    val ty' = ITyVar v' `-> ty;
haftmann@41344
   440
    val program = prepared_program
haftmann@55147
   441
      |> Code_Symbol.Graph.new_node (Code_Symbol.value,
haftmann@55147
   442
          Code_Thingol.Fun (((vs', ty'), [(([IVar NONE], t), (NONE, true))]), NONE))
haftmann@55147
   443
      |> fold (curry (perhaps o try o
haftmann@55147
   444
          Code_Symbol.Graph.add_edge) Code_Symbol.value) syms;
haftmann@41344
   445
    val (program_code, deresolve) = produce (mounted_serializer program);
haftmann@55147
   446
    val value_name = the (deresolve Code_Symbol.value);
haftmann@55147
   447
  in (program_code, value_name) end;
haftmann@41344
   448
haftmann@55147
   449
fun evaluator thy target program syms =
haftmann@41344
   450
  let
haftmann@55147
   451
    val (mounted_serializer, prepared_program) =
haftmann@55147
   452
      mount_serializer thy target NONE generatedN [] program syms;
haftmann@55147
   453
  in evaluation mounted_serializer prepared_program syms end;
haftmann@41344
   454
haftmann@34021
   455
end; (* local *)
haftmann@34021
   456
haftmann@34021
   457
haftmann@34021
   458
(* code generation *)
haftmann@34021
   459
haftmann@38918
   460
fun prep_destination "" = NONE
haftmann@38918
   461
  | prep_destination s = SOME (Path.explode s);
haftmann@38918
   462
haftmann@34021
   463
fun export_code thy cs seris =
haftmann@34021
   464
  let
haftmann@55188
   465
    val program = Code_Thingol.consts_program thy cs;
haftmann@38918
   466
    val _ = map (fn (((target, module_name), some_path), args) =>
haftmann@55150
   467
      export_code_for thy some_path target NONE module_name args program (map Constant cs)) seris;
haftmann@34021
   468
  in () end;
haftmann@34021
   469
haftmann@55188
   470
fun export_code_cmd raw_cs seris thy = export_code thy
haftmann@55188
   471
  (Code_Thingol.read_const_exprs thy raw_cs)
haftmann@38918
   472
  ((map o apfst o apsnd) prep_destination seris);
haftmann@38918
   473
haftmann@38929
   474
fun produce_code thy cs target some_width some_module_name args =
haftmann@38918
   475
  let
haftmann@55188
   476
    val program = Code_Thingol.consts_program thy cs;
haftmann@55150
   477
  in produce_code_for thy target some_width some_module_name args program (map Constant cs) end;
haftmann@38929
   478
haftmann@55147
   479
fun present_code thy cs syms target some_width some_module_name args =
haftmann@38929
   480
  let
haftmann@55188
   481
    val program = Code_Thingol.consts_program thy cs;
haftmann@55150
   482
  in present_code_for thy target some_width some_module_name args program (map Constant cs, syms) end;
haftmann@34021
   483
haftmann@37824
   484
fun check_code thy cs seris =
haftmann@37824
   485
  let
haftmann@55188
   486
    val program = Code_Thingol.consts_program thy cs;
haftmann@38918
   487
    val _ = map (fn ((target, strict), args) =>
haftmann@55150
   488
      check_code_for thy target strict args program (map Constant cs)) seris;
haftmann@37824
   489
  in () end;
haftmann@37824
   490
haftmann@55188
   491
fun check_code_cmd raw_cs seris thy = check_code thy
haftmann@55188
   492
  (Code_Thingol.read_const_exprs thy raw_cs) seris;
haftmann@37824
   493
haftmann@39480
   494
local
haftmann@39480
   495
haftmann@39480
   496
val parse_const_terms = Scan.repeat1 Args.term
haftmann@39480
   497
  >> (fn ts => fn thy => map (Code.check_const thy) ts);
haftmann@39480
   498
haftmann@55147
   499
fun parse_names category parse internalize mark_symbol =
haftmann@39480
   500
  Scan.lift (Args.parens (Args.$$$ category)) |-- Scan.repeat1 parse
haftmann@55147
   501
  >> (fn xs => fn thy => map (mark_symbol o internalize thy) xs);
wenzelm@43564
   502
haftmann@39480
   503
val parse_consts = parse_names "consts" Args.term
haftmann@55150
   504
  Code.check_const Constant;
haftmann@39480
   505
haftmann@39480
   506
val parse_types = parse_names "types" (Scan.lift Args.name)
haftmann@55150
   507
  Sign.intern_type Type_Constructor;
haftmann@39480
   508
haftmann@39480
   509
val parse_classes = parse_names "classes" (Scan.lift Args.name)
haftmann@55150
   510
  Sign.intern_class Type_Class;
haftmann@39480
   511
haftmann@39480
   512
val parse_instances = parse_names "instances" (Scan.lift (Args.name --| Args.$$$ "::" -- Args.name))
haftmann@55147
   513
  (fn thy => fn (raw_tyco, raw_class) => (Sign.intern_class thy raw_tyco, Sign.intern_type thy raw_class))
haftmann@55150
   514
    Class_Instance;
haftmann@39480
   515
haftmann@39480
   516
in
haftmann@39480
   517
wenzelm@43564
   518
val antiq_setup =
wenzelm@43564
   519
  Thy_Output.antiquotation @{binding code_stmts}
wenzelm@43564
   520
    (parse_const_terms --
wenzelm@43564
   521
      Scan.repeat (parse_consts || parse_types || parse_classes || parse_instances)
wenzelm@43564
   522
      -- Scan.lift (Args.parens (Args.name -- Scan.option Parse.int)))
wenzelm@43564
   523
    (fn {context = ctxt, ...} => fn ((mk_cs, mk_stmtss), (target, some_width)) =>
wenzelm@43564
   524
      let val thy = Proof_Context.theory_of ctxt in
wenzelm@43564
   525
        present_code thy (mk_cs thy)
haftmann@55147
   526
          (maps (fn f => f thy) mk_stmtss)
wenzelm@43564
   527
          target some_width "Example" []
wenzelm@43564
   528
      end);
haftmann@39480
   529
haftmann@39480
   530
end;
haftmann@39480
   531
haftmann@34021
   532
haftmann@28054
   533
(** serializer configuration **)
haftmann@27000
   534
haftmann@52138
   535
(* reserved symbol names *)
haftmann@52138
   536
haftmann@52138
   537
fun add_reserved target sym thy =
haftmann@52138
   538
  let
haftmann@52138
   539
    val (_, data) = collapse_hierarchy thy target;
haftmann@52138
   540
    val _ = if member (op =) (the_reserved data) sym
haftmann@52138
   541
      then error ("Reserved symbol " ^ quote sym ^ " already declared")
haftmann@52138
   542
      else ();
haftmann@52138
   543
  in
haftmann@52138
   544
    thy
haftmann@52138
   545
    |> map_reserved target (insert (op =) sym)
haftmann@52138
   546
  end;
haftmann@52138
   547
haftmann@52138
   548
haftmann@52377
   549
(* checking of syntax *)
haftmann@52138
   550
haftmann@55149
   551
fun check_const_syntax thy target c syn =
haftmann@52137
   552
  if Code_Printer.requires_args syn > Code.args_number thy c
haftmann@52137
   553
  then error ("Too many arguments in syntax for constant " ^ quote c)
haftmann@55153
   554
  else Code_Printer.prep_const_syntax thy (the_literals thy target) c syn;
haftmann@52137
   555
haftmann@55149
   556
fun check_tyco_syntax thy target tyco syn =
haftmann@52137
   557
  if fst syn <> Sign.arity_number thy tyco
haftmann@52137
   558
  then error ("Number of arguments mismatch in syntax for type constructor " ^ quote tyco)
haftmann@52137
   559
  else syn;
haftmann@34071
   560
haftmann@52138
   561
haftmann@52138
   562
(* custom symbol names *)
haftmann@52138
   563
wenzelm@52218
   564
fun arrange_name_decls x =
haftmann@52138
   565
  let
haftmann@52138
   566
    fun arrange is_module (sym, target_names) = map (fn (target, some_name) =>
haftmann@52138
   567
      (target, (sym, Option.map (check_name is_module) some_name))) target_names;
haftmann@52138
   568
  in
haftmann@52138
   569
    Code_Symbol.maps_attr' (arrange false) (arrange false) (arrange false)
wenzelm@52218
   570
      (arrange false) (arrange false) (arrange true) x
haftmann@52138
   571
  end;
haftmann@52138
   572
haftmann@52138
   573
fun cert_name_decls thy = cert_syms thy #> arrange_name_decls;
haftmann@52138
   574
haftmann@52138
   575
fun read_name_decls thy = read_syms thy #> arrange_name_decls;
haftmann@52138
   576
haftmann@52138
   577
fun set_identifier (target, sym_name) = map_identifiers target (Code_Symbol.set_data sym_name);
haftmann@52138
   578
haftmann@52138
   579
fun gen_set_identifiers prep_name_decl raw_name_decls thy =
haftmann@52138
   580
  fold set_identifier (prep_name_decl thy raw_name_decls) thy;
haftmann@52138
   581
haftmann@52138
   582
val set_identifiers = gen_set_identifiers cert_name_decls;
haftmann@52138
   583
val set_identifiers_cmd = gen_set_identifiers read_name_decls;
haftmann@52138
   584
wenzelm@54312
   585
fun add_module_alias_cmd target aliasses thy =
haftmann@52435
   586
  let
haftmann@52435
   587
    val _ = legacy_feature "prefer \"code_identifier\" over \"code_modulename\"";
haftmann@52435
   588
  in
haftmann@52435
   589
    fold (fn (sym, name) => set_identifier
wenzelm@54312
   590
      (target, Code_Symbol.Module (sym, if name = "" then NONE else SOME (check_name true name))))
wenzelm@54312
   591
      aliasses thy
haftmann@52435
   592
  end;
haftmann@52138
   593
haftmann@52138
   594
haftmann@52138
   595
(* custom printings *)
haftmann@52138
   596
haftmann@52137
   597
fun arrange_printings prep_const thy =
haftmann@52137
   598
  let
haftmann@52137
   599
    fun arrange check (sym, target_syns) =
haftmann@55149
   600
      map (fn (target, some_syn) => (target, (sym, Option.map (check thy target sym) some_syn))) target_syns;
haftmann@52137
   601
  in
haftmann@52137
   602
    Code_Symbol.maps_attr'
haftmann@52137
   603
      (arrange check_const_syntax) (arrange check_tyco_syntax)
haftmann@55149
   604
        (arrange ((K o K o K) I)) (arrange ((K o K o K) I)) (arrange ((K o K o K) I))
haftmann@55149
   605
        (arrange (fn thy => fn _ => fn _ => fn (raw_content, raw_cs) =>
haftmann@52137
   606
          (Code_Printer.str raw_content, map (prep_const thy) raw_cs)))
haftmann@52137
   607
  end;
haftmann@52137
   608
haftmann@52137
   609
fun cert_printings thy = cert_syms thy #> arrange_printings cert_const thy;
haftmann@24219
   610
haftmann@52137
   611
fun read_printings thy = read_syms thy #> arrange_printings Code.read_const thy;
haftmann@52137
   612
haftmann@52137
   613
fun set_printing (target, sym_syn) = map_printings target (Code_Symbol.set_data sym_syn);
haftmann@52137
   614
haftmann@52137
   615
fun gen_set_printings prep_print_decl raw_print_decls thy =
haftmann@52137
   616
  fold set_printing (prep_print_decl thy raw_print_decls) thy;
haftmann@52137
   617
haftmann@52137
   618
val set_printings = gen_set_printings cert_printings;
haftmann@52137
   619
val set_printings_cmd = gen_set_printings read_printings;
haftmann@52137
   620
haftmann@52137
   621
fun gen_add_syntax Symbol prep_x prep_syn target raw_x some_raw_syn thy =
haftmann@24992
   622
  let
haftmann@52435
   623
    val _ = legacy_feature "prefer \"code_printing\" for custom serialisations"
haftmann@52137
   624
    val x = prep_x thy raw_x;
haftmann@55149
   625
  in set_printing (target, Symbol (x, Option.map (prep_syn thy target x) some_raw_syn)) thy end;
haftmann@28926
   626
haftmann@52137
   627
fun gen_add_const_syntax prep_const =
haftmann@55150
   628
  gen_add_syntax Constant prep_const check_const_syntax;
haftmann@52137
   629
haftmann@52137
   630
fun gen_add_tyco_syntax prep_tyco =
haftmann@55150
   631
  gen_add_syntax Type_Constructor prep_tyco check_tyco_syntax;
haftmann@52137
   632
haftmann@52137
   633
fun gen_add_class_syntax prep_class =
haftmann@55150
   634
  gen_add_syntax Type_Class prep_class ((K o K o K) I);
haftmann@24219
   635
haftmann@52137
   636
fun gen_add_instance_syntax prep_inst =
haftmann@55150
   637
  gen_add_syntax Class_Instance prep_inst ((K o K o K) I);
haftmann@52137
   638
haftmann@52137
   639
fun gen_add_include prep_const target (name, some_content) thy =
haftmann@55150
   640
  gen_add_syntax Module (K I)
haftmann@55149
   641
    (fn thy => fn _ => fn _ => fn (raw_content, raw_cs) => (Code_Printer.str raw_content, map (prep_const thy) raw_cs))
haftmann@52137
   642
    target name some_content thy;
haftmann@52137
   643
haftmann@52137
   644
haftmann@27103
   645
(* concrete syntax *)
haftmann@27000
   646
haftmann@34021
   647
local
haftmann@34021
   648
haftmann@47576
   649
fun zip_list (x :: xs) f g =
haftmann@34152
   650
  f
haftmann@37881
   651
  :|-- (fn y =>
haftmann@34152
   652
    fold_map (fn x => g |-- f >> pair x) xs
haftmann@37881
   653
    :|-- (fn xys => pair ((x, y) :: xys)));
haftmann@34152
   654
haftmann@37881
   655
fun process_multi_syntax parse_thing parse_syntax change =
haftmann@37881
   656
  (Parse.and_list1 parse_thing
wenzelm@46949
   657
  :|-- (fn things => Scan.repeat1 (@{keyword "("} |-- Parse.name --
haftmann@52068
   658
        (zip_list things (Scan.option parse_syntax) @{keyword "and"}) --| @{keyword ")"})))
haftmann@37881
   659
  >> (Toplevel.theory oo fold)
haftmann@37881
   660
    (fn (target, syns) => fold (fn (raw_x, syn) => change target raw_x syn) syns);
haftmann@24219
   661
haftmann@24219
   662
in
haftmann@24219
   663
haftmann@52137
   664
val add_reserved = add_reserved;
haftmann@52137
   665
val add_const_syntax = gen_add_const_syntax (K I);
haftmann@52137
   666
val add_tyco_syntax = gen_add_tyco_syntax cert_tyco;
haftmann@38923
   667
val add_class_syntax = gen_add_class_syntax cert_class;
haftmann@38923
   668
val add_instance_syntax = gen_add_instance_syntax cert_inst;
haftmann@52137
   669
val add_include = gen_add_include (K I);
haftmann@24219
   670
haftmann@52137
   671
val add_const_syntax_cmd = gen_add_const_syntax Code.read_const;
haftmann@52137
   672
val add_tyco_syntax_cmd = gen_add_tyco_syntax read_tyco;
haftmann@38923
   673
val add_class_syntax_cmd = gen_add_class_syntax read_class;
haftmann@38923
   674
val add_instance_syntax_cmd = gen_add_instance_syntax read_inst;
haftmann@52137
   675
val add_include_cmd = gen_add_include Code.read_const;
haftmann@24219
   676
haftmann@28054
   677
fun parse_args f args =
wenzelm@36959
   678
  case Scan.read Token.stopper f args
haftmann@28054
   679
   of SOME x => x
haftmann@28054
   680
    | NONE => error "Bad serializer arguments";
haftmann@28054
   681
haftmann@28054
   682
haftmann@27304
   683
(** Isar setup **)
haftmann@27103
   684
haftmann@52137
   685
fun parse_single_symbol_pragma parse_keyword parse_isa parse_target =
haftmann@52434
   686
  parse_keyword |-- Parse.!!! (parse_isa --| (@{keyword "\<rightharpoonup>"} || @{keyword "=>"})
haftmann@52434
   687
    -- Parse.and_list1 (@{keyword "("} |-- (Parse.name --| @{keyword ")"} -- Scan.option parse_target)));
haftmann@52137
   688
haftmann@52137
   689
fun parse_symbol_pragma parse_const parse_tyco parse_class parse_classrel parse_inst parse_module =
wenzelm@52801
   690
  parse_single_symbol_pragma @{keyword "constant"} Parse.term parse_const
haftmann@55150
   691
    >> Constant
haftmann@52377
   692
  || parse_single_symbol_pragma @{keyword "type_constructor"} Parse.type_const parse_tyco
haftmann@55150
   693
    >> Type_Constructor
haftmann@52377
   694
  || parse_single_symbol_pragma @{keyword "type_class"} Parse.class parse_class
haftmann@55150
   695
    >> Type_Class
haftmann@52137
   696
  || parse_single_symbol_pragma @{keyword "class_relation"} parse_classrel_ident parse_classrel
haftmann@55150
   697
    >> Class_Relation
haftmann@52137
   698
  || parse_single_symbol_pragma @{keyword "class_instance"} parse_inst_ident parse_inst
haftmann@55150
   699
    >> Class_Instance
haftmann@52137
   700
  || parse_single_symbol_pragma @{keyword "code_module"} Parse.name parse_module
haftmann@52137
   701
    >> Code_Symbol.Module;
haftmann@52137
   702
haftmann@52137
   703
fun parse_symbol_pragmas parse_const parse_tyco parse_class parse_classrel parse_inst parse_module =
haftmann@52137
   704
  Parse.enum1 "|" (Parse.group (fn () => "code symbol pragma")
haftmann@52137
   705
    (parse_symbol_pragma parse_const parse_tyco parse_class parse_classrel parse_inst parse_module));
haftmann@52137
   706
haftmann@52138
   707
val code_expr_argsP = Scan.optional (@{keyword "("} |-- Args.parse --| @{keyword ")"}) [];
haftmann@52138
   708
haftmann@52434
   709
fun code_expr_inP raw_cs =
haftmann@52434
   710
  Scan.repeat (@{keyword "in"} |-- Parse.!!! (Parse.name
haftmann@52434
   711
    -- Scan.optional (@{keyword "module_name"} |-- Parse.name) ""
haftmann@52434
   712
    -- Scan.optional (@{keyword "file"} |-- Parse.name) ""
haftmann@52434
   713
    -- code_expr_argsP))
haftmann@52434
   714
      >> (fn seri_args => export_code_cmd raw_cs seri_args);
haftmann@52434
   715
haftmann@52434
   716
fun code_expr_checkingP raw_cs =
haftmann@52434
   717
  (@{keyword "checking"} |-- Parse.!!!
haftmann@52434
   718
    (Scan.repeat (Parse.name -- ((@{keyword "?"} |-- Scan.succeed false) || Scan.succeed true)
haftmann@52434
   719
    -- code_expr_argsP)))
haftmann@52434
   720
      >> (fn seri_args => check_code_cmd raw_cs seri_args);
haftmann@52434
   721
wenzelm@52801
   722
val code_exprP = Scan.repeat1 Parse.term
haftmann@52434
   723
  :|-- (fn raw_cs => (code_expr_checkingP raw_cs || code_expr_inP raw_cs));
haftmann@52138
   724
haftmann@52138
   725
val _ =
haftmann@52138
   726
  Outer_Syntax.command @{command_spec "code_reserved"}
haftmann@52138
   727
    "declare words as reserved for target language"
haftmann@52138
   728
    (Parse.name -- Scan.repeat1 Parse.name
haftmann@52138
   729
      >> (fn (target, reserveds) => (Toplevel.theory o fold (add_reserved target)) reserveds));
haftmann@52138
   730
haftmann@52138
   731
val _ =
haftmann@52138
   732
  Outer_Syntax.command @{command_spec "code_identifier"} "declare mandatory names for code symbols"
haftmann@52138
   733
    (parse_symbol_pragmas Parse.name Parse.name Parse.name Parse.name Parse.name Parse.name
haftmann@52138
   734
      >> (Toplevel.theory o fold set_identifiers_cmd));
haftmann@52138
   735
haftmann@52138
   736
val _ =
haftmann@52138
   737
  Outer_Syntax.command @{command_spec "code_modulename"} "alias module to other name"
haftmann@52138
   738
    (Parse.name -- Scan.repeat1 (Parse.name -- Parse.name)
haftmann@52138
   739
      >> (fn (target, modlnames) => (Toplevel.theory o add_module_alias_cmd target) modlnames));
haftmann@52138
   740
haftmann@52137
   741
val _ =
haftmann@52137
   742
  Outer_Syntax.command @{command_spec "code_printing"} "declare dedicated printing for code symbols"
haftmann@52137
   743
    (parse_symbol_pragmas (Code_Printer.parse_const_syntax) (Code_Printer.parse_tyco_syntax)
haftmann@52137
   744
      Parse.string (Parse.minus >> K ()) (Parse.minus >> K ())
haftmann@52137
   745
      (Parse.text -- Scan.optional (@{keyword "attach"} |-- Scan.repeat1 Parse.term) [])
haftmann@52137
   746
      >> (Toplevel.theory o fold set_printings_cmd));
haftmann@52137
   747
haftmann@52137
   748
val _ =
haftmann@52137
   749
  Outer_Syntax.command @{command_spec "code_const"} "define code syntax for constant"
wenzelm@52801
   750
    (process_multi_syntax Parse.term Code_Printer.parse_const_syntax
haftmann@52137
   751
      add_const_syntax_cmd);
haftmann@52137
   752
haftmann@52137
   753
val _ =
haftmann@52137
   754
  Outer_Syntax.command @{command_spec "code_type"} "define code syntax for type constructor"
haftmann@52377
   755
    (process_multi_syntax Parse.type_const Code_Printer.parse_tyco_syntax
haftmann@52137
   756
      add_tyco_syntax_cmd);
haftmann@52137
   757
wenzelm@24867
   758
val _ =
wenzelm@46961
   759
  Outer_Syntax.command @{command_spec "code_class"} "define code syntax for class"
haftmann@52377
   760
    (process_multi_syntax Parse.class Parse.string
wenzelm@46961
   761
      add_class_syntax_cmd);
haftmann@24219
   762
wenzelm@24867
   763
val _ =
wenzelm@46961
   764
  Outer_Syntax.command @{command_spec "code_instance"} "define code syntax for instance"
haftmann@52377
   765
    (process_multi_syntax parse_inst_ident (Parse.minus >> K ())
wenzelm@46961
   766
      add_instance_syntax_cmd);
haftmann@24219
   767
wenzelm@24867
   768
val _ =
wenzelm@46961
   769
  Outer_Syntax.command @{command_spec "code_include"}
wenzelm@46961
   770
    "declare piece of code to be included in generated code"
wenzelm@46961
   771
    (Parse.name -- Parse.name -- (Parse.text :|--
wenzelm@46961
   772
      (fn "-" => Scan.succeed NONE
wenzelm@46961
   773
        | s => Scan.optional (@{keyword "attach"} |-- Scan.repeat1 Parse.term) [] >> pair s >> SOME))
wenzelm@46961
   774
      >> (fn ((target, name), content_consts) =>
wenzelm@46961
   775
          (Toplevel.theory o add_include_cmd target) (name, content_consts)));
haftmann@24992
   776
haftmann@24992
   777
val _ =
wenzelm@46961
   778
  Outer_Syntax.command @{command_spec "export_code"} "generate executable code for constants"
wenzelm@46961
   779
    (Parse.!!! code_exprP >> (fn f => Toplevel.keep (f o Toplevel.theory_of)));
haftmann@30494
   780
haftmann@24219
   781
end; (*local*)
haftmann@24219
   782
haftmann@39646
   783
haftmann@39646
   784
(** external entrance point -- for codegen tool **)
haftmann@39646
   785
haftmann@39750
   786
fun codegen_tool thyname cmd_expr =
haftmann@39646
   787
  let
haftmann@39646
   788
    val thy = Thy_Info.get_theory thyname;
haftmann@39646
   789
    val parse = Scan.read Token.stopper (Parse.!!! code_exprP) o
haftmann@39646
   790
      (filter Token.is_proper o Outer_Syntax.scan Position.none);
haftmann@39646
   791
  in case parse cmd_expr
haftmann@39646
   792
   of SOME f => (writeln "Now generating code..."; f thy)
haftmann@39646
   793
    | NONE => error ("Bad directive " ^ quote cmd_expr)
haftmann@39646
   794
  end;
haftmann@39646
   795
wenzelm@43564
   796
wenzelm@43564
   797
(** theory setup **)
wenzelm@43564
   798
wenzelm@43564
   799
val setup = antiq_setup;
wenzelm@43564
   800
haftmann@24219
   801
end; (*struct*)