src/Tools/Code/code_haskell.ML
author haftmann
Mon, 30 Aug 2010 16:33:06 +0200
changeset 39142 c0b857a04758
parent 39141 026526cba0e6
child 39149 79d7f2b4cf71
permissions -rw-r--r--
tuned
haftmann@37744
     1
(*  Title:      Tools/Code/code_haskell.ML
haftmann@28054
     2
    Author:     Florian Haftmann, TU Muenchen
haftmann@28054
     3
haftmann@28054
     4
Serializer for Haskell.
haftmann@28054
     5
*)
haftmann@28054
     6
haftmann@28054
     7
signature CODE_HASKELL =
haftmann@28054
     8
sig
haftmann@37744
     9
  val target: string
haftmann@28054
    10
  val setup: theory -> theory
haftmann@28054
    11
end;
haftmann@28054
    12
haftmann@28054
    13
structure Code_Haskell : CODE_HASKELL =
haftmann@28054
    14
struct
haftmann@28054
    15
haftmann@28054
    16
val target = "Haskell";
haftmann@28054
    17
haftmann@28054
    18
open Basic_Code_Thingol;
haftmann@28054
    19
open Code_Printer;
haftmann@28054
    20
haftmann@28054
    21
infixr 5 @@;
haftmann@28054
    22
infixr 5 @|;
haftmann@28054
    23
haftmann@28054
    24
haftmann@28054
    25
(** Haskell serializer **)
haftmann@28054
    26
haftmann@33988
    27
fun print_haskell_stmt labelled_name syntax_class syntax_tyco syntax_const
haftmann@32924
    28
    reserved deresolve contr_classparam_typs deriving_show =
haftmann@28054
    29
  let
wenzelm@30364
    30
    val deresolve_base = Long_Name.base_name o deresolve;
haftmann@28054
    31
    fun class_name class = case syntax_class class
haftmann@28054
    32
     of NONE => deresolve class
haftmann@28687
    33
      | SOME class => class;
haftmann@33988
    34
    fun print_typcontext tyvars vs = case maps (fn (v, sort) => map (pair v) sort) vs
haftmann@28054
    35
     of [] => []
haftmann@37359
    36
      | constraints => enum "," "(" ")" (
haftmann@28054
    37
          map (fn (v, class) =>
haftmann@37359
    38
            str (class_name class ^ " " ^ lookup_var tyvars v)) constraints)
haftmann@28054
    39
          @@ str " => ";
haftmann@33988
    40
    fun print_typforall tyvars vs = case map fst vs
haftmann@28054
    41
     of [] => []
haftmann@28054
    42
      | vnames => str "forall " :: Pretty.breaks
haftmann@32924
    43
          (map (str o lookup_var tyvars) vnames) @ str "." @@ Pretty.brk 1;
haftmann@33988
    44
    fun print_tyco_expr tyvars fxy (tyco, tys) =
haftmann@33988
    45
      brackify fxy (str tyco :: map (print_typ tyvars BR) tys)
haftmann@33988
    46
    and print_typ tyvars fxy (tycoexpr as tyco `%% tys) = (case syntax_tyco tyco
haftmann@33988
    47
         of NONE => print_tyco_expr tyvars fxy (deresolve tyco, tys)
haftmann@33988
    48
          | SOME (i, print) => print (print_typ tyvars) fxy tys)
haftmann@33988
    49
      | print_typ tyvars fxy (ITyVar v) = (str o lookup_var tyvars) v;
haftmann@33988
    50
    fun print_typdecl tyvars (vs, tycoexpr) =
haftmann@33988
    51
      Pretty.block (print_typcontext tyvars vs @| print_tyco_expr tyvars NOBR tycoexpr);
haftmann@33988
    52
    fun print_typscheme tyvars (vs, ty) =
haftmann@33988
    53
      Pretty.block (print_typforall tyvars vs @ print_typcontext tyvars vs @| print_typ tyvars NOBR ty);
haftmann@35228
    54
    fun print_term tyvars some_thm vars fxy (IConst c) =
haftmann@35228
    55
          print_app tyvars some_thm vars fxy (c, [])
haftmann@35228
    56
      | print_term tyvars some_thm vars fxy (t as (t1 `$ t2)) =
haftmann@28054
    57
          (case Code_Thingol.unfold_const_app t
haftmann@35228
    58
           of SOME app => print_app tyvars some_thm vars fxy app
haftmann@28054
    59
            | _ =>
haftmann@28054
    60
                brackify fxy [
haftmann@35228
    61
                  print_term tyvars some_thm vars NOBR t1,
haftmann@35228
    62
                  print_term tyvars some_thm vars BR t2
haftmann@28054
    63
                ])
haftmann@35228
    64
      | print_term tyvars some_thm vars fxy (IVar NONE) =
haftmann@31889
    65
          str "_"
haftmann@35228
    66
      | print_term tyvars some_thm vars fxy (IVar (SOME v)) =
haftmann@32924
    67
          (str o lookup_var vars) v
