Mercurial > urweb

annotate src/cjrize.sml @ 820:91f465ded07e

Find changesets by keywords (author, files, the commit message), revision number or hash, or revset expression.
Change monoization of monads to allow partial applications of operations
author Adam Chlipala <adamc@hcoop.net>
date Sat, 23 May 2009 10:14:51 -0400
parents 493f44759879
children 7a4b026e45dd
rev   line source
adamc@29 1 (* Copyright (c) 2008, Adam Chlipala
adamc@29 2 * All rights reserved.
adamc@29 3 *
adamc@29 4 * Redistribution and use in source and binary forms, with or without
adamc@29 5 * modification, are permitted provided that the following conditions are met:
adamc@29 6 *
adamc@29 7 * - Redistributions of source code must retain the above copyright notice,
adamc@29 8 * this list of conditions and the following disclaimer.
adamc@29 9 * - Redistributions in binary form must reproduce the above copyright notice,
adamc@29 10 * this list of conditions and the following disclaimer in the documentation
adamc@29 11 * and/or other materials provided with the distribution.
adamc@29 12 * - The names of contributors may not be used to endorse or promote products
adamc@29 13 * derived from this software without specific prior written permission.
adamc@29 14 *
adamc@29 15 * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS"
adamc@29 16 * AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
adamc@29 17 * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
adamc@29 18 * ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT OWNER OR CONTRIBUTORS BE
adamc@29 19 * LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
adamc@29 20 * CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
adamc@29 21 * SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
adamc@29 22 * INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
adamc@29 23 * CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
adamc@29 24 * ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
adamc@29 25 * POSSIBILITY OF SUCH DAMAGE.
adamc@29 26 *)
adamc@29 27
adamc@29 28 structure Cjrize :> CJRIZE = struct
adamc@29 29
adamc@109 30 structure L = Mono
adamc@29 31 structure L' = Cjr
adamc@29 32
adamc@196 33 structure IM = IntBinaryMap
adamc@196 34
adamc@29 35 structure Sm :> sig
adamc@29 36 type t
adamc@29 37
adamc@29 38 val empty : t
adamc@29 39 val find : t * (string * L.typ) list * (string * L'.typ) list -> t * int
adamc@757 40 val findList : t * L.typ * L'.typ -> t * int
adamc@29 41
adamc@29 42 val declares : t -> (int * (string * L'.typ) list) list
adamc@453 43 val clearDeclares : t -> t
adamc@29 44 end = struct
adamc@29 45
adamc@29 46 structure FM = BinaryMapFn(struct
adamc@29 47 type ord_key = L.typ
adamc@109 48 val compare = MonoUtil.Typ.compare
adamc@29 49 end)
adamc@29 50
adamc@757 51 type t = {
adamc@757 52 count : int,
adamc@757 53 normal : int FM.map,
adamc@757 54 lists : int FM.map,
adamc@757 55 decls : (int * (string * L'.typ) list) list
adamc@757 56 }
adamc@29 57
adamc@757 58 val empty : t = {
adamc@757 59 count = 1,
adamc@757 60 normal = FM.insert (FM.empty, (L.TRecord [], ErrorMsg.dummySpan), 0),
adamc@757 61 lists = FM.empty,
adamc@757 62 decls = []
adamc@757 63 }
adamc@29 64
adamc@757 65 fun find (v as {count, normal, decls, lists}, xts, xts') =
adamc@29 66 let
adamc@29 67 val t = (L.TRecord xts, ErrorMsg.dummySpan)
adamc@29 68 in
adamc@757 69 case FM.find (normal, t) of
adamc@757 70 SOME i => (v, i)
adamc@757 71 | NONE => ({count = count+1,
adamc@757 72 normal = FM.insert (normal, t, count),
adamc@757 73 lists = lists,
adamc@757 74 decls = (count, xts') :: decls},
adamc@757 75 count)
adamc@29 76 end
adamc@29 77
adamc@757 78 fun findList (v as {count, normal, decls, lists}, t, t') =
adamc@757 79 case FM.find (lists, t) of
adamc@757 80 SOME i => (v, i)
adamc@757 81 | NONE =>
adamc@757 82 let
adamc@757 83 val xts = [("1", t), ("2", (L.TList t, #2 t))]
adamc@757 84 val xts' = [("1", t'), ("2", (L'.TList (t', count), #2 t'))]
adamc@757 85 in
adamc@757 86 ({count = count+1,
adamc@757 87 normal = FM.insert (normal, (L.TRecord xts, ErrorMsg.dummySpan), count),
adamc@757 88 lists = FM.insert (lists, t, count),
adamc@757 89 decls = (count, xts') :: decls},
adamc@757 90 count)
adamc@757 91 end
adamc@29 92
adamc@757 93 fun declares (v : t) = #decls v
adamc@757 94
adamc@757 95 fun clearDeclares (v : t) = {count = #count v,
adamc@757 96 normal = #normal v,
adamc@757 97 lists = #lists v,
adamc@757 98 decls = []}
adamc@453 99
adamc@29 100 end
adamc@29 101
adamc@196 102 fun cifyTyp x =
adamc@196 103 let
adamc@196 104 fun cify dtmap ((t, loc), sm) =
adamc@196 105 case t of
adamc@196 106 L.TFun (t1, t2) =>
adamc@196 107 let
adamc@196 108 val (t1, sm) = cify dtmap (t1, sm)
adamc@196 109 val (t2, sm) = cify dtmap (t2, sm)
adamc@196 110 in
adamc@196 111 ((L'.TFun (t1, t2), loc), sm)
adamc@196 112 end
adamc@196 113 | L.TRecord xts =>
adamc@196 114 let
adamc@196 115 val old_xts = xts
adamc@196 116 val (xts, sm) = ListUtil.foldlMap (fn ((x, t), sm) =>
adamc@196 117 let
adamc@196 118 val (t, sm) = cify dtmap (t, sm)
adamc@196 119 in
adamc@196 120 ((x, t), sm)
adamc@196 121 end)
adamc@196 122 sm xts
adamc@196 123 val (sm, si) = Sm.find (sm, old_xts, xts)
adamc@196 124 in
adamc@196 125 ((L'.TRecord si, loc), sm)
adamc@196 126 end
adamc@196 127 | L.TDatatype (n, ref (dk, xncs)) =>
adamc@196 128 (case IM.find (dtmap, n) of
adamc@196 129 SOME r => ((L'.TDatatype (dk, n, r), loc), sm)
adamc@196 130 | NONE =>
adamc@196 131 let
adamc@196 132 val r = ref []
adamc@196 133 val dtmap = IM.insert (dtmap, n, r)
adamc@196 134
adamc@196 135 val (xncs, sm) = ListUtil.foldlMap (fn ((x, n, to), sm) =>
adamc@196 136 case to of
adamc@196 137 NONE => ((x, n, NONE), sm)
adamc@196 138 | SOME t =>
adamc@196 139 let
adamc@196 140 val (t, sm) = cify dtmap (t, sm)
adamc@196 141 in
adamc@196 142 ((x, n, SOME t), sm)
adamc@196 143 end)
adamc@196 144 sm xncs
adamc@196 145 in
adamc@196 146 r := xncs;
adamc@196 147 ((L'.TDatatype (dk, n, r), loc), sm)
adamc@196 148 end)
adamc@196 149 | L.TFfi mx => ((L'.TFfi mx, loc), sm)
adamc@288 150 | L.TOption t =>
adamc@288 151 let
adamc@288 152 val (t, sm) = cify dtmap (t, sm)
adamc@288 153 in
adamc@288 154 ((L'.TOption t, loc), sm)
adamc@288 155 end
adamc@757 156 | L.TList t =>
adamc@757 157 let
adamc@757 158 val (t', sm) = cify dtmap (t, sm)
adamc@757 159 val (sm, si) = Sm.findList (sm, t, t')
adamc@757 160 in
adamc@757 161 ((L'.TList (t', si), loc), sm)
adamc@757 162 end
adamc@577 163 | L.TSource => ((L'.TFfi ("Basis", "int"), loc), sm)
adamc@568 164 | L.TSignal _ => raise Fail "Cjrize: TSignal remains"
adamc@196 165 in
adamc@196 166 cify IM.empty x
adamc@196 167 end
adamc@29 168
adamc@109 169 val dummye = (L'.EPrim (Prim.Int 0), ErrorMsg.dummySpan)
adamc@109 170
adamc@186 171 fun cifyPatCon (pc, sm) =
adamc@181 172 case pc of
adamc@186 173 L.PConVar n => (L'.PConVar n, sm)
adamc@186 174 | L.PConFfi {mod = m, datatyp, con, arg} =>
adamc@186 175 let
adamc@186 176 val (arg, sm) =
adamc@186 177 case arg of
adamc@186 178 NONE => (NONE, sm)
adamc@186 179 | SOME t =>
adamc@186 180 let
adamc@186 181 val (t, sm) = cifyTyp (t, sm)
adamc@186 182 in
adamc@186 183 (SOME t, sm)
adamc@186 184 end
adamc@186 185 in
adamc@186 186 (L'.PConFfi {mod = m, datatyp = datatyp, con = con, arg = arg}, sm)
adamc@186 187 end
adamc@181 188
adamc@182 189 fun cifyPat ((p, loc), sm) =
adamc@181 190 case p of
adamc@182 191 L.PWild => ((L'.PWild, loc), sm)
adamc@182 192 | L.PVar (x, t) =>
adamc@182 193 let
adamc@182 194 val (t, sm) = cifyTyp (t, sm)
adamc@182 195 in
adamc@182 196 ((L'.PVar (x, t), loc), sm)
adamc@182 197 end
adamc@182 198 | L.PPrim p => ((L'.PPrim p, loc), sm)
adamc@188 199 | L.PCon (dk, pc, NONE) =>
adamc@186 200 let
adamc@186 201 val (pc, sm) = cifyPatCon (pc, sm)
adamc@186 202 in
adamc@188 203 ((L'.PCon (dk, pc, NONE), loc), sm)
adamc@186 204 end
adamc@188 205 | L.PCon (dk, pc, SOME p) =>
adamc@182 206 let
adamc@186 207 val (pc, sm) = cifyPatCon (pc, sm)
adamc@182 208 val (p, sm) = cifyPat (p, sm)
adamc@182 209 in
adamc@188 210 ((L'.PCon (dk, pc, SOME p), loc), sm)
adamc@182 211 end
adamc@182 212 | L.PRecord xps =>
adamc@182 213 let
adamc@182 214 val (xps, sm) = ListUtil.foldlMap (fn ((x, p, t), sm) =>
adamc@182 215 let
adamc@182 216 val (p, sm) = cifyPat (p, sm)
adamc@182 217 val (t, sm) = cifyTyp (t, sm)
adamc@182 218 in
adamc@182 219 ((x, p, t), sm)
adamc@182 220 end) sm xps
adamc@182 221 in
adamc@182 222 ((L'.PRecord xps, loc), sm)
adamc@182 223 end
adamc@288 224 | L.PNone t =>
adamc@288 225 let
adamc@288 226 val (t, sm) = cifyTyp (t, sm)
adamc@288 227 in
adamc@288 228 ((L'.PNone t, loc), sm)
adamc@288 229 end
adamc@288 230 | L.PSome (t, p) =>
adamc@288 231 let
adamc@288 232 val (t, sm) = cifyTyp (t, sm)
adamc@288 233 val (p, sm) = cifyPat (p, sm)
adamc@288 234 in
adamc@288 235 ((L'.PSome (t, p), loc), sm)
adamc@288 236 end
adamc@288 237
adamc@280 238 fun cifyExp (eAll as (e, loc), sm) =
adamc@29 239 case e of
adamc@29 240 L.EPrim p => ((L'.EPrim p, loc), sm)
adamc@29 241 | L.ERel n => ((L'.ERel n, loc), sm)
adamc@29 242 | L.ENamed n => ((L'.ENamed n, loc), sm)
adamc@188 243 | L.ECon (dk, pc, eo) =>
adamc@181 244 let
adamc@181 245 val (eo, sm) =
adamc@181 246 case eo of
adamc@181 247 NONE => (NONE, sm)
adamc@181 248 | SOME e =>
adamc@181 249 let
adamc@181 250 val (e, sm) = cifyExp (e, sm)
adamc@181 251 in
adamc@181 252 (SOME e, sm)
adamc@181 253 end
adamc@186 254 val (pc, sm) = cifyPatCon (pc, sm)
adamc@181 255 in
adamc@188 256 ((L'.ECon (dk, pc, eo), loc), sm)
adamc@181 257 end
adamc@297 258 | L.ENone t =>
adamc@297 259 let
adamc@297 260 val (t, sm) = cifyTyp (t, sm)
adamc@297 261 in
adamc@297 262 ((L'.ENone t, loc), sm)
adamc@297 263 end
adamc@290 264 | L.ESome (t, e) =>
adamc@290 265 let
adamc@290 266 val (t, sm) = cifyTyp (t, sm)
adamc@290 267 val (e, sm) = cifyExp (e, sm)
adamc@290 268 in
adamc@290 269 ((L'.ESome (t, e), loc), sm)
adamc@290 270 end
adamc@53 271 | L.EFfi mx => ((L'.EFfi mx, loc), sm)
adamc@53 272 | L.EFfiApp (m, x, es) =>
adamc@53 273 let
adamc@53 274 val (es, sm) = ListUtil.foldlMap cifyExp sm es
adamc@53 275 in
adamc@53 276 ((L'.EFfiApp (m, x, es), loc), sm)
adamc@53 277 end
adamc@29 278 | L.EApp (e1, e2) =>
adamc@29 279 let
adamc@316 280 fun unravel (e, args) =
adamc@316 281 case e of
adamc@316 282 (L.EApp (e1, e2), _) => unravel (e1, e2 :: args)
adamc@316 283 | _ => (e, args)
adamc@316 284
adamc@316 285 val (f, es) = unravel (e1, [e2])
adamc@316 286
adamc@316 287 val (f, sm) = cifyExp (f, sm)
adamc@316 288 val (es, sm) = ListUtil.foldlMap cifyExp sm es
adamc@29 289 in
adamc@316 290 ((L'.EApp (f, es), loc), sm)
adamc@29 291 end
adamc@109 292 | L.EAbs _ => (ErrorMsg.errorAt loc "Anonymous function remains at code generation";
adamc@280 293 Print.prefaces' [("Function", MonoPrint.p_exp MonoEnv.empty eAll)];
adamc@109 294 (dummye, sm))
adamc@29 295
adamc@387 296 | L.EUnop (s, e1) =>
adamc@387 297 let
adamc@387 298 val (e1, sm) = cifyExp (e1, sm)
adamc@387 299 in
adamc@387 300 ((L'.EUnop (s, e1), loc), sm)
adamc@387 301 end
adamc@387 302 | L.EBinop (s, e1, e2) =>
adamc@387 303 let
adamc@387 304 val (e1, sm) = cifyExp (e1, sm)
adamc@387 305 val (e2, sm) = cifyExp (e2, sm)
adamc@387 306 in
adamc@387 307 ((L'.EBinop (s, e1, e2), loc), sm)
adamc@387 308 end
adamc@387 309
adamc@29 310 | L.ERecord xes =>
adamc@29 311 let
adamc@29 312 val old_xts = map (fn (x, _, t) => (x, t)) xes
adamc@29 313
adamc@29 314 val (xets, sm) = ListUtil.foldlMap (fn ((x, e, t), sm) =>
adamc@29 315 let
adamc@29 316 val (e, sm) = cifyExp (e, sm)
adamc@29 317 val (t, sm) = cifyTyp (t, sm)
adamc@29 318 in
adamc@29 319 ((x, e, t), sm)
adamc@29 320 end)
adamc@29 321 sm xes
adamc@29 322
adamc@29 323 val (sm, si) = Sm.find (sm, old_xts, map (fn (x, _, t) => (x, t)) xets)
adamc@29 324
adamc@29 325 val xes = map (fn (x, e, _) => (x, e)) xets
adamc@29 326 val xes = ListMergeSort.sort (fn ((x1, _), (x2, _)) => String.compare (x1, x2) = GREATER) xes
adamc@29 327 in
adamc@29 328 ((L'.ERecord (si, xes), loc), sm)
adamc@29 329 end
adamc@29 330 | L.EField (e, x) =>
adamc@29 331 let
adamc@29 332 val (e, sm) = cifyExp (e, sm)
adamc@29 333 in
adamc@29 334 ((L'.EField (e, x), loc), sm)
adamc@29 335 end
adamc@29 336
adamc@182 337 | L.ECase (e, pes, {disc, result}) =>
adamc@181 338 let
adamc@181 339 val (e, sm) = cifyExp (e, sm)
adamc@181 340 val (pes, sm) = ListUtil.foldlMap
adamc@181 341 (fn ((p, e), sm) =>
adamc@181 342 let
adamc@181 343 val (e, sm) = cifyExp (e, sm)
adamc@182 344 val (p, sm) = cifyPat (p, sm)
adamc@181 345 in
adamc@182 346 ((p, e), sm)
adamc@181 347 end) sm pes
adamc@182 348 val (disc, sm) = cifyTyp (disc, sm)
adamc@182 349 val (result, sm) = cifyTyp (result, sm)
adamc@181 350 in
adamc@182 351 ((L'.ECase (e, pes, {disc = disc, result = result}), loc), sm)
adamc@181 352 end
adamc@178 353
adamc@283 354 | L.EError (e, t) =>
adamc@283 355 let
adamc@283 356 val (e, sm) = cifyExp (e, sm)
adamc@283 357 val (t, sm) = cifyTyp (t, sm)
adamc@283 358 in
adamc@283 359 ((L'.EError (e, t), loc), sm)
adamc@283 360 end
adamc@741 361 | L.EReturnBlob {blob, mimeType, t} =>
adamc@741 362 let
adamc@741 363 val (blob, sm) = cifyExp (blob, sm)
adamc@741 364 val (mimeType, sm) = cifyExp (mimeType, sm)
adamc@741 365 val (t, sm) = cifyTyp (t, sm)
adamc@741 366 in
adamc@741 367 ((L'.EReturnBlob {blob = blob, mimeType = mimeType, t = t}, loc), sm)
adamc@741 368 end
adamc@283 369
adamc@180 370 | L.EStrcat (e1, e2) =>
adamc@180 371 let
adamc@180 372 val (e1, sm) = cifyExp (e1, sm)
adamc@180 373 val (e2, sm) = cifyExp (e2, sm)
adamc@180 374 in
adamc@180 375 ((L'.EFfiApp ("Basis", "strcat", [e1, e2]), loc), sm)
adamc@180 376 end
adamc@102 377
adamc@102 378 | L.EWrite e =>
adamc@102 379 let
adamc@102 380 val (e, sm) = cifyExp (e, sm)
adamc@102 381 in
adamc@102 382 ((L'.EWrite e, loc), sm)
adamc@102 383 end
adamc@102 384
adamc@106 385 | L.ESeq (e1, e2) =>
adamc@106 386 let
adamc@106 387 val (e1, sm) = cifyExp (e1, sm)
adamc@106 388 val (e2, sm) = cifyExp (e2, sm)
adamc@106 389 in
adamc@106 390 ((L'.ESeq (e1, e2), loc), sm)
adamc@106 391 end
adamc@106 392
adamc@269 393 | L.ELet (x, t, e1, e2) =>
adamc@269 394 let
adamc@269 395 val (t, sm) = cifyTyp (t, sm)
adamc@269 396 val (e1, sm) = cifyExp (e1, sm)
adamc@269 397 val (e2, sm) = cifyExp (e2, sm)
adamc@269 398 in
adamc@269 399 ((L'.ELet (x, t, e1, e2), loc), sm)
adamc@269 400 end
adamc@251 401
adamc@111 402 | L.EClosure _ => (ErrorMsg.errorAt loc "Nested closure remains in code generation";
adamc@111 403 (dummye, sm))
adamc@111 404
adamc@269 405 | L.EQuery {exps, tables, state, query, body, initial} =>
adamc@269 406 let
adamc@269 407 val (exps', sm) = ListUtil.foldlMap (fn ((x, t), sm) =>
adamc@269 408 let
adamc@269 409 val (t, sm) = cifyTyp (t, sm)
adamc@269 410 in
adamc@269 411 ((x, t), sm)
adamc@269 412 end) sm exps
adamc@269 413 val (tables', sm) = ListUtil.foldlMap (fn ((x, xts), sm) =>
adamc@269 414 let
adamc@269 415 val (xts, sm) = ListUtil.foldlMap
adamc@269 416 (fn ((x, t), sm) =>
adamc@269 417 let
adamc@269 418 val (t, sm) = cifyTyp (t, sm)
adamc@269 419 in
adamc@269 420 ((x, t), sm)
adamc@269 421 end) sm xts
adamc@269 422 in
adamc@269 423 ((x, xts), sm)
adamc@269 424 end) sm tables
adamc@269 425
adamc@269 426 val row = exps @ map (fn (x, xts) => (x, (L.TRecord xts, loc))) tables
adamc@269 427 val row = ListMergeSort.sort (fn ((x, _), (y, _)) => String.compare (x, y) = GREATER) row
adamc@269 428
adamc@269 429 val (tableRows, sm) = ListUtil.foldlMap (fn (((x, xts), (_, xts')), sm) =>
adamc@269 430 let
adamc@269 431 val (sm, rnum) = Sm.find (sm, xts, xts')
adamc@269 432 in
adamc@269 433 ((x, rnum), sm)
adamc@269 434 end)
adamc@269 435 sm (ListPair.zip (tables, tables'))
adamc@269 436 val row' = exps' @ map (fn (x, n) => (x, (L'.TRecord n, loc))) tableRows
adamc@269 437 val row' = ListMergeSort.sort (fn ((x, _), (y, _)) => String.compare (x, y) = GREATER) row'
adamc@269 438
adamc@269 439 val (sm, rnum) = Sm.find (sm, row, row')
adamc@269 440
adamc@269 441 val (state, sm) = cifyTyp (state, sm)
adamc@269 442 val (query, sm) = cifyExp (query, sm)
adamc@269 443 val (body, sm) = cifyExp (body, sm)
adamc@269 444 val (initial, sm) = cifyExp (initial, sm)
adamc@269 445 in
adamc@269 446 ((L'.EQuery {exps = exps', tables = tables', rnum = rnum, state = state,
adamc@282 447 query = query, body = body, initial = initial, prepared = NONE}, loc), sm)
adamc@269 448 end
adamc@269 449
adamc@307 450 | L.EDml e =>
adamc@307 451 let
adamc@307 452 val (e, sm) = cifyExp (e, sm)
adamc@307 453 in
adamc@307 454 ((L'.EDml {dml = e, prepared = NONE}, loc), sm)
adamc@307 455 end
adamc@307 456
adamc@338 457 | L.ENextval e =>
adamc@338 458 let
adamc@338 459 val (e, sm) = cifyExp (e, sm)
adamc@338 460 in
adamc@338 461 ((L'.ENextval {seq = e, prepared = NONE}, loc), sm)
adamc@338 462 end
adamc@338 463
adamc@463 464 | L.EUnurlify (e, t) =>
adamc@463 465 let
adamc@463 466 val (e, sm) = cifyExp (e, sm)
adamc@463 467 val (t, sm) = cifyTyp (t, sm)
adamc@463 468 in
adamc@463 469 ((L'.EUnurlify (e, t), loc), sm)
adamc@463 470 end
adamc@252 471
adamc@568 472 | L.EJavaScript _ => raise Fail "Cjrize: EJavaScript remains"
adamc@578 473
adamc@568 474 | L.ESignalReturn _ => raise Fail "Cjrize: ESignalReturn remains"
adamc@572 475 | L.ESignalBind _ => raise Fail "Cjrize: ESignalBind remains"
adamc@574 476 | L.ESignalSource _ => raise Fail "Cjrize: ESignalSource remains"
adamc@566 477
adamc@608 478 | L.EServerCall _ => raise Fail "Cjrize EServerCall"
adamc@670 479 | L.ERecv _ => raise Fail "Cjrize ERecv"
adamc@695 480 | L.ESleep _ => raise Fail "Cjrize ESleep"
adamc@608 481
adamc@29 482 fun cifyDecl ((d, loc), sm) =
adamc@29 483 case d of
adamc@809 484 L.DDatatype dts =>
adamc@165 485 let
adamc@809 486 val (dts, sm) = ListUtil.foldlMap
adamc@809 487 (fn ((x, n, xncs), sm) =>
adamc@809 488 let
adamc@809 489 val dk = ElabUtil.classifyDatatype xncs
adamc@809 490 val (xncs, sm) = ListUtil.foldlMap (fn ((x, n, to), sm) =>
adamc@809 491 case to of
adamc@809 492 NONE => ((x, n, NONE), sm)
adamc@809 493 | SOME t =>
adamc@809 494 let
adamc@809 495 val (t, sm) = cifyTyp (t, sm)
adamc@809 496 in
adamc@809 497 ((x, n, SOME t), sm)
adamc@809 498 end) sm xncs
adamc@809 499 in
adamc@809 500 ((dk, x, n, xncs), sm)
adamc@809 501 end)
adamc@809 502 sm dts
adamc@165 503 in
adamc@809 504 (SOME (L'.DDatatype dts, loc), NONE, sm)
adamc@809 505 end
adamc@164 506
adamc@164 507 | L.DVal (x, n, t, e, _) =>
adamc@29 508 let
adamc@29 509 val (t, sm) = cifyTyp (t, sm)
adamc@109 510
adamc@109 511 val (d, sm) = case #1 t of
adamc@121 512 L'.TFun _ =>
adamc@121 513 let
adamc@121 514 fun unravel (tAll as (t, _), eAll as (e, _)) =
adamc@121 515 case (t, e) of
adamc@121 516 (L'.TFun (dom, ran), L.EAbs (ax, _, _, e)) =>
adamc@121 517 let
adamc@121 518 val (args, t, e) = unravel (ran, e)
adamc@121 519 in
adamc@121 520 ((ax, dom) :: args, t, e)
adamc@121 521 end
adamc@121 522 | (L'.TFun _, _) =>
adamc@121 523 (ErrorMsg.errorAt loc "Function isn't explicit at code generation";
adamc@121 524 ([], tAll, eAll))
adamc@121 525 | _ => ([], tAll, eAll)
adamc@121 526
adamc@121 527 val (args, ran, e) = unravel (t, e)
adamc@121 528 val (e, sm) = cifyExp (e, sm)
adamc@121 529 in
adamc@121 530 (L'.DFun (x, n, args, ran, e), sm)
adamc@121 531 end
adamc@121 532
adamc@109 533 | _ =>
adamc@109 534 let
adamc@109 535 val (e, sm) = cifyExp (e, sm)
adamc@109 536 in
adamc@109 537 (L'.DVal (x, n, t, e), sm)
adamc@109 538 end
adamc@29 539 in
adamc@109 540 (SOME (d, loc), NONE, sm)
adamc@29 541 end
adamc@129 542 | L.DValRec vis =>
adamc@129 543 let
adamc@129 544 val (vis, sm) = ListUtil.foldlMap
adamc@129 545 (fn ((x, n, t, e, _), sm) =>
adamc@129 546 let
adamc@129 547 val (t, sm) = cifyTyp (t, sm)
adamc@129 548
adamc@129 549 fun unravel (tAll as (t, _), eAll as (e, _)) =
adamc@129 550 case (t, e) of
adamc@129 551 (L'.TFun (dom, ran), L.EAbs (ax, _, _, e)) =>
adamc@129 552 let
adamc@129 553 val (args, t, e) = unravel (ran, e)
adamc@129 554 in
adamc@129 555 ((ax, dom) :: args, t, e)
adamc@129 556 end
adamc@129 557 | (L'.TFun _, _) =>
adamc@129 558 (ErrorMsg.errorAt loc "Function isn't explicit at code generation";
adamc@129 559 ([], tAll, eAll))
adamc@129 560 | _ => ([], tAll, eAll)
adamc@129 561
adamc@129 562 val (args, ran, e) = unravel (t, e)
adamc@129 563 val (e, sm) = cifyExp (e, sm)
adamc@129 564 in
adamc@129 565 ((x, n, args, ran, e), sm)
adamc@129 566 end)
adamc@129 567 sm vis
adamc@129 568 in
adamc@129 569 (SOME (L'.DFunRec vis, loc), NONE, sm)
adamc@129 570 end
adamc@129 571
adamc@609 572 | L.DExport (ek, s, n, ts, t) =>
adamc@120 573 let
adamc@120 574 val (ts, sm) = ListUtil.foldlMap cifyTyp sm ts
adamc@609 575 val (t, sm) = cifyTyp (t, sm)
adamc@120 576 in
adamc@693 577 (NONE, SOME (ek, "/" ^ s, n, ts, t, L'.ServerAndPullAndPush), sm)
adamc@120 578 end
adamc@29 579
adamc@707 580 | L.DTable (s, xts, pe, ce) =>
adamc@273 581 let
adamc@273 582 val (xts, sm) = ListUtil.foldlMap (fn ((x, t), sm) =>
adamc@273 583 let
adamc@273 584 val (t, sm) = cifyTyp (t, sm)
adamc@273 585 in
adamc@273 586 ((x, t), sm)
adamc@273 587 end) sm xts
adamc@704 588
adamc@704 589 fun flatten e =
adamc@704 590 case #1 e of
adamc@704 591 L.ERecord [] => []
adamc@704 592 | L.ERecord [(x, (L.EPrim (Prim.String v), _), _)] => [(x, v)]
adamc@704 593 | L.EStrcat (e1, e2) => flatten e1 @ flatten e2
adamc@704 594 | _ => (ErrorMsg.errorAt loc "Constraint has not been fully determined";
adamc@704 595 Print.prefaces "Undetermined constraint"
adamc@707 596 [("e", MonoPrint.p_exp MonoEnv.empty e)];
adamc@704 597 [])
adamc@707 598
adamc@707 599 val pe = case #1 pe of
adamc@707 600 L.EPrim (Prim.String s) => s
adamc@707 601 | _ => (ErrorMsg.errorAt loc "Primary key has not been fully determined";
adamc@707 602 Print.prefaces "Undetermined constraint"
adamc@707 603 [("e", MonoPrint.p_exp MonoEnv.empty pe)];
adamc@707 604 "")
adamc@273 605 in
adamc@707 606 (SOME (L'.DTable (s, xts, pe, flatten ce), loc), NONE, sm)
adamc@273 607 end
adamc@338 608 | L.DSequence s =>
adamc@338 609 (SOME (L'.DSequence s, loc), NONE, sm)
adamc@754 610 | L.DView (s, xts, e) =>
adamc@754 611 let
adamc@754 612 val (xts, sm) = ListUtil.foldlMap (fn ((x, t), sm) =>
adamc@754 613 let
adamc@754 614 val (t, sm) = cifyTyp (t, sm)
adamc@754 615 in
adamc@754 616 ((x, t), sm)
adamc@754 617 end) sm xts
adamc@754 618
adamc@754 619 fun flatten e =
adamc@754 620 case #1 e of
adamc@754 621 L.ERecord [] => []
adamc@754 622 | L.ERecord [(x, (L.EPrim (Prim.String v), _), _)] => [(x, v)]
adamc@754 623 | L.EStrcat (e1, e2) => flatten e1 @ flatten e2
adamc@754 624 | _ => (ErrorMsg.errorAt loc "Constraint has not been fully determined";
adamc@754 625 Print.prefaces "Undetermined constraint"
adamc@754 626 [("e", MonoPrint.p_exp MonoEnv.empty e)];
adamc@754 627 [])
adamc@754 628
adamc@754 629 val e = case #1 e of
adamc@754 630 L.EPrim (Prim.String s) => s
adamc@754 631 | _ => (ErrorMsg.errorAt loc "VIEW query has not been fully determined";
adamc@754 632 Print.prefaces "Undetermined VIEW query"
adamc@754 633 [("e", MonoPrint.p_exp MonoEnv.empty e)];
adamc@754 634 "")
adamc@754 635 in
adamc@754 636 (SOME (L'.DView (s, xts, e), loc), NONE, sm)
adamc@754 637 end
adamc@271 638 | L.DDatabase s => (SOME (L'.DDatabase s, loc), NONE, sm)
adamc@569 639 | L.DJavaScript s => (SOME (L'.DJavaScript s, loc), NONE, sm)
adamc@725 640 | L.DCookie args => (SOME (L'.DCookie args, loc), NONE, sm)
adamc@718 641 | L.DStyle args => (SOME (L'.DStyle args, loc), NONE, sm)
adamc@271 642
adamc@29 643 fun cjrize ds =
adamc@29 644 let
adamc@196 645 val (dsF, ds, ps, sm) = foldl (fn (d, (dsF, ds, ps, sm)) =>
adamc@196 646 let
adamc@196 647 val (dop, pop, sm) = cifyDecl (d, sm)
adamc@453 648
adamc@640 649 val dsF = case dop of
adamc@809 650 SOME (L'.DDatatype dts, loc) =>
adamc@809 651 map (fn (dk, x, n, _) =>
adamc@809 652 (L'.DDatatypeForward (dk, x, n), loc)) dts @ dsF
adamc@640 653 | _ => dsF
adamc@640 654
adamc@640 655 val dsF = map (fn v => (L'.DStruct v, ErrorMsg.dummySpan)) (Sm.declares sm)
adamc@640 656 @ dsF
adamc@640 657
adamc@196 658 val (dsF, ds) = case dop of
adamc@196 659 NONE => (dsF, ds)
adamc@453 660 | SOME (d as (L'.DDatatype _, loc)) =>
adamc@453 661 (d :: dsF, ds)
adamc@196 662 | SOME d => (dsF, d :: ds)
adamc@453 663
adamc@196 664 val ps = case pop of
adamc@196 665 NONE => ps
adamc@196 666 | SOME p => p :: ps
adamc@196 667 in
adamc@453 668 (dsF, ds, ps, Sm.clearDeclares sm)
adamc@196 669 end)
adamc@196 670 ([], [], [], Sm.empty) ds
adamc@29 671 in
adamc@453 672 (List.revAppend (dsF, rev ds),
adamc@101 673 ps)
adamc@29 674 end
adamc@29 675
adamc@29 676 end