adamc@16: (* Copyright (c) 2008, Adam Chlipala
adamc@16:  * All rights reserved.
adamc@16:  *
adamc@16:  * Redistribution and use in source and binary forms, with or without
adamc@16:  * modification, are permitted provided that the following conditions are met:
adamc@16:  *
adamc@16:  * - Redistributions of source code must retain the above copyright notice,
adamc@16:  *   this list of conditions and the following disclaimer.
adamc@16:  * - Redistributions in binary form must reproduce the above copyright notice,
adamc@16:  *   this list of conditions and the following disclaimer in the documentation
adamc@16:  *   and/or other materials provided with the distribution.
adamc@16:  * - The names of contributors may not be used to endorse or promote products
adamc@16:  *   derived from this software without specific prior written permission.
adamc@16:  *
adamc@16:  * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS"
adamc@16:  * AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
adamc@16:  * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
adamc@16:  * ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT OWNER OR CONTRIBUTORS BE
adamc@16:  * LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR 
adamc@16:  * CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
adamc@16:  * SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
adamc@16:  * INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
adamc@16:  * CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
adamc@16:  * ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
adamc@16:  * POSSIBILITY OF SUCH DAMAGE.
adamc@16:  *)
adamc@16: 
adamc@16: structure Corify :> CORIFY = struct
adamc@16: 
adamc@16: structure EM = ErrorMsg
adamc@16: structure L = Elab
adamc@16: structure L' = Core
adamc@16: 
adamc@16: fun corifyKind (k, loc) =
adamc@16:     case k of
adamc@16:         L.KType => (L'.KType, loc)
adamc@16:       | L.KArrow (k1, k2) => (L'.KArrow (corifyKind k1, corifyKind k2), loc)
adamc@16:       | L.KName => (L'.KName, loc)
adamc@16:       | L.KRecord k => (L'.KRecord (corifyKind k), loc)
adamc@16: 
adamc@16:       | L.KError => raise Fail ("corifyKind: KError at " ^ EM.spanToString loc)
adamc@16:       | L.KUnif (_, ref (SOME k)) => corifyKind k
adamc@16:       | L.KUnif _ => raise Fail ("corifyKind: KUnif at " ^ EM.spanToString loc)
adamc@16: 
adamc@16: fun corifyCon (c, loc) =
adamc@16:     case c of
adamc@16:         L.TFun (t1, t2) => (L'.TFun (corifyCon t1, corifyCon t2), loc)
adamc@16:       | L.TCFun (_, x, k, t) => (L'.TCFun (x, corifyKind k, corifyCon t), loc)
adamc@16:       | L.TRecord c => (L'.TRecord (corifyCon c), loc)
adamc@16: 
adamc@16:       | L.CRel n => (L'.CRel n, loc)
adamc@16:       | L.CNamed n => (L'.CNamed n, loc)
adamc@34:       | L.CModProj _ => raise Fail "Corify CModProj"
adamc@34: 
adamc@16:       | L.CApp (c1, c2) => (L'.CApp (corifyCon c1, corifyCon c2), loc)
adamc@16:       | L.CAbs (x, k, c) => (L'.CAbs (x, corifyKind k, corifyCon c), loc)
adamc@16: 
adamc@16:       | L.CName s => (L'.CName s, loc)
adamc@16: 
adamc@16:       | L.CRecord (k, xcs) => (L'.CRecord (corifyKind k, map (fn (c1, c2) => (corifyCon c1, corifyCon c2)) xcs), loc)
adamc@16:       | L.CConcat (c1, c2) => (L'.CConcat (corifyCon c1, corifyCon c2), loc)
adamc@16: 
adamc@16:       | L.CError => raise Fail ("corifyCon: CError at " ^ EM.spanToString loc)
adamc@16:       | L.CUnif (_, _, ref (SOME c)) => corifyCon c
adamc@16:       | L.CUnif _ => raise Fail ("corifyCon: CUnif at " ^ EM.spanToString loc)
adamc@16: 
adamc@16: fun corifyExp (e, loc) =
adamc@16:     case e of
adamc@16:         L.EPrim p => (L'.EPrim p, loc)
adamc@16:       | L.ERel n => (L'.ERel n, loc)
adamc@16:       | L.ENamed n => (L'.ENamed n, loc)
adamc@34:       | L.EModProj _ => raise Fail "Corify EModProj"
adamc@16:       | L.EApp (e1, e2) => (L'.EApp (corifyExp e1, corifyExp e2), loc)
adamc@26:       | L.EAbs (x, dom, ran, e1) => (L'.EAbs (x, corifyCon dom, corifyCon ran, corifyExp e1), loc)
adamc@16:       | L.ECApp (e1, c) => (L'.ECApp (corifyExp e1, corifyCon c), loc)
adamc@16:       | L.ECAbs (_, x, k, e1) => (L'.ECAbs (x, corifyKind k, corifyExp e1), loc)
adamc@16: 
adamc@29:       | L.ERecord xes => (L'.ERecord (map (fn (c, e, t) => (corifyCon c, corifyExp e, corifyCon t)) xes), loc)
adamc@16:       | L.EField (e1, c, {field, rest}) => (L'.EField (corifyExp e1, corifyCon c,
adamc@16:                                                        {field = corifyCon field, rest = corifyCon rest}), loc)
adamc@16: 
adamc@16:       | L.EError => raise Fail ("corifyExp: EError at " ^ EM.spanToString loc)
adamc@16: 
adamc@16: fun corifyDecl (d, loc : EM.span) =
adamc@16:     case d of
adamc@16:         L.DCon (x, n, k, c) => (L'.DCon (x, n, corifyKind k, corifyCon c), loc)
adamc@16:       | L.DVal (x, n, t, e) => (L'.DVal (x, n, corifyCon t, corifyExp e), loc)
adamc@16: 
adamc@31:       | L.DSgn _ => raise Fail "Not ready to corify signature"
adamc@31:       | L.DStr _ => raise Fail "Not ready to corify structure"
adamc@31: 
adamc@16: val corify = map corifyDecl
adamc@16: 
adamc@16: end