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 CjrEnv :> CJR_ENV = struct adamc@29: adamc@29: open Cjr adamc@29: adamc@29: structure IM = IntBinaryMap adamc@29: adamc@29: adamc@29: exception UnboundRel of int adamc@29: exception UnboundNamed of int adamc@29: exception UnboundF of int adamc@101: exception UnboundStruct of int adamc@29: adamc@29: type env = { adamc@166: datatypes : (string * (string * int * typ option) list) IM.map, adamc@181: constructors : (string * typ option * int) IM.map, adamc@29: adamc@29: numRelE : int, adamc@29: relE : (string * typ) list, adamc@29: namedE : (string * typ) IM.map, adamc@29: adamc@101: structs : (string * typ) list IM.map adamc@29: } adamc@29: adamc@29: val empty = { adamc@166: datatypes = IM.empty, adamc@181: constructors = IM.empty, adamc@29: adamc@29: numRelE = 0, adamc@29: relE = [], adamc@29: namedE = IM.empty, adamc@29: adamc@101: structs = IM.empty adamc@29: } adamc@29: adamc@166: fun pushDatatype (env : env) x n xncs = adamc@166: {datatypes = IM.insert (#datatypes env, n, (x, xncs)), adamc@182: constructors = foldl (fn ((x, n', to), constructors) => adamc@182: IM.insert (constructors, n', (x, to, n))) adamc@181: (#constructors env) xncs, adamc@29: adamc@29: numRelE = #numRelE env, adamc@29: relE = #relE env, adamc@29: namedE = #namedE env, adamc@29: adamc@101: structs = #structs env} adamc@29: adamc@166: fun lookupDatatype (env : env) n = adamc@166: case IM.find (#datatypes env, n) of adamc@29: NONE => raise UnboundNamed n adamc@29: | SOME x => x adamc@29: adamc@181: fun lookupConstructor (env : env) n = adamc@181: case IM.find (#constructors env, n) of adamc@181: NONE => raise UnboundNamed n adamc@181: | SOME x => x adamc@181: adamc@29: fun pushERel (env : env) x t = adamc@166: {datatypes = #datatypes env, adamc@181: constructors = #constructors env, adamc@29: adamc@29: numRelE = #numRelE env + 1, adamc@29: relE = (x, t) :: #relE env, adamc@29: namedE = #namedE env, adamc@29: adamc@101: structs = #structs env} adamc@29: adamc@29: fun lookupERel (env : env) n = adamc@29: (List.nth (#relE env, n)) adamc@29: handle Subscript => raise UnboundRel n adamc@29: adamc@29: fun countERels (env : env) = #numRelE env adamc@29: adamc@29: fun listERels (env : env) = #relE env adamc@29: adamc@29: fun pushENamed (env : env) x n t = adamc@166: {datatypes = #datatypes env, adamc@181: constructors = #constructors env, adamc@29: adamc@29: numRelE = #numRelE env, adamc@29: relE = #relE env, adamc@29: namedE = IM.insert (#namedE env, n, (x, t)), adamc@29: adamc@101: structs = #structs env} adamc@29: adamc@29: fun lookupENamed (env : env) n = adamc@29: case IM.find (#namedE env, n) of adamc@29: NONE => raise UnboundNamed n adamc@29: | SOME x => x adamc@29: adamc@101: fun pushStruct (env : env) n xts = adamc@166: {datatypes = #datatypes env, adamc@181: constructors = #constructors env, adamc@101: adamc@101: numRelE = #numRelE env, adamc@101: relE = #relE env, adamc@101: namedE = #namedE env, adamc@101: adamc@101: structs = IM.insert (#structs env, n, xts)} adamc@101: adamc@101: fun lookupStruct (env : env) n = adamc@101: case IM.find (#structs env, n) of adamc@101: NONE => raise UnboundStruct n adamc@101: | SOME x => x adamc@101: adamc@188: fun classifyDatatype xncs = adamc@188: if List.all (fn (_, _, NONE) => true | _ => false) xncs then adamc@188: Enum adamc@188: else adamc@188: Default adamc@188: adamc@109: fun declBinds env (d, loc) = adamc@29: case d of adamc@188: DDatatype (_, x, n, xncs) => adamc@165: let adamc@166: val env = pushDatatype env x n xncs adamc@188: val dt = (TDatatype (classifyDatatype xncs, n, xncs), loc) adamc@165: in adamc@188: foldl (fn ((x', n', NONE), env) => pushENamed env x' n' dt adamc@188: | ((x', n', SOME t), env) => pushENamed env x' n' (TFun (t, dt), loc)) adamc@165: env xncs adamc@165: end adamc@165: | DStruct (n, xts) => pushStruct env n xts adamc@165: | DVal (x, n, t, _) => pushENamed env x n t adamc@121: | DFun (fx, n, args, ran, _) => adamc@121: let adamc@121: val t = foldl (fn ((_, arg), t) => (TFun (arg, t), loc)) ran args adamc@121: in adamc@121: pushENamed env fx n t adamc@121: end adamc@129: | DFunRec vis => adamc@129: foldl (fn ((fx, n, args, ran, _), env) => adamc@129: let adamc@129: val t = foldl (fn ((_, arg), t) => (TFun (arg, t), loc)) ran args adamc@129: in adamc@129: pushENamed env fx n t adamc@129: end) env vis adamc@165: adamc@29: adamc@29: end