src/Tools/Code/code_scala.ML
author haftmann
Thu Jul 08 16:19:24 2010 +0200 (2010-07-08)
changeset 37745 6315b6426200
parent 37669 a5da34455a5c
child 37748 0af0d45257be
permissions -rw-r--r--
checking generated code for various target languages
haftmann@37745
     1
(*  Title:      Tools/Code/code_scala.ML
haftmann@37745
     2
    Author:     Florian Haftmann, TU Muenchen
haftmann@34294
     3
haftmann@34294
     4
Serializer for Scala.
haftmann@34294
     5
*)
haftmann@34294
     6
haftmann@34294
     7
signature CODE_SCALA =
haftmann@34294
     8
sig
haftmann@37745
     9
  val target: string
haftmann@37745
    10
  val check: theory -> Path.T -> unit
haftmann@34294
    11
  val setup: theory -> theory
haftmann@34294
    12
end;
haftmann@34294
    13
haftmann@34294
    14
structure Code_Scala : CODE_SCALA =
haftmann@34294
    15
struct
haftmann@34294
    16
haftmann@34294
    17
val target = "Scala";
haftmann@34294
    18
haftmann@34294
    19
open Basic_Code_Thingol;
haftmann@34294
    20
open Code_Printer;
haftmann@34294
    21
haftmann@34294
    22
infixr 5 @@;
haftmann@34294
    23
infixr 5 @|;
haftmann@34294
    24
haftmann@34294
    25
haftmann@34294
    26
(** Scala serializer **)
haftmann@34294
    27
haftmann@37464
    28
fun print_scala_stmt labelled_name syntax_tyco syntax_const reserved
haftmann@37639
    29
    args_num is_singleton_constr deresolve =
haftmann@34294
    30
  let
haftmann@34294
    31
    val deresolve_base = Long_Name.base_name o deresolve;
haftmann@37639
    32
    fun lookup_tyvar tyvars = lookup_var tyvars o first_upper;
haftmann@37639
    33
    fun intro_tyvars vs = intro_vars (map (first_upper o fst) vs);
haftmann@37639
    34
    fun print_tyco_expr tyvars fxy (tyco, tys) = applify "[" "]"
haftmann@37639
    35
          (print_typ tyvars NOBR) fxy ((str o deresolve) tyco) tys
haftmann@37639
    36
    and print_typ tyvars fxy (tyco `%% tys) = (case syntax_tyco tyco
haftmann@37639
    37
         of NONE => print_tyco_expr tyvars fxy (tyco, tys)
haftmann@34294
    38
          | SOME (i, print) => print (print_typ tyvars) fxy tys)
haftmann@37243
    39
      | print_typ tyvars fxy (ITyVar v) = (str o lookup_tyvar tyvars) v;
haftmann@37639
    40
    fun print_dicttyp tyvars (class, ty) = print_tyco_expr tyvars NOBR (class, [ty]);
haftmann@37639
    41
    fun print_tupled_typ tyvars ([], ty) =
haftmann@37639
    42
          print_typ tyvars NOBR ty
haftmann@37639
    43
      | print_tupled_typ tyvars ([ty1], ty2) =
haftmann@37639
    44
          concat [print_typ tyvars BR ty1, str "=>", print_typ tyvars NOBR ty2]
haftmann@37639
    45
      | print_tupled_typ tyvars (tys, ty) =
haftmann@37639
    46
          concat [enum "," "(" ")" (map (print_typ tyvars NOBR) tys),
haftmann@37639
    47
            str "=>", print_typ tyvars NOBR ty];
haftmann@37639
    48
    fun constraint p1 p2 = Pretty.block [p1, str ":", Pretty.brk 1, p2];
haftmann@34294
    49
    fun print_var vars NONE = str "_"
haftmann@34294
    50
      | print_var vars (SOME v) = (str o lookup_var vars) v
haftmann@35228
    51
    fun print_term tyvars is_pat some_thm vars fxy (IConst c) =
haftmann@35228
    52
          print_app tyvars is_pat some_thm vars fxy (c, [])
haftmann@35228
    53
      | print_term tyvars is_pat some_thm vars fxy (t as (t1 `$ t2)) =
haftmann@34294
    54
          (case Code_Thingol.unfold_const_app t
haftmann@35228
    55
           of SOME app => print_app tyvars is_pat some_thm vars fxy app
haftmann@37639
    56
            | _ => applify "(" ")" (print_term tyvars is_pat some_thm vars NOBR) fxy
haftmann@37639
    57
                (print_term tyvars is_pat some_thm vars BR t1) [t2])
