adamc@29: (* Copyright (c) 2008, Adam Chlipala
adamc@29:  * All rights reserved.
adamc@29:  *
adamc@29:  * Redistribution and use in source and binary forms, with or without
adamc@29:  * modification, are permitted provided that the following conditions are met:
adamc@29:  *
adamc@29:  * - Redistributions of source code must retain the above copyright notice,
adamc@29:  *   this list of conditions and the following disclaimer.
adamc@29:  * - Redistributions in binary form must reproduce the above copyright notice,
adamc@29:  *   this list of conditions and the following disclaimer in the documentation
adamc@29:  *   and/or other materials provided with the distribution.
adamc@29:  * - The names of contributors may not be used to endorse or promote products
adamc@29:  *   derived from this software without specific prior written permission.
adamc@29:  *
adamc@29:  * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS"
adamc@29:  * AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
adamc@29:  * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
adamc@29:  * ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT OWNER OR CONTRIBUTORS BE
adamc@29:  * LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR 
adamc@29:  * CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
adamc@29:  * SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
adamc@29:  * INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
adamc@29:  * CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
adamc@29:  * ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
adamc@29:  * POSSIBILITY OF SUCH DAMAGE.
adamc@29:  *)
adamc@29: 
adamc@29: structure Cjrize :> CJRIZE = struct
adamc@29: 
adamc@109: structure L = Mono
adamc@29: structure L' = Cjr
adamc@29: 
adamc@29: structure Sm :> sig
adamc@29:     type t
adamc@29: 
adamc@29:     val empty : t
adamc@29:     val find : t * (string * L.typ) list * (string * L'.typ) list -> t * int
adamc@29: 
adamc@29:     val declares : t -> (int * (string * L'.typ) list) list
adamc@29: end = struct
adamc@29: 
adamc@29: structure FM = BinaryMapFn(struct
adamc@29:                            type ord_key = L.typ
adamc@109:                            val compare = MonoUtil.Typ.compare
adamc@29:                            end)
adamc@29: 
adamc@29: type t = int * int FM.map * (int * (string * L'.typ) list) list
adamc@29: 
adamc@102: val empty : t = (1, FM.insert (FM.empty, (L.TRecord [], ErrorMsg.dummySpan), 0), [])
adamc@29: 
adamc@29: fun find ((n, m, ds), xts, xts') =
adamc@29:     let
adamc@29:         val t = (L.TRecord xts, ErrorMsg.dummySpan)
adamc@29:     in
adamc@29:         case FM.find (m, t) of
adamc@29:             NONE => ((n+1, FM.insert (m, t, n), (n, xts') :: ds), n)
adamc@29:           | SOME i => ((n, m, ds), i)
adamc@29:     end
adamc@29: 
adamc@29: fun declares (_, _, ds) = ds
adamc@29: 
adamc@29: end
adamc@29: 
adamc@29: fun cifyTyp ((t, loc), sm) =
adamc@29:     case t of
adamc@109:         L.TFun (t1, t2) =>
adamc@29:         let
adamc@29:             val (t1, sm) = cifyTyp (t1, sm)
adamc@29:             val (t2, sm) = cifyTyp (t2, sm)
adamc@29:         in
adamc@109:             ((L'.TFun (t1, t2), loc), sm)
adamc@29:         end
adamc@29:       | L.TRecord xts =>
adamc@29:         let
adamc@29:             val old_xts = xts
adamc@29:             val (xts, sm) = ListUtil.foldlMap (fn ((x, t), sm) =>
adamc@29:                                                   let
adamc@29:                                                       val (t, sm) = cifyTyp (t, sm)
adamc@29:                                                   in
adamc@29:                                                       ((x, t), sm)
adamc@29:                                                   end)
adamc@29:                                               sm xts
adamc@29:             val (sm, si) = Sm.find (sm, old_xts, xts)
adamc@29:         in
adamc@29:             ((L'.TRecord si, loc), sm)
adamc@29:         end
adamc@168:       | L.TDatatype (n, xncs) =>
adamc@168:         let
adamc@168:             val (xncs, sm) = ListUtil.foldlMap (fn ((x, n, to), sm) =>
adamc@168:                                                    case to of
adamc@168:                                                        NONE => ((x, n, NONE), sm)
adamc@168:                                                      | SOME t =>
adamc@168:                                                        let
adamc@168:                                                            val (t, sm) = cifyTyp (t, sm)
adamc@168:                                                        in
adamc@168:                                                            ((x, n, SOME t), sm)
adamc@168:                                                        end)
adamc@168:                              sm xncs
adamc@168:         in
adamc@168:             ((L'.TDatatype (n, xncs), loc), sm)
adamc@168:         end
adamc@53:       | L.TFfi mx => ((L'.TFfi mx, loc), sm)
adamc@29: 
adamc@109: val dummye = (L'.EPrim (Prim.Int 0), ErrorMsg.dummySpan)
adamc@109: 
adamc@29: fun cifyExp ((e, loc), sm) =
adamc@29:     case e of
adamc@29:         L.EPrim p => ((L'.EPrim p, loc), sm)
adamc@29:       | L.ERel n => ((L'.ERel n, loc), sm)
adamc@29:       | L.ENamed n => ((L'.ENamed n, loc), sm)
adamc@53:       | L.EFfi mx => ((L'.EFfi mx, loc), sm)
adamc@53:       | L.EFfiApp (m, x, es) =>
adamc@53:         let
adamc@53:             val (es, sm) = ListUtil.foldlMap cifyExp sm es
adamc@53:         in
adamc@53:             ((L'.EFfiApp (m, x, es), loc), sm)
adamc@53:         end
adamc@29:       | L.EApp (e1, e2) =>
adamc@29:         let
adamc@29:             val (e1, sm) = cifyExp (e1, sm)
adamc@29:             val (e2, sm) = cifyExp (e2, sm)
adamc@29:         in
adamc@29:             ((L'.EApp (e1, e2), loc), sm)
adamc@29:         end
adamc@109:       | L.EAbs _ => (ErrorMsg.errorAt loc "Anonymous function remains at code generation";
adamc@109:                      (dummye, sm))
adamc@29: 
adamc@29:       | L.ERecord xes =>
adamc@29:         let
adamc@29:             val old_xts = map (fn (x, _, t) => (x, t)) xes
adamc@29: 
adamc@29:             val (xets, sm) = ListUtil.foldlMap (fn ((x, e, t), sm) =>
adamc@29:                                                   let
adamc@29:                                                       val (e, sm) = cifyExp (e, sm)
adamc@29:                                                       val (t, sm) = cifyTyp (t, sm)
adamc@29:                                                   in
adamc@29:                                                       ((x, e, t), sm)
adamc@29:                                                   end)
adamc@29:                             sm xes
adamc@29: 
adamc@29:             val (sm, si) = Sm.find (sm, old_xts, map (fn (x, _, t) => (x, t)) xets)
adamc@29: 
adamc@29:             val xes = map (fn (x, e, _) => (x, e)) xets
adamc@29:             val xes = ListMergeSort.sort (fn ((x1, _), (x2, _)) => String.compare (x1, x2) = GREATER) xes
adamc@29:         in
adamc@29:             ((L'.ERecord (si, xes), loc), sm)
adamc@29:         end
adamc@29:       | L.EField (e, x) =>
adamc@29:         let
adamc@29:             val (e, sm) = cifyExp (e, sm)
adamc@29:         in
adamc@29:             ((L'.EField (e, x), loc), sm)
adamc@29:         end
adamc@29: 
adamc@102:       | L.EStrcat _ => raise Fail "Cjrize EStrcat"
adamc@102: 
adamc@102:       | L.EWrite e =>
adamc@102:         let
adamc@102:             val (e, sm) = cifyExp (e, sm)
adamc@102:         in
adamc@102:             ((L'.EWrite e, loc), sm)
adamc@102:         end
adamc@102: 
adamc@106:       | L.ESeq (e1, e2) =>
adamc@106:         let
adamc@106:             val (e1, sm) = cifyExp (e1, sm)
adamc@106:             val (e2, sm) = cifyExp (e2, sm)
adamc@106:         in
adamc@106:             ((L'.ESeq (e1, e2), loc), sm)
adamc@106:         end
adamc@106: 
adamc@111:       | L.EClosure _ => (ErrorMsg.errorAt loc "Nested closure remains in code generation";
adamc@111:                          (dummye, sm))
adamc@111: 
adamc@29: fun cifyDecl ((d, loc), sm) =
adamc@29:     case d of
adamc@165:         L.DDatatype (x, n, xncs) =>
adamc@165:         let
adamc@165:             val (xncs, sm) = ListUtil.foldlMap (fn ((x, n, to), sm) =>
adamc@165:                                                    case to of
adamc@165:                                                        NONE => ((x, n, NONE), sm)
adamc@165:                                                      | SOME t =>
adamc@165:                                                        let
adamc@165:                                                            val (t, sm) = cifyTyp (t, sm)
adamc@165:                                                        in
adamc@165:                                                            ((x, n, SOME t), sm)
adamc@165:                                                        end) sm xncs
adamc@165:         in
adamc@165:             (SOME (L'.DDatatype (x, n, xncs), loc), NONE, sm)
adamc@165:         end
adamc@164: 
adamc@164:       | L.DVal (x, n, t, e, _) =>
adamc@29:         let
adamc@29:             val (t, sm) = cifyTyp (t, sm)
adamc@109: 
adamc@109:             val (d, sm) = case #1 t of
adamc@121:                               L'.TFun _ =>
adamc@121:                               let
adamc@121:                                   fun unravel (tAll as (t, _), eAll as (e, _)) =
adamc@121:                                       case (t, e) of
adamc@121:                                           (L'.TFun (dom, ran), L.EAbs (ax, _, _, e)) =>
adamc@121:                                           let
adamc@121:                                               val (args, t, e) = unravel (ran, e)
adamc@121:                                           in
adamc@121:                                               ((ax, dom) :: args, t, e)
adamc@121:                                           end
adamc@121:                                         | (L'.TFun _, _) =>
adamc@121:                                           (ErrorMsg.errorAt loc "Function isn't explicit at code generation";
adamc@121:                                            ([], tAll, eAll))
adamc@121:                                         | _ => ([], tAll, eAll)
adamc@121: 
adamc@121:                                   val (args, ran, e) = unravel (t, e)
adamc@121:                                   val (e, sm) = cifyExp (e, sm)
adamc@121:                               in
adamc@121:                                   (L'.DFun (x, n, args, ran, e), sm)
adamc@121:                               end
adamc@121: 
adamc@109:                             | _ =>
adamc@109:                               let
adamc@109:                                   val (e, sm) = cifyExp (e, sm)
adamc@109:                               in
adamc@109:                                   (L'.DVal (x, n, t, e), sm)
adamc@109:                               end
adamc@29:         in
adamc@109:             (SOME (d, loc), NONE, sm)
adamc@29:         end
adamc@129:       | L.DValRec vis =>
adamc@129:         let
adamc@129:             val (vis, sm) = ListUtil.foldlMap
adamc@129:                             (fn ((x, n, t, e, _), sm) =>
adamc@129:                                 let                                    
adamc@129:                                     val (t, sm) = cifyTyp (t, sm)
adamc@129: 
adamc@129:                                     fun unravel (tAll as (t, _), eAll as (e, _)) =
adamc@129:                                         case (t, e) of
adamc@129:                                             (L'.TFun (dom, ran), L.EAbs (ax, _, _, e)) =>
adamc@129:                                             let
adamc@129:                                                 val (args, t, e) = unravel (ran, e)
adamc@129:                                             in
adamc@129:                                                 ((ax, dom) :: args, t, e)
adamc@129:                                             end
adamc@129:                                           | (L'.TFun _, _) =>
adamc@129:                                             (ErrorMsg.errorAt loc "Function isn't explicit at code generation";
adamc@129:                                              ([], tAll, eAll))
adamc@129:                                           | _ => ([], tAll, eAll)
adamc@129:                                                  
adamc@129:                                     val (args, ran, e) = unravel (t, e)
adamc@129:                                     val (e, sm) = cifyExp (e, sm)
adamc@129:                               in
adamc@129:                                   ((x, n, args, ran, e), sm)
adamc@129:                               end)
adamc@129:                             sm vis
adamc@129:         in
adamc@129:             (SOME (L'.DFunRec vis, loc), NONE, sm)
adamc@129:         end        
adamc@129: 
adamc@144:       | L.DExport (ek, s, n, ts) =>
adamc@120:         let
adamc@120:             val (ts, sm) = ListUtil.foldlMap cifyTyp sm ts
adamc@120:         in
adamc@144:             (NONE, SOME (ek, "/" ^ s, n, ts), sm)
adamc@120:         end
adamc@29: 
adamc@29: fun cjrize ds =
adamc@29:     let
adamc@101:         val (ds, ps, sm) = foldl (fn (d, (ds, ps, sm)) =>
adamc@101:                                      let
adamc@101:                                          val (dop, pop, sm) = cifyDecl (d, sm)
adamc@101:                                          val ds = case dop of
adamc@101:                                                       NONE => ds
adamc@101:                                                     | SOME d => d :: ds
adamc@101:                                          val ps = case pop of
adamc@101:                                                       NONE => ps
adamc@101:                                                     | SOME p => p :: ps 
adamc@101:                                      in
adamc@101:                                          (ds, ps, sm)
adamc@101:                                      end)
adamc@101:                            ([], [], Sm.empty) ds
adamc@29:     in
adamc@101:         (List.revAppend (map (fn v => (L'.DStruct v, ErrorMsg.dummySpan)) (Sm.declares sm),
adamc@101:                          rev ds),
adamc@101:          ps)
adamc@29:     end
adamc@29: 
adamc@29: end