haftmann@35228
    68
      | print_term tyvars some_thm vars fxy (t as _ `|=> _) =
haftmann@28054
    69
          let
haftmann@31873
    70
            val (binds, t') = Code_Thingol.unfold_pat_abs t;
haftmann@35228
    71
            val (ps, vars') = fold_map (print_bind tyvars some_thm BR o fst) binds vars;
haftmann@35228
    72
          in brackets (str "\\" :: ps @ str "->" @@ print_term tyvars some_thm vars' NOBR t') end
haftmann@35228
    73
      | print_term tyvars some_thm vars fxy (ICase (cases as (_, t0))) =
haftmann@28054
    74
          (case Code_Thingol.unfold_const_app t0
haftmann@28054
    75
           of SOME (c_ts as ((c, _), _)) => if is_none (syntax_const c)
haftmann@35228
    76
                then print_case tyvars some_thm vars fxy cases
haftmann@35228
    77
                else print_app tyvars some_thm vars fxy c_ts
haftmann@35228
    78
            | NONE => print_case tyvars some_thm vars fxy cases)
haftmann@37424
    79
    and print_app_expr tyvars some_thm vars ((c, (_, function_typs)), ts) = case contr_classparam_typs c
haftmann@35228
    80
     of [] => (str o deresolve) c :: map (print_term tyvars some_thm vars BR) ts
haftmann@28054
    81
      | fingerprint => let
haftmann@33956
    82
          val ts_fingerprint = ts ~~ take (length ts) fingerprint;
haftmann@28054
    83
          val needs_annotation = forall (fn (_, NONE) => true | (t, SOME _) =>
haftmann@28054
    84
            (not o Code_Thingol.locally_monomorphic) t) ts_fingerprint;
haftmann@35228
    85
          fun print_term_anno (t, NONE) _ = print_term tyvars some_thm vars BR t
haftmann@33988
    86
            | print_term_anno (t, SOME _) ty =
haftmann@35228
    87
                brackets [print_term tyvars some_thm vars NOBR t, str "::", print_typ tyvars NOBR ty];
haftmann@28054
    88
        in
haftmann@28054
    89
          if needs_annotation then
haftmann@37424
    90
            (str o deresolve) c :: map2 print_term_anno ts_fingerprint (take (length ts) function_typs)
haftmann@35228
    91
          else (str o deresolve) c :: map (print_term tyvars some_thm vars BR) ts
haftmann@28054
    92
        end
haftmann@33988
    93
    and print_app tyvars = gen_print_app (print_app_expr tyvars) (print_term tyvars) syntax_const
haftmann@35228
    94
    and print_bind tyvars some_thm fxy p = gen_print_bind (print_term tyvars) some_thm fxy p
haftmann@35228
    95
    and print_case tyvars some_thm vars fxy (cases as ((_, [_]), _)) =
haftmann@28054
    96
          let
haftmann@29889
    97
            val (binds, body) = Code_Thingol.unfold_let (ICase cases);
haftmann@33988
    98
            fun print_match ((pat, ty), t) vars =
haftmann@28054
    99
              vars
haftmann@35228
   100
              |> print_bind tyvars some_thm BR pat
haftmann@35228
   101
              |>> (fn p => semicolon [p, str "=", print_term tyvars some_thm vars NOBR t])
haftmann@33988
   102
            val (ps, vars') = fold_map print_match binds vars;
haftmann@31665
   103
          in brackify_block fxy (str "let {")
haftmann@31665
   104
            ps
haftmann@35228
   105
            (concat [str "}", str "in", print_term tyvars some_thm vars' NOBR body])
haftmann@28054
   106
          end
haftmann@35228
   107
      | print_case tyvars some_thm vars fxy (((t, ty), clauses as _ :: _), _) =
haftmann@28054
   108
          let
haftmann@33988
   109
            fun print_select (pat, body) =
haftmann@28054
   110
              let
haftmann@35228
   111
                val (p, vars') = print_bind tyvars some_thm NOBR pat vars;
haftmann@35228
   112
              in semicolon [p, str "->", print_term tyvars some_thm vars' NOBR body] end;
haftmann@36574
   113
          in Pretty.block_enclose
haftmann@36574
   114
            (concat [str "(case", print_term tyvars some_thm vars NOBR t, str "of", str "{"], str "})")
haftmann@33988
   115
            (map print_select clauses)
haftmann@28054
   116
          end
haftmann@35228
   117
      | print_case tyvars some_thm vars fxy ((_, []), _) =
haftmann@31376
   118
          (brackify fxy o Pretty.breaks o map str) ["error", "\"empty case\""];
haftmann@37412
   119
    fun print_stmt (name, Code_Thingol.Fun (_, (((vs, ty), raw_eqs), _))) =
haftmann@28054
   120
          let
haftmann@32924
   121
            val tyvars = intro_vars (map fst vs) reserved;
haftmann@34262
   122
            fun print_err n =
haftmann@33988
   123
              semicolon (
haftmann@28054
   124
                (str o deresolve_base) name
haftmann@28054
   125
                :: map str (replicate n "_")
haftmann@28054
   126
                @ str "="
haftmann@28054
   127
                :: str "error"
haftmann@33988
   128
                @@ (str o ML_Syntax.print_string
wenzelm@30364
   129
                    o Long_Name.base_name o Long_Name.qualifier) name
haftmann@34262
   130
              );
haftmann@35228
   131
            fun print_eqn ((ts, t), (some_thm, _)) =
haftmann@28054
   132
              let
haftmann@32913
   133
                val consts = fold Code_Thingol.add_constnames (t :: ts) [];
haftmann@32924
   134
                val vars = reserved
haftmann@32924
   135
                  |> intro_base_names
haftmann@32913
   136
                      (is_none o syntax_const) deresolve consts
haftmann@32924
   137
                  |> intro_vars ((fold o Code_Thingol.fold_varnames)
haftmann@32925
   138
                      (insert (op =)) ts []);
haftmann@28054
   139
              in
haftmann@28054
   140
                semicolon (
haftmann@28054
   141
                  (str o deresolve_base) name
haftmann@35228
   142
                  :: map (print_term tyvars some_thm vars BR) ts
haftmann@28054
   143
                  @ str "="
haftmann@35228
   144
                  @@ print_term tyvars some_thm vars NOBR t
haftmann@28054
   145
                )
haftmann@28054
   146
              end;
haftmann@28054
   147
          in
haftmann@28054
   148
            Pretty.chunks (
haftmann@33988
   149
              semicolon [
haftmann@28054
   150
                (str o suffix " ::" o deresolve_base) name,
haftmann@33988
   151
                print_typscheme tyvars (vs, ty)
haftmann@28054
   152
              ]
haftmann@34262
   153
              :: (case filter (snd o snd) raw_eqs
haftmann@34262
   154
               of [] => [print_err ((length o fst o Code_Thingol.unfold_fun) ty)]
haftmann@34262
   155
                | eqs => map print_eqn eqs)
haftmann@28054
   156
            )
haftmann@28054
   157
          end
haftmann@33988
   158
      | print_stmt (name, Code_Thingol.Datatype (_, (vs, []))) =
haftmann@28054
   159
          let
haftmann@32924
   160
            val tyvars = intro_vars (map fst vs) reserved;
haftmann@28054
   161
          in
haftmann@28054
   162
            semicolon [
haftmann@28054
   163
              str "data",
haftmann@33988
   164
              print_typdecl tyvars (vs, (deresolve_base name, map (ITyVar o fst) vs))
haftmann@28054
   165
            ]
haftmann@28054
   166
          end
haftmann@37424
   167
      | print_stmt (name, Code_Thingol.Datatype (_, (vs, [((co, _), [ty])]))) =
haftmann@28054
   168
          let
haftmann@32924
   169
            val tyvars = intro_vars (map fst vs) reserved;
haftmann@28054
   170
          in
haftmann@28054
   171
            semicolon (
haftmann@28054
   172
              str "newtype"
haftmann@33988
   173
              :: print_typdecl tyvars (vs, (deresolve_base name, map (ITyVar o fst) vs))
haftmann@28054
   174
              :: str "="
haftmann@28054
   175
              :: (str o deresolve_base) co
haftmann@33988
   176
              :: print_typ tyvars BR ty
haftmann@28054
   177
              :: (if deriving_show name then [str "deriving (Read, Show)"] else [])
haftmann@28054
   178
            )
haftmann@28054
   179
          end
haftmann@33988
   180
      | print_stmt (name, Code_Thingol.Datatype (_, (vs, co :: cos))) =
haftmann@28054
   181
          let
haftmann@32924
   182
            val tyvars = intro_vars (map fst vs) reserved;
haftmann@37424
   183
            fun print_co ((co, _), tys) =
haftmann@28054
   184
              concat (
haftmann@28054
   185
                (str o deresolve_base) co
haftmann@33988
   186
                :: map (print_typ tyvars BR) tys
haftmann@28054
   187
              )
haftmann@28054
   188
          in
haftmann@28054
   189
            semicolon (
haftmann@28054
   190
              str "data"
haftmann@33988
   191
              :: print_typdecl tyvars (vs, (deresolve_base name, map (ITyVar o fst) vs))
haftmann@28054
   192
              :: str "="
haftmann@33988
   193
              :: print_co co
haftmann@33988
   194
              :: map ((fn p => Pretty.block [str "| ", p]) o print_co) cos
haftmann@28054
   195
              @ (if deriving_show name then [str "deriving (Read, Show)"] else [])
haftmann@28054
   196
            )
haftmann@28054
   197
          end
haftmann@37422
   198
      | print_stmt (name, Code_Thingol.Class (_, (v, (super_classes, classparams)))) =
haftmann@28054
   199
          let
haftmann@32924
   200
            val tyvars = intro_vars [v] reserved;
haftmann@33988
   201
            fun print_classparam (classparam, ty) =
haftmann@28054
   202
              semicolon [
haftmann@28687
   203
                (str o deresolve_base) classparam,
haftmann@28054
   204
                str "::",
haftmann@33988
   205
                print_typ tyvars NOBR ty
haftmann@28054
   206
              ]
haftmann@28054
   207
          in
haftmann@28054
   208
            Pretty.block_enclose (
haftmann@28054
   209
              Pretty.block [
haftmann@28054
   210
                str "class ",
haftmann@37359
   211
                Pretty.block (print_typcontext tyvars [(v, map fst super_classes)]),
haftmann@32924
   212
                str (deresolve_base name ^ " " ^ lookup_var tyvars v),
haftmann@28054
   213
                str " where {"
haftmann@28054
   214
              ],
haftmann@28054
   215
              str "};"
haftmann@33988
   216
            ) (map print_classparam classparams)
haftmann@28054
   217
          end
haftmann@37424
   218
      | print_stmt (_, Code_Thingol.Classinst ((class, (tyco, vs)), (_, (classparam_instances, _)))) =
haftmann@28054
   219
          let
haftmann@32924
   220
            val tyvars = intro_vars (map fst vs) reserved;
haftmann@37854
   221
            fun requires_args classparam = case syntax_const classparam
haftmann@37854
   222
             of NONE => 0
haftmann@37854
   223
              | SOME (Code_Printer.Plain_const_syntax _) => 0
haftmann@37854
   224
              | SOME (Code_Printer.Complex_const_syntax (k,_ )) => k;
haftmann@37854
   225
            fun print_classparam_instance ((classparam, const), (thm, _)) =
haftmann@37854
   226
              case requires_args classparam
haftmann@37854
   227
               of 0 => semicolon [
haftmann@37854
   228
                      (str o deresolve_base) classparam,
haftmann@28687
   229
                      str "=",
haftmann@37854
   230
                      print_app tyvars (SOME thm) reserved NOBR (const, [])
haftmann@28687
   231
                    ]
haftmann@37854
   232
                | k =>
haftmann@37854
   233
                    let
haftmann@37854
   234
                      val (c, (_, tys)) = const;
haftmann@37854
   235
                      val (vs, rhs) = (apfst o map) fst
haftmann@37854
   236
                        (Code_Thingol.unfold_abs (Code_Thingol.eta_expand k (const, [])));
haftmann@37854
   237
                      val s = if (is_some o syntax_const) c
haftmann@37854
   238
                        then NONE else (SOME o Long_Name.base_name o deresolve) c;
haftmann@37854
   239
                      val vars = reserved
haftmann@37854
   240
                        |> intro_vars (map_filter I (s :: vs));
haftmann@37854
   241
                      val lhs = IConst (classparam, (([], []), tys)) `$$ map IVar vs;
haftmann@37854
   242
                        (*dictionaries are not relevant at this late stage*)
haftmann@37854
   243
                    in
haftmann@37854
   244
                      semicolon [
haftmann@37854
   245
                        print_term tyvars (SOME thm) vars NOBR lhs,
haftmann@37854
   246
                        str "=",
haftmann@37854
   247
                        print_term tyvars (SOME thm) vars NOBR rhs
haftmann@37854
   248
                      ]
haftmann@37854
   249
                    end;
haftmann@28054
   250
          in
haftmann@28054
   251
            Pretty.block_enclose (
haftmann@28054
   252
              Pretty.block [
haftmann@28054
   253
                str "instance ",
haftmann@33988
   254
                Pretty.block (print_typcontext tyvars vs),
haftmann@28054
   255
                str (class_name class ^ " "),
haftmann@33988
   256
                print_typ tyvars BR (tyco `%% map (ITyVar o fst) vs),
haftmann@28054
   257
                str " where {"
haftmann@28054
   258
              ],
haftmann@28054
   259
              str "};"
haftmann@37359
   260
            ) (map print_classparam_instance classparam_instances)
haftmann@28054
   261
          end;
haftmann@33988
   262
  in print_stmt end;
haftmann@28054
   263
haftmann@39012
   264
fun haskell_program_of_program labelled_name module_prefix reserved module_alias program =
haftmann@28054
   265
  let
haftmann@32924
   266
    val reserved = Name.make_context reserved;
haftmann@32924
   267
    val mk_name_module = mk_name_module reserved module_prefix module_alias program;
haftmann@28054
   268
    fun add_stmt (name, (stmt, deps)) =
haftmann@28054
   269
      let
haftmann@32924
   270
        val (module_name, base) = dest_name name;
haftmann@28054
   271
        val module_name' = mk_name_module module_name;
haftmann@28054
   272
        val mk_name_stmt = yield_singleton Name.variants;
haftmann@28054
   273
        fun add_fun upper (nsp_fun, nsp_typ) =
haftmann@28054
   274
          let
haftmann@28054
   275
            val (base', nsp_fun') =
haftmann@32924
   276
              mk_name_stmt (if upper then first_upper base else base) nsp_fun
haftmann@28054
   277
          in (base', (nsp_fun', nsp_typ)) end;
haftmann@28054
   278
        fun add_typ (nsp_fun, nsp_typ) =
haftmann@28054
   279
          let
haftmann@32924
   280
            val (base', nsp_typ') = mk_name_stmt (first_upper base) nsp_typ
haftmann@28054
   281
          in (base', (nsp_fun, nsp_typ')) end;
haftmann@28054
   282
        val add_name = case stmt
haftmann@37414
   283
         of Code_Thingol.Fun (_, (_, SOME _)) => pair base
haftmann@37414
   284
          | Code_Thingol.Fun _ => add_fun false
haftmann@28054
   285
          | Code_Thingol.Datatype _ => add_typ
haftmann@28054
   286
          | Code_Thingol.Datatypecons _ => add_fun true
haftmann@28054
   287
          | Code_Thingol.Class _ => add_typ
haftmann@28054
   288
          | Code_Thingol.Classrel _ => pair base
haftmann@28054
   289
          | Code_Thingol.Classparam _ => add_fun false
haftmann@28054
   290
          | Code_Thingol.Classinst _ => pair base;
haftmann@28054
   291
        fun add_stmt' base' = case stmt
haftmann@37414
   292
         of Code_Thingol.Fun (_, (_, SOME _)) =>
haftmann@37414
   293
              I
haftmann@37414
   294
          | Code_Thingol.Datatypecons _ =>
wenzelm@30364
   295
              cons (name, (Long_Name.append module_name' base', NONE))
haftmann@28054
   296
          | Code_Thingol.Classrel _ => I
haftmann@28054
   297
          | Code_Thingol.Classparam _ =>
wenzelm@30364
   298
              cons (name, (Long_Name.append module_name' base', NONE))
wenzelm@30364
   299
          | _ => cons (name, (Long_Name.append module_name' base', SOME stmt));
haftmann@28054
   300
      in
haftmann@32924
   301
        Symtab.map_default (module_name', ([], ([], (reserved, reserved))))
haftmann@28054
   302
              (apfst (fold (insert (op = : string * string -> bool)) deps))
haftmann@28054
   303
        #> `(fn program => add_name ((snd o snd o the o Symtab.lookup program) module_name'))
haftmann@28054
   304
        #-> (fn (base', names) =>
haftmann@28054
   305
              (Symtab.map_entry module_name' o apsnd) (fn (stmts, _) =>
haftmann@28054
   306
              (add_stmt' base' stmts, names)))
haftmann@28054
   307
      end;
haftmann@28054
   308
    val hs_program = fold add_stmt (AList.make (fn name =>
haftmann@28054
   309
      (Graph.get_node program name, Graph.imm_succs program name))
haftmann@28054
   310
      (Graph.strong_conn program |> flat)) Symtab.empty;
haftmann@28663
   311
    fun deresolver name = (fst o the o AList.lookup (op =) ((fst o snd o the
haftmann@32924
   312
      o Symtab.lookup hs_program) ((mk_name_module o fst o dest_name) name))) name
haftmann@28663
   313
      handle Option => error ("Unknown statement name: " ^ labelled_name name);
haftmann@28054
   314
  in (deresolver, hs_program) end;
haftmann@28054
   315
haftmann@39012
   316
fun serialize_haskell module_prefix module_name string_classes labelled_name
haftmann@39012
   317
    raw_reserved includes module_alias
haftmann@39138
   318
    syntax_class syntax_tyco syntax_const program
haftmann@39139
   319
    (stmt_names, presentation_stmt_names) =
haftmann@28054
   320
  let
haftmann@32924
   321
    val reserved = fold (insert (op =) o fst) includes raw_reserved;
haftmann@28054
   322
    val (deresolver, hs_program) = haskell_program_of_program labelled_name
haftmann@39012
   323
      module_prefix reserved module_alias program;
haftmann@28054
   324
    val contr_classparam_typs = Code_Thingol.contr_classparam_typs program;
haftmann@28054
   325
    fun deriving_show tyco =
haftmann@28054
   326
      let
haftmann@28054
   327
        fun deriv _ "fun" = false
haftmann@34085
   328
          | deriv tycos tyco = not (tyco = Code_Thingol.fun_tyco)
haftmann@34085
   329
              andalso (member (op =) tycos tyco
haftmann@34085
   330
              orelse case try (Graph.get_node program) tyco
haftmann@28663
   331
                of SOME (Code_Thingol.Datatype (_, (_, cs))) => forall (deriv' (tyco :: tycos))
haftmann@28054
   332
                    (maps snd cs)
haftmann@34085
   333
                 | NONE => true)
haftmann@28054
   334
        and deriv' tycos (tyco `%% tys) = deriv tycos tyco
haftmann@28054
   335
              andalso forall (deriv' tycos) tys
haftmann@28054
   336
          | deriv' _ (ITyVar _) = true
haftmann@28054
   337
      in deriv [] tyco end;
haftmann@32924
   338
    val reserved = make_vars reserved;
haftmann@33988
   339
    fun print_stmt qualified = print_haskell_stmt labelled_name
haftmann@32924
   340
      syntax_class syntax_tyco syntax_const reserved
wenzelm@30364
   341
      (if qualified then deresolver else Long_Name.base_name o deresolver)
haftmann@32903
   342
      contr_classparam_typs
haftmann@28054
   343
      (if string_classes then deriving_show else K false);
haftmann@33988
   344
    fun print_module name content =
haftmann@39001
   345
      (name, Pretty.chunks2 [
haftmann@28054
   346
        str ("module " ^ name ^ " where {"),
haftmann@28054
   347
        content,
haftmann@28054
   348
        str "}"
haftmann@28054
   349
      ]);
haftmann@28054
   350
    fun serialize_module1 (module_name', (deps, (stmts, _))) =
haftmann@28054
   351
      let
haftmann@28054
   352
        val stmt_names = map fst stmts;
haftmann@33410
   353
        val qualified = is_none module_name;
haftmann@33410
   354
        val imports = subtract (op =) stmt_names deps
haftmann@28054
   355
          |> distinct (op =)
haftmann@33410
   356
          |> map_filter (try deresolver)
wenzelm@30364
   357
          |> map Long_Name.qualifier
haftmann@28054
   358
          |> distinct (op =);
haftmann@33988
   359
        fun print_import_include (name, _) = str ("import qualified " ^ name ^ ";");
haftmann@34045
   360
        fun print_import_module name = str ((if qualified
haftmann@34045
   361
          then "import qualified "
haftmann@34045
   362
          else "import ") ^ name ^ ";");
haftmann@33988
   363
        val import_ps = map print_import_include includes @ map print_import_module imports
haftmann@34073
   364
        val content = Pretty.chunks2 ((if null import_ps then [] else [Pretty.chunks import_ps])
haftmann@33988
   365
            @ map_filter
haftmann@33988
   366
              (fn (name, (_, SOME stmt)) => SOME (print_stmt qualified (name, stmt))
haftmann@33988
   367
                | (_, (_, NONE)) => NONE) stmts
haftmann@33988
   368
          );
haftmann@33988
   369
      in print_module module_name' content end;
haftmann@33988
   370
    fun serialize_module2 (_, (_, (stmts, _))) = Pretty.chunks2 (map_filter
haftmann@36535
   371
        (fn (name, (_, SOME stmt)) => if null presentation_stmt_names
haftmann@36535
   372
              orelse member (op =) presentation_stmt_names name
haftmann@33988
   373
              then SOME (print_stmt false (name, stmt))
haftmann@28054
   374
              else NONE
haftmann@33988
   375
          | (_, (_, NONE)) => NONE) stmts);
haftmann@28054
   376
    val serialize_module =
haftmann@36535
   377
      if null presentation_stmt_names then serialize_module1 else pair "" o serialize_module2;
haftmann@39141
   378
    fun write_module width (SOME destination) (modlname, content) =
haftmann@39141
   379
          let
haftmann@39141
   380
            val _ = File.check destination;
haftmann@39141
   381
            val filename = case modlname
haftmann@39141
   382
             of "" => Path.explode "Main.hs"
haftmann@39141
   383
              | _ => (Path.ext "hs" o Path.explode o implode o separate "/"
haftmann@39141
   384
                    o Long_Name.explode) modlname;
haftmann@39141
   385
            val pathname = Path.append destination filename;
haftmann@39141
   386
            val _ = File.mkdir_leaf (Path.dir pathname);
haftmann@39141
   387
          in File.write pathname
haftmann@39141
   388
            ("{-# OPTIONS_GHC -fglasgow-exts #-}\n\n"
haftmann@39141
   389
              ^ string_of_pretty width content)
haftmann@39141
   390
          end
haftmann@39141
   391
      | write_module width NONE (_, content) = writeln_pretty width content;
haftmann@28054
   392
  in
haftmann@39142
   393
    Code_Target.serialization
haftmann@39141
   394
      (fn width => fn destination => K () o map (write_module width destination))
haftmann@39141
   395
      (fn width => rpair [] o cat_lines o map (string_of_pretty width o snd))
haftmann@33988
   396
      (map (uncurry print_module) includes
haftmann@28054
   397
        @ map serialize_module (Symtab.dest hs_program))
haftmann@28054
   398
  end;
haftmann@28054
   399
haftmann@28064
   400
val literals = let
haftmann@28064
   401
  fun char_haskell c =
haftmann@28064
   402
    let
haftmann@28064
   403
      val s = ML_Syntax.print_char c;
haftmann@28064
   404
    in if s = "'" then "\\'" else s end;
haftmann@34931
   405
  fun numeral_haskell k = if k >= 0 then string_of_int k
haftmann@34931
   406
    else Library.enclose "(" ")" (signed_string_of_int k);
haftmann@28064
   407
in Literals {
haftmann@34178
   408
  literal_char = Library.enclose "'" "'" o char_haskell,
haftmann@28064
   409
  literal_string = quote o translate_string char_haskell,
haftmann@34931
   410
  literal_numeral = numeral_haskell,
haftmann@34931
   411
  literal_positive_numeral = numeral_haskell,
haftmann@38195
   412
  literal_alternative_numeral = numeral_haskell,
haftmann@34931
   413
  literal_naive_numeral = numeral_haskell,
haftmann@34178
   414
  literal_list = enum "," "[" "]",
haftmann@28064
   415
  infix_cons = (5, ":")
haftmann@28064
   416
} end;
haftmann@28064
   417
haftmann@28054
   418
haftmann@28054
   419
(** optional monad syntax **)
haftmann@28054
   420
haftmann@28054
   421
fun pretty_haskell_monad c_bind =
haftmann@28054
   422
  let
haftmann@31873
   423
    fun dest_bind t1 t2 = case Code_Thingol.split_pat_abs t2
haftmann@31889
   424
     of SOME ((pat, ty), t') =>
haftmann@31889
   425
          SOME ((SOME ((pat, ty), true), t1), t')
haftmann@28145
   426
      | NONE => NONE;
haftmann@28663
   427
    fun dest_monad c_bind_name (IConst (c, _) `$ t1 `$ t2) =
haftmann@28663
   428
          if c = c_bind_name then dest_bind t1 t2
haftmann@28145
   429
          else NONE
haftmann@28663
   430
      | dest_monad _ t = case Code_Thingol.split_let t
haftmann@28145
   431
         of SOME (((pat, ty), tbind), t') =>
haftmann@31889
   432
              SOME ((SOME ((pat, ty), false), tbind), t')
haftmann@28145
   433
          | NONE => NONE;
haftmann@28663
   434
    fun implode_monad c_bind_name = Code_Thingol.unfoldr (dest_monad c_bind_name);
haftmann@33988
   435
    fun print_monad print_bind print_term (NONE, t) vars =
haftmann@33988
   436
          (semicolon [print_term vars NOBR t], vars)
haftmann@33988
   437
      | print_monad print_bind print_term (SOME ((bind, _), true), t) vars = vars
haftmann@33988
   438
          |> print_bind NOBR bind
haftmann@33988
   439
          |>> (fn p => semicolon [p, str "<-", print_term vars NOBR t])
haftmann@33988
   440
      | print_monad print_bind print_term (SOME ((bind, _), false), t) vars = vars
haftmann@33988
   441
          |> print_bind NOBR bind
haftmann@37832
   442
          |>> (fn p => semicolon [str "let", str "{", p, str "=", print_term vars NOBR t, str "}"]);
haftmann@33988
   443
    fun pretty _ [c_bind'] print_term thm vars fxy [(t1, _), (t2, _)] = case dest_bind t1 t2
haftmann@28145
   444
     of SOME (bind, t') => let
haftmann@31054
   445
          val (binds, t'') = implode_monad c_bind' t'
haftmann@33988
   446
          val (ps, vars') = fold_map (print_monad (gen_print_bind (K print_term) thm) print_term)
haftmann@33988
   447
            (bind :: binds) vars;
haftmann@33988
   448
        in
haftmann@37833
   449
          (brackify fxy o single o enclose "do { " " }" o Pretty.breaks)
haftmann@33988
   450
            (ps @| print_term vars' NOBR t'')
haftmann@33988
   451
        end
haftmann@28145
   452
      | NONE => brackify_infix (1, L) fxy
haftmann@37239
   453
          (print_term vars (INFX (1, L)) t1, str ">>=", print_term vars (INFX (1, X)) t2)
haftmann@31054
   454
  in (2, ([c_bind], pretty)) end;
haftmann@28054
   455
haftmann@28145
   456
fun add_monad target' raw_c_bind thy =
haftmann@28054
   457
  let
haftmann@31156
   458
    val c_bind = Code.read_const thy raw_c_bind;
haftmann@28054
   459
  in if target = target' then
haftmann@28054
   460
    thy
haftmann@28145
   461
    |> Code_Target.add_syntax_const target c_bind
haftmann@37854
   462
        (SOME (Code_Printer.complex_const_syntax (pretty_haskell_monad c_bind)))
haftmann@28054
   463
  else error "Only Haskell target allows for monad syntax" end;
haftmann@28054
   464
haftmann@28054
   465
haftmann@28054
   466
(** Isar setup **)
haftmann@28054
   467
haftmann@37821
   468
fun isar_serializer module_name =
haftmann@28054
   469
  Code_Target.parse_args (Scan.option (Args.$$$ "root" -- Args.colon |-- Args.name)
haftmann@28054
   470
    -- Scan.optional (Args.$$$ "string_classes" >> K true) false
haftmann@28054
   471
    >> (fn (module_prefix, string_classes) =>
haftmann@33988
   472
      serialize_haskell module_prefix module_name string_classes));
haftmann@28054
   473
haftmann@28054
   474
val _ =
wenzelm@36970
   475
  Outer_Syntax.command "code_monad" "define code syntax for monads" Keyword.thy_decl (
wenzelm@36970
   476
    Parse.term_group -- Parse.name >> (fn (raw_bind, target) =>
haftmann@28145
   477
      Toplevel.theory  (add_monad target raw_bind))
haftmann@28054
   478
  );
haftmann@28054
   479
haftmann@28054
   480
val setup =
haftmann@37821
   481
  Code_Target.add_target
haftmann@37822
   482
    (target, { serializer = isar_serializer, literals = literals,
haftmann@37822
   483
      check = { env_var = "EXEC_GHC", make_destination = I,
haftmann@39092
   484
        make_command = fn ghc => fn module_name =>
haftmann@37822
   485
          ghc ^ " -fglasgow-exts -odir build -hidir build -stubdir build -e \"\" " ^ module_name ^ ".hs" } })
haftmann@33988
   486
  #> Code_Target.add_syntax_tyco target "fun" (SOME (2, fn print_typ => fn fxy => fn [ty1, ty2] =>
haftmann@37239
   487
      brackify_infix (1, R) fxy (
haftmann@33988
   488
        print_typ (INFX (1, X)) ty1,
haftmann@28054
   489
        str "->",
haftmann@33988
   490
        print_typ (INFX (1, R)) ty2
haftmann@37239
   491
      )))
haftmann@28054
   492
  #> fold (Code_Target.add_reserved target) [
haftmann@28054
   493
      "hiding", "deriving", "where", "case", "of", "infix", "infixl", "infixr",
haftmann@28054
   494
      "import", "default", "forall", "let", "in", "class", "qualified", "data",
haftmann@28054
   495
      "newtype", "instance", "if", "then", "else", "type", "as", "do", "module"
haftmann@28054
   496
    ]
haftmann@28054
   497
  #> fold (Code_Target.add_reserved target) [
haftmann@28054
   498
      "Prelude", "Main", "Bool", "Maybe", "Either", "Ordering", "Char", "String", "Int",
haftmann@28054
   499
      "Integer", "Float", "Double", "Rational", "IO", "Eq", "Ord", "Enum", "Bounded",
haftmann@28054
   500
      "Num", "Real", "Integral", "Fractional", "Floating", "RealFloat", "Monad", "Functor",
haftmann@28054
   501
      "AlreadyExists", "ArithException", "ArrayException", "AssertionFailed", "AsyncException",
haftmann@28054
   502
      "BlockedOnDeadMVar", "Deadlock", "Denormal", "DivideByZero", "DotNetException", "DynException",
haftmann@28054
   503
      "Dynamic", "EOF", "EQ", "EmptyRec", "ErrorCall", "ExitException", "ExitFailure",
haftmann@28054
   504
      "ExitSuccess", "False", "GT", "HeapOverflow",
haftmann@28054
   505
      "IOError", "IOException", "IllegalOperation",
haftmann@28054
   506
      "IndexOutOfBounds", "Just", "Key", "LT", "Left", "LossOfPrecision", "NoMethodError",
haftmann@28054
   507
      "NoSuchThing", "NonTermination", "Nothing", "Obj", "OtherError", "Overflow",
haftmann@28054
   508
      "PatternMatchFail", "PermissionDenied", "ProtocolError", "RecConError", "RecSelError",
haftmann@28054
   509
      "RecUpdError", "ResourceBusy", "ResourceExhausted", "Right", "StackOverflow",
haftmann@28054
   510
      "ThreadKilled", "True", "TyCon", "TypeRep", "UndefinedElement", "Underflow",
haftmann@28054
   511
      "UnsupportedOperation", "UserError", "abs", "absReal", "acos", "acosh", "all",
haftmann@28054
   512
      "and", "any", "appendFile", "asTypeOf", "asciiTab", "asin", "asinh", "atan",
haftmann@28054
   513
      "atan2", "atanh", "basicIORun", "blockIO", "boundedEnumFrom", "boundedEnumFromThen",
haftmann@28054
   514
      "boundedEnumFromThenTo", "boundedEnumFromTo", "boundedPred", "boundedSucc", "break",
haftmann@28054
   515
      "catch", "catchException", "ceiling", "compare", "concat", "concatMap", "const",
haftmann@28054
   516
      "cos", "cosh", "curry", "cycle", "decodeFloat", "denominator", "div", "divMod",
haftmann@28054
   517
      "doubleToRatio", "doubleToRational", "drop", "dropWhile", "either", "elem",
haftmann@28054
   518
      "emptyRec", "encodeFloat", "enumFrom", "enumFromThen", "enumFromThenTo",
haftmann@28054
   519
      "enumFromTo", "error", "even", "exp", "exponent", "fail", "filter", "flip",
haftmann@28054
   520
      "floatDigits", "floatProperFraction", "floatRadix", "floatRange", "floatToRational",
haftmann@28054
   521
      "floor", "fmap", "foldl", "foldl'", "foldl1", "foldr", "foldr1", "fromDouble",
haftmann@28054
   522
      "fromEnum", "fromEnum_0", "fromInt", "fromInteger", "fromIntegral", "fromObj",
haftmann@28054
   523
      "fromRational", "fst", "gcd", "getChar", "getContents", "getLine", "head",
haftmann@28054
   524
      "id", "inRange", "index", "init", "intToRatio", "interact", "ioError", "isAlpha",
haftmann@28054
   525
      "isAlphaNum", "isDenormalized", "isDigit", "isHexDigit", "isIEEE", "isInfinite",
haftmann@28054
   526
      "isLower", "isNaN", "isNegativeZero", "isOctDigit", "isSpace", "isUpper", "iterate", "iterate'",
haftmann@28054
   527
      "last", "lcm", "length", "lex", "lexDigits", "lexLitChar", "lexmatch", "lines", "log",
haftmann@28054
   528
      "logBase", "lookup", "loop", "map", "mapM", "mapM_", "max", "maxBound", "maximum",
haftmann@28054
   529
      "maybe", "min", "minBound", "minimum", "mod", "negate", "nonnull", "not", "notElem",
haftmann@28054
   530
      "null", "numerator", "numericEnumFrom", "numericEnumFromThen", "numericEnumFromThenTo",
haftmann@28054
   531
      "numericEnumFromTo", "odd", "or", "otherwise", "pi", "pred", 
haftmann@28054
   532
      "print", "product", "properFraction", "protectEsc", "putChar", "putStr", "putStrLn",
haftmann@28054
   533
      "quot", "quotRem", "range", "rangeSize", "rationalToDouble", "rationalToFloat",
haftmann@28054
   534
      "rationalToRealFloat", "read", "readDec", "readField", "readFieldName", "readFile",
haftmann@28054
   535
      "readFloat", "readHex", "readIO", "readInt", "readList", "readLitChar", "readLn",
haftmann@28054
   536
      "readOct", "readParen", "readSigned", "reads", "readsPrec", "realFloatToRational",
haftmann@28054
   537
      "realToFrac", "recip", "reduce", "rem", "repeat", "replicate", "return", "reverse",
haftmann@28054
   538
      "round", "scaleFloat", "scanl", "scanl1", "scanr", "scanr1", "seq", "sequence",
haftmann@28054
   539
      "sequence_", "show", "showChar", "showException", "showField", "showList",
haftmann@28054
   540
      "showLitChar", "showParen", "showString", "shows", "showsPrec", "significand",
haftmann@28054
   541
      "signum", "signumReal", "sin", "sinh", "snd", "span", "splitAt", "sqrt", "subtract",
haftmann@28054
   542
      "succ", "sum", "tail", "take", "takeWhile", "takeWhile1", "tan", "tanh", "threadToIOResult",
haftmann@28054
   543
      "throw", "toEnum", "toInt", "toInteger", "toObj", "toRational", "truncate", "uncurry",
haftmann@28054
   544
      "undefined", "unlines", "unsafeCoerce", "unsafeIndex", "unsafeRangeSize", "until", "unwords",
haftmann@28054
   545
      "unzip", "unzip3", "userError", "words", "writeFile", "zip", "zip3", "zipWith", "zipWith3"
haftmann@28054
   546
    ] (*due to weird handling of ':', we can't do anything else than to import *all* prelude symbols*);
haftmann@28054
   547
haftmann@28054
   548
end; (*struct*)