haftmann@35228
    58
      | print_term tyvars is_pat some_thm vars fxy (IVar v) =
haftmann@34294
    59
          print_var vars v
haftmann@35228
    60
      | print_term tyvars is_pat some_thm vars fxy ((v, ty) `|=> t) =
haftmann@34294
    61
          let
haftmann@34294
    62
            val vars' = intro_vars (the_list v) vars;
haftmann@34294
    63
          in
haftmann@34294
    64
            concat [
haftmann@37639
    65
              enclose "(" ")" [constraint (print_var vars' v) (print_typ tyvars NOBR ty)],
haftmann@34294
    66
              str "=>",
haftmann@35228
    67
              print_term tyvars false some_thm vars' NOBR t
haftmann@34294
    68
            ]
haftmann@34294
    69
          end 
haftmann@35228
    70
      | print_term tyvars is_pat some_thm vars fxy (ICase (cases as (_, t0))) =
haftmann@34294
    71
          (case Code_Thingol.unfold_const_app t0
haftmann@34294
    72
           of SOME (c_ts as ((c, _), _)) => if is_none (syntax_const c)
haftmann@35228
    73
                then print_case tyvars some_thm vars fxy cases
haftmann@35228
    74
                else print_app tyvars is_pat some_thm vars fxy c_ts
haftmann@35228
    75
            | NONE => print_case tyvars some_thm vars fxy cases)
haftmann@37464
    76
    and print_app tyvars is_pat some_thm vars fxy
haftmann@37464
    77
        (app as ((c, ((arg_typs, _), function_typs)), ts)) =
haftmann@34294
    78
      let
haftmann@34294
    79
        val k = length ts;
haftmann@34294
    80
        val l = case syntax_const c of NONE => args_num c | SOME (l, _) => l;
haftmann@37450
    81
        val arg_typs' = if is_pat orelse
haftmann@37639
    82
          (is_none (syntax_const c) andalso is_singleton_constr c) then [] else arg_typs;
haftmann@34294
    83
        val (no_syntax, print') = case syntax_const c
haftmann@37639
    84
         of NONE => (true, fn ts => applify "(" ")"
haftmann@37639
    85
              (print_term tyvars is_pat some_thm vars NOBR) fxy
haftmann@37639
    86
                (applify "[" "]" (print_typ tyvars NOBR)
haftmann@37639
    87
                  NOBR ((str o deresolve) c) arg_typs') ts)
haftmann@34294
    88
          | SOME (_, print) => (false, fn ts =>
haftmann@37464
    89
              print (print_term tyvars is_pat some_thm) some_thm vars fxy
haftmann@37464
    90
                (ts ~~ take l function_typs));
haftmann@34294
    91
      in if k = l then print' ts
haftmann@34294
    92
      else if k < l then
haftmann@35228
    93
        print_term tyvars is_pat some_thm vars fxy (Code_Thingol.eta_expand l app)
haftmann@34294
    94
      else let
haftmann@34294
    95
        val (ts1, ts23) = chop l ts;
haftmann@34294
    96
      in
haftmann@34294
    97
        Pretty.block (print' ts1 :: map (fn t => Pretty.block
haftmann@35228
    98
          [str ".apply(", print_term tyvars is_pat some_thm vars NOBR t, str ")"]) ts23)
haftmann@34294
    99
      end end
haftmann@37464
   100
    and print_bind tyvars some_thm fxy p =
haftmann@37464
   101
      gen_print_bind (print_term tyvars true) some_thm fxy p
haftmann@35228
   102
    and print_case tyvars some_thm vars fxy (cases as ((_, [_]), _)) =
haftmann@34294
   103
          let
haftmann@34294
   104
            val (binds, body) = Code_Thingol.unfold_let (ICase cases);
haftmann@34294
   105
            fun print_match ((pat, ty), t) vars =
haftmann@34294
   106
              vars
haftmann@35228
   107
              |> print_bind tyvars some_thm BR pat
haftmann@37639
   108
              |>> (fn p => concat [str "val", constraint p (print_typ tyvars NOBR ty),
haftmann@35228
   109
                  str "=", print_term tyvars false some_thm vars NOBR t])
haftmann@37639
   110
            val (all_ps, vars') = fold_map print_match binds vars;
haftmann@37639
   111
            val (ps, p) = split_last all_ps;
haftmann@37639
   112
          in brackify_block fxy
haftmann@37639
   113
            (str "{")
haftmann@37639
   114
            (ps @ Pretty.block [p, str ";"] @@ print_term tyvars false some_thm vars' NOBR body)
haftmann@37639
   115
            (str "}")
haftmann@34294
   116
          end
haftmann@35228
   117
      | print_case tyvars some_thm vars fxy (((t, ty), clauses as _ :: _), _) =
haftmann@34294
   118
          let
haftmann@34294
   119
            fun print_select (pat, body) =
haftmann@34294
   120
              let
haftmann@37464
   121
                val (p_pat, vars') = print_bind tyvars some_thm NOBR pat vars;
haftmann@37464
   122
                val p_body = print_term tyvars false some_thm vars' NOBR body
haftmann@37464
   123
              in concat [str "case", p_pat, str "=>", p_body] end;
haftmann@34294
   124
          in brackify_block fxy
haftmann@35228
   125
            (concat [print_term tyvars false some_thm vars NOBR t, str "match", str "{"])
haftmann@34294
   126
            (map print_select clauses)
haftmann@34294
   127
            (str "}") 
haftmann@34294
   128
          end
haftmann@35228
   129
      | print_case tyvars some_thm vars fxy ((_, []), _) =
haftmann@34294
   130
          (brackify fxy o Pretty.breaks o map str) ["error(\"empty case\")"];
haftmann@37639
   131
    fun print_context tyvars vs name = applify "[" "]"
haftmann@37639
   132
      (fn (v, sort) => (Pretty.block o map str)
haftmann@37639
   133
        (lookup_tyvar tyvars v :: maps (fn sort => [": ", deresolve sort]) sort))
haftmann@37639
   134
          NOBR ((str o deresolve) name) vs;
haftmann@37639
   135
    fun print_defhead tyvars vars name vs params tys ty =
haftmann@37639
   136
      Pretty.block [str "def ", constraint (applify "(" ")" (fn (param, ty) =>
haftmann@37639
   137
        constraint ((str o lookup_var vars) param) (print_typ tyvars NOBR ty))
haftmann@37639
   138
          NOBR (print_context tyvars vs name) (params ~~ tys)) (print_typ tyvars NOBR ty),
haftmann@37639
   139
            str " ="];
haftmann@37639
   140
    fun print_def name (vs, ty) [] =
haftmann@37639
   141
          let
haftmann@37639
   142
            val (tys, ty') = Code_Thingol.unfold_fun ty;
haftmann@37639
   143
            val params = Name.invents (snd reserved) "a" (length tys);
haftmann@37639
   144
            val tyvars = intro_tyvars vs reserved;
haftmann@37639
   145
            val vars = intro_vars params reserved;
haftmann@37639
   146
          in
haftmann@37639
   147
            concat [print_defhead tyvars vars name vs params tys ty',
haftmann@37639
   148
              str ("error(\"" ^ name ^ "\")")]
haftmann@37639
   149
          end
haftmann@37639
   150
      | print_def name (vs, ty) eqs =
haftmann@37639
   151
          let
haftmann@37639
   152
            val tycos = fold (fn ((ts, t), _) =>
haftmann@37639
   153
              fold Code_Thingol.add_tyconames (t :: ts)) eqs [];
haftmann@37639
   154
            val tyvars = reserved
haftmann@37639
   155
              |> intro_base_names
haftmann@37639
   156
                   (is_none o syntax_tyco) deresolve tycos
haftmann@37639
   157
              |> intro_tyvars vs;
haftmann@37639
   158
            val simple = case eqs
haftmann@37639
   159
             of [((ts, _), _)] => forall Code_Thingol.is_IVar ts
haftmann@37639
   160
              | _ => false;
haftmann@37639
   161
            val consts = fold Code_Thingol.add_constnames
haftmann@37639
   162
              (map (snd o fst) eqs) [];
haftmann@37639
   163
            val vars1 = reserved
haftmann@37639
   164
              |> intro_base_names
haftmann@37639
   165
                   (is_none o syntax_const) deresolve consts
haftmann@37639
   166
            val params = if simple
haftmann@37639
   167
              then (map (fn IVar (SOME x) => x) o fst o fst o hd) eqs
haftmann@37639
   168
              else aux_params vars1 (map (fst o fst) eqs);
haftmann@37639
   169
            val vars2 = intro_vars params vars1;
haftmann@37639
   170
            val (tys', ty') = Code_Thingol.unfold_fun_n (length params) ty;
haftmann@37639
   171
            fun print_tuple [p] = p
haftmann@37639
   172
              | print_tuple ps = enum "," "(" ")" ps;
haftmann@37639
   173
            fun print_rhs vars' ((_, t), (some_thm, _)) =
haftmann@37639
   174
              print_term tyvars false some_thm vars' NOBR t;
haftmann@37639
   175
            fun print_clause (eq as ((ts, _), (some_thm, _))) =
haftmann@34294
   176
              let
haftmann@37639
   177
                val vars' = intro_vars ((fold o Code_Thingol.fold_varnames)
haftmann@37639
   178
                  (insert (op =)) ts []) vars1;
haftmann@37639
   179
              in
haftmann@37639
   180
                concat [str "case",
haftmann@37639
   181
                  print_tuple (map (print_term tyvars true some_thm vars' NOBR) ts),
haftmann@37639
   182
                  str "=>", print_rhs vars' eq]
haftmann@37639
   183
              end;
haftmann@37639
   184
            val head = print_defhead tyvars vars2 name vs params tys' ty';
haftmann@37639
   185
          in if simple then
haftmann@37639
   186
            concat [head, print_rhs vars2 (hd eqs)]
haftmann@37639
   187
          else
haftmann@37639
   188
            Pretty.block_enclose
haftmann@37639
   189
              (concat [head, print_tuple (map (str o lookup_var vars2) params),
haftmann@37639
   190
                str "match", str "{"], str "}")
haftmann@37639
   191
              (map print_clause eqs)
haftmann@37639
   192
          end;
haftmann@37639
   193
    val print_method = (str o Library.enclose "`" "+`" o deresolve_base);
haftmann@37639
   194
    fun print_stmt (name, Code_Thingol.Fun (_, (((vs, ty), raw_eqs), _))) =
haftmann@37639
   195
          print_def name (vs, ty) (filter (snd o snd) raw_eqs)
haftmann@34294
   196
      | print_stmt (name, Code_Thingol.Datatype (_, (vs, cos))) =
haftmann@34294
   197
          let
haftmann@37639
   198
            val tyvars = intro_tyvars vs reserved;
haftmann@37450
   199
            fun print_co ((co, _), []) =
haftmann@34294
   200
                  concat [str "final", str "case", str "object", (str o deresolve_base) co,
haftmann@37639
   201
                    str "extends", applify "[" "]" I NOBR ((str o deresolve_base) name)
haftmann@34294
   202
                      (replicate (length vs) (str "Nothing"))]
haftmann@37450
   203
              | print_co ((co, vs_args), tys) =
haftmann@37639
   204
                  concat [applify "(" ")"
haftmann@37639
   205
                    (fn (v, arg) => constraint (str v) (print_typ tyvars NOBR arg)) NOBR
haftmann@37639
   206
                    (applify "[" "]" (str o lookup_tyvar tyvars) NOBR ((concat o map str)
haftmann@37639
   207
                      ["final", "case", "class", deresolve_base co]) vs_args)
haftmann@37639
   208
                    (Name.names (snd reserved) "a" tys),
haftmann@37639
   209
                    str "extends",
haftmann@37639
   210
                    applify "[" "]" (str o lookup_tyvar tyvars o fst) NOBR
haftmann@37639
   211
                      ((str o deresolve_base) name) vs
haftmann@37639
   212
                  ];
haftmann@34294
   213
          in
haftmann@37639
   214
            Pretty.chunks (applify "[" "]" (str o prefix "+" o lookup_tyvar tyvars o fst)
haftmann@37639
   215
              NOBR ((concat o map str) ["sealed", "class", deresolve_base name]) vs
haftmann@37639
   216
                :: map print_co cos)
haftmann@34294
   217
          end
haftmann@37447
   218
      | print_stmt (name, Code_Thingol.Class (_, (v, (super_classes, classparams)))) =
haftmann@34294
   219
          let
haftmann@37639
   220
            val tyvars = intro_tyvars [(v, [name])] reserved;
haftmann@37639
   221
            fun add_typarg s = Pretty.block
haftmann@37639
   222
              [str s, str "[", (str o lookup_tyvar tyvars) v, str "]"];
haftmann@37384
   223
            fun print_super_classes [] = NONE
haftmann@37384
   224
              | print_super_classes classes = SOME (concat (str "extends"
haftmann@37639
   225
                  :: separate (str "with") (map (add_typarg o deresolve o fst) classes)));
haftmann@34294
   226
            fun print_classparam_val (classparam, ty) =
haftmann@37639
   227
              concat [str "val", constraint (print_method classparam)
haftmann@37639
   228
                ((print_tupled_typ tyvars o Code_Thingol.unfold_fun) ty)];
haftmann@34294
   229
            fun print_classparam_def (classparam, ty) =
haftmann@34294
   230
              let
haftmann@34294
   231
                val (tys, ty) = Code_Thingol.unfold_fun ty;
haftmann@37639
   232
                val [implicit_name] = Name.invents (snd reserved) (lookup_tyvar tyvars v) 1;
haftmann@37639
   233
                val proto_vars = intro_vars [implicit_name] reserved;
haftmann@37639
   234
                val auxs = Name.invents (snd proto_vars) "a" (length tys);
haftmann@37639
   235
                val vars = intro_vars auxs proto_vars;
haftmann@34294
   236
              in
haftmann@37639
   237
                concat [str "def", constraint (Pretty.block [applify "(" ")"
haftmann@37639
   238
                  (fn (aux, ty) => constraint ((str o lookup_var vars) aux)
haftmann@37639
   239
                  (print_typ tyvars NOBR ty)) NOBR (add_typarg (deresolve classparam))
haftmann@37639
   240
                  (auxs ~~ tys), str "(implicit ", str implicit_name, str ": ",
haftmann@37639
   241
                  add_typarg (deresolve name), str ")"]) (print_typ tyvars NOBR ty), str "=",
haftmann@37639
   242
                  applify "(" ")" (str o lookup_var vars) NOBR
haftmann@37639
   243
                  (Pretty.block [str implicit_name, str ".", print_method classparam]) auxs]
haftmann@34294
   244
              end;
haftmann@34294
   245
          in
haftmann@34294
   246
            Pretty.chunks (
haftmann@34294
   247
              (Pretty.block_enclose
haftmann@37639
   248
                (concat ([str "trait", (add_typarg o deresolve_base) name]
haftmann@37384
   249
                  @ the_list (print_super_classes super_classes) @ [str "{"]), str "}")
haftmann@34294
   250
                (map print_classparam_val classparams))
haftmann@34294
   251
              :: map print_classparam_def classparams
haftmann@34294
   252
            )
haftmann@34294
   253
          end
haftmann@34294
   254
      | print_stmt (name, Code_Thingol.Classinst ((class, (tyco, vs)),
haftmann@37450
   255
            (super_instances, (classparam_instances, further_classparam_instances)))) =
haftmann@34294
   256
          let
haftmann@37639
   257
            val tyvars = intro_tyvars vs reserved;
haftmann@37639
   258
            val classtyp = (class, tyco `%% map (ITyVar o fst) vs);
