adam@1294: (* Copyright (c) 2008-2010, Adam Chlipala
adamc@38:  * All rights reserved.
adamc@38:  *
adamc@38:  * Redistribution and use in source and binary forms, with or without
adamc@38:  * modification, are permitted provided that the following conditions are met:
adamc@38:  *
adamc@38:  * - Redistributions of source code must retain the above copyright notice,
adamc@38:  *   this list of conditions and the following disclaimer.
adamc@38:  * - Redistributions in binary form must reproduce the above copyright notice,
adamc@38:  *   this list of conditions and the following disclaimer in the documentation
adamc@38:  *   and/or other materials provided with the distribution.
adamc@38:  * - The names of contributors may not be used to endorse or promote products
adamc@38:  *   derived from this software without specific prior written permission.
adamc@38:  *
adamc@38:  * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS"
adamc@38:  * AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
adamc@38:  * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
adamc@38:  * ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT OWNER OR CONTRIBUTORS BE
adamc@38:  * LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR 
adamc@38:  * CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
adamc@38:  * SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
adamc@38:  * INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
adamc@38:  * CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
adamc@38:  * ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
adamc@38:  * POSSIBILITY OF SUCH DAMAGE.
adamc@38:  *)
adamc@38: 
adamc@38: structure Explify :> EXPLIFY = struct
adamc@38: 
adamc@38: structure EM = ErrorMsg
adamc@38: structure L = Elab
adamc@38: structure L' = Expl
adamc@38: 
adamc@38: fun explifyKind (k, loc) =
adamc@38:     case k of
adamc@38:         L.KType => (L'.KType, loc)
adamc@38:       | L.KArrow (k1, k2) => (L'.KArrow (explifyKind k1, explifyKind k2), loc)
adamc@38:       | L.KName => (L'.KName, loc)
adamc@38:       | L.KRecord k => (L'.KRecord (explifyKind k), loc)
adamc@38: 
adamc@87:       | L.KUnit => (L'.KUnit, loc)
adamc@213:       | L.KTuple ks => (L'.KTuple (map explifyKind ks), loc)
adamc@82: 
adamc@38:       | L.KError => raise Fail ("explifyKind: KError at " ^ EM.spanToString loc)
adam@1639:       | L.KUnif (_, _, ref (L.KKnown k)) => explifyKind k
adamc@38:       | L.KUnif _ => raise Fail ("explifyKind: KUnif at " ^ EM.spanToString loc)
adam@1639:       | L.KTupleUnif (loc, _, ref (L.KKnown k)) => explifyKind k
adam@1302:       | L.KTupleUnif _ => raise Fail ("explifyKind: KTupleUnif at " ^ EM.spanToString loc)
adamc@38: 
adamc@624:       | L.KRel n => (L'.KRel n, loc)
adamc@624:       | L.KFun (x, k) => (L'.KFun (x, explifyKind k), loc)
adamc@624: 
adamc@38: fun explifyCon (c, loc) =
adamc@38:     case c of
adamc@38:         L.TFun (t1, t2) => (L'.TFun (explifyCon t1, explifyCon t2), loc)
adamc@38:       | L.TCFun (_, x, k, t) => (L'.TCFun (x, explifyKind k, explifyCon t), loc)
adamc@628:       | L.TDisjoint (_, _, t) => explifyCon t
adamc@38:       | L.TRecord c => (L'.TRecord (explifyCon c), loc)
adamc@38: 
adamc@38:       | L.CRel n => (L'.CRel n, loc)
adamc@38:       | L.CNamed n => (L'.CNamed n, loc)
adamc@38:       | L.CModProj (m, ms, x) => (L'.CModProj (m, ms, x), loc)
adamc@38: 
adamc@38:       | L.CApp (c1, c2) => (L'.CApp (explifyCon c1, explifyCon c2), loc)
adamc@38:       | L.CAbs (x, k, c) => (L'.CAbs (x, explifyKind k, explifyCon c), loc)
adamc@38: 
adamc@38:       | L.CName s => (L'.CName s, loc)
adamc@38: 
adamc@38:       | L.CRecord (k, xcs) => (L'.CRecord (explifyKind k, map (fn (c1, c2) => (explifyCon c1, explifyCon c2)) xcs), loc)
adamc@38:       | L.CConcat (c1, c2) => (L'.CConcat (explifyCon c1, explifyCon c2), loc)
adamc@621:       | L.CMap (dom, ran) => (L'.CMap (explifyKind dom, explifyKind ran), loc)
adamc@38: 
adamc@87:       | L.CUnit => (L'.CUnit, loc)
adamc@82: 
adamc@213:       | L.CTuple cs => (L'.CTuple (map explifyCon cs), loc)
adamc@213:       | L.CProj (c, n) => (L'.CProj (explifyCon c, n), loc)
adamc@208: 
adamc@38:       | L.CError => raise Fail ("explifyCon: CError at " ^ EM.spanToString loc)
adam@1639:       | L.CUnif (nl, _, _, _, ref (L.Known c)) => explifyCon (ElabEnv.mliftConInCon nl c)
adamc@38:       | L.CUnif _ => raise Fail ("explifyCon: CUnif at " ^ EM.spanToString loc)
adamc@38: 
adamc@624:       | L.CKAbs (x, c) => (L'.CKAbs (x, explifyCon c), loc)
adamc@624:       | L.CKApp (c, k) => (L'.CKApp (explifyCon c, explifyKind k), loc)
adamc@624:       | L.TKFun (x, c) => (L'.TKFun (x, explifyCon c), loc)
adamc@624: 
adamc@176: fun explifyPatCon pc =
adamc@176:     case pc of
adamc@176:         L.PConVar n => L'.PConVar n
adamc@176:       | L.PConProj x => L'.PConProj x
adamc@176: 
adamc@176: fun explifyPat (p, loc) =
adamc@176:     case p of
adamc@176:         L.PWild => (L'.PWild, loc)
adamc@182:       | L.PVar (x, t) => (L'.PVar (x, explifyCon t), loc)
adamc@176:       | L.PPrim p => (L'.PPrim p, loc)
adamc@191:       | L.PCon (dk, pc, cs, po) => (L'.PCon (dk, explifyPatCon pc, map explifyCon cs, Option.map explifyPat po), loc)
adamc@182:       | L.PRecord xps => (L'.PRecord (map (fn (x, p, t) => (x, explifyPat p, explifyCon t)) xps), loc)
adamc@176: 
adamc@38: fun explifyExp (e, loc) =
adamc@38:     case e of
adamc@38:         L.EPrim p => (L'.EPrim p, loc)
adamc@38:       | L.ERel n => (L'.ERel n, loc)
adamc@38:       | L.ENamed n => (L'.ENamed n, loc)
adamc@38:       | L.EModProj (m, ms, x) => (L'.EModProj (m, ms, x), loc)
adamc@38:       | L.EApp (e1, e2) => (L'.EApp (explifyExp e1, explifyExp e2), loc)
adamc@38:       | L.EAbs (x, dom, ran, e1) => (L'.EAbs (x, explifyCon dom, explifyCon ran, explifyExp e1), loc)
adamc@38:       | L.ECApp (e1, c) => (L'.ECApp (explifyExp e1, explifyCon c), loc)
adamc@38:       | L.ECAbs (_, x, k, e1) => (L'.ECAbs (x, explifyKind k, explifyExp e1), loc)
adamc@38: 
adamc@38:       | L.ERecord xes => (L'.ERecord (map (fn (c, e, t) => (explifyCon c, explifyExp e, explifyCon t)) xes), loc)
adamc@38:       | L.EField (e1, c, {field, rest}) => (L'.EField (explifyExp e1, explifyCon c,
adamc@38:                                                        {field = explifyCon field, rest = explifyCon rest}), loc)
adamc@445:       | L.EConcat (e1, c1, e2, c2) => (L'.EConcat (explifyExp e1, explifyCon c1, explifyExp e2, explifyCon c2),
adamc@445:                                        loc)
adamc@149:       | L.ECut (e1, c, {field, rest}) => (L'.ECut (explifyExp e1, explifyCon c,
adamc@149:                                                      {field = explifyCon field, rest = explifyCon rest}), loc)
adamc@493:       | L.ECutMulti (e1, c, {rest}) => (L'.ECutMulti (explifyExp e1, explifyCon c,
adamc@493:                                                       {rest = explifyCon rest}), loc)
adamc@182:       | L.ECase (e, pes, {disc, result}) =>
adamc@182:         (L'.ECase (explifyExp e,
adamc@182:                    map (fn (p, e) => (explifyPat p, explifyExp e)) pes,
adamc@182:                    {disc = explifyCon disc, result = explifyCon result}), loc)
adamc@171: 
adamc@38:       | L.EError => raise Fail ("explifyExp: EError at " ^ EM.spanToString loc)
adamc@228:       | L.EUnif (ref (SOME e)) => explifyExp e
adamc@228:       | L.EUnif _ => raise Fail ("explifyExp: Undetermined EUnif at " ^ EM.spanToString loc)
adamc@38: 
adamc@825:       | L.ELet (des, e, t) =>
adamc@449:         foldr (fn ((de, loc), e) =>
adamc@449:                   case de of
adamc@449:                       L.EDValRec _ => raise Fail "explifyExp: Local 'val rec' remains"
adamc@825:                     | L.EDVal ((L.PVar (x, _), _), t', e') => (L'.ELet (x, explifyCon t', explifyExp e', e), loc)
adamc@825:                     | L.EDVal (p, t', e') => (L'.ECase (explifyExp e',
adamc@825:                                                         [(explifyPat p, e)],
adamc@825:                                                         {disc = explifyCon t', result = explifyCon t}), loc))
adamc@449:         (explifyExp e) des
adamc@449: 
adamc@624:       | L.EKAbs (x, e) => (L'.EKAbs (x, explifyExp e), loc)
adamc@624:       | L.EKApp (e, k) => (L'.EKApp (explifyExp e, explifyKind k), loc)
adamc@624: 
adamc@38: fun explifySgi (sgi, loc) =
adamc@38:     case sgi of
adamc@88:         L.SgiConAbs (x, n, k) => SOME (L'.SgiConAbs (x, n, explifyKind k), loc)
adamc@88:       | L.SgiCon (x, n, k, c) => SOME (L'.SgiCon (x, n, explifyKind k, explifyCon c), loc)
adamc@806:       | L.SgiDatatype dts => SOME (L'.SgiDatatype (map (fn (x, n, xs, xncs) =>
adamc@806:                                                            (x, n, xs,
adamc@806:                                                             map (fn (x, n, co) =>
adamc@806:                                                                     (x, n, Option.map explifyCon co)) xncs)) dts), loc)
adamc@191:       | L.SgiDatatypeImp (x, n, m1, ms, s, xs, xncs) =>
adamc@191:         SOME (L'.SgiDatatypeImp (x, n, m1, ms, s, xs, map (fn (x, n, co) =>
adamc@191:                                                               (x, n, Option.map explifyCon co)) xncs), loc)
adamc@88:       | L.SgiVal (x, n, c) => SOME (L'.SgiVal (x, n, explifyCon c), loc)
adam@2190:       | L.SgiStr (_, x, n, sgn) => SOME (L'.SgiStr (x, n, explifySgn sgn), loc)
adamc@88:       | L.SgiSgn (x, n, sgn) => SOME (L'.SgiSgn (x, n, explifySgn sgn), loc)
adamc@88:       | L.SgiConstraint _ => NONE
adamc@563:       | L.SgiClassAbs (x, n, k) => SOME (L'.SgiConAbs (x, n, (L'.KArrow (explifyKind k, (L'.KType, loc)), loc)), loc)
adamc@563:       | L.SgiClass (x, n, k, c) => SOME (L'.SgiCon (x, n, (L'.KArrow (explifyKind k, (L'.KType, loc)), loc),
adamc@563:                                                     explifyCon c), loc)
adamc@38: 
adamc@38: and explifySgn (sgn, loc) =
adamc@38:     case sgn of
adamc@88:         L.SgnConst sgis => (L'.SgnConst (List.mapPartial explifySgi sgis), loc)
adamc@38:       | L.SgnVar n => (L'.SgnVar n, loc)
adamc@45:       | L.SgnFun (m, n, dom, ran) => (L'.SgnFun (m, n, explifySgn dom, explifySgn ran), loc)
adam@1864:       | L.SgnWhere (sgn, ms, x, c) => (L'.SgnWhere (explifySgn sgn, ms, x, explifyCon c), loc)
adamc@64:       | L.SgnProj x => (L'.SgnProj x, loc)
adamc@38:       | L.SgnError => raise Fail ("explifySgn: SgnError at " ^ EM.spanToString loc)
adamc@38: 
adamc@38: fun explifyDecl (d, loc : EM.span) =
adamc@38:     case d of
adamc@88:         L.DCon (x, n, k, c) => SOME (L'.DCon (x, n, explifyKind k, explifyCon c), loc)
adamc@806:       | L.DDatatype dts => SOME (L'.DDatatype (map (fn (x, n, xs, xncs) =>
adamc@806:                                                        (x, n, xs,
adamc@806:                                                         map (fn (x, n, co) =>
adamc@806:                                                                 (x, n, Option.map explifyCon co)) xncs)) dts), loc)
adamc@191:       | L.DDatatypeImp (x, n, m1, ms, s, xs, xncs) =>
adamc@191:         SOME (L'.DDatatypeImp (x, n, m1, ms, s, xs,
adamc@191:                                map (fn (x, n, co) =>
adamc@191:                                        (x, n, Option.map explifyCon co)) xncs), loc)
adamc@88:       | L.DVal (x, n, t, e) => SOME (L'.DVal (x, n, explifyCon t, explifyExp e), loc)
adamc@124:       | L.DValRec vis => SOME (L'.DValRec (map (fn (x, n, t, e) => (x, n, explifyCon t, explifyExp e)) vis), loc)
adamc@38: 
adamc@88:       | L.DSgn (x, n, sgn) => SOME (L'.DSgn (x, n, explifySgn sgn), loc)
adamc@88:       | L.DStr (x, n, sgn, str) => SOME (L'.DStr (x, n, explifySgn sgn, explifyStr str), loc)
adamc@88:       | L.DFfiStr (x, n, sgn) => SOME (L'.DFfiStr (x, n, explifySgn sgn), loc)
adamc@88:       | L.DConstraint (c1, c2) => NONE
adamc@109:       | L.DExport (en, sgn, str) => SOME (L'.DExport (en, explifySgn sgn, explifyStr str), loc)
adamc@707:       | L.DTable (nt, x, n, c, pe, pc, ce, cc) =>
adamc@707:         SOME (L'.DTable (nt, x, n, explifyCon c,
adamc@707:                          explifyExp pe, explifyCon pc,
adamc@707:                          explifyExp ce, explifyCon cc), loc)
adamc@754:       | L.DView (nt, x, n, e, c) =>
adamc@754:         SOME (L'.DView (nt, x, n, explifyExp e, explifyCon c), loc)
adamc@338:       | L.DSequence (nt, x, n) => SOME (L'.DSequence (nt, x, n), loc)
adamc@271:       | L.DDatabase s => SOME (L'.DDatabase s, loc)
adamc@460:       | L.DCookie (nt, x, n, c) => SOME (L'.DCookie (nt, x, n, explifyCon c), loc)
adamc@720:       | L.DStyle (nt, x, n) => SOME (L'.DStyle (nt, x, n), loc)
adamc@1075:       | L.DTask (e1, e2) => SOME (L'.DTask (explifyExp e1, explifyExp e2), loc)
adamc@1199:       | L.DPolicy e1 => SOME (L'.DPolicy (explifyExp e1), loc)
adam@1294:       | L.DOnError v => SOME (L'.DOnError v, loc)
adam@2010:       | L.DFfi (x, n, modes, t) => SOME (L'.DFfi (x, n, modes, explifyCon t), loc)
adamc@38: 
adamc@38: and explifyStr (str, loc) =
adamc@38:     case str of
adamc@88:         L.StrConst ds => (L'.StrConst (List.mapPartial explifyDecl ds), loc)
adamc@38:       | L.StrVar n => (L'.StrVar n, loc)
adamc@38:       | L.StrProj (str, s) => (L'.StrProj (explifyStr str, s), loc)
adamc@45:       | L.StrFun (m, n, dom, ran, str) => (L'.StrFun (m, n, explifySgn dom, explifySgn ran, explifyStr str), loc)
adamc@45:       | L.StrApp (str1, str2) => (L'.StrApp (explifyStr str1, explifyStr str2), loc)
adamc@38:       | L.StrError => raise Fail ("explifyStr: StrError at " ^ EM.spanToString loc)
adamc@38: 
adamc@88: val explify = List.mapPartial explifyDecl
adamc@38: 
adamc@38: end