adamc@2: (* Copyright (c) 2008, Adam Chlipala
adamc@2:  * All rights reserved.
adamc@2:  *
adamc@2:  * Redistribution and use in source and binary forms, with or without
adamc@2:  * modification, are permitted provided that the following conditions are met:
adamc@2:  *
adamc@2:  * - Redistributions of source code must retain the above copyright notice,
adamc@2:  *   this list of conditions and the following disclaimer.
adamc@2:  * - Redistributions in binary form must reproduce the above copyright notice,
adamc@2:  *   this list of conditions and the following disclaimer in the documentation
adamc@2:  *   and/or other materials provided with the distribution.
adamc@2:  * - The names of contributors may not be used to endorse or promote products
adamc@2:  *   derived from this software without specific prior written permission.
adamc@2:  *
adamc@2:  * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS"
adamc@2:  * AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
adamc@2:  * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
adamc@2:  * ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT OWNER OR CONTRIBUTORS BE
adamc@2:  * LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR 
adamc@2:  * CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
adamc@2:  * SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
adamc@2:  * INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
adamc@2:  * CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
adamc@2:  * ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
adamc@2:  * POSSIBILITY OF SUCH DAMAGE.
adamc@2:  *)
adamc@2: 
adamc@2: structure ElabEnv :> ELAB_ENV = struct
adamc@2: 
adamc@2: open Elab
adamc@2: 
adamc@2: structure IM = IntBinaryMap
adamc@2: structure SM = BinaryMapFn(struct
adamc@2:                            type ord_key = string
adamc@2:                            val compare = String.compare
adamc@2:                            end)
adamc@2: 
adamc@2: exception UnboundRel of int
adamc@2: exception UnboundNamed of int
adamc@2: 
adamc@9: datatype 'a var' =
adamc@9:          Rel' of int * 'a
adamc@9:        | Named' of int * 'a
adamc@2: 
adamc@9: datatype 'a var =
adamc@9:          NotBound
adamc@9:        | Rel of int * 'a
adamc@9:        | Named of int * 'a
adamc@2: 
adamc@2: type env = {
adamc@9:      renameC : kind var' SM.map,
adamc@2:      relC : (string * kind) list,
adamc@9:      namedC : (string * kind) IM.map,
adamc@9: 
adamc@9:      renameE : con var' SM.map,
adamc@9:      relE : (string * con) list,
adamc@9:      namedE : (string * con) IM.map
adamc@2: }
adamc@2: 
adamc@2: val namedCounter = ref 0
adamc@2: 
adamc@2: val empty = {
adamc@2:     renameC = SM.empty,
adamc@2:     relC = [],
adamc@9:     namedC = IM.empty,
adamc@9: 
adamc@9:     renameE = SM.empty,
adamc@9:     relE = [],
adamc@9:     namedE = IM.empty
adamc@2: }
adamc@2: 
adamc@2: fun pushCRel (env : env) x k =
adamc@2:     let
adamc@9:         val renameC = SM.map (fn Rel' (n, k) => Rel' (n+1, k)
adamc@2:                                | x => x) (#renameC env)
adamc@2:     in
adamc@9:         {renameC = SM.insert (renameC, x, Rel' (0, k)),
adamc@2:          relC = (x, k) :: #relC env,
adamc@9:          namedC = #namedC env,
adamc@9: 
adamc@9:          renameE = #renameE env,
adamc@9:          relE = #relE env,
adamc@9:          namedE = #namedE env}
adamc@2:     end
adamc@2: 
adamc@2: fun lookupCRel (env : env) n =
adamc@2:     (List.nth (#relC env, n))
adamc@2:     handle Subscript => raise UnboundRel n
adamc@2: 
adamc@5: fun pushCNamedAs (env : env) x n k =
adamc@9:     {renameC = SM.insert (#renameC env, x, Named' (n, k)),
adamc@5:      relC = #relC env,
adamc@9:      namedC = IM.insert (#namedC env, n, (x, k)),
adamc@9: 
adamc@9:      renameE = #renameE env,
adamc@9:      relE = #relE env,
adamc@9:      namedE = #namedE env}
adamc@5: 
adamc@5: fun pushCNamed env x k =
adamc@2:     let
adamc@2:         val n = !namedCounter
adamc@2:     in
adamc@2:         namedCounter := n + 1;
adamc@5:         (pushCNamedAs env x n k, n)
adamc@2:     end
adamc@2: 
adamc@2: fun lookupCNamed (env : env) n =
adamc@2:     case IM.find (#namedC env, n) of
adamc@2:         NONE => raise UnboundNamed n
adamc@2:       | SOME x => x
adamc@2: 
adamc@2: fun lookupC (env : env) x =
adamc@2:     case SM.find (#renameC env, x) of
adamc@9:         NONE => NotBound
adamc@9:       | SOME (Rel' x) => Rel x
adamc@9:       | SOME (Named' x) => Named x
adamc@9: 
adamc@9: fun pushERel (env : env) x t =
adamc@9:     let
adamc@9:         val renameE = SM.map (fn Rel' (n, t) => Rel' (n+1, t)
adamc@9:                                | x => x) (#renameE env)
adamc@9:     in
adamc@9:         {renameC = #renameC env,
adamc@9:          relC = #relC env,
adamc@9:          namedC = #namedC env,
adamc@9: 
adamc@9:          renameE = SM.insert (renameE, x, Rel' (0, t)),
adamc@9:          relE = (x, t) :: #relE env,
adamc@9:          namedE = #namedE env}
adamc@9:     end
adamc@9: 
adamc@9: fun lookupERel (env : env) n =
adamc@9:     (List.nth (#relE env, n))
adamc@9:     handle Subscript => raise UnboundRel n
adamc@9: 
adamc@9: fun pushENamedAs (env : env) x n t =
adamc@9:     {renameC = #renameC env,
adamc@9:      relC = #relC env,
adamc@9:      namedC = #namedC env,
adamc@9: 
adamc@9:      renameE = SM.insert (#renameE env, x, Named' (n, t)),
adamc@9:      relE = #relE env,
adamc@9:      namedE = IM.insert (#namedE env, n, (x, t))}
adamc@9: 
adamc@9: fun pushENamed env x t =
adamc@9:     let
adamc@9:         val n = !namedCounter
adamc@9:     in
adamc@9:         namedCounter := n + 1;
adamc@9:         (pushENamedAs env x n t, n)
adamc@9:     end
adamc@9: 
adamc@9: fun lookupENamed (env : env) n =
adamc@9:     case IM.find (#namedE env, n) of
adamc@9:         NONE => raise UnboundNamed n
adamc@9:       | SOME x => x
adamc@9: 
adamc@9: fun lookupE (env : env) x =
adamc@9:     case SM.find (#renameE env, x) of
adamc@9:         NONE => NotBound
adamc@9:       | SOME (Rel' x) => Rel x
adamc@9:       | SOME (Named' x) => Named x
adamc@2: 
adamc@2: end