haftmann@37450
   259
            fun print_classparam_instance ((classparam, const as (_, (_, tys))), (thm, _)) =
haftmann@37450
   260
              let
haftmann@37639
   261
                val aux_tys = Name.names (snd reserved) "a" tys;
haftmann@37639
   262
                val auxs = map fst aux_tys;
haftmann@37450
   263
                val vars = intro_vars auxs reserved;
haftmann@37639
   264
                val aux_abstr = if null auxs then [] else [enum "," "(" ")"
haftmann@37639
   265
                  (map (fn (aux, ty) => constraint ((str o lookup_var vars) aux)
haftmann@37639
   266
                  (print_typ tyvars NOBR ty)) aux_tys), str "=>"];
haftmann@37450
   267
              in 
haftmann@37639
   268
                concat ([str "val", print_method classparam, str "="]
haftmann@37639
   269
                  @ aux_abstr @| print_app tyvars false (SOME thm) vars NOBR
haftmann@37639
   270
                    (const, map (IVar o SOME) auxs))
haftmann@37450
   271
              end;
haftmann@37639
   272
          in
haftmann@37639
   273
            Pretty.block_enclose (concat [str "implicit def",
haftmann@37639
   274
              constraint (print_context tyvars vs name) (print_dicttyp tyvars classtyp),
haftmann@37639
   275
              str "=", str "new", print_dicttyp tyvars classtyp, str "{"], str "}")
haftmann@37639
   276
                (map print_classparam_instance (classparam_instances @ further_classparam_instances))
haftmann@37639
   277
          end;
haftmann@34294
   278
  in print_stmt end;
haftmann@34294
   279
haftmann@34294
   280
fun scala_program_of_program labelled_name module_name reserved raw_module_alias program =
haftmann@34294
   281
  let
haftmann@34294
   282
    val the_module_name = the_default "Program" module_name;
haftmann@34294
   283
    val module_alias = K (SOME the_module_name);
haftmann@34294
   284
    val reserved = Name.make_context reserved;
haftmann@34294
   285
    fun prepare_stmt (name, stmt) (nsps, stmts) =
haftmann@34294
   286
      let
haftmann@34294
   287
        val (_, base) = Code_Printer.dest_name name;
haftmann@34294
   288
        val mk_name_stmt = yield_singleton Name.variants;
haftmann@34294
   289
        fun add_class ((nsp_class, nsp_object), nsp_common) =
haftmann@34294
   290
          let
haftmann@34294
   291
            val (base', nsp_class') = mk_name_stmt base nsp_class
haftmann@34294
   292
          in (base', ((nsp_class', nsp_object), Name.declare base' nsp_common)) end;
haftmann@34294
   293
        fun add_object ((nsp_class, nsp_object), nsp_common) =
haftmann@34294
   294
          let
haftmann@34294
   295
            val (base', nsp_object') = mk_name_stmt base nsp_object
haftmann@34294
   296
          in (base', ((nsp_class, nsp_object'), Name.declare base' nsp_common)) end;
haftmann@34294
   297
        fun add_common upper ((nsp_class, nsp_object), nsp_common) =
haftmann@34294
   298
          let
haftmann@34294
   299
            val (base', nsp_common') =
haftmann@34294
   300
              mk_name_stmt (if upper then first_upper base else base) nsp_common
haftmann@37464
   301
          in
haftmann@37464
   302
            (base',
haftmann@37464
   303
              ((Name.declare base' nsp_class, Name.declare base' nsp_object), nsp_common'))
haftmann@37464
   304
          end;
haftmann@34294
   305
        val add_name = case stmt
haftmann@34294
   306
         of Code_Thingol.Fun _ => add_object
haftmann@34294
   307
          | Code_Thingol.Datatype _ => add_class
haftmann@34294
   308
          | Code_Thingol.Datatypecons _ => add_common true
haftmann@34294
   309
          | Code_Thingol.Class _ => add_class
haftmann@34294
   310
          | Code_Thingol.Classrel _ => add_object
haftmann@34294
   311
          | Code_Thingol.Classparam _ => add_object
haftmann@34294
   312
          | Code_Thingol.Classinst _ => add_common false;
haftmann@34294
   313
        fun add_stmt base' = case stmt
haftmann@34294
   314
         of Code_Thingol.Datatypecons _ => cons (name, (base', NONE))
haftmann@34294
   315
          | Code_Thingol.Classrel _ => cons (name, (base', NONE))
haftmann@34294
   316
          | Code_Thingol.Classparam _ => cons (name, (base', NONE))
haftmann@34294
   317
          | _ => cons (name, (base', SOME stmt));
haftmann@34294
   318
      in
haftmann@34294
   319
        nsps
haftmann@34294
   320
        |> add_name
haftmann@34294
   321
        |-> (fn base' => rpair (add_stmt base' stmts))
haftmann@34294
   322
      end;
haftmann@37439
   323
    val stmts = AList.make (Graph.get_node program) (Graph.strong_conn program |> flat)
haftmann@37439
   324
      |> filter_out (Code_Thingol.is_case o snd);
haftmann@37439
   325
    val (_, sca_program) = fold prepare_stmt stmts (((reserved, reserved), reserved), []);
haftmann@34294
   326
    fun deresolver name = (fst o the o AList.lookup (op =) sca_program) name
haftmann@34294
   327
      handle Option => error ("Unknown statement name: " ^ labelled_name name);
haftmann@34294
   328
  in (deresolver, (the_module_name, sca_program)) end;
haftmann@34294
   329
haftmann@34294
   330
fun serialize_scala raw_module_name labelled_name
haftmann@34294
   331
    raw_reserved includes raw_module_alias
haftmann@37464
   332
    _ syntax_tyco syntax_const (code_of_pretty, code_writeln)
haftmann@37464
   333
    program stmt_names destination =
haftmann@34294
   334
  let
haftmann@36535
   335
    val presentation_stmt_names = Code_Target.stmt_names_of_destination destination;
haftmann@36535
   336
    val module_name = if null presentation_stmt_names then raw_module_name else SOME "Code";
haftmann@34294
   337
    val reserved = fold (insert (op =) o fst) includes raw_reserved;
haftmann@34294
   338
    val (deresolver, (the_module_name, sca_program)) = scala_program_of_program labelled_name
haftmann@34294
   339
      module_name reserved raw_module_alias program;
haftmann@34294
   340
    val reserved = make_vars reserved;
haftmann@37639
   341
    fun lookup_constr tyco constr = case Graph.get_node program tyco
haftmann@37639
   342
     of Code_Thingol.Datatype (_, (_, constrs)) =>
haftmann@37639
   343
          the (AList.lookup (op = o apsnd fst) constrs constr);
haftmann@37639
   344
    fun classparams_of_class class = case Graph.get_node program class
haftmann@37639
   345
     of Code_Thingol.Class (_, (_, (_, classparams))) => classparams;
haftmann@34294
   346
    fun args_num c = case Graph.get_node program c
haftmann@37464
   347
     of Code_Thingol.Fun (_, (((_, ty), []), _)) =>
haftmann@37464
   348
          (length o fst o Code_Thingol.unfold_fun) ty
haftmann@37437
   349
      | Code_Thingol.Fun (_, ((_, ((ts, _), _) :: _), _)) => length ts
haftmann@37639
   350
      | Code_Thingol.Datatypecons (_, tyco) => length (lookup_constr tyco c)
haftmann@34294
   351
      | Code_Thingol.Classparam (_, class) =>
haftmann@37639
   352
          (length o fst o Code_Thingol.unfold_fun o the o AList.lookup (op =)
haftmann@37639
   353
            (classparams_of_class class)) c;
haftmann@37639
   354
    fun is_singleton_constr c = case Graph.get_node program c
haftmann@37639
   355
     of Code_Thingol.Datatypecons (_, tyco) => null (lookup_constr tyco c)
haftmann@34294
   356
      | _ => false;
haftmann@34294
   357
    val print_stmt = print_scala_stmt labelled_name syntax_tyco syntax_const
haftmann@37639
   358
      reserved args_num is_singleton_constr deresolver;
haftmann@34294
   359
    fun print_module name content =
haftmann@34294
   360
      (name, Pretty.chunks [
haftmann@34294
   361
        str ("object " ^ name ^ " {"),
haftmann@34294
   362
        str "",
haftmann@34294
   363
        content,
haftmann@34294
   364
        str "",
haftmann@34294
   365
        str "}"
haftmann@34294
   366
      ]);
haftmann@34294
   367
    fun serialize_module the_module_name sca_program =
haftmann@34294
   368
      let
haftmann@34294
   369
        val content = Pretty.chunks2 (map_filter
haftmann@34294
   370
          (fn (name, (_, SOME stmt)) => SOME (print_stmt (name, stmt))
haftmann@34294
   371
            | (_, (_, NONE)) => NONE) sca_program);
haftmann@34294
   372
      in print_module the_module_name content end;
haftmann@34294
   373
    fun check_destination destination =
haftmann@34294
   374
      (File.check destination; destination);
haftmann@34294
   375
    fun write_module destination (modlname, content) =
haftmann@34294
   376
      let
haftmann@34294
   377
        val filename = case modlname
haftmann@34294
   378
         of "" => Path.explode "Main.scala"
haftmann@34294
   379
          | _ => (Path.ext "scala" o Path.explode o implode o separate "/"
haftmann@34294
   380
                o Long_Name.explode) modlname;
haftmann@34294
   381
        val pathname = Path.append destination filename;
haftmann@37669
   382
        val _ = File.mkdir_leaf (Path.dir pathname);
haftmann@34294
   383
      in File.write pathname (code_of_pretty content) end
haftmann@34294
   384
  in
haftmann@34294
   385
    Code_Target.mk_serialization target NONE
haftmann@34294
   386
      (fn NONE => K () o map (code_writeln o snd) | SOME file => K () o map
haftmann@34294
   387
        (write_module (check_destination file)))
haftmann@34294
   388
      (rpair [] o cat_lines o map (code_of_pretty o snd))
haftmann@34294
   389
      (map (uncurry print_module) includes
haftmann@34294
   390
        @| serialize_module the_module_name sca_program)
haftmann@34294
   391
      destination
haftmann@34294
   392
  end;
haftmann@34294
   393
haftmann@34294
   394
val literals = let
haftmann@37224
   395
  fun char_scala c = if c = "'" then "\\'"
haftmann@37224
   396
    else if c = "\"" then "\\\""
haftmann@37224
   397
    else if c = "\\" then "\\\\"
haftmann@37224
   398
    else let val k = ord c
haftmann@37224
   399
    in if k < 32 orelse k > 126 then "\\" ^ radixstring (8, "0", k) else c end
haftmann@34944
   400
  fun numeral_scala k = if k < 0
haftmann@34944
   401
    then if k <= 2147483647 then "- " ^ string_of_int (~ k)
haftmann@34944
   402
      else quote ("- " ^ string_of_int (~ k))
haftmann@34944
   403
    else if k <= 2147483647 then string_of_int k
haftmann@34944
   404
      else quote (string_of_int k)
haftmann@34294
   405
in Literals {
haftmann@34294
   406
  literal_char = Library.enclose "'" "'" o char_scala,
haftmann@34294
   407
  literal_string = quote o translate_string char_scala,
haftmann@34944
   408
  literal_numeral = fn k => "BigInt(" ^ numeral_scala k ^ ")",
haftmann@34944
   409
  literal_positive_numeral = fn k => "Nat.Nat(" ^ numeral_scala k ^ ")",
haftmann@34944
   410
  literal_naive_numeral = fn k => if k >= 0
haftmann@34944
   411
    then string_of_int k else "(- " ^ string_of_int (~ k) ^ ")",
haftmann@34888
   412
  literal_list = fn [] => str "Nil" | ps => Pretty.block [str "List", enum "," "(" ")" ps],
haftmann@34294
   413
  infix_cons = (6, "::")
haftmann@34294
   414
} end;
haftmann@34294
   415
haftmann@34294
   416
haftmann@37745
   417
(** formal checking of compiled code **)
haftmann@37745
   418
haftmann@37745
   419
fun check thy = Code_Target.external_check thy target
haftmann@37745
   420
  "SCALA_HOME" I (fn scala_home => fn p => fn _ =>
haftmann@37745
   421
    Path.implode (Path.append (Path.explode scala_home) (Path.explode "bin/scalac")) ^ " *.scala");
haftmann@37745
   422
haftmann@37745
   423
haftmann@37745
   424
haftmann@34294
   425
(** Isar setup **)
haftmann@34294
   426
haftmann@34294
   427
fun isar_seri_scala module_name =
haftmann@34294
   428
  Code_Target.parse_args (Scan.succeed ())
haftmann@34294
   429
  #> (fn () => serialize_scala module_name);
haftmann@34294
   430
haftmann@34294
   431
val setup =
haftmann@34294
   432
  Code_Target.add_target (target, (isar_seri_scala, literals))
haftmann@37464
   433
  #> Code_Target.add_syntax_tyco target "fun"
haftmann@37464
   434
     (SOME (2, fn print_typ => fn fxy => fn [ty1, ty2] =>
haftmann@37464
   435
        brackify_infix (1, R) fxy (
haftmann@37464
   436
          print_typ BR ty1 (*product type vs. tupled arguments!*),
haftmann@37464
   437
          str "=>",
haftmann@37464
   438
          print_typ (INFX (1, R)) ty2
haftmann@37464
   439
        )))
haftmann@34294
   440
  #> fold (Code_Target.add_reserved target) [
haftmann@34294
   441
      "abstract", "case", "catch", "class", "def", "do", "else", "extends", "false",
haftmann@34294
   442
      "final", "finally", "for", "forSome", "if", "implicit", "import", "lazy",
haftmann@34294
   443
      "match", "new", "null", "object", "override", "package", "private", "protected",
haftmann@34294
   444
      "requires", "return", "sealed", "super", "this", "throw", "trait", "try",
haftmann@37243
   445
      "true", "type", "val", "var", "while", "with", "yield"
haftmann@34294
   446
    ]
haftmann@34294
   447
  #> fold (Code_Target.add_reserved target) [
haftmann@37639
   448
      "apply", "error", "BigInt", "Nil", "List"
haftmann@34294
   449
    ];
haftmann@34294
   450
haftmann@34294
   451
end; (*struct*)